torchmetrics
torch
huggingface_hub
datasets
scipy
gdown

[dev]
pytest
