Jean-Baptiste Cordonnier
@jbcordonnier.bsky.social
ML researcher in bio 🧬 at inceptive.life. PhD from EPFL🇨🇭Mountain lover 🏔
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.
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
github.com
November 18, 2024 at 3:05 PM
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.
November 18, 2024 at 2:51 PM
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.