Replace deprecated jax.tree_*
functions with jax.tree.*
#206
The logs for this run have expired and are no longer available.
Loading
jax.tree_*
functions with jax.tree.*
#206