DeepSpeed

a0bd66c6 - support higher sequence length in softmax backward (>2048 & <8192)