Fix MoERagged docstring return reference#663

dtype2100 wants to merge 1 commit into

google-deepmind:maingoogle-deepmind/gemma:mainfrom

dtype2100:fix/moe-ragged-docstringdtype2100/gemma:fix/moe-ragged-docstringCopy head branch name to clipboard