Helix AI Engineer, Generative 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, 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 US base salary range for this full-time position is between $200,000 - $400,000

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.