https://reviewbook.org/at-attention/