Two heads of attention over three positions: each row is how much one query reads each key. Switch on the causal mask and a position can no longer read the future.
This widget runs in your browser and needs JavaScript.