Learning Probabilistic Representations For Inference, Training And Interpretability