HMCKernel.log_prob_fn()# HMCKernel.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]