Senior/Staff ML Engineer, 3D/4D World Modeling, Simulation

Highlights

• Build ultra-realistic controllable virtual worlds to test the Waymo Driver.
• Model full sensor suite and traffic to validate autonomous ride-hail service.
• Partner with world-class research teams across Waymo and Alphabet.

What they offer

• Discretionary annual bonus program
• Equity incentive plan
• Generous company benefits program

Ideal candidate

Experienced ML builder passionate about photorealistic simulation and autonomous driving, eager to turn frontier research into production and mentor teammates.

Requirements

• MS or PhD in Computer Science, Machine Learning, Robotics, or related field
• 5+ years of experience in ML engineering and applied deep learning with shipped products or publications
• Proven experience developing and training large-scale generative video models or Vision Language Models
• Deep expertise in 3D world modeling or 3D computer vision
• Familiarity with 3D reconstruction and rendering techniques (e.g., 3D Gaussian Splatting)
• Strong Python programming skills and experience with JAX/Flax, PyTorch or TensorFlow