Cerebrasは公式ブログで、兆パラメータ級モデルのウェハースケール展開計画を公開し、オープンモデルKimi K3(約3兆パラメータ、93層)を試算例とした。4ビット量子化では1層あたり約16GBで、69層のリニアアテンション層は1枚のウェハーに2層ずつ、24層のフルアテンション層はKVキャッシュ用の余地を残して1枚ずつ占有し、35+24で計59枚になる。各WSE-3Tウェハーは44GBのオンチップメモリを持ち、層全体をチップ上に直接置ける。実行時は層がパイプラインを組み、ウェハー間では中間結果だけを1ホップ約2マイクロ秒で伝送する。Cerebrasによれば、このクラスのモデルは毎秒1000〜2000超のトークンに達するという。ただし重要な注意点がある。K3はあくまで例示であり、これは展開レシピであって、59枚のウェハーでの実測結果ではない。モデルが大きいほど、展開のエンジニアリングが重要になる。
原文リンク:原文を見るにはここをクリック





コメント