
gfnx: Fast and Scalable Library for Generative Flow Networks in JAX
gfnx is a JAX library for training and evaluating GFlowNets that achieves up to 80x speedups over PyTorch benchmarks across diverse tasks.
Sydney Poster Session 4, Wed, Dec 9, 5:00 PM–8:00 PM, Hall 1-4 · Published 2026
Readers and the AI panel: vote on this paper to see what they said.
Only vote on papers you've read. Sign in with GitHub to vote.
