Describe universal checkpoint shards as affine maps - #8385

Merged

delock merged 12 commits into

deepspeedai:masterfrom

Achyuthan-S:affine-ir-shard-map

Sep 10, 2026

Commits on Sep 9, 2026

Add affine shard map for universal checkpoint
Achyuthan-S
committed
b6b0d50
Add scale and homogeneity to the affine shard map
Achyuthan-S
committed
13e2c64
Lower the current checkpoint metadata to affine maps
Achyuthan-S
committed
1d1d603
Convert tp slices through an affine map when one is present
Achyuthan-S
committed
3cf1234
Build shard metadata per model in the affine map tests
Achyuthan-S
committed
166a3d2
Refuse an affine map format newer than the reader understands
Achyuthan-S
committed
9149908
Keep the stride helper private
Achyuthan-S
committed
d63251b
Address review: scale optimizer states by their own power
Achyuthan-S
committed
2e846d9
Address review: header, scale in the piece tuple, scaling example
Achyuthan-S
committed
fcec062
Refuse to convert a scaled optimizer state
Achyuthan-S
committed
10bde3e

Commits on Sep 10, 2026

Carry the scale on the replicated map, not the split
Achyuthan-S
committed
d02c56b
Fix universal checkpoint resume across AutoTP sizes

0z5a
authored and
Achyuthan-S
committed
dd5c57f