O

OpenAI

Builds AI models and the ChatGPT assistant

Software Engineer - Model Runtime

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Francisco, CA, USA$266k - $445k /yrHybrid

About the Job

Requirements

  • Strong systems programming experience in C++, Rust, Python, or comparable performance-oriented environments.
  • Experience building or optimizing runtimes, distributed systems, compilers, kernels, model-serving infrastructure, or adjacent systems software.
  • Understanding of modern large language model inference, including prefill and decode behavior, batching, key-value cache tradeoffs, and model parallelism.
  • Ability to reason quantitatively about latency, throughput, compute intensity, memory bandwidth, communication, and utilization.
  • Ability to profile and debug performance across multiple layers of a hardware-software stack.
  • Ability to design clean abstractions while retaining low-level control to extract performance from specialized hardware.
  • Ability to work across model, systems, compiler, kernel, and hardware teams to resolve ambiguous technical problems.
  • Commitment to production quality, including correctness, observability, reliability, maintainability, and graceful behavior at scale.

Responsibilities

  • Design and implement the large language model inference runtime for frontier models running on custom silicon.
  • Build scheduling, continuous batching, memory management, key-value cache management, and execution orchestration for high-performance inference.
  • Develop distributed execution strategies across chips, hosts, and racks, including model partitioning, communication, and synchronization.
  • Optimize end-to-end latency, throughput, memory efficiency, and hardware utilization across diverse model architectures and serving workloads.
  • Partner with kernel, compiler, architecture, and silicon teams to co-design interfaces and remove performance bottlenecks across the stack.
  • Enable new model features, execution patterns, numerical formats, and hardware capabilities in a reliable production runtime.
  • Create profiling, observability, benchmarking, and performance-modeling tools that make runtime behavior measurable and actionable.
  • Debug correctness, performance, and reliability issues spanning model code, runtime software, communication layers, and hardware.
  • Translate workload insights into clear requirements for future generations of silicon and system architecture.

About the Company

OpenAI

Builds AI models and the ChatGPT assistant

OpenAI researches and builds artificial intelligence models and turns them into products for people, developers, and businesses, with a stated mission of making sure artificial general intelligence benefits all of humanity. Its launch of ChatGPT in 2022 brought generative AI into everyday use. People use ChatGPT to ask questions, write, code, and analyze files, developers build OpenAI's models into their own apps through its API, and its Codex agent works through coding tasks on its own. OpenAI earns money from ChatGPT subscriptions, business plans, and API usage. Sam Altman, Elon Musk, and others co-founded it in 2015; it is based in San Francisco and overseen by a nonprofit, the OpenAI Foundation.

Company size
1,001-5,000
Company stage
Late Stage VC
Total funding
$207.5B
Headquarters
San Francisco, CA
Founded
2015

Recently Posted Jobs

Explore more opportunities at OpenAI

See more jobs