jax>=0.4.30
numpy>=1.26.0
