sae_stats.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.sae_stats

Functions
Path	aquin.compute.sae_stats._sae_cache_dir (str model_id)
list[int]|None	aquin.compute.sae_stats._parse_layers_wanted (str|None layers)
str	aquin.compute.sae_stats._format_load_sae_commands (str model_id, list[int] layers)
str	aquin.compute.sae_stats._missing_sae_error (str model_id, *, str mode, list[int] available, str|None layers_arg)
Path	aquin.compute.sae_stats._expand (str path)
list[dict[str, Any]]	aquin.compute.sae_stats._rows_to_probes (list[Any] rows)
list[dict[str, Any]]	aquin.compute.sae_stats.load_probes (str|list[Any] path)
list[int]	aquin.compute.sae_stats.parse_layers_arg (str|None layers, list[int] available, *, str|None model_id=None, str mode="llm")
list[int]	aquin.compute.sae_stats.list_llm_sae_layers (str model_id)
dict[str, Any]	aquin.compute.sae_stats._feature_stats (torch.Tensor mean_acts, *, int top_k)
torch.Tensor	aquin.compute.sae_stats._llm_probe_acts (Any model, Any sae, str model_id, int layer, str text)
dict[str, Any]	aquin.compute.sae_stats.run_llm_sae_stats (Any model, str model_id, list[dict[str, Any]] probes, *, list[int] layers, int top_k=10)
dict[str, Any]	aquin.compute.sae_stats._build_payload (*, str model_id, str mode, list[dict[str, Any]] probes, list[int] layers, list[dict[str, Any]] layer_stats, list[str] heatmap_rows, list[str] heatmap_cols, list[list[float]] heatmap_values, int top_k)
dict[str, Any]	aquin.compute.sae_stats.run_sae_stats (dict[str, Any] args)

Variables
	aquin.compute.sae_stats.DEVICE = resolve_compute_device()
tuple	aquin.compute.sae_stats._PROBE_TEXT_KEYS = ("text", "prompt", "input", "query", "instruction")
tuple	aquin.compute.sae_stats._META_KEYS = ("id", "stressor", "lang", "quant_run_id", "condition", "label", "group")