numpy
matplotlib
qutip
jax[cpu]
diffrax
flax

[dev]
jupyterlab>=3.1.0
mypy
pylint
black

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