程式忍者PGNinja

GitHub 熱門專案

vLLM:柏克萊團隊打造的大型語言模型推理加速引擎

· 柏克萊實驗室開發的開源推理引擎 vLLM,靠 PagedAttention 技術壓低大型語言模型服務的記憶體成本,提升回應速度。

vLLM 是由美國加州大學柏克萊分校 Sky Computing Lab 發起的開源軟體專案,目標是打造「簡單、快速且便宜」的大型語言模型推理服務引擎。隨著各種 AI 聊天與應用興起,企業要讓模型在真實流量下快速回應、同時控制伺服器成本變得困難,vLLM 就是為了解決這個推理效能與記憶體浪費的問題而生,現在已成長為由兩千多名貢獻者共同維護的大型開源專案。

技術上,vLLM 的核心是 PagedAttention 演算法,把語言模型推理時佔用大量顯示卡記憶體的「鍵值快取」切成固定大小的區塊來管理,減少浪費並支援連續批次處理與前綴快取。專案同時支援 FP8、INT4 等量化方式,整合 FlashAttention 等加速核心,程式碼主要用 Python 撰寫核心框架,再用 Rust 處理效能敏感的部分,可在 NVIDIA、AMD、Intel 的顯示卡與一般 CPU 上執行。

vLLM 這幾天衝上 GitHub 排行榜前段,和它已經支援超過 200 種模型架構有關,從 Llama、Qwen、Gemma 到 DeepSeek-V3 這類混合專家模型與多模態模型都能跑,也提供和 OpenAI API 相容的介面,方便開發者直接替換既有服務。專案背後有 2023 年發表的學術論文當基礎,兼具研究嚴謹度與 production 等級的穩定度,是社群長期信任它的原因。

這則比較偏產業面,不是給剛入門的孩子馬上能動手玩的工具,但背後的概念值得知道:寫出一個程式能動,和讓很多人同時、快速地使用它,是兩個不同層次的工程問題。對已經會寫 Python、對 AI 有興趣的青少年來說,vLLM 的原始碼和技術文件是觀察「真正的大型開源專案怎麼運作」的好教材,例如怎麼分工、怎麼寫測試、怎麼讓程式碼被全世界的人一起維護。

開源AIPython開發工具

其他新聞