Check out The Annotated Transformer, it’s one of my favorite references! It contains straightforward python code side by side with excerpts from the original paper.

http://nlp.seas.harvard.edu/2018/04/03/attention.html