mpi4py>=3.1
torch<2.11.1,~=2.3
torchvision~=0.18
scipy~=1.14

[dev]
pre-commit
ruff
mypy
pytest
coverage
perun

[docs]
sphinx
sphinx_rtd_theme
sphinx-autoapi
nbsphinx
sphinx-autobuild
sphinx-copybutton
myst-parser
sphinxcontrib-mermaid

[examples]
scikit-learn
matplotlib~=3.8
jupyter
ipyparallel
pillow
h5py>=3.11

[hdf5]
h5py>=3.11

[netcdf]
netCDF4>=1.7

[pandas]
pandas

[zarr]
zarr
