View source on GitHub |
Experimental methods and objectives for variational inference.
Modules
util
module: Experimental methods and objectives for variational inference.
Functions
build_affine_surrogate_posterior(...)
: Builds a joint variational posterior with a given event_shape
.
build_affine_surrogate_posterior_from_base_distribution(...)
: Builds a variational posterior by linearly transforming base distributions.
build_affine_surrogate_posterior_from_base_distribution_stateless(...)
: Builds a variational posterior by linearly transforming base distributions.
build_affine_surrogate_posterior_stateless(...)
: Builds a joint variational posterior with a given event_shape
.
build_asvi_surrogate_posterior(...)
: Builds a structured surrogate posterior inspired by conjugate updating.
build_asvi_surrogate_posterior_stateless(...)
: Builds a structured surrogate posterior inspired by conjugate updating.
build_factored_surrogate_posterior(...)
: Builds a joint variational posterior that factors over model variables.
build_factored_surrogate_posterior_stateless(...)
: Builds a joint variational posterior that factors over model variables.
build_split_flow_surrogate_posterior(...)
: Builds a joint variational posterior by splitting a normalizing flow.