Search
Go
This job is no longer accepting applications
This listing was closed on Sep 21, 2026.
Browse Open JobsHelix AI Engineer, Training Performance
Summary by AIEngineer.careers
Helix AI Engineer, Training Performance is a AI Infrastructure Engineer role (full-time). with Figure. in SAN JOSE, US. Compensation shown: $200K–$400K. Imported listing (source: newsbreak.com). Apply on the employer's site (newsbreak.com).
Imported listing (source: newsbreak.com) · Apply on newsbreak.com
SAN JOSE · US
$200K–$400K
Full-time
The sections below reproduce the third-party job description for reference. AIEngineer.careers does not write or control this text.
Imported job description
Sourced from newsbreak.com
Optimize the training performance of 100B+ parameter models on 100k+ GPU clusters for humanoid robots at Figure. The role involves tuning distributed training systems, writing custom CUDA kernels, and evaluating new hardware accelerators. This is a high-impact engineering position focused on scaling AI training infrastructure.
Responsibilities
- Tune distributed training for 100B+ parameter models
- Write custom kernels for optimized performance
- Evaluate and benchmark new accelerators
Requirements
- Experience with distributed training of large-scale models
- Proficiency in GPU programming and CUDA
- Strong understanding of deep learning frameworks like PyTorch
Tech Stack
CUDADistributed TrainingGPU
Figure
San Jose · US · 845+ employees
Figure AI is an AI robotics company dedicated to building the world's first commercially viable autonomous humanoid robot. The company focuses on developing general-purpose robotic solutions for commercial deployment.
