What problem does it solve?
This Skill optimizes LLM inference with NVIDIA TensorRT for maximum throughput and lowest latency, ideal for production deployment on NVIDIA GPUs and serving models with quantization and in-flight batching.
Core Features & Use Cases
- High Throughput and Low Latency: Accelerates inference on NVIDIA GPUs by up to 100x faster than PyTorch.
- Quantization: Supports FP8, INT4, and FP4 quantization for reduced memory usage and faster inference.
- In-Flight Batching: Dynamically batches requests during generation for higher throughput.
- Multi-GPU Scaling: Scales across multiple GPUs or nodes for large-scale inference.
- Use Case: Use this Skill to serve a large language model like Meta-Llama 3-70B in a production environment, achieving real-time latency and high throughput.
Quick Start
Install TensorRT-LLM and serve a Meta-Llama 3-8B model:
pip install tensorrt_llm==1.2.0rc3
trtllm-serve meta-llama/Meta-Llama-3-8B \
--tp_size 4 \
--max_batch_size 256 \
--max_num_tokens 4096