TensorFlow Probability on JAX

Run in Google Colab

View source on GitHub
Download notebook
TensorFlow Probability (TFP) is a library for probabil