跳至主要内容

DeepSpeed

DeepSpeed​

DeepSpeed 是 Microsoft 開發的深度學習優化函式庫,透過 ZeRO-3 等技術把模型參數、梯度與優化器狀態分散到多 GPU/節點,大幅降低單卡顯存壓力,是 LLM 大規模訓練的常用框架。

TODO Stub 待擴充。被以下筆記引用:

  • Ray 進行 LLM 微調

待補​

  • ZeRO-1/2/3 差別與 trade-off
  • 與 PyTorch FSDP / Ray Train 的搭配
  • 實戰啟動指令範例