Software Engineer, Triton Compiler jobs in United States
cer-icon
Apply on Employer Site
company-logo

OpenAI · 5 days ago

Software Engineer, Triton Compiler

OpenAI is an AI research and deployment company dedicated to ensuring that general-purpose artificial intelligence benefits all of humanity. As a Software Engineer, you will help build AI systems and optimize core ML systems while working closely with the Compiler/Kernels team to enhance the performance-critical software stack for custom silicon.

Agentic AIArtificial Intelligence (AI)Foundational AIGenerative AIMachine LearningNatural Language ProcessingSaaS
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Help build AI systems that achieve levels of performance that were previously impossible
Designing and optimizing core ML systems
Writing highly reliable low-level code
Advancing the algorithms and infrastructure that power our models
Responsible for the performance-critical software stack that brings OpenAI’s custom silicon to life
Design and build the compilers, languages, and high-performance kernels that allow researchers to fully exploit our first-party accelerators
Advancing Triton and its backend
Developing new compiler passes
Creating the tooling needed to write fast, correct, and deeply optimized kernels for brand-new hardware
Partner closely with the hardware team to unlock new capabilities and ensure our custom silicon can support the next generation of frontier models

Qualification

Compiler developmentPerformance engineeringMachine learning frameworksCustom acceleratorsProblem ownershipSelf-directedAmbiguous technical spaces

Required

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

Preferred

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

Company

OpenAI is an AI research and deployment company that develops advanced AI models, including ChatGPT. It is a sub-organization of OpenAI Foundation.

H1B Sponsorship

OpenAI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1)
2024 (1)
2023 (1)
2022 (18)
2021 (10)
2020 (6)

Funding

Current Stage
Growth Stage
Total Funding
$79B
Key Investors
The Walt Disney CompanySoftBankThrive Capital
2025-12-11Corporate Round· $1B
2025-10-02Secondary Market· $6.6B
2025-03-31Series Unknown· $40B

Leadership Team

leader-logo
Sam Altman
CEO & Co-Founder
leader-logo
Greg Brockman
President, Chairman, & Co-Founder
linkedin
Company data provided by crunchbase