Software Engineer, Model Runtime

OpenAI Inc.
San Francisco, CA, United States
4 days ago
Apply on www.careerbuilder.com
Prepare application

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours

Tech stack

Artificial Intelligence C++ (Programming Language) Computer Programming Software Debugging Distributed Systems Memory Management Python (Programming Language) OpenAI Software Engineering Supercomputing Systems Architecture System Programming
+6 more
Systems Integration Rust (Programming Language) Low Latency SGLang Machine Learning Operations Model Inference

Job description

You will build the model runtime within the inference engine that executes complex, frontier models at scale on OpenAI’s custom silicon. The runtime will sit between models running on the hardware and the upper layers of the cluster serving software stack, translating demanding inference workloads into efficient execution while optimizing for throughput, latency, utilization, and reliability.

You will work across model architecture, distributed systems, compilers, kernels, and silicon to design a production-grade runtime comparable in ambition to systems such as vLLM and SGLang, but customized and optimized for OpenAI’s AI accelerator. Your work will shape how new model capabilities map onto the platform and how quickly custom silicon can deliver meaningful performance in production.

In this role, you will:

  • Design and implement the LLM inference runtime for frontier models running on custom silicon.
  • Build scheduling, continuous batching, memory management, KV-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 complex correctness, performance, and reliability issues spanning model code, runtime software, communication layers, and hardware.
  • Turn workload insights into clear requirements for future generations of silicon and system architecture.

You might thrive in this role if:

  • Have strong systems programming experience in C++, Rust, Python, or comparable performance-oriented environments.
  • Have built or optimized runtimes, distributed systems, compilers, kernels, model-serving infrastructure, or adjacent systems software.
  • Understand modern LLM inference, including prefill and decode behavior, batching, KV-cache tradeoffs, and model parallelism.
  • Can reason quantitatively about latency, throughput, compute intensity, memory bandwidth, communication, and utilization.
  • Are comfortable profiling and debugging performance across multiple layers of a hardware-software stack.
  • Can design clean abstractions while retaining the low-level control needed to extract performance from specialized hardware.
  • Work effectively across model, systems, compiler, kernel, and hardware teams to drive ambiguous technical problems to closure.
  • Care about production quality, including correctness, observability, reliability, maintainability, and graceful behavior at scale.

To comply with U.S. export control laws and regulations, candidates for this role may need to meet certain legal status requirements as provided in those laws and regulations.

Requirements

Software Engineering, Artificial Intelligence (AI), Supercomputing, System Integration (SI), Power Generation, Inference Engine, Distributed Computing, Kernel Programming, Memory Management, Strategic Planning

About the company

OpenAI is an AI research and deployment company dedicated to ensuring that general-purpose artificial intelligence benefits all of humanity. We push the boundaries of the capabilities of AI systems and seek to safely deploy them to the world through our products. AI is an extremely powerful tool that must be created with safety and human needs at its core, and to achieve our mission, we must encompass and value the many different perspectives, voices, and experiences that form the full spectrum of humanity., At OpenAI, we believe artificial intelligence has the potential to help people solve immense global challenges, and we want the upside of AI to be widely shared. Join us in shaping the future of technology.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on www.careerbuilder.com
Prepare application

Good distractions

Loading talks and stories from around this role…