Skip to content

Commit

Permalink
Merge pull request #4047 from rajasekharporeddy:main
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 648833213
  • Loading branch information
Flax Authors committed Jul 2, 2024
2 parents 37123d5 + 956ad6b commit 3112da6
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion docs/nnx/nnx_basics.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@
"\n",
"To actually initialize a Module you simply call the constructor, all the parameters \n",
"of a Module are usually created eagerly. Since Modules hold their own state methods \n",
"can be called directly without the no need for a separate `apply` method, this is very \n",
"can be called directly without the need for a separate `apply` method, this is very \n",
"convenient for debugging as entire structure of the model can be inspected directly."
]
},
Expand Down
2 changes: 1 addition & 1 deletion docs/nnx/nnx_basics.md
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ inner value is a JAX array).

To actually initialize a Module you simply call the constructor, all the parameters
of a Module are usually created eagerly. Since Modules hold their own state methods
can be called directly without the no need for a separate `apply` method, this is very
can be called directly without the need for a separate `apply` method, this is very
convenient for debugging as entire structure of the model can be inspected directly.

```{code-cell} ipython3
Expand Down

0 comments on commit 3112da6

Please sign in to comment.