Research codebase accompanying the ICLR 2026 paper DriftLite: Lightweight Drift Control for Inference-Time Scaling of Diffusion Models.