apache / tvmLinks
Open deep learning compiler stack for cpu, gpu and specialized accelerators
☆12,435Updated this week
Alternatives and similar repositories for tvm
Users that are interested in tvm are comparing it to the libraries listed below
Sorting:
- oneAPI Deep Neural Network Library (oneDNN)☆3,830Updated this week
- Open standard for machine learning interoperability☆19,234Updated this week
- Compiler for Neural Network hardware accelerators☆3,312Updated last year
- a language for fast, portable data-parallel computation☆6,124Updated this week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,531Updated last week
- Development repository for the Triton language and compiler☆16,114Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,828Updated this week
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆9,443Updated this week
- High-efficiency floating-point neural network inference operators for mobile, server, and Web☆2,060Updated this week
- CUDA Templates for Linear Algebra Subroutines☆7,808Updated this week
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,457Updated this week
- Transformer related optimization, including BERT, GPT☆6,231Updated last year
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,209Updated this week
- Low-precision matrix multiplication☆1,809Updated last year
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,341Updated this week
- A collection of pre-trained, state-of-the-art models in the ONNX format☆8,793Updated 2 weeks ago
- A high performance and generic framework for distributed DNN training☆3,683Updated last year
- Tutorials for creating and using ONNX models☆3,565Updated 11 months ago
- Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Juli…☆20,805Updated last year
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,655Updated 3 months ago
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆17,136Updated this week
- "Multi-Level Intermediate Representation" Compiler Infrastructure☆1,750Updated 4 years ago
- Optimized primitives for collective multi-GPU communication☆3,848Updated 3 weeks ago
- MACE is a deep learning inference framework optimized for mobile heterogeneous computing platforms.☆5,012Updated last year
- Simplify your onnx model☆4,114Updated 10 months ago
- NumPy & SciPy for GPU☆10,323Updated this week
- An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.☆2,906Updated 2 years ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆91,472Updated this week
- Visualizer for neural network, deep learning and machine learning models☆30,800Updated this week
- OpenBLAS is an optimized BLAS library based on GotoBLAS2 1.13 BSD version.☆6,857Updated this week