numpy
numba
torch>=2.0