equinox>=0.13.4
jax>=0.9.0.1
statedict2pytree>=2.0.1
