diff --git a/docs_nnx/nnx_basics.ipynb b/docs_nnx/nnx_basics.ipynb index ccacf2b58..458184b8c 100644 --- a/docs_nnx/nnx_basics.ipynb +++ b/docs_nnx/nnx_basics.ipynb @@ -453,7 +453,7 @@ "\n", "The Flax NNX Functional API establishes a clear boundary between reference/object semantics and value/pytree semantics. It also allows the same amount of fine-grained control over the state that Flax Linen and Haiku users are used to. The Flax NNX Functional API consists of three basic methods: `nnx.split`, `nnx.merge`, and `nnx.update`.\n", "\n", - "Below is an example of of `StatefulLinear` `Module` that uses the Functional API. It contains:\n", + "Below is an example of `StatefulLinear` `Module` that uses the Functional API. It contains:\n", "\n", "- Some `Param` Variables; and\n", "- A custom `Count` Variable type, which is used to track the integer scalar state that increases on every forward pass." diff --git a/docs_nnx/nnx_basics.md b/docs_nnx/nnx_basics.md index 81bf5bcc3..f29074de7 100644 --- a/docs_nnx/nnx_basics.md +++ b/docs_nnx/nnx_basics.md @@ -224,7 +224,7 @@ How do Flax NNX transforms achieve this? To understand how Flax NNX objects inte The Flax NNX Functional API establishes a clear boundary between reference/object semantics and value/pytree semantics. It also allows the same amount of fine-grained control over the state that Flax Linen and Haiku users are used to. The Flax NNX Functional API consists of three basic methods: `nnx.split`, `nnx.merge`, and `nnx.update`. -Below is an example of of `StatefulLinear` `Module` that uses the Functional API. It contains: +Below is an example of `StatefulLinear` `Module` that uses the Functional API. It contains: - Some `Param` Variables; and - A custom `Count` Variable type, which is used to track the integer scalar state that increases on every forward pass.