Module: tfp.experimental.substrates.jax.util

TensorFlow Probability python utilities.

Modules

deferred_tensor module: Class definitions for declaritive (vs imperative) Tensors & Variables.

seed_stream module: Local PRNG for amplifying seed entropy into seeds for base operations.

Classes

class SeedStream: Local PRNG for amplifying seed entropy into seeds for base operations.

Functions

DeferredTensor(...)

TransformedVariable(...)

remove_undocumented(...)