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

[dev]
pytest
mypy
hf_transfer
