中文

PARAGEN:一个并行生成工具包

计算与语言 2022-10-10 v1

摘要

PARAGEN 是一个基于 PyTorch 的 NLP 工具包,用于并行生成方向的进一步开发。PARAGEN 提供十三种可定制插件,帮助用户跨模型架构、优化和学习策略快速试验新想法。我们实现了无限数据加载和自动模型选择等多种功能,以增强其工业用途。ParaGen 现已部署至字节跳动,支持多种研究与工业应用。PARAGEN 可在 https://github.com/bytedance/ParaGen 获取。

关键词

引用

@article{arxiv.2210.03405,
  title  = {PARAGEN : A Parallel Generation Toolkit},
  author = {Jiangtao Feng and Yi Zhou and Jun Zhang and Xian Qian and Liwei Wu and Zhexi Zhang and Yanming Liu and Mingxuan Wang and Lei Li and Hao Zhou},
  journal= {arXiv preprint arXiv:2210.03405},
  year   = {2022}
}

备注

9 pages, 1 figure, 6 tables