Index

A

AbstractCoreset (class in coreax.coreset) AdditiveKernel (class in coreax.kernels) ANNchorApproximateKernel (class in coreax.approximation)

apply_negative_precision_threshold() (in module coreax.util) ApproximateKernel (class in coreax.approximation) as_data() (in module coreax.data) as_supervised_data() (in module coreax.data)

B

base_kernel (coreax.kernels.UniCompositeKernel attribute) batch_size (coreax.score_matching.SlicedScoreMatching attribute)

build() (coreax.coreset.Coresubset class method) (coreax.coreset.PseudoCoreset class method)

C

CaratheodoryRecombination (class in coreax.solvers) CompositeSolver (class in coreax.solvers) compute() (coreax.kernels.ScalarValuedKernel method) (coreax.metrics.KSD method) (coreax.metrics.Metric method) (coreax.metrics.MMD method) compute_elementwise() (coreax.approximation.ApproximateKernel method) (coreax.kernels.AdditiveKernel method) (coreax.kernels.ExponentialKernel method) (coreax.kernels.LaplacianKernel method) (coreax.kernels.LinearKernel method) (coreax.kernels.MaternKernel method) (coreax.kernels.PCIMQKernel method) (coreax.kernels.PeriodicKernel method) (coreax.kernels.PoissonKernel method) (coreax.kernels.PolynomialKernel method) (coreax.kernels.PowerKernel method) (coreax.kernels.ProductKernel method) (coreax.kernels.RationalQuadraticKernel method) (coreax.kernels.ScalarValuedKernel method) (coreax.kernels.SquaredExponentialKernel method) (coreax.kernels.SteinKernel method) compute_mean() (coreax.kernels.ScalarValuedKernel method) compute_metric() (coreax.coreset.AbstractCoreset method) constant (coreax.kernels.LinearKernel attribute) (coreax.kernels.PolynomialKernel attribute) convert_stein_kernel() (in module coreax.score_matching)

coreax.approximation module coreax.coreset module coreax.data module coreax.kernels module coreax.least_squares 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.AbstractCoreset 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) degree (coreax.kernels.MaternKernel attribute) (coreax.kernels.PolynomialKernel attribute) difference() (in module coreax.util) divergence_x_grad_y() (coreax.kernels.ScalarValuedKernel method) divergence_x_grad_y_elementwise() (coreax.approximation.ApproximateKernel method) (coreax.kernels.AdditiveKernel method) (coreax.kernels.ExponentialKernel method) (coreax.kernels.LaplacianKernel method) (coreax.kernels.LinearKernel method) (coreax.kernels.PCIMQKernel method) (coreax.kernels.PeriodicKernel method) (coreax.kernels.PoissonKernel method) (coreax.kernels.PolynomialKernel method) (coreax.kernels.PowerKernel method) (coreax.kernels.ProductKernel method) (coreax.kernels.RationalQuadraticKernel method) (coreax.kernels.ScalarValuedKernel method) (coreax.kernels.SquaredExponentialKernel method)

downsample_opencv() (in module examples.david_map_reduce_weighted) dtype (coreax.data.Data property) DuoCompositeKernel (class in coreax.kernels)

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.kernels)

F

first_kernel (coreax.kernels.DuoCompositeKernel attribute)

format_time() (in module coreax.util)

G

gamma (coreax.score_matching.SlicedScoreMatching attribute) get_baseline_coreset() (coreax.solvers.KernelThinning method) get_sqrt_kernel() (coreax.kernels.SquaredExponentialKernel method) grad_x() (coreax.kernels.ScalarValuedKernel method) grad_x_elementwise() (coreax.approximation.ApproximateKernel method) (coreax.kernels.AdditiveKernel method) (coreax.kernels.ExponentialKernel method) (coreax.kernels.LaplacianKernel method) (coreax.kernels.LinearKernel method) (coreax.kernels.PCIMQKernel method) (coreax.kernels.PeriodicKernel method) (coreax.kernels.PoissonKernel method) (coreax.kernels.PolynomialKernel method) (coreax.kernels.PowerKernel method) (coreax.kernels.ProductKernel method) (coreax.kernels.RationalQuadraticKernel method) (coreax.kernels.ScalarValuedKernel method) (coreax.kernels.SquaredExponentialKernel method) grad_y() (coreax.kernels.ScalarValuedKernel method)

grad_y_elementwise() (coreax.approximation.ApproximateKernel method) (coreax.kernels.AdditiveKernel method) (coreax.kernels.ExponentialKernel method) (coreax.kernels.LaplacianKernel method) (coreax.kernels.LinearKernel method) (coreax.kernels.PCIMQKernel method) (coreax.kernels.PeriodicKernel method) (coreax.kernels.PoissonKernel method) (coreax.kernels.PolynomialKernel method) (coreax.kernels.PowerKernel method) (coreax.kernels.ProductKernel method) (coreax.kernels.RationalQuadraticKernel method) (coreax.kernels.ScalarValuedKernel method) (coreax.kernels.SquaredExponentialKernel method) gramian_row_mean() (coreax.approximation.ANNchorApproximateKernel method) (coreax.approximation.MonteCarloApproximateKernel method) (coreax.approximation.NystromApproximateKernel method) (coreax.kernels.ScalarValuedKernel method) GreedyKernelPoints (class in coreax.solvers) GreedyKernelPointsState (class in coreax.solvers)

H

HerdingState (class in coreax.solvers)

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

I

index (coreax.kernels.PoissonKernel attribute)

indices (coreax.coreset.Coresubset property) InvalidKernel (class in coreax.util)

J

jit_test() (in module coreax.util)

JITCompilableFunction (class in coreax.util)

K

kernel (coreax.metrics.KSD attribute) (coreax.metrics.MMD attribute) (coreax.score_matching.KernelDensityMatching attribute) (coreax.weights.MMDWeightsOptimiser attribute) (coreax.weights.SBQWeightsOptimiser attribute) KernelDensityMatching (class in coreax.score_matching) KernelHerding (class in coreax.solvers)

KernelThinning (class in coreax.solvers) KeyArray (in module coreax.util) KSD (class in coreax.metrics) kt_choose() (coreax.solvers.KernelThinning method) kt_half() (coreax.solvers.KernelThinning method) kt_half_recursive() (coreax.solvers.KernelThinning method) kt_refine() (coreax.solvers.KernelThinning method)

L

LaplacianKernel (class in coreax.kernels) learning_rate (coreax.score_matching.SlicedScoreMatching attribute) length_scale (coreax.kernels.ExponentialKernel attribute) (coreax.kernels.LaplacianKernel attribute) (coreax.kernels.MaternKernel attribute) (coreax.kernels.PCIMQKernel attribute) (coreax.kernels.PeriodicKernel attribute) (coreax.kernels.RationalQuadraticKernel attribute) (coreax.kernels.SquaredExponentialKernel attribute)

LinearKernel (class in coreax.kernels) LocallyPeriodicKernel (class in coreax.kernels)

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) MaternKernel (class in coreax.kernels) median_heuristic() (in module coreax.kernels) Metric (class in coreax.metrics) MinimalEuclideanNormSolver (class in coreax.least_squares) MMD (class in coreax.metrics) MMDWeightsOptimiser (class in coreax.weights) module coreax.approximation coreax.coreset coreax.data coreax.kernels coreax.least_squares 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

nodes (coreax.coreset.AbstractCoreset property) (coreax.coreset.Coresubset property) (coreax.coreset.PseudoCoreset property) noise_conditioning (coreax.score_matching.SlicedScoreMatching attribute) normalize() (coreax.data.Data method) NotCalculatedError

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

O

optimiser (coreax.score_matching.SlicedScoreMatching attribute) output_scale (coreax.kernels.ExponentialKernel attribute) (coreax.kernels.LaplacianKernel attribute) (coreax.kernels.LinearKernel attribute) (coreax.kernels.MaternKernel attribute) (coreax.kernels.PCIMQKernel attribute) (coreax.kernels.PeriodicKernel attribute) (coreax.kernels.PoissonKernel attribute) (coreax.kernels.PolynomialKernel attribute) (coreax.kernels.RationalQuadraticKernel attribute) (coreax.kernels.SquaredExponentialKernel attribute)

oversampling_parameter (coreax.least_squares.RandomisedEigendecompositionSolver attribute)

P

PaddingInvariantSolver (class in coreax.solvers) pairwise() (in module coreax.util) PCIMQKernel (class in coreax.kernels) periodicity (coreax.kernels.PeriodicKernel attribute) PeriodicKernel (class in coreax.kernels) points (coreax.coreset.AbstractCoreset property) (coreax.coreset.Coresubset property) (coreax.coreset.PseudoCoreset property) PoissonKernel (class in coreax.kernels) PolynomialKernel (class in coreax.kernels)

power (coreax.kernels.PowerKernel attribute) power_iterations (coreax.least_squares.RandomisedEigendecompositionSolver attribute) PowerKernel (class in coreax.kernels) pre_coreset_data (coreax.coreset.AbstractCoreset property) (coreax.coreset.Coresubset property) (coreax.coreset.PseudoCoreset property) precision_threshold (coreax.metrics.KSD attribute) ProductKernel (class in coreax.kernels) progress_bar (coreax.score_matching.SlicedScoreMatching attribute) PseudoCoreset (class in coreax.coreset)

R

random_generator (coreax.score_matching.SlicedScoreMatching attribute) random_key (coreax.approximation.RandomRegressionKernel attribute) (coreax.least_squares.RandomisedEigendecompositionSolver attribute) (coreax.score_matching.SlicedScoreMatching attribute) randomised_eigendecomposition() (coreax.least_squares.RandomisedEigendecompositionSolver method) RandomisedEigendecompositionSolver (class in coreax.least_squares) RandomRegressionKernel (class in coreax.approximation) RandomSample (class in coreax.solvers) RationalQuadraticKernel (class in coreax.kernels) rcond (coreax.least_squares.MinimalEuclideanNormSolver attribute) (coreax.least_squares.RandomisedEigendecompositionSolver attribute) RecombinationSolver (class in coreax.solvers) reduce() (coreax.solvers.CaratheodoryRecombination method) (coreax.solvers.GreedyKernelPoints method) (coreax.solvers.KernelHerding method) (coreax.solvers.KernelThinning method) (coreax.solvers.MapReduce method) (coreax.solvers.RandomSample method) (coreax.solvers.RPCholesky method) (coreax.solvers.Solver method) (coreax.solvers.SteinThinning method) (coreax.solvers.TreeRecombination method)

refine() (coreax.solvers.GreedyKernelPoints method) (coreax.solvers.KernelHerding method) (coreax.solvers.RefinementSolver method) (coreax.solvers.SteinThinning method) RefinementSolver (class in coreax.solvers) RegularisedLeastSquaresSolver (class in coreax.least_squares) relative_weighting (coreax.kernels.RationalQuadraticKernel attribute) RPCholesky (class in coreax.solvers) RPCholeskyState (class in coreax.solvers)

S

sample_batch_indices() (in module coreax.util) SBQWeightsOptimiser (class in coreax.weights) ScalarValuedKernel (class in coreax.kernels) score_function (coreax.kernels.SteinKernel attribute) score_matching (coreax.metrics.KSD attribute) ScoreMatching (class in coreax.score_matching) ScoreNetwork (class in coreax.networks) second_kernel (coreax.kernels.DuoCompositeKernel attribute) shape (coreax.data.Data property) sigma (coreax.score_matching.SlicedScoreMatching attribute) SilentTQDM (class in coreax.util) SlicedScoreMatching (class in coreax.score_matching) solve() (coreax.least_squares.MinimalEuclideanNormSolver method) (coreax.least_squares.RandomisedEigendecompositionSolver method) (coreax.least_squares.RegularisedLeastSquaresSolver method) (coreax.weights.MMDWeightsOptimiser method) (coreax.weights.SBQWeightsOptimiser method) (coreax.weights.WeightsOptimiser method)

solve_qp() (in module coreax.weights) solve_stack() (coreax.least_squares.RegularisedLeastSquaresSolver method) solve_weights() (coreax.coreset.AbstractCoreset method) (coreax.coreset.Coresubset method) (coreax.coreset.PseudoCoreset method) Solver (class in coreax.solvers) speed_comparison_test() (in module coreax.util) squared_distance() (in module coreax.util) SquaredExponentialKernel (class in coreax.kernels) SteinKernel (class in coreax.kernels) 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) TreeRecombination (class in coreax.solvers)

U

UniCompositeKernel (class in coreax.kernels)

unweighted_indices (coreax.coreset.Coresubset property) use_analytic (coreax.score_matching.SlicedScoreMatching attribute)

W

weights (coreax.data.Data attribute) WeightsOptimiser (class in coreax.weights)

without_name() (coreax.util.JITCompilableFunction method) write() (coreax.util.SilentTQDM static method)