2026 Summer Intern, PhD, ML Training Infra and Efficiency jobs in United States
cer-icon
Apply on Employer Site
company-logo

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
check
H1B Sponsor Likelynote

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

PythonC++Machine Learning frameworksReinforcement learningML accelerator profiling toolsML compiler optimizationML compiler infrastructure

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 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 Stage
Total Funding
$11.1B
Key Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M

Leadership Team

leader-logo
Tekedra Mawakana
Co-Chief Executive Officer
linkedin
leader-logo
Elisa de Martel
Chief Financial Officer
linkedin
Company data provided by crunchbase