Figure

AI Training Infrastructure Engineer - Humanoid Whole Body Control

Figure$200K — $300K *
Consumer Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Strong software engineering fundamentals with production experience in Python and PyTorch
  • Experience building or scaling training infrastructure for robotics, control systems, or large-scale ML workloads
  • Familiarity with physics simulation tools such as NVIDIA PhysX, MuJoCo, Warp, or PyBullet
  • Working knowledge of dynamics, controls, and robotics systems
  • Experience with reinforcement learning, imitation learning, or policy distillation
  • Strong ownership mindset with a focus on system reliability
  • Experience modeling contact interactions and photorealistic simulation environments for complex manipulation tasks

Responsibilities

  • Own and scale the infrastructure for training whole-body control policies
  • Design fast, reliable, and configurable systems for control engineers
  • Ensure high cluster utilization and minimal downtime to support team efficiency
  • Evaluate and integrate various physics engines and simulation environments
  • Optimize hyperparameters and infrastructure for maximum training efficiency
  • Build robust tooling for transitioning policies from training to deployment on hardware

Benefits

  • Comprehensive health benefits
  • Flexible work hours
  • Collaborative work environment
  • Professional development opportunities
  • Access to cutting-edge technology and tools
Full Job Description
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 engineers
  • Ensure high cluster utilization and minimal downtime-unblocking the team and accelerating iteration cycles
  • Evaluate and integrate physics engines, simulation environments, and parameterizations to balance realism and training speed
  • Optimize hyperparameters and infrastructure to maximize training speed and efficiency and final model performance
  • Build robust tooling to take policies from training  validation  deployment on hardware

Requirements:
  • Strong software engineering fundamentals with production experience in Python and PyTorch
  • Experience building or scaling training infrastructure for robotics, control systems, or large-scale ML workloads
  • Familiarity with physics simulation tools such as NVIDIA PhysX, MuJoCo, Warp, or PyBullet
  • Working knowledge of dynamics, controls, and robotics systems
  • Experience with reinforcement learning, imitation learning, or policy distillation
  • Strong ownership mindset-you own systems that your teammates rely on every day
  • Experience modeling contact interactions and photorealistic simulation environments for complex manipulation tasks

Bonus Qualifications:
  • Experience with humanoid or legged robot control
  • Background in distributed systems, job schedulers, or cluster management
  • Experience deploying ML models or control policies to real-world systems

The US base salary range for this full-time position is between $200,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.

About Figure

Figure is a financial technology company that provides home equity loans and other financial services. The company was founded in 2018 by Mike Cagney, a former CEO of the online lending platform, SoFi. Figure's platform uses blockchain technology to streamline the loan application process and reduce costs. The company has raised over $225 million in funding and has partnerships with several major financial institutions, including Jefferies and WSFS Bank. Figure was named one of the 50 most innovative fintech companies in the world by Forbes in 2019.
Learn more about Figure
Size
200 employees
Industry
Founded
2018

Similar Jobs

More Jobs at Figure

More Consumer Technology Jobs

Find similar AI Training Infrastructure Engineer - Humanoid Whole Body Control jobs: