Machine Learning Engineer Intern, Waymo
MS/PhD, 2027 Summer
Hourly Masters Pay: $70—$70 USD. Hourly PhD Pay: $85—$85 USD
Autonomous driving technology
Be an early applicant
Autonomous driving technology
1001+ employees
Be an early applicant
Hourly Masters Pay: $70—$70 USD. Hourly PhD Pay: $85—$85 USD
1001+ employees
Company mission
To make it simpler and safer for people to move things and themselves around.
Role
Who you are
- Currently enrolled in an PhD or MS program in Computer Science, Machine Learning or a related field, returning to the program after the internship
- Hands-on experience training and evaluating deep learning models in a modern framework (JAX, PyTorch or TensorFlow), including building data pipelines, choosing losses, and debugging training runs
- Strong programming skills in C++/Python, plus a solid grounding in ML fundamentals: precision/recall trade-offs, class imbalance, evaluation metric selection, and rigorous experiment design
Desirable
- Authorship of published papers in top-tier AI/ML, data mining, or computer vision conferences (e.g., NeurIPS, ICML, ICLR, KDD, CVPR, CoRL, SIGMOD, VLDB, ACL)
- Research or applied experience with transformer and sequence models, multi-task learning, transfer learning or domain adaptation, and parameter-efficient fine-tuning of large pretrained models
- Experience with JAX/Flax, distributed training on TPUs or GPUs, and large-scale data processing (MapReduce-style pipelines, SQL) for building training and evaluation datasets
- Familiarity with autonomous driving, robotics, or simulation; and/or with probability calibration, uncertainty quantification, importance sampling, active learning, or rare-event and imbalanced-data modeling
What the job involves
- Train and fine-tune a large multi-task transformer over driving-log sequences, in JAX/Flax on TPUs - iterating on fine-tuning strategies, training data mixtures, and losses to improve evaluation quality for the hillclimbing workflow
- Design and run rigorous offline and end-to-end evaluations - PR-AUC, calibration quality, and metric sensitivity on real hillclimbing A/B runs - and build the dataset and evaluation pipelines needed to produce them
- Land production-quality code in a shared, high-traffic codebase, and communicate results through a design doc, team deep dives, and a final intern presentation, partnering with UEM Core, Data Science, and release-eval stakeholders
Our take
Waymo, Alphabet's self-driving car company, is changing the way people get around. Its robotaxis, which don't need a human driver, are already operating in cities like Phoenix, San Francisco, Los Angeles, Atlanta, and Austin. Soon, Waymo will expand to Detroit, Las Vegas, and San Diego, showing just how autonomous cars are moving from experiments to real-world transportation.
The company has already completed over 10M fully driverless rides and driven more than 100M miles. Its cars use advanced sensors and AI to navigate streets safely, while partnerships with Uber and DoorDash show that Waymo's technology works for both passengers and deliveries. And it's feeling ambitious too, hoping to offer one million trips per week by the end of 2026.
Even with competitors like Tesla and Zoox, Waymo is leading the autonomous ride-hailing market. Its cars have fewer crashes and keep passengers safer than traditional vehicles. With international plans for London and Japan, Waymo isn't just growing in the US; it's setting its sights on shaping the future of transportation worldwide.
Steph
Company Specialist at Welcome to the Jungle
Insights
-1% employee growth in 12 months
Company
Funding (last 2 of 5 rounds)
Feb 2026
$16bn
LATE VC
Aug 2024
$5.6bn
SERIES C
Total funding: $27.1bn
Company benefits
- Medical, dental, and vision insurance for employees and dependents
- Employee assistance programs focused on mental health
- Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations
- Access to mental health apps
- Onsite wellness centers
- Second medical opinion for you and your loved ones
- Medical advocacy program for transgender employees
- Counseling services
- Support programs including menopause benefit
- Competitive compensation
- Regular bonus and equity performance grant opportunities
- Generous 401(k) and regional retirement plans
- Annual cross-company compensation review and pay equity analysis
- 1-on-1 financial coaching
- Fertility and growing family assistance
- Parental leave and baby bonding leave
- Caregiver leave
- Elder care and support
- Backup childcare
- Survivor income benefit
- Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays
- Jury duty leave
- Military leave
- Hybrid work model with remote work opportunities also available
- Educational reimbursement
- Peer learning and coaching platform
- Donation matching programs
- Employee resource groups
- Volunteer hours
- Internal community groups and local culture clubs
- Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts
- On-site meals and snacks
- Fitness centers, massage programs, and ergonomic support
- On-demand fitness, wellbeing, and cooking classes
- Commuter benefits
Company HQ
Mountain View, CA
Leadership
Dmitri Dolgov
(Co-CEO)Dmitri joined Waymo in 2016 as CTO, becoming Co-CEO in 2021. Prior to this, he spent nearly 8 years at Google as a Distinguished Software Engineer and 2 years at Toyota Research Institute as a Senior Research Scientist.
Tekedra N. Mawakana
(Co-CEO)Tekedra joined Waymo in 2017 as the Global Head of Policy, holding various roles before becoming Co-CEO in 2021. Previously, she was the VP of Global Government Relations and Public Policy at eBay and VP and Deputy General Counsel of GLobal Public Policy at Yahoo.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 206 more jobs at Waymo