Sr. Engineer, Software - AI Compiler

Tenstorrent · Semiconductors · Belgrade, Serbia · AI Compiler

Sr. Engineer, Software - AI Compiler role at Tenstorrent focused on developing TT-Forge, an MLIR-based compiler for Tenstorrent hardware, optimizing AI models for training and inference.

What you'd actually do

  1. 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.
  2. A drive to solve novel challenges in AI compilation, from optimizing computational graphs to creating custom dialects and transformation passes.
  3. Experience or strong interest in MLIR and how modular compiler frameworks connect AI models to advanced hardware.
  4. How to build open-source compiler frameworks supporting diverse AI models and workloads, including training and multi-chip scaling.
  5. New methods for human-in-the-loop compiler optimization using TT-Explorer, making advanced tuning tools usable by all developers.

Skills

Required

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

Nice to have

  • RISC-V
  • open-source development

What the JD emphasized

  • leading the industry on cutting-edge AI technology
  • revolutionizing performance expectations
  • unify innovations in software models, compilers, platforms, networking, and semiconductors
  • high performance RISC-V CPU from scratch
  • passion for AI
  • deep desire to build the best AI platform possible
  • revolutionizing AI computing
  • open-source, performant, and general-purpose compiler
  • forefront of the AI hardware revolution
  • building compiler technologies that redefine what’s possible
  • passionate software engineer eager to work on compiler technologies and the challenges of AI hardware
  • Fluent in C++ and Python
  • experience building complex systems that bridge high-level frameworks to low-level execution
  • Excited by compiler optimization and machine learning
  • collaborative problem-solver who thrives in open-source
  • novel challenges in AI compilation
  • optimizing computational graphs
  • creating custom dialects and transformation passes
  • strong interest in MLIR
  • modular compiler frameworks connect AI models to advanced hardware
  • Motivation to build technology that impacts the future of AI
  • enable the next wave of breakthroughs
  • build open-source compiler frameworks supporting diverse AI models and workloads
  • training and multi-chip scaling
  • Deep expertise in compiler technologies
  • custom MLIR dialects (TTIR, TTNN, TTKernel)
  • transformation passes
  • New methods for human-in-the-loop compiler optimization
  • TT-Explorer
  • making advanced tuning tools usable by all developers
  • compiler technology powers Tenstorrent’s mission
  • deliver affordable, open-source AI computing
  • highly competitive space
  • highly competitive compensation package and benefits
  • applicant being eligible to access U.S. export-controlled technology
  • U.S. export laws
  • U.S. Export Administration Regulations (EAR)
  • compliance with these laws
  • nationals of certain countries (such as EAR Country Groups D:1, E1, and E2)
  • access to information, systems, or technologies subject to these laws
  • contingent upon your citizenship/permanent residency status
  • ability to obtain prior license approval from the U.S. Commerce Department or applicable federal agency
  • employment is not possible due to U.S. export laws
  • offer of employment will be rescinded

Other signals

  • MLIR-based compiler
  • AI hardware
  • compiler optimization