# Senior / Staff Machine Learning Engineer - Perception Attributes

**Company:** [Zoox](https://hotfix.jobs/companies/zoox)
**Location:** Foster City, CA
**Role:** ML Engineering
**Salary:** $242k – $333k/yr
**Experience:** 7+ years
**Skills:** Deep Learning, Machine Learning Pipelines, Python, Knowledge Distillation, Vision Language Models, Vlms, Vision Transformers, Vits, C++, Data Labeling, Auto-Labeling, Dataset Creation
**Posted:** 2026-05-02

> Develops and maintains behavioral models for road users (vehicles, pedestrians, cyclists) in autonomous driving perception stack. Requires MS/PhD, 7+ years experience, deep learning expertise, Python fluency, and production ML pipelines.

## Job Description

## Responsibilities
- Lead the development of sophisticated behavioral models for vehicles, pedestrians, and cyclists as a key member of the Attributes team within Zoox's Perception department.
- Create and maintain perception attribute models that generate essential signals, enabling our autonomous vehicles to understand and predict the behavior of various road users.
- Collaborate closely with Prediction and Planning teams to optimize your models' outputs, directly influencing how our autonomous vehicles make real-time driving decisions.
- Work with data labeling and ontology teams on data labeling and ontology definitions of the road users in different attributes and generate auto-labeling or data mining strategies for different attributes.

## Qualifications
- MS/PhD in computer science or related fields with a minimum of 7 years of relevant experience
- Experience with training and deploying Deep Learning models
- Experience with knowledge distillations from large foundation models
- Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
- Fluency programming in Python and extensive experience with algorithm design
- Strong mathematics skills

## Bonus Qualifications
- Familiarity of VLMs/VLAs/ViTs
- Experience with large model distillation in a production environment
- Familiarity with C++

## Similar roles

- [Senior/Staff Machine Learning Engineer - Offline Driving Intelligence](https://hotfix.jobs/jobs/577e2e7c-a8f4-4921-a5e7-8a1942c543f6) - Zoox - Foster City, CA - $242k – $333k/yr
- [Staff Software Engineer, ML Performance Optimization](https://hotfix.jobs/jobs/7f64aaee-53c4-4b47-9175-575e6f5e51c9) - Zoox - Foster City, CA - $242k – $389k/yr
- [Staff Software Engineer](https://hotfix.jobs/jobs/3ff71c05-9160-4c22-8a7d-cabd5b98e092) - Traba - New York, NY - $240k – $300k/yr
- [Staff Software Engineer](https://hotfix.jobs/jobs/60c76f48-09f3-413f-a9d8-8204704ba5fb) - Traba - New York, NY - $240k – $300k/yr
- [Senior Staff Machine Learning Engineer, Trust](https://hotfix.jobs/jobs/c7192112-1fc2-4b9e-a003-7c07858108f7) - Airbnb - Remote - $244k – $305k/yr

**Apply:** https://hotfix.jobs/jobs/6920b277-a720-4c46-b1bd-50d17bf4afb0
**Canonical:** https://hotfix.jobs/jobs/6920b277-a720-4c46-b1bd-50d17bf4afb0