What problem does it solve?
Deploying and serving large language models on NVIDIA GPUs for production use requires extreme throughput, low latency, efficient memory usage, and careful handling of quantization and parallelism; achieving this with PyTorch alone is often too slow or memory-intensive. TensorRT-LLM compiles and optimizes models for NVIDIA hardware, enabling dramatic speedups (10–100× vs PyTorch), reduced memory footprint with FP8/INT4, and reliable multi-GPU scaling for production inference.
Core Features & Use Cases
- High-performance inference: Model compilation, CUDA graphs, flash attention, and in-flight batching to maximize tokens/sec and stabilize P99 latency.
- Quantized serving: FP8, INT4 and AWQ/GPTQ workflows to reduce memory and increase throughput for H100/A100 deployments.
- Multi-GPU & multi-node scaling: Tensor parallelism, pipeline parallelism, expert parallelism, NVLink/InfiniBand guidance and Kubernetes/Ray deployment patterns.
- Production readiness: OpenAI-compatible trtllm-serve, Prometheus metrics, health probes, autoscaling examples, and best-practice security and cost-optimization tips.
- Use case example: Serve a Meta Llama 3-70B model across 4 A100/H100 GPUs with FP8 quantization for high-throughput chat completions and low-latency streaming.
Quick Start
Start a TensorRT-LLM server hosting meta-llama/Meta-Llama-3-8B with FP8 and tensor parallelism across 4 GPUs for high-throughput, low-latency inference.