
[cpu]
jax[cpu]

[doc]
mkdocs
mkdocs-material
mkdocstrings
mkdocstrings-python
matplotlib
tqdm

[format]
isort
black

[full]
jax[cpu]
pytest
pytest-cases
pre-commit
ruff
mypy
mkdocs
mkdocs-material
mkdocstrings
mkdocstrings-python
matplotlib
tqdm
isort
black

[lint]
pytest
pytest-cases
pre-commit
ruff
mypy

[test]
pytest
pytest-cases
