DSV4: Add shard_indexer_acts to shard the CSA indexer scores across heads - #4909

DSV4: Add shard_indexer_acts to shard the CSA indexer scores across heads#4909

systalyze-ai wants to merge 1 commit into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

systalyze-ai:upstream-pr/indexer-activation-shardingsystalyze-ai/maxtext:upstream-pr/indexer-activation-shardingCopy head branch name to clipboard