Open theorashid opened 3 months ago
Hey @theorashid, great to see you here!
I wasn't familiar with the geometric kernels package you referenced, but it seems really interesting! I had a pure PyMC implementation in mind since everything listed in the roadmap here is pretty simple to do by hand. But we certainly don't need to stick to just that!
This is meant to be a temporary-ish functional fork of pymc.gp
that we can use to mess around and hopefully contribute back into PyMC. The naming conflict you point out is an unfortunate coincidence. I don't think it's just you, since import gpjax as gpx
is featured in their docs. (I haven't used it so I wasn't aware. :see_no_evil:)
If you have any ideas, please feel free to incubate them here!!!
Is there any way to connect the jax-based geometric kernels package to pymc for this? It already connects to torch, tfp, gpjax
Secondly, I'm used to
import gpjax as gpx
so the gpx is personally confusing for me, but perhaps not anyone else