Documentation

• kernel
  • KernelInterface
    • KernelInterface.__init__()
    • KernelInterface.base_kernel_function()
    • KernelInterface.partial_derivative_x_kernel_function()
    • KernelInterface.partial_derivative_y_kernel_function()
    • KernelInterface.cross_partial_derivative_kernel_function()
  • KernelJax
    • KernelJax.__init__()
    • KernelJax.base_kernel_function()
    • KernelJax.partial_derivative_x_kernel_function()
    • KernelJax.partial_derivative_y_kernel_function()
    • KernelJax.cross_partial_derivative_kernel_function()
• stein
  • KernelSteinDiscrepancyInterface
    • KernelSteinDiscrepancyInterface.__init__()
    • KernelSteinDiscrepancyInterface.stein_kernel()
    • KernelSteinDiscrepancyInterface.kernel_stein_discrepancy()
  • KernelSteinDiscrepancyJax
    • KernelSteinDiscrepancyJax.__init__()
    • KernelSteinDiscrepancyJax.stein_kernel()
    • KernelSteinDiscrepancyJax.kernel_stein_discrepancy()
• target
  • TargetDistributionInterface
    • TargetDistributionInterface.log_target_pdf()
    • TargetDistributionInterface.grad_log_target_pdf()
  • TargetDistributionJax
    • TargetDistributionJax.__init__()
    • TargetDistributionJax.log_target_pdf()
    • TargetDistributionJax.grad_log_target_pdf()
• utils
  • JaxKernelFunction
    • JaxKernelFunction.rbf()
    • JaxKernelFunction.linear()
    • JaxKernelFunction.imq()