Skip to content

Replace deprecated jax.tree_* functions with jax.tree.* #125

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

Replace deprecated jax.tree_* functions with jax.tree.* #125

Annotations

1 warning

test on 3.11 on ubuntu-latest

succeeded May 15, 2024 in 1m 48s