Helix AI Engineer, Training Performance

From the posting

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

Figure's vision is to deploy autonomous humanoids at a global scale. Our Helix team is looking for an experienced AI Training Performance Engineer to take our model training to the next level. This role is focused on improving distributed training frameworks for large scale model training, optimizing GPU kernels, exploring the relative gains of different accelerator types and co-designing our models to maximize utilization of our hardware.

Responsibilities:

Optimize training performance for a 100B+ parameter models across 100k+ GPUs.

Collaborate with the broader team on accelerator choice, cluster topology, scheduling, and hardware procurement decisions to inform future scaling.

Write and optimize custom kernels (Triton/CUDA).

Build tooling and dashboards for continuous performance monitoring, regression detection, and root-cause analysis across training jobs.

Optimize data loading and preprocessing pipelines so I/O never gates the accelerators.

Improve checkpointing, fault tolerance, and elastic restart so large jobs recover quickly from node failures without losing significant wall-clock time. …

Excerpt from Figure AI's public job board. Read the full posting.