Update pytree guide to use nnx.capture#5370

samanklesaria wants to merge 1 commit intogoogle:maingoogle/flax:mainfrom

samanklesaria:pytree_guide_capturesamanklesaria/flax:pytree_guide_captureCopy head branch name to clipboard