"""Transformer modules.""" from text_recognizer.networks.transformer.embeddings.rotary import RotaryEmbedding from text_recognizer.networks.transformer.attention import Attention from text_recognizer.networks.transformer.embeddings.axial import ( AxialPositionalEmbedding, )