ModelMixin.log_prob_fn()# ModelMixin.log_prob_fn(model_state)[source]# Returns the log-probability function with the position as the only argument. Return type: Callable[[Position (dict[str, Any])], float]