Module: tfp.substrates.jax

TFP for JAX.

View aliases

Main aliases

tfp.experimental.substrates.jax

Modules

bijectors module: Bijective transformations.

distributions module: Statistical distributions.

math module: TensorFlow Probability math functions.

mcmc module: TensorFlow Probability MCMC python package.

monte_carlo module: Support for Monte Carlo expectations.

optimizer module: TensorFlow Probability Optimizer python package.

random module: TensorFlow Probability random samplers/utilities.

stats module: Statistical functions.

sts module: Framework for Bayesian structural time series models.

util module: TensorFlow Probability python utilities.

vi module: Methods and objectives for variational inference.

Other Members
version	'0.17.0'