transformer Trasnsformer implementation with tensorflow 2.0 inspired from https://www.tensorflow.org/tutorials/text/transformer#encoder