YaFSDP

YaFSDP是一个分布式数据并行框架,专为与transformer类神经网络结构良好协作而设计。它在预训练大型语言模型(Large Language Models, LLMs)时比传统的FSDP快20%,并且在高内存压力条件下表现更佳。YaFSDP旨在减少通信和内存操作的开销。

需求人群:

"YaFSDP框架适用于需要处理大规模数据和模型的机器学习研究人员和工程师。它特别适合于那些需要在高内存压力环境下进行深度学习模型训练的场景,例如大型语言模型的预训练和微调。"

使用场景示例:

使用YaFSDP进行7B至70B参数规模的语言模型预训练。

在64至256个设备上应用YaFSDP进行模型训练,以提高效率。

利用YaFSDP进行2048至8192个token序列的模型训练。

产品特色:

支持大规模语言模型的高效预训练。

优化了内存和通信操作,提高了训练效率。

提供了详细的使用示例,包括因果预训练和监督微调。

基于NVIDIA PyTorch图像构建,集成了必要的补丁库。

支持自定义事件通知,便于开发者根据需要接收更新。

在A100 80G集群上进行了性能评估,确保了框架的高性能。

使用教程:

1. 克隆YaFSDP的GitHub仓库到本地环境。

2. 根据示例文件夹中的指导文档设置Docker环境。

3. 运行docker/build.sh脚本来构建所需的Docker镜像。

4. 根据具体的训练需求,选择合适的示例脚本进行模型训练。

5. 监控训练过程中的内存和通信开销,确保系统稳定运行。

6. 根据需要调整YaFSDP的配置参数,优化模型训练性能。

浏览量:31

打开站点

网站流量情况

最新流量情况

月访问量

4.92m

平均访问时长

00:06:33

每次访问页数

6.11

跳出率

36.20%

流量来源

直接访问

51.61%

自然搜索

33.46%

邮件

0.04%

外链引荐

12.58%

社交媒体

2.19%

展示广告

0

截止目前所有流量趋势图

地理流量分布情况

中国

12.55%

德国

3.84%

印度

9.38%

俄罗斯

4.61%

美国

18.64%

类似产品

透明跟踪和触发,细粒度计算与集合的重叠

大型语言模型在训练和推断中越来越依赖于分布式技术。这些技术需要在设备之间进行通信,随着设备数量的增加,这可能会降低扩展效率。虽然一些分布式技术可以重叠,从而隐藏独立计算的通信,但类似张量并行(TP)的技术固有地将通信与模型执行串行化。隐藏这种串行化通信的一种方法是以细粒度的方式将其与生产者操作(通信数据的产生)交错在一起。然而,在软件中实现这种细粒度的通信和计算交错可能很困难。此外,与任何并发执行一样,它需要在计算和通信之间共享计算和内存资源,导致资源争用,从而降低了重叠效率。为了克服这些挑战,我们提出了T3,它应用硬件-软件共同设计,透明地重叠串行通信,同时最小化与计算的资源争用。T3通过简单配置生产者的输出地址空间,透明地融合了生产者操作和随后的通信,需要进行轻微的软件更改。在硬件层面,T3添加了轻量级的跟踪和触发机制,以编排生产者的计算和通信。它进一步利用增强计算的存储器来进行通信的相关计算。因此,T3减少了资源争用,并有效地将串行通信与计算重叠。对于重要的Transformer模型,如T-NLG,T3将通信密集型子层的速度提高了30%的几何平均值(最大47%),并将数据移动减少了22%的几何平均值(最大36%)。此外,随着模型的扩展,T3的好处仍然存在:对于sim500亿参数模型的子层,几何平均值为29%,PALM和MT-NLG。

© 2025     AIbase    备案号:闽ICP备08105208号-14

隐私政策

用户协议

意见反馈 网站地图