Module: tfp.experimental.vi

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_asvi_surrogate_posterior(...): Builds a structured surrogate posterior inspired by conjugate updating.

build_factored_surrogate_posterior(...): Builds a joint variational posterior that factors over model variables. (deprecated arguments) (deprecated)

build_split_flow_surrogate_posterior(...): Builds a joint variational posterior by splitting a normalizing flow.

build_trainable_location_scale_distribution(...): Builds a variational distribution from a location-scale family.

register_asvi_substitution_rule(...): Registers a rule for substituting distributions in ASVI surrogates.