Figure.ai Logo

Figure.ai

Helix AI Engineer, Reinforcement Learning

Reposted 29 Days Ago
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 San Jose, California, USA Office

San Jose, CA, United States

Similar Jobs

3 Hours Ago
Hybrid
San Francisco, CA, USA
175K-200K Annually
Senior level
175K-200K Annually
Senior level
Fintech • Mobile • Social Impact • Financial Services
Build, test, deploy, and monitor ML models for credit risk, fraud detection, and transaction timing. Create training datasets, run experiments and A/B tests, analyze customer trends, and collaborate with Engineering, Product, and Business teams to deliver end-to-end data science solutions.
Top Skills: JupyterlabMatplotlibPandasPythonScikit-LearnSQLStatsmodels
3 Hours Ago
Hybrid
San Francisco, CA, USA
185K-215K Annually
Senior level
185K-215K Annually
Senior level
Fintech • Mobile • Social Impact • Financial Services
Lead development, deployment, and monitoring of underwriting, risk, fraud, and payments ML models. Own full modeling lifecycle, feature engineering, validation, A/B testing, cross-functional collaboration, and mentor junior data scientists.
Top Skills: A/B TestingJupyterlabMatplotlibPandasPythonScikit-LearnSQLStatsmodels
3 Hours Ago
Hybrid
33-50 Hourly
Mid level
33-50 Hourly
Mid level
Automotive • Hardware • Robotics • Software • Transportation • Manufacturing
Program, troubleshoot, and maintain PLC-based automation (primarily Rockwell) to optimize uptime, quality, and throughput. Integrate FANUC robots, welding/laser systems, vision-assisted handling, HMIs (Ignition/Rockwell), industrial networks, VFD/motion control, and data historian archiving. Perform mechanical/pneumatic/hydraulic support, preventive maintenance, root-cause analysis, continuous improvement, and coordinate apprentices/contractors.
Top Skills: AutocadData MatrixEthernetFanucHistorianHmiIgnitionIndustrial RoboticsIo-LinkLaser CuttingMachine VisionMig WeldingPlcProfibusProfinetRockwell PlcSafety PlcServo Motion ControlVfd

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