2026. július 11., szombat · Kutatás
NVIDIA: JAX host offloading akár 57%-kal javítja a nagy nyelvi modellek tanítási átvitelét Blackwell GPU-kon
Az NVIDIA technikai blogja szerint a JAX keretrendszerben alkalmazott host offloading technika jelentősen csökkenti a GPU HBM-szűk keresztmetszetét LLM-tanítás során: a forward pass közben kiválasztott aktivációkat hoszt memóriába mozgatja, majd a backward passnál visszatölti. Az NVIDIA GB200 NVL72 rendszeren végzett MaxText kísérletekben a DeepSeek-V3 671B és Llama 3.1 405B modellekkel az aktiváció-újraszámításhoz képest akár 57%-os átviteli javulást mértek, különösen ritka MoE modellek esetén. A megoldás a Grace Blackwell NVLink-C2C 900 GB/s kétirányú sávszélességére épít, és az optimális eredményhez XLA ütemezési beállítások és Nsight Systems profilozás szükséges.
Egy forrás Ehhez a hírhez egy forráslink áll rendelkezésre.
Miért fontos?
A módszer nagyobb modellek és batch-méretek tanítását teszi lehetővé a meglévő GPU-memórián belül, közvetlen skálázási előnyt nyújtva.
Források
Kapcsolódó témák
Napi összefoglaló
Ez a hír a 2026. július 11., szombat napi AI összefoglaló része.