ML Infrastructure Engineer: OpenAI

OpenAI is actively hiring across multiple research teams, each dedicated to advancing artificial general intelligence (AGI). As an engineer supporting these teams, you'll be crucial in building the critical infrastructure that powers their most ambitious experiments. Your work will involve developing robust, scalable, and high-performance components for distributed training and inference. The overarching mission of this team is to dramatically accelerate research velocity by maximizing the efficiency of both researchers and hardware.

About the Role
As an ML Infrastructure Engineer at OpenAI, your primary focus will be on improving the training or inference throughput for their internal training framework, while simultaneously enabling researchers to innovate with new ideas. This involves strong engineering practices, including designing, implementing, and optimizing state-of-the-art AI models, writing precise machine learning code, and developing a deep understanding of supercomputer performance. The ultimate goal of all projects in this role is to push the field of AI forward.

The ideal candidate will be passionate about optimizing performance, understanding distributed systems, and committed to writing bug-free code. Given that the training framework is used for large runs with a massive number of GPUs, performance improvements made in this role will have a significant impact.

Responsibilities
In this role, you will:

Apply the latest techniques in OpenAI's internal training framework to achieve impressive hardware efficiency for training runs.

Profile and optimize the training framework.

Work collaboratively with researchers to enable them to develop the next generation of models.

Requirements
You might thrive in this role if you:

Have run small-scale ML experiments.

Love figuring out how systems work and consistently propose ideas to make them faster while minimizing complexity and maintenance burden.

Possess strong software engineering skills and are proficient in Python.

About OpenAI & Benefits
OpenAI is an AI research and deployment company committed to ensuring that general-purpose artificial intelligence benefits all of humanity. They push the boundaries of AI system capabilities and aim to safely deploy them to the world through their products. OpenAI emphasizes that AI is a powerful tool that must prioritize safety and human needs, and achieving their mission requires valuing diverse perspectives, voices, and experiences.

OpenAI is an equal opportunity employer and does not discriminate based on race, religion, color, national origin, sex, sexual orientation, age, veteran status, disability, genetic information, or other legally protected characteristics. They also consider qualified applicants with arrest or conviction records in accordance with applicable laws, including the San Francisco Fair Chance Ordinance, the Los Angeles County Fair Chance Ordinance for Employers, and the California Fair Chance Act. They are committed to providing reasonable accommodations to applicants with disabilities.

OpenAI believes AI has the potential to solve immense global challenges and aims for the upside of AI to be widely shared.