2027 Summer Intern MS PhD Software Engineer ML Frameworks and Efficiency
1 hour agoInternSalary: 146K - 177KMountain View, California, United StatesHybridInternshipEngineeringJobs by Waymo
WaymoVisit Waymo website
Waymo is an Alphabet company operating a public, fully autonomous ride-hailing service and developing the AI-powered Waymo Driver autonomous-driving system.
0 current maintainers0 active leads
Mountain View, United States
Funding history
About Waymo
Waymo develops and deploys the Waymo Driver, an autonomous-driving technology using AI and a sensor suite, and operates its ride-hailing service for public riders.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will support machine learning frameworks and efficiency initiatives for LPM model training. You will understand the current model’s problem space, extend reinforcement learning infrastructure, profile and address training bottlenecks, and evaluate and integrate promising reinforcement learning technologies.
Requirements
- Currently pursuing a Master's or PhD degree in Computer Science, Electrical Engineering, Machine Learning, Robotics, or a related technical field
- Fundamental understanding of machine learning, deep learning, and reinforcement learning
- Familiarity with JAX, Flax, PyTorch, or TensorFlow
- Strong Python software development skills
Responsibilities
- Understand the current LPM model and its requirements
- Improve and extend reinforcement learning infrastructure to support LPM model training
- Profile training performance and improve training efficiency
- Identify and integrate promising reinforcement learning technologies
Benefits
- Housing or relocation bonus, if applicable
- Medical, dental, and vision insurance
- Intern events and networking opportunities
- Free breakfast, lunch, dinner, and snacks
- Free access to Google shuttles
- Onsite gym
