Waymo · 2 months ago
Senior Machine Learning Infrastructure Engineer, Simulation
Mountain View, CA
Full-time
Onsite
Mid, Senior Level
$120K/yr - $130K/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 Senior Machine Learning Infrastructure Engineer to lead the development of advanced AI/ML infrastructure for multi-billion parameter foundation models in ML accelerator-friendly simulations.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
Be part of a world-class, research engineering team to improve the state-of-the-art of ultra realistic multi-agent simulations using foundation models
Collaborate with the core Waymo Realism Modeling team in London and Waymo Oxford to use large foundation models to improve sim realism
Work at the intersection of data engineering, model development, and simulations, and drive architectural decisions. Manage complex systems, driving architectures and designs that meet technical and goals
Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation, model training, and evaluation
Collaborate to derive performance and system-level requirements for large ML systems. Translate product/business goals into measurable technical deliverables, ensuring system component agreement
Qualification
Required
5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure such as developing, scaling, training, deploying, and optimizing large-scale machine learning systems from data to model
Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, Ray, or similar frameworks
Experience with machine learning models and algorithms such as autoregressive transformers and familiarity scaling large models across ML accelerator profiling tools to uncover performance bottlenecks
Strong leadership skills, experience driving ambiguous problems end-to-end, with a willingness and independence to pick up whatever knowledge
Passionate about building infrastructure, libraries, tools, and pipelines for engineers and scientists
Excellent communication skills with the ability to translate complex technical concepts for a broad audience
Practical familiarity in Autonomous Driving, Simulations, and ML accelerators
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
Elisa de Martel
Chief Financial Officer
Recent News
2026-01-12
PYMNTS.com
2026-01-12
Company data provided by crunchbase