tenstorrent / tt-metal
TT-NN operator library, and TT-Metalium low level kernel programming model.
☆829Updated this week
Alternatives and similar repositories for tt-metal:
Users that are interested in tt-metal are comparing it to the libraries listed below
- Tenstorrent TT-BUDA Repository☆313Updated 3 weeks ago
- Tenstorrent MLIR compiler☆120Updated this week
- ⭐️ TTNN Compiler for PyTorch 2 ⭐️ It enables running PyTorch models on Tenstorrent hardware using torch.compile path☆36Updated this week
- An MLIR-based toolchain for AMD AI Engine-enabled devices.☆363Updated this week
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆389Updated this week
- Backward compatible ML compute opset inspired by HLO/MHLO☆467Updated this week
- torchtrail: trace the graph of torch functions and modules for visualization, reports, etc☆25Updated 10 months ago
- Awesome resources for GPUs☆559Updated last year
- The TT-Forge FE is a graph compiler designed to optimize and transform computational graphs for deep learning models, enhancing their per…☆40Updated this week
- An unofficial cuda assembler, for all generations of SASS, hopefully ：）☆486Updated 2 years ago
- Stores documents and resources used by the OpenXLA developer community☆120Updated 8 months ago
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,503Updated this week
- Frontend integration for PyTorch with tt-mlir☆14Updated this week
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆594Updated 2 months ago
- Fast CUDA matrix multiplication from scratch☆697Updated last year
- IREE's PyTorch Frontend, based on Torch Dynamo.☆79Updated this week
- Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure☆846Updated this week
- Tenstorrent Kernel Module☆41Updated this week
- Nvidia Instruction Set Specification Generator☆256Updated 9 months ago
- AI Tensor Engine for ROCm☆180Updated this week
- Shared Middle-Layer for Triton Compilation☆246Updated last week
- hipBLASLt is a library that provides general matrix-matrix operations with a flexible API and extends functionalities beyond a traditiona…☆91Updated this week
- Stretching GPU performance for GEMMs and tensor contractions.☆237Updated this week
- A comprehensive tool for visualizing and analyzing model execution, offering interactive graphs, memory plots, tensor details, buffer ove…☆31Updated this week
- GPUOcelot: A dynamic compilation framework for PTX☆187Updated 2 months ago
- PyTorch emulation library for Microscaling (MX)-compatible data formats☆217Updated last week
- Tenstorrent's MLIR Based Compiler. We aim to enable developers to run AI on all configurations of Tenstorrent hardware, through an open-s…☆43Updated this week
- An experimental CPU backend for Triton☆105Updated 2 weeks ago
- Repository of model demos using TT-Buda☆63Updated 3 weeks ago
- ctypes wrappers for HIP, CUDA, and OpenCL☆129Updated 9 months ago