liesel.liesel.goose.make_log_prob_fn#

liesel.liesel.goose.make_log_prob_fn(model, jaxify=True)[source]#

Returns a pure and jittable log_prob function for the provided model.

Return type

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