Figure
Figure’s vision is to deploy autonomous humanoids at a global scale. 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.
Responsibilities
- Design, deploy, and maintain Figure’s training clusters
- Architect, optimize, 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, advanced parallelization strategies, and high-performance data loaders to reduce model development cycles
- Profile, identify, and eliminate training bottlenecks at the hardware and software levels to maximize Model FLOPs Utilization (MFU)
- Implement tooling for data processing, model experimentation, and continuous integration
- Strong software engineering fundamentals
- Bachelor’s or Master’s degree in Computer Science, Robotics, Engineering, or a related field
- Extensive professional experience with Python and PyTorch
- Proven track record of scaling and running large-scale training experiments personally on 800+ GPUs
- Experience managing HPC clusters for deep neural network training
- Minimum of 4 years of professional, full-time experience building reliable backend systems and infrastructure
- Experience contributing to or maintaining open-source distributed training frameworks (Megatron-LM, DeepSpeed, TorchTitan)
- Experience managing cloud infrastructure (AWS, Azure, GCP)
- Experience with job scheduling / orchestration tools (SLURM, Kubernetes, LSF, etc.)
- Experience with configuration management tools (Ansible, Terraform, Puppet, Chef, etc.)
- Deep understanding of CUDA and hands-on experience writing custom GPU kernels to optimize training
The US base salary range for this full-time position is between $150,000 – $350,000 annually.
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.
To apply for this job please visit job-boards.greenhouse.io.
Explore Figure online
Working in SAN JOSE
Market context
- Similar listings in San Jose, CA 1
Job details above are provided by the employer/source. The sections on this page are compiled from public data sources with AI assistance.
Accommodations: if you need a workplace accommodation to apply for or perform this job, see ADA.gov or EEOC.gov for guidance on your rights and how to request one.
Keep exploring on Get A Job.ai
Not quite the right fit? Your next opportunity is a click away.
- Browse all jobs
- More jobs by category
- Remote jobs you can do from anywhere
- Research typical pay for this role
- Set a job alert so new matches reach you first
- Upload your resume to apply faster
Hiring instead? Post a job and reach candidates searching right now.