Zoox

Senior / Staff Software Engineer - Learned Trajectory Machine Learning Engineer

Zoox
Apply
almost 2 years ago
Foster City, CA, USASenior / Staff+
H1B sponsor

Base Salary

$242k - $349k/yr

Responsibilities

  • Develop deep learning models using imitation learning and reinforcement learning to generate autonomous vehicle driving plans.
  • Develop techniques to evaluate driving-plan quality across safety, progress, comfort, and related dimensions.
  • Use large-scale machine learning infrastructure to develop new solutions.
  • Create metrics and tools for error analysis and system improvement.
  • Collaborate with Perception, Planning, Simulation, and other engineering teams on autonomous driving in complex urban environments.

Requirements

  • BS, MS, or PhD in computer science or a related field.
  • Experience training and deploying transformer-based model architectures and reinforcement learning systems.
  • Experience with production machine learning pipelines, including dataset creation, training frameworks, and metrics pipelines.
  • Fluency in Python and basic understanding of C++.
  • Extensive programming and algorithm-design experience with strong mathematics skills.
  • Conference or journal publications in machine learning or robotics are preferred.
  • Prior experience with prediction, autonomous vehicles, or robotics is preferred.

Benefits

  • The role contributes to Zoox's ground-up autonomous vehicle fleet and mobility-as-a-service ecosystem.
  • Accommodations are available for the application and interview process.

Tech Stack

Categories

Zoox

About Zoox

1,001-5,000 employees

Zoox builds a fully autonomous, purpose-built electric vehicle and the end-to-end platform to operate it as an urban robotaxi service. The company designs the vehicle, self-driving software, and fleet operations in-house to offer mobility-as-a-service rather than selling cars. Founded in 2014 and headquartered in Foster City, California, Zoox is a subsidiary of Amazon.

Contact me