figureai
AI Training Infrastructure Engineer
At a Glance
- Location
- San Jose, California, United States
- Experience
- 4+ years
- Compensation
- full-time position is between $150,000 - $350,000 annually. The pay offered for
- Posted
- 2026-03-19T16:56:17-04:00
Key Requirements
Required Skills
Domain Knowledge
- Cloud
- Engineering
- Robotics
Requirements
Strong software engineering fundamentals
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
Experience managing cloud infrastructure (AWS, Azure, GCP)
Experience with job scheduling / orchestration tools (SLURM, Kubernetes, LSF, etc.)
Compensation & Benefits
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.
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