Unify input types kernels of different backends

[arnaudvl]

Ensure that PyTorch, TensorFlow and KeOps kernels all take native tensors from the relevant backend as inputs. Currently PyTorch allows to pass NumPy arrays in the forward pass but doesn't handle proper device placement afterwards.