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:
- The Genred module: our low-level pythonic syntax, compatible with NumPy and PyTorch.
- The generic_reduction functions: a math-friendly set of helper routines for the Genred module.
- kernel_product: a specific syntax with convenient aliases for operations in kernel spaces. It is only compatible with PyTorch.
- Python install
- Generic reductions
- Generic solver
- Math-friendly syntax
- The kernel_product helper
- Block-sparse reductions