torch
torchvision
torchmetrics
pillow
pytest
numpy