transformer_lens.components#
Submodules#
- transformer_lens.components.abstract_attention
- transformer_lens.components.attention
- transformer_lens.components.bert_block
- transformer_lens.components.bert_embed
- transformer_lens.components.bert_mlm_head
- transformer_lens.components.embed
- transformer_lens.components.grouped_query_attention
- transformer_lens.components.layer_norm
- transformer_lens.components.layer_norm_pre
- transformer_lens.components.pos_embed
- transformer_lens.components.rms_norm
- transformer_lens.components.rms_norm_pre
- transformer_lens.components.t5_attention
- transformer_lens.components.t5_block
- transformer_lens.components.token_typed_embed
- transformer_lens.components.transformer_block
- transformer_lens.components.unembed