numba
torch
