DeepSpeed

4217be24 - Merge branch 'master' into olruwase/elastic-ckpt-refresh