DeepSpeed

8ad187d8 - Universal ckp fixes (#4588)

Universal ckp fixes (#4588) Signed-off-by: Moshe Island <misland@habana.ai> Co-authored-by: Moshe Island <misland@habana.ai> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>