flax
Add intermediate value captures based off hijax.Box
#4925

Open

Add intermediate value captures based off hijax.Box #4925

IvyZX wants to merge 1 commit into google:main from IvyZX:capture

IvyZX requested a review from

cgarciae 122 days ago

cgarciae commented on 2025-09-08

IvyZX force pushed from 191c4f27 to e095e8a7 116 days ago

IvyZX force pushed from e095e8a7 to e87102b7 116 days ago

cgarciae commented on 2025-09-10

IvyZX force pushed from e87102b7 to 70e081dc 114 days ago

Add nnx.capture_intermediates based off hijax.Box

79cd1e78

IvyZX force pushed from 70e081dc to 79cd1e78 113 days ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

None yet

Milestone

No milestone