ML Framework Engineer: Accelerate Large-Scale Training

OpenAI; California (MO); Hybrid; USD 150,000 - 210,000; Full time

Benefits offered by this job

Relocation assistance

Hybrid work model

OpenAI team culture

Job summary

OpenAI in San Francisco, CA is seeking a Training: ML Framework Engineer to push the boundaries of our internal training runtime. The role focuses on increasing training throughput for large-scale experiments and requires deep knowledge of Python, distributed systems, and modern ML frameworks.

You will work in a hybrid setup (3 days in the office) with relocation assistance for new hires, collaborating closely with researchers to enable the next generation of models.

Qualifications

• Experience building and optimizing ML training frameworks.
• Strong knowledge of distributed systems and performance profiling.
• Proficiency in Python and research-to-prod execution.

Responsibilities

• Apply latest techniques to improve training throughput on large-scale runs.
• Profile and optimize the training framework for high GPU utilization.
• Collaborate with researchers to enable next-generation models.

Skills

Python

Distributed systems

Performance optimization

ML frameworks

GPU/HPC awareness

Job description

OpenAI in San Francisco, CA is seeking a Training: ML Framework Engineer to push the boundaries of our internal training runtime. The role focuses on increasing training throughput for large-scale experiments and requires deep knowledge of Python, distributed systems, and modern ML frameworks.

You will work in a hybrid setup (3 days in the office) with relocation assistance for new hires, collaborating closely with researchers to enable the next generation of models.