Helix AI Engineer, Generative AI

Figureai; San Jose (CA); On-site; USD 120,000 - 160,000; Full time

Job summary

Figureai, headquartered in San Jose, CA, is seeking a Helix AI Engineer to design and deploy generative models that enhance robotic understanding and interaction with the real world.

The ideal candidate will have experience with training generative models, be proficient in Python and deep learning frameworks like PyTorch, and possess strong software engineering skills. This position is in-office 5 days a week, focusing on building more capable humanoid robots through advanced AI techniques.

Qualifications

• Experience training and deploying generative models (diffusion, autoregressive, etc.) at scale.
• Strong understanding of deep learning techniques for vision and/or multimodal systems.
• Proficiency in Python and in using frameworks like PyTorch.
• Experience with large-scale datasets and distributed training systems.
• Ability to iterate quickly on model performance.
• Solid software engineering skills to build reliable systems.
• Ability to own ambiguous, high-impact technical problems.

Responsibilities

• Design, train, and deploy large-scale generative models.
• Improve robot perception and world modeling.
• Create synthetic data and scale datasets for robot learning.
• Explore techniques in generative modeling and multimodal systems.
• Optimize training pipelines across distributed systems.
• Integrate generative models into the autonomy stack.
• Evaluate model quality and generalization in real-world scenarios.
• Contribute to scalable experimentation frameworks.

Skills

Training and deploying generative models

Deep learning techniques

Python proficiency

Large-scale datasets management

Software engineering skills

Independent problem solving

Tools

PyTorch

Job description

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, Generative AI to build and scale generative models that enable robots to understand, simulate, and interact with the physical world. This role focuses on training and deploying diffusion and generative models across vision, video, and multimodal domains, with applications spanning perception, data generation, and model-based reasoning.

Responsibilities

• Design, train, and deploy large-scale generative models, with a focus on diffusion-based approaches for vision, video, and multimodal data
• Develop models that 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
• Explore and implement state-of-the-art techniques in diffusion, generative modeling, and multimodal foundation models
• Optimize training pipelines for large-scale generative models across distributed systems
• Work closely with data, training infrastructure, and agent teams to integrate generative models into the full autonomy stack
• Evaluate model quality, robustness, and generalization across real-world scenarios
• Contribute to the design of scalable experimentation frameworks for generative model development

Requirements

• Experience training and deploying generative models (diffusion, autoregressive, or related approaches) 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
• Strong experimental rigor and ability to iterate quickly on model performance
• Solid software engineering skills and ability to build reliable, maintainable systems
• Ability to operate independently and own ambiguous, high-impact technical problems

Bonus 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 (multi-node, GPU clusters, etc.)
• Familiarity with 3D, video prediction, or world models
• Prior work in robotics, embodied AI, or real-world ML systems
• Publication record in machine learning, computer vision, or generative modeling

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.