Official code for "The Overthinker's DIET: Cutting Token Calories with DIfficulty-AwarE Training" - View it on GitHub