Pytorch implementation of "Adapting Text Embeddings for Causal Inference"