Transformer Architecture In Python