absl-py
arviz==0.11.2
immutabledict>=2.0.0
jax>=0.3.14
jaxlib>=0.3.14
matplotlib==3.3.4
numpy>=1.12
numpyro>=0.9.2
pandas>=1.1.5
scipy
seaborn==0.11.1
sklearn
tensorflow>=2.7.2

[:python_version < "3.7"]
dataclasses

[:python_version < "3.8"]
typing_extensions
