Awesome
Neural Machine Translation Implementations
Contributions
Feel free add to this via a pull request, with each section alphabetically ordered.
Project Stats
- 23 project names include nmt
- 6 project names include seq2seq
- 2 project names include dl4mt
- 3 projects are derived from Kyunghyun Cho's tutorial at the 2015 DL4MT winter school
The Lists
Caffe2-based NMT
| Software | Who | Comments |
|---|---|---|
| seq2seq | Deepak, Yury |
Chainer-based NMT
| Software | Who | Comments |
|---|---|---|
| attention_is_all_you_need | Sosuke | impements the Transformer |
| chainn | Philip | |
| KyotoNMT | Fabien |
Custom C/C++ NMT
| Software | Who | Comments |
|---|---|---|
| EUREKA-MangoNMT | Jiajun | |
| Mantidae | Vu | A successor of Mantis |
| Marian | Marcin, Tomasz | Compatible with Nematus models |
| N3LP | Kazuma, Akiko | |
| Zoph_RNN | Barret | Supports multi-GPU parallelism |
DyNet-based NMT
| Software | Who | Comments |
|---|---|---|
| dynmt-py | Roee | Python |
| lamtram | Graham | C++ |
| mantis | Trevor | C++ |
| NMTKit | Yusuke | C++ |
| Transformer-DyNet | Vu | C++; implements the Transformer |
| xnmt | NeuLab | Python |
Keras-based NMT
| Software | Who | Comments |
|---|---|---|
| Keras seq2seq | Fariz | |
| NMT-Keras | Álvaro |
MXNet-based NMT
| Software | Who | Comments |
|---|---|---|
| MXNMT | magic282 | |
| sockeye | AWS Labs |
PyTorch-based NMT
| Software | Who | Comments |
|---|---|---|
| attention-is-all-you-need-pytorch | Yu-Hsiang | |
| fairseq-py | Myle & Facebook Research | |
| Joey NMT | Heidelberg University & University of Amsterdam | minimalist, beginner-friendly |
| NJUNMT-pytorch | Haoran Wei & Zaixiang Zheng | |
| nmtpytorch | Ozan | |
| nonauto-nmt | Salesforce Research | |
| OpenNMT-py | Harvard, Systran | |
| pytorch_nmt | Pengcheng | |
| seq2seq.pytorch | Elad | |
| UNdreaMT | Mikel | Unsupervised |
TensorFlow-based NMT
Theano-based NMT
Note: Theano will be discontinued
| Software | Who | Comments |
|---|---|---|
| DCNMT | SwordYork | |
| dl4mt-tutorial | Cho, Orhan | |
| dl4mt-c2c | Jason, Cho | |
| HNMT | Robert | |
| Nematus | Rico, Cho, Orhan | |
| neuralmt | Raphael | |
| NMT | Zhaopeng | Adds coverage and context gate |
| nmtpy | LIUM | refactored dl4mt-tutorial |
| SGNMT | Cambridge | Syntactically-guided NMT |
| THUMT | Tsinghua University |
Torch-based NMT
| Software | Who | Comments |
|---|---|---|
| fairseq | Novel CNN+attention | |
| nmt-android | Yoon, Sasha | |
| NPMT | Po-Sen, Fairseq team | Phrase-based NMT |
| OpenNMT | Harvard, Systran | |
| seq2seq-attn | Yoon | Superseded by OpenNMT, except for character-level stuff |
Other
| Software | Who | Backend | Comments |
|---|---|---|---|
| nmt.hybrid | Thang | Matlab | also word-based |
| textsum | TensorFlow | text summarization demo using seq2seq+attn |