DeepSpeed

0bd1826c - Merge branch 'master' into olruwase/refactor_universal_checkpoint