Zoox

Machine Learning Engineer - ML Agents and Planning

Zoox
Apply
2 months 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 human-like agents.
  • Develop techniques to estimate driving-plan quality across safety, progress, comfort, and realism.
  • Contribute to large-scale machine learning infrastructure.
  • Develop metrics and tools to analyze errors and measure system improvements.
  • Collaborate with Perception, Planning, Simulation, and Validation engineers on autonomous driving.

Requirements

  • PhD in computer science or a related field, or a master’s degree and 5+ years of professional experience in a relevant field.
  • Experience with planning and/or prediction using reinforcement learning techniques.
  • Experience training and deploying transformer-based model architectures.
  • Experience with production machine learning pipelines, including dataset creation, training frameworks, and metrics pipelines.
  • Fluency in Python and a basic understanding of C++.
  • Top-tier publications such as NeurIPS, ICML, or CVPR are a bonus.

Benefits

  • Paid time off, including sick leave, vacation, and bereavement leave.
  • Unpaid time off.
  • Zoox Stock Appreciation Rights and Amazon Restricted Stock Units.
  • Health insurance, long-term care insurance, disability insurance, and life insurance.

Tech Stack

Categories

Zoox

About Zoox

1,001-5,000 employees

Zoox is transforming mobility-as-a-service by developing a fully autonomous, purpose-built fleet designed for AI to drive and humans to enjoy.

Contact me