Devotd Github
新发现推理与服务

nvidia/tensorrt-llm v1.2.1 — 面向 NVIDIA GPU 的 LLM 推理优化

梳理用于优化 NVIDIA GPU 上 LLM 与 Visual Gen 模型推理的 nvidia/tensorrt-llm 的构成及其最新版本 v1.2.1。

仓库信息nvidia/tensorrt-llm ↗

梳理用于优化 NVIDIA GPU 上 LLM 与 Visual Gen 模型推理的 nvidia/tensorrt-llm 的构成及其最新版本 v1.2.1。

GitHub 项目说明: TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.

星标
★ 14.9K
复刻
2.8K
语言
Python
许可证
NOASSERTION
最新版本
v1.2.1
查看原始来源 ↗在 GitHub 查看 ↗

nvidia/tensorrt-llm 是一款在 NVIDIA GPU 上提升大规模语言模型(LLM)推理效率的工具。它把定义 LLM 的 Python API,以及协调推理执行的 Python 与 C++ 运行时组件集中在一个仓库中。该仓库的最新版本是 v1.2.1,而本次发布的变更记录在所提供的资料中没有记载。

README 说明该项目用于优化 LLM 与 Visual Gen 模型的推理。其核心构成包括面向通用运算的专用内核(specialized kernel)、高效的运行时,以及便于用户定制和扩展系统的 Pythonic 框架。这些优化以 NVIDIA GPU 为前提。

该仓库以 Python 编写,Star 数为 14902,GitHub 记录的创建日期为 2023 年 8 月 16 日。许可证字段标记为 NOASSERTION,因此无法确认具体的许可证类型。

不过,性能指标和基准测试在所提供的资料中均不存在,与其他硬件的对比也未给出。评估项中只有 momentum 的值为 5.5,其余项均为空。v1.2.1 究竟有哪些改动,资料中没有说明,需要直接查看仓库的发布页面。其适用范围限定于已经拥有 NVIDIA GPU 的环境,不适合希望获得不依赖硬件的通用推理栈的用户。