Training: ML Framework Engineer

As a Training: ML Framework Engineer, you will focus on enhancing the training throughput of our internal training framework, empowering researchers to explore novel ideas. This role demands strong engineering skills, including the design, implementation, and optimization of state-of-the-art AI models, coupled with meticulous machine learning code development and a deep understanding of supercomputer performance. The ultimate objective is to advance the field of AI. We are seeking individuals passionate about performance optimization, distributed systems, and writing bug-free code, as improvements in this framework, used for large-scale GPU runs, will have a significant impact.