Autonomous Driving ML Engineer: Behavior & Planning

Zoox; Foster City (CA); On-site; USD 189,000 - 270,000; Full time

Benefits offered by this job

Health insurance

Stock Appreciation Rights

Amazon RSUs

Paid time off

Job summary

Zoox is seeking a Research Engineer for Prediction & Behavior ML to develop deep learning models that learn and predict driving behaviors for autonomous vehicles, using imitation and reinforcement learning to generate driving plans.

You will work with large-scale ML infrastructure, build metrics, and collaborate with Perception, Planning, and Simulation teams to validate driving performance in urban environments.

Qualifications

• BS, MS, or PhD in computer science or related field.
• Experience with training and deploying transformer-based model architectures and reinforcement learning.
• Experience with production ML pipelines: dataset creation, training frameworks, metrics pipelines.
• Fluency in C++ or Fluency in Python with a basic understanding of C++.
• Extensive experience with programming and algorithm design.
• Strong mathematics skills.

Responsibilities

• You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for our autonomous vehicle.
• You will leverage our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
• You will develop metrics and tools to analyze errors and understand improvements of our systems
• You will collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments

Skills

Transformer-based models

Reinforcement learning

ML deployment pipelines

C++

Python

Mathematics

Education

BS/MS/PhD in CS or related field

Job description

Zoox is seeking a Research Engineer for Prediction & Behavior ML to develop deep learning models that learn and predict driving behaviors for autonomous vehicles, using imitation and reinforcement learning to generate driving plans.

You will work with large-scale ML infrastructure, build metrics, and collaborate with Perception, Planning, and Simulation teams to validate driving performance in urban environments.