Helix AI Engineer, Reinforcement Learning

Figure AI

$130K — $180K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • 5-7 years of experience with reinforcement learning in complex environments
  • Proficient in Python and deep learning frameworks like PyTorch
  • Strong understanding of reinforcement learning fundamentals
  • Skilled in building scalable and reliable training systems
  • Experience training policies in both simulated and real-world applications
  • Familiarity with large-scale experimentation and distributed training systems
  • Ability to independently resolve complex technical challenges

Responsibilities

  • Design and implement reinforcement learning algorithms for robots in real and simulated environments
  • Train learning policies using interaction and feedback across various tasks
  • Develop strategies for reward modeling and exploration in long-horizon tasks
  • Enhance policy robustness against real-world challenges such as noise and variability
  • Work in both online and offline reinforcement learning contexts using extensive logged robot data
  • Collaborate with other teams to integrate reinforcement learning into a complete autonomy stack
  • Establish evaluation frameworks to assess policy performance and generalization

Benefits

  • Collaborative environment with cross-disciplinary teams
  • Opportunities to work on cutting-edge AI and robotics projects
  • Access to large-scale robot data and infrastructure
  • Potential exposure to top-tier AI research and technologies
  • Dynamic work culture focused on innovation and skill advancement
Full Job Description
Our Helix team is responsible for developing the core AI systems that power humanoid autonomy. We are looking for a Helix AI Engineer, Reinforcement Learning to develop learning systems that enable robots to acquire skills through interaction, feedback, and experience.

This role focuses on applying and advancing reinforcement learning across simulation and real-world environments-improving policy performance, robustness, and long-horizon decision-making in embodied systems.
Responsibilities
  • Design and implement reinforcement learning algorithms for embodied agents operating in real-world and simulated environments
  • Train policies that learn from interaction, feedback, and large-scale experience across diverse tasks
  • Develop reward modeling, credit assignment, and exploration strategies for complex, long-horizon behaviors
  • Improve policy robustness to real-world challenges such as noise, partial observability, and environment variability
  • Work across online and offline RL settings, including learning from large-scale logged robot data
  • Collaborate closely with pretraining, video, generative, agent, and robot learning teams to integrate RL into the full autonomy stack
  • Build scalable training systems for RL, including distributed rollouts, simulation infrastructure, and experiment management
  • Design evaluation frameworks to measure policy performance, stability, and generalization
Requirements
  • Experience developing and applying reinforcement learning algorithms in complex environments
  • Strong understanding of RL fundamentals (e.g., policy optimization, value methods, model-based RL)
  • Experience training policies in simulation and/or real-world systems
  • Proficiency in Python and deep learning frameworks such as PyTorch
  • Experience with large-scale experimentation and distributed training systems
  • Strong experimental rigor and ability to diagnose and improve learning systems
  • Solid software engineering skills and ability to build scalable, reliable systems
  • Ability to operate independently and drive ambiguous, high-impact technical problems
Bonus Qualifications
  • Experience applying RL to robotics, control systems, or embodied AI
  • Experience with large-scale RL infrastructure (distributed rollouts, simulation at scale)
  • Background in offline RL, imitation learning, or hybrid learning approaches
  • Experience with reward modeling or human-in-the-loop learning
  • Experience at leading AI labs such as OpenAI, Google DeepMind, Anthropic, or xAI
  • Familiarity with robotics systems, simulation environments, or real-world deployment constraints
  • Publication record in reinforcement learning, machine learning, or robotics

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.

Similar Jobs

More Jobs at Figure AI

More Information Technology Jobs

Find similar Helix AI Engineer, Reinforcement Learning jobs: