jax>=0.7.1
numpy>=2.3.2
