AKG (Auto Kernel Generator) is an optimizer for operators in Deep Learning Networks, which provides the ability to automatically fuse ops with specific patterns.
☆244Dec 13, 2025Updated 2 months ago
Alternatives and similar repositories for akg
Users that are interested in akg are comparing it to the libraries listed below
Sorting:
- MindSpore is a new open source deep learning training/inference framework that could be used for mobile, edge and cloud scenarios.☆4,677Jul 29, 2024Updated last year
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆1,005Sep 19, 2024Updated last year
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆121Oct 26, 2022Updated 3 years ago
- BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.☆917Dec 30, 2024Updated last year
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆182Apr 25, 2022Updated 3 years ago
- A model compilation solution for various hardware☆464Aug 20, 2025Updated 6 months ago
- ☆192Mar 28, 2023Updated 2 years ago
- The Tensor Algebra SuperOptimizer for Deep Learning☆739Jan 26, 2023Updated 3 years ago
- A home for the final text of all TVM RFCs.☆109Sep 24, 2024Updated last year
- A language and compiler for irregular tensor programs.☆152Nov 29, 2024Updated last year
- Bridging polyhedral analysis tools to the MLIR framework☆119Sep 9, 2023Updated 2 years ago
- ☆423Feb 24, 2026Updated last week
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆125Jun 23, 2022Updated 3 years ago
- Shared Middle-Layer for Triton Compilation☆331Dec 5, 2025Updated 3 months ago
- Symbolic Expression and Statement Module for new DSLs☆205Oct 6, 2020Updated 5 years ago
- A list of tutorials, paper, talks, and open-source projects for emerging compiler and architecture☆520Jan 15, 2025Updated last year
- Pluto: An automatic polyhedral parallelizer and locality optimizer☆324Aug 21, 2025Updated 6 months ago
- A list of awesome compiler projects and papers for tensor computation and deep learning.☆2,733Oct 19, 2024Updated last year
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,760Updated this week
- C/C++ frontend for MLIR. Also features polyhedral optimizations, parallel optimizations, and more!☆605Jun 19, 2025Updated 8 months ago
- An MLIR-based compiler framework bridges DSLs (domain-specific languages) to DSAs (domain-specific architectures).☆696Updated this week
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆478Mar 15, 2024Updated last year
- ☆13Nov 25, 2019Updated 6 years ago
- An Attention Superoptimizer☆22Jan 20, 2025Updated last year
- MegCC是一个运行时超轻量，高效，移植简单的深度学习模型编译器☆485Oct 23, 2024Updated last year
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆199Apr 27, 2022Updated 3 years ago
- HeteroHalide: From Image Processing DSL to Efficient FPGA Acceleration☆15Sep 14, 2020Updated 5 years ago
- heterogeneity-aware-lowering-and-optimization☆257Jan 20, 2024Updated 2 years ago
- ☆250Jul 27, 2025Updated 7 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆143Mar 31, 2023Updated 2 years ago
- A Easy-to-understand TensorOp Matmul Tutorial☆409Updated this week
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Feb 20, 2026Updated 2 weeks ago
- Tengine 管子是用来快速生产 demo 的辅助工具☆12Jul 15, 2021Updated 4 years ago
- An open-source efficient deep learning framework/compiler, written in python.☆737Sep 4, 2025Updated 6 months ago
- ☆95Nov 4, 2022Updated 3 years ago
- Polyhedral Parallel Code Generation (source repository: http://repo.or.cz/ppcg.git)☆130Jul 22, 2022Updated 3 years ago
- play gemm with tvm☆92Jul 22, 2023Updated 2 years ago
- Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure☆981Updated this week
- ☆12Jul 9, 2021Updated 4 years ago