[KongchangAI]
Productpytorch/TorchTitan

TorchTitan

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

Timeline (last 90 days)

Oct 4

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

Unverified50%
Oct 4

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

Unverified50%
Oct 4

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

Unverified50%
Sep 11

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

Unverified50%
Sep 11

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

Unverified50%
Sep 11

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

Unverified50%
Sep 11

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

Unverified50%

All Facts (7)

Source Articles