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) 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 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() (gpjax.gps.ConjugateModel method) (gpjax.gps.HeteroscedasticModel method) (gpjax.gps.JointModel method) (gpjax.gps.NonConjugateModel method) (gpjax.models.OILMMModel method) (gpjax.state_space.StateSpaceConjugateModel method) (gpjax.variational_families.AbstractVariationalFamily method) (gpjax.variational_families.CollapsedVariationalGaussian method) (gpjax.variational_families.HeteroscedasticVariationalFamily method) (gpjax.variational_families.VariationalGaussian method) (gpjax.variational_families.WhitenedVariationalGaussian method) condition_on_observations() (gpjax.models.OILMMModel method) conjugate conjugate_loocv() (in module gpjax.objectives) conjugate_mll() (in module gpjax.objectives) ConjugateModel (class in gpjax.gps) Constant (class in gpjax.kernels) (class in gpjax.mean_functions) ConstantDiagonalKernelComputation (class in gpjax.kernels) construct_model() (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) evidence_correction (gpjax.models.OILMMPosterior attribute) ExactPosterior (class in gpjax.conditioning) 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_natgrads() (in module gpjax.fit) fit_scipy() (in module gpjax.fit) (in module gpjax.state_space) from_icm_components() (gpjax.kernels.LCMKernel class method) full_size (gpjax.dataset.Dataset property) 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) HeteroscedasticModel (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 init_latent() (gpjax.gps.NonConjugateModel method) 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 JointModel (class in gpjax.gps) K KeyArray (in module gpjax.typing) kl_divergence() (gpjax.distributions.GaussianDistribution method) Kronecker (class in gpjax.linalg) L 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.conditioning) 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_marginal_likelihood (gpjax.models.OILMMPosterior property) log_posterior_density (gpjax.conditioning.LatentPosterior property) 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) loo() (gpjax.conditioning.ExactPosterior method) 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 noise_prior (gpjax.gps.HeteroscedasticModel property) 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) NonConjugateModel (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.OILMMPosterior 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) Posterior (class in gpjax.conditioning) PoweredExponential (class in gpjax.kernels) predict() (gpjax.conditioning.Posterior method) (gpjax.gps.JointModel 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.StateSpaceConjugateModel method) (gpjax.state_space.StateSpacePrior method) (gpjax.variational_families.AbstractVariationalFamily method) (gpjax.variational_families.CollapsedVariationalGaussian method) (gpjax.variational_families.HeteroscedasticVariationalFamily method) (gpjax.variational_families.VariationalGaussian method) predict_filter() (gpjax.state_space.StateSpaceConjugateModel method) prepare_targets() (gpjax.likelihoods.Gaussian method) (gpjax.likelihoods.MultiOutputGaussian method) Prior (class in gpjax.gps) prior_kl() (gpjax.variational_families.AbstractVariationalFamily method) (gpjax.variational_families.CollapsedVariationalGaussian method) (gpjax.variational_families.HeteroscedasticVariationalFamily 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.conditioning.ExactPosterior method) (gpjax.gps.ConjugateModel method) (gpjax.gps.Prior method) (gpjax.state_space.StateSpaceConjugateModel 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) stabilised_cholesky() (in module gpjax.linalg) state_space_mll() (in module gpjax.state_space) StateSpaceConjugateModel (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)