Add tip to tree reduce and reduce_associative about how to exclude leaves from the reduction.#30172

copybara-service[bot] merged 1 commit intojax-ml:mainjax-ml/jax:mainfrom

carlosgmartin:tree_reduce_exclude_leaves_tipcarlosgmartin/jax:tree_reduce_exclude_leaves_tipCopy head branch name to clipboard