diff --git a/flax/linen/module.py b/flax/linen/module.py index 8fb4497d9c..31c955ac54 100644 --- a/flax/linen/module.py +++ b/flax/linen/module.py @@ -508,7 +508,10 @@ def nowrap(fun: _CallableT) -> _CallableT: - if you're subclassing a method like Module.param and don't want this overriden core function decorated with the state management wrapper. - If you want a method to be callable from an unbound Module (e.g.: a - function of construction of arguments that doesn't depend on params/RNGs) + function of construction of arguments that doesn't depend on params/RNGs). + If you want to learn more about how Flax Modules manage their state read the + [The Flax Module lifecycle](https://flax.readthedocs.io/en/latest/developer_notes/module_lifecycle.html) + guide. For instance::