torch>=1.9
torchvision
scipy
numpy>=1.19
xgboost
