contourpy<=1.1.1;python_version=="3.8"

etils<=1.5.2;python_version=="3.9"

jax[cpu]==0.4.13;python_version=="3.8"
jax[cpu]>=0.4.13,<=0.4.30;python_version=="3.9"
jax[cpu]>=0.4.13,<0.5.0;python_version>"3.9"

matplotlib>=3.7.0,<=3.7.2;python_version=="3.8"
matplotlib>=3.7.0;python_version>="3.9"

ml-dtypes<=0.2.0;python_version=="3.8"

numpy>=1.22,<1.25;python_version=="3.8"
numpy>=1.22,<2.0.2;python_version>="3.9"
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;python_version=="3.8"

pillow<=10.4.0;python_version=="3.8"

scipy>=1.9,<1.11;python_version=="3.8"
scipy>=1.9,<=1.13.1;python_version=="3.9"
scipy>=1.9;python_version>"3.9"

seaborn>=0.12.2