SEA: State-Exchange Attention for High-Fidelity Physics Based Transformers - Esmati, Parsa; Dadashzadeh, Amirhossein; Ardakani, Vahid Goodarzi; Larrosa, Nicolas; Grilli, Nicolò (Advances in Neural Information Processing Systems, 2024)