transformer_lens.factories package

Submodules

• transformer_lens.factories.activation_function_factory module
  • ActivationFunctionFactory
    • ActivationFunctionFactory.pick_activation_function()
• transformer_lens.factories.architecture_adapter_factory module
  • ArchitectureAdapterFactory
    • ArchitectureAdapterFactory.discover_entry_points()
    • ArchitectureAdapterFactory.register_adapter()
    • ArchitectureAdapterFactory.select_architecture_adapter()
• transformer_lens.factories.mlp_factory module
  • MLPFactory
    • MLPFactory.create_mlp()

Module contents