Skip to content
ZooxZooxFoster City, CA

Senior Machine Learning Engineer - ML Agents and Planning

Develops ML models using imitation and reinforcement learning for agent behavior prediction and autonomous vehicle planning. Requires PhD +1 year or MSc +5 years experience, expertise in RL, transformers, and production ML pipelines.

242k – 290k/yr
On-site1+ YOEML Engineering

About the role

Responsibilities

  • Develop deep learning models using imitation learning and reinforcement learning for human-like agent driving plans.
  • Create novel techniques to estimate driving plan quality (safety, progress, comfort, realism).
  • Contribute to large-scale ML infrastructure to discover solutions and advance the field.
  • Develop metrics and tools to analyze errors and improvements.
  • Collaborate with Perception, Planning, Simulation, and Validation engineers on autonomous driving.

Qualifications

  • PhD in computer science or related field +1 year professional experience (top-tier publications may substitute), or MSc +5 years professional experience.
  • Experience in Planning/Prediction using Reinforcement Learning.
  • Experience training and deploying transformer-based models.
  • Experience with production ML pipelines: dataset creation, training frameworks, metrics.
  • Fluency in Python; basic C++ understanding.

Bonus Qualifications

  • Top-tier publications (NeurIPS, ICML, CVPR).

Skills

Reinforcement LearningImitation LearningTransformersPythonC++Machine Learning PipelinesDeep LearningPlanningPredictionAutonomous Driving

Similar roles

ML Engineering jobs
Zoox

Senior Machine Learning Engineer - Perception Mapping

ZooxFoster City, CA

Design, train, and integrate ML models for semantic map element detection in autonomous vehicles. Requires 5+ years experience, MS/PhD in CS, expertise in computer vision, deep learning, and PyTorch.

242k – 290k/yr
On-site5+ YOEML Engineering
Zoox

Senior Machine Learning Engineer - Perception

ZooxFoster City, CA +1

Develops cutting-edge perception models using fused sensor data for autonomous vehicles, focusing on joint detection/tracking and embeddings for downstream systems. Requires MS/PhD in CS, PyTorch experience, and production deployment skills.

242k – 290k/yr
HybridML Engineering
Sigma

Senior AI/ML Engineer

SigmaNew York, NY +1

Build and productionize AI/ML systems including recommendations, NLP interfaces, and agentic workflows. Requires 10+ years building production-grade AI/ML systems and experience with foundation models.

240k – 270k/yr
On-site10+ YOEML Engineering
Attentive

Senior Machine Learning Engineer

AttentiveSan Francisco, CA +1

Builds and operates scalable production ML systems for real-time personalization and targeting. Requires 6+ years experience with Python, PyTorch/TensorFlow, data pipelines, and cross-functional collaboration.

244k – 320k/yr
Hybrid6+ YOEML Engineering
Sentry

Senior Software Engineer, AI Evals

SentrySan Francisco, CA

Build evaluation infrastructure for AI systems at Sentry, designing datasets, benchmarks, and test harnesses to measure accuracy and reliability of debugging agents. Requires 5+ years experience, Python/TypeScript proficiency, and AI/ML background.

240k – 280k/yr
Hybrid5+ YOEML Engineering