Software Engineer, ML Performance Optimization

What this role involves

Zoox is hiring a Software Engineer for ML Performance Optimization to accelerate and optimize ML models for autonomous driving. The role involves working with state-of-the-art accelerators and techniques like distributed training, quantization, distillation, and pruning. The engineer will collaborate with autonomy teams and contribute to the ML platform that supports autonomous robotaxis.

Skills this posting asks for

• machine learning
• distributed training
• quantization
• distillation
• pruning
• accelerators
• autonomous driving
• perception
• prediction
• planning
• simulation
• collision avoidance
• ml infrastructure
• model serving
• foundation models
• vlms
• vlas

From the employer’s posting

Zoox is on a mission to reimagine transportation and ground-up build autonomous robotaxis that are safe, reliable, clean, and enjoyable for everyone. We are still in the early stages of deploying our robotaxis on public roads, and it is a great time to join Zoox and have a significant impact in exe…