A per-rank memory budget narrows the feasible training layouts

Derive training-state bytes from the actual precision and optimizer recipe. Separate persistent shards from transient gathers and activations, then compare candidate layouts against each rank's peak memory rather than aggregate capacity alone.

17 MIN · PLUS