jax>=0.2.27
jaxlib>=0.1.76
equinox>=0.1.6
