394,870open jobs
13,845companies
76,904added this week
Browse all
Salary
$291k – $350k per year
Location
Remote/Hybrid (Foster City, United States)
Seniority
Senior · 5+ years exp
Employment
Full-Time
Overview
Company
Impact
Profile match
Zoox is an American autonomous vehicle company founded in 2014 that designs a purpose-built robotaxi rather than retrofitting an existing car. Its vehicle has no steering wheel or driver position, seats four passengers facing each other, drives equally well in either direction and was developed alongside the sensors, compute and software that operate it, which required the company to certify the design under its own self-certification of federal motor vehicle safety standards. Amazon acquired the company in 2020 and it has since begun public robotaxi service in Las Vegas and San Francisco while testing in several other American cities.

The Onboard Behavior Model Architecture team is responsible for developing deep learning models that leverage data and compute at large scale to train driving models. We learn and predict behaviors from large scale expert data and large scale reinforcement learning to produce a ML driver that is safe, comfortable, and completes the mission. In this role, you will collaborate closely with the Onboard Perception, Cost Planner, Simulation, Validation, Data Science, Systems Engineering, QA, and ML Infra teams.

In this role, you will:

  • Lead a Team: Manage, mentor, and grow a team of individual contributors, fostering a culture of innovation and continuous improvement.

  • Develop Strategy: Develop and organize our overall strategy for Onboard Behavior ML Models for generating driving plans for our autonomous vehicle. You will interface with multiple partner teams to identify opportunities for model improvements within their problem area. You’ll be setting the short and long term technical direction for the team and collaborate on broader company-wide directions.

  • Provide technical guidance and leadership in the design and development of training models at large scale and work with partner teams on ensuring their efficient inference.

  • Monitor Performance: Establish and monitor key performance indicators (KPIs) to measure the effectiveness of work packages and drive continuous improvement.

  • Manage Resources: Manage the allocation of resources within the team, ensuring that projects are staffed appropriately and that team members have the necessary tools and support to succeed.

Qualifications

  • Expertise with Reinforcement Learning and Machine Learning for at least one of these areas: Planning, LLMs, VLAs/VLMs, recommendation systems.

  • Extensive experience with programming and algorithm design, strong mathematics skills.

  • MS or PhD degree in computer science or related field.

  • 5+ years of experience with production Machine Learning pipelines, with at least 3 years in a leadership or management role.

Bonus Qualifications

  • Conference or Journal publications in Machine Learning or Robotics related venues.

  • Prior experience working with autonomous vehicles or robotics, diffusion models, large scale training.

Free account
Stop reading job ads. Get the ones that fit.
One free account turns this page into a shortlist built around your stack, your level and your pay.
Match on every job. Stack, seniority, pay and location, scored against your profile.
394,870 open roles. Read straight off company career pages, refreshed every day.
Unlimited applications. Every one you send is tracked in one place, on-site or on a company board.
3 tailored CVs a month. Rewritten for the exact job you are applying to. Included free.
Create a free account
Free forever. No card. Under a minute.

Your match

How well do you fit this role?
Two answers are enough for a real match. No account needed.
Check my fit
Answers stay in this browser until you create an account.

Recommended for you based on this role

Similar stack
Same company
Foster City
In office • Full-Time • 5+ years exp • Master's Degree • Shanghai
C++
Python
AI/ML
AI Agents
Copilot
Cursor
LLM
Reinforcement Learning
Robotics
Isaac Lab
Isaac Sim
Perception
Reinforcement Learning
ROS
Sim-to-Real
Teleoperation
Apply
Chief of staff 1 day ago
$93k – $116k per year • Equity 0.5–1% • In office • Full-Time • 6+ years exp • Paris
AI/ML
AI Agents
Reinforcement Learning
Robotics
Digital Twin
Management
Jira
Notion
Slack
Apply
$60k – $300k per year • Equity 0.1–0.5% • Remote • Full-Time • 6+ years exp • San Francisco
Python
Rust
AI/ML
AI Agents
Contrastive Learning
Embeddings
Fine-tuning
Knowledge Distillation
Multimodal AI
PyTorch
Quantization
RAG
Reinforcement Learning
Reranking
Speech Recognition
Synthetic Data
Transformers
Whisper
Model Distillation
World Models
DevOps
Vector
Apply
$272k – $431k per year • In office • Full-Time • 15+ years exp • PhD • Santa Clara • New York
AI/ML
AI Agents
Fine-tuning
Function Calling
LLM
Multimodal AI
NVIDIA NeMo
Post-training
Pre-training
Reinforcement Learning
Structured Outputs
Synthetic Data
TGI
vLLM
DevOps
CI/CD
Git
Apply
$155k – $291k per year (Estimated) • Equity • Remote/Hybrid • Full-Time • 7+ years exp • Salt Lake City
Python
SQL
AI/ML
Reinforcement Learning
DevOps
AWS
CI/CD
Docker
GCP
GitHub
GitHub Actions
Kubernetes
Pulumi
Terraform
Apply
$170k – $267k per year • Remote/Hybrid • Full-Time • 3+ years exp • Bachelor's Degree • Foster City
Python
Databases
Databricks
AI/ML
LLM
Robotics
Sensor Fusion
Apply
$230k – $277k per year • Remote/Hybrid • Full-Time • 10+ years exp • Foster City
Databases
Databricks
AI/ML
AI Agents
DevOps
Amazon EKS
AWS
CI/CD
IAM
Platform Engineering
Self-Healing
Terraform
Kubernetes
Apply
$185k – $251k per year • Remote/Hybrid • Full-Time • Bachelor's Degree • Foster City
C++
Python
Databases
Databricks
AI/ML
Spark
Apply
$135k – $299k per year (Estimated) • In office • Contractor • Master's Degree • Foster City
Python
Scala
SQL
Python
pySpark
AI/ML
Spark
Apply
$160k – $215k per year • Remote/Hybrid • Full-Time • 3+ years exp • Bachelor's Degree • Foster City
C++
JavaScript
Kotlin
Python
Rust
TypeScript
Python
FastAPI
Flask
Frontend
React.js
Svelte
Vue.js
Apply
Cohort 0 2 days ago
In office • Full-Time • Bachelor's Degree • Foster City
AI/ML
AI Agents
Replit
Apply
$170k – $267k per year • Remote/Hybrid • Full-Time • 3+ years exp • Bachelor's Degree • Foster City
Python
Databases
Databricks
AI/ML
LLM
Robotics
Sensor Fusion
Apply
$215k – $230k per year • In office • 8+ years exp • Bachelor's Degree • Foster City
SQL
Apex
Apex
Salesforce Data Cloud
Databases
Snowflake
Analytics
ETL/ELT
Power BI
Tableau
Marketing
Salesforce
Apply
$175k – $195k per year • In office • 3+ years exp • Bachelor's Degree • Foster City
Apply
$215k – $230k per year • In office • 7+ years exp • Bachelor's Degree • Foster City
Apply
See all jobs
This is one of many
394,870 more open roles from verified company boards, updated every day.