DeepSpeed

217338be - Refactor dist tests: Checkpointing (#2202)

Refactor dist tests: Checkpointing (#2202) Refactor distributed tests: checkpointing Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>