本文將分享阿里云人工智能平臺 PAI 團隊與 NVIDIA Megatron-Core 團隊在 MoE (Mixture of Experts) 大語言模型(LLM)實現(xiàn)與訓(xùn)練優(yōu)化上的創(chuàng)新工作。分享內(nèi)容將按以下脈絡(luò)展開:
首先簡短回顧 MoE 技術(shù)的發(fā)展歷程,提煉核心概念及其在實踐應(yīng)用中亟待解決的關(guān)鍵挑戰(zhàn)。
接著詳述雙方合作研發(fā)的 MoE 框架所具有的獨特能力和卓越性能,展示其在提升模型訓(xùn)練效率、資源利用以及模型表現(xiàn)等方面取得的驗證結(jié)果。
最后,扼要介紹阿里云基于此合作成果所搭建的平臺工具及推薦的最佳實踐方案,賦能開發(fā)者高效運用 MoE 技術(shù),促進大規(guī)模模型訓(xùn)練的深入探索與廣泛應(yīng)用。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIA Megatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
MoE 是一種模型結(jié)構(gòu),它將稠密模型結(jié)構(gòu)拆分為多個子結(jié)構(gòu),每個子結(jié)構(gòu)成為一個專家,通過在訓(xùn)練和推理過程中動態(tài)選擇一組專家進行計算,實現(xiàn)了模型參數(shù)的稀疏更新。
簡而言之,MoE 模型將整體模型拆分為多個專業(yè)子模塊(專家),每次僅激活和更新少數(shù)與輸入相關(guān)的專家子結(jié)構(gòu),而選擇哪些專家參與計算則是通過一個路由機制決定。在 GPT 等超大規(guī)模語言模型中,采用MoE 技術(shù)能夠顯著減少訓(xùn)練和推理時的計算負擔(dān),因為不是所有參數(shù)都需要在每次操作時都更新,這極大提高了訓(xùn)練效率,并且在推理階段僅使用部分活躍的網(wǎng)絡(luò)參數(shù),極大地削減了計算資源需求。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
2023 年末,Mistral AI 推出開源 MoE 架構(gòu)大語言模型 Mixtral 8x7B,憑借 46.7B 參數(shù)量,在多項下游任務(wù)榜單的效果勝過當(dāng)時的最佳開源稠密模型 Llama-2 70B。右側(cè)圖表揭示,在同等推理資源條件下,MoE 模型性能顯著優(yōu)于稠密模型。這一成就引發(fā)業(yè)界對 MoE 模型的強烈關(guān)注,進而推動阿里云與 NVIDIA Megatron-Core 團隊,共同深化在大模型領(lǐng)域的 MoE 技術(shù)合作與應(yīng)用。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
為了描述使用 MoE 結(jié)構(gòu)來實現(xiàn)工業(yè)級應(yīng)用需要解決的問題,本文首先回顧 MoE 的發(fā)展歷史。90 年代初提出的多專家模型概念為 MoE 打下了理論基礎(chǔ),即通過集合多個專家模型協(xié)同解決任務(wù),并采用門控路由進行專家選擇。
隨著 2017 年深度學(xué)習(xí)突飛猛進,參數(shù)量對模型性能的重要性日益顯現(xiàn),傳統(tǒng)稠密結(jié)構(gòu)深度學(xué)習(xí)框架面臨容量瓶頸。谷歌率先將 MoE 與深度學(xué)習(xí)結(jié)合,首次在 RNN 中驗證了 MoE 的可行性。至 2020 年,Transformer 結(jié)構(gòu)在語言模型領(lǐng)域展現(xiàn)出卓越效果和優(yōu)秀的可擴展性。此后,Google Gshard 項目將 MoE 融入 Transformer,通過深度設(shè)計與實驗顯著提升效果和性能,并在 Switch Transformers 中將參數(shù)總量推向萬億級別,奠定了 MoE 架構(gòu)在大模型領(lǐng)域的基礎(chǔ)。
后續(xù)研究如 ST-MoE、Tutel MoE、FasterMOE 和 MegaBlocks 等,針對速度和穩(wěn)定性進行深入探索。2023 年,GPT-4 的驚艷表現(xiàn)引發(fā)了業(yè)內(nèi)對其是否采用 MoE 結(jié)構(gòu)的熱議。同年,Mixtral 的實現(xiàn)似乎驗證了這一猜想,表明 MoE 在大模型架構(gòu)中的應(yīng)用正逐步走向成熟和廣泛認可。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
后續(xù)我們將簡要剖析幾個標(biāo)志性工作,首先是將 MoE 與深度學(xué)習(xí)融為一體的 Sparsely-Gated MoE。該研究將 RNN 網(wǎng)絡(luò)劃分為多個專家子網(wǎng)絡(luò),并采用 Softmax Gating 機制來控制 token 的路由分配。然而,原始 Softmax Gating 邏輯可能導(dǎo)致負載不均衡,隨著訓(xùn)練推進,部分專家網(wǎng)絡(luò)可能過于活躍,而其他專家的參數(shù)訓(xùn)練不足。
為解決此問題,研究者在計算 Softmax 之前,對門控矩陣參數(shù)和輸入特征進行噪聲 (noise) 注入,以實現(xiàn)更為均衡的專家選擇。此外,在門控網(wǎng)絡(luò)損失函數(shù) (loss function) 中增設(shè)輔助損失,引導(dǎo)模型實現(xiàn)更佳的負載分配。同時,路由策略上嘗試每次挑選 top-2 至 top-4 的專家,這些優(yōu)化舉措最終助力模型參數(shù)容量首次突破千億門檻。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
而谷歌的 Gshard 項目進一步將 MoE 應(yīng)用于 Transformer 架構(gòu)中,其主要創(chuàng)新包括:
設(shè)計了 MoE Transformer 的基本結(jié)構(gòu)和并行模式,每間隔一層前向?qū)硬捎?MoE 層替代,專家分布在不同設(shè)備上獨立計算,其他層的參數(shù)共享。
針對專家負載均衡問題,提出了“專家容量(expert capacity)”概念,限制每個專家處理的 token 數(shù)量,并采用殘差連接策略繞過已滿負荷的專家。
引入輔助損失與隨機 top-k 路由策略,以優(yōu)化專家選擇過程。
在通信效率上,創(chuàng)新提出“l(fā)ocal group dispatching”方案,通過門控網(wǎng)絡(luò)預(yù)篩選后再按專家 ID 分組傳輸數(shù)據(jù),有效提升通信效率。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
實驗結(jié)果顯示,Gshard 成功將 Transformer 參數(shù)量從 150B 提升至 600B,而計算開銷僅增加 50%,首次驗證了 MoE 在模型擴展性上的顯著優(yōu)勢。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
之后的 Switch Transformer 工作對 MoE Transformer 的訓(xùn)練流程進行了深度優(yōu)化,以實現(xiàn)模型參數(shù)規(guī)模的極限拓展。首先,路由策略選擇了更為激進的 top-1 方案,即每次僅選擇一個專家進行參數(shù)更新。此舉不僅能顯著降低路由的計算負擔(dān),與 top-2 策略相比計算量近乎減半,同時每個專家處理的 batch size 也因此減少,進一步減輕了通信成本。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
其次,針對先前存在的專家負載均衡問題,工作引入了輔助損失函數(shù)。該函數(shù)中,N 代表專家總數(shù),f 表示第 i 個專家分配到的 token 比例,P 代表路由器分配給第i個專家的概率。優(yōu)化目標(biāo)旨在促使 token 均勻分配至每個專家。通過最小化損失函數(shù),使得 f 和 P 趨向于 N 分之一,此時損失函數(shù)中的調(diào)節(jié)參數(shù) α 通常取值 0.01 左右,能夠?qū)崿F(xiàn)較好的負載均衡效果。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
在分布式訓(xùn)練方面,除了先前提及的數(shù)據(jù)并行和專家并行外,該工作對非專家結(jié)構(gòu)部分實施了模型并行策略,來進一步降低單卡顯存需求,使得模型參數(shù)量得以顯著提升。
針對路由負載均衡引起的模型訓(xùn)練隨機性與不穩(wěn)定問題,該工作提出了一系列穩(wěn)定訓(xùn)練的策略。其中包括在本地計算時采用較高的 FP32 精度,而在通信階段使用較低的 BF16 精度的隨機精度策略;采用具有較小縮放系數(shù)的正態(tài)分布初始化權(quán)重;引入專家 dropout 即專家內(nèi)部的 dropout 技術(shù)以減少過擬合現(xiàn)象。
通過這些優(yōu)化措施,MoE transformer 的訓(xùn)練收斂速度相較于稠密模型在迭代次數(shù)和時間上均有數(shù)倍提升。其中,最大的 Switch-C 模型參數(shù)量高達 1.5 萬億。后續(xù)的 ST-MoE 工作則更深入地探究了如何進一步改善 MoE 模型的訓(xùn)練穩(wěn)定性和提升模型性能。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
在預(yù)訓(xùn)練穩(wěn)定性方面,為解決專家路由隨機性和數(shù)值精度造成的 roundoff 誤差,即路由網(wǎng)絡(luò)計算 logits 值過大導(dǎo)致的訓(xùn)練不收斂問題,工作提出了 z-loss 函數(shù)。該函數(shù)中,B 表示一個 batch 的 token 數(shù)量,N 代表專家數(shù)量,X 則對應(yīng)輸入路由網(wǎng)絡(luò)的 logits 的維度。通過引入 z-loss 有效抑制了由精度等因素導(dǎo)致的logits 值過高現(xiàn)象,從而增強了模型訓(xùn)練的穩(wěn)定性,z-loss 同樣適用于稠密 LLM 訓(xùn)練的穩(wěn)定性提升。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
此外,該工作還深入探究了影響 MoE 訓(xùn)練穩(wěn)定性和效果均衡的多種因素,并通過細致實驗歸納出一系列最佳實踐。
譬如,向路由網(wǎng)絡(luò)引入適量噪聲 (noise),例如 dropout 操作,能夠增強訓(xùn)練穩(wěn)定性,但可能模型效果受損。在調(diào)優(yōu) (Fine-tuning) 階段,ST-MoE 的研究進一步發(fā)現(xiàn),增大 dropout 參數(shù)對稠密 MoE 模型和非稠密 MoE 模型的影響存在顯著差異,對于稠密模型影響更大。同時,與稠密模型不同,小型 batch size 和較大學(xué)習(xí)率 (learning rate) 對 MoE 模型訓(xùn)練效果的提升更為有利。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
Mixtral 模型在實現(xiàn)上借鑒了 Megablocks 框架的獨特思路,該框架的主要特點是將 MoE 層的計算表述為塊稀疏運算。面對前面提及的動態(tài)路由問題,即不同專家處理的 token 數(shù)量各異,導(dǎo)致用戶在計算過程中需抉擇使用 drop token 或 padding,前者關(guān)乎模型性能,后者則會增加訓(xùn)練成本。
Megablocks 創(chuàng)新性地將多個矩陣乘的操作統(tǒng)一定義為一個大型塊稀疏矩陣,即將多位專家的計算視作一個固定尺寸的大矩陣,其中的計算任務(wù)則細分為多個小矩陣塊的 GEMM(General Matrix Multiply)操作。
為優(yōu)化這種塊稀疏矩陣的處理,Megablocks 框架利用了 Block Compressed Sparse Row (BCSR) 數(shù)據(jù)結(jié)構(gòu),對矩陣的行和列訪問速度以及轉(zhuǎn)換操作進行了優(yōu)化。通過這種方法實現(xiàn)了路由矩陣的高效操作。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
上述 MoE 發(fā)展史詳細說明了其開發(fā)需要解決的問題,下文將簡介阿里云為解決這些問題,與 NVIDIA Megatron-Core 團隊的技術(shù)合作及產(chǎn)出。Megatron-Core 是一個成熟且輕量化的大規(guī)模 LLM 訓(xùn)練框架,集成了訓(xùn)練大規(guī)模 LLM 的核心技術(shù),比如多元化的模型并行支持、算子優(yōu)化、通信優(yōu)化、顯存優(yōu)化,以及低精度訓(xùn)練(如 FP8)等先進技術(shù)。
Megatron-Core 沿襲了 Megatron-LM 的優(yōu)秀能力,并在代碼質(zhì)量、穩(wěn)定性、功能完備度及測試覆蓋范圍等維度全面提升。尤為關(guān)鍵的是,該框架設(shè)計上更注重解耦和模塊化,開發(fā)者在做二次開發(fā)或探索新模型架構(gòu)時享有高度靈活性。因此我們選擇了與 Megatron-Core 團隊開展合作。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
接下來探討 Megatron-Core 對 MoE 架構(gòu)的主要支持特性。在并行化方面,Megatron-Core MoE 不僅支持專家并行,還支持 3D 并行,包括數(shù)據(jù)并行、張量并行、流水并行以及序列并行等。對于超大規(guī)模 MoE 模型,它能夠靈活地將專家并行與其他并行策略有機結(jié)合。
在 token 分發(fā)機制上,Megatron-Core MoE 采用了 dropless MoE 操作,即不丟棄任何 token。在路由和負載均衡優(yōu)化層面,它支持多種路由策略,如通用的 top-k,并在負載均衡算法上支持 Sinkhorn 算法、z-loss 以及 load balancing loss 等多種方案。
此外,為解決多個專家接收變長輸入問題,Megatron-Core MoE 引入了 GroupedGMM 技術(shù),并優(yōu)化效率較低的操作,將其替換為優(yōu)化的 CUDA kernel。
同時,在模型遷移與適配上,Megatron-Core MoE 提供了豐富的模型 checkpoint 轉(zhuǎn)換功能,允許用戶導(dǎo)入HuggingFace 模型,并自由調(diào)整 TP(tensor parallelism)、PP(pipeline parallelism)和 EP(expert parallelism)等結(jié)構(gòu),隨后利用 Megatron-Core 高效啟動模型訓(xùn)練任務(wù)。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
阿里云人工智能平臺 PAI 團隊基于 Megatron-Core,開發(fā)了一套簡易、高效的大模型訓(xùn)練工具。依托阿里云靈駿計算服務(wù),團隊推出了 PAI-Megatron-Patch 工具庫,可實現(xiàn)從十余種主流開源大模型的模型格式到Megatron-LM 和 Megatron-Core 的無縫轉(zhuǎn)換。用戶通過 PAI DSW 和 DLC 等產(chǎn)品,能輕松啟動 Megatron-LM 及 Megatron-Core 進行大規(guī)模稠密模型和 MoE 模型訓(xùn)練。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
針對開源大模型多采用 HuggingFace 格式,與 Megatron 訓(xùn)練框架存在一定差異的問題,PAI-Megatron-patch 的主要功能在于提供從 HuggingFace 模型到 Megatron 框架的權(quán)重轉(zhuǎn)換服務(wù)。該轉(zhuǎn)換實質(zhì)上是對模型命名空間進行映射,涵蓋了 layernorm 層、attention 層以及 MLP 層等核心組件的定義轉(zhuǎn)換。PAI-Megatron-patch 內(nèi)置簡潔的權(quán)重轉(zhuǎn)換腳本,使得用戶能夠便捷地執(zhí)行模型格式轉(zhuǎn)換操作,極大地簡化了遷移流程。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
在進行 MoE 模型轉(zhuǎn)換時,需確保轉(zhuǎn)換后的文件能直接用于分布式訓(xùn)練,為此我們也做了大量驗證工作。以Mixtral 模型為例,在擁有 8 個專家和 16 張卡的場景下,采取 TP=4 和 EP=4 的切分策略。每個 TP 分區(qū)內(nèi)的文件夾中包含四個文件,存儲全部八個專家的 FFN 權(quán)重,即每個文件承載兩位專家的 FFN 權(quán)重信息。
經(jīng)過轉(zhuǎn)換,模型在零樣本(Zero-shot)損失精度方面的表現(xiàn)如途中下表所示,數(shù)據(jù)顯示轉(zhuǎn)換前后模型的精度差異非常微小,確保了模型轉(zhuǎn)換的有效性和準(zhǔn)確性。
我們的工作不僅關(guān)注模型的轉(zhuǎn)換環(huán)節(jié),同樣嚴(yán)謹(jǐn)?shù)仳炞C了整個訓(xùn)練流程的穩(wěn)健性。仍以 Mixtral 8x7B 模型為例,我們在三個訓(xùn)練階段——從頭預(yù)訓(xùn)練、基于 checkpoint 的續(xù)訓(xùn)及指令微調(diào) (Finetune),均進行了細致測試。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
在從頭預(yù)訓(xùn)練階段,我們設(shè)定的參數(shù)包括:global_size=256,LR (learning rate) =1e-4,seq_len=2048,TP=4。經(jīng)過 24 小時至 2.4K 個訓(xùn)練步驟后,損失 loss 成功收斂至約 1.9。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
在繼續(xù)預(yù)訓(xùn)練階段,設(shè)定 global_size = 256,LR=5e-5,seq_len=2048,TP=4。在 18 小時達到 2,000 個訓(xùn)練步驟后,loss 亦表現(xiàn)出正常的收斂行為。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
針對 Finetune 階段,我們做了三組測試:
橙色曲線代表 finetune base 模型,設(shè)定 global_size=128,LR=1e-5,seq_len=2048,TP=4。
藍色曲線代表 finetune instruct 模型,其參數(shù)配置與橙色曲線一致。
紅色曲線代表僅計算 answer loss 的 finetune instruct 模型,其 global_size=32,LR 依舊為1e-5,seq_len 調(diào)整為 128,TP 依舊為 4。
所有模型在訓(xùn)練過程中均呈現(xiàn)出正常收斂態(tài)勢,其中 finetune base 模型導(dǎo)致 loss 下降幅度較大,這一現(xiàn)象符合預(yù)期。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
為了進一步驗證 Finetune 的效果,我們選取了一個代碼生成任務(wù)作為評估基準(zhǔn)。我們利用大約 80K 條公開的代碼訓(xùn)練樣本,測試集在 HumanEvol 平臺上進行評估,此處的參數(shù)配置沿襲了上述 SFT 設(shè)定。經(jīng)過 2,500 步訓(xùn)練后,模型在 HumanEvol 上的性能指標(biāo)從最初的 45.73% 顯著提升至 53.05%,有力證明了訓(xùn)練流程的合理性和有效性。此外,在速度對比方面,該方法優(yōu)于同等資源條件下運行的 Megablocks。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
我們通過阿里云人工智能平臺 PAI 提供了一套官方最佳實踐指南,指南是對我們從數(shù)據(jù)處理、訓(xùn)練、推理、評估、直至在線服務(wù)部署的完整 AI 開發(fā)流程的測試成果總結(jié):
https://mp.weixin.qq.com/s?__biz=Mzg4MzgxNDk2OA==&mid=2247492431&idx=1&sn=1e135a9e61ac65e88a156969d66ab5be&chksm=cf430418f8348d0e1ed97db178536e73d3ae7093e428c6e3e25bc0dc84a5754f49e3c20ce98e&cur_album_id=2918365856378880004&scene=189#wechat_redirect
該指南涵蓋了從數(shù)據(jù)準(zhǔn)備階段起,指導(dǎo)用戶如何從 OSS (對象存儲服務(wù))和 NAS (網(wǎng)絡(luò)附加存儲)讀取原始數(shù)據(jù),并在 PAI DSW 環(huán)境中執(zhí)行高效的數(shù)據(jù)預(yù)處理操作。預(yù)處理后的數(shù)據(jù)能夠方便地回存至 NAS 或 OSS,以為后續(xù)的模型訓(xùn)練做好準(zhǔn)備。
這種方式的模型訓(xùn)練支持在 PAI DLC 進行大規(guī)模分布式訓(xùn)練,同時也兼容 DSW 提供的單機訓(xùn)練環(huán)境,確保用戶可根據(jù)實際需求靈活選擇訓(xùn)練方式。訓(xùn)練完成后,模型的 checkpoint 可以直接導(dǎo)出至 OSS 或 NAS 存儲系統(tǒng)。
完成訓(xùn)練的 checkpoint 可先進行離線推理和模型性能評估。一旦推理和評估驗證無誤,開發(fā)者只需一鍵即可部署至 EAS 提供的模型在線服務(wù)。依托此服務(wù)的接口,開發(fā)者可以輕松構(gòu)建各類 APP 和業(yè)務(wù)場景。
圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講
《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》
這份最佳實踐指南提供了詳細的步驟說明,確保用戶能清晰掌握每一步驟的操作方法。
未來,阿里云人工智能平臺 PAI 團隊將繼續(xù)深化與 NVIDIA Megatron-Core 團隊的合作,致力于在密集型和稀疏型模型的訓(xùn)練表現(xiàn)和效率上取得更大突破,為推進 AGI(通用人工智能)技術(shù)的發(fā)展貢獻力量。我們熱忱歡迎全球開發(fā)者共同參與到開源社區(qū)項目以及阿里云的建設(shè)之中,攜手共進,共創(chuàng)智能未來。
審核編輯:劉清
-
NVIDIA
+關(guān)注
關(guān)注
14文章
5078瀏覽量
103770 -
人工智能
+關(guān)注
關(guān)注
1796文章
47734瀏覽量
240411 -
GPT
+關(guān)注
關(guān)注
0文章
360瀏覽量
15523 -
阿里云
+關(guān)注
關(guān)注
3文章
976瀏覽量
43247 -
LLM
+關(guān)注
關(guān)注
0文章
300瀏覽量
407
原文標(biāo)題:GTC24|SE63222 線上演講回顧:基于 NVIDIA Megatron Core 的 MOE LLM 實現(xiàn)和訓(xùn)練優(yōu)化
文章出處:【微信號:NVIDIA-Enterprise,微信公眾號:NVIDIA英偉達企業(yè)解決方案】歡迎添加關(guān)注!文章轉(zhuǎn)載請注明出處。
發(fā)布評論請先 登錄
相關(guān)推薦
解析DeepSeek MoE并行計算優(yōu)化策略
![解析DeepSeek <b class='flag-5'>MoE</b>并行計算<b class='flag-5'>優(yōu)化</b>策略](https://file1.elecfans.com/web3/M00/07/65/wKgZO2elYF-AGKjNAABJXwhqoLs327.png)
在NVIDIA TensorRT-LLM中啟用ReDrafter的一些變化
![在<b class='flag-5'>NVIDIA</b> TensorRT-<b class='flag-5'>LLM</b>中啟用ReDrafter的一些變化](https://file1.elecfans.com/web3/M00/03/BF/wKgZPGdr0WaAWiI-AAATzNjpSz0481.png)
解鎖NVIDIA TensorRT-LLM的卓越性能
NVIDIA TensorRT-LLM Roadmap現(xiàn)已在GitHub上公開發(fā)布
![<b class='flag-5'>NVIDIA</b> TensorRT-<b class='flag-5'>LLM</b> Roadmap現(xiàn)已在GitHub上公開發(fā)布](https://file1.elecfans.com/web3/M00/00/42/wKgZO2dH2UeALMQdAABTS5uU1Rg392.png)
TensorRT-LLM低精度推理優(yōu)化
![TensorRT-<b class='flag-5'>LLM</b>低精度推理<b class='flag-5'>優(yōu)化</b>](https://file1.elecfans.com/web1/M00/F5/62/wKgZoWc8MUuACQJOAAARWw9PFe0355.jpg)
如何訓(xùn)練自己的LLM模型
LLM和傳統(tǒng)機器學(xué)習(xí)的區(qū)別
端到端InfiniBand網(wǎng)絡(luò)解決LLM訓(xùn)練瓶頸
![端到端InfiniBand網(wǎng)絡(luò)解決<b class='flag-5'>LLM</b><b class='flag-5'>訓(xùn)練</b>瓶頸](https://file1.elecfans.com/web2/M00/0A/02/wKgZomcYbZ2AQS_eAAApjamVIRI773.png)
NVIDIA Nemotron-4 340B模型幫助開發(fā)者生成合成訓(xùn)練數(shù)據(jù)
![<b class='flag-5'>NVIDIA</b> Nemotron-4 340B模型幫助開發(fā)者生成合成<b class='flag-5'>訓(xùn)練</b>數(shù)據(jù)](https://file1.elecfans.com/web2/M00/05/86/wKgZombaqFaAYqNYAAEgri7KFqc769.jpg)
LLM預(yù)訓(xùn)練的基本概念、基本原理和主要優(yōu)勢
llm模型訓(xùn)練一般用什么系統(tǒng)
llm模型和chatGPT的區(qū)別
進一步解讀英偉達 Blackwell 架構(gòu)、NVlink及GB200 超級芯片
NVIDIA加速微軟最新的Phi-3 Mini開源語言模型
MOE與MOT:提升LLM效能的關(guān)鍵策略比較
![<b class='flag-5'>MOE</b>與MOT:提升<b class='flag-5'>LLM</b>效能的關(guān)鍵策略比較](https://file1.elecfans.com/web2/M00/C8/79/wKgZomYciRKADBy3AAAmknEk_yw738.png)
評論