Module: tfp.substrates.jax.experimental

TensorFlow Probability API-unstable package.

This package contains potentially useful code which is under active development with the intention eventually migrate to TFP proper. All code in tfp.experimental should be of production quality, i.e., idiomatically consistent, well tested, and extensively documented. tfp.experimental code relaxes the TFP non-experimental contract in two regards:

  1. tfp.experimental has no API stability guarantee. The public footprint of tfp.experimental code may change without notice or warning.
  2. Code outside tfp.experimental cannot depend on code within tfp.experimental.

You are welcome to try any of this out (and tell us how well it works for you!).


distributions module: TensorFlow Probability experimental distributions package.

mcmc module: TensorFlow Probability experimental MCMC package.

unnest module: Utilities for accessing nested attributes.

util module: TensorFlow Probability experimental python utilities.