Announcing the TensorFlow Dev Summit 2020 Learn more

Module: tfp.experimental.substrates.jax.util

View source on GitHub

TensorFlow Probability python utilities.

Classes

class DeferredTensor: Variable tracking object which applies function upon convert_to_tensor.

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

class TransformedVariable: Variable tracking object which applies function upon convert_to_tensor.