Search
Go
Software Engineer, AI Compiler
AUSTIN · US
$100K–$500K
Full-time
About the Role
Tenstorrent is hiring a Software Engineer, AI Compiler to lead development of TT-Forge, an MLIR-based compiler. The role involves scaling graph transformations, lowering passes, and kernel-level optimizations for AI models on custom hardware. This hybrid position in Austin, TX is open to candidates at various seniority levels.
Responsibilities
- Lead development on TT-Forge MLIR-based compiler
- Manage team focused on graph transformations, lowering passes, and kernel optimizations
- Collaborate with hardware, software, and ML teams
- Drive compiler roadmap and mentor engineers
Requirements
- Experience building compilers with strong proficiency in C++ and Python
- Knowledge of MLIR, LLVM, or similar infrastructure
- Understanding of modern AI frameworks like PyTorch, TensorFlow, or JAX
- Background in cross-team collaboration for reliable, scalable tools
Tech Stack
PythonPyTorchTensorFlowJAXC++MLIRLLVM
Tenstorrent
Toronto · CA · 1005+ employees
Tenstorrent is a next-generation computing company that builds computers for AI, specializing in computer architecture, ASIC design, RISC-V technology, and neural network compilers. They provide computing systems designed to support AI workloads and offer an open-source compiler, TT-Forge2, to optimize AI workloads for their hardware.
