typing_extensions
dm-tree

[dev]
torch
pyro-ppl
setuptools
sphinx
sphinxcontrib-bibtex
sphinx_rtd_theme
myst-parser
nbsphinx
pytest
pytest-cov
pytest-xdist
pytest-benchmark
mypy
black
flake8
isort
nbval
nbqa

[docs]
setuptools
sphinx
sphinxcontrib-bibtex
sphinx_rtd_theme
myst-parser
nbsphinx

[pyro]
torch
pyro-ppl

[torch]
torch
