Fix masked softmax in inference#9

mcognetta wants to merge 2 commits intoCSSLab:mainCSSLab/maia2:mainfrom

mcognetta:fix_masked_softmaxmcognetta/maia2:fix_masked_softmaxCopy head branch name to clipboard