Engineering Manager, ML Performance Optimization
zoox
- Salary
- $276k - $343k
- Experience
- 8+ yrs
- Location
- Foster City, California, United States
- Job type
- Full-time
Required skills
- PyTorch
- JAX
- TensorRT
- Ray Serve
- CUDA
- Triton
- torch.compile
- XLA
- TVM
About the role
8+ years relevant experience including 3+ years managing engineers; strong technical background in ML performance optimization, distributed training, GPU acceleration, and ML infrastructure; experience with PyTorch, JAX, TensorRT, Ray Serve.
About zoox
Autonomous mobility company developing a fully electric robotaxi fleet.
This page is fully interactive when JavaScript is enabled. Please enable JavaScript to apply or browse related roles.
