jax==0.2.26
matplotlib==3.5.0
numpy==1.21.4
pandas==1.3.4
scipy==1.7.3
seaborn==0.11.2
