GitHub

Module: tfp.substrates.jax

TFP for JAX.

Main aliases

tfp.experimental.substrates.jax

Modules

bijectors module: Bijective transformations.

distributions module: Statistical distributions.

experimental module: TensorFlow Probability API-unstable package.

math module: TensorFlow Probability math functions.

mcmc module: TensorFlow Probability MCMC python package.

optimizer module: TensorFlow Probability Optimizer python package.

random module: TensorFlow Probability random samplers/utilities.

stats module: Statistical functions.

util module: TensorFlow Probability python utilities.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-09-12 UTC.