nibabel
torch
pytorch_lightning
scipy
transformers
torchvision
scikit-image