description: TensorFlow Probability alternative substrates.
TensorFlow Probability alternative substrates.
jax module: TFP for JAX.
jax
numpy module: TFP for Numpy.
numpy