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

Loading