Zyphra 训练手册:如何构建 SSM、Transformer 与 MoE 混合模型
Zyphra 发布《Zyphra Training Cookbook》,分享构建混合(SSM、Transformer、MoE)模型的工程经验,供其他技术团队参考。Zyphra 已构建 BlackMamba、Zamba-7B、Zamba2-2.7B、Zamba2-1.2B 等混合模型,以及 Zyda 数据集和 Tree Attention 工程优化。手册聚焦 1B-15B 端侧小模型,建议每 1/4 到 1/6 的序列混合层使用全注意力,其余采用 Mamba 等线性混合器以降低 FLOP 和内存开销。

into building good ones. The purpose of this cookbook is to enable other technical groups to hit the ground running when building their own hybrid (SSM, Transformer, MoE) models. GitHub For context, we at Zyphra have built the following hybrid models: BlackMamba; Zamba-7B; Zamba2-2.7B; Zamba2-1.2B The following datasets: Zyda And the following engineering optimizations Tree Attention Dense transformer models (i.e. alternating multi-head attention (MHA) and multilayer perceptron (MLP) blocks) ha
匹配字段正文




