Waymo · 2 weeks ago
Senior Software Engineer, Simulation ML Infrastructure
Mountain View, CA
Full-time
Onsite
Senior Level
$204K/yr - $259K/yr
5+ years expWaymo is an autonomous driving technology company with a mission to improve access to mobility while saving lives. The role involves leading the development of advanced AI/ML infrastructure for multi-billion parameter foundation models in ML accelerator-friendly simulations, working with a high-performing research engineering team to enhance simulation realism.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
Be part of a world-class, high-performing research engineering team to advance the state of the art of ultra realistic multi-agent simulations using foundation models
Collaborate closely 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. Own large, complex systems, driving architectures and designs that meet technical and business objectives
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
Your expertise in massive model scaling, ML accelerators, and large-scale distributed systems will be required for designing and scaling our systems
Preferred
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
Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, Ray, or similar frameworks
Strong understanding of state-of-the-art 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 with experience driving ambiguous problems end-to-end, with a willingness and independence to pick up whatever knowledge to get the job done. Passionate about building infrastructure, libraries, tools, and pipelines for engineers and scientists
Excellent communication skills, both verbal and written, with the ability to translate complex technical concepts for a broad audience
Practical familiarity in Autonomous Driving, Simulations, and ML accelerators is a plus
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-09
Sherwood News
2026-01-09
2026-01-09
Company data provided by crunchbase