torch torchvision xgboost==0.90 sklearn boto3