numpy
matplotlib
qutip
jax[cpu]
diffrax
flax
tqdm
pytest

[dev]
jupyterlab>=3.1.0
mypy
pylint
black
coverage
pytest

[docs]
mkdocs
mkdocs-material
mkdocs-literate-nav
mkdocs-section-index
mkdocs-gen-files
mkdocstrings-python
mkdocs-jupyter
