jax[cpu]>=0.4.6
matplotlib>=3.7.0
numpy<1.25,>=1.22
optax>=0.1.5
scipy<1.11,>=1.9
seaborn>=0.12.2
