KeOps 101: Working with LazyTensors
Thanks to the pykeops.numpy.LazyTensor
and pykeops.torch.LazyTensor
decorator,
KeOps can now speed-up your
NumPy or PyTorch programs without any overhead.
A wrapper for NumPy and PyTorch arrays
A wrapper for NumPy and PyTorch arrays
Fancy reductions, solving linear systems
Fancy reductions, solving linear systems
Advanced usage: Vi, Vj, Pm helpers and symbolic variables
Advanced usage: Vi, Vj, Pm helpers and symbolic variables
TensorDot