Drimva

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

First seen Oct 28, 2026 · Last verified Nov 4, 2026 (UTC)