DeepSpeed

44e5b269 - Merge branch 'master' into olruwase/refactor_universal_checkpoint