Waymo · 1 month ago
Senior Machine Learning Infrastructure Engineer, Simulation
Mountain View, CA
Full-time
Onsite
Senior Level
$204K/yr - $259K/yr
5+ years expWaymo is an autonomous driving technology company focused on building the Waymo Driver to improve mobility and safety. They are seeking a Senior Machine Learning Infrastructure Engineer to lead the development of AI/ML infrastructure for advanced simulations used in training the Waymo Driver, collaborating with various teams to enhance simulation realism and mentoring junior engineers.
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 Google DeepMind and Waymo Realism Modeling teams in London, and Waymo Oxford to use the large models to improve sim realism
Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models. Work at the intersection of data engineering, model development, and deployment, and provide guidance on architectural decisions and technical directions. Own large, complex systems, driving architectures that meet technical and business objectives
Design and scale large distributed systems covering the ML lifecycle, supporting planet-scale dataset generation and model training
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
Mentor junior engineers, growing their expertise and fostering a collaborative culture
Qualification
Required
BS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience
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
Preferred
10+ 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
Solid experience in the development and optimization of machine learning infrastructure tools like DeepSpeed, PyTorch, TensorFlow, or similar frameworks
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 auto-regressive transformers and familiarity with custom-kernels for diverse h/w compute based efficiency
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
Annabel Chang
Head of State Policy & Government Relations
Recent News
San Bernardino Sun
2026-02-04
2026-02-04
Company data provided by crunchbase