https://medium.com/@monocosmo77/working-with-variational-inference-in-machine-learning-part4-f3eb29878257