
[docs]
effectful[jax,numpyro,pyro,torch]
sphinx
sphinxcontrib-bibtex
sphinx_rtd_theme
myst-parser
nbsphinx
sphinx_autodoc_typehints
pypandoc_binary

[jax]
jax
dm-tree

[numpyro]
numpyro>=0.19
dm-tree

[pyro]
pyro-ppl>=1.9.1
dm-tree

[test]
effectful[docs,jax,numpyro,pyro,torch]
pytest
pytest-cov
pytest-xdist
pytest-benchmark
mypy
ruff
nbval
nbqa

[torch]
torch
dm-tree
