numpy
matplotlib
pymc==5.*
pytensor
arviz
numpyro
ipywidgets
graphviz
diffrax
optax
jaxopt

[dev]
pytest>=7.3.2
black>=23.3.0
isort>=5.12.0
pre-commit>=3.3.2
ruff>=0.0.272
mypy>=1.3.0
pydocstyle>=6.3.0
ipython<8
ipykernel>=6.0.0
Sphinx>=6.2.1
sphinx-book-theme
importlib-metadata>=6.5.1
myst-parser>=0.19.2
nbsphinx>=0.9.2
pandoc>=2.3
