jax>=0.4.19
numpy
equinox>=0.11.2
