tensorflow>=2.6.0
numpy
einops
nibabel
tqdm
