DeepSpeed

9d9d4cd3 - Merge branch 'master' into olruwase/non_tensor_activation_checkpoint