ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfp.substrates.jax.util

TensorFlow Probability python utilities.


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

class ParameterProperties: Annotates expected properties of a Tensor-valued distribution parameter.

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

class TransformedVariable: Variable tracking object which applies a bijector upon convert_to_tensor.