geometric-kernels / GeometricKernels

Geometric kernels on manifolds, meshes and graphs

https://geometric-kernels.github.io/

Apache License 2.0

214 stars 18 forks source link

Integrate with GPJax #74

Closed thomaspinder closed 1 year ago

thomaspinder commented 1 year ago

Feature request

Geometric kernels can be easily supported by GPJax. There is an open PR to GPJax that demonstrates this:

https://github.com/JaxGaussianProcesses/GPJax/pull/159

Motivation

Geometric kernels currently have a TensorFlow and GPyTorch backend. Supporting GPJax would add JAX to this list.

Proposal

Describe the solution you would like I'd suggest lifting the GeometricComputation and GeometricKernel out of GPJax into Geometric kernels.

Are you willing to open a pull request? Yes

stoprightthere commented 1 year ago

@thomaspinder This would be awesome! Feel free to open a PR