mi-crow 0.1.1.post12__py3-none-any.whl → 0.1.1.post14__py3-none-any.whl

{amber → mi_crow}/__init__.py

@@ -1,4 +1,4 @@
-"""Amber: helper package for the Engineer Thesis project.
+"""mi_crow: helper package for the Engineer Thesis project.
 
 This module is intentionally minimal. It exists to define the top-level package
 and to enable code coverage to include the package. Importing it should succeed
@@ -6,7 +6,7 @@ without side effects.
 """
 
 # A tiny bit of executable code to make the package measurable by coverage.
-PACKAGE_NAME = "amber"
+PACKAGE_NAME = "mi_crow"
 __version__ = "0.0.0"
 
 

mi_crow/datasets/__init__.py

@@ -0,0 +1,11 @@
+from mi_crow.datasets.base_dataset import BaseDataset
+from mi_crow.datasets.text_dataset import TextDataset
+from mi_crow.datasets.classification_dataset import ClassificationDataset
+from mi_crow.datasets.loading_strategy import LoadingStrategy
+
+__all__ = [
+    "BaseDataset",
+    "TextDataset",
+    "ClassificationDataset",
+    "LoadingStrategy",
+]

{amber → mi_crow}/datasets/base_dataset.py

@@ -9,8 +9,8 @@ from typing import Any, Dict, Iterator, List, Optional, Union
 
 from datasets import Dataset, IterableDataset, load_dataset, load_from_disk
 
-from amber.datasets.loading_strategy import IndexLike, LoadingStrategy
-from amber.store.store import Store
+from mi_crow.datasets.loading_strategy import IndexLike, LoadingStrategy
+from mi_crow.store.store import Store
 
 
 class BaseDataset(ABC):

{amber → mi_crow}/datasets/classification_dataset.py

@@ -5,9 +5,9 @@ from typing import Any, Dict, Iterator, List, Optional, Sequence, Union
 
 from datasets import Dataset, IterableDataset, load_dataset
 
-from amber.datasets.base_dataset import BaseDataset
-from amber.datasets.loading_strategy import IndexLike, LoadingStrategy
-from amber.store.store import Store
+from mi_crow.datasets.base_dataset import BaseDataset
+from mi_crow.datasets.loading_strategy import IndexLike, LoadingStrategy
+from mi_crow.store.store import Store
 
 
 class ClassificationDataset(BaseDataset):

{amber → mi_crow}/datasets/text_dataset.py

@@ -5,9 +5,9 @@ from typing import Any, Dict, Iterator, List, Optional, Sequence, Union
 
 from datasets import Dataset, IterableDataset, load_dataset
 
-from amber.datasets.base_dataset import BaseDataset
-from amber.datasets.loading_strategy import IndexLike, LoadingStrategy
-from amber.store.store import Store
+from mi_crow.datasets.base_dataset import BaseDataset
+from mi_crow.datasets.loading_strategy import IndexLike, LoadingStrategy
+from mi_crow.store.store import Store
 
 
 class TextDataset(BaseDataset):

mi_crow/hooks/__init__.py

@@ -0,0 +1,20 @@
+from mi_crow.hooks.hook import Hook, HookType, HookError
+from mi_crow.hooks.detector import Detector
+from mi_crow.hooks.controller import Controller
+from mi_crow.hooks.implementations.layer_activation_detector import LayerActivationDetector
+from mi_crow.hooks.implementations.model_input_detector import ModelInputDetector
+from mi_crow.hooks.implementations.model_output_detector import ModelOutputDetector
+from mi_crow.hooks.implementations.function_controller import FunctionController
+
+__all__ = [
+    "Hook",
+    "HookType",
+    "HookError",
+    "Detector",
+    "Controller",
+    "LayerActivationDetector",
+    "ModelInputDetector",
+    "ModelOutputDetector",
+    "FunctionController",
+]
+

{amber → mi_crow}/hooks/controller.py

@@ -6,9 +6,9 @@ from typing import TYPE_CHECKING
 import torch
 import torch.nn as nn
 
-from amber.hooks.hook import Hook, HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
-from amber.hooks.utils import extract_tensor_from_input, extract_tensor_from_output
-from amber.utils import get_logger
+from mi_crow.hooks.hook import Hook, HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.hooks.utils import extract_tensor_from_input, extract_tensor_from_output
+from mi_crow.utils import get_logger
 
 if TYPE_CHECKING:
     pass

{amber → mi_crow}/hooks/detector.py

@@ -5,8 +5,8 @@ from typing import Any, TYPE_CHECKING, Dict
 
 import torch
 
-from amber.hooks.hook import Hook, HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
-from amber.store.store import Store
+from mi_crow.hooks.hook import Hook, HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.store.store import Store
 
 if TYPE_CHECKING:
     pass

{amber → mi_crow}/hooks/hook.py

@@ -10,7 +10,7 @@ from torch import nn, Tensor
 from torch.types import _TensorOrTensors
 
 if TYPE_CHECKING:
-    from amber.language_model.context import LanguageModelContext
+    from mi_crow.language_model.context import LanguageModelContext
 
 
 class HookType(str, Enum):

{amber → mi_crow}/hooks/implementations/function_controller.py

@@ -3,8 +3,8 @@ from __future__ import annotations
 from typing import Callable, TYPE_CHECKING
 import torch
 
-from amber.hooks.controller import Controller
-from amber.hooks.hook import HookType
+from mi_crow.hooks.controller import Controller
+from mi_crow.hooks.hook import HookType
 
 if TYPE_CHECKING:
     from torch import nn

{amber → mi_crow}/hooks/implementations/layer_activation_detector.py

@@ -3,9 +3,9 @@ from __future__ import annotations
 from typing import TYPE_CHECKING
 import torch
 
-from amber.hooks.detector import Detector
-from amber.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
-from amber.hooks.utils import extract_tensor_from_output
+from mi_crow.hooks.detector import Detector
+from mi_crow.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.hooks.utils import extract_tensor_from_output
 
 if TYPE_CHECKING:
     from torch import nn

{amber → mi_crow}/hooks/implementations/model_input_detector.py

@@ -3,8 +3,8 @@ from __future__ import annotations
 from typing import TYPE_CHECKING, Dict, Set, List, Optional
 import torch
 
-from amber.hooks.detector import Detector
-from amber.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.hooks.detector import Detector
+from mi_crow.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
 
 if TYPE_CHECKING:
     from torch import nn

{amber → mi_crow}/hooks/implementations/model_output_detector.py

@@ -3,8 +3,8 @@ from __future__ import annotations
 from typing import TYPE_CHECKING
 import torch
 
-from amber.hooks.detector import Detector
-from amber.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.hooks.detector import Detector
+from mi_crow.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
 
 if TYPE_CHECKING:
     from torch import nn

{amber → mi_crow}/hooks/utils.py

@@ -6,7 +6,7 @@ from typing import Any
 
 import torch
 
-from amber.hooks.hook import HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.hooks.hook import HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
 
 
 def extract_tensor_from_input(input: HOOK_FUNCTION_INPUT) -> torch.Tensor | None:

{amber → mi_crow}/language_model/activations.py

@@ -4,16 +4,16 @@ from typing import TYPE_CHECKING, Any, Dict, Sequence
 import torch
 from torch import nn
 
-from amber.datasets import BaseDataset
-from amber.hooks import HookType
-from amber.hooks.implementations.layer_activation_detector import LayerActivationDetector
-from amber.hooks.implementations.model_input_detector import ModelInputDetector
-from amber.store.store import Store
-from amber.utils import get_logger
-from amber.language_model.utils import get_device_from_model
+from mi_crow.datasets import BaseDataset
+from mi_crow.hooks import HookType
+from mi_crow.hooks.implementations.layer_activation_detector import LayerActivationDetector
+from mi_crow.hooks.implementations.model_input_detector import ModelInputDetector
+from mi_crow.store.store import Store
+from mi_crow.utils import get_logger
+from mi_crow.language_model.utils import get_device_from_model
 
 if TYPE_CHECKING:
-    from amber.language_model.context import LanguageModelContext
+    from mi_crow.language_model.context import LanguageModelContext
 
 logger = get_logger(__name__)
 
@@ -170,7 +170,7 @@ class LanguageModelActivations:
             meta: Metadata dictionary
             verbose: Whether to log
         """
-        from amber.language_model.inference import InferenceEngine
+        from mi_crow.language_model.inference import InferenceEngine
         InferenceEngine._save_run_metadata(store, run_name, meta, verbose)
 
     def _process_batch(

{amber → mi_crow}/language_model/context.py

@@ -2,11 +2,11 @@ from dataclasses import dataclass, field
 from typing import Optional, Dict, Any, TYPE_CHECKING, List, Set
 
 if TYPE_CHECKING:
-    from amber.language_model.language_model import LanguageModel
+    from mi_crow.language_model.language_model import LanguageModel
     from torch import nn
     from transformers import PreTrainedTokenizerBase
-    from amber.hooks.hook import Hook
-    from amber.store.store import Store
+    from mi_crow.hooks.hook import Hook
+    from mi_crow.store.store import Store
 
 
 @dataclass

{amber → mi_crow}/language_model/hook_metadata.py

@@ -6,7 +6,7 @@ from collections import defaultdict
 from typing import Dict, List, Any, TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from amber.language_model.context import LanguageModelContext
+    from mi_crow.language_model.context import LanguageModelContext
 
 
 def collect_hooks_metadata(context: "LanguageModelContext") -> Dict[str, List[Dict[str, Any]]]:

{amber → mi_crow}/language_model/inference.py

@@ -8,14 +8,14 @@ from typing import Sequence, Any, Dict, List, TYPE_CHECKING
 import torch
 from torch import nn
 
-from amber.language_model.utils import get_device_from_model, move_tensors_to_device, extract_logits_from_output
-from amber.utils import get_logger
+from mi_crow.language_model.utils import get_device_from_model, move_tensors_to_device, extract_logits_from_output
+from mi_crow.utils import get_logger
 
 if TYPE_CHECKING:
-    from amber.language_model.language_model import LanguageModel
-    from amber.hooks.controller import Controller
-    from amber.datasets import BaseDataset
-    from amber.store.store import Store
+    from mi_crow.language_model.language_model import LanguageModel
+    from mi_crow.hooks.controller import Controller
+    from mi_crow.datasets import BaseDataset
+    from mi_crow.store.store import Store
 
 logger = get_logger(__name__)
 
@@ -80,7 +80,7 @@ class InferenceEngine:
         Args:
             enc: Encoded inputs dictionary
         """
-        from amber.hooks.implementations.model_input_detector import ModelInputDetector
+        from mi_crow.hooks.implementations.model_input_detector import ModelInputDetector
         
         detectors = self.lm.layers.get_detectors()
         for detector in detectors:

{amber → mi_crow}/language_model/initialization.py

@@ -9,11 +9,11 @@ import torch
 from torch import nn
 from transformers import AutoTokenizer, AutoModelForCausalLM, PreTrainedTokenizerBase
 
-from amber.store.store import Store
-from amber.language_model.utils import extract_model_id
+from mi_crow.store.store import Store
+from mi_crow.language_model.utils import extract_model_id
 
 if TYPE_CHECKING:
-    from amber.language_model.language_model import LanguageModel
+    from mi_crow.language_model.language_model import LanguageModel
 
 
 def initialize_model_id(

{amber → mi_crow}/language_model/language_model.py

@@ -8,18 +8,18 @@ import torch
 from torch import nn, Tensor
 from transformers import PreTrainedTokenizerBase
 
-from amber.language_model.layers import LanguageModelLayers
-from amber.language_model.tokenizer import LanguageModelTokenizer
-from amber.language_model.activations import LanguageModelActivations
-from amber.language_model.context import LanguageModelContext
-from amber.language_model.inference import InferenceEngine
-from amber.language_model.persistence import save_model, load_model_from_saved_file
-from amber.language_model.initialization import initialize_model_id, create_from_huggingface, create_from_local_torch
-from amber.store.store import Store
-from amber.utils import get_logger
+from mi_crow.language_model.layers import LanguageModelLayers
+from mi_crow.language_model.tokenizer import LanguageModelTokenizer
+from mi_crow.language_model.activations import LanguageModelActivations
+from mi_crow.language_model.context import LanguageModelContext
+from mi_crow.language_model.inference import InferenceEngine
+from mi_crow.language_model.persistence import save_model, load_model_from_saved_file
+from mi_crow.language_model.initialization import initialize_model_id, create_from_huggingface, create_from_local_torch
+from mi_crow.store.store import Store
+from mi_crow.utils import get_logger
 
 if TYPE_CHECKING:
-    from amber.mechanistic.sae.concepts.input_tracker import InputTracker
+    from mi_crow.mechanistic.sae.concepts.input_tracker import InputTracker
 
 logger = get_logger(__name__)
 
@@ -308,7 +308,7 @@ class LanguageModel:
         if self._input_tracker is not None:
             return self._input_tracker
 
-        from amber.mechanistic.sae.concepts.input_tracker import InputTracker
+        from mi_crow.mechanistic.sae.concepts.input_tracker import InputTracker
 
         self._input_tracker = InputTracker(language_model=self)
 

{amber → mi_crow}/language_model/layers.py

@@ -2,12 +2,12 @@ from typing import Dict, List, Callable, TYPE_CHECKING
 
 from torch import nn
 
-from amber.hooks.hook import Hook, HookType
-from amber.hooks.detector import Detector
-from amber.hooks.controller import Controller
+from mi_crow.hooks.hook import Hook, HookType
+from mi_crow.hooks.detector import Detector
+from mi_crow.hooks.controller import Controller
 
 if TYPE_CHECKING:
-    from amber.language_model.context import LanguageModelContext
+    from mi_crow.language_model.context import LanguageModelContext
 
 
 class LanguageModelLayers:

{amber → mi_crow}/language_model/persistence.py

@@ -9,13 +9,13 @@ from typing import TYPE_CHECKING
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-from amber.language_model.contracts import ModelMetadata
-from amber.language_model.hook_metadata import collect_hooks_metadata
-from amber.language_model.utils import extract_model_id
+from mi_crow.language_model.contracts import ModelMetadata
+from mi_crow.language_model.hook_metadata import collect_hooks_metadata
+from mi_crow.language_model.utils import extract_model_id
 
 if TYPE_CHECKING:
-    from amber.language_model.language_model import LanguageModel
-    from amber.store.store import Store
+    from mi_crow.language_model.language_model import LanguageModel
+    from mi_crow.store.store import Store
 
 
 def save_model(
@@ -78,7 +78,7 @@ def save_model(
             f"Failed to save model to {save_path}. Error: {e}"
         ) from e
     
-    from amber.utils import get_logger
+    from mi_crow.utils import get_logger
     logger = get_logger(__name__)
     logger.info(f"Saved model to {save_path}")
     
@@ -169,7 +169,7 @@ def load_model_from_saved_file(
     # Note: Hooks are not automatically restored as they require hook instances
     # The hook metadata is available in metadata_dict["hooks"] if needed
     
-    from amber.utils import get_logger
+    from mi_crow.utils import get_logger
     logger = get_logger(__name__)
     logger.info(f"Loaded model from {saved_path} (model_id: {model_id})")
     

{amber → mi_crow}/language_model/tokenizer.py

@@ -4,7 +4,7 @@ from torch import nn
 from transformers import AutoTokenizer
 
 if TYPE_CHECKING:
-    from amber.language_model.context import LanguageModelContext
+    from mi_crow.language_model.context import LanguageModelContext
 
 
 class LanguageModelTokenizer:

{amber → mi_crow}/mechanistic/sae/autoencoder_context.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from typing import Optional, TYPE_CHECKING
 
-from amber.store.store import Store
+from mi_crow.store.store import Store
 
 if TYPE_CHECKING:
     pass

{amber → mi_crow}/mechanistic/sae/concepts/autoencoder_concepts.py

@@ -9,12 +9,12 @@ import heapq
 import torch
 from torch import nn
 
-from amber.mechanistic.sae.concepts.concept_models import NeuronText
-from amber.mechanistic.sae.autoencoder_context import AutoencoderContext
-from amber.utils import get_logger
+from mi_crow.mechanistic.sae.concepts.concept_models import NeuronText
+from mi_crow.mechanistic.sae.autoencoder_context import AutoencoderContext
+from mi_crow.utils import get_logger
 
 if TYPE_CHECKING:
-    from amber.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
+    from mi_crow.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
 
 logger = get_logger(__name__)
 
@@ -59,12 +59,12 @@ class AutoencoderConcepts:
 
     def _ensure_dictionary(self):
         if self.dictionary is None:
-            from amber.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
+            from mi_crow.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
             self.dictionary = ConceptDictionary(self._n_size)
         return self.dictionary
 
     def load_concepts_from_csv(self, csv_filepath: str | Path):
-        from amber.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
+        from mi_crow.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
         self.dictionary = ConceptDictionary.from_csv(
             csv_filepath=csv_filepath,
             n_size=self._n_size,
@@ -72,7 +72,7 @@ class AutoencoderConcepts:
         )
 
     def load_concepts_from_json(self, json_filepath: str | Path):
-        from amber.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
+        from mi_crow.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
         self.dictionary = ConceptDictionary.from_json(
             json_filepath=json_filepath,
             n_size=self._n_size,
@@ -84,7 +84,7 @@ class AutoencoderConcepts:
         if self._top_texts_heaps is None:
             raise ValueError("No top texts available. Enable text tracking and run inference first.")
 
-        from amber.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
+        from mi_crow.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
         neuron_texts = self.get_all_top_texts()
 
         self.dictionary = ConceptDictionary.from_llm(

{amber → mi_crow}/mechanistic/sae/concepts/concept_dictionary.py

@@ -6,7 +6,7 @@ from typing import Dict, Sequence, TYPE_CHECKING, Optional
 import json
 import csv
 
-from amber.store.store import Store
+from mi_crow.store.store import Store
 
 if TYPE_CHECKING:
     pass

{amber → mi_crow}/mechanistic/sae/concepts/input_tracker.py

@@ -1,9 +1,9 @@
 from typing import TYPE_CHECKING, Sequence
 
-from amber.utils import get_logger
+from mi_crow.utils import get_logger
 
 if TYPE_CHECKING:
-    from amber.language_model.language_model import LanguageModel
+    from mi_crow.language_model.language_model import LanguageModel
 
 logger = get_logger(__name__)
 

mi_crow/mechanistic/sae/modules/__init__.py

@@ -0,0 +1,5 @@
+from mi_crow.mechanistic.sae.modules.topk_sae import TopKSae, TopKSaeTrainingConfig
+from mi_crow.mechanistic.sae.modules.l1_sae import L1Sae, L1SaeTrainingConfig
+
+__all__ = ["TopKSae", "TopKSaeTrainingConfig", "L1Sae", "L1SaeTrainingConfig"]
+

{amber → mi_crow}/mechanistic/sae/modules/l1_sae.py

@@ -3,11 +3,11 @@ from typing import Any
 
 import torch
 from overcomplete import SAE as OvercompleteSAE
-from amber.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
-from amber.mechanistic.sae.sae import Sae
-from amber.mechanistic.sae.sae_trainer import SaeTrainingConfig
-from amber.store.store import Store
-from amber.utils import get_logger
+from mi_crow.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.mechanistic.sae.sae import Sae
+from mi_crow.mechanistic.sae.sae_trainer import SaeTrainingConfig
+from mi_crow.store.store import Store
+from mi_crow.utils import get_logger
 
 logger = get_logger(__name__)
 
@@ -293,7 +293,7 @@ class L1Sae(Sae):
         # Save overcomplete model state dict
         sae_state_dict = self.sae_engine.state_dict()
 
-        amber_metadata = {
+        mi_crow_metadata = {
             "concepts_state": {
                 'multiplication': self.concepts.multiplication.data,
                 'bias': self.concepts.bias.data,
@@ -307,7 +307,7 @@ class L1Sae(Sae):
 
         payload = {
             "sae_state_dict": sae_state_dict,
-            "amber_metadata": amber_metadata,
+            "mi_crow_metadata": mi_crow_metadata,
         }
 
         torch.save(payload, save_path)
@@ -336,16 +336,16 @@ class L1Sae(Sae):
         payload = torch.load(p, map_location=map_location)
 
         # Extract our metadata
-        if "amber_metadata" not in payload:
-            raise ValueError(f"Invalid L1SAE save format: missing 'amber_metadata' key in {p}")
-
-        amber_meta = payload["amber_metadata"]
-        n_latents = int(amber_meta["n_latents"])
-        n_inputs = int(amber_meta["n_inputs"])
-        device = amber_meta.get("device", "cpu")
-        layer_signature = amber_meta.get("layer_signature")
-        model_id = amber_meta.get("model_id")
-        concepts_state = amber_meta.get("concepts_state", {})
+        if "mi_crow_metadata" not in payload:
+            raise ValueError(f"Invalid L1SAE save format: missing 'mi_crow_metadata' key in {p}")
+
+        mi_crow_meta = payload["mi_crow_metadata"]
+        n_latents = int(mi_crow_meta["n_latents"])
+        n_inputs = int(mi_crow_meta["n_inputs"])
+        device = mi_crow_meta.get("device", "cpu")
+        layer_signature = mi_crow_meta.get("layer_signature")
+        model_id = mi_crow_meta.get("model_id")
+        concepts_state = mi_crow_meta.get("concepts_state", {})
 
         # Create L1Sae instance
         l1_sae = L1Sae(

{amber → mi_crow}/mechanistic/sae/modules/topk_sae.py

@@ -7,11 +7,11 @@ from overcomplete import (
     TopKSAE as OvercompleteTopkSAE,
     SAE as OvercompleteSAE
 )
-from amber.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
-from amber.mechanistic.sae.sae import Sae
-from amber.mechanistic.sae.sae_trainer import SaeTrainingConfig
-from amber.store.store import Store
-from amber.utils import get_logger
+from mi_crow.hooks.hook import HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.mechanistic.sae.sae import Sae
+from mi_crow.mechanistic.sae.sae_trainer import SaeTrainingConfig
+from mi_crow.store.store import Store
+from mi_crow.utils import get_logger
 
 logger = get_logger(__name__)
 
@@ -338,7 +338,7 @@ class TopKSae(Sae):
         # Save overcomplete model state dict
         sae_state_dict = self.sae_engine.state_dict()
 
-        amber_metadata = {
+        mi_crow_metadata = {
             "concepts_state": {
                 'multiplication': self.concepts.multiplication.data,
                 'bias': self.concepts.bias.data,
@@ -353,7 +353,7 @@ class TopKSae(Sae):
 
         payload = {
             "sae_state_dict": sae_state_dict,
-            "amber_metadata": amber_metadata,
+            "mi_crow_metadata": mi_crow_metadata,
         }
 
         torch.save(payload, save_path)
@@ -382,17 +382,17 @@ class TopKSae(Sae):
         payload = torch.load(p, map_location=map_location)
 
         # Extract our metadata
-        if "amber_metadata" not in payload:
-            raise ValueError(f"Invalid TopKSAE save format: missing 'amber_metadata' key in {p}")
-
-        amber_meta = payload["amber_metadata"]
-        n_latents = int(amber_meta["n_latents"])
-        n_inputs = int(amber_meta["n_inputs"])
-        k = int(amber_meta["k"])
-        device = amber_meta.get("device", "cpu")
-        layer_signature = amber_meta.get("layer_signature")
-        model_id = amber_meta.get("model_id")
-        concepts_state = amber_meta.get("concepts_state", {})
+        if "mi_crow_metadata" not in payload:
+            raise ValueError(f"Invalid TopKSAE save format: missing 'mi_crow_metadata' key in {p}")
+
+        mi_crow_meta = payload["mi_crow_metadata"]
+        n_latents = int(mi_crow_meta["n_latents"])
+        n_inputs = int(mi_crow_meta["n_inputs"])
+        k = int(mi_crow_meta["k"])
+        device = mi_crow_meta.get("device", "cpu")
+        layer_signature = mi_crow_meta.get("layer_signature")
+        model_id = mi_crow_meta.get("model_id")
+        concepts_state = mi_crow_meta.get("concepts_state", {})
 
         # Create TopKSAE instance
         topk_sae = TopKSae(

{amber → mi_crow}/mechanistic/sae/sae.py

@@ -5,15 +5,15 @@ from typing import Any, TYPE_CHECKING, Literal
 import torch
 from torch import nn
 
-from amber.hooks.controller import Controller
-from amber.hooks.detector import Detector
-from amber.hooks.hook import Hook, HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
-from amber.mechanistic.sae.autoencoder_context import AutoencoderContext
-from amber.mechanistic.sae.concepts.autoencoder_concepts import AutoencoderConcepts
-from amber.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
-from amber.mechanistic.sae.sae_trainer import SaeTrainer
-from amber.store.store import Store
-from amber.utils import get_logger
+from mi_crow.hooks.controller import Controller
+from mi_crow.hooks.detector import Detector
+from mi_crow.hooks.hook import Hook, HookType, HOOK_FUNCTION_INPUT, HOOK_FUNCTION_OUTPUT
+from mi_crow.mechanistic.sae.autoencoder_context import AutoencoderContext
+from mi_crow.mechanistic.sae.concepts.autoencoder_concepts import AutoencoderConcepts
+from mi_crow.mechanistic.sae.concepts.concept_dictionary import ConceptDictionary
+from mi_crow.mechanistic.sae.sae_trainer import SaeTrainer
+from mi_crow.store.store import Store
+from mi_crow.utils import get_logger
 
 from overcomplete.sae import SAE as OvercompleteSAE
 

{amber → mi_crow}/mechanistic/sae/sae_trainer.py

@@ -9,12 +9,12 @@ import gc
 
 import torch
 
-from amber.store.store_dataloader import StoreDataloader
-from amber.utils import get_logger
+from mi_crow.store.store_dataloader import StoreDataloader
+from mi_crow.utils import get_logger
 
 if TYPE_CHECKING:
-    from amber.mechanistic.sae.sae import Sae
-    from amber.store.store import Store
+    from mi_crow.mechanistic.sae.sae import Sae
+    from mi_crow.store.store import Store
 
 
 @dataclass
@@ -528,7 +528,7 @@ class SaeTrainer:
 
             sae_state_dict = self.sae.sae_engine.state_dict()
 
-            amber_metadata = {
+            mi_crow_metadata = {
                 "concepts_state": {
                     'multiplication': self.sae.concepts.multiplication.data.cpu().clone(),
                     'bias': self.sae.concepts.bias.data.cpu().clone(),
@@ -541,11 +541,11 @@ class SaeTrainer:
             }
 
             if hasattr(self.sae, 'k'):
-                amber_metadata["k"] = self.sae.k
+                mi_crow_metadata["k"] = self.sae.k
 
             payload = {
                 "sae_state_dict": sae_state_dict,
-                "amber_metadata": amber_metadata,
+                "mi_crow_metadata": mi_crow_metadata,
             }
 
             torch.save(payload, model_path)

mi_crow/mechanistic/sae/training/__init__.py

@@ -0,0 +1,6 @@
+"""Training utilities for SAE models."""
+
+from mi_crow.mechanistic.sae.training.wandb_logger import WandbLogger
+
+__all__ = ["WandbLogger"]
+

{amber → mi_crow}/mechanistic/sae/training/wandb_logger.py

@@ -2,8 +2,8 @@
 
 from typing import Any, Optional
 
-from amber.mechanistic.sae.sae_trainer import SaeTrainingConfig
-from amber.utils import get_logger
+from mi_crow.mechanistic.sae.sae_trainer import SaeTrainingConfig
+from mi_crow.utils import get_logger
 
 logger = get_logger(__name__)
 

mi_crow/store/__init__.py

@@ -0,0 +1,5 @@
+from mi_crow.store.store import Store
+from mi_crow.store.local_store import LocalStore
+
+__all__ = ["Store", "LocalStore"]
+

{amber → mi_crow}/store/local_store.py

@@ -5,7 +5,7 @@ import shutil
 
 import torch
 
-from amber.store.store import Store, TensorMetadata
+from mi_crow.store.store import Store, TensorMetadata
 import safetensors.torch as storch
 
 

{amber → mi_crow}/store/store_dataloader.py

@@ -2,12 +2,12 @@ from typing import Optional, Iterator, TYPE_CHECKING
 
 import torch
 
-from amber.utils import get_logger
+from mi_crow.utils import get_logger
 
 logger = get_logger(__name__)
 
 if TYPE_CHECKING:
-    from amber.store.store import Store
+    from mi_crow.store.store import Store
 
 
 class StoreDataloader:

{amber → mi_crow}/utils.py

@@ -27,7 +27,7 @@ def set_seed(seed: int, deterministic: bool = True) -> None:
         torch.backends.cudnn.benchmark = False  # type: ignore[attr-defined]
 
 
-def get_logger(name: str = "amber", level: int | str = logging.INFO) -> logging.Logger:
+def get_logger(name: str = "mi_crow", level: int | str = logging.INFO) -> logging.Logger:
     """Get a configured logger with a simple format. Idempotent."""
     logger = logging.getLogger(name)
     if isinstance(level, str):

{mi_crow-0.1.1.post12.dist-info → mi_crow-0.1.1.post14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mi-crow
-Version: 0.1.1.post12
+Version: 0.1.1.post14
 Summary: Engineer Thesis: Explaining and modifying LLM responses using SAE and concepts.
 Author-email: Hubert Kowalski <your.email@example.com>, Adam Kaniasty <adam.kaniasty@gmail.com>
 Requires-Python: >=3.10
@@ -96,7 +96,7 @@ uv run --group server pytest tests/server/test_api.py --cov=server --cov-fail-un
 
 ### SAE API usage
 
-- Configure artifact location (optional): `export SERVER_ARTIFACT_BASE_PATH=/path/to/amber_artifacts` (defaults to `~/.cache/amber_server`)
+- Configure artifact location (optional): `export SERVER_ARTIFACT_BASE_PATH=/path/to/mi_crow_artifacts` (defaults to `~/.cache/mi_crow_server`)
 - Load a model: `curl -X POST http://localhost:8000/models/load -H "Content-Type: application/json" -d '{"model_id":"bielik"}'`
 - Save activations from dataset (stored in `LocalStore` under `activations/<model>/<run_id>`):
   - HF dataset: `{"dataset":{"type":"hf","name":"ag_news","split":"train","text_field":"text"}}`

mi_crow-0.1.1.post14.dist-info/RECORD

@@ -0,0 +1,52 @@
+mi_crow/__init__.py,sha256=J7aXVlAicbjvk5630rhDxx0ATsvZnihud5u_aQpAwY8,487
+mi_crow/utils.py,sha256=LTfh2Ep87lAgPBaZkrQPP9caXFJoS9zUxu4qFuV4kzM,1549
+mi_crow/datasets/__init__.py,sha256=lCAc3nFlvoERrBPAan6C9YFmDx86W2gbIAy267Rb2Sk,349
+mi_crow/datasets/base_dataset.py,sha256=vYx-oj3jVhLZD1-xGSO4K4ZIsQtYpHP5zHmg7jd4FE0,22512
+mi_crow/datasets/classification_dataset.py,sha256=nL_xndJHyf8hlLxKBe_ZO2YLYsXQjGyeY6csqGTTzEY,21706
+mi_crow/datasets/loading_strategy.py,sha256=17VM3Td8lqDllGIx9DHI6WiXmSKKQHDHbfe4ZeM8ATA,1206
+mi_crow/datasets/text_dataset.py,sha256=5FzHWkMWWK0yP69O48S3fUj5KgHb8qo3mkvvZihHFuU,16781
+mi_crow/hooks/__init__.py,sha256=KYy5qcbEpnJceNH86ofy43Suu_36QXjj0HYl79rVyls,693
+mi_crow/hooks/controller.py,sha256=eo8LMERORXYUjH4-_R6DHk5JKN6O8SW6PlnuBFrlNqg,6063
+mi_crow/hooks/detector.py,sha256=Bj3xz56cSgRvbcoQBsHIdlJdf0dtgVLw3l1pOSRvRAg,3114
+mi_crow/hooks/hook.py,sha256=JrCyPptXzHICAToxug7FD8zKWKZcLxbXfIe7UHCkh34,7542
+mi_crow/hooks/utils.py,sha256=GdUAqL9InCsthjBVWbZtVQp2VtQLLOMaJy8S8Tc7WX0,2024
+mi_crow/hooks/implementations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mi_crow/hooks/implementations/function_controller.py,sha256=Rwu1Ghffqm-Jc4mqniRdIsXQeV24JKlx9CMO8jP8iv0,3092
+mi_crow/hooks/implementations/layer_activation_detector.py,sha256=mFqLkWrBlNYzYYvwI8O-ymnpwP1UJSX7eWIQ7Fp-lxE,3167
+mi_crow/hooks/implementations/model_input_detector.py,sha256=4AFA88eQ-zM9Kz8bpv4Bl5cvlZF5ZyIRXNi7Znd6V98,10078
+mi_crow/hooks/implementations/model_output_detector.py,sha256=BmD8oJ1k_nv4hIzG-9SGxSCRARQkLRI4d_XQmKs6jjE,4883
+mi_crow/language_model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mi_crow/language_model/activations.py,sha256=mJhuXkQ_JKzQsrFossnXVTdq4WgBI10Gi0ZYeQ9USFY,16989
+mi_crow/language_model/context.py,sha256=TJSe6IruGktePpQ0dtHIoeatSY72qAhEKeTvLizCdlw,1124
+mi_crow/language_model/contracts.py,sha256=6ij7rzJcpSAKgYx-fiefg0Fi8TsFugaM5IlihEm_2VM,268
+mi_crow/language_model/hook_metadata.py,sha256=GACZjZUneo2l5j7DCFycLAunTm0etdMQ2YB_xgueUuk,1394
+mi_crow/language_model/inference.py,sha256=-Kpm85jM8y6-GyDgrvIczitBIwGh8grJP8aYuXsLV-g,19082
+mi_crow/language_model/initialization.py,sha256=e_Vkk-p9KWRt6-Hmkm6I29dTf20jzEAyNF9CG4nc48M,3704
+mi_crow/language_model/language_model.py,sha256=a6CcklVA65oYtFxGXiwQrOKMPZj6eb7LOiT1zJ5-guo,13965
+mi_crow/language_model/layers.py,sha256=1yExHodMyqr_Yk4W-2HiSGnRs2sYOA7swsxI8u0Uvfk,15914
+mi_crow/language_model/persistence.py,sha256=9wQE6tRvLg7BgdLlkKRTOfRwXb5Q0LsEgg8B9J7Yos0,5881
+mi_crow/language_model/tokenizer.py,sha256=uZbMDVNnzu8WZINUaR1tLFXiuk9V5pAoahwnJOUvEuE,7379
+mi_crow/language_model/utils.py,sha256=5Y7scRvvudUjKDV8QPhC3HAc2S-dCuqbm6xEjRr0fRM,2630
+mi_crow/mechanistic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mi_crow/mechanistic/sae/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mi_crow/mechanistic/sae/autoencoder_context.py,sha256=u5WzSlLb8_HaJF9LwGqe-J_rE-iRCoXbvwEhTZIArkw,947
+mi_crow/mechanistic/sae/sae.py,sha256=R2IckZ-UDVXFURUSRoIdX0MGLe0TWU5JCVeDXPz1Wv4,6053
+mi_crow/mechanistic/sae/sae_trainer.py,sha256=VqN9UgEPtgYItzAS0RqMfOhTnTxIEuModO0noHYieFA,26327
+mi_crow/mechanistic/sae/concepts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mi_crow/mechanistic/sae/concepts/autoencoder_concepts.py,sha256=SE7A6y6cZiMGhylljNdZz0D8lhrulUh1FYiKhnPPRvY,14387
+mi_crow/mechanistic/sae/concepts/concept_dictionary.py,sha256=aUXTe4Fy7Oe7iYJOcQImEXYMZAXhnkfAszZNwivk4eg,7673
+mi_crow/mechanistic/sae/concepts/concept_models.py,sha256=HGyPoMSmj8CAg9joIa6fV_FfWAY2XN4LFPIkNQiJWbY,133
+mi_crow/mechanistic/sae/concepts/input_tracker.py,sha256=kIiqt7guv_-9-UPYtefAFJbHkWtAS_mnqYVvRU4eb2o,1890
+mi_crow/mechanistic/sae/modules/__init__.py,sha256=e0lkCALQZcJN7KpYyTtXx3OD2NhBxV_kOZLLJ6EWaTE,243
+mi_crow/mechanistic/sae/modules/l1_sae.py,sha256=qqw0iTWLSmWAlz5kgfw_mex8LeecFWM1FobyUteMqmM,15388
+mi_crow/mechanistic/sae/modules/topk_sae.py,sha256=pK_ajKTQb0wGAftzb6AE5ZZthV3aFLr6G3avOVclSHE,17313
+mi_crow/mechanistic/sae/training/__init__.py,sha256=5flCJVkOyKizY0FZy1OP5v0EI6bPEayunpnUPp82a6s,140
+mi_crow/mechanistic/sae/training/wandb_logger.py,sha256=YlSJd5CaNa35RmIgf1FD_gSEDyhGRa2UdHo_Ofrplos,8558
+mi_crow/store/__init__.py,sha256=DrYTpdgzrRzjHm9bigy-GiP0BGxzjmD3-lJCthtgxbE,123
+mi_crow/store/local_store.py,sha256=XmguFvdrUi6NHzvV_bLaDJzpk5KWU_-ObkzhICcLu6g,17216
+mi_crow/store/store.py,sha256=VuDe9Git0glND3TTHh0zhDJNxdQY3dCp0cURhApYQbU,9334
+mi_crow/store/store_dataloader.py,sha256=UkZhHCOTg56ozomPtU9vHBhxIMOPcOiyfMqiAxgqtQs,4341
+mi_crow-0.1.1.post14.dist-info/METADATA,sha256=Dd5lhIR9XmTNgdTDgZBhSsFVAvih80juP77KxHHfjFQ,6584
+mi_crow-0.1.1.post14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mi_crow-0.1.1.post14.dist-info/top_level.txt,sha256=DTuNo2VWgrH6jQKY19NciReSpLwGKKIRzJ3WbpspLlE,8
+mi_crow-0.1.1.post14.dist-info/RECORD,,

mi_crow-0.1.1.post14.dist-info/top_level.txt

@@ -0,0 +1 @@
+mi_crow

amber/datasets/__init__.py

@@ -1,11 +0,0 @@
-from amber.datasets.base_dataset import BaseDataset
-from amber.datasets.text_dataset import TextDataset
-from amber.datasets.classification_dataset import ClassificationDataset
-from amber.datasets.loading_strategy import LoadingStrategy
-
-__all__ = [
-    "BaseDataset",
-    "TextDataset",
-    "ClassificationDataset",
-    "LoadingStrategy",
-]

amber/hooks/__init__.py

@@ -1,20 +0,0 @@
-from amber.hooks.hook import Hook, HookType, HookError
-from amber.hooks.detector import Detector
-from amber.hooks.controller import Controller
-from amber.hooks.implementations.layer_activation_detector import LayerActivationDetector
-from amber.hooks.implementations.model_input_detector import ModelInputDetector
-from amber.hooks.implementations.model_output_detector import ModelOutputDetector
-from amber.hooks.implementations.function_controller import FunctionController
-
-__all__ = [
-    "Hook",
-    "HookType",
-    "HookError",
-    "Detector",
-    "Controller",
-    "LayerActivationDetector",
-    "ModelInputDetector",
-    "ModelOutputDetector",
-    "FunctionController",
-]
-

amber/mechanistic/sae/modules/__init__.py

@@ -1,5 +0,0 @@
-from amber.mechanistic.sae.modules.topk_sae import TopKSae, TopKSaeTrainingConfig
-from amber.mechanistic.sae.modules.l1_sae import L1Sae, L1SaeTrainingConfig
-
-__all__ = ["TopKSae", "TopKSaeTrainingConfig", "L1Sae", "L1SaeTrainingConfig"]
-

amber/store/__init__.py

@@ -1,5 +0,0 @@
-from amber.store.store import Store
-from amber.store.local_store import LocalStore
-
-__all__ = ["Store", "LocalStore"]
-

mi_crow-0.1.1.post12.dist-info/RECORD

@@ -1,51 +0,0 @@
-amber/__init__.py,sha256=5nh0D8qvFgOhBEQj00Rm06T1iY5VcSiifAg9SoY1LLA,483
-amber/utils.py,sha256=oER2LA_alUjaIk_xCAyP2V54ywjqsg00I4KvitYnJPc,1547
-amber/datasets/__init__.py,sha256=zhqgbm5zMBsRbmPNfjlYNJwGWOLuCNf5jEj0P8aopRU,341
-amber/datasets/base_dataset.py,sha256=X2wt3GdjgAOY24_vOqrD5gVFxGplSRMCb69CoQtj0xw,22508
-amber/datasets/classification_dataset.py,sha256=x_ZQ4dMzoY3Nn8V1I01xvzJK_IcHcDcm8dIxYeXzV5g,21700
-amber/datasets/loading_strategy.py,sha256=17VM3Td8lqDllGIx9DHI6WiXmSKKQHDHbfe4ZeM8ATA,1206
-amber/datasets/text_dataset.py,sha256=ly0GHCS28Rg5ZluaafjavhcbvSD9-6ryovd_Y1ZIMms,16775
-amber/hooks/__init__.py,sha256=9H08ZVoTK6TzYJXjEP2aqdHfoyLfdXvg6eOv3K1zNps,679
-amber/hooks/controller.py,sha256=hc8FrrDosFYLrEGsEZmx1KsJ77F4p_gMKcF2WzHiURY,6057
-amber/hooks/detector.py,sha256=5drJFrdrjseVjRNT-cq-U8XCt8AXV04YY2YrkQz4eFk,3110
-amber/hooks/hook.py,sha256=-Qi-GJqRuIskXMuHUzp9_ESbbZi5tLSAFMWoBmrP3io,7540
-amber/hooks/utils.py,sha256=wtsrjsMt-bXR3NshkwyZmfLre3IE3S4E5EoKppQrYOo,2022
-amber/hooks/implementations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-amber/hooks/implementations/function_controller.py,sha256=66FFx_7sU7b0_FFQFFkAOqQm3aGsiyIRuZ64hIv-0w8,3088
-amber/hooks/implementations/layer_activation_detector.py,sha256=bzoW6V8NNDNgRASs1YN_1TjEXaK3ahoNWiZ-ODfjB6I,3161
-amber/hooks/implementations/model_input_detector.py,sha256=cYRVfyBEHi-1qg6F-4Q0vKEae6gYtq_3g1j3rOOCQdA,10074
-amber/hooks/implementations/model_output_detector.py,sha256=iN-twt7Chc9ODmj-iei7_Ah7GqvE-knTVWi4C9kNye4,4879
-amber/language_model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-amber/language_model/activations.py,sha256=7VRDlCM-IctF1ee0H68_Dd7xeXHyUSQe1SOPb28Ej18,16971
-amber/language_model/context.py,sha256=koslpikCcu9Svsopboa1wd7Fv0R2-4sI2whOCvVWvT8,1118
-amber/language_model/contracts.py,sha256=6ij7rzJcpSAKgYx-fiefg0Fi8TsFugaM5IlihEm_2VM,268
-amber/language_model/hook_metadata.py,sha256=9Xyfiu4ekCZj79zG4gZfLk-850AO2iKDE24FDXe7q7s,1392
-amber/language_model/inference.py,sha256=l8BASS8E9B4VWJHucEqF_G_zqOzlKeG4KEvGameBbMw,19068
-amber/language_model/initialization.py,sha256=hfrKdI_fsmaxk0p9q4wN7EFxq_lSXs8BXGlxwKJ21Qw,3698
-amber/language_model/language_model.py,sha256=MXoaXYbNBUxHw4sxtWkVFLbXiw9tFEx9GI78sCiESuQ,13943
-amber/language_model/layers.py,sha256=Ob7QZl8i236ALLklY9o_xtjDZSt6FD8sqdmFy_YLgN0,15906
-amber/language_model/persistence.py,sha256=i2ibDH1OABM5-ZNNLh7h4rOYWPsg3aaeYhmB_xWYDZw,5867
-amber/language_model/tokenizer.py,sha256=9eKNOHvUjIJhJbj7M-tN7jWU5lWhOeCY_cssa4exQ1g,7377
-amber/language_model/utils.py,sha256=5Y7scRvvudUjKDV8QPhC3HAc2S-dCuqbm6xEjRr0fRM,2630
-amber/mechanistic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-amber/mechanistic/sae/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-amber/mechanistic/sae/autoencoder_context.py,sha256=cn0mv9COqT3jNcvXBfce70ankVEmw9kNE3Mu-knugoc,945
-amber/mechanistic/sae/sae.py,sha256=ha6rXGsOXE59E_ohTH0vJh6M4rQh3Xw0GfmCkSgeYS4,6035
-amber/mechanistic/sae/sae_trainer.py,sha256=GMrPz9SpSuANA0tJt3IkyIOOqVr2k7apE0w4CqL92gM,26311
-amber/mechanistic/sae/concepts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-amber/mechanistic/sae/concepts/autoencoder_concepts.py,sha256=dTB9v5zK5lcKc4_zbYtWoCy8fA6ycj1-FJid4-7hQ04,14371
-amber/mechanistic/sae/concepts/concept_dictionary.py,sha256=9px845gODuufW2koym-5426q0ijdpxvS7avH1vu8-Ls,7671
-amber/mechanistic/sae/concepts/concept_models.py,sha256=HGyPoMSmj8CAg9joIa6fV_FfWAY2XN4LFPIkNQiJWbY,133
-amber/mechanistic/sae/concepts/input_tracker.py,sha256=81FrOv9AAC7ejhryOWDTZ7Hlt3B2WoANx-wiO0KLr24,1886
-amber/mechanistic/sae/modules/__init__.py,sha256=xpoz0HtPWoJD4dPj1qHaxtXDr7J0ERn30CX3m1dz21s,239
-amber/mechanistic/sae/modules/l1_sae.py,sha256=_BebvpB9iUCTDjSiYNzDBM4l9sU_wadAdypwXaSb4ww,15352
-amber/mechanistic/sae/modules/topk_sae.py,sha256=GQA8hYb6Fw7U1e5ExZjzZBjAZRhGk-VwqqYREVCQ_u8,17275
-amber/mechanistic/sae/training/wandb_logger.py,sha256=d3vVBIQrnsJurX5HNVu7OYW4DqNgv18UZDpV8ddfN9k,8554
-amber/store/__init__.py,sha256=UW4Hqyu-_qgnZ-gN_mk97OaWSrlPERcNi5YjnXMKeOU,119
-amber/store/local_store.py,sha256=1pJbizZKrzNt_IQFnCFYjApPXs9ot-G1H8adeR7Qi50,17214
-amber/store/store.py,sha256=VuDe9Git0glND3TTHh0zhDJNxdQY3dCp0cURhApYQbU,9334
-amber/store/store_dataloader.py,sha256=QyYHSgOos8e-yzaEE_rySSVlGKaRNybURSDCgNrTIVM,4337
-mi_crow-0.1.1.post12.dist-info/METADATA,sha256=1_SPBcnK8j_scOIWwnPqQGjlUpNnXxGGtvkYRZHeDQ8,6580
-mi_crow-0.1.1.post12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mi_crow-0.1.1.post12.dist-info/top_level.txt,sha256=FNP1x_ePvcW9Jsr7J9gCBARdDC-gqxIYtWF6HGNxtnI,6
-mi_crow-0.1.1.post12.dist-info/RECORD,,

mi_crow-0.1.1.post12.dist-info/top_level.txt

@@ -1 +0,0 @@
-amber