TorchTitan
PyTorch官方推出的大规模模型训练参考实现,专注于利用原生PyTorch特性高效训练大语言模型,支持多种分布式训练策略和低精度训练格式
Timeline (last 90 days)
AMD团队将RCCL团队在GPU发起网络方面的成果集成到TorchTitan中
AMD将GIN应用于TorchTitan的all-to-all操作后,内核层面性能提升最高可达30%
AMD将GIN应用于TorchTitan的all-to-all操作后,端到端整体训练性能提升12%到15%
Primus-Turbo是AMD为TorchTitan等训练框架打造的优化库
TorchTitan是PyTorch官方推出的大规模训练参考实现,专注于用原生PyTorch特性高效训练大模型
AMD已将FP8训练优化正式上游合并到PyTorch官方的pytorch/AO和pytorch/TorchTitan仓库
TorchTitan能够直接支持AMD Instinct GPU并开箱即用地提供FP8训练性能
All Facts (7)
AMD将GIN应用于TorchTitan的all-to-all操作后,端到端整体训练性能提升12%到15%
50%UnverifiedAMD将GIN应用于TorchTitan的all-to-all操作后,内核层面性能提升最高可达30%
50%UnverifiedPrimus-Turbo是AMD为TorchTitan等训练框架打造的优化库
50%UnverifiedTorchTitan是PyTorch官方推出的大规模训练参考实现,专注于用原生PyTorch特性高效训练大模型
50%UnverifiedTorchTitan能够直接支持AMD Instinct GPU并开箱即用地提供FP8训练性能
50%UnverifiedAMD团队将RCCL团队在GPU发起网络方面的成果集成到TorchTitan中
50%UnverifiedAMD已将FP8训练优化正式上游合并到PyTorch官方的pytorch/AO和pytorch/TorchTitan仓库
50%