jax>=0.4.3
equinox>=0.10.4
