pykeops (pypi): Python bindings of KeOps: KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
Keywords: kernels|gpu|autodiff