RAAS [ICLR 2026] Official Implementation for "Characterizing and Mitigating Reasoning Drift in Large Language Models"