Multi-head Transformers Provably Learn Symbolic Multi-step Reasoning via Gradient Descent - Yang, Tong; Huang, Yu; Liang, Yingbin; Chi, Yuejie (Advances in Neural Information Processing Systems, 2025)