И вот очень красивый код: весь трансформер в 40 строк на PyTorch 1.1