中文

Sockeye 3:基于PyTorch的快速神经机器翻译

计算与语言 2022-08-03 v4

摘要

Sockeye 3是Sockeye神经机器翻译(NMT)工具包的最新版本。Sockeye 3现基于PyTorch,提供更快速的模型实现与更先进的功能,并进一步精简了代码库。这使得更广泛的实验成为可能,迭代更快,可高效训练更强且更快的模型,并能灵活地将新想法从研究快速推向生产。在运行可比模型时,Sockeye 3在GPU上比其他PyTorch实现快至多126%,在CPU上快至多292%。Sockeye 3是以Apache 2.0许可证发布的开源软件。

关键词

引用

@article{arxiv.2207.05851,
  title  = {Sockeye 3: Fast Neural Machine Translation with PyTorch},
  author = {Felix Hieber and Michael Denkowski and Tobias Domhan and Barbara Darques Barros and Celina Dong Ye and Xing Niu and Cuong Hoang and Ke Tran and Benjamin Hsu and Maria Nadejde and Surafel Lakew and Prashant Mathur and Anna Currey and Marcello Federico},
  journal= {arXiv preprint arXiv:2207.05851},
  year   = {2022}
}