Some cool examples, like an animated attention head layer: https://bhosmer.github.io/mm/index.html?0=out+%3D+%28attn+%3...