typing.BlackjaxBuildFn

A blackjax sampler build function (rng_key, logdensity_fn, position, num_warmup).

typing.BlackjaxBuildFn=Callable[…, object]

Runs the adaptation and returns (inner_state, params), with params a pytree of numeric arrays holding the tuned parameters. Static settings that blackjax needs as concrete Python values (for example an integer max_num_doublings) do not go in params: bind them in the BlackjaxStepFn. rng_key is first, matching the package’s rng-key-first convention. Consumed by BlackjaxCustomKernel.