These short examples showcase the features of the pykeops.numpy and pykeops.torch modules, from automatic differentiation to block-sparse reductions.


If you run a KeOps script for the first time, the internal engine may take a few minutes to compile all the relevant formulas. Do not worry: this work is done once and for all as KeOps stores the resulting shared object files (‘.so’) in a cache directory.