sklearn
numpy
numba
torch
pytorch_lightning
torchvision
imblearn
xgboost==1.5.0
