DeepSpeed

Fix softmax dim of Residual MoE implementation in moe/layer.py

#2110

Merged