pyro-ppl / numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
https://num.pyro.ai
Apache License 2.0
2.09k stars 227 forks source link

HSGP utility functions in the `contrib` module? #1788

Closed juanitorduz closed 1 month ago

juanitorduz commented 2 months ago

I have been using the Hilbert Space approximation for Gaussian Processes a lot. The main ingredients are the spectral densities of the kernels and the eigenvalues and eigenvectors of the Laplacian in a box, see https://num.pyro.ai/en/stable/examples/hsgp.html

Would you like me to add these functions (and spectral densities from other common kernels) to the contrib module so other users can import them?

fehiepsi commented 2 months ago

That would be awesome, @juanitorduz!

juanitorduz commented 2 months ago

ok! I will start working on it in the upcoming weeks 😄 !