Open Machine Learning Compiler Framework
☆13,156Mar 3, 2026Updated last week
Alternatives and similar repositories for tvm
Users that are interested in tvm are comparing it to the libraries listed below
Sorting:
- Development repository for the Triton language and compiler☆18,573Updated this week
- Open standard for machine learning interoperability☆20,445Updated this week
- a language for fast, portable data-parallel computation☆6,591Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,753Updated this week
- Compiler for Neural Network hardware accelerators☆3,326May 11, 2024Updated last year
- ncnn is a high-performance neural network inference framework optimized for the mobile platform☆22,868Updated this week
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,348Mar 2, 2026Updated last week
- ☆1,655Sep 11, 2018Updated 7 years ago
- A list of awesome compiler projects and papers for tensor computation and deep learning.☆2,733Oct 19, 2024Updated last year
- compiler learning resources collect.☆2,688Mar 19, 2025Updated 11 months ago
- Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Juli…☆20,828Oct 25, 2023Updated 2 years ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,675Dec 1, 2025Updated 3 months ago
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,621Updated this week
- oneAPI Deep Neural Network Library (oneDNN)☆3,958Updated this week
- MNN: A blazing-fast, lightweight inference engine battle-tested by Alibaba, powering high-performance on-device LLMs and Edge AI.☆14,453Updated this week
- Visualizer for neural network, deep learning and machine learning models☆32,543Updated this week
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆19,435Updated this week
- Transformer related optimization, including BERT, GPT☆6,398Mar 27, 2024Updated last year
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,987Updated this week
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆10,406Updated this week
- Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.☆41,617Updated this week
- 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
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,759Updated this week
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,760Updated this week
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆98,039Updated this week
- "Multi-Level Intermediate Representation" Compiler Infrastructure☆1,764Apr 22, 2021Updated 4 years ago
- Ongoing research training transformer models at scale☆15,535Updated this week
- Dive into Deep Learning Compiler☆646Jun 19, 2022Updated 3 years ago
- A high performance and generic framework for distributed DNN training☆3,717Oct 3, 2023Updated 2 years ago
- TVM integration into PyTorch☆456Jan 15, 2020Updated 6 years ago
- OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.☆9,391Dec 4, 2025Updated 3 months ago
- Optimized primitives for collective multi-GPU communication☆4,495Updated this week
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,642Updated this week
- Fast and memory-efficient exact attention☆22,460Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,928Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆4,049Updated this week
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆14,341Jul 3, 2024Updated last year
- Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators☆1,547Aug 28, 2019Updated 6 years ago
- Simplify your onnx model☆4,304Feb 26, 2026Updated last week