scipy
matplotlib
torch
