# Senior/Staff Software Engineer, ML Training Performance at Zoox

- Company: Zoox
- What the company does: Zoox is a purpose-built autonomous vehicle designed for riders, not drivers. Learn more about the Zoox robotaxi and the future of ride-hailing. Backed by Lux.
- Company website: https://zoox.com
- Type: Startups (AI role)
- Level: Senior
- Location: Foster City, CA
- Work setup: On-site
- Pay: $254K to $350K base salary per year (USD)
- Posted: 2026-10-16
- Apply by: 2026-11-30
- Apply: https://jobs.lever.co/zoox/2ae1f38f-1b7e-4ec1-8a20-cc3816f96878?utm_source=1752vc&utm_medium=careers
- Page: https://www.1752.vc/careers/jobs/zoox-senior-staff-software-engineer-ml-training-performance/

## About the role

Zoox is on a mission to reimagine transportation and build ground-up autonomous robotaxis that are safe, reliable, clean, and enjoyable for everyone. We’re still early in deploying on public roads, so this is a great time to join and have a significant impact. The ML Platform team enables innovation in large-scale Foundation models, VLMs, and VLAs to make autonomous driving as seamless as possible. Found on 1752vc Careers, the job board for startup and VC roles.

## What they're looking for

- 7+ years of total experience , including 4+ years on large-scale model training platforms, with a record of leading projects end to end
- Deep experience with distributed training frameworks (Megatron-LM, TorchTitan) and efficiently utilizing GPU/accelerator clusters at scale
- Experience with RL frameworks (Verl/MILES) and algorithms (GRPO, PPO)
- Experience with collective communication libraries (e.g., NCCL) and optimizing communication/compute overlap in multi-node training
- Strong command of profiling tools (NVIDIA Nsight, PyTorch Profiler, or XLA/JAX profiling) to find and resolve training bottlenecks
- Proficiency in Python or C++

Tags: Software Systems

Source: 1752vc Careers, https://www.1752.vc/careers/jobs/zoox-senior-staff-software-engineer-ml-training-performance/