setup.py
src/vit_prisma/__init__.py
src/vit_prisma.egg-info/PKG-INFO
src/vit_prisma.egg-info/SOURCES.txt
src/vit_prisma.egg-info/dependency_links.txt
src/vit_prisma.egg-info/not-zip-safe
src/vit_prisma.egg-info/requires.txt
src/vit_prisma.egg-info/top_level.txt
src/vit_prisma/configs/CircleConfig.py
src/vit_prisma/configs/DSpritesConfig.py
src/vit_prisma/configs/HookedViTConfig.py
src/vit_prisma/configs/InductionConfig.py
src/vit_prisma/configs/MNISTConfig.py
src/vit_prisma/configs/__init__.py
src/vit_prisma/configs/timmConfigs/__init__.py
src/vit_prisma/configs/timmConfigs/vit_base_patch32_224_Config.py
src/vit_prisma/dataloaders/__init__.py
src/vit_prisma/dataloaders/cifar10.py
src/vit_prisma/dataloaders/circle.py
src/vit_prisma/dataloaders/dsprites.py
src/vit_prisma/dataloaders/induction.py
src/vit_prisma/dataloaders/polygenic_induction.py
src/vit_prisma/models/__init__.py
src/vit_prisma/models/activation_fns.py
src/vit_prisma/models/base_vit.py
src/vit_prisma/models/dictionaries.py
src/vit_prisma/models/prisma_net.py
src/vit_prisma/models/layers/__init__.py
src/vit_prisma/models/layers/attention.py
src/vit_prisma/models/layers/head.py
src/vit_prisma/models/layers/layer_norm.py
src/vit_prisma/models/layers/mlp.py
src/vit_prisma/models/layers/patch_embedding.py
src/vit_prisma/models/layers/position_embedding.py
src/vit_prisma/models/layers/transformer_block.py
src/vit_prisma/prisma_tools/__init__.py
src/vit_prisma/prisma_tools/activation_cache.py
src/vit_prisma/prisma_tools/factored_matrix.py
src/vit_prisma/prisma_tools/hook_point.py
src/vit_prisma/prisma_tools/hooked_root_module.py
src/vit_prisma/prisma_tools/lens_handle.py
src/vit_prisma/prisma_tools/loading_from_pretrained.py
src/vit_prisma/training/__init__.py
src/vit_prisma/training/early_stopping.py
src/vit_prisma/training/schedulers.py
src/vit_prisma/training/trainer.py
src/vit_prisma/training/training_dictionary.py
src/vit_prisma/training/training_utils.py
src/vit_prisma/utils/__init__.py
src/vit_prisma/utils/devices.py
src/vit_prisma/utils/get_activations.py
src/vit_prisma/utils/prisma_utils.py
src/vit_prisma/utils/saving_utils.py
src/vit_prisma/utils/wandb_utils.py
src/vit_prisma/utils/data_utils/__init__.py
src/vit_prisma/utils/data_utils/imagenet_dict.py
src/vit_prisma/visualization/__init__.py
src/vit_prisma/visualization/visualize_attention.py
src/vit_prisma/visualization/visualize_attention_js.py
tests/test_cache_hook_names.py
tests/test_hooks.py
tests/test_induction_dataset.py
tests/test_loading_timm.py
tests/test_models.py
tests/test_preprocessing.py
tests/test_trainer_circle.py
tests/test_trainer_circle_timm.py
tests/test_trainer_induction.py
tests/test_trainer_induction_timm.py
tests/test_trainer_mnist.py