From 3ef9b2a7ac3a92528b283821fb72d28ca0901d8a Mon Sep 17 00:00:00 2001 From: gileshd Date: Tue, 16 Jul 2024 14:15:50 +0100 Subject: [PATCH] Fix broken link to NNX Basics in the docs --- docs/nnx/haiku_linen_vs_nnx.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/nnx/haiku_linen_vs_nnx.rst b/docs/nnx/haiku_linen_vs_nnx.rst index 9f07e5e34d..571d2cd01a 100644 --- a/docs/nnx/haiku_linen_vs_nnx.rst +++ b/docs/nnx/haiku_linen_vs_nnx.rst @@ -199,7 +199,7 @@ simply inspect the ``params`` object returned from ``.init()``. To see the parameter structure in NNX, the user can call ``nnx.split`` to generate ``Graphdef`` and ``State`` objects. The ``Graphdef`` is a static pytree denoting the structure of the model (for example usages, see -`NNX Basics `__). +`NNX Basics `__). ``State`` objects contains all the module variables (i.e. any class that sub-classes ``nnx.Variable``). If we filter for ``nnx.Param``, we will generate a ``State`` object of all the learnable module parameters.