Fix: Numeric divergence between the mHC layer and its Pallas kernel - #5179

Fix: Numeric divergence between the mHC layer and its Pallas kernel#5179

denis-mil wants to merge 1 commit into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

denis-mil:dsv4-bug-fixdenis-mil/AI-Hypercomputer-maxtext:dsv4-bug-fixCopy head branch name to clipboard