Add alias for for functions in `optax.tree_utils` like `jax.tree`#1166

Open

Ata-Shaker wants to merge 3 commits intogoogle-deepmind:mainfrom Ata-Shaker:dev

+42

Commits on Jan 2, 2025

Created alias file for functions in optax.tree_utils and dropped "tree" from them
Ata-Shaker
committed

Commits on Jan 3, 2025

Move 'optax.tree' to '_src'
Ata-Shaker
committed

Commits on Jan 9, 2025

Reformat the aliases as seperate imports
Ata-Shaker
committed