Module: tfp.experimental.substrates.jax.util

View source on GitHub

JAX util.

Modules

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.