tqdm
torch
torchvision
nibabel
einops
torchio
timm==0.6.12
numpy
segmentation-models-pytorch==0.3.2
huggingface_hub
nnunetv2
