Tensorflow Probability Save Model. Consider the following minimal VAE: import tensorflow as tf impo