tfp-nightly
flax
jax
optax
tqdm

[dev]
pytest
ruff
pre-commit

[docs]
mkdocs
mkdocs-material
mkdocstrings
mkdocstrings-python
mkdocs-include-markdown-plugin
pymdown-extensions
