Add citation info for Penzai/Treescope paper.

PiperOrigin-RevId: 660538262
google-deepmind · Aug 7, 2024 · 807e80d · 807e80d
1 parent e52547b
commit 807e80d
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -95,6 +95,20 @@ Looking for a neural network library with first-class support for Treescope's
 visualization features?
 Try [Penzai](https://penzai.readthedocs.io/en/stable)!
 
+## Citation
+
+If you have found Treescope to be useful for your research, please consider
+citing the following writeup (also avaliable on [arXiv](https://arxiv.org/abs/2408.00211)):
+
+```
+@article{johnson2024penzai,
+    author={Daniel D. Johnson},
+    title={{Penzai} + {Treescope}: A Toolkit for Interpreting, Visualizing, and Editing Models As Data},
+    year={2024},
+    journal={ICML 2024 Workshop on Mechanistic Interpretability}
+}
+```
+
 ---
 
 *This is not an officially supported Google product.*
diff --git a/docs/index.rst b/docs/index.rst
@@ -125,6 +125,22 @@ Try `Penzai <https://penzai.readthedocs.io/en/stable>`_!
    api/treescope.type_registries
 
 
+Citation
+--------
+
+If you have found Treescope to be useful for your research, please consider
+citing the following writeup (also avaliable on `arXiv <https://arxiv.org/abs/2408.00211>`_):
+
+.. code-block:: none
+
+  @article{johnson2024penzai,
+      author={Daniel D. Johnson},
+      title={{Penzai} + {Treescope}: A Toolkit for Interpreting, Visualizing, and Editing Models As Data},
+      year={2024},
+      journal={ICML 2024 Workshop on Mechanistic Interpretability}
+  }
+
+
 License
 -------