amortized-sampling
A modular, easy to extend GFlowNet library
Environments and Algorithms for Generative Flow Networks in JAX