pymc-devs / pymc

Bayesian Modeling and Probabilistic Programming in Python
https://docs.pymc.io/
Other
8.53k stars 1.98k forks source link

ENH: Extend `gp.Latent` to allow for multiple outputs #7152

Open bwengals opened 5 months ago

bwengals commented 5 months ago

Context for the issue:

Currently gp.Latent can only handle a single GP observation, so f ~ GP(0, k(x, x')). It should be generalized to allow f_i ~ GP(0, k(x, x')), which is multiple observations of the same GP.

It "should" be pretty straightforward to extend all the GP implementations to handle this case, but it's probably most important for gp.Latent.