Smoke render: optimisation and MCMC are shrunk for CI speed, so figures and diagnostics are not publication-fidelity. The production docs are built with full-length inference.

GPJax

heteroscedastic_elbo_chained#

gpjax.objectives.heteroscedastic_elbo_chained(variational_family, data)[source]#

Generic chained bound for heteroscedastic likelihoods.

Parameters:

variational_family (HVF)
data (Dataset)

Return type:

float | Float[jaxlib._jax.Array, ‘’] | Float[ndarray, ‘’]