Migrate away from private jax._src.tree_util._registry#5464

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

maingoogle/flax:mainfrom

test_918133843google/flax:test_918133843Copy head branch name to clipboard