https://jalammar.github.io/visualizing-neural-machine-transl... and https://jalammar.github.io/illustrated-transformer/ for anyone looking