NUTSKernel.log_prob_fn()

NUTSKernel.log_prob_fn(model_state)

Returns the log-probability function with the position as the only argument.

Return type

Callable[NewType(Position, dict[str, Any]), float]