chengzeyi/stable-fast

TensorRT Extension for Stable Diffusion Web UI

🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization tools

[ICLR2025 Spotlight] SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Models

xDiT: A Scalable Inference Engine for Diffusion Transformers (DiTs) with Massive Parallelism

Scalable and memory-optimized training of diffusion models

A unified library of SOTA model optimization techniques like quantization, distillation, pruning, neural architecture search, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM, TensorRT, vLLM, etc. to optimize inference speed.

Karras et al. (2022) diffusion models for PyTorch

TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT