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