TensorFlow Probability
github.com