AI Training Infrastructure Engineer Humanoid Whole Body Control
Figure is an AI robotics company developing general-purpose humanoid robots and its Helix vision-language-action AI system.
Funding history
About Figure
Figure develops and deploys general-purpose humanoid robots for commercial and household tasks. Its current Figure 03 robot is powered by Helix, an onboard generalist vision-language-action model; the company also operates the Index data-collection service.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will own and scale infrastructure for training whole-body control policies. You will design configurable and reliable systems, optimize cluster utilization and training performance, integrate physics simulations, and build tooling that moves policies from training through validation to hardware deployment.
Requirements
- Production software engineering experience with Python and PyTorch
- Experience building or scaling training infrastructure for robotics, control systems, or large-scale machine learning workloads
- Familiarity with NVIDIA PhysX, MuJoCo, Warp, or PyBullet
- Knowledge of dynamics, controls, and robotics systems
- Experience with reinforcement learning, imitation learning, or policy distillation
- Experience modeling contact interactions and photorealistic simulation environments for complex manipulation tasks
Responsibilities
- Own and scale infrastructure for training whole-body control policies
- Design fast, reliable, and highly configurable systems for controls engineers
- Ensure high cluster utilization and minimize downtime
- Evaluate and integrate physics engines, simulation environments, and parameterizations
- Optimize hyperparameters and infrastructure for training speed, efficiency, and model performance
- Build tooling to move policies from training through validation to hardware deployment