Variational Auto-Encoder Architectures that Excel at Causal Inference