PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
☆2,947Updated this week
Alternatives and similar repositories for TensorRT
Users that are interested in TensorRT are comparing it to the libraries listed below
Sorting:
- An easy to use PyTorch to TensorRT converter☆4,857Aug 17, 2024Updated last year
- ONNX-TensorRT: TensorRT backend for ONNX☆3,188Feb 3, 2026Updated 3 weeks ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,702Feb 13, 2026Updated 2 weeks ago
- The Triton Inference Server provides an optimized cloud and edge inferencing solution.☆10,375Updated this week
- Simplify your onnx model☆4,298Jan 29, 2026Updated 3 weeks ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,634Feb 19, 2026Updated last week
- Transformer related optimization, including BERT, GPT☆6,394Mar 27, 2024Updated last year
- Implementation of popular deep learning networks with TensorRT network definition API☆7,680Updated this week
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,926Feb 16, 2026Updated last week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,706Jan 12, 2026Updated last month
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,075Apr 17, 2024Updated last year
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on H…☆3,170Updated this week
- Development repository for the Triton language and compiler☆18,460Updated this week
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.☆2,652Jan 22, 2026Updated last month
- TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizat…☆12,938Updated this week
- Open Machine Learning Compiler Framework☆13,142Updated this week
- Deploy your model with TensorRT quickly.☆764Nov 21, 2023Updated 2 years ago
- A pytorch to tensorrt convert with dynamic shape support☆266Feb 18, 2024Updated 2 years ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,397Updated this week
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,315Updated this week
- ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator☆19,389Updated this week
- convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.☆600Feb 18, 2024Updated 2 years ago
- Serve, optimize and scale PyTorch models in production☆4,359Aug 6, 2025Updated 6 months ago
- Fast and memory-efficient exact attention☆22,361Updated this week
- A primitive library for neural network☆1,367Nov 24, 2024Updated last year
- 🐍 Geometric Computer Vision Library for Spatial AI☆11,079Feb 17, 2026Updated last week
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,353Updated this week
- A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.☆1,006Sep 19, 2024Updated last year
- OpenMMLab Model Deployment Framework☆3,100Sep 30, 2024Updated last year
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,497Feb 18, 2026Updated last week
- Accelerate PyTorch models with ONNX Runtime☆368Feb 5, 2026Updated 3 weeks ago
- ONNX Optimizer☆797Feb 4, 2026Updated 3 weeks ago
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,860Updated this week
- A GPU performance profiling tool for PyTorch models☆509Jul 13, 2021Updated 4 years ago
- State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…☆14,735Aug 12, 2024Updated last year
- TVM integration into PyTorch☆456Jan 15, 2020Updated 6 years ago
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,648Updated this week
- A unified library of SOTA model optimization techniques like quantization, pruning, distillation, speculative decoding, etc. It compresse…☆2,031Updated this week