Lead ML Training Systems Engineer — Scalable Multimodal AI

Rhoda AI; Mountain View (CA); On-site; USD 140,000 - 180,000; Full time

Job summary

Rhoda AI is seeking a Staff / Principal ML Training Systems Engineer in Mountain View to enhance the training systems performance. This role focuses on large-scale multimodal training, driving efficiency, and scalability in compute use across thousands of GPUs.

The ideal candidate should have a proven track record in improving distributed training performance, significant experience with PyTorch, and strong systems intuition. Join us to contribute to the future of generalist robotics.

Qualifications

• Proven track record improving large-scale distributed training performance.
• Deep hands-on experience with modern ML stacks, PyTorch required.
• Strong understanding of scaling behavior across large GPU clusters.

Responsibilities

• Own training performance end-to-end.
• Design training systems with efficiency and communication improvements.
• Partner with researchers to translate innovations into efficient implementations.

Skills

Large-scale distributed training performance improvement

ML stacks (PyTorch)

Data / tensor / pipeline parallelism

Debugging and measurement ability

Systems intuition

Job description

Rhoda AI is seeking a Staff / Principal ML Training Systems Engineer in Mountain View to enhance the training systems performance. This role focuses on large-scale multimodal training, driving efficiency, and scalability in compute use across thousands of GPUs.

The ideal candidate should have a proven track record in improving distributed training performance, significant experience with PyTorch, and strong systems intuition. Join us to contribute to the future of generalist robotics.