Training: ML Framework Engineer

OpenAI
San Francisco
Workplace: HybridFull timeUSD 205,000 - 445,000 annuallyFunction: Education & TrainingSkills: ["Python","Problem-solving","Communication","Team collaboration","Adaptability"]

The Training: ML Framework Engineer role focuses on improving the training throughput for an internal distributed ML training framework, enabling researchers to experiment with new ideas. It emphasizes high-performance optimization, Python software engineering, and deep knowledge of GPU-scale performance, with a hybrid in-person/office setup in San Francisco and relocation assistance.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
OpenAI
OpenAI
9 months ago

Training: ML Framework Engineer

āœ“ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 1 hour agoStatus: Live

Job Summary

The Training: ML Framework Engineer role focuses on improving the training throughput for an internal distributed ML training framework, enabling researchers to experiment with new ideas. It emphasizes high-performance optimization, Python software engineering, and deep knowledge of GPU-scale performance, with a hybrid in-person/office setup in San Francisco and relocation assistance.
Location: San Francisco
Workplace: Hybrid
Employment Type: Full time
Job Function: Education & Training

Key Responsibilities

  • ā€¢Improve training throughput by applying techniques to the internal training framework and optimize hardware efficiency
  • ā€¢Profile and optimize the training framework to identify bottlenecks and improve performance
  • ā€¢Collaborate with researchers to enable development of the next generation of models
  • ā€¢Assist in designing and implementing state-of-the-art AI model components within the training runtime
  • ā€¢Contribute to maintaining high availability, fault-tolerance, and observability of the training system

Pay and Benefits

Salary: USD 205,000 - 445,000 annually
Equity and Bonus:Equity
Perks:Equity

Key Requirements

  • ā€¢Strong software engineering skills with proficiency in Python
  • ā€¢Experience optimizing performance for distributed systems or large-scale ML workloads
  • ā€¢Ability to design, implement, and optimize AI models and training frameworks
  • ā€¢Experience profiling and optimizing ML training frameworks
  • ā€¢Experience collaborating with researchers to enable model development
Experience:Machine learningDistributed systemsAI research
Skills:PythonProblem-solvingCommunicationTeam collaborationAdaptability
Languages:English
Tech Stack:PythonGPUsDistributed systemsTraining frameworkCheckpointing

Company Brief

OpenAI
Develops and deploys advanced generative AI models (including ChatGPT and DALLĀ·E) and AI infrastructure, providing APIs and consumer products to accelerate safe AGI for broad benefit.
Industry: AI & Machine Learning
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Scaleup
Valuation: Hectocorn (USD 100B+)
Funding: Series E+
Headquarters: San Francisco, United States
Founded: 2015
Glassdoor
Glassdoor: 4.4
WebsiteLinkedInGlassdoor