What problem does it solve?
TensorRT-LLM solves the challenge of slow and costly LLM inference by optimizing model execution on NVIDIA GPUs to achieve much higher throughput and lower latency than typical PyTorch serving.
Core Features & Use Cases
- High-performance inference optimization: Uses in-flight batching, paged KV cache, Flash Attention kernels, CUDA graphs, and quantization (FP8/INT4/FP4) to maximize tokens/sec and reduce tail latency.
- Scalable parallel serving: Supports tensor parallelism (TP), pipeline parallelism (PP), and expert parallelism (EP) for single-node and multi-node scaling of large models.
- Production-ready serving: Provides an OpenAI-compatible server mode (trtllm-serve) with monitoring endpoints and operational settings for production deployments.
- Use Case: You need real-time chat responses for a production app on NVIDIA A100/H100 hardware, and you must reduce per-request latency while increasing sustained throughput under concurrent traffic.
Quick Start
Use the tensorrt-llm skill to start an OpenAI-compatible server for a Llama 3 model by running the trtllm-serve command against meta-llama/Meta-Llama-3-8B with tensor parallel size and batching settings, then send a POST request to the /v1/chat/completions endpoint.