numpy>=1.22

[jax]
jax>=0.4.30
