jax>=0.3.15
jaxlib
fastprogress
optax
tensorflow_probability
scikit-learn
jaxtyping

[dev]
matplotlib
seaborn
flax
blackjax
graphviz
scikit-learn
sphinx
sphinx-autobuild
myst-nb
jupytext
sphinx-book-theme
codecov
coverage
pytest>=3.9
pytest-cov

[doc]
matplotlib
seaborn
flax
scikit-learn
blackjax
sphinx
sphinx-autobuild
myst-nb
jupytext
sphinx-book-theme

[notebooks]
flax
blackjax
graphviz

[plotting]
matplotlib
seaborn

[test]
codecov
coverage
pytest>=3.9
pytest-cov
