chex>=0.0.4
jax>=0.2.5
jaxlib>=0.1.57
ml-collections==0.1.0
multipledispatch==0.6.0
numpy>=1.18.5
numpyro==0.6.0
packaging==20.4
tfp-nightly==0.12.0.dev20201123

[dev]
black
flake8
isort
pylint

[docs]
furo==2020.12.30b24
matplotlib==3.3.3
nb-black==1.0.7
nbsphinx==0.8.1
sphinx-copybutton==0.3.5

[tests]
pytest
