Module: tfp.experimental.substrates.jax.math.psd_kernels

Positive-semidefinite kernels package.


exp_sin_squared module: ExpSinSquared kernel.

exponentiated_quadratic module: The ExponentiatedQuadratic kernel.

feature_scaled module: Feature scaled kernel.

feature_transformed module: FeatureTransformed kernel.

kumaraswamy_transformed module: Kumaraswamy CDF transformed kernel.

matern module: Matern kernels.

polynomial module: Polynomial and Linear kernel.

positive_semidefinite_kernel module: PositiveSemidefiniteKernel base.

rational_quadratic module: Rational Quadratic kernel.

schur_complement module: The SchurComplement kernel.


class ExpSinSquared: Exponentiated Sine Squared Kernel.

class ExponentiatedQuadratic: The ExponentiatedQuadratic kernel.

class FeatureScaled: Kernel that first rescales all feature dimensions.

class FeatureTransformed: Input transformed kernel.

class KumaraswamyTransformed: Transform inputs by Kumaraswamy bijector.

class Linear: Linear Kernel.

class MaternFiveHalves: Matern 5/2 Kernel.

class MaternOneHalf: Matern Kernel with parameter 1/2.

class MaternThreeHalves: Matern Kernel with parameter 3/2.

class Polynomial: Polynomial Kernel.

class PositiveSemidefiniteKernel: Abstract base class for positive semi-definite kernel functions.

class RationalQuadratic: RationalQuadratic Kernel.

class SchurComplement: The SchurComplement kernel.