Merge pull request #4209 from IvyZX:linen-nnx-guide

PiperOrigin-RevId: 677141281
google · Sep 21, 2024 · 268366f · 268366f
2 parents e356acb + fe2e586
commit 268366f
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