UNDO: Understanding Distillation as Optimization. - Kushal Jain; Piyushi Goyal; Kumar Shridhar (CoRR, 2025)