numpy
scipy
torch
numba
