scipy
torch  # pytorch pytorch-cuda
torchvision
transformers
datasets