torch
torchvision
pillow
scikit-learn

[dev]
pytest
ruff
