ziatdinovmax / gpax

Gaussian Processes for Experimental Sciences
http://gpax.rtfd.io
MIT License
205 stars 27 forks source link

Cholesky decomposition for ExactGP class #115

Open mjbajwa opened 1 month ago

mjbajwa commented 1 month ago

Hi,

Really useful library! I'm using GPs for some large scale problems and wanted to be able to use your code. For exact GPs, it is well known that Cholesky decomposition is a better alternative than the base matrix inversion. I think the NumPyro example, which the ExactGP class borrows from, also reflects this now. Just added this slight change.

Let me know if I missed something. Testing the new test_gp.py runs successfully on my end.

I'm not sure if other parts of the codebase would also benefit from this.

matthewcarbone commented 1 month ago

I have no problems with this. @ziatdinovmax?