Senior/Staff Software Engineer (Machine Learning)

TL;DR

Senior/Staff Software Engineer (Machine Learning): Developing deep learning models for autonomous vehicles with an accent on behavior prediction and motion planning. Focus on leveraging large-scale ML infrastructure and collaborating with cross-functional teams to solve complex driving challenges.

Location: Hybrid (Foster City, CA)

Company

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market.

What you will do

• Develop new deep learning models using imitation and reinforcement learning for autonomous vehicle driving plans.
• Leverage large-scale machine learning infrastructure to discover new solutions.
• Develop metrics and tools to analyze errors and system improvements.
• Collaborate with engineers on Perception, Planning, and Simulation to address autonomous driving challenges.

Requirements

• BS, MS, or PhD in computer science or related field.
• Experience with transformer-based model architectures and reinforcement learning.
• Experience with production ML pipelines.
• Fluency in Python and basic understanding of C++.
• Strong mathematics skills.

Nice to have

• Conference or Journal publications in Machine Learning or Robotics.
• Prior experience with Prediction and/or autonomous vehicles.