Hi Reddit,

KeOps v1.1 is out! This package provides a new LazyTensor wrapper for PyTorch and Numpy that speeds up all kernel-related computations with an efficient, tailor-made CUDA scheme. It should be especially useful to people who work with kernel matrices, geometric deep learning or point clouds.

https://i.redd.it/9uozdr0tjva31.png

https://i.redd.it/084g9udunva31.png

​

To use it: just install CUDA, nvcc, and type pip install pykeops[full]. This works out-of-the-box on Google Colab !

Our documentation is available at www.kernel-operations.io , with numerous examples and tutorials (spectral coordinates, Sinkhorn, K-NN, K-Means…). If you have any remark or suggestion, please let us know 🙂