Update README.md

README.md

@@ -1,3 +1,34 @@
----
-license: cc-by-4.0
----
+---
+license: cc-by-4.0
+---
+
+# 1. AxBench
+
+AxBench evaluates interpretability methods in terms of concept detection and model steering. AxBench releases two supervised dictionary learning methods that outperforms existing methods including SAEs. These dictionaries contain 1D subspaces that map to high-level concepts.
+
+# 2. What is `gemma-diffmean-9b-it-res`?
+
+- `gemma-`: Refer to Gemma 2 models
+- `diffmean-` : The dictionary learning model is taking the difference in mean between two contrastive groups.
+- `9b-it-`: The dictionary is for Gemma 2 9B instruction-tuning model
+- `res`   : The dictionary is trained on the model's residual stream.
+- We release the weights as well as the annotated concepts for all subspaces.
+
+# 3. How can I use these dictionaries straight away?
+
+```python
+import pyvene as pv
+
+```
+
+# 4. Point of Contact
+
+Point of contact: Zhengxuan Wu or Aryaman Arora
+
+Contact by email:
+
+{wuzhengx, aryamana}@stanford.edu
+
+# 5. Citation
+
+Paper: