Samplers

sgmcmcjax.samplers.sgmcmc_sampler(build_kernel_fn: Callable) → Callable

Decorator that turns a kernel factory into a sampler factory.

These samplers have exactly the same signatures as the kernels they’re built from (see sgmcmcjax.kernels).

Parameters

build_kernel_fn (Callable) – kernel factory

Returns

sampling function with the same signature as build_kernel_fn

Return type

Callable