Instructions to use decoderesearch/gemma-4-saes with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- SAELens
How to use decoderesearch/gemma-4-saes with SAELens:
# pip install sae-lens from sae_lens import SAE sae, cfg_dict, sparsity = SAE.from_pretrained( release = "RELEASE_ID", # e.g., "gpt2-small-res-jb". See other options in https://github.com/jbloomAus/SAELens/blob/main/sae_lens/pretrained_saes.yaml sae_id = "SAE_ID", # e.g., "blocks.8.hook_resid_pre". Won't always be a hook point ) - Notebooks
- Google Colab
- Kaggle
Gemma-4 SAEs for use with the SAELens library
This repository contains the following SAEs:
Gemma-4-E2B
- gemma-4-e2b/btk-mat-layer-6-k-100
- gemma-4-e2b/btk-mat-layer-17-k-100
- gemma-4-e2b/btk-mat-layer-28-k-100
Gemma-4-E4B
- gemma-4-e4b/btk-mat-layer-7-k-100
- gemma-4-e4b/btk-mat-layer-21-k-100
- gemma-4-e4b/btk-mat-layer-35-k-100
Load these SAEs using SAELens as below:
from sae_lens import SAE
sae = SAE.from_pretrained("decoderesearch/gemma-4-saes", "<sae_id>")
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support