Solved by AbdulsalamBande in post #2 
              
              
                Try attention is all you need paper. Attention Is All You Need 
If you find it hard to implement, you can check  Video on Attention is all you need. It explains the code in detail.