numpy
fitsio
torch
