Helix AI Engineer, Training Infrastructure

Responsibilities

• Design, deploy, and maintain Figure's training clusters
• Architect and maintain scalable deep learning frameworks for training on massive robot datasets
• Work together with AI researchers to implement training of new model architectures at a large scale
• Implement distributed training and parallelization strategies to reduce model development cycles
• Implement tooling for data processing, model experimentation, and continuous integration

Requirements

• Strong software engineering fundamentals
• Bachelor's or Master's degree in Computer Science, Robotics, Engineering, or a related field
• Experience with Python and PyTorch
• Experience managing HPC clusters for deep neural network training
• Minimum of 4 years of professional, full-time experience building reliable backend systems

Preferred Qualifications

• Experience managing cloud infrastructure (AWS, Azure, GCP)
• Experience with job scheduling or orchestration tools such as SLURM, Kubernetes, or LSF
• Experience with configuration management tools like Ansible, Terraform, Puppet, or Chef

About the Company

Figure is an AI robotics company developing autonomous general-purpose humanoid robots. Our goal is to ship humanoid robots with human-level intelligence engineered to perform a variety of tasks in the home and commercial markets. We are headquartered in San Jose, CA.