DeepSpeed v0.19.8 擴充 AutoTP、AutoEP 與加速器支援

快訊 大型語言模型訓練基礎設施

  • DeepSpeed v0.19.8 為 AutoTP 加入詞彙平行 LM head 與 Embedding,並強化 tied model 和 row-parallel output head 訓練支援。
  • AutoEP 新增跨 MoE layers 共用 DeepEP buffer、選擇性區域編譯與融合運算等功能,也調整 checkpoint placement 和 dispatch replay。
  • 此版本為 NPU 加入 graph operations、FusedLion、FusedLamb 與 Quantizer 運算,並新增 KLXPU 加速器。
  • DeepSpeed 移除 elastic training、Progressive Layer Dropping 與 graph harvesting,並將 APEX amp 列為棄用功能。

本站見解

若工作流程仰賴已移除的功能,升級前需先調整訓練設定或評估是否暫留舊版;APEX amp 也應納入後續遷移規劃。發布說明沒有指出這些功能的替代方案,因此具體遷移方式仍待確認。

來源

  1. DeepSpeed v0.19.8 Patch Release · DeepSpeed Releases

本文摘要與見解由 AI 自動生成,事實以原文為準。 AI 內容政策