jax>=0.4.31
jaxlib>=0.4.31
jaxopt>=0.8.3
multiprocess>=0.70.15
packaging
numpy>=1.26
pandas>=2.2
scipy>=1.12

[all]
pymedm[notebooks,tests]

[cuda_gpu]
pymedm[jax_gpu,notebooks,tests]

[jax_gpu]
jax[cuda12]<=0.4.31
jaxlib[cuda12]<=0.4.31
cuda-nvcc
cudatoolkit

[notebooks]
ipywidgets
jupyterlab

[tests]
pre-commit
pytest
pytest-cov
pytest-xdist
ruff
setuptools_scm
watermark
