Two papers are accepted by NeurIPS 2023. (Spotlight paper!) Max-margin token selection in attention mechanism Dissecting chain-of-thought: Compositionality through in-context filtering and learning