numpy
matplotlib
torch
scipy
