Software Engineer, AI Compiler

About the role

Lead development on TT-Forge, our MLIR-based compiler, and manage a team focused on scaling graph transformations, lowering passes, and kernel-level optimizations. Help shape the future of AI computing through compiler technology that is fast, flexible, and built for real-world models. This role is hybrid based in Austin, TX. We welcome candidates at various experience levels.

Responsibilities

• Drive compiler development across multiple layers of the stack, including custom dialects, transformation passes, and runtime integration.
• Align with hardware, software, and ML teams to ensure the compiler supports realistic performance and deployment needs.
• Take technical ownership of roadmap priorities, planning, and mentorship for a high-performing engineering team.
• Identify bottlenecks, debug performance issues, and improve developer workflows across the stack.

Requirements

• Experience building compilers or similar systems, with strong fluency in C++ and Python.
• Strong knowledge of MLIR, LLVM, or related infrastructure, with hands-on work in dialect design or optimization passes.
• Understanding of modern AI frameworks like PyTorch, TensorFlow, or JAX, including how models are transformed and executed.
• Background in working across teams to ship reliable, scalable tools in fast-paced engineering environments.

What You’ll Learn

• How to build compiler infrastructure that supports both training and inference at scale across custom chip architectures.
• New approaches for human-in-the-loop optimization using TT-Explorer and other internal tooling.
• Best practices for evolving MLIR dialects like TTIR, TTNN, and TTKernel to support next-generation AI workloads.
• How Tenstorrent delivers open, performant, and developer-friendly AI infrastructure that scales with industry demands.

Pay

Compensation for all engineers at Tenstorrent ranges from $100k - $500k including base and variable compensation targets. Experience, skills, education, background, and location all impact the actual offer made.

Tenstorrent offers a highly competitive compensation package and benefits. This offer of employment is contingent upon the applicant being eligible to access U.S. export-controlled technology.