CoTFormer: A Chain of Thought Driven Architecture with Budget-Adaptive Computation Cost at Inference - Mohtashami, Amirkeivan; Pagliardini, Matteo; Jaggi, Martin (International Conference on Learning Representations, 2025)