README.md
pyproject.toml
src/neo_taker/__init__.py
src/neo_taker/data_classes.py
src/neo_taker/hooks.py
src/neo_taker/model.py
src/neo_taker/model_maps.py
src/neo_taker.egg-info/PKG-INFO
src/neo_taker.egg-info/SOURCES.txt
src/neo_taker.egg-info/dependency_links.txt
src/neo_taker.egg-info/requires.txt
src/neo_taker.egg-info/top_level.txt
src/neo_taker/component_maps/__init__.py
src/neo_taker/component_maps/map_data_classes.py
src/neo_taker/component_maps/map_utils.py
src/neo_taker/component_maps/maps.py
src/neo_taker/component_maps/modelling_gemma3.py
src/neo_taker/component_maps/modelling_gpt2.py
src/neo_taker/component_maps/modelling_llama.py
src/neo_taker/component_maps/modelling_mistral.py
src/taker/__init__.py
src/taker/activations.py
src/taker/data_classes.py
src/taker/detoxify.py
src/taker/eval.py
src/taker/hooks.py
src/taker/mia.py
src/taker/model.py
src/taker/model_maps.py
src/taker/model_repos.py
src/taker/nn.py
src/taker/parser.py
src/taker/prune.py
src/taker/scoring.py
src/taker/texts.py
src/taker/vit_processor.py
src/transformer_lens/ActivationCache.py
src/transformer_lens/BertNextSentencePrediction.py
src/transformer_lens/FactoredMatrix.py
src/transformer_lens/HookedEncoder.py
src/transformer_lens/HookedEncoderDecoder.py
src/transformer_lens/HookedTransformer.py
src/transformer_lens/HookedTransformerConfig.py
src/transformer_lens/SVDInterpreter.py
src/transformer_lens/__init__.py
src/transformer_lens/evals.py
src/transformer_lens/head_detector.py
src/transformer_lens/hook_points.py
src/transformer_lens/loading_from_pretrained.py
src/transformer_lens/past_key_value_caching.py
src/transformer_lens/patching.py
src/transformer_lens/train.py
src/transformer_lens/utils.py
src/transformer_lens/components/__init__.py
src/transformer_lens/components/abstract_attention.py
src/transformer_lens/components/attention.py
src/transformer_lens/components/bert_block.py
src/transformer_lens/components/bert_embed.py
src/transformer_lens/components/bert_mlm_head.py
src/transformer_lens/components/bert_nsp_head.py
src/transformer_lens/components/bert_pooler.py
src/transformer_lens/components/embed.py
src/transformer_lens/components/grouped_query_attention.py
src/transformer_lens/components/layer_norm.py
src/transformer_lens/components/layer_norm_pre.py
src/transformer_lens/components/pos_embed.py
src/transformer_lens/components/rms_norm.py
src/transformer_lens/components/rms_norm_pre.py
src/transformer_lens/components/t5_attention.py
src/transformer_lens/components/t5_block.py
src/transformer_lens/components/token_typed_embed.py
src/transformer_lens/components/transformer_block.py
src/transformer_lens/components/unembed.py
src/transformer_lens/components/mlps/can_be_used_as_mlp.py
src/transformer_lens/components/mlps/gated_mlp.py
src/transformer_lens/components/mlps/gated_mlp_4bit.py
src/transformer_lens/components/mlps/mlp.py
src/transformer_lens/components/mlps/moe.py
src/transformer_lens/factories/activation_function_factory.py
src/transformer_lens/factories/mlp_factory.py
src/transformer_lens/pretrained/__init__.py
src/transformer_lens/pretrained/weight_conversions/__init__.py
src/transformer_lens/pretrained/weight_conversions/bert.py
src/transformer_lens/pretrained/weight_conversions/bloom.py
src/transformer_lens/pretrained/weight_conversions/coder.py
src/transformer_lens/pretrained/weight_conversions/gemma.py
src/transformer_lens/pretrained/weight_conversions/gpt2.py
src/transformer_lens/pretrained/weight_conversions/gptj.py
src/transformer_lens/pretrained/weight_conversions/llama.py
src/transformer_lens/pretrained/weight_conversions/mingpt.py
src/transformer_lens/pretrained/weight_conversions/mistral.py
src/transformer_lens/pretrained/weight_conversions/mixtral.py
src/transformer_lens/pretrained/weight_conversions/nanogpt.py
src/transformer_lens/pretrained/weight_conversions/neel_solu_old.py
src/transformer_lens/pretrained/weight_conversions/neo.py
src/transformer_lens/pretrained/weight_conversions/neox.py
src/transformer_lens/pretrained/weight_conversions/opt.py
src/transformer_lens/pretrained/weight_conversions/phi.py
src/transformer_lens/pretrained/weight_conversions/phi3.py
src/transformer_lens/pretrained/weight_conversions/qwen.py
src/transformer_lens/pretrained/weight_conversions/qwen2.py
src/transformer_lens/pretrained/weight_conversions/qwen3.py
src/transformer_lens/pretrained/weight_conversions/t5.py
src/transformer_lens/utilities/__init__.py
src/transformer_lens/utilities/activation_functions.py
src/transformer_lens/utilities/addmm.py
src/transformer_lens/utilities/attention.py
src/transformer_lens/utilities/devices.py
tests/test_user_example.py