Waymo · 5 hours ago
Senior/Staff Machine Learning Engineer
Mountain View, CA
Full-time
Hybrid
Senior Level, Lead/Staff
$204K/yr - $259K/yr
5+ years expWaymo is an autonomous driving technology company focused on building the Waymo Driver to improve access to mobility and save lives. The role involves leading the design and development of advanced 4D world models and generative systems for simulation, while mentoring other engineers and collaborating with research teams.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
Lead the design, development and deployment of cutting-edge 4D world models and generative systems for ultra-realistic and controllable sensor and semantics generation for simulation use cases at Waymo
Architect and implement scalable and robust ML pipelines for training, evaluating, and deploying large-scale generative models into our simulation infrastructure, including techniques like model distillation and quantization
Build and scale production-ready video generation techniques (e.g., Diffusion, Flow Matching) to create dynamic and interactive simulation environments
Apply Vision Language Models (VLMs) to enhance the semantic understanding and controllability of our world simulation products
Partner with world class research teams across Waymo and Alphabet to leverage State-of-The-Art research in 4D world modeling and generative AI into robust, production-ready solutions
Mentor and provide technical guidance to other engineers on the team
Qualification
Required
MS or PhD in Computer Science, Machine Learning, Robotics, or a related field
5+ years of experience in ML engineering and applied Deep Learning, with a strong portfolio of shipped products or publication record
Proven experience in developing and training large-scale generative models for video generation (e.g., Diffusion models, Flow Matching) or Vision Language Models (VLMs) and their applications
Deep expertise in 3D World Modeling or 3D computer vision
Familiarity with 3D reconstruction and rendering techniques (e.g., 3D Gaussian Splatting)
Strong programming skills in Python and experience with ML frameworks such as Jax/Flax, PyTorch or Tensorflow
Preferred
PhD and a strong track record of delivering impactful ML products in 3D generative models, world models, or video generation
Experience in simulating sensor data (Camera, Lidar, Radar) and/or semantic scenes
Experience with autonomous systems, robotics, or autonomous vehicle simulation
Experience in training and optimizing large scale models on GPU/TPU clusters for efficient serving
Experience in C++ for production systems
Benefits
Waymo employees are also eligible to participate in 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
Director of U.S. State and Local Public Policy and Political Campaigns
Recent News
2026-01-12
PYMNTS.com
2026-01-12
Company data provided by crunchbase