jax
numpyro
optax

[arviz]
arviz>=1.0

[dev]
optax
arviz>=1.0
pytest
pytest-check
pytest-xdist
mypy
black
isort
autoflake
sphinx
sphinx-book-theme
furo
coverage
pytest-cov
pre-commit
myst-parser
