numpy
torch

[full]
pykeops
