h5py
mypy
numpy>=2.1
pandas
pydantic<3,>=2
pytest>=8.4.1
ruff
scikit-image
scikit-learn
scipy
torch>=2.4
torchvision
tqdm
matplotlib>=3.10.6

[cpu]
torch>=2.4
torchvision

[cuda126]
torch>=2.4
torchvision

[cuda132]
torch>=2.4
torchvision

[docs]
sphinx
sphinx-rtd-theme

[movies]
opencv-python

[nn]
torchinfo

[rocm72]
torch>=2.4
torchvision

[rocm72:sys_platform == "linux"]
triton-rocm
