Senior / Staff Software Engineer - Learned Trajectory Machine Learning Engineer

The role

You will develop deep learning models that produce driving trajectories for autonomous vehicles. You will use imitation learning and reinforcement learning, and build metrics to judge plan quality on safety, progress and comfort. You will work with the Planner, Perception and Simulation teams on urban driving.

What you will do

• Develop deep learning models that generate driving plans using imitation learning and reinforcement learning
• Develop techniques to estimate the quality of driving plans for safety, progress and comfort
• Use large-scale machine learning infrastructure to find new solutions
• Develop metrics and tools to analyze errors and track improvements
• Collaborate with Perception, Planning and Simulation engineers on autonomous driving in complex urban environments

Required qualifications

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

Nice to have

• Conference or journal publications in machine learning or robotics venues
• Prior experience with prediction and/or autonomous vehicles or robotics

Before you apply

• You do not need to match every listed expectation to apply

About the company

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem to bring it to market. It works at the intersection of robotics, machine learning and design, aiming to provide mobility-as-a-service in urban environments.