jax[cpu]==0.4.13;python_version=="3.8"
jax[cpu]>=0.4.13;python_version>="3.9"
matplotlib>=3.7.0,<=3.7.2;python_version=="3.8"
matplotlib>=3.7.0;python_version>="3.9"
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,<=0.1.5;python_version=="3.8"
optax>=0.1.4;python_version>="3.9"
pandas>=0.25,<=1.5.3
scipy>=1.9,<1.11;python_version=="3.8"
scipy>=1.9;python_version>="3.9"
seaborn>=0.12.2
