Engineering Manager, ML Performance Optimization
Core
Lead the design and operation of an ML platform to enable efficient training, validation, serving, and optimization of models for autonomous robotaxis.
Role type
Senior Engineering Manager, ML Performance Optimization (via careerplan.io/jobs/222978c8-80d6-4a55-aa40-3bb6eb15671c-engineering-manager-ml-performance-optimization-at-zoox)
Builds
Scalable ML training and inference infrastructure for autonomous driving
Domain
Autonomous vehicles / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
Distributed training strategies, Kernel-level optimization, Model compression, Low-latency inference, GPU utilization, Cross-functional leadership
Preferred skills
Compiler stack expertise, Embedded accelerator optimization, Team hiring and mentorship
Technologies
PyTorch, JAX, CUDA, Triton, torch.compile, XLA, TVM, TensorRT, Ray Serve
Responsibilities
Develop strategic vision for ML performance optimization; Lead design and operation of robust ML platforms; Drive end-to-end performance optimization for training and inference; Hire and mentor world-class engineering teams; Collaborate with cross-functional teams on architectural decisions.
Seniority
Senior, hands-on IC with management responsibilities
