Skip to content
VibekollenBETAVibekollen
VideoAI Engineer

Taking Reinforcement Learning Cross Datacenter — Nan Jiang, Modal

Nan Jiang från Modal presenterar en teknik för att köra reinforcement learning (AI-träning genom trial-and-error) på GPU-servrar spridda över flera världsdelar.

Problemet är att checkpoints — sparade versioner av en tränad AI-modell — väger omkring 500 GB och tar timmar att skicka mellan datacenter, vilket förstör möjligheten att uppdatera modellen snabbt. Hans lösning är att skicka endast omkring 500 MB istället: en liten patch som innehåller bara de viktigaste ändringarna. Det fungerar för att mindre än 1% av vikterna faktiskt förändras mellan versioner — inte för att gradienter är glesa, utan för att uppdateringarna är mycket små, ofta tusen gånger mindre än precisionen tillåter. Modal kallar implementationen Stitch.

Öppna på YouTube →

Sammanfattningen är skriven av Vibekollen utifrån källans egen publicering. Innehållet tillhör AI Engineer.

Mer från AI Engineer