From 807e80d75484575cebc33515e57e71bd0d7846de Mon Sep 17 00:00:00 2001
From: Daniel Johnson <ddjohnson@google.com>
Date: Wed, 7 Aug 2024 14:25:10 -0700
Subject: [PATCH] Add citation info for Penzai/Treescope paper.

PiperOrigin-RevId: 660538262
---
 README.md      | 14 ++++++++++++++
 docs/index.rst | 16 ++++++++++++++++
 2 files changed, 30 insertions(+)

diff --git a/README.md b/README.md
index a7c623b..2b67d9c 100644
--- 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
index 6389cc2..0a55b14 100644
--- 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
 -------