PyKeOps
RKeOps
KeOpsLab
KeOps++
KeOps formulas
How does it work?
Sum reduction
GPU Selection
Block-sparse reductions
Arg-K-Min reduction
KernelSolve reduction
KernelSolve reduction (with LazyTensors)
SumSoftMaxWeight reduction
SumSoftMaxWeight reduction (with LazyTensors)