flax
Update pytree guide to use nnx.capture
#5370
Merged

Loading