Distributional Associations vs In-Context Reasoning: A Study of Feed-forward and Attention Layers - Chen, Lei; Bruna, Joan; Bietti, Alberto (International Conference on Learning Representations, 2025)