Principal Engineer, AI Compiler

Key Skills

Aartificial intelligence, C++, Infrastructure as code, JAX framework, Machine learning techniques, Optimization, Python Programming, PyTorch, TensorFlow

Job Description

We welcome candidates at various experience levels for this role. During the interview process, candidates will be assessed for the appropriate level, and offers will align with that level, which may differ from the one in this posting.

Who You Are

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

What We Need

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

What You Will 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.

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, and we are an equal opportunity employer.