ML Training Infrastructure and Efficiency Intern, Waymo
2026 Summer, PhD
Expected hourly rate $85—$85 USD
Office located in Mountain View, CA
Autonomous driving technology
Be an early applicant
Autonomous driving technology
1001+ employees
Be an early applicant
Expected hourly rate $85—$85 USD
Office located in Mountain View, CA
1001+ employees
Company mission
To make it simpler and safer for people to move things and themselves around.
Role
Who you are
- Python or C++ experience
- Scale model training to multiple accelerators
- Reinforcement learning
- Machine Learning frameworks (e.g., TensorFlow, JAX, or PyTorch) and distributed training algorithms
- Practical familiarity using ML accelerator profiling tools to uncover performance bottlenecks
Desirable
- ML compiler optimization such as TVM or XLA
- Understand ML compiler infrastructure, such as of HLO and MLIR
What the job involves
- Waymo interns partner with leaders in the industry on projects that create impact to the company
- We believe learning is a two-way street: applying your knowledge while providing you with opportunities to expand your skill-set
- Interns are an important part of our culture and our recruiting pipeline. Join us at Waymo for a fun and rewarding internship!
- Identify model performance bottlenecks and optimization opportunities
- Improve the developer experience and performance of our scalable ML framework
- Apply model optimization, efficient deep learning techniques and ML software improvements to Waymo's ML systems
- This will be a hybrid onsite internship position
Application process
- We will accept resumes on a rolling basis until the role is filled
- To be in consideration for multiple roles, you will need to apply to each one individually - please apply to the top 3 roles you are interested in
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, its 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 4 rounds)
Aug 2024
$5.6bn
SERIES C
Jun 2021
$2.5bn
GROWTH EQUITY VC
Total funding: $11.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 124 more jobs at Waymo