mpi4py>=3.0.0
torch<2.10,~=2.0
torchvision~=0.15
scipy~=1.14

[dev]
pre-commit
ruff
mypy
pytest
coverage
perun

[docs]
sphinx
sphinx_rtd_theme
sphinx-autoapi
nbsphinx
sphinx-autobuild
sphinx-copybutton

[examples]
scikit-learn~=0.24
matplotlib~=3.1
jupyter
ipyparallel
pillow

[hdf5]
h5py>=2.8.0

[netcdf]
netCDF4>=1.5.6

[zarr]
zarr
