vit-prisma

View on PyPIReverse Dependencies (0)

0.1.4 vit_prisma-0.1.4-py3-none-any.whl

Wheel Details

Project: vit-prisma
Version: 0.1.4
Filename: vit_prisma-0.1.4-py3-none-any.whl
Download: [link]
Size: 111790
MD5: b2667a024d3e3937ce4fe0ece8ab8197
SHA256: d952a570ebbb674929811cf3455a0f8ada6c1e8bc1a37dbb75a6c751be96b91d
Uploaded: 2024-03-10 01:04:28 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: vit-prisma
Version: 0.1.4
Summary: A Vision Transformer library for mechanistic interpretability.
Author: Sonia Joseph
Author-Email: soniamollyjoseph[at]gmail.com
Home-Page: https://github.com/soniajoseph/vit-prisma
Keywords: vision-transformer,clip,multimodal,machine-learning,mechanistic interpretability
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.6
Requires-Dist: pytest (>=6.0)
Requires-Dist: torch
Requires-Dist: numpy
Requires-Dist: jaxtyping
Requires-Dist: einops
Requires-Dist: fancy-einsum
Requires-Dist: plotly (==5.19.0)
Requires-Dist: timm
Requires-Dist: transformers
Requires-Dist: scikit-learn
Requires-Dist: datasets
Description-Content-Type: text/markdown
[Description omitted; length: 9802 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.42.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
vit_prisma/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/configs/CircleConfig.py sha256=8U9p6ldoic14gU7NZX3iPs_AgPIqTeIxQMHZILtexp4 2249
vit_prisma/configs/DSpritesConfig.py sha256=Flr0RvaIzGjmvLW9gUIlipbcGqX4Q8ZIvXLremjUzfg 2523
vit_prisma/configs/HookedViTConfig.py sha256=uOgMDAvCt2TqP3mcTxcdkzTKBqogedZO8GmJSofMhKA 3316
vit_prisma/configs/InductionConfig.py sha256=dtcY6D927o_O6RbN_1e6jccV8rHWv22k9D5v5iF_aTw 2390
vit_prisma/configs/MNISTConfig.py sha256=Az-YwngHBbnmo29LWo7OyHi17tkvSZvR-euV6zBjyN0 2208
vit_prisma/configs/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/configs/timmConfigs/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/configs/timmConfigs/vit_base_patch32_224_Config.py sha256=RnuxOeps4UZXamvjrnmh2vVBhgcvVXCIZdAYn-ivGf0 2349
vit_prisma/dataloaders/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/dataloaders/cifar10.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/dataloaders/circle.py sha256=ECMYqvKQtg97hG9NRvp83Ws_y1vW0cjbddp80URaGx8 5864
vit_prisma/dataloaders/dsprites.py sha256=FoPGADpQhrXHeofeZQnDv0oDAo6TwXI2Vvcl53zwRy4 1268
vit_prisma/dataloaders/induction.py sha256=JCg--VbbVZkSzNq53io5AFrRLoxkD1Jij5gjNIvpY5k 7724
vit_prisma/dataloaders/polygenic_induction.py sha256=awKBQdXL-JOTl5zTjaDQsJ_wej2-EnykFXm4oGcAeAY 7047
vit_prisma/models/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/models/activation_fns.py sha256=s8fFyxNEbfedDPWFLFX9hWabBQ4YGKoJ7GCxrrLc6hY 1381
vit_prisma/models/base_vit.py sha256=yNTHJbP7JecWdFpHt_bxe0dc0V7_gQtHyKTVz5U3Muw 34657
vit_prisma/models/dictionaries.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/models/prisma_net.py sha256=zgnZ-3JpQUDm8vtRpCZRWTXXnMT5sw-Qtk-wUSGMkEc 730
vit_prisma/models/layers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/models/layers/attention.py sha256=C5_l8n7S7wudIY1GEtc-iLXCM7g0Whb1lfd6Q7X_GIw 10867
vit_prisma/models/layers/head.py sha256=55rBtaDfwtSkT7q1oiplonENfll75kR3k0nWRS-VPFg 1078
vit_prisma/models/layers/layer_norm.py sha256=c2foia9m72AVlbNoP8e3lly3Nl9oTOVcrXIq6qkscck 3507
vit_prisma/models/layers/mlp.py sha256=kifpzNJIbWQ8YnHASiqvaq46MRMGFGgDY901FDZMwSo 2922
vit_prisma/models/layers/patch_embedding.py sha256=tEQpQIbGRfNY1W-79PINLro1iSZyFEnPqeOZvtZVVQU 1827
vit_prisma/models/layers/position_embedding.py sha256=JepNoGL3Xh-K-XTTy6jMPwj38fQFSnDyTcNk5JPtX-I 1138
vit_prisma/models/layers/transformer_block.py sha256=NOz5WittGl7HKKXU5wFbum4OwL6K3UcKvosnz2IrzW4 7769
vit_prisma/prisma_tools/__init__.py sha256=UCP3TrtCiFN6Fof4Ml29He1m8D7lLla-UemrtFmzPTU 127
vit_prisma/prisma_tools/activation_cache.py sha256=6BGYVKXsSN3L35_t71lkTOk3UBHO_9fk6zB_KoGrSF8 34650
vit_prisma/prisma_tools/factored_matrix.py sha256=dLyW13STZqYaErIFNPIe4xlzsTEnnc7AbrFa0oPrA4E 9529
vit_prisma/prisma_tools/hook_point.py sha256=JTTyRL-IBxfg6wX1G50Ta7Zeeu0Yr26tqi5RA260HMg 3886
vit_prisma/prisma_tools/hooked_root_module.py sha256=IEFWR1LxN4JOiCXlVpBmiAbOoGx6VQEpWH4G26b4OiU 9710
vit_prisma/prisma_tools/lens_handle.py sha256=t7y2hNGZrJyy-1AMeEIF_8q-lbWLtJeFG8h9piEtZhg 766
vit_prisma/prisma_tools/loading_from_pretrained.py sha256=R1jIBPP0S4-Vi-LqWwJkJd-llLiXjHY_gMtdg67nzPE 18892
vit_prisma/prisma_tools/logit_lens.py sha256=_hmNGeROzqU2ewdgTKr31Fe6bMhxPMsl3ab3p5dsJ60 2253
vit_prisma/training/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/training/early_stopping.py sha256=vUFdZ1mLapEmDKLIE-_gzKjnCeR3iSgSbEcUMpQ2NMc 827
vit_prisma/training/schedulers.py sha256=lG0GGaNmRq5SBMBEd_2OL0_rnF_ZGh-mA4PbMoXA6Ow 1144
vit_prisma/training/trainer.py sha256=tPFGUx3MU3z3jXZZztI5GYB-pK0HCpl86A8Udt93irM 6647
vit_prisma/training/training_dictionary.py sha256=Z_0XN0MizEIefGrmB7QcwgSorkQWQ6FkOJeBv5cuj_s 433
vit_prisma/training/training_utils.py sha256=J0Bqr2KaJ86ddmXtci3_snVZRO0-3M_1xp2UbiFBxuU 1468
vit_prisma/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/utils/devices.py sha256=PqtQvc7reb8Svph6f7PUHoEKiRDlfBOBI0CAYE0Eat0 2501
vit_prisma/utils/get_activations.py sha256=prpzRItV_bwXVNIzrZg_I7otBetHSaODcskcavw0Ahs 5870
vit_prisma/utils/prisma_utils.py sha256=q9weTSzD2EEYAHwm9MyN3mTgeo7evNMQbuetnI2LfUE 11077
vit_prisma/utils/saving_utils.py sha256=hABvNzT9rLxrgYPGyVyCturJLAmV5w_kd__HnzrJw-o 902
vit_prisma/utils/wandb_utils.py sha256=bx5wh1ofrl0dfm5eUQpfI9r_tRvsRWaEF6JU6us7utA 629
vit_prisma/utils/data_utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/utils/data_utils/imagenet_dict.py sha256=msgFX-eMFgK4TrG-V2fxXdIJIaIGa984GppfOCHwZhc 30581
vit_prisma/utils/data_utils/imagenet_emoji.py sha256=VAMi4RSVugNUaXdD6vwvj1tkwezw9cI4TLyFx0RW7T8 40652
vit_prisma/utils/data_utils/imagenet_utils.py sha256=RtNe_WEvhpAjJBfhEPxxEQxs60ZzhT4m2sbQlwsmDKA 662
vit_prisma/utils/imagenet/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/visualization/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
vit_prisma/visualization/get_color.js sha256=mdKyRtxZ8yyx0Pu4hxPh5PY9-9owzbDOZ2vW_kxy5P4 1961
vit_prisma/visualization/main_visualize.js sha256=LmN5OkwyoDQWr_o62FdizJunYg0mtQb5gUFcRY-rJLI 7958
vit_prisma/visualization/patch_level_logit_lens.py sha256=jD_5_mD9QM2uAXPpBQ6j9I2Z6Wb8_-Cvo4nSFv9btT0 6874
vit_prisma/visualization/patch_to_img.js sha256=cNf8HXaUhlZg0C9_bzzz47s7NxunK3c3aOiEbKN6c8w 283
vit_prisma/visualization/template.html sha256=n86xs8uICa09A-yCp5hAtzc8Ne0jBH7GMpT5hmX4r6A 1040
vit_prisma/visualization/visualize_attention.py sha256=ZOZHFslY0HpHUdje42gCB4N3hRFgu9vxxeNOxC5x0lc 3193
vit_prisma/visualization/visualize_attention_js.py sha256=bn9Fs6KlPFYEFf07Es6CKBMJqozkBHPXXtSp4RNNDn8 7489
vit_prisma/visualization/visualize_image.py sha256=OXxRRkGWsOrL6Md74TlkJeki5yVHzK-VJlKbhrKry8k 1540
vit_prisma-0.1.4.dist-info/METADATA sha256=WaCCj1_Mu9UEAkiKUGqj6xtuS-i3xNLUI1VuAvOScOE 10750
vit_prisma-0.1.4.dist-info/WHEEL sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM 92
vit_prisma-0.1.4.dist-info/top_level.txt sha256=7425OkRsZ7uP2HLX-Rp3M3amT0pL_i-Jcd1QNOISF30 11
vit_prisma-0.1.4.dist-info/RECORD

top_level.txt

vit_prisma