numpy>=1.22.0
scipy>=1.8.0
ipdb
jax>=0.4.18
