typing_extensions
dm-tree

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

[pyro]
torch
pyro-ppl

[torch]
torch
