jax>=0.3.4
jaxlib>=0.3.0
equinox>=0.1.6
