fix partial energy forward looking trick in backward sampling and make it efficient.#4

hyeok9855 wants to merge 3 commits intoGFNOrg:staggerGFNOrg/gfn-diffusion:staggerfrom

hyeok9855:staggerhyeok9855/gfn-diffusion:staggerCopy head branch name to clipboard