maresb / pymc-gpx

Apache License 2.0
4 stars 0 forks source link

other packages #2

Open theorashid opened 3 months ago

theorashid commented 3 months ago

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

maresb commented 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!!!