[ engineering ]

Helix Ai Engineer, Training Performance

Figure AI
Location
San Jose, California
Work mode
Remote
Type
full time
Listed

Applications go directly to Figure AI. Ramai takes no fee for placement and no money for ranking.

[ About this role ]

Figure AI is seeking a Senior AI Training Performance Engineer to optimize large-scale model training for their humanoid robots. This role involves improving distributed training frameworks, optimizing GPU kernels, and co-designing models for hardware utilization.

Skills and focus areas

  • distributed training
  • gpu kernel optimization
  • model co-design
  • hardware utilization
  • custom kernel writing
  • triton
  • cuda
  • performance monitoring
  • data loading optimization
  • fault tolerance
  • elastic restart
  • mixed precision

Summary written by Ramai from the original posting. Full details and requirements are on Figure AI’s careers page.

[ More at Figure AI ]

See all 98 open roles at Figure AI →

[ Similar roles at other startups ]

Browse all open engineering roles →

[ Not quite right? ]

Tell the assistant what you’re after — stage, location, stack, visa sponsorship — and it searches every open role for you. A few questions are free, no account needed.