/keras-transformer-flex

Transformer block in tf.keras similar to PyTorch's nn.Transformer block.

Primary LanguageJupyter Notebook

keras-transformer-flex

Flexible Transformer block in tf.keras inspired by PyTorch's nn.Transformer block

Using it for NLP and CV purposes