liesel.liesel.goose#

The Liesel-Goose interface.

Functions

get_log_prob(model_state)

Computes and returns the log-probability from a model state.

get_position(position_keys, model_state)

Extracts a position from a model state.

make_log_prob_fn(model[, jaxify])

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

make_update_state_fn(model[, jaxify])

Returns a pure and jittable update_state() function for the provided model.

update_state(position, model_state, model)

Updates and returns a model state given a position.

Classes

GooseModel(model)

A ModelInterface for a Liesel Model.