AI Grimoire
02 — siblings
objectives/optimisers/scaling/memory/
distributed/

Training / distributed

Distributed

Once the model, its gradients and its optimiser state exceed a single accelerator, the question becomes what to shard and when to gather it again. The answers differ in how much communication they add to buy how much memory back.

1 entry.

Entries

02.05.1
ZeRO Shardingstandard
Partition optimiser state, gradients and parameters across ranks.
O(d/N)