[控场AI]
产品pytorch/TorchTitan

TorchTitan

PyTorch官方推出的大规模模型训练参考实现,专注于利用原生PyTorch特性高效训练大语言模型,支持多种分布式训练策略和低精度训练格式

时间轴 (近 90 天)

10月4日

AMD团队将RCCL团队在GPU发起网络方面的成果集成到TorchTitan中

待验证50%
10月4日

AMD将GIN应用于TorchTitan的all-to-all操作后,内核层面性能提升最高可达30%

待验证50%
10月4日

AMD将GIN应用于TorchTitan的all-to-all操作后,端到端整体训练性能提升12%到15%

待验证50%
9月11日

Primus-Turbo是AMD为TorchTitan等训练框架打造的优化库

待验证50%
9月11日

TorchTitan是PyTorch官方推出的大规模训练参考实现,专注于用原生PyTorch特性高效训练大模型

待验证50%
9月11日

AMD已将FP8训练优化正式上游合并到PyTorch官方的pytorch/AO和pytorch/TorchTitan仓库

待验证50%
9月11日

TorchTitan能够直接支持AMD Instinct GPU并开箱即用地提供FP8训练性能

待验证50%

全部知识事实 (7)

来源文章