← Search

Training: ML Framework Engineer

OpenAI · Scaling

Location
San Francisco, California, United States
Metro
San Francisco Bay Area
Workplace
Hybrid
Level
Unspecified
Type
Full-time
Pay
$295K–$500K/yr (published by employer)
Posted
12 months ago
Source
Ashby

Read the full posting & apply ↗

Python

What the posting covers

About the Team

Training Runtime designs the core distributed machine-learning training runtime that powers everything from early research experiments to frontier-scale model runs. With a dual mandate to accelerate researchers and enable frontier scale, we’re building a…

About the Role

As a Training: ML Framework Engineer, you will work on improving the training throughput for our internal training framework, while enabling researchers to experiment with new ideas. This requires good engineering (for example designing, implementing, and…

In this role, you will

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

You might thrive in this role if you

Have run small scale ML experiments Love figuring out how systems work and continuously come up with ideas for how to make them faster while minimizing complexity and maintenance burden Have strong software engineering skills and are proficient in Python

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…