An implemenetation of a Generative Flow Network (GFlowNet) for sampling a sequence of bits via the trajectory balance objective.
Code is adapted from Emmanuel Bengio's Google Colab tutorial for implementing a GFlowNet.
TO-DOs:
- Modularize design
- Vectorize to sample N trajectories at once
- Store training result graphs