This is a pytorch implementation of the ProdLDA model in https://arxiv.org/pdf/1703.01488.pdf. Original Tensorflow implementation here Another pytorch implementation here