6 月 11 日消息，俄羅斯科技巨頭 Yandex 推出了一款開源的大語(yǔ)言模型訓(xùn)練工具 ——YaFSDP，號(hào)稱與現(xiàn)有工具相比速度最高可提升 26%。

據(jù)介紹，YaFSDP 在訓(xùn)練速度方面優(yōu)于傳統(tǒng)的 FSDP 方法，尤其適用于大型模型。在預(yù)訓(xùn)練 LLM 方面，YaFSDP 速度提高了 20%，并且在高內(nèi)存壓力條件下表現(xiàn)更佳。

例如，YaFSDP 對(duì)具有 700 億參數(shù)的 Llama 2 可以實(shí)現(xiàn) 21% 的效率提升，對(duì)具有同級(jí)參數(shù)的 Llama 3 也實(shí)現(xiàn) 了 26% 的效率提升。IT之家附官方數(shù)據(jù)一覽：

Yandex 表示，通過優(yōu)化 GPU 使用率，YaFSDP 可以為開發(fā)者和公司節(jié)省大量資金 —— 每月可能節(jié)省數(shù)十萬(wàn)美元。

Yandex 的資深開發(fā)人員、YaFSDP 團(tuán)隊(duì)成員之一 Mikhail Khruschev 還提到，“目前，我們正在積極嘗試各種模型架構(gòu)和參數(shù)大小，以擴(kuò)展 YaFSDP 的通用性” 。