graphpatch

View on PyPIReverse Dependencies (0)

0.2.3 graphpatch-0.2.3-py3-none-any.whl

Wheel Details

Project: graphpatch
Version: 0.2.3
Filename: graphpatch-0.2.3-py3-none-any.whl
Download: [link]
Size: 64792
MD5: e853fb475ef8b999c5cf2e69e3857957
SHA256: a736f607121bc4b9348bc95126eda60d5583fcf097935bb9da95979d88c137b5
Uploaded: 2024-10-19 01:30:57 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: graphpatch
Version: 0.2.3
Summary: graphpatch is a library for activation patching on PyTorch neural network models.
Author: Evan Lloyd
Author-Email: evan.t.lloyd[at]gmail.com
Home-Page: https://www.graphpatch.dev
Project-Url: Documentation, https://graphpatch.readthedocs.io/en/latest/index.html
Project-Url: Repository, https://github.com/evan-lloyd/graphpatch
License: MIT
Keywords: mechanistic interpretability,interpretability,pytorch,torch,activation patch,ablation,transformer,large language model,llm
Classifier: Development Status :: 4 - Beta
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.8.1,<3.13
Requires-Dist: accelerate (>=0.15.0); extra == "transformers"
Requires-Dist: bitsandbytes (>=0.35.4); sys_platform != "darwin" and extra == "transformers"
Requires-Dist: numpy (>=1.17); python_version >= "3.10"
Requires-Dist: numpy (<1.25,>=1.17); python_version < "3.9"
Requires-Dist: numpy (<2.1,>=1.17); python_version >= "3.9" and python_version < "3.10"
Requires-Dist: sentencepiece (>=0.1.94); extra == "transformers"
Requires-Dist: torch (<2.5,>=2.0); python_version >= "3.8" and python_version < "3.9"
Requires-Dist: torch (<2.6,>=2.0); python_version >= "3.9" and python_version < "3.11"
Requires-Dist: torch (<2.6,>=2.1); python_version >= "3.11" and python_version < "3.12"
Requires-Dist: torch (<2.6,>=2.4); python_version >= "3.12"
Requires-Dist: transformer-lens (>=1.16.0); extra == "transformer-lens"
Requires-Dist: transformers (>=4.28.0); extra == "transformers"
Provides-Extra: transformer-lens
Provides-Extra: transformers
Description-Content-Type: text/markdown
[Description omitted; length: 9824 characters]

WHEEL

Wheel-Version: 1.0
Generator: poetry-core 1.9.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
graphpatch/__init__.py sha256=aYGH9swRIni5QM7UNQgZ7Kf53tCGv6LPYx3x4GU_tXs 696
graphpatch/exceptions.py sha256=RkWjpoiSUp-27zm4SkPcGhr7JNMYUq2QnNbEbIupjfY 92
graphpatch/extraction/__init__.py sha256=zJugCDpSUwN804wFVIEyVfWAw7eUInd-bT5HPf9Kuec 495
graphpatch/extraction/compiled_graph_module.py sha256=HBn47jodhXPSiLuOzzpo18WU-hSQASJr2mQBOY75_v8 2667
graphpatch/extraction/extraction_context.py sha256=yBNMTVPIzb1ZfOWNfkKRfY2ydEbyEXCWNoSU30mXj1o 9652
graphpatch/extraction/extraction_options.py sha256=NTTP2lDA6PLBJLMr1zqBQSOXKHcjDXsIAlnOKKBxfG4 3661
graphpatch/extraction/graph_extraction.py sha256=_Dtnvk77pCAgnieoP0EaPeiDahH_ErN_ct_RjAUsxLw 27015
graphpatch/extraction/graphpatch_module.py sha256=bBi6IEG09AS0nXnZbFqUTV9PaFmh_sRINCd3visusyA 9442
graphpatch/extraction/multiply_invoked_module.py sha256=TetlUU9UZDRTdpAiqmg0RDR1asiEOjktL9hM8fyef18 2590
graphpatch/extraction/opaque_graph_module.py sha256=HEsE8nS_OP9YLvyaNRr2a2flhglNeHPV45NsaGG209k 16106
graphpatch/extraction/wrapped_8_bit_linear.py sha256=tylFrHQWTWBKA2riVBLiXw07Z1J6rRu6P8AyjbjWTao 2836
graphpatch/extraction/wrapped_layer_norm.py sha256=Vv5lce9jIF8EW0f671NRuTtzshI2rxTdW_GcPEvAQ8g 959
graphpatch/hacks.py sha256=R2o-mqATuy59y1A9WNn-VcRbprDK4HnpzszzhAnAoc4 26642
graphpatch/meta/__init__.py sha256=_CcCnOwdz2AFD_W3oEO5JK_rWlf5nQm5OCogbWtw8PQ 684
graphpatch/meta/graph_meta.py sha256=4UuHiSh1njaxDM_QIMLVW_MekPKoluX3dwuxaNkYCzU 18800
graphpatch/meta/node_data.py sha256=86fOrJjK0V4IYn1bMWVvj99oROj83N90PxD0Dh0GDyE 19551
graphpatch/meta/node_path.py sha256=D9z9T3r-Wlz6EjA9iEnFh2hLWi2VFZsZnByBF_aBa-M 4858
graphpatch/optional/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
graphpatch/optional/accelerate.py sha256=PZyYVurIWM3ZpeaC8wr2RtDIWQJ8dKp4SP_O3ZD4Lms 567
graphpatch/optional/bitsandbytes.py sha256=WGnveENq5UCu38097Q9tQtC29d1cfCBL4x1mhrzWlOk 909
graphpatch/optional/dataclasses.py sha256=YbSSecU46wmnv-L_gMgC08oSKDF0AK2O8OS6SIqE8tQ 1780
graphpatch/optional/transformer_lens.py sha256=Yfj2sr9lv3tmLP9rGamK0NbakPcOGVMo-JpH09IT4gs 285
graphpatch/optional/transformers.py sha256=cOB1F6ClmCvY77u9-Et4wloWUTPKq_HWcM-LABQH2p0 1905
graphpatch/optional/typing_extensions.py sha256=qI4ebbSJTlNgVRB0eXkzoQlML4dvpa8cc-7utqAvv24 328
graphpatch/patch.py sha256=WPCDQe9RnczEE9yiEdHogyCyUMXwxlk-BLfvjAz2X5k 7967
graphpatch/patchable_graph.py sha256=-cSe7eaDa8vwkK-F8C4Rorxulz44qyOxmjFp851GQaA 32724
graphpatch-0.2.3.dist-info/LICENSE sha256=zslGVuY0j8IjDx-N2r-6SPcSU6SkBPe9Qi9xKs71EqU 1072
graphpatch-0.2.3.dist-info/METADATA sha256=VdJkczes0ezxITPESLykYdxaEJbNpLW7Cdb81OP4hg0 11768
graphpatch-0.2.3.dist-info/WHEEL sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg 88
graphpatch-0.2.3.dist-info/RECORD