Senior ML Training Systems Engineer, Large-Scale Robotics

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

Job summary

Rhoda AI in Mountain View is seeking a Staff / Principal ML Training Systems Engineer to lead the performance of large-scale multimodal training systems. This role involves improving training efficiency and collaborating closely with research teams to accelerate model iteration.

The ideal candidate will have proven experience in boosting distributed training performance and hands-on skills with modern ML platforms like PyTorch. You will contribute significantly to the scalability and effectiveness of our robotics systems, driving impactful innovations.

Qualifications

• Proven track record in distributed training performance.
• Comfortable in a fast-moving environment.
• Strong ability to identify and address bottlenecks.

Responsibilities

• Own training performance end-to-end.
• Design training systems and improve execution efficiency.
• Make performance observable and measurable.

Skills

Large-scale distributed training performance improvement

Hands-on experience with modern ML stacks (PyTorch required)

Understanding of data / tensor / pipeline parallelism

Exceptional debugging and measurement ability

Tools

PyTorch

JAX

Job description

Rhoda AI in Mountain View is seeking a Staff / Principal ML Training Systems Engineer to lead the performance of large-scale multimodal training systems. This role involves improving training efficiency and collaborating closely with research teams to accelerate model iteration.

The ideal candidate will have proven experience in boosting distributed training performance and hands-on skills with modern ML platforms like PyTorch. You will contribute significantly to the scalability and effectiveness of our robotics systems, driving impactful innovations.