Refer to @DrJonnyT's gist

wesselb · Jun 2, 2024 · f32f081 · f32f081
1 parent 14c2bed
commit f32f081
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -189,4 +189,4 @@ print(dist.logpdf(B.randn(tf.float32, 16, 3, 15)))
 print(dist.sample())
 print(dist.kl(dist))
 print(dist.entropy())
-```
+```