transformer/transformer_block library

Classes

TransformerBlock
A single Transformer block, which combines attention and feed-forward layers.