sae/autoencoder/__init__.py

from .Standard import SAE_Template, SAE
from .TopK import TopKSAE, TopKTranscoder, TopKCrosscoder
from .Utils import standardize, unit_norm_decoder, unit_norm_decoder_gradient
from .geom_median import geometric_median
from .types import TopK, SAEOut, Stats
from .SAE_Wrapper import HookedSAE
from .BatchTopK import BatchTopKSAE

__all__ = [
    "SAE_Template",
    "SAE",
    "TopKSAE",
    "TopK",
    "SAEOut",
    "Stats",
    "standardize",
    "unit_norm_decoder",
    "unit_norm_decoder_gradient",
    "geometric_median",
    "HookedSAE",
    "TopKTranscoder",
    "TopKCrosscoder",
    "BatchTopKSAE",
]