mlsanityData to 14 Oct 2026

torchgfn: A PyTorch GFlowNet library

arXiv 2305.14594 ↗First posted 24 May 2023cs.LGcited by 11

Not on the map. Why?

Abstract (arXiv)

The growing popularity of generative flow networks (GFlowNets or GFNs) from a range of researchers with diverse backgrounds and areas of expertise necessitates a library that facilitates the testing of new features (e.g., training losses and training policies) against standard benchmark implementations, or on a set of common environments. We present torchgfn, a PyTorch library that aims to address this need. Its core contribution is a modular and decoupled architecture which treats environments, neural network modules, and training objectives as interchangeable components. This provides users with a simple yet powerful API to facilitate rapid prototyping and novel research. Multiple examples are provided, replicating and unifying published results. The library is available on GitHub (https://github.com/GFNOrg/torchgfn) and on pypi (https://pypi.org/project/torchgfn/).

Key facts. Cited by 11 papers in the map (data through 14 Oct 2026). Builds on 9 of its 19 references that are in the map. First posted on arXiv 24 May 2023.

Builds on

9 of 19 references.

References · 9

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  2. 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 392
  3. 2021GFlowNet Foundationscited by 290
  4. 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 251
  5. 2022Bayesian Structure Learning with Generative Flow Networkscited by 153
  6. 2022Learning GFlowNets from partial episodes for improved convergence and stabilitycited by 132
  7. 2023A theory of continuous generative flow networkscited by 101
  8. 2023Better Training of GFlowNets with Local Credit and Incomplete Trajectoriescited by 85
  9. 2023Robust Scheduling with GFlowNetscited by 69

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2023: 3; 2024: 0; 2025: 3; 2026: 5.

11 papers.

  1. 2023Generative Flow Networks as Entropy-Regularized RLcited by 64
  2. 2023Order-Preserving GFlowNetscited by 13
  3. 2025gfnx: Fast and Scalable Library for Generative Flow Networks in JAXcited by 3
  4. 2023BatchGFN: Generative Flow Networks for Batch Active Learningcited by 2
  5. 2025Curriculum-Augmented GFlowNets For mRNA Sequence Generationcited by 2
  6. 2025Improved Exploration in GFlownets via Enhanced Epistemic Neural Networkscited by 1
  7. 2026Evaluating GFlowNet from partial episodes for stable and flexible policy-based trainingcited by 1
  8. 2026Stable GFlowNets with TV Monitoring and Probabilistic Guaranteescited by 1
  9. 2026Particle GFlowNets: Rethinking Generative Marginalization Modelscited by 0
  10. 2026Learning a Mixture of GFlowNetscited by 0
  11. 2026Mixture of GFlowNetscited by 0