ML Engineer, Foundation Model Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waymo · 15 hours ago

ML Engineer, Foundation Model Infrastructure

positionMountain View, CA
timeFull-time
remoteHybrid
seniorityMid, Senior Level
money$204K/yr - $259K/yr

Waymo is an autonomous driving technology company focused on building the Waymo Driver to improve mobility and save lives. The ML Engineer will develop machine learning solutions, build data systems, and enhance the efficiency of the ML development lifecycle.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
check
H1B Sponsor Likelynote

Responsibilities

Build and operate the petabyte-scale data systems and ML pipelines at the heart of Waymo's foundation model development
Shepherd cutting-edge foundation models from research prototypes to robust components within the Waymo Driver
Create the automated infrastructure for rigorously benchmarking, continuously monitoring, and safely releasing models
Wield large-scale compute and frameworks like Flume and JAX to process massive datasets and train/deploy complex models
Drive significant leaps in the speed, reliability, and efficiency of the end-to-end ML development lifecycle
Partner with AI Foundations, ML, and Platform experts to transform model innovations into tangible on-road improvements

Qualification

PythonC++Deep learning frameworksLarge-scale data pipelinesMLOps platformsHands-on SWE skillsAV planning experienceDistributed systems designML model evaluation methodologies

Required

Masters degree in Computer Science, Machine Learning, Robotics, similar technical field of study, or equivalent practical experience
Proficiency in Python
Proficiency in C++
Familiarity with one of the modern deep learning frameworks (e.g. Pytorch, JAX, Tensorflow)
Experience building or maintaining large-scale data pipelines or ML infrastructure (e.g., Flume, Spark, Borg, Kubeflow)

Preferred

Strong hands-on SWE skills, able to drive development of large, complex shared codebases
Experience in AV planning and related research
Experience designing and building distributed systems or MLOps platforms (e.g., model versioning, experiment tracking, CI/CD for ML)
Prior work in an industrial or research setting developing methodologies for the evaluation of ML models

Benefits

Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program

Company

Waymo

twittertwittertwitter
Glassdoor3.7
company-logo
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.
dateFounded in 2009
location
Mountain View, California, USA
people-size1001-5000 employees
web-link
https://waymo.com

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
Annabel Chang
Director of U.S. State and Local Public Policy and Political Campaigns
linkedin

Recent News

Korea Herald
Hyundai’s Ioniq 5 robotaxi is finally ready to hit the streets
2026-01-12
PYMNTS.com
Legacy Business Models Break
2026-01-12
Benzinga.com
Weekend Round-Up: GM's $7.1 Billion EV Charge, Polestar's Record Year, Tesla's Best Month In China And More
2026-01-11
Company data provided by crunchbase