Helix AI Engineer, Generative AI

About this role

Figure AI is seeking a Helix AI Engineer to develop and deploy large-scale generative models—including diffusion and multimodal approaches—that enable humanoid robots to perceive, reason, and interact with the physical world. The role involves training models for vision, video, and perception tasks, optimizing distributed training pipelines, and integrating generative systems into the full autonomy stack.

What you'll do

• Design, train, and deploy large-scale generative models with focus on diffusion-based approaches for vision, video, and multimodal data
• Develop models that enhance robot perception, world modeling, and prediction from sensory inputs
• Build generative systems for synthetic data creation and dataset scaling for robot learning
• Optimize training pipelines for generative models across distributed systems
• Integrate generative models into the autonomy stack working with data, infrastructure, and agent teams
• Evaluate model quality, robustness, and generalization in real-world robot scenarios

What they're looking for

• Generative modeling (diffusion, autoregressive methods)
• Deep learning for vision and multimodal systems
• Python and PyTorch
• Large-scale distributed training
• Experimental rigor and rapid iteration
• Software engineering and system design
• Independent problem-solving