Software Engineer, Triton Compiler

• 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
• 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

• 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

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

• Offers Equity
• Medical, dental, and vision insurance for you and your family, with employer contributions to Health Savings Accounts
• Pre-tax accounts for Health FSA, Dependent Care FSA, and commuter expenses (parking and transit)
• 401(k) retirement plan with employer match
• Paid parental leave (up to 24 weeks for birth parents and 20 weeks for non-birthing parents), plus paid medical and caregiver leave (up to 8 weeks)
• Paid time off: flexible PTO for exempt employees and up to 15 days annually for non-exempt employees
• 13+ paid company holidays, and multiple paid coordinated company office closures throughout the year for focus and recharge, plus paid sick or safe time (1 hour per 30 hours worked, or more, as required by applicable state or local law)
• Mental health and wellness support
• Employer-paid basic life and disability coverage
• Annual learning and development stipend to fuel your professional growth
• Daily meals in our offices, and meal delivery credits as eligible
• Relocation support for eligible employees
• Additional taxable fringe benefits, such as charitable donation matching and wellness stipends