Have you see this one? https://jalammar.github.io/illustrated-transformer/