DeepSpeed

7752dc5e - Fix layout bug in ZeRO Stage 1 checkpoint logic (#531)

Fix layout bug in ZeRO Stage 1 checkpoint logic (#531) * Fix layout bug in ZeRO Stage 1 checkpoint logic Add elastic checkpoint option for ZeRO stage 1, default to True * Format fixes