Oossa

Google、KVキャッシュ転送を高速化するTPU Syncを公開

Googleの新しいオープンソースC++ライブラリは、大規模なMoEモデルでKVキャッシュの転送遅延を90%超削減する。

短信著者: Oossa公開日: 1 分で読める

Googleは、TPUの高帯域幅メモリー、ホストRAM、ストレージ、ネットワーク間でキー・バリュー(KV)キャッシュやモデルの重みを移動させるオープンソースのC++ライブラリ「TPU Sync」を公開した。同社によると、2350億〜4800億パラメーターのMixture of Experts(MoE)モデルで、KVキャッシュの転送遅延を90%超削減できるという。

なぜ重要か

開発者はTPU上で大規模言語モデルの状態をより速く更新でき、推論やファインチューニング時の停止時間を短縮できる。

この記事は役に立ちましたか?
共有

次に読む

Oossa · ニュースレター

今週のAIを、わかりやすく

毎週月曜日、知っておきたいニュースをやさしい言葉で。無料・スパムなし。