Skip to content
ZooxZooxFoster City, CA

Machine Learning Engineer

Develop deep learning models using imitation and reinforcement learning for autonomous driving agents and planning. Requires advanced degree plus experience with RL, transformers, and production ML pipelines.

189k – 270k/yr
Hybrid5+ YOEML Engineering

About the role

Responsibilities

  • Develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like agents.
  • Work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
  • Contribute to our 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 our systems.
  • Collaborate with engineers on Perception, Planning, Simulation, and Validation to solve the overall Autonomous Driving problem.

Requirements

  • PhD degree in computer science or related field or master's degree and 5+ years of professional experience in a relevant field.
  • Experience in Planning and / or Prediction using Reinforcement Learning techniques.
  • Experience with training and deploying transformer-based model architectures.
  • Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines.
  • Fluency in Python with a basic understanding of C++.

Nice-to-Haves

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

Skills

Reinforcement Learningimitation learningTransformersPythonC++machine learning pipelinesplanningpredictionDeep Learning

Similar roles

ML Engineering jobs
Zoox

Software Engineer - Behavior Capabilities

ZooxFoster City, CA

As a Software Engineer on the Behavior Capabilities team, you will develop and implement algorithmic advancements to expand the robot's driving abilities in complex scenarios, focusing on improving trip progress and vehicle uptime.

189k – 270k/yr
Hybrid4+ YOEML Engineering
Zoox

Software Engineer - Learned Trajectory Machine Learning Engineer

ZooxFoster City, CA

Develops deep learning models using imitation and reinforcement learning to generate safe, efficient trajectories for autonomous vehicles. Collaborates with perception and planning teams, leveraging large-scale ML infrastructure for production deployment in urban environments.

189k – 270k/yr
HybridML Engineering
Zoox

AI Engineer

ZooxFoster City, CA

Build and deploy production GenAI/LLM applications and intelligent agents to automate workflows across Procurement, Supply Chain, Legal, Finance, HR, and Marketing. Requires 5+ years engineering experience including 2+ years production LLMs, Python, LangChain/LlamaIndex, and cloud AI/vector DB tools.

190k – 250k/yr
Hybrid5+ YOEML Engineering
Norm Ai

Software Engineer, Applied AI

Norm AiNew York, NY

Build and ship reliable, scalable LLM-powered AI agents for high-stakes legal workflows at an agentic law platform. Requires 4+ years software/AI engineering experience, strong Python skills, production ML/LLM integration experience, and AI fluency.

190k – 280k/yr
Hybrid4+ YOEML Engineering
Sesame

ML Engineer

SesameSan Francisco, CA +2

Research Engineer building and deploying production voice and multimodal ML models. Requires expert PyTorch, large-scale model training experience, and shipping user-facing ML systems.

190k – 320k/yr
On-site5+ YOEML Engineering