apache / tvmLinks
Open deep learning compiler stack for cpu, gpu and specialized accelerators
☆12,758Updated this week
Alternatives and similar repositories for tvm
Users that are interested in tvm are comparing it to the libraries listed below
Sorting:
- Open standard for machine learning interoperability☆19,759Updated last week
- oneAPI Deep Neural Network Library (oneDNN)☆3,901Updated this week
- Compiler for Neural Network hardware accelerators☆3,313Updated last year
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,413Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,276Updated last month
- Development repository for the Triton language and compiler☆17,289Updated last week
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆18,186Updated this week
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,531Updated last week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,617Updated last month
- Transformer related optimization, including BERT, GPT☆6,331Updated last year
- A high performance and generic framework for distributed DNN training☆3,708Updated 2 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,832Updated this week
- Optimized primitives for collective multi-GPU communication☆4,182Updated last week
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆9,938Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,610Updated last week
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆8,665Updated last week
- Simplify your onnx model☆4,217Updated 2 months ago
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,688Updated this week
- Ongoing research training transformer models at scale☆13,976Updated this week
- A list of awesome compiler projects and papers for tensor computation and deep learning.☆2,665Updated last year
- High-efficiency floating-point neural network inference operators for mobile, server, and Web☆2,144Updated this week
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆14,287Updated last year
- Visualizer for neural network, deep learning and machine learning models☆31,624Updated 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,815Updated 2 months ago
- PyTorch extensions for high performance and large scale training.☆3,384Updated 6 months ago
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,657Updated this week
- Reference implementations of MLPerf® training benchmarks☆1,719Updated last week
- ☆1,932Updated 2 years ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,874Updated this week
- NumPy & SciPy for GPU☆10,563Updated this week