https://www.tensorflow.org/probability/api_docs/python/tfp/substrates/jax/distributions/GaussianProcessRegressionModel
tfp.substrates.jax.distributions.GaussianProcessRegressionModel | TensorFlow Probability
Posterior predictive distribution in a conjugate GP regression model.
tfpsubstratesjaxdistributionstensorflow