NVIDIA / TensorRT-Model-Optimizer
TensorRT Model Optimizer is a unified library of state-of-the-art model optimization techniques such as quantization, pruning, distillation, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM or TensorRT to optimize inference speed on NVIDIA GPUs.
☆534Updated this week
Related projects ⓘ
Alternatives and complementary repositories for TensorRT-Model-Optimizer
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆611Updated 2 months ago
- FlashInfer: Kernel Library for LLM Serving☆1,395Updated this week
- QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving☆433Updated 2 months ago
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆1,936Updated this week
- A pytorch quantization backend for optimum☆818Updated this week
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,242Updated 3 months ago
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆407Updated this week
- Pipeline Parallelism for PyTorch☆725Updated 2 months ago
- The Triton TensorRT-LLM Backend☆703Updated this week
- Microsoft Automatic Mixed Precision Library☆522Updated last month
- A throughput-oriented high-performance serving framework for LLMs☆629Updated last month
- A parser, editor and profiler tool for ONNX models.☆397Updated last month
- Mirage: Automatically Generating Fast GPU Kernels without Programming in Triton/CUDA☆597Updated this week
- FlagGems is an operator library for large language models implemented in Triton Language.☆328Updated this week
- This repository contains tutorials and examples for Triton Inference Server☆558Updated this week
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆457Updated 7 months ago
- Common utilities for ONNX converters☆251Updated 4 months ago
- [EMNLP 2024 Industry Track] This is the official PyTorch implementation of "LLMC: Benchmarking Large Language Model Quantization with a V…☆315Updated this week
- Code for Neurips24 paper: QuaRot, an end-to-end 4-bit inference of large language models.☆278Updated 3 months ago
- LLaMa/RWKV onnx models, quantization and testcase☆350Updated last year
- PyTorch native quantization and sparsity for training and inference☆1,549Updated this week
- Advanced Quantization Algorithm for LLMs. This is official implementation of "Optimize Weight Rounding via Signed Gradient Descent for t…☆245Updated this week
- ☆283Updated 7 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆479Updated 2 weeks ago
- depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.☆499Updated this week
- A Easy-to-understand TensorOp Matmul Tutorial☆287Updated last month
- Common source, scripts and utilities for creating Triton backends.☆293Updated this week
- TensorRT Plugin Autogen Tool☆367Updated last year
- Flash Attention in ~100 lines of CUDA (forward pass only)☆609Updated 7 months ago
- This repository contains the experimental PyTorch native float8 training UX☆211Updated 3 months ago