A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
☆1,003Sep 19, 2024Updated last year
Alternatives and similar repositories for nnfusion
Users that are interested in nnfusion are comparing it to the libraries listed below
Sorting:
- The Tensor Algebra SuperOptimizer for Deep Learning☆740Jan 26, 2023Updated 3 years ago
- A list of awesome compiler projects and papers for tensor computation and deep learning.☆2,734Oct 19, 2024Updated last year
- BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.☆921Dec 30, 2024Updated last year
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆199Apr 27, 2022Updated 3 years ago
- Antares: an automatic engine for multi-platform kernel generation and optimization. Supporting CPU, CUDA, ROCm, DirectX12, GraphCore, SYC…☆464Apr 20, 2025Updated 11 months ago
- Automatic Schedule Exploration and Optimization Framework for Tensor Computations☆183Apr 25, 2022Updated 3 years ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆125Jun 23, 2022Updated 3 years ago
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,770Mar 13, 2026Updated last week
- ☆423Feb 24, 2026Updated 3 weeks ago
- Open Machine Learning Compiler Framework☆13,197Updated this week
- MegCC是一个运行时超轻量，高效，移植简单的深度学习模型编译器☆484Oct 23, 2024Updated last year
- A performant and modular runtime for TensorFlow☆753Sep 4, 2025Updated 6 months ago
- Dive into Deep Learning Compiler☆647Jun 19, 2022Updated 3 years ago
- Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training☆1,864Mar 12, 2026Updated last week
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,661Updated this week
- Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure☆983Mar 13, 2026Updated last week
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆121Oct 26, 2022Updated 3 years ago
- An open-source efficient deep learning framework/compiler, written in python.☆738Sep 4, 2025Updated 6 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆144Mar 31, 2023Updated 2 years ago
- A primitive library for neural network☆1,367Nov 24, 2024Updated last year
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆477Mar 15, 2024Updated 2 years ago
- ☆1,995Jul 29, 2023Updated 2 years ago
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,442Updated this week
- A model compilation solution for various hardware☆467Aug 20, 2025Updated 7 months ago
- An unofficial cuda assembler, for all generations of SASS, hopefully ：）☆572Apr 20, 2023Updated 2 years ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆50Jul 23, 2024Updated last year
- Graphiler is a compiler stack built on top of DGL and TorchScript which compiles GNNs defined using user-defined functions (UDFs) into ef…☆59Oct 3, 2022Updated 3 years ago
- compiler learning resources collect.☆2,693Mar 19, 2025Updated last year
- ☆192Mar 28, 2023Updated 2 years ago
- Distributed Compiler based on Triton for Parallel Systems☆1,386Mar 11, 2026Updated last week
- ☆145Jan 30, 2025Updated last year
- The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs☆1,350Apr 14, 2025Updated 11 months ago
- ☆166Jul 22, 2024Updated last year
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆752Aug 6, 2025Updated 7 months ago
- DietCode Code Release☆65Jul 21, 2022Updated 3 years ago
- common in-memory tensor structure☆1,177Jan 26, 2026Updated last month
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,543Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,709Updated this week
- A language and compiler for irregular tensor programs.☆152Nov 29, 2024Updated last year