transformer_lens.factories.mlp_factory module

MLP Factory

Centralized location for creating any MLP needed within TransformerLens

class transformer_lens.factories.mlp_factory.MLPFactory

Bases: object

static create_mlp(cfg: HookedTransformerConfig) → CanBeUsedAsMLP