# Senior/Staff Machine Learning Engineer - Offline Driving Intelligence

**Company:** [Zoox](https://hotfix.jobs/companies/zoox)
**Location:** Foster City, CA, Seattle, WA, San Diego, CA
**Role:** ML Engineering
**Salary:** $242k – $333k/yr
**Experience:** 5+ years
**Skills:** PyTorch, TensorFlow, JAX, Python, Deep Learning, Machine Learning, SQL, Distributed Computing, Bayesian Optimization, Foundation Models
**Posted:** 2025-11-13

> Leads ML initiatives to enhance validation processes using large-scale fleet and synthetic data for autonomous vehicle safety. Requires PhD or 5+ years ML experience, Python/PyTorch expertise, and AV domain knowledge.

## Job Description

## Responsibilities
- Lead technical initiatives applying modern machine learning and large-scale data to critical validation problems at the intersection of ML and data science. Serve as key contributor and tech lead on a small team.
- Improve feature representation by extending and refining features and embedding space to identify and cluster interesting driving scenarios using first-principles thinking.
- Integrate AV performance data and metrics into models for more accurate risk predictions.
- Collaborate cross-functionally with system safety, data science, software, and fleet operations teams.

## Qualifications
- PhD in a relevant field and/or **5+ years** of experience with machine learning models and data science in industry.
- Expertise in machine learning concepts, including training deep learning models, evaluation, and optimization.
- Strong programming skills in **Python** and experience with **PyTorch**, **TensorFlow**, **Jax**.
- Experience with large-scale data processing and distributed computing.
- Domain knowledge in robotics, autonomous vehicles, perception, prediction, and planning.
- Proven ability to drive progress independently, lead technical projects, and apply critical thinking.
- Excellent communication skills for cross-functional collaboration.

## Bonus Qualifications
- Real-world impact via patents, presentations, blog posts, publications at top ML conferences (Neurips, ICML, CORL, ICLR).
- Familiarity with encoder-decoder or foundation models for prediction and planning.
- Experience with test scripting and data analysis languages like **SQL**.
- Familiarity with fleet data collection and validation challenges in autonomous vehicles.
- Background in Bayesian optimization, online learning, and adaptive search.

## Similar roles

- [Senior / Staff Machine Learning Engineer - Perception Attributes](https://hotfix.jobs/jobs/6920b277-a720-4c46-b1bd-50d17bf4afb0) - 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/577e2e7c-a8f4-4921-a5e7-8a1942c543f6
**Canonical:** https://hotfix.jobs/jobs/577e2e7c-a8f4-4921-a5e7-8a1942c543f6