Transformer neural network