Figure, an AI robotics company headquartered in San Jose, CA, is seeking an experienced AI Training Performance Engineer to advance our large-scale model training. You will optimize distributed training across thousands of GPUs, tune GPU kernels, and co-design training recipes with researchers.
You will build tooling and dashboards, improve data pipelines, and extend kernel compilers for non-NVIDIA accelerators.
#J-18808-LjbffrAI Training Performance Engineer — Large-Scale GPU Training in san jose at Unknown Company
This position is listed as full time and onsite.