apache / tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
☆12,259Updated 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,895Updated this week
- Compiler for Neural Network hardware accelerators☆3,289Updated 11 months ago
- oneAPI Deep Neural Network Library (oneDNN)☆3,783Updated this week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,459Updated 2 weeks ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,543Updated this week
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,111Updated this week
- Development repository for the Triton language and compiler☆15,447Updated this week
- a language for fast, portable data-parallel computation☆6,047Updated this week
- NumPy & SciPy for GPU☆10,181Updated this week
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆16,455Updated this week
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,381Updated this week
- Simplify your onnx model☆4,065Updated 8 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,645Updated 3 weeks ago
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆32,148Updated 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,811Updated 11 months ago
- Visualizer for neural network, deep learning and machine learning models☆30,133Updated this week
- A library for efficient similarity search and clustering of dense vectors.☆34,661Updated this week
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆14,179Updated 10 months ago
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆9,157Updated this week
- ncnn is a high-performance neural network inference framework optimized for the mobile platform☆21,418Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,135Updated this week
- Quantized Neural Network PACKage - mobile-optimized implementation of quantized neural network operators☆1,540Updated 5 years ago
- MNN is a blazing fast, lightweight deep learning framework, battle-tested by business-critical use cases in Alibaba. Full multimodal LLM …☆10,662Updated this week
- State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…☆14,217Updated 8 months ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆89,697Updated this week
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,596Updated this week
- Low-precision matrix multiplication☆1,802Updated last year
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,390Updated 2 years ago
- Flax is a neural network library for JAX that is designed for flexibility.☆6,527Updated this week
- Build and run Docker containers leveraging NVIDIA GPUs☆17,366Updated last year