seaborn>=0.12.2

[:python_version == "3.8"]
contourpy<=1.1.1
jax[cpu]==0.4.13
matplotlib<=3.7.2,>=3.7.0
ml-dtypes<=0.2.0
numpy<1.25,>=1.22
chex<=0.1.7
optax<=0.1.5,>=0.1.4
pandas<=1.5.3,>=0.25
pillow<=10.4.0
scipy<1.11,>=1.9

[:python_version == "3.9"]
etils<=1.5.2
jax[cpu]<=0.4.30,>=0.4.13
scipy<=1.13.1,>=1.9

[:python_version > "3.9"]
jax[cpu]<0.5.0,>=0.4.13
numpy>=1.22
scipy>=1.9

[:python_version >= "3.9"]
matplotlib>=3.7.0
numpy<2.0.2,>=1.22
optax>=0.1.4
