From f32f081be94dfd622209651562359248921ba404 Mon Sep 17 00:00:00 2001
From: Wessel Bruinsma <wessel.p.bruinsma@gmail.com>
Date: Sun, 2 Jun 2024 18:06:02 +0200
Subject: [PATCH] Refer to @DrJonnyT's gist

---
 docs/advanced_usage.md | 6 ++++++
 docs/basic_usage.md    | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/docs/advanced_usage.md b/docs/advanced_usage.md
index 99fcaf32..35f910b5 100644
--- a/docs/advanced_usage.md
+++ b/docs/advanced_usage.md
@@ -116,3 +116,9 @@ xc_grid_merged, yc_grid_merged = nps.merge_contexts(
     B.concat(xt, xt, axis=0)
 )
 ```
+
+## Equivalence of PyTorch and TensorFlow Architectures
+
+Jonny Taylor has a created a very helpful [Gist](https://gist.github.com/DrJonnyT/c946044591fb4ce922b0f5e7fd0f047a)
+which can be used to verify the equivalence of PyTorch and TensorFlow versions of architectures.
+
diff --git a/docs/basic_usage.md b/docs/basic_usage.md
index a29f85d6..0280d2a5 100644
--- a/docs/basic_usage.md
+++ b/docs/basic_usage.md
@@ -189,4 +189,4 @@ print(dist.logpdf(B.randn(tf.float32, 16, 3, 15)))
 print(dist.sample())
 print(dist.kl(dist))
 print(dist.entropy())
-```
\ No newline at end of file
+```