pyTorchNCE

github.com ↗

An implementation of the Noise Contrastive Estimation algorithm for pyTorch. Working, yet not very efficient.