Index

_

_analytic_objective() (coreax.score_matching.SlicedScoreMatching static method) _compact_name_scope_methods (coreax.networks.ScoreNetwork attribute) _general_objective() (coreax.score_matching.SlicedScoreMatching static method) _loss() (coreax.score_matching.SlicedScoreMatching method) _loss_element() (coreax.score_matching.SlicedScoreMatching method) _noise_conditional_loop_body() (coreax.score_matching.SlicedScoreMatching method)

_noise_conditional_train_step() (coreax.score_matching.SlicedScoreMatching method) _objective_function() (coreax.score_matching.SlicedScoreMatching method) _prepare_kernel_system() (in module coreax.weights) _random_indices() (in module coreax.approximation) _random_least_squares() (in module coreax.approximation) _train_step() (coreax.score_matching.SlicedScoreMatching method)

A

AdditiveKernel (class in coreax.kernel) ANNchorApproximateKernel (class in coreax.approximation) apply_negative_precision_threshold() (in module coreax.util) approximate() (coreax.inverses.LeastSquareApproximator method) (coreax.inverses.RandomisedEigendecompositionApproximator method) (coreax.inverses.RegularisedInverseApproximator method)

approximate_stack() (coreax.inverses.RegularisedInverseApproximator method) ApproximateKernel (class in coreax.approximation) as_data() (in module coreax.data)

B

batch_size (coreax.score_matching.SlicedScoreMatching attribute)

C

CompositeKernel (class in coreax.kernel) CompositeSolver (class in coreax.solvers) compute() (coreax.kernel.Kernel method) (coreax.metrics.Metric method) (coreax.metrics.MMD method) compute_elementwise() (coreax.approximation.ApproximateKernel method) (coreax.kernel.AdditiveKernel method) (coreax.kernel.ExponentialKernel method) (coreax.kernel.Kernel method) (coreax.kernel.LaplacianKernel method) (coreax.kernel.LinearKernel method) (coreax.kernel.PCIMQKernel method) (coreax.kernel.PeriodicKernel method) (coreax.kernel.PolynomialKernel method) (coreax.kernel.ProductKernel method) (coreax.kernel.RationalQuadraticKernel method) (coreax.kernel.SquaredExponentialKernel method) (coreax.kernel.SteinKernel method) compute_mean() (coreax.kernel.Kernel method) compute_metric() (coreax.coreset.Coreset method) convert_stein_kernel() (in module coreax.score_matching) coreax.approximation module coreax.coreset module

coreax.data module coreax.inverses module coreax.kernel module coreax.metrics module coreax.networks module coreax.score_matching module coreax.solvers module coreax.util module coreax.weights module Coreset (class in coreax.coreset) coreset (coreax.coreset.Coreset property) (coreax.coreset.Coresubset property) Coresubset (class in coreax.coreset) CoresubsetSolver (class in coreax.solvers) create_train_state() (in module coreax.networks)

D

Data (class in coreax.data) data (coreax.data.Data attribute) difference() (in module coreax.util) divergence_x_grad_y() (coreax.kernel.Kernel method) divergence_x_grad_y_elementwise() (coreax.approximation.ApproximateKernel method) (coreax.kernel.AdditiveKernel method) (coreax.kernel.ExponentialKernel method) (coreax.kernel.Kernel method) (coreax.kernel.LaplacianKernel method) (coreax.kernel.LinearKernel method) (coreax.kernel.PCIMQKernel method) (coreax.kernel.PeriodicKernel method) (coreax.kernel.PolynomialKernel method) (coreax.kernel.ProductKernel method) (coreax.kernel.RationalQuadraticKernel method) (coreax.kernel.SquaredExponentialKernel method)

E

examples.david_map_reduce_weighted module examples.herding_stein_weighted module examples.pounce module

examples.pounce_map_reduce module ExplicitSizeSolver (class in coreax.solvers) ExponentialKernel (class in coreax.kernel)

G

gamma (coreax.score_matching.SlicedScoreMatching attribute) grad_x() (coreax.kernel.Kernel method) grad_x_elementwise() (coreax.approximation.ApproximateKernel method) (coreax.kernel.AdditiveKernel method) (coreax.kernel.ExponentialKernel method) (coreax.kernel.Kernel method) (coreax.kernel.LaplacianKernel method) (coreax.kernel.LinearKernel method) (coreax.kernel.PCIMQKernel method) (coreax.kernel.PeriodicKernel method) (coreax.kernel.PolynomialKernel method) (coreax.kernel.ProductKernel method) (coreax.kernel.RationalQuadraticKernel method) (coreax.kernel.SquaredExponentialKernel method) grad_y() (coreax.kernel.Kernel method)

grad_y_elementwise() (coreax.approximation.ApproximateKernel method) (coreax.kernel.AdditiveKernel method) (coreax.kernel.ExponentialKernel method) (coreax.kernel.Kernel method) (coreax.kernel.LaplacianKernel method) (coreax.kernel.LinearKernel method) (coreax.kernel.PCIMQKernel method) (coreax.kernel.PeriodicKernel method) (coreax.kernel.PolynomialKernel method) (coreax.kernel.ProductKernel method) (coreax.kernel.RationalQuadraticKernel method) (coreax.kernel.SquaredExponentialKernel method) gramian_row_mean() (coreax.approximation.ANNchorApproximateKernel method) (coreax.approximation.MonteCarloApproximateKernel method) (coreax.approximation.NystromApproximateKernel method) (coreax.kernel.Kernel method)

H

HerdingState (class in coreax.solvers)

hidden_dims (coreax.networks.ScoreNetwork attribute) (coreax.score_matching.SlicedScoreMatching attribute)

I

InvalidKernel (class in coreax.util)

is_data() (in module coreax.data)

J

jit_test() (in module coreax.util)

K

Kernel (class in coreax.kernel) kernel (coreax.score_matching.KernelDensityMatching attribute) (coreax.weights.WeightsOptimiser attribute)

KernelDensityMatching (class in coreax.score_matching) KernelHerding (class in coreax.solvers) KeyArray (in module coreax.util)

L

LaplacianKernel (class in coreax.kernel) learning_rate (coreax.score_matching.SlicedScoreMatching attribute)

LeastSquareApproximator (class in coreax.inverses) LinearKernel (class in coreax.kernel) LocallyPeriodicKernel (class in coreax.kernel)

M

main() (in module examples.david_map_reduce_weighted) (in module examples.herding_stein_weighted) (in module examples.pounce) (in module examples.pounce_map_reduce) MapReduce (class in coreax.solvers) match() (coreax.score_matching.KernelDensityMatching method) (coreax.score_matching.ScoreMatching method) (coreax.score_matching.SlicedScoreMatching method) median_heuristic() (in module coreax.kernel) Metric (class in coreax.metrics) MMD (class in coreax.metrics) (class in coreax.weights) MMDWeightsOptimiser (class in coreax.weights) module coreax.approximation coreax.coreset coreax.data coreax.inverses coreax.kernel coreax.metrics coreax.networks coreax.score_matching coreax.solvers coreax.util coreax.weights examples.david_map_reduce_weighted examples.herding_stein_weighted examples.pounce examples.pounce_map_reduce

MonteCarloApproximateKernel (class in coreax.approximation)

N

noise_conditioning (coreax.score_matching.SlicedScoreMatching attribute) normalize() (coreax.data.Data method) NotCalculatedError

num_epochs (coreax.score_matching.SlicedScoreMatching attribute) num_noise_models (coreax.score_matching.SlicedScoreMatching attribute) num_random_vectors (coreax.score_matching.SlicedScoreMatching attribute) NystromApproximateKernel (class in coreax.approximation)

O

optimiser (coreax.score_matching.SlicedScoreMatching attribute)

P

PaddingInvariantSolver (class in coreax.solvers) PairedKernel (class in coreax.kernel) pairwise() (in module coreax.util) pairwise_difference() (in module coreax.util)

PCIMQKernel (class in coreax.kernel) PeriodicKernel (class in coreax.kernel) PolynomialKernel (class in coreax.kernel) ProductKernel (class in coreax.kernel)

R

random_generator (coreax.score_matching.SlicedScoreMatching attribute) random_key (coreax.score_matching.SlicedScoreMatching attribute) randomised_eigendecomposition() (in module coreax.inverses) RandomisedEigendecompositionApproximator (class in coreax.inverses) RandomRegressionKernel (class in coreax.approximation) RandomSample (class in coreax.solvers) RationalQuadraticKernel (class in coreax.kernel) reduce() (coreax.solvers.KernelHerding method) (coreax.solvers.MapReduce method) (coreax.solvers.RandomSample method) (coreax.solvers.RPCholesky method) (coreax.solvers.Solver method) (coreax.solvers.SteinThinning method)

refine() (coreax.solvers.KernelHerding method) (coreax.solvers.RefinementSolver method) (coreax.solvers.SteinThinning method) RefinementSolver (class in coreax.solvers) RegularisedInverseApproximator (class in coreax.inverses) RPCholesky (class in coreax.solvers) RPCholeskyState (class in coreax.solvers)

S

sample_batch_indices() (in module coreax.util) SBQ (class in coreax.weights) SBQWeightsOptimiser (class in coreax.weights) ScoreMatching (class in coreax.score_matching) ScoreNetwork (class in coreax.networks) sigma (coreax.score_matching.SlicedScoreMatching attribute) SilentTQDM (class in coreax.util) SlicedScoreMatching (class in coreax.score_matching) solve() (coreax.weights.MMDWeightsOptimiser method) (coreax.weights.SBQWeightsOptimiser method) (coreax.weights.WeightsOptimiser method)

solve_qp() (in module coreax.util) solve_weights() (coreax.coreset.Coreset method) Solver (class in coreax.solvers) squared_distance() (in module coreax.util) squared_distance_pairwise() (in module coreax.util) SquaredExponentialKernel (class in coreax.kernel) SteinKernel (class in coreax.kernel) SteinThinning (class in coreax.solvers) SupervisedData (class in coreax.data) supervision (coreax.data.SupervisedData attribute)

T

tree_leaves_repeat() (in module coreax.util)

tree_type (coreax.solvers.MapReduce attribute) tree_zero_pad_leading_axis() (in module coreax.util)

U

unweighted_indices (coreax.coreset.Coresubset property)

use_analytic (coreax.score_matching.SlicedScoreMatching attribute)

W

weights (coreax.data.Data attribute)

WeightsOptimiser (class in coreax.weights) write() (coreax.util.SilentTQDM method)