Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | Z A AbstractCitation (class in gpjax.citation) AbstractHeteroscedasticLikelihood (class in gpjax.likelihoods) AbstractIntegrator (class in gpjax.integrators) AbstractKernel (class in gpjax.kernels) AbstractLikelihood (class in gpjax.likelihoods) AbstractMeanFunction (class in gpjax.mean_functions) AbstractNoiseTransform (class in gpjax.likelihoods) AbstractPosterior (class in gpjax.gps) AbstractPrior (class in gpjax.gps) AbstractVariationalFamily (class in gpjax.variational_families) AbstractVariationalGaussian (class in gpjax.variational_families) add_jitter() (in module gpjax.linalg) AnalyticalGaussianIntegrator (class in gpjax.integrators) ArcCosine (class in gpjax.kernels) ARD as_matrix() (gpjax.linalg.BlockDiag method) (gpjax.linalg.Kronecker method) B BasisFunctionComputation (class in gpjax.kernels) Bernoulli (class in gpjax.likelihoods) BlockDiag (class in gpjax.linalg) BookCitation (class in gpjax.citation) C ChainedPosterior (class in gpjax.gps) Cholesky factor cholesky_factor() (in module gpjax.linalg) CitationType (in module gpjax.citation) cite() (in module gpjax.citation) collapsed_elbo() (in module gpjax.objectives) CollapsedVariationalGaussian (class in gpjax.variational_families) CombinationMeanFunction (class in gpjax.mean_functions) components (gpjax.kernels.ICMKernel property) (gpjax.kernels.LCMKernel property) (gpjax.kernels.MultiOutputKernel property) compute_features() (gpjax.kernels.BasisFunctionComputation method) (gpjax.kernels.RFF method) condition_on_observations() (gpjax.models.OILMMModel method) conjugate conjugate_loocv() (in module gpjax.objectives) conjugate_mll() (in module gpjax.objectives) ConjugatePosterior (class in gpjax.gps) Constant (class in gpjax.kernels) (class in gpjax.mean_functions) ConstantDiagonalKernelComputation (class in gpjax.kernels) construct_posterior() (in module gpjax.gps) CoregionalizationMatrix (class in gpjax.parameters) covariance() (gpjax.distributions.GaussianDistribution method) covariance_matrix (gpjax.distributions.GaussianDistribution property) create_oilmm() (in module gpjax.models) create_oilmm_from_data() (in module gpjax.models) create_oilmm_with_kernels() (in module gpjax.models) cross_covariance() (gpjax.kernels.AbstractKernel method) (gpjax.kernels.MultiOutputKernel method) (gpjax.kernels.MultiOutputKernelComputation method) D Dataset (class in gpjax.dataset) DenseKernelComputation (class in gpjax.kernels) diagonal() (gpjax.kernels.AbstractKernel method) (gpjax.kernels.BasisFunctionComputation method) (gpjax.kernels.MultiOutputKernelComputation method) DiagonalKernelComputation (class in gpjax.kernels) E EigenKernelComputation (class in gpjax.kernels) ELBO elbo() (in module gpjax.objectives) entropy() (gpjax.distributions.GaussianDistribution method) ExpectationVariationalGaussian (class in gpjax.variational_families) expected_log_likelihood() (gpjax.likelihoods.AbstractHeteroscedasticLikelihood method) (gpjax.likelihoods.AbstractLikelihood method) (gpjax.likelihoods.HeteroscedasticGaussian method) F fit() (in module gpjax.fit) (in module gpjax.state_space) fit_lbfgs() (in module gpjax.fit) (in module gpjax.state_space) fit_scipy() (in module gpjax.fit) (in module gpjax.state_space) from_icm_components() (gpjax.kernels.LCMKernel class method) FunctionalSample (in module gpjax.typing) G Gaussian (class in gpjax.likelihoods) GaussianDistribution (class in gpjax.distributions) get_batch() (in module gpjax.fit) GHQuadratureIntegrator (class in gpjax.integrators) Gram matrix gram() (gpjax.kernels.AbstractKernel method) (gpjax.kernels.ConstantDiagonalKernelComputation method) (gpjax.kernels.DiagonalKernelComputation method) (gpjax.kernels.MultiOutputKernelComputation method) GraphKernel (class in gpjax.kernels) GraphVariationalGaussian (class in gpjax.variational_families) H H (gpjax.models.OrthogonalMixingMatrix property) H_squared (gpjax.models.OrthogonalMixingMatrix property) heteroscedastic_elbo() (in module gpjax.objectives) heteroscedastic_elbo_chained() (in module gpjax.objectives) heteroscedastic_elbo_conjugate() (in module gpjax.objectives) HeteroscedasticGaussian (class in gpjax.likelihoods) HeteroscedasticPosterior (class in gpjax.gps) HeteroscedasticPrediction (class in gpjax.variational_families) HeteroscedasticVariationalFamily (class in gpjax.variational_families) I ICMKernel (class in gpjax.kernels) in_dim (gpjax.dataset.Dataset property) in_structure() (gpjax.linalg.BlockDiag method) (gpjax.linalg.Kronecker method) inducing points integrate() (gpjax.integrators.AbstractIntegrator method) (gpjax.integrators.AnalyticalGaussianIntegrator method) (gpjax.integrators.GHQuadratureIntegrator method) inv_probit() (in module gpjax.likelihoods) inv_sqrt_S (gpjax.models.OrthogonalMixingMatrix property) is_supervised() (gpjax.dataset.Dataset method) is_unsupervised() (gpjax.dataset.Dataset method) J jitter K KeyArray (in module gpjax.typing) kl_divergence() (gpjax.distributions.GaussianDistribution method) Kronecker (class in gpjax.linalg) L latent_datasets (gpjax.models.OILMMPosterior attribute) latent_kernels (gpjax.kernels.ICMKernel property) (gpjax.kernels.LCMKernel property) (gpjax.kernels.MultiOutputKernel property) latent_noise_variance (gpjax.models.OrthogonalMixingMatrix attribute) latent_posteriors (gpjax.models.OILMMPosterior attribute) latent_priors (gpjax.models.OILMMModel attribute) LatentPosterior (class in gpjax.gps) LCMKernel (class in gpjax.kernels) lengthscale Linear (class in gpjax.kernels) link_function() (gpjax.likelihoods.AbstractLikelihood method) (gpjax.likelihoods.Bernoulli method) (gpjax.likelihoods.Gaussian method) (gpjax.likelihoods.HeteroscedasticGaussian method) (gpjax.likelihoods.Poisson method) log_posterior_density() (in module gpjax.objectives) log_prob() (gpjax.distributions.GaussianDistribution method) logdet() (in module gpjax.linalg) logdet_from_factor() (in module gpjax.linalg) LogNormalTransform (class in gpjax.likelihoods) LowerTriangular (class in gpjax.parameters) M marginal log-likelihood Matern12 (class in gpjax.kernels) Matern32 (class in gpjax.kernels) Matern52 (class in gpjax.kernels) mean (gpjax.distributions.GaussianDistribution property) mean_f (gpjax.variational_families.HeteroscedasticPrediction attribute) mean_g (gpjax.variational_families.HeteroscedasticPrediction attribute) median() (gpjax.distributions.GaussianDistribution method) mixing_matrix (gpjax.models.OILMMModel attribute) (gpjax.models.OILMMPosterior attribute) mode() (gpjax.distributions.GaussianDistribution method) moments() (gpjax.likelihoods.AbstractNoiseTransform method) (gpjax.likelihoods.LogNormalTransform method) (gpjax.likelihoods.SoftplusTransform method) multi_output (gpjax.dataset.Dataset property) MultiOutputGaussian (class in gpjax.likelihoods) MultiOutputKernel (class in gpjax.kernels) MultiOutputKernelComputation (class in gpjax.kernels) mv() (gpjax.linalg.BlockDiag method) (gpjax.linalg.Kronecker method) N n (gpjax.dataset.Dataset property) natural parameters NaturalVariationalGaussian (class in gpjax.variational_families) noise_statistics() (gpjax.likelihoods.AbstractHeteroscedasticLikelihood method) noise_vector() (gpjax.likelihoods.Gaussian method) (gpjax.likelihoods.MultiOutputGaussian method) NoiseMoments (class in gpjax.likelihoods) non_conjugate_mll() (in module gpjax.objectives) NonConjugatePosterior (class in gpjax.gps) NonGaussian (in module gpjax.likelihoods) NonNegativeReal (class in gpjax.parameters) NullCitation (class in gpjax.citation) num_inducing (gpjax.variational_families.AbstractVariationalGaussian property) (gpjax.variational_families.GraphVariationalGaussian property) num_latent_gps (gpjax.kernels.ICMKernel property) (gpjax.kernels.LCMKernel property) (gpjax.kernels.MultiOutputKernel property) (gpjax.models.OILMMModel attribute) (gpjax.models.OILMMPosterior attribute) (gpjax.models.OrthogonalMixingMatrix attribute) num_outputs (gpjax.dataset.Dataset property) (gpjax.kernels.ICMKernel property) (gpjax.kernels.LCMKernel property) (gpjax.kernels.MultiOutputKernel property) (gpjax.models.OILMMModel attribute) (gpjax.models.OrthogonalMixingMatrix attribute) O Objective (in module gpjax.objectives) obs_noise_variance (gpjax.models.OrthogonalMixingMatrix attribute) oilmm_mll() (in module gpjax.models) OILMMModel (class in gpjax.models) OILMMPosterior (class in gpjax.models) OrthogonalAdditiveKernel (class in gpjax.kernels) OrthogonalMixingMatrix (class in gpjax.models) out_structure() (gpjax.linalg.BlockDiag method) (gpjax.linalg.Kronecker method) P PaperCitation (class in gpjax.citation) Periodic (class in gpjax.kernels) PhDThesisCitation (class in gpjax.citation) Poisson (class in gpjax.likelihoods) Polynomial (class in gpjax.kernels) PositiveReal (class in gpjax.parameters) PoweredExponential (class in gpjax.kernels) predict() (gpjax.gps.AbstractPosterior method) (gpjax.gps.AbstractPrior method) (gpjax.gps.ConjugatePosterior method) (gpjax.gps.LatentPosterior method) (gpjax.gps.NonConjugatePosterior method) (gpjax.gps.Prior method) (gpjax.likelihoods.AbstractLikelihood method) (gpjax.likelihoods.Bernoulli method) (gpjax.likelihoods.Gaussian method) (gpjax.likelihoods.HeteroscedasticGaussian method) (gpjax.likelihoods.Poisson method) (gpjax.models.OILMMPosterior method) (gpjax.state_space.StateSpaceConjugatePosterior method) (gpjax.state_space.StateSpacePrior method) (gpjax.variational_families.AbstractVariationalFamily method) (gpjax.variational_families.CollapsedVariationalGaussian method) (gpjax.variational_families.ExpectationVariationalGaussian method) (gpjax.variational_families.HeteroscedasticVariationalFamily method) (gpjax.variational_families.NaturalVariationalGaussian method) (gpjax.variational_families.VariationalGaussian method) (gpjax.variational_families.WhitenedVariationalGaussian method) prepare_targets() (gpjax.likelihoods.Gaussian method) (gpjax.likelihoods.MultiOutputGaussian method) Prior (class in gpjax.gps) prior_kl() (gpjax.variational_families.ExpectationVariationalGaussian method) (gpjax.variational_families.NaturalVariationalGaussian method) (gpjax.variational_families.VariationalGaussian method) (gpjax.variational_families.WhitenedVariationalGaussian method) ProductKernel (class in gpjax.kernels) ProductMeanFunction (class in gpjax.mean_functions) projected_noise_variance (gpjax.models.OrthogonalMixingMatrix property) R RationalQuadratic (class in gpjax.kernels) RBF (class in gpjax.kernels) Real (class in gpjax.parameters) RFF (class in gpjax.kernels) S S (gpjax.models.OrthogonalMixingMatrix attribute) sample() (gpjax.distributions.GaussianDistribution method) sample_approx() (gpjax.gps.ConjugatePosterior method) (gpjax.gps.Prior method) ScalarBool (in module gpjax.typing) ScalarFloat (in module gpjax.typing) ScalarInt (in module gpjax.typing) scaling() (gpjax.kernels.BasisFunctionComputation method) SigmoidBounded (class in gpjax.parameters) slice_input() (gpjax.kernels.AbstractKernel method) SoftplusTransform (class in gpjax.likelihoods) spectral_density (gpjax.kernels.Matern12 property) (gpjax.kernels.Matern32 property) (gpjax.kernels.Matern52 property) (gpjax.kernels.RBF property) sqrt_S (gpjax.models.OrthogonalMixingMatrix property) state_space_mll() (in module gpjax.state_space) StateSpaceConjugatePosterior (class in gpjax.state_space) StateSpacePrior (class in gpjax.state_space) stddev() (gpjax.distributions.GaussianDistribution method) SumKernel (class in gpjax.kernels) summarise() (in module gpjax.summary) SumMeanFunction (class in gpjax.mean_functions) support (gpjax.distributions.GaussianDistribution attribute) supports_tight_bound() (gpjax.likelihoods.AbstractHeteroscedasticLikelihood method) (gpjax.likelihoods.HeteroscedasticGaussian method) T T (gpjax.models.OrthogonalMixingMatrix property) to_sde() (in module gpjax.state_space) transpose() (gpjax.linalg.BlockDiag method) (gpjax.linalg.Kronecker method) TruncatedPeriodic (class in gpjax.state_space) U U (gpjax.models.OrthogonalMixingMatrix property) U_latent (gpjax.models.OrthogonalMixingMatrix attribute) unwrap() (gpjax.parameters.LowerTriangular method) (gpjax.parameters.NonNegativeReal method) (gpjax.parameters.PositiveReal method) (gpjax.parameters.Real method) (gpjax.parameters.SigmoidBounded method) V variance (gpjax.distributions.GaussianDistribution property) variance_f (gpjax.variational_families.HeteroscedasticPrediction attribute) variance_g (gpjax.variational_families.HeteroscedasticPrediction attribute) variational inference variational_expectation() (in module gpjax.objectives) VariationalGaussian (class in gpjax.variational_families) VariationalGaussianInit (class in gpjax.variational_families) vscan() (in module gpjax.scan) W White (class in gpjax.kernels) WhitenedVariationalGaussian (class in gpjax.variational_families) whitening Z Zero (class in gpjax.mean_functions)