keras-transformer-flex Flexible Transformer block in tf.keras inspired by PyTorch's nn.Transformer block Using it for NLP and CV purposes