apache / tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
☆12,197Updated this week
Alternatives and similar repositories for tvm:
Users that are interested in tvm are comparing it to the libraries listed below
- Open standard for machine learning interoperability☆18,802Updated this week
- Compiler for Neural Network hardware accelerators☆3,279Updated 11 months ago
- oneAPI Deep Neural Network Library (oneDNN)☆3,769Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,454Updated last month
- MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Co…☆5,811Updated 10 months ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,448Updated 2 months ago
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆16,248Updated this week
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆9,060Updated this week
- a language for fast, portable data-parallel computation☆6,022Updated 2 weeks ago
- Transformer related optimization, including BERT, GPT☆6,123Updated last year
- Visualizer for neural network, deep learning and machine learning models☆29,925Updated this week
- CUDA Templates for Linear Algebra Subroutines☆7,275Updated last week
- Optimized primitives for collective multi-GPU communication☆3,641Updated 3 weeks ago
- A flexible framework of neural networks for deep learning☆5,911Updated last year
- Tutorials for creating and using ONNX models☆3,489Updated 9 months ago
- Development repository for the Triton language and compiler☆15,217Updated this week
- Simplify your onnx model☆4,050Updated 7 months ago
- Low-precision matrix multiplication☆1,799Updated last year
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆89,003Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,616Updated this week
- Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Juli…☆20,797Updated last year
- Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators☆1,538Updated 5 years ago
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,092Updated this week
- ncnn is a high-performance neural network inference framework optimized for the mobile platform☆21,308Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,627Updated 2 weeks ago
- Caffe: a fast open framework for deep learning.☆34,302Updated 8 months ago
- A high performance and generic framework for distributed DNN training☆3,675Updated last year
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,087Updated this week
- High-efficiency floating-point neural network inference operators for mobile, server, and Web☆1,998Updated this week
- A collection of pre-trained, state-of-the-art models in the ONNX format☆8,473Updated 11 months ago