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:
The LazyTensor wrapper: our high-level interface, which provides a pythonic support for the most useful features of KeOps.
The Genred module: our low-level syntax, which interacts directly with the KeOps++ engine.
The generic_reduction functions: a legacy collection of helper routines for the Genred module.
- Python install
- Generic reductions
- Generic solver
- Math-friendly syntax
- Block-sparse reductions