Ragged buffer size control#3569

copybara-service[bot] wants to merge 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

test_885683593AI-Hypercomputer/maxtext:test_885683593Copy head branch name to clipboard