jiazhihao / TASOView external linksLinks
The Tensor Algebra SuperOptimizer for Deep Learning
☆741Jan 26, 2023Updated 3 years ago
Alternatives and similar repositories for TASO
Users that are interested in TASO are comparing it to the libraries listed below
Sorting:
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆1,006Sep 19, 2024Updated last year
- Re-implementation of the TASO compiler using equality saturation☆138Jun 28, 2021Updated 4 years ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆125Jun 23, 2022Updated 3 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆199Apr 27, 2022Updated 3 years ago
- Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training☆1,860Updated this week
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆182Apr 25, 2022Updated 3 years ago
- A list of awesome compiler projects and papers for tensor computation and deep learning.☆2,731Oct 19, 2024Updated last year
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆142Mar 31, 2023Updated 2 years ago
- BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.☆916Dec 30, 2024Updated last year
- ☆422Jan 4, 2026Updated last month
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,745Updated this week
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆121Oct 26, 2022Updated 3 years ago
- Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure☆973Feb 6, 2026Updated last week
- Dive into Deep Learning Compiler☆644Jun 19, 2022Updated 3 years ago
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,348Apr 14, 2025Updated 10 months ago
- Repository for SysML19 Artifacts Evaluation☆53Feb 28, 2019Updated 6 years ago
- A language and compiler for irregular tensor programs.☆151Nov 29, 2024Updated last year
- ☆192Mar 28, 2023Updated 2 years ago
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆477Mar 15, 2024Updated last year
- Open Machine Learning Compiler Framework☆13,117Updated this week
- ☆145Jan 30, 2025Updated last year
- Lightweight and Parallel Deep Learning Framework☆264Nov 26, 2022Updated 3 years ago
- Place for meetup slides☆140Oct 11, 2020Updated 5 years ago
- A polyhedral compiler for expressing fast and portable data parallel algorithms☆955Nov 20, 2024Updated last year
- ☆68Mar 4, 2023Updated 2 years ago
- Benchmark scripts for TVM☆74Mar 15, 2022Updated 3 years ago
- AKG (Auto Kernel Generator) is an optimizer for operators in Deep Learning Networks, which provides the ability to automatically fuse ops…☆245Dec 13, 2025Updated 2 months ago
- HeteroCL: A Multi-Paradigm Programming Infrastructure for Software-Defined Heterogeneous Computing (FPGA'19 Best Paper)☆341Apr 20, 2024Updated last year
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,530Updated this week
- A schedule language for large model training☆152Aug 21, 2025Updated 5 months ago
- "Multi-Level Intermediate Representation" Compiler Infrastructure☆1,762Apr 22, 2021Updated 4 years ago
- An experimental ahead of time compiler for Relay.☆49Apr 21, 2020Updated 5 years ago
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,604Updated this week
- Symbolic Expression and Statement Module for new DSLs☆205Oct 6, 2020Updated 5 years ago
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,266Updated this week
- heterogeneity-aware-lowering-and-optimization☆257Jan 20, 2024Updated 2 years ago
- Compiler for Neural Network hardware accelerators☆3,324May 11, 2024Updated last year
- DietCode Code Release☆65Jul 21, 2022Updated 3 years ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,074Apr 17, 2024Updated last year