default partition metadata in vmap/scan/pmap#5508

mohsinm-dev wants to merge 1 commit into

google:maingoogle/flax:mainfrom

mohsinm-dev:fix-default-partition-metadatamohsinm-dev/flax:fix-default-partition-metadataCopy head branch name to clipboard