View source on GitHub |
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
make_trainable(...)
: Constructs a distribution or bijector instance with trainable parameters.
make_trainable_stateless(...)
: Constructs a distribution or bijector instance with trainable parameters.