- Salary
- $150k - $450k
- Experience
- 5+ yrs
- Location
- Sunnyvale, California, United States
- Job type
- Full-time
Required skills
- DeepSpeed
- FSDP
- FairScale
- Horovod
- vLLM
- SGLang
- PyTorch
- JAX
- Slurm
- Kubernetes
- Ray
- NCCL
- GLOO
- CUDA
- Triton
About the role
5+ years ML systems/infra/distributed training experience; experience modifying distributed ML frameworks (DeepSpeed, FSDP, FairScale, Horovod); strong Python and systems design; multi-node experience (Slurm, Kubernetes, Ray); LLM RL pipeline ownership; PPO/GRPO knowledge.
About MBZUAI (Mohamed bin Zayed University of Artificial Intelligence)
Government-funded UAE public university educating students and conducting advanced artificial-intelligence research.
This page is fully interactive when JavaScript is enabled. Please enable JavaScript to apply or browse related roles.
