absl-py==0.15.0
arviz==0.12.0
jax==0.3.7
jaxlib==0.3.7+cuda11.cudnn82
matplotlib==3.5.1
numpy==1.22.3
numpyro==0.9.2
pandas==1.4.2
scikit-learn==1.0.2
scipy==1.8.0
seaborn==0.11.2
sklearn