Waymo · 3 months ago
Machine Learning Engineer, Runtime & Optimization
Mountain View, California, USA
Full-time
Onsite
Senior Level
$204K/yr - $259K/yr
5+ years expWaymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The role involves leading collaboration with ML scientists to optimize machine learning workloads and improve compute performance on cloud and self-driving cars.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
Lead the collaboration with the world-class Waymo ML scientists in perception, planner, research and simulation. Identify opportunities in both systems and models to make ML workloads faster
Lead projects from proposals through execution by developing junior engineers
Analyze and improve ML system workloads on both cloud and self-driving cars
Apply model optimization, efficient deep learning techniques and ML software improvements to Waymo's ML systems
Qualification
Required
M.S. in CS, EE, Deep Learning or a related field
2+ years of experience as a technical lead, including writing project plans, engaging with customer teams, mentoring, responsible for goals & execution, reporting status
5+ years of experience developing solutions in ML systems or ML software stack (Pytorch/JAX/TF, runtime libraries, ML compiler)
Deep understanding of ML system architecture, performance analysis and tools
Strong Python or C++ programming skills
Preferred
PhD in CS, EE, Deep Learning or a related field
Familiarity with the HW architecture of ML hardware accelerators (e.g., GPU/TPU)
Deep knowledge of model optimization or efficient deep learning techniques for foundation models or LLM
Experience with GPU HW or TPU HW and related system software
Benefits
Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program
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.
1001-5000 employees
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
$27.1BKey Investors
Dragoneer Investment Group,DST Global,Sequoia CapitalAlphabetFidelity,Perry Creek Capital,T. Rowe Price
2026-02-02Series D· $16B
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
Leadership Team
Tekedra Mawakana
Co-Chief Executive Officer
Annabel Chang
Director of U.S. State and Local Public Policy and Political Campaigns
Recent News
PCMag.com - Technology Product Reviews, News, Prices & Tips
2026-02-07
2026-02-07
2026-02-07
Company data provided by crunchbase