AI Training Infrastructure Engineer, Figure

• The ideal candidate has experience building tools and infrastructure for a large-scale deep learning system
• 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

Desirable

• Our Helix team is looking for an experienced Training Infrastructure Engineer, to take our infrastructure to the next level
• This role is focused on managing the training cluster, implementing distributed training algorithms, data loaders, and developer tools for AI researchers
• 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