Index

_

__init__() (ksd_metric.kernel.KernelInterface method) (ksd_metric.kernel.KernelJax method) (ksd_metric.stein.KernelSteinDiscrepancyInterface method) (ksd_metric.stein.KernelSteinDiscrepancyJax method) (ksd_metric.target.TargetDistributionJax method)

B

base_kernel_function() (ksd_metric.kernel.KernelInterface method) (ksd_metric.kernel.KernelJax method)

C

cross_partial_derivative_kernel_function() (ksd_metric.kernel.KernelInterface method) (ksd_metric.kernel.KernelJax method)

G

grad_log_target_pdf() (ksd_metric.target.TargetDistributionInterface method) (ksd_metric.target.TargetDistributionJax method)

I

imq() (ksd_metric.utils.JaxKernelFunction static method)

J

JaxKernelFunction (class in ksd_metric.utils)

K

kernel_stein_discrepancy() (ksd_metric.stein.KernelSteinDiscrepancyInterface method) (ksd_metric.stein.KernelSteinDiscrepancyJax method) KernelInterface (class in ksd_metric.kernel) KernelJax (class in ksd_metric.kernel) KernelSteinDiscrepancyInterface (class in ksd_metric.stein) KernelSteinDiscrepancyJax (class in ksd_metric.stein) ksd_metric.kernel module

ksd_metric.stein module ksd_metric.target module ksd_metric.utils module

L

linear() (ksd_metric.utils.JaxKernelFunction static method)

log_target_pdf() (ksd_metric.target.TargetDistributionInterface method) (ksd_metric.target.TargetDistributionJax method)

M

module ksd_metric.kernel ksd_metric.stein ksd_metric.target ksd_metric.utils

P

partial_derivative_x_kernel_function() (ksd_metric.kernel.KernelInterface method) (ksd_metric.kernel.KernelJax method)

partial_derivative_y_kernel_function() (ksd_metric.kernel.KernelInterface method) (ksd_metric.kernel.KernelJax method)

R

rbf() (ksd_metric.utils.JaxKernelFunction static method)

S

stein_kernel() (ksd_metric.stein.KernelSteinDiscrepancyInterface method) (ksd_metric.stein.KernelSteinDiscrepancyJax method)

T

TargetDistributionInterface (class in ksd_metric.target)

TargetDistributionJax (class in ksd_metric.target)