Skip to content

Update pytree guide to use nnx.capture#5370

Open
samanklesaria wants to merge 1 commit intogoogle:mainfrom
samanklesaria:pytree_guide_capture
Open

Update pytree guide to use nnx.capture#5370
samanklesaria wants to merge 1 commit intogoogle:mainfrom
samanklesaria:pytree_guide_capture

Conversation

@samanklesaria
Copy link
Collaborator

The pytree guide currently demonstrates using the deprecated Module.sow() with nnx.pop pattern. This PR updates it to use nnx.capture.

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

1 participant