Comment by abakus

If you just want to understand the Transformer, here is a clean implementation:

https://github.com/blue-season/pywarm/blob/master/examples/t...