numpy
pillow
torch
torchvision
astropy
matplotlib
h5py
