ML Infrastructure Engineer
Description For ML Infrastructure Engineer
OpenAI is seeking a ML Infrastructure Engineer to join their Runtime team, focusing on building and optimizing low-level framework components for ML training systems. This role combines advanced engineering skills with machine learning expertise to improve training throughput and enable cutting-edge research. The position offers an opportunity to work on large-scale GPU implementations and directly impact the development of next-generation AI models. Based in San Francisco with a hybrid work model, the role offers competitive compensation ($380K-$555K plus equity) and comprehensive benefits. The ideal candidate should have experience with ML experiments, system optimization, and strong Python programming skills. OpenAI's mission focuses on developing safe and beneficial artificial intelligence, making this an opportunity to contribute to groundbreaking advancements in AI technology while ensuring it benefits humanity.
Responsibilities For ML Infrastructure Engineer
- Apply the latest techniques in internal training framework to achieve hardware efficiency for training runs
- Profile and optimize training framework
- Work with researchers to enable them to develop the next generation of models
Requirements For ML Infrastructure Engineer
- Have run small scale ML experiments
- Love figuring out how systems work and continuously come up with ideas for optimization
- Have strong software engineering skills and are proficient in Python
Benefits For ML Infrastructure Engineer
- Medical, dental, and vision insurance for you and your family
- Mental health and wellness support
- 401(k) plan with 50% matching
- Generous time off and company holidays
- Paid parental leave (24 weeks paid birth-parent leave & 20-week paid parental leave)
- Annual learning & development stipend ($1,500 per year)
- Equity
Jobs Related To OpenAI ML Infrastructure Engineer
