Wheelodex » Projects » sparse_autoencoder

sparse_autoencoder

View on PyPI — Reverse Dependencies (0)

1.10.0 sparse_autoencoder-1.10.0-py3-none-any.whl

Wheel Details

Project: sparse_autoencoder
Version: 1.10.0
Filename: sparse_autoencoder-1.10.0-py3-none-any.whl
Download: [link]
Size: 137304
MD5: 1d0f0d98c4a72f9d783823099d025fa0
SHA256: 2013aebe12dcc94a101f9e2ab73789646c8c21870780746ba68e68ac6bf3f8f7
Uploaded: 2024-01-26 13:41:43 +0000

dist-info

METADATA · WHEEL · RECORD · entry_points.txt

METADATA

Metadata-Version: 2.1
Name: sparse_autoencoder
Version: 1.10.0
Summary: Sparse Autoencoder for Mechanistic Interpretability
Author: Alan Cooney
Author-Email: 41682961+alan-cooney[at]users.noreply.github.com
License: MIT
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.10,<3.12
Requires-Dist: datasets (>=2.16.1)
Requires-Dist: einops (>=0.7.0)
Requires-Dist: pydantic (>=2.5.3)
Requires-Dist: strenum (>=0.4.15)
Requires-Dist: tokenizers (>=0.15.1)
Requires-Dist: torch (>=2.1.2)
Requires-Dist: transformers (>=4.37.1)
Requires-Dist: wandb (>=0.16.2)
Requires-Dist: zstandard (>=0.22.0)
Description-Content-Type: text/markdown
[Description omitted; length: 3789 characters]

WHEEL

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

RECORD

Path Digest Size
sparse_autoencoder/__init__.py sha256=RsHi5P7OgkXwtxZozlhtVqLqttXa2Qk_5YaUWPaUAQI 2672
sparse_autoencoder/activation_resampler/__init__.py sha256=8_TDh3ka-gJINK2Y22henByjenPAdyb6S-4YyAevYTs 28
sparse_autoencoder/activation_resampler/activation_resampler.py sha256=tUHXvV1ss-99rgD794lQNfLR3akFmzEybtn7Mpx075c 26204
sparse_autoencoder/activation_resampler/tests/test_activation_resampler.py sha256=buX21WU6vyk-q3IaH84vh0nOUHgOn6Lpm0Y949w2_L0 19707
sparse_autoencoder/activation_resampler/utils/__init__.py sha256=IQS2BspCx6VLTrn-UnW5THvashvTdL9DnIaSxNqBoGk 34
sparse_autoencoder/activation_resampler/utils/component_slice_tensor.py sha256=Y5BNWPhp3iAAw3pWdpQH5d6fwgbKAoPiz2-ap-ELeG0 1705
sparse_autoencoder/activation_store/__init__.py sha256=hm_MR38P0o3k1kJiqYsTCwpYqQq5fZdxx6Da08UC0qE 25
sparse_autoencoder/activation_store/base_store.py sha256=ok7HJ6-9lMeeCLHtbErSPdbpHcKTDAcIAC9LLDw8JDk 4837
sparse_autoencoder/activation_store/tensor_store.py sha256=ASHx6c0FKWBWdApEE3auq80aGVFtX-BzgTeNQqv4Vyo 8531
sparse_autoencoder/activation_store/tests/test_tensor_store.py sha256=5z3gvt1A0bYM9D3y8AKlab_KH-7Bu6w6xvUlwMYVr8c 1967
sparse_autoencoder/autoencoder/__init__.py sha256=B5mqdbIZlcXfQK6vZ6MGAM0a18zyVEngt6WgiQEUx0w 45
sparse_autoencoder/autoencoder/components/__init__.py sha256=m0VkyJQ5hoEnLYBAsMwJ9oOxVD7kekWauED_8jIZWRk 37
sparse_autoencoder/autoencoder/components/linear_encoder.py sha256=sfryNrXeLp038QZM42PAZU2G9msiK2OVqT7zMjtYFpE 7807
sparse_autoencoder/autoencoder/components/tests/__snapshots__/test_linear_encoder.ambr sha256=CdWlBXDev__MTwJ05nEkCOU6MlmnkXapyxtmwTwLraU 1137
sparse_autoencoder/autoencoder/components/tests/test_compare_neel_implementation.py sha256=b0xO7bk4ajpX4Y3GEElWIHNJkcIfuZ1PWlRbDL4abZA 7990
sparse_autoencoder/autoencoder/components/tests/test_linear_encoder.py sha256=Om-J3ijh7ef5oKwTZP6k1L1OxHq6PgXpKQwaA9LtEIE 5329
sparse_autoencoder/autoencoder/components/tests/test_tied_bias.py sha256=KUXOc-MAhdbTqYCeMj8U9CBTmPLBUOUbxZ3-nP9pwSQ 1219
sparse_autoencoder/autoencoder/components/tests/test_unit_norm_decoder.py sha256=iaaHTxzSVHbLdFSHlE-oadWPDWl7F4GICXT3gG5zOOg 6516
sparse_autoencoder/autoencoder/components/tied_bias.py sha256=eTXqrqHuJ3w4qGR6CZtv8LTveu1fQUVqism3AoIMds8 2708
sparse_autoencoder/autoencoder/components/unit_norm_decoder.py sha256=TGbTVzHmoyTobmUnejZi0RvPQbGdqpkJerisA4Fr7sM 13069
sparse_autoencoder/autoencoder/model.py sha256=9ceCyjzFfFvO2_gMUVLrg_7-smTe2MzNbHmu8J3ODqc 14766
sparse_autoencoder/autoencoder/tests/__snapshots__/test_model.ambr sha256=7SIyQ02IdsBhBsPOpF4zXy8LP8WqAKxMpL4pE7jfAk0 429
sparse_autoencoder/autoencoder/tests/test_model.py sha256=EgyQ8Gbx9FuTD0J_-rEqPqjZIH1_KH1KQoMYtia4A_8 7690
sparse_autoencoder/autoencoder/types.py sha256=OE8UfmsRMxPZAh84YRz6GwD0bzsD7q128FcjfRD31WY 427
sparse_autoencoder/loss/__init__.py sha256=psj8kGh8E9gZcC1roscQGPvlEAqbyPJdYW57YoGsf7M 387
sparse_autoencoder/loss/abstract_loss.py sha256=z34JwZ9KcGVPnuRSTiPFp51Ynf3bpNSvYlOgYZ4rXkU 8517
sparse_autoencoder/loss/decoded_activations_l2.py sha256=ReuHoDFtOCiAjCvcTV3oxRdOLW2E7vKiOoAJuQAi6EA 3234
sparse_autoencoder/loss/learned_activations_l1.py sha256=Slh6JkkwpBpeXgl8qWagZ4MauvDsTetzOTl76I3WMEE 8809
sparse_autoencoder/loss/reducer.py sha256=_cxQLnoTrPR_4EhRcV3iqO9eK90749Arqdd6-RIph80 3448
sparse_autoencoder/loss/tests/__snapshots__/test_decoded_activations_l2.ambr sha256=6ZhJsrScEHCgNb9jvgTSe126LqKUAGQWX-rbgBTG7F0 293
sparse_autoencoder/loss/tests/test_abstract_loss.py sha256=69dkV2YETEV07kaZNhVxCvC8ov2Jik9PCBBvhFY8QWQ 4035
sparse_autoencoder/loss/tests/test_decoded_activations_l2.py sha256=Gky0eWpJXKkXI09Ukw4ic8Q207BFEO45VYiu3Eb3kqQ 3227
sparse_autoencoder/loss/tests/test_learned_activations_l1.py sha256=kK6X24aRSSXJidNBXcx_IqVrcCwWEpaIvRfcx_z_qqg 4650
sparse_autoencoder/loss/tests/test_neel_loss.py sha256=b0UxYoOBwtFy-VzuBF053jpbz8tsO6cipI68YIX_4N0 5899
sparse_autoencoder/loss/tests/test_towards_monosemanticity_loss.py sha256=PsUsf6vDVI3OeEdX_kCqTk1_l-Zap4yhQCW0MbFfk5U 1572
sparse_autoencoder/metrics/__init__.py sha256=wHI9Vd3O4YdwCzO-wQYkPwl9BY_SoEMg0_DduoU1Xeo 15
sparse_autoencoder/metrics/abstract_metric.py sha256=rl-n_eQG5KIAawsjObD4BW862Q8TDsOqipzPb0kUwyc 12417
sparse_autoencoder/metrics/metrics_container.py sha256=12c4cbfKPuS_4rf4GtWbtXKu6sw_w4fAW2CsXsAXJcc 1416
sparse_autoencoder/metrics/train/__init__.py sha256=kxQrlUc6LOYRvOyocZOvpkZNqpMoHbx0vIsWhLK_j8U 149
sparse_autoencoder/metrics/train/abstract_train_metric.py sha256=haqRXn55VvQk7driySWPr4x_LUpdcd9X53QWfysoJCU 2396
sparse_autoencoder/metrics/train/capacity.py sha256=sq5XG81rrTs-beeO0pC6sdT-F3aRADqeXK1YekV3Kv4 3923
sparse_autoencoder/metrics/train/feature_density.py sha256=1Bw9uWJ3bFCm3kVFsUYf3B5Y9Rsa_6BdmeH_OOxhCR4 4762
sparse_autoencoder/metrics/train/l0_norm_metric.py sha256=XJF8e4DJlwno-oEV0Z3kFMzqwrwj1azCDnvwq6GD08c 1489
sparse_autoencoder/metrics/train/neuron_activity_metric.py sha256=vVk4OTOXkbyYLnG772nUBbSAHZ-1Hx5J5iJRyr1VS_8 10053
sparse_autoencoder/metrics/train/tests/__snapshots__/test_capacities.ambr sha256=Ebv7WEHTqH6gkStND7wPNXnOAl3wyw_L-g_-iGnslMM 5932
sparse_autoencoder/metrics/train/tests/__snapshots__/test_feature_density.ambr sha256=PGbzGMw7FwTthLHVVrYdRBbwpzAmqxrYAkb6tnfu9_I 13810
sparse_autoencoder/metrics/train/tests/__snapshots__/test_l0_norm_metric.ambr sha256=m8iUfmKTPohyLiu5vWGlmBe9HMleI3aW8cMqVNAyxxI 584
sparse_autoencoder/metrics/train/tests/__snapshots__/test_neuron_activity_metric.ambr sha256=gLTMEhSP1v8WhrYtGtq2vAN0S7Go_vyPlzuJ6PyRIhc 9872
sparse_autoencoder/metrics/train/tests/test_abstract_train_metric.py sha256=pOmyyk7WAmohKI2FFXBGWvEl-15pHlnIQ4IAdvcTQ-k 1763
sparse_autoencoder/metrics/train/tests/test_capacities.py sha256=-TKfZt7w4n1RnI9OTR1GYCdgM3cXj77MXT5f1cZzHm8 3245
sparse_autoencoder/metrics/train/tests/test_feature_density.py sha256=4_wxX5VcmGPftyEi3ZY4tDDc3L26fAhGcFjZma5Qezg 2851
sparse_autoencoder/metrics/train/tests/test_l0_norm_metric.py sha256=doQSiIveZ2QlIwV3XsS3YWagdp_eL5nXfnosU7oB0h0 1799
sparse_autoencoder/metrics/train/tests/test_neuron_activity_metric.py sha256=Q-9UDbxhQxQbpUvKs0vOAmHf9dLijroCKvNn4OW3XPw 8481
sparse_autoencoder/metrics/utils/__init__.py sha256=AzDmaoqQbHAAV2m1mBXD1-vhmJVBPWaUuQopg--emg8 20
sparse_autoencoder/metrics/utils/add_component_axis_if_missing.py sha256=PK9zbZLIbfvO_MuNSMXxoHZXxmbOWzxfYL-7delIfc8 1530
sparse_autoencoder/metrics/utils/find_metric_result.py sha256=Dw9ltMObxNlmdRmX7qLn6qyzaFrXLK-AJUoL9AlK4rg 2563
sparse_autoencoder/metrics/validate/__init__.py sha256=xVEdjUSWT7bk9sFpkc9jMPZPUybzLpLgvMcsKVKvz8w 168
sparse_autoencoder/metrics/validate/abstract_validate_metric.py sha256=wFkLSdy-a9hWtZmAQ9OzJCgHcOmx0ccJE5xwAHtKq5s 2235
sparse_autoencoder/metrics/validate/model_reconstruction_score.py sha256=3tqa3sPmpg0Vs_jUosrqeAVXIDro4f0CjEfTJ1GHDDg 3921
sparse_autoencoder/metrics/validate/tests/__snapshots__/test_model_reconstruction_score.ambr sha256=rETqRIEgtCnux4iaUqxFaDaRZE-iI1wI3KRzWV96uJM 2502
sparse_autoencoder/metrics/validate/tests/test_model_reconstruction_score.py sha256=7adJ4A__zjHzZhv5Yb8zJiqJwU3e9nOmbk8OhRc3ZSs 3261
sparse_autoencoder/optimizer/__init__.py sha256=ZYQyzh9zoBF888FBtkFWdNnqPkofE-ix0qLAjY1UAbw 826
sparse_autoencoder/optimizer/abstract_optimizer.py sha256=imd6IEiuDgvXVQuGs4BJrxdarlFVoJhIMTSgJRF7cZc 1636
sparse_autoencoder/optimizer/adam_with_reset.py sha256=CciYN05RZ_ksgnKi01a9sHU5_cDq3EerscFVxDZ-mU4 11100
sparse_autoencoder/optimizer/tests/test_adam_with_reset.py sha256=9ss-JIFMXK6Fgyqp77gOkE0acIZlIVRxgOGibiFY7Eg 3356
sparse_autoencoder/source_data/__init__.py sha256=wietwzMKjV66bLKLEqBaAMME7O25fqvQUNtY1UvRzjU 19
sparse_autoencoder/source_data/abstract_dataset.py sha256=7--5gPmo2wL-DukbvnHQtJ64UUTmYpltoTY_t8goJ3A 9513
sparse_autoencoder/source_data/mock_dataset.py sha256=tQg06HWs4I3NYgKi8qVLquTS6CGEI5M7WHxVTlHxSmE 5911
sparse_autoencoder/source_data/pretokenized_dataset.py sha256=GR-4EtakgfyY38zh3X5QKcYrg3s_lgn3hY_LBa0htWs 4647
sparse_autoencoder/source_data/tests/test_abstract_dataset.py sha256=0K6XddyHmG5v6W0Ezsc6uvHy-IyXAa2y6lpntvlQdJs 1218
sparse_autoencoder/source_data/tests/test_mock_dataset.py sha256=4ePb3YSphJujC3M1zKdKd7DvTlhhHvWW2_TlwHvC6vk 2876
sparse_autoencoder/source_data/tests/test_pretokenized_dataset.py sha256=O-cyBRfIqBwUyCbHUSk3P8KgYyZ5PRLun9LmwdMh2nk 1196
sparse_autoencoder/source_data/tests/test_text_dataset.py sha256=WcZ6oMXGIsadzDv-Rj0FDdRa6Zgdw74enBAkqUrZky0 1550
sparse_autoencoder/source_data/text_dataset.py sha256=a7zCTpNsTur-6fRbYLKBGDzEuX-AAnIdHkdDTL_JE3w 7177
sparse_autoencoder/source_model/__init__.py sha256=CNvkbOOfuFuNvDd27Hpk6YgfJIF6mKWQNYhfaHfIEck 20
sparse_autoencoder/source_model/replace_activations_hook.py sha256=230yyVHPs6lIOCElswUGfzvwskzxGygoIPLX6sVobvU 2705
sparse_autoencoder/source_model/reshape_activations.py sha256=7MFz-9paLR2y1XbGtk2WgfZOIOV6dbMoSCZUcm1lnBM 3222
sparse_autoencoder/source_model/store_activations_hook.py sha256=sfZaKVaj8n6QNf1Rs_liR51vPxelcyMmGb-sd_qcWio 2789
sparse_autoencoder/source_model/tests/test_replace_activations_hook.py sha256=9uEbOuauAj5ymPNT3bZYi-AN2fCVNf97iUvE8CvOvyY 2791
sparse_autoencoder/source_model/tests/test_store_activations_hook.py sha256=FtxHjNpZuX_ZB0qjhup0JkbuLZyddFEdOiuuIW1rJTI 1243
sparse_autoencoder/source_model/tests/test_zero_ablate_hook.py sha256=YcnLM--4CWM01rnMyReO0nOfD709w2ABZey-nhLuX9A 1810
sparse_autoencoder/source_model/zero_ablate_hook.py sha256=QkeAQsF1JNpmPwdQhRAbBKAh4hHd8RckC4FScU9PCs0 609
sparse_autoencoder/tensor_types.py sha256=cUEwlWQtQuECMqDJ5da-64LfRksyBuQtZMCgbEyu0GY 2450
sparse_autoencoder/train/__init__.py sha256=eUu_Mde8WORnhx6NRgQpbZ2W9NzHtep9O6YLH9xFtXc 13
sparse_autoencoder/train/join_sweep.py sha256=GKvBX0d6r-4_YgJ_nEMTT92hWbglxUijFa1lq8PMjh4 688
sparse_autoencoder/train/pipeline.py sha256=fSIxgMBOiMKkEQKOGEH3mfOZfsFI1Qbv9WBysBtxE4A 21660
sparse_autoencoder/train/sweep.py sha256=QnfTCjq8WZmF6nioyvGq_upcLt-6LBdXKaKL3r3eHnw 12855
sparse_autoencoder/train/sweep_config.py sha256=_CEkS7BSfC9pGLxc8c8eenBADqXh2ypcTympYyNj4Y4 12383
sparse_autoencoder/train/tests/__snapshots__/test_sweep.ambr sha256=4905SC3qEkBM7xTfWvYeAMo9TcWwyRa64915nseGXBY 833
sparse_autoencoder/train/tests/test_pipeline.py sha256=5_11IqTMbX5Q5NUNlFc-puUbrj0wHjm6pHomBtdfRjc 17146
sparse_autoencoder/train/tests/test_sweep.py sha256=yAnCUEySAyaL9bYoZf0dPL9vQFWRKW4M1N7Lyl0GHDs 3100
sparse_autoencoder/train/utils/__init__.py sha256=GCA0_-zhndiDfPoBAYH_P-JUeQlif5e7HW2nfyVKSHo 19
sparse_autoencoder/train/utils/get_model_device.py sha256=3cD28g8lnglseziShYWtTes-iBZzXxO9S7gw8WrgCu4 862
sparse_autoencoder/train/utils/round_down.py sha256=_jWdo_D9LfIneGyY3aRyc0-BMDsdTEPgLaRZQYT6zCA 756
sparse_autoencoder/train/utils/tests/test_get_model_device.py sha256=qkFMVYUEsrG7Xgaf8drOwSlno1QNQZeMaoEfX1In2hE 1130
sparse_autoencoder/train/utils/tests/test_wandb_sweep_types.py sha256=ZJ-98NBqqjw1iz6-KzHBjHoJhxl-JPc0u3x-u8IAT7I 1720
sparse_autoencoder/train/utils/wandb_sweep_types.py sha256=C7XyDexvqEl9CvCYdyT19xqW4YbvaVsRNOFrK9tpSFc 13390
sparse_autoencoder/training_runs/__init__.py sha256=w-L9BIHA6YMNcRU2qO1IvtG1eMbufUHgNJs92b_Fr_o 21
sparse_autoencoder/training_runs/gpt2.py sha256=czZ8vH-DgR65VvH9pbuWv5EusR0tDUzxlNAsgPTVcmU 2299
sparse_autoencoder/utils/__init__.py sha256=LIdR_iCizJH5_znPF0m6bjU81G8HuzUN0503Pvh0tWE 20
sparse_autoencoder/utils/tensor_shape.py sha256=hJmq9bmnDoGuXTZDr_bzQhoqGzSyam7eEMdGmQHzSmk 863
sparse_autoencoder-1.10.0.dist-info/LICENSE sha256=8LZ3OhIJImJM6ZJ9EuXKRstpLZmTEZ3qUiRZLTu7AxY 1068
sparse_autoencoder-1.10.0.dist-info/METADATA sha256=eM1lifbFpN5wsQbtpOjGI1xGCiMNY5c61vVS8gLZ_lg 4588
sparse_autoencoder-1.10.0.dist-info/WHEEL sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4 88
sparse_autoencoder-1.10.0.dist-info/entry_points.txt sha256=PljBNIKSxaPFE5m92j7rpk40-MYOyTaSGyX6Nu_hUMU 74
sparse_autoencoder-1.10.0.dist-info/RECORD — —

entry_points.txt

[console_scripts]
join-sae-sweep = sparse_autoencoder.train.join_sweep:run