liesel.goose.models#

Model interfaces.

Classes

DataClassModel(log_prob_fn)

A model interface for a model state represented by a dataclass and a corresponding log-probability function.

DictModel(log_prob_fn)

A model interface for a model state represented by a dict[str, Array] and a corresponding log-probability function.