torch
torchvision
matplotlib
pillow
torchmetrics
numpy
huggingface_hub[cli]
transformers
accelerate
termcolor

[dev]
pytest
mypy
hf_transfer
