What problem does it solve?
Reduce the time, complexity, and risk of deploying high-throughput LLM inference on NVIDIA GPUs by providing a clear workflow for converting HuggingFace checkpoints, building optimized TensorRT-LLM engines, and configuring production serving options.
Core Features & Use Cases
- Convert HuggingFace model checkpoints into TensorRT-LLM checkpoint format suitable for engine building.
- Build optimized TensorRT engines with support for FP8, INT4, and INT8 quantization, GEMM and attention kernel fusion, and multi-profile builds.
- Configure in-flight batching, paged KV cache, and other runtime tuning parameters to maximize tokens/sec throughput.
- Support multi-GPU tensor parallelism and pipeline parallelism for large-model builds and Triton backend integration for production serving.
- Use case: convert a Llama-3 checkpoint, apply FP8 quantization on Hopper GPUs, build rank*.engine files, and deploy the model behind Triton for scalable streaming inference.
Quick Start
Use the tensorrt-llm skill to convert a HuggingFace checkpoint, build an optimized TensorRT engine with the desired quantization, and prepare the engine for Triton serving.