tenstorrent/tt-torch

Frontend integration for PyTorch with tt-mlir

26

stars

889

commits

Python

primary language

Aug 28, 2026

updated

docs.tenstorrent.com/tt-torch/

README

Hardware | Documentation | Discord | Join Us | Bounty $

tt-torch

NOTE: TT-Torch is deprecated. To work with PyTorch and the various features available in TT-Torch, please see the documentation for TT-XLA.

tt-torch is a PyTorch2.0 and torch-mlir based front-end for tt-mlir.

Quick Links

What is this Repo?

The TT-Torch repository is a PyTorch-based front end compiler that lets developers write standard PyTorch models as well as compile and run those models on Tenstorrent AI accelerators. TT-Torch is a bridge between PyTorch models, MLIR dialects (Tenstorrent-specific IRs like ttir and ttnn), and low-level hardware execution on Tenstorrent chips.

Tenstorrent Bounty Program Terms and Conditions

This repo is a part of Tenstorrent’s bounty program. If you are interested in helping to improve tt-forge, please make sure to read the Tenstorrent Bounty Program Terms and Conditions before heading to the issues tab. Look for the issues that are tagged with both “bounty” and difficulty level!

Contributors

(top 30 of 31)

221 commits

123 commits

97 commits

97 commits

Similar repos

tenstorrent/tt-xla

Repo for AI Compiler team. The intended purpose of this repo is for implementation of a PJRT…

75

shared contributors

tenstorrent/tt-forge

Tenstorrent's MLIR Based Compiler. We aim to enable developers to run AI on all configurations of…

362

shared contributors

tenstorrent/tt-mlir

Tenstorrent MLIR compiler

306

shared contributors

tenstorrent/tt-forge-fe

The TT-Forge ONNX is a graph compiler designed to optimize and transform computational graphs for…

66

shared contributors

tenstorrent/tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.

1,666

shared contributors

llvm/torch-mlir

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR…

1,907

ucb-bar/model2MLIR

Any model in Torch or JAX format to MLIR frontend dialects.

1

nod-ai/AMD-SHARK-ModelDev

Unified compiler/runtime for interfacing with PyTorch Dynamo.

100

Languages

Python

96.0%

C++

3.5%

tenstorrent/tt-torch

Frontend integration for PyTorch with tt-mlir

26

stars

889

commits

Python

primary language

Aug 28, 2026

updated

docs.tenstorrent.com/tt-torch/

README

Hardware | Documentation | Discord | Join Us | Bounty $

tt-torch

NOTE: TT-Torch is deprecated. To work with PyTorch and the various features available in TT-Torch, please see the documentation for TT-XLA.

tt-torch is a PyTorch2.0 and torch-mlir based front-end for tt-mlir.

Quick Links

What is this Repo?

The TT-Torch repository is a PyTorch-based front end compiler that lets developers write standard PyTorch models as well as compile and run those models on Tenstorrent AI accelerators. TT-Torch is a bridge between PyTorch models, MLIR dialects (Tenstorrent-specific IRs like ttir and ttnn), and low-level hardware execution on Tenstorrent chips.

Tenstorrent Bounty Program Terms and Conditions

This repo is a part of Tenstorrent’s bounty program. If you are interested in helping to improve tt-forge, please make sure to read the Tenstorrent Bounty Program Terms and Conditions before heading to the issues tab. Look for the issues that are tagged with both “bounty” and difficulty level!

Similar repos

tenstorrent/tt-xla

Repo for AI Compiler team. The intended purpose of this repo is for implementation of a PJRT…

75

shared contributors

tenstorrent/tt-forge

Tenstorrent's MLIR Based Compiler. We aim to enable developers to run AI on all configurations of…

362

shared contributors

tenstorrent/tt-mlir

Tenstorrent MLIR compiler

306

shared contributors

tenstorrent/tt-forge-fe

The TT-Forge ONNX is a graph compiler designed to optimize and transform computational graphs for…

66

shared contributors

tenstorrent/tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.

1,666

shared contributors

llvm/torch-mlir

The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR…

1,907

ucb-bar/model2MLIR

Any model in Torch or JAX format to MLIR frontend dialects.

1

nod-ai/AMD-SHARK-ModelDev

Unified compiler/runtime for interfacing with PyTorch Dynamo.

100

Contributors

(top 30 of 31)

221 commits

123 commits

97 commits

97 commits

Languages

Python

96.0%

C++

3.5%