Sadegh Mahdavi

Latest

• Memorization Capacity of Multi-Head Attention in Transformers
• Towards Better Out-of-Distribution Generalization of Neural Algorithmic Reasoning Tasks