DeepSpeed
Fix numerical stability in Triton masked softmax
#8782
Merged

Loading