Destava

Senior/Staff Software Engineer, ML Training Optimization

Zoox is hiring for Senior/Staff Software Engineer, ML Training Optimization in Foster City, CA. Staff level. First seen October 17, 2026; open as of November 11, 2026. Jobs taken straight from company career pages — no aggregator in between.

Save to applyRead more and apply on Zoox's site

With Destava: How well do you fit? · Tailor my résumé · Write a cover letter

Company
Zoox
Where
Foster City, CA
Level
Staff
First seen
October 17, 2026
Listed via
Lever

What the posting says

In this role, you will: Lead the design, implementation, and operation of training performance optimizations (data/tensor/pipeline parallelism, mixed precision, gradient checkpointing, GPU kernels) to scale VLM, VLA, and Foundation model training. Own the technical strategy and multi-quarter roadmap for RL framework and RL performance optimization for driving foundation models. Drive profiling and root-cause analysis of throughput bottlenecks across the compute, compiler, framework, and cluster-networking stack. Partner with ML researchers, data engineers, and hardware engineers to define…

Read more and apply on Zoox's site

This page includes information from the O*NET 31.0 Database by the U.S. Department of Labor, Employment and Training Administration (USDOL/ETA). Used under the CC BY 4.0 license. O*NET® is a trademark of USDOL/ETA.