Revealing example of self-attention, the building block of transformer AI models - View it on GitHub