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

[jax]
jax

[llm]
litellm
pillow
pydantic

[numpyro]
numpyro>=0.19

[prettyprinter]
prettyprinter

[pyro]
pyro-ppl>=1.9.1

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

[torch]
torch
