JoMA: Demystifying Multilayer Transformers via Joint Dynamics of MLP and Attention - Tian, Yuandong; Wang, Yiping; Zhang, Zhenyu; Chen, Beidi; Du, Simon (International Conference on Learning Representations, 2024)