Helix AI Engineer, Generative AI

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