I’m using the dm-tree lib github.com/google-deepm... I wonder if it’s the same as jax.tree_util but packaged independently. Will have a look.
1 year ago
1
0
0
0
I’m using the dm-tree lib github.com/google-deepm... I wonder if it’s the same as jax.tree_util but packaged independently. Will have a look.
Is it possible to install equinox without the dependency on JAX? I’m particularly interested in the pytree utils you were using to code lenses.