tensorrt_llm

Contents:

  • TensorRT-LLM Architecture
  • C++ GPT Runtime
  • The Batch Manager in TensorRT-LLM
  • Multi-head, Multi-query and Group-query Attention
  • Numerical Precision
  • Build TensorRT-LLM
  • Performance of TensorRT-LLM
  • How to debug
  • How to add a new model
  • Graph Rewriting Module
  • Memory Usage of TensorRT-LLM

Python API

  • Layers
  • Functionals
  • Models
  • Plugin
  • Quantization
  • Runtime

C++ API

  • Runtime

Blogs

  • H100 has 4.6x A100 Performance in TensorRT-LLM, achieving 10,000 tok/s at 100ms to first token
  • H200 achieves nearly 12,000 tokens/sec on Llama2-13B with TensorRT-LLM
tensorrt_llm
  • Python Module Index

Python Module Index

t
 
t
- tensorrt_llm
    tensorrt_llm.functional
    tensorrt_llm.layers.activation
    tensorrt_llm.layers.attention
    tensorrt_llm.layers.cast
    tensorrt_llm.layers.conv
    tensorrt_llm.layers.embedding
    tensorrt_llm.layers.linear
    tensorrt_llm.layers.mlp
    tensorrt_llm.layers.normalization
    tensorrt_llm.layers.pooling
    tensorrt_llm.models
    tensorrt_llm.plugin
    tensorrt_llm.quantization
    tensorrt_llm.runtime

© Copyright 2023, NVidia.

Built with Sphinx using a theme provided by Read the Docs.