Skip to content
ZooxZooxFoster City, CA

Manager, Onboard Behavior Encoder

Lead and grow a team developing large-scale Reinforcement Learning and ML models for onboard autonomous vehicle behavior and driving plans. Requires MS/PhD, 5+ years production ML experience (3+ in leadership), and expertise in RL/ML for planning, LLMs, or related areas.

277k – 349k/yr
Hybrid5+ YOEML Engineering

About the role

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 overall strategy for Onboard Behavior ML Models for generating driving plans for autonomous vehicles.
  • Interface with multiple partner teams (Onboard Perception, Cost Planner, Simulation, Validation, Data Science, Systems Engineering, QA, and ML Infra) to identify opportunities for model improvements.
  • Set short and long term technical direction for the team and collaborate on broader company-wide directions.

Technical Guidance

  • Provide technical guidance and leadership in the design and development of training models at large scale.
  • Work with partner teams on ensuring 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.

Skills

Reinforcement LearningMachine LearningplanningLLMsvlasvlmsRecommendation SystemsPythonalgorithm designmathematicsproduction ml pipelineslarge scale training

Similar roles

ML Engineering jobs
Zoox

Manager, RL Algorithms & Decoder

ZooxFoster City, CA

Lead a team developing large-scale RL and ML models for autonomous vehicle behavior planning and driving decisions. Requires RL/ML expertise, production ML pipeline experience, and 3+ years in leadership.

277k – 349k/yr
Hybrid5+ YOEML Engineering
Anthropic

TPU Kernel Engineer

AnthropicSan Francisco, CA +2

Designs and optimizes TPU kernels to address performance issues in ML research, training, and inference systems. Provides feedback on model impacts and solves large-scale systems problems, requiring deep accelerator expertise.

280k – 850k/yr
HybridML Engineering
Fluidstack

Machine Learning Engineer

FluidstackAustin, TX +3

Build and own ML/LLM systems for internal operations including forecasting, risk flagging, and document extraction. Ship production agentic systems end-to-end with guardrails and partner with data engineering to integrate predictions into tools.

269k – 317k/yr
On-site5+ YOEML Engineering
OpenAI

Machine Learning Engineer, Integrity

OpenAISan Francisco, CA

Deploys state-of-the-art ML models and fine-tunes LLMs in production to enhance platform integrity and safety against adversarial threats. Requires Master's/PhD, deep learning expertise, PyTorch/TensorFlow proficiency.

266k – 555k/yr
On-siteML Engineering
OpenAI

Software Engineer, Workload Enablement

OpenAISan Francisco, CA +1

Software Engineer enabling production AI workloads on new hardware platforms through porting, benchmarking, stress testing, and performance optimization. Requires 5+ years in ML systems, distributed training, PyTorch, and RDMA/NCCL expertise.

293k – 455k/yr
Hybrid5+ YOEML Engineering