Training: ML Framework Engineer

The role

You will improve training throughput for the internal training framework and enable researchers to experiment with new ideas. This involves applying techniques to achieve hardware efficiency, profiling and optimizing the framework, and working with researchers.

What you will do

• Apply the latest techniques in our internal training framework to achieve impressive hardware efficiency for our training runs
• Profile and optimize our training framework
• Work with researchers to enable them to develop the next generation of models

Required qualifications

• Strong software engineering skills
• Proficient in Python

Nice to have

• Have run small scale ML experiments

Benefits

• Relocation assistance

Before you apply

• Background checks for applicants will be administered in accordance with applicable law

About the company

OpenAI is an AI research and deployment company dedicated to ensuring that general-purpose artificial intelligence benefits all of humanity.