Yoshua Bengio

torchgfn: A PyTorch GFlowNet library

torchgfn：一个PyTorch GFlowNet库

Joseph D. Viviano, Omar G. Younis, Sanghyeok Choi, Victor Schmidt, Yoshua Bengio, Salem Lahlou

机构 * MBZUAI（穆罕默德·本·拉希德智能研究院） 纠错

AI总结 本文提出torchgfn库，通过模块化架构实现环境、神经网络模块和训练目标的可替换性，简化了GFlowNet的研究与实验流程。

Comments 15 pages, 3 figures, 3 tables. Submitted