DeepSpeed

backwards compatibility w. v020 ckpts, fix issue with zero-1 ckpts

#543

Merged