在纯AMD平台完成首个大规模MoE模型训练,验证其竞争力。
Training Foundation Models on a Full-Stack AMD Platform: Compute, Networking, and System Design
- 基于AMD MI300X与Pollara网络构建完整训练系统,提供底层性能基准。
- 推出适配AMD硬件的Transformer结构设计规则,优化训练效率与推理延迟。
- 发布760M激活参数的ZAYA1-base模型,性能媲美主流大模型。
我们报告了首个在纯AMD硬件上进行的大规模混合专家(MoE)预训练研究,采用MI300X GPU和Pollara网络。在系统层面,我们提供了完整的集群与网络特性分析:对所有核心集体通信操作(all-reduce、reduce-scatter、all-gather、broadcast)在不同消息大小和GPU数量下的微基准测试,据我们所知,这是该规模下的首次。同时提供MI300X在核尺寸与内存带宽方面的微基准测试数据,用于指导模型设计。在建模方面,提出并应用适配MI300X的Transformer结构尺寸规则,针对注意力与MLP模块,并验证能兼顾训练吞吐与推理延迟的MoE宽度配置。深入描述了训练栈,包括常被忽略的容错机制与检查点重塑功能,以及详细的训练方案。还公布了模型架构与基础模型ZAYA1(760M活跃参数,8.3B总参数,可于https://huggingface.co/Zyphra/ZAYA1-base获取),未来将在后续论文中进一步优化。ZAYA1-base在同等规模下性能可比肩Qwen3-4B与Gemma3-12B,且在推理、数学和编码任务上优于Llama-3-8B与OLMoE。
原文摘要 · Abstract (English)
We report on the first large-scale mixture-of-experts (MoE) pretraining study on pure AMD hardware, utilizing both MI300X GPUs and Pollara networking. We distill practical guidance for both systems and model design. On the systems side, we deliver a comprehensive cluster and networking characterization: microbenchmarks for all core collectives (all-reduce, reduce-scatter, all-gather, broadcast) across message sizes and GPU counts over Pollara. To our knowledge, this is the first at this scale. We further provide MI300X microbenchmarks on kernel sizing and memory bandwidth to inform model design. On the modeling side, we introduce and apply MI300X-aware transformer sizing rules for attention and MLP blocks and justify MoE widths that jointly optimize training throughput and inference latency. We describe our training stack in depth, including often-ignored utilities such as fault-tolerance and checkpoint-reshaping, as well as detailed information on our training recipe. We also provide a preview of our model architecture and base model - ZAYA1 (760M active, 8.3B total parameters MoE, available at https://huggingface.co/Zyphra/ZAYA1-base) - which will be further improved upon in forthcoming papers. ZAYA1-base achieves performance comparable to leading base models such as Qwen3-4B and Gemma3-12B at its scale and larger, and outperforms models including Llama-3-8B and OLMoE across reasoning, mathematics, and coding benchmarks. Together, these results demonstrate that the AMD hardware, network, and software stack are mature and optimized enough for competitive large-scale pretraining.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。