Skip to content
ZooxZooxFoster City, CA

Principal Software Engineer - Autonomy Behaviors

Lead technical development of ML algorithms for next-generation ML Planner. Drive innovations in imitation learning, reinforcement learning, and model scaling while mentoring ML developers.

324k – 492k/yr
Hybrid10+ YOEML Engineering

About the role

Responsibilities

  • Set the technical direction of ML development for the future of our Planner
  • Push the frontiers of imitation learning, reinforcement learning, and model scaling to develop new ML models for our onboard Planner
  • Identify and drive force-multipliers for ML development, including metrics, data, and training frameworks
  • Develop new methods to leverage Zoox’s Foundation Models into the ML Planner
  • Provide technical mentorship to the broader group of ML developers at Zoox
  • Collaborate with engineers across Autonomy Software, Simulation, System Design and Mission Assurance, and other departments to solve the overall Autonomous Driving problem in complex urban environments

Requirements

  • BS, MS, or PhD degree in computer science or related field
  • Proven track record of technical leadership in large-scale AI organizations
  • Significant experience with training and deploying Deep Learning models to production
  • Fluency in C++ or fluency in Python with a basic understanding of C++
  • Extensive experience with programming and algorithm design
  • Strong mathematics skills
  • 10+ years of experience

Nice-to-Haves

  • Conference or Journal publications in Machine Learning or Robotics related venues
  • Prior experience with Prediction and Planning for autonomous vehicles

Skills

PythonC++Deep LearningMachine LearningImitation LearningReinforcement LearningModel ScalingAlgorithm DesignMathematics

Similar roles

ML Engineering jobs
Shield AI

Principal Engineer, AI and Data Platform Engineering (R4941)

Shield AISan Francisco, CA

Leads the design, build, and operation of AI and data platform for autonomy systems, managing training, simulation, data pipelines, MLOps, and deployment across on-premise, cloud, and edge environments. Requires deep expertise in scalable ML infrastructure and compute strategy.

320k – 490k/yr
On-siteML Engineering
Square

Principal Engineer, AI Systems

SquareUnited States

Principal-level IC building and scaling production autonomous agents and agentic workflows across Block's ecosystem using frontier LLMs. Requires 15+ years experience shipping AI systems from zero to production scale.

319k – 479k/yr
Remote15+ YOEML Engineering
Pinterest

Principal Machine Learning Engineer, Ads Delivery

PinterestSan Francisco, CA +2

Develop backend systems, statistical models, and experiments to optimize Pinterest's ads marketplace, balancing short- and long-term objectives. Requires 10+ years experience, CS/ML degree, and strong engineering/math skills.

315k – 551k/yr
Hybrid10+ YOEML Engineering
OpenAI

Principal Software Engineer, Codex Cyber

OpenAISan Francisco, CA

Principal engineer on OpenAI's Codex Cyber team building AI-powered security products for trusted defenders. Define roadmaps, shape model training and safeguards, build evaluations for cyber capabilities and risks, and translate frontier research into production tools while collaborating across product, research, safety, and external partners.

347k – 490k/yr
On-site10+ YOEML Engineering
Airbnb

Principal Machine Learning Engineer- LLM Fine-tuning and Optimization

AirbnbUnited States

Principal MLE focused on fine-tuning and optimizing large language models for production, building scalable AI pipelines, and shaping Airbnb's ML platform strategy. Requires 10+ years experience and a PhD.

292k – 365k/yr
Remote10+ YOEML Engineering