jax>=0.6.2
jaxlib>=0.6.2
numpy
scipy
