DSV4: Add indexer_save_selection and indexer_sharded_topk for the CSA indexer - #4915

DSV4: Add indexer_save_selection and indexer_sharded_topk for the CSA indexer#4915

systalyze-ai wants to merge 2 commits into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

systalyze-ai:upstream-pr/indexer-selection-remat-topksystalyze-ai/maxtext:upstream-pr/indexer-selection-remat-topkCopy head branch name to clipboard