Zoox logo
ZooxPosted 1 month ago

Machine Learning Engineer - ML Agents and Planning

On-siteFoster City, California, United States

Full TimeDoctorate Or Professional DegreeLarge

Job Summary

Develop deep learning models using imitation learning and reinforcement learning to generate driving plans for human-like agents. Estimate plan quality across safety, progress, comfort, and realism dimensions while contributing to large-scale ML infrastructure. Develop metrics and tools to analyze system errors and improvements. Collaborate with Perception, Planning, Simulation, and Validation engineers to solve the autonomous driving problem. Work on the Offline Driving Intelligence team, applying foundation models off-vehicle to enhance simulation and validation generalization.

Required Qualifications

  • PhD degree in computer science or related field
  • master's degree and 5+ years of professional experience in a relevant field
  • Experience in Planning and / or Prediction using Reinforcement Learning techniques
  • Experience with training and deploying transformer-based model architectures
  • Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
  • Fluency in Python with a basic understanding of C++

Desired Qualifications

  • Top tier publications (NeurIPS, ICML, CVPR)

Hiring someone like this?

Get your role in front of qualified candidates on Sorce.

Get started

Apply to this job in one click with Sorce

Apply on Sorce