Senior Machine Learning Engineer - ML Agents and Planning

Zoox

2 months ago

Foster City, CA, USA

Senior / Staff+

H1B Sponsor

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 discover new solutions.
Develop metrics and tools to analyze errors and system improvements.
Collaborate with engineers in Perception, Planning, Simulation, and Validation to solve autonomous driving problems.

PhD in computer science or related field with 1+ years of experience, or MSc with 5+ years of relevant experience.
Experience in Planning and/or Prediction using Reinforcement Learning techniques.
Experience with 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++.
Top tier publications can substitute for experience.

C++Python