KeopsLab logo

Python bindings for KeOps

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

  1. The LazyTensor wrapper: our high-level interface, which provides a pythonic support for the most useful features of KeOps.

  2. The Genred module: our low-level syntax, which interacts directly with the KeOps++ engine.

  3. The generic_reduction functions: a legacy collection of helper routines for the Genred module.