OpenAI

Training: ML Framework Engineer

OpenAI
Apply
11 months ago

Base Salary

$205k - $445k/yr

Responsibilities

  • Apply advanced techniques in the internal training framework to improve hardware efficiency for training runs.
  • Profile and optimize the training framework.
  • Work with researchers to enable development of next-generation models.

Requirements

  • Strong software engineering skills.
  • Proficiency in Python.
  • Experience running small-scale machine-learning experiments.
  • Interest in understanding distributed systems and optimizing performance while minimizing complexity and maintenance burden.

Benefits

  • Hybrid work model based in San Francisco, California, with 3 days in the office per week.
  • Relocation assistance is offered to new employees.

Tech Stack

OpenAI

About OpenAI

10,000+ employees

OpenAI builds and deploys large-scale AI models and tools—including ChatGPT, GPT-4–class models, DALL·E, and Whisper—sold via APIs and enterprise subscriptions to developers and businesses. It monetizes through usage-based API pricing and ChatGPT Plus/Team/Enterprise, and also reaches customers via Microsoft’s Azure OpenAI Service. Founded in 2015 and headquartered in San Francisco, it operates as a private partnership.

Contact me