What problem does it solve?
TensorRT-LLM reduces latency and increases throughput for large language model inference on NVIDIA GPUs by leveraging TensorRT optimizations, enabling production-grade deployment on GPUs such as A100 and H100. It supports quantization (FP8/INT4), in-flight batching, and multi-GPU scaling to meet real-time or high-throughput requirements.
Core Features & Use Cases
- Performance optimizations: In-flight batching, paged KV cache, CUDA graphs, and quantization for speedups and reduced memory footprint.
- Parallelism & scaling: Tensor parallelism, pipeline parallelism, expert parallelism for Mixture-of-Experts models, and multi-node deployments for very large models.
- Production-ready serving: OpenAI-compatible API, automatic model download/compilation, metrics, health checks, and deployment patterns for robust inference services.
Quick Start
Start a TensorRT-LLM deployment on NVIDIA GPUs to serve an LLM with FP8 quantization and multi-GPU parallelism.