Skip to content

Handle captures in nnx.scan#5515

Open
samanklesaria wants to merge 2 commits into
google:mainfrom
samanklesaria:handle-captures-scan
Open

Handle captures in nnx.scan#5515
samanklesaria wants to merge 2 commits into
google:mainfrom
samanklesaria:handle-captures-scan

Conversation

@samanklesaria

@samanklesaria samanklesaria commented Jun 25, 2026

Copy link
Copy Markdown
Collaborator

This PR allows nnx.scan to transform closures that mutate their closed over values. This PR depends on #5490 and should only be merged after it. This PR incorporate's that PR's changes, because there isn't a way to set the original PR as the base branch in github.

@samanklesaria samanklesaria requested a review from cgarciae June 25, 2026 18:53
@samanklesaria samanklesaria force-pushed the handle-captures-scan branch 2 times, most recently from a1dd22e to 760826a Compare June 30, 2026 15:58
@samanklesaria samanklesaria marked this pull request as ready for review June 30, 2026 16:00
@samanklesaria samanklesaria force-pushed the handle-captures-scan branch 5 times, most recently from 494cf0d to e6c5179 Compare July 2, 2026 21:09
@samanklesaria samanklesaria force-pushed the handle-captures-scan branch from e6c5179 to 4aaf86c Compare July 7, 2026 04:03
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant