Sr. Software Engineer, AI Compiler

Tenstorrent · Semiconductors · Toronto, ON · AI Compiler

Software Engineer role focused on developing and optimizing Tenstorrent's MLIR-based AI compiler (TT-Forge) to run AI models efficiently on Tenstorrent hardware. Responsibilities include optimizing computational graphs, creating custom dialects and transformation passes, and potentially developing human-in-the-loop tuning tools.

What you'd actually do

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

Skills

Required

  • C++
  • Python
  • compiler optimization
  • machine learning
  • PyTorch
  • JAX
  • TensorFlow
  • deep systems programming
  • MLIR

Nice to have

  • experience building complex systems that bridge high-level frameworks to low-level execution
  • experience in open-source
  • experience working closely with hardware and software engineers
  • experience in training and multi-chip scaling
  • custom MLIR dialects (TTIR, TTNN, TTKernel)
  • TT-Explorer

What the JD emphasized

  • MLIR-based compiler
  • optimizing computational graphs
  • custom dialects and transformation passes
  • AI hardware revolution
  • U.S. export-controlled technology

Other signals

  • MLIR-based compiler
  • optimizing computational graphs
  • custom dialects and transformation passes
  • enables developers to run AI on Tenstorrent hardware