Author's PyTorch implementation of SR-DICE for marginalized importance sampling - View it on GitHub