Waymo · 4 weeks ago
2026 Summer Intern, PhD, ML Training Infra and Efficiency
Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The internship focuses on improving machine learning systems by identifying performance bottlenecks and enhancing the developer experience.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
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
Qualification
Required
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
Preferred
ML compiler optimization such as TVM or XLA
Understand ML compiler infrastructure, such as of HLO and MLIR
Benefits
Company’s generous benefits programs
Company
Waymo
Waymo is a mobility technology company that improves transportation by developing self-driving solutions for travelers and daily commuters. It is a sub-organization of Alphabet.
H1B Sponsorship
Waymo has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (231)
2024 (175)
2023 (268)
2022 (306)
2021 (298)
2020 (317)
Funding
Current Stage
Late StageTotal Funding
$11.1BKey Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M
Recent News
2026-01-12
PYMNTS.com
2026-01-12
Company data provided by crunchbase