torch
numpy
numba
