crosscoder-learning

A toolkit for sparse autoencoders, crosscoders, and dictionary learning on the internal activity of neural networks. These methods help represent and compare patterns inside language models.

In plain words
What is it for?
Use it to train or load dictionaries, cache model activity, compare models, evaluate representations, and work with language-model internals through nnsight.
Why use it?
It helps researchers inspect model features, compare a base model with a fine-tuned version, and measure which internal patterns changed.

Skill for Claude CodeCodex

Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

agentmods
npx agentmods add skills/zjunlp/mechanist/crosscoder
Any agent
npx skills add zjunlp/Mechanist --skill crosscoder
Clone the repo
git clone --depth 1 https://github.com/zjunlp/Mechanist

Made for: Claude Code, Codex.

Per session 70 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 5,486 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. Scan, not verified.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5 $0.00070 $0.05486
Opus 5 $0.00035 $0.02743
Sonnet 5 $0.00014 $0.01097
Haiku 4.5 $0.00007 $0.00549

Measured 2d ago against content hash e983ce79e9f0, method: parsed. Prices are Anthropic first-party input rates as of 2026-08-30, from the pricing page.

Security

Grade A, and why

crosscoder-learning scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 2d ago.

The scan reads SKILL.md. This mod also ships 2 executable files (scripts/train_and_evaluate_sae.py, scripts/train_sae_demo.py), listed below but not scanned — reading those needs a real analyzer, not pattern matching.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

skills/mechanism-skills/feature-dictionary-learning/crosscoder/SKILL.md · 642 lines

How it starts

The opening of the file, as written. The whole thing — 642 lines — stays where its author put it; the contents beside it link to each section on GitHub.

Dictionary Learning & Crosscoders Skill

When to Use

Activate this skill when the user needs to:

  • Train Sparse Autoencoders (SAEs) on neural network activations (MLP outputs, attention outputs, residual streams)
  • Train CrossCoders to compare activations across two related models (e.g., a base model and its fine-tuned variant)
  • Load and use pretrained dictionaries (AutoEncoders, CrossCoders, JumpReLU SAEs) from local disk or the Hugging Face Hub
  • Cache activations from a language model for later offline training
  • Evaluate dictionaries using MSE loss, L1/L0 sparsity, CE diff, and variance-explained metrics
  • Push or pull dictionary weights to/from the Hugging Face Hub
  • Perform model diffing to identify which features change between a base model and a fine-tuned model
  • Use BatchTopKCrossCoder for batch-level top-k sparsity across model pairs
  • Work with the nnsight library to hook into language model internals

Keywords: sparse autoencoder, SAE, crosscoder, dictionary learning, activation buffer, neural network features, mechanistic interpretability, model diffing, JumpReLU, GatedSAE, TopK SAE, nnsight, Pythia, Gemma, residual stream, MLP output, attention output


Quick Reference

Resource URL
Repository (fork) https://github.com/jkminder/dictionary_learning
Original Repository https://github.com/saprmarks/dictionary_learning
Anthropic CrossCoder Paper https://transformer-circuits.pub/drafts/crosscoders/index.html#model-diffing
BatchTopKCrossCoder Paper https://arxiv.org/pdf/2504.02922
nnsight Documentation https://nnsight.net/
nnsight Walkthrough https://nnsight.net/notebooks/tutorials/walkthrough/
Pretrained Dictionaries (Pythia) https://baulab.us/u/smarks/autoencoders/
Pretrained CrossCoder (HF Hub) Butanium/gemma-2-2b-crosscoder-l13-mu4.1e-02-lr1e-04
Pretrained SAE Downloader Script ./pretrained_dictionary_downloader.sh

Installation / Setup

Read the full file on GitHub · 642 lines

Files

What ships with it

3 files beside SKILL.md in the same directory: the scripts, references and assets a skill reads on demand. Not counted in the per-session cost; read them before you install if any of them is executable.

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 2d ago First seen · 642 lines · 70 tokens per session scan A e983ce79e9f0

Subscribe to this mod's changes

crosscoder-learning is a skill published in the GitHub repository zjunlp/Mechanist (51 stars, last pushed 6d ago), licensed MIT. It adds 70 tokens to every session and 5,486 once invoked, about $0.0003 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.