dmipy-sim<3,>=2.0
dmipy-fit<3,>=2.0

[cpu]
jax[cpu]>=0.6.2
jaxopt>=0.8

[cuda12]
jax[cuda12]>=0.6.2
jaxopt>=0.8

[docs]
mkdocs-material>=9.5
mkdocstrings[python]>=0.24
mkdocs-jupyter>=0.24
