Probabilistic Deep Learning with TensorFlow 2