numpy
torch
triton
