torchgfn：一个PyTorch GFlowNet库
机器学习
2026-03-23 v4
摘要
生成流网络（GFlowNets或GFNs）日益受到来自不同背景和专业领域研究人员的欢迎，这亟需一个库来支撑将新特性（如训练损失与训练策略）与标准基准实现或一组通用环境进行对比测试。我们提出torchgfn，一个旨在满足此需求的PyTorch库。其核心贡献是一种模块化且解耦的架构，将环境、神经网络模块与训练目标视为可互换组件。这为使用者提供了简洁而强大的API，以加速原型设计与新颖研究。我们提供了多个示例，复现并统一了已发表的结果。该库可在GitHub（https://github.com/GFNOrg/torchgfn）与pypi（https://pypi.org/project/torchgfn/）获取。
引用
@article{arxiv.2305.14594,
title = {torchgfn: A PyTorch GFlowNet library},
author = {Joseph D. Viviano and Omar G. Younis and Sanghyeok Choi and Victor Schmidt and Yoshua Bengio and Salem Lahlou},
journal= {arXiv preprint arXiv:2305.14594},
year = {2026}
}
备注
15 pages, 3 figures, 3 tables. Submitted