Add concat_then_split packing in Grain pipeline#2786

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

aireen/concat_splitAI-Hypercomputer/maxtext:aireen/concat_splitCopy head branch name to clipboard