Waymo · 3 weeks ago
Machine Learning SWE Infra
Mountain View, CA, USA
Full-time
Onsite
Senior Level
$238K/yr - $302K/yr
5+ years expWaymo is an autonomous driving technology company with the mission to be the world's most trusted driver. They are seeking a Machine Learning Software Engineer to provide technical leadership on large-scale machine learning systems and to build scalable systems for training and evaluation of autonomous vehicle models.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models
Build scalable systems for training and fine-tuning large-scale models to evaluate interesting driving behaviors
Work at the intersection of data engineering, model development, and simulation Provide guidance on architectural decisions and technical directions
Own large, complex systems, driving architectures that meet technical and business objectives
Oversee the production and optimization of machine learning models aiming to assess Waymo’s expansive fleet of vehicles that cumulatively travel millions of miles
Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation, model training, and evaluation
Collaborate cross-functionally to derive performance and system-level requirements for large ML systems
Translate product/business goals into measurable technical deliverables, ensuring system component alignment
Qualification
Required
M.S. or Ph.D. degree Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience
5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure such as developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model
A history of contributions to machine learning tooling and frameworks e.g. PyTorch, Jax, Tensorflow, Ray, or similar. The candidate should understand both the user facing API and the internal workings
Strong expertise in distributed training techniques, including gradient sharding and optimization strategies for scaling large models across ML accelerator profiling tools to uncover performance bottlenecks
Deep understanding of state-of-the-art machine learning models such as autoregressive transformers
Strong leadership skills with experience navigating cross-functional teams and providing technical leadership projects across multiple organizations
Preferred
7+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure such as developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model
Experience in the autonomous vehicles domain, robotics, or complex simulation environments
Deep understanding of state-of-the-art RL techniques, including those used for fine-tuning large models (e.g., from human feedback/preferences)
Familiarity with large-scale simulation platforms and their integration with ML training workflows
Experience designing and using metrics for evaluating complex AI systems
Track record of technical leadership, influencing senior stakeholders, and driving innovation across team boundaries
Excellent communication skills, with the ability to articulate complex technical concepts clearly
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
$11.1BKey Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M
Leadership Team
Tekedra Mawakana
Co-Chief Executive Officer
Annabel Chang
Head of State Policy & Government Relations
Recent News
2026-01-16
2026-01-16
Company data provided by crunchbase