AUTO-ENCODING VARIATIONAL BAYES