2 months ago
Mountain View, CA, USASenior / Staff+
H1B sponsor
Base Salary
$213k - $263k/yr
Responsibilities
- Lead the design, development, and deployment of 4D world models and generative systems for realistic and controllable simulation.
- Architect scalable ML pipelines for training, evaluating, and deploying large-scale generative models, including model distillation and quantization.
- Build production-ready video generation systems using techniques such as Diffusion and Flow Matching.
- Apply Vision Language Models to improve semantic understanding and controllability of simulation products.
- Collaborate with Waymo and Alphabet research teams to productionize 4D world modeling and generative AI research.
- Mentor and provide technical guidance to other engineers.
Requirements
- Master’s or PhD in Computer Science, Machine Learning, Robotics, or a related field.
- At least 5 years of experience in ML engineering and applied deep learning.
- A strong portfolio of shipped products or a publication record.
- Experience developing and training large-scale generative video models or Vision Language Models and applying them to products.
- Deep expertise in 3D world modeling or 3D computer vision.
- Familiarity with 3D reconstruction and rendering techniques such as 3D Gaussian Splatting.
- Strong Python programming skills and experience with Jax/Flax, PyTorch, or TensorFlow.
- Preferred: PhD and a record of impactful ML products involving 3D generative models, world models, or video generation.
- Preferred: experience simulating camera, lidar, radar, or semantic scene data.
- Preferred: experience with autonomous systems, robotics, or autonomous vehicle simulation.
- Preferred: experience training and optimizing large-scale models on GPU/TPU clusters for efficient serving.
- Preferred: C++ experience for production systems.
Benefits
- Full-time hybrid role, indicated by #LI-Hybrid.
- Eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program.
- Base salary range of $213,000–$263,000 USD across US locations, with actual pay determined by location, experience, education, training, and skill level.
Categories
About Waymo
Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.