jax==0.2.3
jaxlib==0.1.56
tqdm

[dev]
funsor
ipython
isort
flax
dm-haiku
tfp-nightly

[doc]
sphinx
sphinx_rtd_theme
sphinx-gallery

[examples]
matplotlib
seaborn
graphviz
arviz

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