fairseq: A Fast, Extensible Toolkit for Sequence Modeling
Abstract
fairseq is an open-source sequence modeling toolkit that allows researchers and developers to train custom models for translation, summarization, language modeling, and other text generation tasks. The toolkit is based on PyTorch and supports distributed training across multiple GPUs and machines. We also support fast mixed-precision training and inference on modern GPUs. A demo video can be found at https://www.youtube.com/watch?v=OtgDdWtHvto
- Publication:
-
arXiv e-prints
- Pub Date:
- April 2019
- arXiv:
- arXiv:1904.01038
- Bibcode:
- 2019arXiv190401038O
- Keywords:
-
- Computer Science - Computation and Language
- E-Print:
- NAACL 2019 Demo paper