jax
jaxlib
optax
tqdm

[dev]
ruff
pre-commit

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