Sr. Software Engineer, AI Compiler

Join the team revolutionizing AI computing at Tenstorrent. You'll work on TT-Forge, our MLIR-based compiler that enables developers to run AI on all configurations of Tenstorrent hardware using an open-source, performant, and general-purpose compiler. You will be at the forefront of the AI hardware revolution, building compiler technologies that redefine what’s possible.

 This role is hybrid and based out of Toronto, ON.

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

• A passionate software engineer eager to work on compiler technologies and the challenges of AI hardware, whether from compilers, systems, or broader software backgrounds.
• Fluent in C++ and Python, with experience building complex systems that bridge high-level frameworks to low-level execution.
• Excited by compiler optimization and machine learning, with experience in PyTorch, JAX, TensorFlow, or deep systems programming.
• A collaborative problem-solver who thrives in open-source and enjoys working closely with hardware and software engineers.

 

What We Need

• A drive to solve novel challenges in AI compilation, from optimizing computational graphs to creating custom dialects and transformation passes.
• Experience or strong interest in MLIR and how modular compiler frameworks connect AI models to advanced hardware.
• Motivation to build technology that impacts the future of AI, knowing your work will enable the next wave of breakthroughs.

 

What You Will Learn

• How to build open-source compiler frameworks supporting diverse AI models and workloads, including training and multi-chip scaling.
• Deep expertise in compiler technologies including custom MLIR dialects (TTIR, TTNN, TTKernel) and transformation passes.
• New methods for human-in-the-loop compiler optimization using TT-Explorer, making advanced tuning tools usable by all developers.
• How compiler technology powers Tenstorrent’s mission to deliver affordable, open-source AI computing in a highly competitive space.