apache / tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
☆11,798Updated this week
Related projects ⓘ
Alternatives and complementary repositories for tvm
- Open standard for machine learning interoperability☆17,949Updated this week
- Compiler for Neural Network hardware accelerators☆3,236Updated 6 months ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆10,820Updated 2 weeks ago
- Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Juli…☆20,782Updated last year
- oneAPI Deep Neural Network Library (oneDNN)☆3,635Updated this week
- MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Co…☆5,796Updated 5 months ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,265Updated 2 months ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,158Updated this week
- Development repository for the Triton language and compiler☆13,443Updated this week
- Visualizer for neural network, deep learning and machine learning models☆28,167Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,415Updated 2 weeks ago
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆8,348Updated this week
- Transformer related optimization, including BERT, GPT☆5,890Updated 7 months ago
- High-efficiency floating-point neural network inference operators for mobile, server, and Web☆1,885Updated this week
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆14,722Updated this week
- CUDA Templates for Linear Algebra Subroutines☆5,679Updated this week
- Serve, optimize and scale PyTorch models in production☆4,218Updated 3 weeks ago
- State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…☆13,580Updated 3 months ago
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,710Updated this week
- Optimized primitives for collective multi-GPU communication☆3,253Updated 2 months ago
- MNN is a blazing fast, lightweight deep learning framework, battle-tested by business-critical use cases in Alibaba☆8,749Updated this week
- ☆1,655Updated 6 years ago
- ncnn is a high-performance neural network inference framework optimized for the mobile platform☆20,503Updated this week
- a language for fast, portable data-parallel computation☆5,906Updated this week
- A high performance and generic framework for distributed DNN training☆3,630Updated last year
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆14,056Updated 4 months ago
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆2,846Updated this week
- Low-precision matrix multiplication☆1,780Updated 9 months ago
- Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators☆1,528Updated 5 years ago
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,351Updated last year