進階Python

TensorRT-LLMNVIDIA 優化大模型推理的開源庫

TensorRT-LLM 是 NVIDIA 的開源庫,用於在 NVIDIA GPU 上優化大語言模型和視覺生成模型的推理。它提供 PyTorch 原生的 Python API,底層為 C++ 運行時,並包含用於注意力、GEMM 和混合專家的自定義 CUDA 內核。運行時特性包括預填充解碼分離、推測解碼和廣泛專家並行,支持多 GPU 和多節點部署。倉庫內置 DeepSeek-V3 等預構建模型配方,並與 Triton 推理集成。主要語言為 Python,許可證為 Other。

13.9K 星標
2.5K 分叉
1.4K 問題
135 流覽
Python
Other
收錄日期

專案概述

TensorRT-LLM 是 NVIDIA 的開源庫,用於在 NVIDIA GPU 上優化大語言模型和視覺生成模型的推理。它提供 PyTorch 原生的 Python API,底層為 C++ 運行時,並包含用於注意力、GEMM 和混合專家的自定義 CUDA 內核。運行時特性包括預填充解碼分離、推測解碼和廣泛專家並行,支持多 GPU 和多節點部署。倉庫內置 DeepSeek-V3 等預構建模型配方,並與 Triton 推理集成。主要語言為 Python,許可證為 Other。

項目簡介

TensorRT-LLM 是 NVIDIA 推出的開源庫,專注於在 NVIDIA GPU 上優化大語言模型和視覺生成模型的推理性能。

核心功能

  • 提供 PyTorch 原生的 Python API,基於 C++ 運行時。
  • 包含自定義 CUDA 內核,用於注意力機制、GEMM 和混合專家(MoE)。
  • 運行時支持預填充解碼分離、推測解碼和廣泛專家並行。
  • 支持通過多種並行策略進行多 GPU 和多節點部署。

技術棧

主要程式語言為 Python,底層為 C++ 運行時和 CUDA 內核,併集成 Triton 推理伺服器。

許可證

許可證類型為 Other。

獲取與上手

倉庫中提供了預構建模型配方,例如 DeepSeek-V3,用戶可參考這些配方開始使用。

TensorRT-LLMNVIDIA大語言模型推理GPU 優化開源高效能推理LLM 部署Python API量化推理

項目評分

0.0 (0 評價)

分享

常見問題

TensorRT-LLM: NVIDIA 優化大模型推理的開源庫 是什麼?

TensorRT-LLM 是 NVIDIA 的開源庫,用於在 NVIDIA GPU 上優化大語言模型和視覺生成模型的推理。它提供 PyTorch 原生的 Python API,底層為 C++ 運行時,並包含用於注意力、GEMM 和混合專家的自定義 CUDA 內核。運行時特性包括預填充解碼分離、推測解碼和廣泛專家並行,支持多 GPU 和多節點部署。倉庫內置 DeepSeek-V3 等預構建模型配方,並與 Triton 推理集成。主要語言為 Python,許可證為 Other。

TensorRT-LLM: NVIDIA 優化大模型推理的開源庫 用什麼語言開發?

TensorRT-LLM: NVIDIA 優化大模型推理的開源庫 主要使用 Python 開發。

TensorRT-LLM: NVIDIA 優化大模型推理的開源庫 使用什麼開源授權?

TensorRT-LLM: NVIDIA 優化大模型推理的開源庫 基於 Other 授權開源。

相關專案

暫無結果

探索更多

相似工具

Cursor

Cursor

一款基於 VS Code 二次開發的智慧程式碼編輯器,以「原生內建 AI」為核心賣點。它不依賴外掛,而是將 AI 深度植入編輯器底層,能夠理解整個專案的上下文程式碼庫,支援無縫遷移 VS Code 的所有配置和外掛。

Google Antigravity

Google Antigravity

Antigravity 支援多模型,包括 Gemini 3 Pro、Claude Sonnet 4.5、GPT-OSS,開發者可以在同一環境中選擇最適合任務的模型。

Codex

Codex

OpenAI Codex 是由 OpenAI 開發的 AI 程式設計模型和助手,可將自然語言指令翻譯成對應的原始碼,為開發者提供智慧補全、程式碼生成等功能。它最初於 2021 年作為 OpenAI API 的程式碼模型推出,曾為 GitHub Copilot 提供核心支援。隨著 OpenAI 技術的迭代,Codex 在 2025 年以「AI 程式設計智慧體」的全新姿態迴歸,能夠理解複雜需求並自動編寫、除錯程式碼,顯著提升開發效率和軟體交付速度。

Kiro

Kiro

Kiro 是由 AWS 推出的 AI 程式設計 IDE,採用規範驅動的開發模式,將自然語言需求轉化為明確的規格文件和任務,再由內建 AI 代理生成程式碼並除錯優化,全流程輔助大型專案開發。

Trae

Trae

Trae(官網 trae.ai)是由 位元組跳動(ByteDance)推出的一款 AI 原生整合開發環境(IDE)。它不是簡單地作為一個程式設計助手,而是一個「協作夥伴」,通過深度整合大型語言模型(LLM),幫助開發者從需求、構建程式碼,到除錯和部署,實現更智慧化、自動化的軟體開發。

Claude

Claude

Claude 是由美國人工智慧公司 Anthropic 打造的智慧語言互動平臺,它融合了深度文字理解、資訊整理、程式碼輔助和任務分析等能力,能在聊天對話之外應對更復雜的問題,例如長文摘要、影象解析、邏輯推理及程式設計協助等。相比一些單一問答機器人,Claude 更像一個具備推理邏輯、可擴充套件功能的智慧工具。

評論

評論

0
0/500 字元

暫無評論

成為第一個評論的人

開源專案

探索、學習和貢獻開源 AI 專案,推動人工智慧技術的發展

查看全部