Skip to content
ZooxZooxFoster City, CA

Software Engineer - Learned Trajectory Machine Learning Engineer

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

About the role

Responsibilities

  • Develop new deep learning models using imitation learning and reinforcement learning to generate driving plans for autonomous vehicles.
  • Work on techniques to estimate the quality of driving plans along dimensions of safety, progress, comfort, etc.
  • Leverage large-scale machine learning infrastructure to discover new solutions and push field boundaries.
  • Develop metrics and tools to analyze errors and understand system improvements.
  • Collaborate with engineers on Perception, Planning, and Simulation to solve autonomous driving in complex urban environments.

Qualifications

  • BS, MS, or PhD degree in computer science or related field.
  • Experience with training and deploying transformer-based model architectures and reinforcement learning.
  • Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines.
  • Fluency in C++ or fluency in Python with basic understanding of C++.
  • Extensive experience with programming and algorithm design.
  • Strong mathematics skills.

Bonus Qualifications

  • Conference or Journal publications in Machine Learning or Robotics related venues.
  • Prior experience with Prediction and/or autonomous vehicles or robotics in general.

Skills

PyTorchTensorFlowTransformersReinforcement LearningImitation LearningC++PythonMachine Learning PipelinesDeep LearningComputer Vision

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

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
Parafin

Data Scientist, Underwriting

ParafinSan Francisco, CA

Develops and deploys production machine learning models for underwriting small business loans, focusing on risk assessment, pricing optimization, and automated systems. Requires strong statistical reasoning, ML expertise, engineering skills, and business acumen.

190k – 300k/yr
HybridML Engineering