Senior/Staff Machine Learning Engineer (Reinforcement Learning), Motion Planning

What you'll do

• Design, develop, and deploy behavior-planning models using reinforcement learning, behavior cloning, and imitation learning.
• Own the reinforcement-learning pipeline from environment construction through training and modeling; contribute to infrastructure for distributed training.
• Analyze model performance and failure modes, improve generalization, and collaborate with simulation and planning teams on deployment.

What they ask for

• MS or PhD in Computer/Software Engineering, Robotics, Machine Learning, or a related field.
• Production-code experience in C/C++ and Python.
• Experience with machine-learning frameworks such as PyTorch.
• Experience with reinforcement learning; strong oral and written communication skills.

Nice to have

• Experience with high-throughput batched driving simulators such as GPUDrive and PufferDrive, self-play reinforcement learning, or sim-to-real in robotics/autonomous driving.
• Prior experience in learned behavior planning or prediction for autonomous vehicles.

Pay

USD 130,000–220,000 per-year-salary (base salary)

Tools

C/C++, Python, PyTorch, GPUDrive, PufferDrive

AI focus

Reinforcement learning and other learning approaches for autonomous-vehicle behavior planning, including model training, generalization, and deployment.