Helix AI Engineer, Generative AI

Responsibilities

• Design, train, and deploy large-scale generative models, focusing on diffusion-based approaches for vision, video, and multimodal data
• Develop models to improve robot perception, world modeling, and prediction from raw sensory inputs
• Build generative systems for synthetic data creation, augmentation, and dataset scaling for robot learning
• Optimize training pipelines for large-scale generative models across distributed systems
• Work closely with data, training infrastructure, and agent teams to integrate models into the full autonomy stack
• Evaluate model quality, robustness, and generalization across real-world scenarios

Requirements

• Experience training and deploying generative models (diffusion, autoregressive, or related) at scale
• Strong understanding of modern deep learning techniques for vision and/or multimodal systems
• Proficiency in Python and deep learning frameworks such as PyTorch
• Experience working with large-scale datasets and distributed training systems
• Solid software engineering skills and the ability to build reliable, maintainable systems
• Ability to operate independently and own ambiguous, high-impact technical problems

Preferred Qualifications

• Experience with diffusion models for image or video generation
• Experience with multimodal foundation models (vision-language or vision-language-action)
• Background in synthetic data generation or simulation for robotics or embodied AI
• Experience optimizing large-scale training on multi-node GPU clusters
• Familiarity with 3D, video prediction, or world models
• Publication record in machine learning, computer vision, or generative modeling

About the Company

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.