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.
☆567Updated this week
Related projects ⓘ
Alternatives and complementary repositories for TensorRT-Model-Optimizer
- QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving☆443Updated last week
- A pytorch quantization backend for optimum☆824Updated last week
- FlashInfer: Kernel Library for LLM Serving☆1,452Updated this week
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆624Updated 2 months ago
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆420Updated this week
- A throughput-oriented high-performance serving framework for LLMs☆636Updated 2 months ago
- TensorRT Plugin Autogen Tool☆367Updated last year
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆1,979Updated this week
- FlagGems is an operator library for large language models implemented in Triton Language.☆342Updated this week
- Advanced Quantization Algorithm for LLMs. This is official implementation of "Optimize Weight Rounding via Signed Gradient Descent for t…☆248Updated this week
- Pipeline Parallelism for PyTorch☆726Updated 2 months ago
- The Triton TensorRT-LLM Backend☆706Updated this week
- Applied AI experiments and examples for PyTorch☆166Updated 2 weeks ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,257Updated 4 months ago
- This repository contains tutorials and examples for Triton Inference Server☆566Updated this week
- [EMNLP 2024 Industry Track] This is the official PyTorch implementation of "LLMC: Benchmarking Large Language Model Quantization with a V…☆322Updated this week
- Common source, scripts and utilities for creating Triton backends.☆295Updated this week
- Microsoft Automatic Mixed Precision Library☆525Updated last month
- A parser, editor and profiler tool for ONNX models.☆400Updated this week
- Common utilities for ONNX converters☆251Updated 5 months ago
- Triton Model Analyzer is a CLI tool to help with better understanding of the compute and memory requirements of the Triton Inference Serv…☆433Updated last week
- Code for Neurips24 paper: QuaRot, an end-to-end 4-bit inference of large language models.☆284Updated 3 months ago
- Mirage: Automatically Generating Fast GPU Kernels without Programming in Triton/CUDA☆636Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆211Updated 3 months ago
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆302Updated 2 months ago
- Actively maintained ONNX Optimizer☆647Updated 8 months ago
- List of papers related to neural network quantization in recent AI conferences and journals.☆458Updated last month
- Model Compression Toolbox for Large Language Models and Diffusion Models☆222Updated last week
- Transformers-compatible library for applying various compression algorithms to LLMs for optimized deployment with vLLM☆685Updated this week
- xDiT: A Scalable Inference Engine for Diffusion Transformers (DiTs) with Massive Parallelism☆714Updated this week