2027 Summer Intern MS PhD Machine Learning Engineer
Waymo is an Alphabet-owned autonomous-driving company operating a public autonomous ride-hailing service and developing the Waymo Driver.
Funding history
About Waymo
Waymo LLC develops and deploys the Waymo Driver, its fully autonomous driving technology, and operates autonomous ride-hailing services. It was established as an Alphabet self-driving technology company in 2016, building on Google’s self-driving car project begun in 2009.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will train and fine-tune a multi-task transformer using JAX and Flax on TPUs, design offline and end-to-end evaluations, build data and evaluation pipelines, deliver production-quality code, and communicate results through technical documentation and presentations.
Requirements
- Enrollment in a PhD or MS program in Computer Science, Machine Learning, or a related field
- Deep learning model training and evaluation
- JAX, PyTorch, or TensorFlow
- Data pipeline development
- C++
- Python
- Machine learning fundamentals
- Experiment design
Responsibilities
- Train and fine-tune a multi-task transformer over driving-log sequences
- Design and run offline and end-to-end evaluations
- Build datasets and evaluation pipelines
- Deliver production-quality code in a shared codebase
- Communicate results through a design document, deep dives, and a final presentation
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
- Mental wellness support
- Gym membership
- Wellness programs
- Bonus opportunities
- Equity
- Employee discounts
- Paid time off
- Bereavement leave
- Sick leave
- Parental leave
- Education reimbursement
- Employee Resource Groups
- Volunteer time off