Oossa

Google veröffentlicht TPU Sync für schnellere KV-Cache-Transfers

Googles neue Open-Source-C++-Bibliothek verkürzt die Übertragungsdauer von KV-Caches bei großen MoE-Modellen um mehr als 90 Prozent.

KurzmeldungVon Von Oossa veröffentlicht: 1 Min. Lesezeit

Google hat TPU Sync veröffentlicht, eine Open-Source-C++-Bibliothek, die Key-Value-Caches (KV-Caches) und Modellgewichte zwischen dem High-Bandwidth Memory von TPUs, dem Arbeitsspeicher des Hostsystems, Speichergeräten und dem Netzwerk verschiebt. Laut Google verkürzt die Bibliothek die Übertragungsdauer von KV-Caches bei Mixture-of-Experts-Modellen mit 235 bis 480 Milliarden Parametern um mehr als 90 Prozent.

Warum es wichtig ist

Entwickler können den Zustand großer Sprachmodelle auf TPUs deutlich schneller aktualisieren und so Unterbrechungen bei der Inferenz oder beim Fine-Tuning verkürzen.

War dieser Artikel hilfreich?
Teilen

Als Nächstes lesen

Oossa · Newsletter

Die KI-Woche, erklärt

Jeden Montag: die Nachrichten, die man kennen sollte – verständlich. Kostenlos, kein Spam.