Module: tfp.experimental.util

TensorFlow Probability experimental python utilities.

Classes

class DeferredModule: Wrapper to defer initialization of a tf.Module instance.

class JitPublicMethods: Wrapper to compile an object's public methods using XLA.

Functions

as_composite(...)

make_trainable(...): Constructs a distribution or bijector instance with trainable parameters.

make_trainable_stateless(...): Constructs a distribution or bijector instance with trainable parameters.

register_composite(...)

DEFAULT_METHODS_EXCLUDED_FROM_JIT ('event_shape', 'event_shape_tensor', 'batch_shape', 'batch_shape_tensor', 'dtype', 'kl_divergence', 'experimental_default_event_space_bijector', 'experimental_local_measure', 'forward_event_shape', 'forward_event_shape_tensor', 'inverse_event_shape', 'inverse_event_shape_tensor', 'forward_dtype', 'inverse_dtype', 'forward_event_ndims', 'inverse_event_ndims', 'experimental_compute_density_correction')