pymc==5.*
equinox
diffrax
pytensor
graphviz
jax
numpy

[dev]
icomo[opt]
pytest>=7.3.2
pytest-xdist
pre-commit>=3.3.2
ruff>=0.0.272
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
pytest-beartype
sphinx-autodoc-typehints

[opt]
numpyro
optax
jaxopt
equinox
jaxtyping
optimistix
