kd.nn.TransformerBlock

class kauldron.modules.TransformerBlock(*args, **kwargs)

Bases: Protocol

Interface definition for transformer blocks (for use in type annotations).

Examples include: PreNormBlock, PostNormBlock and ParallelAttentionBlock.