numpy
Pillow
scipy
astropy
torch
torchvision
einops