numpy>=1.17.0
jax>=0.4
jaxlib
