flax
Replace deprecated `jax.tree_` functions with `jax.tree.`
#3926

Merged

Commits

Replace deprecated `jax.tree_*` functions with `jax.tree.*`

a-googler committed 1 year ago

Loading