Merge pull request #4770 from IvyZX:fix-links

PiperOrigin-RevId: 767766488

Author: Flax Authors

docs_nnx/index.rst

@@ -63,7 +63,7 @@ Features
          .. div:: sd-font-normal
 
             Flax NNX allows fine-grained control of the model's state via
-            its `Filter <https://flax.readthedocs.io/en/latest/nnx/filters_guide.html>`__
+            its `Filter <https://flax.readthedocs.io/en/latest/guides/filters_guide.html>`__
             system.
 
    .. grid-item::

flax/nnx/transforms/iteration.py

@@ -236,7 +236,7 @@ def vmap(
       addition to integers and None, :class:`StateAxes`  can be used to control
       how graph nodes like Modules are vectorized by specifying the axes to be
       applied to substates of the graph node given a `Filter
-      <https://flax.readthedocs.io/en/latest/nnx/filters_guide.html>`__.
+      <https://flax.readthedocs.io/en/latest/guides/filters_guide.html>`__.
     out_axes: An integer, None, or pytree indicating where the mapped axis
       should appear in the output (see `jax.vmap
       <https://jax.readthedocs.io/en/latest/_autosummary/jax.vmap.html>`__).
@@ -454,7 +454,7 @@ def pmap(
       addition to integers and None, :class:`StateAxes`  can be used to control
       how graph nodes like Modules are vectorized by specifying the axes to be
       applied to substates of the graph node given a `Filter
-      <https://flax.readthedocs.io/en/latest/nnx/filters_guide.html>`__.
+      <https://flax.readthedocs.io/en/latest/guides/filters_guide.html>`__.
     out_axes: An integer, None, or pytree indicating where the mapped axis
       should appear in the output (see `jax.vmap
       <https://jax.readthedocs.io/en/latest/_autosummary/jax.vmap.html>`__).