astropy
scipy
torch
opt_einsum
numpy
tqdm
