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