DeepSpeed

dce054db - backwards compatability w. v020 ckpts, fix issue with zero-1 ckpts (#543)