You can deploy this model and start earning money today!

kitft/
nla-gemma3-12b-L32-ar

Name: nla-gemma3-12b-L32-ar
Availability: InStock

google/gemma-3-12b-it

finetune

gemma

Views

No views yet

nla-gemma3-12b-L32-ar

The AR (activation reconstructor, text → vector) half of a Natural Language Autoencoder (NLA) pair, fine-tuned from google/gemma-3-12b-it. The other half is kitft/nla-gemma3-12b-L32-av; both are released together and are intended to be used as a pair.

NLA pairs are interpretability tools: the AV (activation verbalizer) maps a hidden-state vector to a natural-language description; the AR (activation reconstructor) maps that description back to a vector. Together they let you read out what a residual-stream activation "means" and measure how much of it the description captured. These checkpoints are not useful as general-purpose language models — the fine-tuning repurposes them entirely for activation decoding.

📄 Paper: Natural Language Autoencoders Produce Unsupervised Explanations of LLM Activations
Inference code + worked examples: kitft/nla-inference
Training code: kitft/natural_language_autoencoders
Extraction layer: residual stream output of block 32
In-distribution fve_nrm: 0.768 (training set, 50/50 WildChat + Ultra-FineWeb)

Usage

See the nla-inference README for the full recipe (SGLang launch, NLAClient/NLACritic, embedding-injection details).

Citation

bibtex

@article{frasertaliente2026nla,
  author  = {Fraser-Taliente, Kit and Kantamneni, Subhash and Ong, Euan and Mossing, Dan and Lu, Christina and Bogdan, Paul C. and Ameisen, Emmanuel and Chen, James and Kishylau, Dzmitry and Pearce, Adam and Tarng, Julius and Wu, Alex and Wu, Jeff and Zhang, Yang and Ziegler, Daniel M. and Hubinger, Evan and Batson, Joshua and Lindsey, Jack and Zimmerman, Samuel and Marks, Samuel},
  title   = {Natural Language Autoencoders Produce Unsupervised Explanations of LLM Activations},
  journal = {Transformer Circuits Thread},
  year    = {2026},
  url     = {https://transformer-circuits.pub/2026/nla/index.html}
}

License & use restrictions

This model is a derivative of Gemma 3 and is provided under and subject to the Gemma Terms of Use. By using this model you agree to those terms and the Gemma Prohibited Use Policy. See NOTICE in this repository.

Training data attribution

The fine-tuning data was derived from two public datasets:

WildChat-1M (allenai/WildChat-1M). Contains information from WildChat-1M which is made available under the ODC Attribution License.
Ultra-FineWeb (openbmb/Ultra-FineWeb, Apache-2.0), a filtered derivative of HuggingFaceFW/fineweb (ODC-BY).

kitft/nla-gemma3-12b-L32-ar