optimize transform updates#5453

copybara-service[bot] wants to merge 1 commit into

maingoogle/flax:mainfrom

test_902283858google/flax:test_902283858Copy head branch name to clipboard