AI Training Infrastructure Engineer – Humanoid Whole Body Control

AI Training Infrastructure Engineer – Humanoid Whole Body ControlSan Jose, CAFigure is an AI Robotics company developing autonomous general-purpose humanoid robots. The goal of the company is to ship humanoid robots with human level intelligence. Its robots are engineered to perform a variety of tasks in the home and commercial markets.

We are based in North San Jose, CA and require 5 days/week in-office collaboration. It's time to build.We're looking for an engineer to own the training and deployment backbone behind our RL-based whole-body control systems. This role sits at the intersection of robotics, machine learning, controls, and software systems engineering, and is critical to how quickly we can iterate, train, and deploy new capability to our fleet of humanoid robots.Key Responsibilities:Own and scale the infrastructure used to train whole-body control policies (simulation, data pipelines, orchestration, visualizations)Design systems that are fast, reliable, and highly configurable for our controls engineersEnsure high cluster utilization and minimal downtime—unblocking the team and accelerating iteration cyclesEvaluate and integrate physics engines, simulation environments, and parameterizations to balance realism and training speedOptimize hyperparameters and infrastructure to maximize training speed and efficiency and final model performanceBuild robust tooling to take policies from training ?

validation ? deployment on hardwareRequirements:Strong software engineering fundamentals with production experience in Python and PyTorchExperience building or scaling training infrastructure for robotics, control systems, or large-scale ML workloadsFamiliarity with physics simulation tools such as NVIDIA PhysX, MuJoCo, Warp, or PyBulletWorking knowledge of dynamics, controls, and robotics systemsExperience with reinforcement learning, imitation learning, or policy distillationStrong ownership mindset—you own systems that your teammates rely on every dayExperience modeling contact interactions and photorealistic simulation environments for complex manipulation tasksBonus Qualifications:Experience with humanoid or legged robot controlBackground in distributed systems, job schedulers, or cluster managementExperience deploying ML models or control policies to real-world systemsThe US base salary range for this full-time position is between $150,000 and $300,000 annually.The pay offered for this position may vary based on several individual factors, including job-related knowledge, skills, and experience. The total compensation package may also include additional components/benefits depending on the specific role.

This information will be shared if an employment offer is extended.