Foster City or Seattle
Senior/Staff Software Engineer, ML Training Performance
Job details
- Location
- Foster City or Seattle
- Work type
- Onsite
- Compensation
- $254,000 - $350,000/yr
- Posted
- 2 weeks ago
- Apply on
- jobs.lever.co
About this 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++.
What you'll do at Zoox:
- leading training optimizations
- driving performance analysis
- mentoring engineers
Apply to this Senior/Staff Software Engineer, ML Training Performance role at Zoox with a tailored resume on ApplyBolt.
Ready to apply to Zoox?
ApplyBolt finds matching jobs, tailors your resume, and submits applications for you.