reward-guidance
sanjitdp/reward-guidance
Python
Experiment code for 'Are we really tilting? The mechanics of reward guidance in flow and diffusion models' β plug-in Doob h-transform sampling, reward damping, best-of-n, and flow map reward guidance for Gaussian mixtures, a 2D checkerboard, and FLUX.1 text-to-image generation.