GrepJob
Zoox

Machine Learning Engineer - ML Agents and Planning

Zoox
Apply
about 16 hours ago
Foster City, CA, USASenior / Staff+
H1B Sponsor

Base Salary

$189k - $270k/yr

Responsibilities

  • Develop new deep learning models using imitation and reinforcement learning for driving plans.
  • Estimate the quality of driving plans focusing on safety, progress, comfort, and realism.
  • Contribute to large-scale machine learning infrastructure to explore new solutions.
  • Develop metrics and tools for error analysis and system improvements.
  • Collaborate with engineers across various teams to tackle the Autonomous Driving problem.

Requirements

  • PhD in computer science or related field, or a master's degree with 5+ years of relevant experience.
  • Experience in Planning and/or Prediction using Reinforcement Learning techniques.
  • Proficiency in training and deploying transformer-based model architectures.
  • Experience with production Machine Learning pipelines including dataset creation and training frameworks.
  • Fluency in Python and basic understanding of C++.

Benefits

  • Comprehensive benefits package including paid time off and health insurance.
  • Zoox Stock Appreciation Rights and Amazon RSUs.
  • Long-term care insurance and disability insurance.
  • Life insurance and unpaid time off options.

Tech Stack

Categories

AI & MLData Science