Research Member of Technical Staff - Training Platform
Rhoda AI · Mountain View, CA · 1 wk ago
On-siteEngineeringFull-time
What You'll Do
- Build and maintain training orchestration systems for large-scale distributed model training across GPU clusters
- Develop experiment management tooling: job configuration, tracking, reproducibility, and artifact management
- Create observability infrastructure for training runs: loss curves, compute utilization, gradient statistics, and anomaly detection
- Optimize and automate the research iteration loop from experiment launch to results analysis
- Manage job scheduling and cluster utilization for efficient use of GPU compute
- Build internal tooling and interfaces that help researchers move faster
- Collaborate with training systems, data infrastructure, and research teams to support their platform needs
What We're Looking For
- Strong software engineering skills with experience in MLOps or ML platform engineering
- Familiarity with distributed training frameworks (PyTorch DDP, FSDP, DeepSpeed, Megatron, or similar)
- Experience building experiment tracking, reproducibility, and artifact management systems
- Comfortable managing and operating GPU cluster environments (Slurm, Kubernetes, or similar)
- Strong reliability engineering instincts: monitoring, alerting, and failure recovery
Nice To Have (But Not Required)
- Experience with training orchestration tools (Slurm, Ray, Kubernetes, or similar schedulers)
- Familiarity with experiment tracking tools (Weights & Biases, MLflow, or custom solutions)
- Experience supporting large model training pipelines (LLMs, VLMs, or video models)
- Understanding of parallelism strategies and how they affect training efficiency and debugging
- Experience with cloud-based training infrastructure (AWS, GCP, or Azure)