Senior ML Systems Engineer: Scalable Training Frameworks

A leading AI research firm located in San Francisco is seeking a Senior ML Systems Engineer to build and maintain the training framework for large-scale language models. The role involves designing distributed training solutions and improving training throughput across multi-node clusters. The ideal candidate will have strong engineering experience in distributed training, familiarity with JAX, and excellent collaboration skills. This position promises significant ownership over critical components and engagement with cutting-edge AI technologies while offering a flexible work environment. #J-18808-Ljbffr