reduce tfds packing intermediate batch_size#2847

aireenmei wants to merge 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

aireen/tfds_packingAI-Hypercomputer/maxtext:aireen/tfds_packingCopy head branch name to clipboard