Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

Module: tfp.experimental.substrates.jax

View source on GitHub

TFP for JAX.

Modules

bijectors module: Bijective transformations.

distributions module: Statistical distributions.

math module: TensorFlow Probability math functions.

mcmc module: TensorFlow Probability MCMC python package.

stats module: Statistical functions.

util module: TensorFlow Probability python utilities.