Training: ML Framework Engineer
Training: ML Framework Engineer
OpenAI
The Training: ML Framework Engineer role at OpenAI involves enhancing the performance of the internal training framework for machine learning, focusing on optimizing training throughput and enabling researchers to experiment with new ideas. The position is based in San Francisco, CA, and follows a hybrid work model. Candidates should have strong software engineering skills, particularly in Python, and a passion for optimizing performance in distributed systems.
Qualification
- Strong software engineering skills
- Proficiency in Python
- Experience running small scale ML experiments
- Understanding of distributed systems
- Ability to optimize performance and minimize complexity
Responsibility
- Apply the latest techniques in the internal training framework to achieve hardware efficiency for training runs
- Profile and optimize the training framework
- Collaborate with researchers to develop next-generation models
- Design, implement, and optimize state-of-the-art AI models
- Ensure bug-free machine learning code
