/word2vec-pytorch

A fast, modern, high-quality implementation of word2vec in PyTorch.

Primary LanguagePythonMIT LicenseMIT

word2vec PyTorch

GitHub Action test status

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.