# Staff / Senior Staff Software Engineer - Learned Trajectory Machine Learning Engineer

**Company:** [Zoox](https://hotfix.jobs/companies/zoox)
**Location:** Foster City, CA
**Role:** ML Engineering
**Salary:** $277k – $407k/yr
**Skills:** Python, C++, Transformers, Reinforcement Learning, Imitation Learning, Machine Learning Pipelines, Deep Learning, Mathematics, Autonomous Driving, Robotics
**Posted:** 2024-11-07

> Develops deep learning models using imitation and reinforcement learning to generate safe, efficient driving trajectories for autonomous vehicles. Collaborates with Perception, Planning, and Simulation teams; requires ML expertise, Python fluency, and transformer experience.

## Job Description

## Responsibilities
- Develop new deep learning models using imitation learning and reinforcement learning to generate driving plans for autonomous vehicles.
- Work on techniques to estimate the quality of driving plans along dimensions of safety, progress, comfort, etc.
- Leverage large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field.
- Develop metrics and tools to analyze errors and understand improvements of systems.
- Collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments.

## Qualifications
- BS, MS, or PhD degree in computer science or related field.
- Experience with training and deploying **transformer-based model architectures** and **reinforcement learning**.
- Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines.
- Fluency in **Python** with a basic understanding of **C++**.
- Extensive experience with programming, algorithm design, and strong mathematics skills.

## Bonus Qualifications
- Conference or Journal publications in Machine Learning or Robotics related venues.
- Prior experience with Prediction and/or autonomous vehicles or robotics in general.

## Similar roles

- [Senior Staff Machine Learning Engineer - Perception](https://hotfix.jobs/jobs/ed07dc78-49c2-4be4-89e7-c698011ce10b) - Zoox - Boston, MA - $277k – $389k/yr
- [Staff Machine Learning Engineer, Underwriting and Credit](https://hotfix.jobs/jobs/b61761e2-2aef-46fd-888b-9f738b8c47f2) - Square - Remote - $277k – $415k/yr
- [Staff Applied Machine Learning Engineer - Fraud & Abuse](https://hotfix.jobs/jobs/c8ea2c50-bd8e-410a-9116-6abf45a49a47) - Square - San Francisco, CA - $277k – $415k/yr
- [Staff Applied Machine Learning Engineer](https://hotfix.jobs/jobs/e1528eb9-6d81-4ecf-b9d5-ebc31ee66a75) - Square - Remote - $277k – $415k/yr
- [Staff Machine Learning Engineer](https://hotfix.jobs/jobs/522da530-b698-444d-80e5-307cd4eeafe0) - Otter - Mountain View, CA - $278k – $330k/yr

**Apply:** https://hotfix.jobs/jobs/9449191b-9b08-4799-997f-7a53378184e1
**Canonical:** https://hotfix.jobs/jobs/9449191b-9b08-4799-997f-7a53378184e1