Training: ML Framework Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

OpenAI · 3 hours ago

Training: ML Framework Engineer

positionSan Francisco, CA
timeFull-time
remoteHybrid
seniorityMid Level

OpenAI is an AI research and deployment company dedicated to ensuring that general-purpose artificial intelligence benefits all of humanity. As a Training: ML Framework Engineer, you will improve the training throughput for the internal training framework while enabling researchers to experiment with new ideas, focusing on optimizing performance and understanding distributed systems.

Artificial Intelligence (AI)Generative AIMachine LearningNatural Language ProcessingSaaS
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

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

Qualification

Machine LearningPythonDistributed SystemsPerformance OptimizationSoftware EngineeringSupercomputer KnowledgeBug-Free CodeCollaboration with ResearchersExperimentation

Required

Good engineering skills for designing, implementing, and optimizing state-of-the-art AI models
Ability to write bug-free machine learning code
Deep knowledge of the performance of supercomputers
Experience running small scale ML experiments
Strong software engineering skills
Proficiency in Python

Preferred

Love figuring out how systems work and continuously come up with ideas for how to make them faster while minimizing complexity and maintenance burden

Benefits

Relocation assistance

Company

OpenAI

twittertwittertwitter
Glassdoor4.2
company-logo
OpenAI is an AI research and deployment company that develops advanced AI models, including ChatGPT. It is a sub-organization of OpenAI Foundation.
dateFounded in 2015
location
San Francisco, California, USA
people-size201-500 employees
web-link
https://www.openai.com

H1B Sponsorship

OpenAI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)
2024 (1)
2023 (1)
2022 (18)
2021 (10)
2020 (6)

Funding

Current Stage
Growth Stage
Total Funding
$79B
Key Investors
The Walt Disney CompanySoftBankThrive Capital
2025-12-11Corporate Round· $1B
2025-10-02Secondary Market· $6.6B
2025-03-31Series Unknown· $40B

Leadership Team

leader-logo
Sam Altman
CEO & Co-Founder
leader-logo
Greg Brockman
President, Chairman, & Co-Founder
linkedin

Recent News

SiliconANGLE
SoftBank reportedly finalizes OpenAI investment with $22.5B cash infusion
2025-12-31
BBC
China to crack down on AI firms to protect kids
2025-12-31
Android Authority
OpenAI's mysterious hardware project could be a... pen?
2025-12-31
Company data provided by crunchbase