Paper you’re looking for is called “Attention is all you need”