Software Engineer, Ai Compiler

Tenstorrent

Austin, United States
Base: $100k - $500k; bonus/equity: not specified; ...
On-site
Mlir-based compiler development
Graph transformations and optimization
Risc-v cpu architecture knowledge
Lead development on TT-Forge, an MLIR-based compiler, managing a team focused on scaling graph transformations, lowering passes, and kernel-level optimizations

Job Summary

  • Lead development on TT-Forge, an MLIR-based compiler, managing a team focused on scaling graph transformations, lowering passes, and kernel-level optimizations.
  • Shape the future of AI computing through compiler technology that is fast, flexible, and built for real-world models.
  • Learn to build compiler infrastructure supporting training and inference at scale across custom chip architectures and evolve MLIR dialects.

Matching Summary

Lead development on TT-Forge, an MLIR-based compiler, managing a team focused on scaling graph transformations, lowering passes, and kernel-level optimizations.

Salary

Base: $100k - $500k; Bonus/Equity: Not specified; Benefits: Not specified

Skills & Requirements

Must-have

  • MLIR-based compiler development
  • Graph transformations and optimization
  • RISC-V CPU architecture knowledge
  • C++ and Python fluency
  • Modern AI frameworks (PyTorch, TensorFlow, JAX)

Nice-to-have

  • Leadership and team mentorship
  • Performance debugging and bottleneck identification
  • Collaboration with hardware and ML teams
  • Developer workflow improvement

Key Requirements

  • Experience building compilers or similar systems
  • Strong knowledge of MLIR, LLVM, or related infrastructure
  • Hands-on work in dialect design or optimization passes
  • Experience shipping reliable, scalable tools in fast-paced environments

Work Rights

Eligibility to access U.S. export-controlled technology

Tailored Resume

Cover Letter