Figure.ai Logo

Figure.ai

Helix AI Engineer, Reinforcement Learning

Reposted 9 Days Ago
Be an Early Applicant
In-Office
San Jose, CA, USA
Mid level
In-Office
San Jose, CA, USA
Mid level
Develop learning systems using reinforcement learning for autonomous robots, focusing on policy optimization and real-world application.
The summary above was generated by AI

Figure is an AI robotics company developing autonomous general-purpose humanoid robots. Our goal is to build embodied AI systems that can perceive, reason, and act in the real world. Figure is headquartered in San Jose, CA, and this role requires 5 days/week in-office collaboration.

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. 

HQ

Figure.ai Sunnyvale, California, USA Office

Sunnyvale, California, United States

Similar Jobs

An Hour Ago
In-Office
101K-134K Annually
Mid level
101K-134K Annually
Mid level
Gaming
Responsibilities include developing scalable server-side features, optimizing performance, and collaborating with teams to enhance online services for games.
Top Skills: .NetApi DesignC#IisSQL ServerWindows Server
An Hour Ago
Easy Apply
Hybrid
San Francisco, CA, USA
Easy Apply
200K-350K Annually
Senior level
200K-350K Annually
Senior level
Fintech • Financial Services
The Director will manage client relationships, execute transactions, and drive revenue growth in the private markets, focusing on institutional sales and private placements.
Top Skills: Finra Series 63Finra Series 7
An Hour Ago
Remote or Hybrid
US
110K-130K Annually
Senior level
110K-130K Annually
Senior level
Big Data • Fintech • Information Technology • Insurance • Software
The role involves leading bug fixes and solutions, collaborating with engineering and product teams, mentoring peers, building documentation, and ensuring product integrity for partners.
Top Skills: GoMicroservice ArchitecturePythonTypescript

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account