Megatron-DeepSpeed

Fix mixed fused layer norm to mimick nn.LayerNorm for torch>1.11

#281

Merged