Sr. Software Engineer (AI Compiler)

Location: Hybrid and based out of Toronto, ON, Canada

Compensation: $100k - $500k including base and variable

Company

Tenstorrent leads in cutting-edge AI technology, developing high-performance RISC-V CPUs, compilers, platforms, and semiconductors to revolutionize AI computing.

What you will do

• Develop TT-Forge, the open-source MLIR-based compiler for all Tenstorrent hardware configurations.
• Optimize computational graphs, create custom dialects (TTIR, TTNN, TTKernel), and build transformation passes.
• Bridge high-level AI frameworks like PyTorch, JAX, TensorFlow to low-level hardware execution.
• Enable training, multi-chip scaling, and human-in-the-loop optimization with tools like TT-Explorer.
• Collaborate with hardware and software engineers on novel AI compilation challenges.

Requirements

• Fluent in C++ and Python with experience building complex systems bridging frameworks to execution.
• Experience in compiler optimization, machine learning frameworks (PyTorch, JAX, TensorFlow), or deep systems programming.
• Strong interest in MLIR and modular compiler frameworks for AI hardware.
• Eligible to access U.S. export-controlled technology (may require citizenship/permanent residency or U.S. license approval).
• Collaborative problem-solver thriving in open-source environments.

Culture & Benefits

• Highly competitive compensation package and benefits.
• Value collaboration, curiosity, and solving hard problems in a diverse team.
• Equal opportunity employer focused on open-source AI computing.