numpy>=1.21
torch>=1.12
torchvision>=0.13
scikit-learn>=1.1
