Add jax.tree_util.tree_traverse#19695

jakevdp wants to merge 1 commit intojax-ml:mainjax-ml/jax:mainfrom

jakevdp:tree-traversejakevdp/jax:tree-traverseCopy head branch name to clipboard