Open supervised dictionary learning models and datasets for Gemma 2 2B and 9B instruction-tuned models.

pyvene
university
AI & ML interests
interpretability
Recent Activity
View all activity
Organization Card
Who are we?
We are a group of hackers from Stanford's NLP group, and we are interested in LLM interpretability.
pyvene
is where we started, which stands for pytorch model intervenetion.
Resources
Supervised dictionary learning models (SDLs) and datasets releases for Gemma 2 2B and 9B: AxBench Collection
.
Benchmark interpretability methods at scale (AxBench) library: AxBench
.
Representation finetuning (ReFT) library: pyreft
.
PyTorch model intervention library: pyvene
.
Collections
1
spaces
6
Running
on
Zero
2
SDL-ReFT-cr1
ðŸ«
Guide chatbot with specific topics
Running
on
Zero
2
SDL-ReFT-r1
🤔
Guide conversations with specific topics
Sleeping
4
ReFT-Golden-Gate-Bridge
ðŸ«
Converse with an AI assistant that mimics the Golden Gate Bridge
Sleeping
6
ReFT-Chat7B
🫡
Generate responses to chat messages using ReFT-Chat
Running
on
Zero
3
ReFT-Emoji
🫡
Chat with an emoji-enhanced assistant
Sleeping
2
ReFT-Ethos
🚀
Converse with a helpful assistant in text form
models
12

pyvene/gemma-reft-r1-9b-it-res
Updated

pyvene/gemma-reft-r1-2b-it-res
Updated

pyvene/gemma-diffmean-2b-it-res
Updated

pyvene/gemma-diffmean-9b-it-res
Updated

pyvene/gemma-reft-9b-it-res-generator
Updated

pyvene/gemma-reft-2b-it-res-generator
Updated

pyvene/reft_golden_gate_bridge_llama3
Updated
•
37

pyvene/reft_goody2_llama3
Updated
•
8

pyvene/reft_emoji_chat_llama3
Updated
•
4

pyvene/reft_emoji_chat
Updated
•
15
•
2
datasets
5
pyvene/axbench-concept16k_v2
Viewer
•
Updated
•
3.46M
•
23
pyvene/axbench-conceptFD
Viewer
•
Updated
•
5.33k
•
100
•
2
pyvene/axbench-concept16k
Viewer
•
Updated
•
2.27M
•
179
•
3
pyvene/axbench-concept500
Viewer
•
Updated
•
297k
•
173
•
1
pyvene/axbench-concept10
Viewer
•
Updated
•
6.8k
•
253
•
1