Sr. Engineer, Software - AI Compiler

Tenstorrent · Semiconductors · Santa Clara, CA · AI Compiler

Software Engineer role focused on developing and optimizing an MLIR-based AI compiler (TT-Forge) to run AI models efficiently on Tenstorrent hardware. Involves optimizing computational graphs, creating custom dialects, and transformation passes, with a focus on training and multi-chip scaling.

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

  • RISC-V CPU
  • open-source
  • human-in-the-loop compiler optimization

What the JD emphasized

  • This offer of employment is contingent upon the applicant being eligible to access U.S. export-controlled technology.

Other signals

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