jax==0.2.8
jaxlib==0.1.59
tqdm

[dev]
dm-haiku
flax
funsor
ipython
isort
tensorflow_probability

[doc]
nbsphinx
sphinx
sphinx_rtd_theme
sphinx-gallery

[examples]
matplotlib
seaborn
graphviz
arviz

[test]
flake8
pytest>=4.1
pyro-api>=0.1.1
