Environments and Algorithms for Generative Flow Networks in JAX

Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.

Code repository for Trajectory Flow Matching

🌺 Population-Based Reinforcement Learning for Combinatorial Optimization

Official implementation of the ICLR 2023 paper "Policy-Based Self-Competition for Planning Problems"

Code for our TMLR paper "Distributional GFlowNets with Quantile Flows".

Evolution Strategies in JAX 🦎

Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselines.

Code for "Bayesian Structure Learning with Generative Flow Networks"

Simple single-file baselines for Q-Learning in pure-GPU setting

Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023

This repository implements variational graph auto encoder by Thomas Kipf.

Code of NeurIPS paper: arxiv.org/abs/2302.08224

PyTorch implementation for our NeurIPS 2023 spotlight paper "Let the Flows Tell: Solving Graph Combinatorial Optimization Problems with GFlowNets".