Helix AI Engineer, Pretraining

About this role

Figure AI is seeking a Helix AI Engineer to develop large-scale foundation models that power autonomous humanoid robots. You'll design pretraining strategies across multimodal data and build distributed training systems that enable robots to perceive, reason, and act in the real world.

What you'll do

• Design and train large-scale multimodal foundation models using text, vision, and robot-collected data
• Develop pretraining strategies to improve generalization and transfer to embodied AI tasks
• Implement and explore transformer-based and emerging foundation model architectures
• Build and optimize distributed training pipelines across multi-node GPU clusters
• Collaborate with video, generative, agent, and robot learning teams to integrate models into the autonomy stack
• Design evaluation frameworks and contribute to post-training approaches like fine-tuning and alignment

What they're looking for

• Large-scale foundation model training experience
• Deep learning architectures, especially transformers
• Distributed training and optimization at scale
• Python and PyTorch proficiency
• Experimental rigor and model design iteration
• Scalable software engineering practices
• Multimodal pretraining (bonus)
• Scaling laws and dataset curation (bonus)