jax[cpu]>=0.4.6
matplotlib>=3.7.0
numpy>=1.22,<1.25;python_version=="3.8"
numpy>=1.22;python_version>="3.9"
chex<=0.1.7;python_version=="3.8"
optax>=0.1.4
scipy>=1.9,<1.11;python_version=="3.8"
scipy>=1.9;python_version>="3.9"
seaborn>=0.12.2
