Robotics ML Infra Engineer - Humanoid Whole-Body Control

Figure; San Jose (CA); On-site; USD 200,000 - 300,000; Full time

Job summary

An AI Robotics company in San Jose seeks an AI Training Infrastructure Engineer to develop the backbone for RL-based whole-body control systems. Responsibilities include scaling training infrastructure and optimizing performance for humanoid robots. Ideal candidates will have a strong background in Python, PyTorch, and robotics. The position offers a competitive salary between $200,000 and $300,000 annually and requires on-site collaboration in a rapidly evolving field.

Qualifications

• Production experience in Python and PyTorch.
• Experience building/training infrastructure for robotics or large-scale ML workloads.
• Familiarity with physics simulation tools.

Responsibilities

• Own and scale the infrastructure used to train control policies.
• Design fast, reliable systems for controls engineers.
• Ensure high cluster utilization with minimal downtime.

Skills

Software engineering fundamentals

Python

PyTorch

Physics simulation tools (e.g., MuJoCo)

Reinforcement learning

Ownership mindset

Tools

NVIDIA PhysX

PyBullet

Job description

An AI Robotics company in San Jose seeks an AI Training Infrastructure Engineer to develop the backbone for RL-based whole-body control systems. Responsibilities include scaling training infrastructure and optimizing performance for humanoid robots. Ideal candidates will have a strong background in Python, PyTorch, and robotics. The position offers a competitive salary between $200,000 and $300,000 annually and requires on-site collaboration in a rapidly evolving field.