Zoox

Software Engineer - Learned Trajectory Machine Learning Engineer

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

Base Salary

$189k - $270k/yr

Responsibilities

  • Develop deep learning models using imitation learning and reinforcement learning to generate driving plans for autonomous vehicles.
  • Develop techniques and metrics to evaluate driving-plan safety, progress, comfort, and overall quality.
  • Leverage large-scale machine learning infrastructure to develop new solutions.
  • Build tools to analyze system errors and measure improvements.
  • Collaborate with Perception, Planning, Simulation, and Systems Engineering teams on autonomous driving in complex urban environments.

Requirements

  • Bachelor’s, master’s, or PhD degree 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 C++ or Python, with a basic understanding of C++.
  • Extensive programming and algorithm-design experience.
  • Strong mathematics skills.
  • Conference or journal publications in machine learning or robotics and experience with prediction, autonomous vehicles, or robotics are preferred.

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