2027 Summer Intern, MS/PhD, Machine Learning, Driver Refinement Foundations
About this role
The Driver Refinement Foundations subteam builds fundamental platforms like the Data Flywheel and Reasoning platform to support driving behavior improvements across the Planner organization. Graduate students in this internship will mine and curate dataset samples, leverage large language models and world foundation models, and optimize multimodal generative models using reinforcement learning from feedback. Candidates who thrive in this environment possess strong programming abilities and a background in model evaluation within an autonomous driving context.
Waymo operates an autonomous driving technology company focused on developing the Waymo Driver to improve mobility and safety. Interns partner with industry leaders on impactful projects while expanding their technical skill sets and contributing to the company culture.
What you will do
- Mine and curate dataset samples from a large corpus of driving data.
- Leverage large language models and world foundation models for reasoning, decision-making, and multimodality generation.
- Optimize multimodal generative models using RLFT and explore reward modeling.
- Design and implement evaluation metrics while iterating on ongoing improvements.
What you need
- Enrollment in a graduate program such as a Master's or PhD in Computer Science or a related technical field.
- Strong problem-solving and programming skills utilizing Python and/or C++.
- Good understanding of Jax and PyTorch frameworks.
- Prior experience in model evaluation and developing metrics.
- Demonstrated experience in developing models.
Nice to have
- Some understanding of SQL
- Previous internship experience
- PhD degree
Benefits
- Competitive compensation packages with a housing or relocation bonus if applicable
- Medical, dental, and vision insurance
- Fun intern events and networking opportunities
- Free breakfast, lunch, dinner, and snacks
- Free access to Google shuttles
- Onsite gym access
Skills and tech stack
Questions candidates ask
What is the work policy for this internship?
This will be a hybrid onsite internship position located in Mountain View, California. Resumes are accepted on a rolling basis until the role is filled.
What is the compensation for interns?
The expected hourly rate is $70 for Master's students and $85 for PhD students. Interns are also eligible to participate in company benefit programs subject to requirements.
What tools and technologies will I use?
Engineers on this team utilize Python, C++, Jax, PyTorch, and large language models or world foundation models. Familiarity with SQL is also preferred.
What does the Driver Refinement Foundations team do?
The team is accountable for providing the common, consistent framework and enabling expertise for post-training and heuristic hill climbing across the entire Planner organization. They build fundamental platforms, including the Data Flywheel and the Reasoning platform, to assist engineers in delivering driving behavior improvements.