✨ About The Role
- The role involves working on the capture and optimization of model graphs for training runs
- Responsibilities include applying advanced optimizations to maximize hardware resource utilization
- Collaboration with researchers to accelerate progress towards Artificial General Intelligence (AGI)
- Focus on maximizing training throughput and researcher throughput to develop new models efficiently
- Opportunity to work on a state-of-the-art internal training framework for cutting-edge models
âš¡ Requirements
- Ideal candidate has strong software engineering skills and proficiency in Python
- Experience running machine learning experiments is a must
- Deep knowledge of GPU performance and systems-level optimization is required
- Ability to work on a training framework to enable easy graph capture of research experiments
- Excitement to work with cutting-edge ML researchers and contribute to the development of new capabilities