torch
numpy
triton

[sfno]
tensorly
tensorly-torch
