numpy
matplotlib
torch
numba
