Skip to content
OpenAIOpenAISan Francisco, CA

Software Engineer, ML Systems & Training Architecture

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/yr
On-site3+ YOEML Engineering

About the role

Responsibilities

  • Review, improve, and clean up code across training frameworks and adjacent infrastructure
  • Identify risky or low-quality changes before they land, and raise the code quality bar without slowing the team down
  • Debug issues across ML training systems, GPUs, clusters, networking, and related infrastructure
  • Help researchers and engineers unblock broken training jobs, flaky workflows, and brittle internal tooling
  • Improve the reliability, maintainability, and usability of the robotics team's training framework
  • Move quickly on practical engineering problems that directly affect team velocity

Requirements

  • Strong software engineering fundamentals and excellent code review judgment
  • Experience with ML systems, training frameworks, GPUs, distributed systems, infrastructure, or similarly complex technical environments
  • Ability to read and debug unfamiliar codebases quickly, and enjoy getting to root cause
  • Ship high-quality code with strong velocity and pragmatic judgment
  • Low-ego, responsive, and motivated by helping researchers and engineers move faster
  • Prefer being a highly effective hands-on IC over driving broad process-heavy initiatives
  • Experience reviewing messy, fast-moving, or AI-generated codebases

Skills

PythonPyTorchTensorFlowCUDADistributed SystemsGpu ProgrammingMl Training FrameworksCode ReviewDebuggingInfrastructure

Similar roles

ML Engineering jobs
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/yr
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/yr
HybridML Engineering
OpenAI

Software Engineer, Inference - Performance Optimization

OpenAISan Francisco, CA

Models inference performance across application, model, and fleet layers using microbenchmarks to build cost-to-serve estimates. Analyzes workloads end-to-end, enhances bottleneck detection tools, and collaborates on optimizations for latency, throughput, and cost.

295k – 555k/yr
On-siteML Engineering
OpenAI

Machine Learning Engineer, Distributed Data Systems

OpenAISan Francisco, CA

Designs and scales distributed data infrastructure for large-scale multimodal AI training and evaluation. Collaborates with researchers to build reliable, high-performance systems in a fast-paced environment.

295k – 445k/yr
HybridML Engineering
OpenAI

Software Engineer, Inference – AMD GPU Enablement

OpenAISan Francisco, CA

Develops and optimizes OpenAI's inference infrastructure for AMD GPUs, handling low-level kernel performance, distributed execution, and integration with serving frameworks like vLLM and Triton. Requires expertise in GPU programming with HIP/CUDA and distributed systems scaling.

295k – 555k/yr
On-siteML Engineering