pytorch implementation of Attention is all you need - View it on GitHub