Skip to content
OpenAIOpenAISan Francisco, CA

Distributed Training Engineer, Sora

Distributed Training Engineer optimizes training throughput for video foundation models, enabling researcher experiments with efficient architectures. Requires strong Python skills, distributed systems expertise, and performance profiling.

293k – 490k
HybridML Engineering

About the role

Responsibilities

  • Collaborate with researchers to enable them to develop systems-efficient video models and architectures
  • Apply the latest techniques to our internal training framework to achieve impressive hardware efficiency for our training runs
  • Profile and optimize our training framework

Requirements

  • Experience working with multi-modal ML pipelines
  • Love diving deep into systems implementations and understanding their fundamentals in order to improve their performance and maintainability
  • Strong software engineering skills and are proficient in Python
  • Experience understanding and optimizing training kernels
  • Passionate about understanding stable training dynamics

Skills

PythonDistributed SystemsMachine LearningMulti-Modal MlTraining FrameworksPerformance OptimizationPyTorchSupercomputingTraining KernelsVideo Models

Similar roles

ML Engineering jobs
OpenAI

Software Engineer, Workload Enablement

OpenAISan Francisco, CA +1

Software Engineer enabling production AI workloads on new hardware platforms through porting, benchmarking, stress testing, and performance optimization. Requires 5+ years in ML systems, distributed training, PyTorch, and RDMA/NCCL expertise.

293k – 455k
Hybrid5+ YOEML Engineering
OpenAI

Research Engineer, Retrieval & Search, Applied Engineering

OpenAISan Francisco, CA

Develop and deploy retrieval and search algorithms for OpenAI's API and ChatGPT, collaborating with research teams on production systems for millions of users. Requires experience with ML systems, vector databases, and large-scale search.

293k – 585k
On-siteML Engineering
OpenAI

Agent Post-Training, API & Power Users

OpenAISan Francisco, CA

Improve agentic model capabilities, reliability, and product fit for power users and API developers through evals, training data, and post-training interventions.

295k – 445k
On-siteML Engineering
OpenAI

Software Engineer, RL Training Infra

OpenAISan Francisco, CA

Build and maintain infrastructure for large-scale RL training runs of frontier OpenAI models. Debug across training, inference, and distributed systems while supporting research integrations.

295k – 445k
HybridML Engineering
OpenAI

Software Engineer, ML Systems & Training Architecture

OpenAISan Francisco, CA

Hands-on senior software engineer focused on maintaining and improving ML training infrastructure, debugging training systems, and unblocking researchers on the robotics team.

295k – 380k
On-site3+ YOEML Engineering