DeepSeek“開源周”第二日,DeepSeek宣布開源DeepEP,第一個用于MoE模型訓(xùn)練和推理的開源EP通信庫。
昨天,DeepSeek則開源了代碼庫Flash MLA,這是針對Hopper GPU優(yōu)化的高效MLA解碼內(nèi)核,針對可變長度序列作了優(yōu)化。
MoE(混合專家架構(gòu))和MLA(多頭潛在注意力機制)被認為是DeepSeek以低成本實現(xiàn)杰出表現(xiàn)的核心原因。
簡單理解,MoE架構(gòu)是由多個專注于垂直技能的專家模型分工協(xié)作來實現(xiàn)最終輸出結(jié)果,訓(xùn)練成本和推理成本更低。有消息稱,GPT-4就使用了MoE架構(gòu),由8個220B模型組成。但MoE架構(gòu)的缺點之一是會增加通信成本。
DeepEP通信庫就是針對通信環(huán)節(jié)的優(yōu)化,其特點包括:高效、優(yōu)化的全員溝通;節(jié)點內(nèi)和節(jié)點間均支持 NVLink 和 RDMA;用于訓(xùn)練和推理預(yù)填充的高吞吐量內(nèi)核;用于推理解碼的低延遲內(nèi)核;原生 FP8 調(diào)度支持;靈活的 GPU 資源控制,實現(xiàn)計算-通信重疊。
MLA則是讓模型預(yù)測更遠位置的token,從而增強語義理解能力。DeepSeek的Flash MLA專為英偉達Hopper GPU打造了高效MLA解碼內(nèi)核,特別針對變長序列進行了優(yōu)化。
責(zé)任編輯:黃學(xué)焚
特別聲明:本網(wǎng)登載內(nèi)容出于更直觀傳遞信息之目的。該內(nèi)容版權(quán)歸原作者所有,并不代表本網(wǎng)贊同其觀點和對其真實性負責(zé)。如該內(nèi)容涉及任何第三方合法權(quán)利,請及時與ts@hxnews.com聯(lián)系或者請點擊右側(cè)投訴按鈕,我們會及時反饋并處理完畢。
- 字節(jié)跳動豆包疑接入deepseek?相關(guān)負責(zé)人:小范圍測試自有模型2025-02-26
- 爆火的騰訊元寶,接入DeepSeek后超越字節(jié)豆包升至免費榜第二2025-02-24
- 本地化部署DeepSeek 福大團隊構(gòu)建醫(yī)療超算大模型2025-02-21
已有0人發(fā)表了評論