torch>=2.3.0
torchvision>=0.18.0
numpy==1.26.4
Pillow
matplotlib
tqdm

[dev]
pytest
build
