DeepSpeed

54796bb8 - Update comments and move custom comm ops to internal functions