KeopsLab logo

Python bindings for KeOps

We now fully document the public interface of the pykeops module, which is a NumPy or PyTorch front-end for the KeOps C++/Cuda library. It contains three sets of instructions:

  1. The Genred module: our low-level pythonic syntax, compatible with NumPy and PyTorch.
  2. The generic_reduction functions: a math-friendly set of helper routines for the Genred module.
  3. kernel_product: a specific syntax with convenient aliases for operations in kernel spaces. It is only compatible with PyTorch.