https://medium.com/@monocosmo77/working-with-variational-inference-in-machine-learning-part3-4eced773e0a2