Engineering Manager, ML Performance Optimization
Hybrid (Foster City, CA)$276k - $343kFull-timePosted 2w ago
About the role
Engineering Manager leading ML performance optimization across Zoox's ML research teams, responsible for vision, platform design, and cross-functional collaboration to enable efficient training and low-latency on-vehicle inference for autonomous driving.
Responsibilities
- Vision: Develop and execute a strategic vision and roadmap for ML Training and Inference Performance Optimization, ensuring scalability, reliability, and performance to support autonomous driving.
- Technical acumen: Lead the design, implementation, and operation of a robust and efficient ML platform to enable the training, validation, serving, optimization and monitoring of ML models.
- ML Performance Optimization: Drive end-to-end performance optimization for large-scale model training and inference, including distributed training efficiency, GPU utilization, memory and communication optimization, model compression (quantization, pruning, distillation), and low-latency on-vehicle inference that meets strict real-time and compute budgets.
- Hiring: Attract, hire, and inspire a diverse world-class engineering team, fostering a culture of innovation, collaboration, and excellence.
- Partnership: Collaborate closely with cross-functional teams to define requirements and align on architectural decisions.
- Mentorship: Enable engineers on the team to grow their careers by providing the right opportunities and clear, timely feedback.
Qualifications
- 8+ years of relevant experience, including 3+ years of management experience managing engineers.
- Strong technical background in ML performance optimization (distributed training strategies, mixed-precision training, kernel-level optimization, compiler stacks, quantization, profiling/benchmarking across GPU and embedded accelerators).
- Experience building user-friendly ML Infrastructure for large-scale model training and high-throughput, low-latency serving.
- Experience with training frameworks like PyTorch, JAX, etc., leveraging GPUs for distributed model training.
- Experience with GPU-accelerated inference using TensorRT, Ray Serve, or similar frameworks.
- Proven track record of extensive cross-functional collaboration with research, product, hardware, and platform teams.
Skills
Required:ML performance optimizationML platform design and operationCross-functional collaboration
Benefits
- paid time off (sick leave, vacation, bereavement)
- unpaid time off
- Zoox Stock Appreciation Rights
- Amazon RSUs
- health insurance
- long-term care insurance
- long-term and short-term disability insurance
- life insurance
Last updated Oct 21, 2026. Interested? Apply on the company site.
About Zoox
It's Not a Car | Zoox
Zoox is a purpose-built autonomous vehicle designed for riders, not drivers. Learn more about the Zoox robotaxi and the future of ride-hailing.
autonomous vehiclerobotaxiride-hailing
More open roles at Zoox
- Contract Student Worker - Freeways Data Evaluation (20 hrs/wk)5d agoHybrid (Foster City, CA) · Zoox Internships – Software
- Software Engineer, ML Training Optimization5d agoFoster City, CA · Software Systems
- Strategic Sourcing Manager, IT Software & Services5d agoHybrid (Foster City, CA) · Supply Chain Quality and Reliability – Indirect Procurement
- Program Analyst5d agoFoster City, CA · Vehicle Development – Vehicle Program Management
- Hardware Engineer - Test Development5d agoFoster City, CA · Vehicle Development – Vehicle Integration, Testing & Validation