Module: tfp.substrates.jax.util

TensorFlow Probability python utilities.

View aliases

Main aliases

tfp.experimental.substrates.jax.util

Classes

class BatchedComponentProperties: Alias to assist in defining properties of non-Tensor parameters.

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.