Causal Effect Inference with Deep Latent-Variable Models
auto - encoding variational bayes .