sae-lens 6.20.1__tar.gz → 6.22.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (41) hide show
  1. {sae_lens-6.20.1 → sae_lens-6.22.0}/PKG-INFO +1 -1
  2. {sae_lens-6.20.1 → sae_lens-6.22.0}/pyproject.toml +1 -1
  3. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/__init__.py +1 -1
  4. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/pretrained_saes.yaml +26 -0
  5. {sae_lens-6.20.1 → sae_lens-6.22.0}/LICENSE +0 -0
  6. {sae_lens-6.20.1 → sae_lens-6.22.0}/README.md +0 -0
  7. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/analysis/__init__.py +0 -0
  8. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/analysis/hooked_sae_transformer.py +0 -0
  9. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/analysis/neuronpedia_integration.py +0 -0
  10. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/cache_activations_runner.py +0 -0
  11. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/config.py +0 -0
  12. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/constants.py +0 -0
  13. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/evals.py +0 -0
  14. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/llm_sae_training_runner.py +0 -0
  15. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/load_model.py +0 -0
  16. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/loading/__init__.py +0 -0
  17. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/loading/pretrained_sae_loaders.py +0 -0
  18. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/loading/pretrained_saes_directory.py +0 -0
  19. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/pretokenize_runner.py +0 -0
  20. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/registry.py +0 -0
  21. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/__init__.py +0 -0
  22. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/batchtopk_sae.py +0 -0
  23. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/gated_sae.py +0 -0
  24. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/jumprelu_sae.py +0 -0
  25. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/matryoshka_batchtopk_sae.py +0 -0
  26. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/sae.py +0 -0
  27. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/standard_sae.py +0 -0
  28. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/temporal_sae.py +0 -0
  29. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/topk_sae.py +0 -0
  30. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/saes/transcoder.py +0 -0
  31. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/tokenization_and_batching.py +0 -0
  32. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/training/__init__.py +0 -0
  33. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/training/activation_scaler.py +0 -0
  34. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/training/activations_store.py +0 -0
  35. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/training/mixing_buffer.py +0 -0
  36. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/training/optim.py +0 -0
  37. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/training/sae_trainer.py +0 -0
  38. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/training/types.py +0 -0
  39. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/training/upload_saes_to_huggingface.py +0 -0
  40. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/tutorial/tsea.py +0 -0
  41. {sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/util.py +0 -0
{sae_lens-6.20.1 → sae_lens-6.22.0}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: sae-lens
3
- Version: 6.20.1
3
+ Version: 6.22.0
4
4
  Summary: Training and Analyzing Sparse Autoencoders (SAEs)
5
5
  License: MIT
6
6
  License-File: LICENSE
{sae_lens-6.20.1 → sae_lens-6.22.0}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "sae-lens"
3
- version = "6.20.1"
3
+ version = "6.22.0"
4
4
  description = "Training and Analyzing Sparse Autoencoders (SAEs)"
5
5
  authors = ["Joseph Bloom"]
6
6
  readme = "README.md"
{sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/__init__.py RENAMED
@@ -1,5 +1,5 @@
1
1
  # ruff: noqa: E402
2
- __version__ = "6.20.1"
2
+ __version__ = "6.22.0"
3
3
 
4
4
  import logging
5
5
 
{sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/pretrained_saes.yaml RENAMED
@@ -14916,6 +14916,30 @@ qwen2.5-7b-instruct-andyrdt:
14916
14916
  path: resid_post_layer_27/trainer_1
14917
14917
  neuronpedia: qwen2.5-7b-it/27-resid-post-aa
14918
14918
 
14919
+ gpt-oss-20b-andyrdt:
14920
+ conversion_func: dictionary_learning_1
14921
+ model: openai/gpt-oss-20b
14922
+ repo_id: andyrdt/saes-gpt-oss-20b
14923
+ saes:
14924
+ - id: resid_post_layer_3_trainer_0
14925
+ path: resid_post_layer_3/trainer_0
14926
+ neuronpedia: gpt-oss-20b/3-resid-post-aa
14927
+ - id: resid_post_layer_7_trainer_0
14928
+ path: resid_post_layer_7/trainer_0
14929
+ neuronpedia: gpt-oss-20b/7-resid-post-aa
14930
+ - id: resid_post_layer_11_trainer_0
14931
+ path: resid_post_layer_11/trainer_0
14932
+ neuronpedia: gpt-oss-20b/11-resid-post-aa
14933
+ - id: resid_post_layer_15_trainer_0
14934
+ path: resid_post_layer_15/trainer_0
14935
+ neuronpedia: gpt-oss-20b/15-resid-post-aa
14936
+ - id: resid_post_layer_19_trainer_0
14937
+ path: resid_post_layer_19/trainer_0
14938
+ neuronpedia: gpt-oss-20b/19-resid-post-aa
14939
+ - id: resid_post_layer_23_trainer_0
14940
+ path: resid_post_layer_23/trainer_0
14941
+ neuronpedia: gpt-oss-20b/23-resid-post-aa
14942
+
14919
14943
  goodfire-llama-3.3-70b-instruct:
14920
14944
  conversion_func: goodfire
14921
14945
  model: meta-llama/Llama-3.3-70B-Instruct
@@ -14924,6 +14948,7 @@ goodfire-llama-3.3-70b-instruct:
14924
14948
  - id: layer_50
14925
14949
  path: Llama-3.3-70B-Instruct-SAE-l50.pt
14926
14950
  l0: 121
14951
+ neuronpedia: llama3.3-70b-it/50-resid-post-gf
14927
14952
 
14928
14953
  goodfire-llama-3.1-8b-instruct:
14929
14954
  conversion_func: goodfire
@@ -14933,3 +14958,4 @@ goodfire-llama-3.1-8b-instruct:
14933
14958
  - id: layer_19
14934
14959
  path: Llama-3.1-8B-Instruct-SAE-l19.pth
14935
14960
  l0: 91
14961
+ neuronpedia: llama3.1-8b-it/19-resid-post-gf
{sae_lens-6.20.1 → sae_lens-6.22.0}/LICENSE RENAMED
File without changes
{sae_lens-6.20.1 → sae_lens-6.22.0}/README.md RENAMED
File without changes
{sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/config.py RENAMED
File without changes
{sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/evals.py RENAMED
File without changes
{sae_lens-6.20.1 → sae_lens-6.22.0}/sae_lens/util.py RENAMED
File without changes