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

[dev]
matplotlib
seaborn
flax
blackjax
graphviz
scipy
sphinx
sphinx-autobuild
sphinx_autodoc_typehints
sphinx-math-dollar
myst-nb
jupytext
sphinx-book-theme
codecov
coverage
pytest>=3.9
pytest-cov

[doc]
matplotlib
seaborn
flax
blackjax
graphviz
scipy
sphinx
sphinx-autobuild
sphinx_autodoc_typehints
sphinx-math-dollar
myst-nb
jupytext
sphinx-book-theme

[notebooks]
matplotlib
seaborn
flax
blackjax
graphviz
scipy

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