Add weight for ragged gather kernel and enable fan out in bwd ragged sort#4166

copybara-service[bot] merged 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

chengnuojin-fix-raggedAI-Hypercomputer/maxtext:chengnuojin-fix-raggedCopy head branch name to clipboard