In this role, you will:
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, including ML researchers, software engineers, data engineers, and hardware engineers, 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, such as distributed training strategies (data, tensor, pipeline parallelism, FSDP/ZeRO), mixed-precision training, kernel-level optimization (CUDA, Triton), compiler stacks (torch.compile, XLA, TVM), quantization, and profiling/benchmarking across GPU and embedded accelerators.
- Experience building user-friendly ML Infrastructure that enabled large-scale model training and high-throughput, low-latency serving use cases.
- 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, partnering with research, product, hardware, and platform teams to align priorities, influence technical direction, and deliver measurable performance improvements across organizational boundaries.
Skills Required
- 8+ years of relevant professional experience
- 3+ years of experience managing engineers
- Strong technical background in ML performance optimization, including distributed training, mixed-precision training, CUDA, Triton, compiler stacks, quantization, and profiling 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 such as PyTorch and JAX using GPUs for distributed model training
- Experience with GPU-accelerated inference using TensorRT, Ray Serve, or similar frameworks
- Proven extensive cross-functional collaboration with research, product, hardware, and platform teams
Zoox Compensation & Benefits Highlights
The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Zoox and has not been reviewed or approved by Zoox.
-
Healthcare Strength — Healthcare is extensive, with broad medical and vision options, company‑paid disability coverage, and multiple mental‑health resources. Feedback suggests coverage breadth and auxiliary programs support a wide range of needs.
-
Parental & Family Support — Family supports include paid parental leave, additional pregnancy disability time, fertility coverage, and adoption/surrogacy assistance. Backup care and family‑oriented programs further reinforce support across life stages.
-
Wellbeing & Lifestyle Benefits — Day‑to‑day perks are robust, including free daily meals, fitness subsidies, commuter support, and on‑site amenities. Feedback suggests these lifestyle benefits enhance convenience and workplace experience, especially for office‑based roles.
Zoox Insights
What We Do
Zoox is an autonomous mobility company that was founded to provide a safer, cleaner, and more enjoyable future on the road. To achieve that goal, the company has spent the past 10 years creating a purpose-built robotaxi that gives the world a better way to ride.
Why Work With Us
At Zoox, we are working to solve one of the greatest technological challenges of our generation. From the beginning, we have been focused on our goal of reimagining transportation from the ground up. We are a mission-driven community of innovators working together to create a safer, cleaner, and more enjoyable future on the road.
Gallery
