nibabel
torch
pytorch_lightning
scipy
transformers
torchvision