Add trimmed Linen to NNX guide

google · Sep 19, 2024 · fe2e586 · fe2e586
1 parent e356acb
commit fe2e586
Show file tree

Hide file tree

Showing 4 changed files with 753 additions and 3 deletions.
diff --git a/docs_nnx/guides/haiku_linen_vs_nnx.rst → docs_nnx/guides/haiku_to_flax.rst b/docs_nnx/guides/haiku_linen_vs_nnx.rst → docs_nnx/guides/haiku_to_flax.rst
@@ -1,5 +1,5 @@
 
-Migrating from Haiku/Linen to NNX
+Migrating from Haiku to Flax
 =================================
 
 This guide will showcase the differences between Haiku, Flax Linen and Flax NNX.

diff --git a/docs_nnx/guides/index.rst b/docs_nnx/guides/index.rst
@@ -5,8 +5,9 @@ Guides
    :maxdepth: 2
 
    filters_guide
-   haiku_linen_vs_nnx
+   linen_to_nnx
    bridge_guide
    surgery
    graph_mutations
-   jax_and_nnx_transforms
+   jax_and_nnx_transforms
+   haiku_to_flax