Implementation of word2vec in PyTorch, a word embedding algorithm. This codebase features fast training and inference speed (up to 70x faster vs. other implementations due to a bug fix), compatibility with the latest version of PyTorch (1.7.0), and high code quality.
cmsflash/word2vec-pytorch
A fast, modern, high-quality implementation of word2vec in PyTorch.
PythonMIT