There are three files in this supplementary material.

sd3_sde_with_logprob.py : The core implementation of our sampling algorithm. Copy it to `flow_grpo/diffusers_patch` in flow-grpo's repository, then our results can be reproduced by running the code of flow-grpo.

sde_noise_level.ipynb : The scripts for drawing Figure 2. 

readme.txt : current readme file.