tensorrt-llm

Optimizes LLM inference with NVIDIA TensorRT for maximum throughput and lowest latency. Use for production deployment on NVIDIA GPUs (A100/H100), when you need 10-100x faster inference than PyTorch, or for serving models with quantization (FP8/INT4), in-flight batching, and multi-GPU scaling.

By orchestra-research · 851 installs

npx skills add orchestra-research/ai-research-skills --skill tensorrt-llm

Source repository · Upstream listing

TensorRT LLM NVIDIA's open source library for optimizing LLM inference with state of the art performance on NVIDIA GPUs. When to use TensorRT LLM Use TensorRT LLM when: Deploying on NVIDIA GPUs (A100, H100, GB200) Need maximum throughput (24,000+ tokens/sec on Llama 3) Require low latency for real time applications Working with quantized models (FP8, INT4, FP4) Scaling across multiple GPUs or nodes Use vLLM instead when: Need simpler setup and Python first API Want PagedAttention without TensorRT compilation Working with AMD GPUs or non NVIDIA hardware Use llama.cpp instead when: Deploying on CPU or Apple Silicon Need edge deployment without NVIDIA GPUs Want simpler GGUF quantization format Quick start Installation Basic inference Serving with trtllm serve Key features Performance optimizations In flight batching : Dynamic batching during generation Paged KV cache : Efficient memory management Flash Attention : Optimized attention kernels Quantization : FP8, INT4, FP4 for 2 4× faster inference CUDA graphs : Reduced kernel launch overhead Parallelism Tensor parallelism (TP) : Split model across GPUs Pipeline parallelism (PP) : Layer wise distribution Expert parallelism : For Mixture of Experts models Multi node : Scale beyond single machine Advanced features Speculative decoding : Faster generation with draft models LoRA serving : Efficient multi adapter deployment Disaggregated serving : Separate prefill and generation Common patterns Quantized model (FP8) Multi GPU deployment Batch inference Performance benchmarks Meta Llama 3 8B (H100 GPU): Throughput: 24,000 tokens/sec Latency: ~10ms per token vs PyTorch: 100× faster Llama 3 70B (8× A100 80GB): FP8 quantization: 2× faster than FP16 Memory: 50% reduction with FP8 Supported models LLaMA family : Llama 2, Llama 3, CodeLlama GPT family : GPT 2, GPT J, GPT NeoX Qwen : Qwen, Qwen2, QwQ DeepSeek : DeepSeek V2, DeepSeek V3 Mixtral : Mixtral 8x7B, Mixtral 8x22B Vision : LLaVA, Phi 3 vision 100+ models on HuggingFace References [Optimization Guide](references/optimization.md) Quantization, batching, KV cache tuning [Multi GPU Setup](references/multi gpu.md) Tensor/pipeline parallelism, multi node [Serving Guide](references/serving.md) Production deployment, monitoring, autoscaling Resources Docs : https://nvidia.github.io/TensorRT LLM/ GitHub : https://github.com/NVIDIA/TensorRT LLM Models : https://huggingface.co/models?library=tensorrt llm