diff --git a/flax/nnx/module.py b/flax/nnx/module.py index a0eeb857c..ce693edf7 100644 --- a/flax/nnx/module.py +++ b/flax/nnx/module.py @@ -177,7 +177,7 @@ def sow( setattr(self, name, variable_type(reduced_value)) warnings.warn( """Using 'Module.sow()' outside of 'nnx.capture()' is deprecated; see - https://flax.readthedocs.io/en/stable/capturing_intermediates.html for more information. + https://flax.readthedocs.io/en/latest/guides/extracting_intermediates.html for more information. """, DeprecationWarning, stacklevel=2, @@ -270,7 +270,7 @@ def perturb( setattr(self, name, variable_type(old_value)) warnings.warn(""" Using 'Module.perturb()' outside of 'nnx.capture()' is deprecated; see - https://flax.readthedocs.io/en/stable/capturing_intermediates.html for more information. + https://flax.readthedocs.io/en/latest/guides/extracting_intermediates.html for more information. """, DeprecationWarning, stacklevel=2)