Aurora
ML Engineer, Behavior Planning
$139K–$223K
Mountain View, CA · Remote · Full-time · Posted Oct 12, 2026
- Experience
- 2+ years developing state-of-the-art ML models.
- Education
- BS, MS, or PhD in Robotics, Machine Learning, Computer Science, or a related quantitative field, or equivalent practical experience.
About the job
Develop and deploy machine-learning models for autonomous-vehicle behavior planning, evaluation, and simulation, from research through production vehicles.
What you'll do
- Design onboard planning models trained with imitation and reinforcement learning.
- Develop offboard critic models to evaluate driving behavior and foundation models that create realistic simulation world models.
- Advance autonomous-driving methods and deploy models on production vehicles.
What they ask for
- Degree in Robotics, Machine Learning, Computer Science, or a related quantitative field, or equivalent practical experience.
- 2+ years developing state-of-the-art ML models in research or production.
- Hands-on experience with Imitation Learning or Reinforcement Learning for physical or simulated agents, and training large models on massive datasets using distributed computing.
- Fluency in Python; deep experience with PyTorch or another modern ML framework; mastery of Transformers and Diffusion Models.
Nice to have
- Publications at top-tier ML conferences, production deployment experience, or experience developing generative models or neural simulators.
- Fluency in C++.
Pay
$139,000 - $223,000 per Year (base salary)
Tools
Python, PyTorch, Transformers, Diffusion Models, C++
AI focus
Develop ML planning models using imitation and reinforcement learning, offboard critic models, and foundation models for simulation world models; deploy models in autonomous vehicles.
Model-written summary · the original posting is authoritative