Port softmax ops to libtorch stable ABI#2830

pstjohn wants to merge 1 commit intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom

pstjohn:pstjohn/stable-abi-v2pstjohn/TransformerEngine:pstjohn/stable-abi-v2Copy head branch name to clipboard