Senior/Staff Software Engineer, ML Training Performance
zoox
- Salary
- $254k - $350k
- Experience
- 7+ yrs
- Location
- Foster City or Seattle
- Job type
- Full-time
Required skills
- Megatron-LM
- TorchTitan
- Verl
- MILES
- NCCL
- NVIDIA Nsight
- PyTorch Profiler
- XLA
- JAX
- Python
- C++
About the role
7+ years total experience including 4+ years on large-scale model training platforms; deep experience with distributed training frameworks (Megatron-LM, TorchTitan), RL frameworks (Verl/MILES), collective communication libraries (NCCL), profiling tools (NVIDIA Nsight, PyTorch Profiler, XLA/JAX), and Python or C++.
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.
