Senior AI Training Performance Engineer (GPU & Scale)

Figure is seeking an experienced AI Training Performance Engineer to advance distributed training for massive models. The role focuses on optimizing GPU kernels, accelerator selection, and co-designing models to maximize hardware utilization.

You will develop kernels, build performance dashboards, and collaborate across teams to enable scalable, reliable training at scale. Applicants should have 3+ years in AI performance engineering, deep GPU knowledge, and strong Python and CUDA/C++ skills,