Software Engineer, Triton Compiler

OpenAI • San Francisco, CA

1d

About The Position

Our mission at OpenAI is to discover and enact the path to safe, beneficial AGI. To do this, we believe that many technical breakthroughs are needed in generative modeling, reinforcement learning, large scale optimization, active learning, among other topics. About the Role As a Software Engineer, you will help build AI systems that achieve levels of performance that were previously impossible. This means strong engineering fundamentals—designing and optimizing core ML systems, writing highly reliable low-level code, and advancing the algorithms and infrastructure that power our models. The end goal in every project is the same: push the frontier forward. The Compiler/Kernels team is responsible for the performance-critical software stack that brings OpenAI’s custom silicon to life. We design and build the compilers, languages, and high-performance kernels that allow researchers to fully exploit our first-party accelerators. This includes advancing Triton and its backend, developing new compiler passes, and creating the tooling needed to write fast, correct, and deeply optimized kernels for brand-new hardware. We partner closely with the hardware team to unlock new capabilities and ensure our custom silicon can support the next generation of frontier models. If you love compilers, kernels, and shaping how entirely new AI hardware is programmed, this role is for you. We’re looking for engineers who are self-directed, comfortable in ambiguous technical spaces, and excited to identify and solve the highest-leverage problems.

Requirements

• 3+ years of relevant engineering experience, ideally in systems, compilers, ML frameworks, or performance engineering
• Owning problems end-to-end, including learning new hardware and software domains as needed

Nice To Haves

• contributions to ML frameworks (e.g., PyTorch, JAX, TensorFlow) or compilers (e.g., LLVM, MLIR, XLA), or experience programming custom or specialized accelerators

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: No Education Listed
• Number of Employees: 1,001-5,000 employees