torch
numpy
einops
packaging
pystrum>=0.2
nibabel
scipy
matplotlib
tqdm
six
