astropy
numpy
torch
