GitHub - google-deepmind/tree: tree is a library for working with nested data structures tree is a library for working with nested data structures - google-deepmind/tree

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.