Machine Learning Engineer, Simulation Realism

Highlights

• Drive innovation in ultra-realistic world simulation for autonomous vehicles.
• Test and train the Waymo Driver with billions of miles in simulation.
• Model realistic agents, roads, traffic systems and weather conditions.

What they offer

• Discretionary annual bonus program
• Equity incentive plan
• Company benefits program

Ideal candidate

A collaborative ML engineer with 5+ years building real-world systems, passionate about simulation realism and autonomous driving. You love turning cutting-edge GenAI research into scalable, production-ready impact.

Requirements

• 5+ years of experience with ML software engineering
• 5+ years of experience with Python
• 5+ years of ML software experience with machine learning framework including Pytorch, Tensorflow, Jax/Flax
• Experience working with, creating and/or developing ML models for robotics and/or self driving vehicles
• Strong C++ in production setting, with experience in integrating ML models in production system
• Experience in training large scale models on GPU/TPU clusters