All Jobs
ML Framework Engineer

ML Framework Engineer

Posted 817 days ago

Job Description

This job posting has expired and no longer accepting applications.

About the Team

The Platform ML team builds the ML side of our state-of-the-art internal training framework used to train our cutting-edge models. We work on distributed model execution as well as the interfaces and implementation for model code, training, and inference. Our priorities are to maximize training throughput (how quickly we can train a new model) and researcher throughput (how quickly we can develop new models) with the goal of accelerating progress towards AGI. We frequently collaborate with other teams to speed up the development of new capabilities.

About the Role

As a ML Framework Engineer, you will work on our training framework to create the abstractions that allow researchers to perform experiments using cutting-edge models with impressive ease-of-use. Many of our research teams build on our framework for running their experiments, and they should be able to write their experiments such that they get both flexibility and high scalability.

We’re looking for someone who wants to work at the interface between research and compute and who loves building a great experience for researchers designing their next experiment. This is a high leverage area due to the large algorithmic gains that can be achieved when research productivity is increased.

This role is based in San Francisco, CA. We use a hybrid work model of 3 days in the office per week and offer relocation assistance to new employees.

In this role, you will:

Build out our internal training framework to make research with large distributed models easy and fun
Work closely with researchers to ensure that the framework meets their needs and enables the next generation of experiments
Design and build the next generation of training infrastructure at the forefront of language model research

You might thrive in this role if you:

Have worked on ML tools that researchers love
Have a good understanding of current models and training algorithms
Are a strong software engineer and proficient in Python
Are obsessed with correctness and beautiful interfaces

Compensation, Benefits and Perks

Total compensation also includes generous equity and benefits.

Medical, dental, and vision insurance for you and your family
Mental health and wellness support
401(k) plan with 4% matching
Unlimited time off and 18+ company holidays per year
Paid parental leave (20 weeks) and family-planning support
Annual learning & development stipend ($1,500 per year)

Annual Salary Range

$245,000—$385,000 USD

About OpenAI

OpenAI is an AI research and deployment company dedicated to ensuring that general-purpose artificial intelligence benefits all of humanity. We push the boundaries of the capabilities of AI systems and seek to safely deploy them to the world through our products. AI is an extremely powerful tool that must be created with safety and human needs at its core, and to achieve our mission, we must encompass and value the many different perspectives, voices, and experiences that form the full spectrum of humanity.

We are an equal opportunity employer and do not discriminate on the basis of race, religion, national origin, gender, sexual orientation, age, veteran status, disability or any other legally protected status.

For US Based Candidates: Pursuant to the San Francisco Fair Chance Ordinance, we will consider qualified applicants with arrest and conviction records.

We are committed to providing reasonable accommodations to applicants with disabilities, and requests can be made via this link.

OpenAI Global Applicant Privacy Policy

At OpenAI, we believe artificial intelligence has the potential to help people solve immense global challenges, and we want the upside of AI to be widely shared. Join us in shaping the future of technology.

This job posting has expired and no longer accepting applications. Please check out our latest AI jobs.