Oossa

Google brengt TPU Sync uit voor snellere verplaatsing van KV-caches

Google zegt dat de nieuwe open-source C++-bibliotheek de overdracht van KV-caches bij grote MoE-modellen met meer dan 90% versnelt.

Kort berichtDoor Gepubliceerd door Oossa: 1 min lezen

Google heeft TPU Sync uitgebracht, een open-source C++-bibliotheek die key-value (KV)-caches en modelgewichten verplaatst tussen het high-bandwidth memory van TPU’s, het RAM-geheugen van hostsystemen, opslag en het netwerk. Volgens het bedrijf verlaagt de bibliotheek de latentie bij KV-overdracht met meer dan 90% voor mixture-of-experts-modellen met 235 miljard tot 480 miljard parameters.

Waarom het ertoe doet

Ontwikkelaars kunnen de status van grote taalmodellen op TPU’s veel sneller verversen, waardoor de onderbreking tijdens inferentie of finetuning korter wordt.

Was dit artikel nuttig?
Delen

Lees ook

Oossa · Nieuwsbrief

De AI-week, uitgelegd

Elke maandag: het nieuws dat ertoe doet, in gewone taal. Gratis, geen spam.