flax
Add intermediate value captures (extends #4925)
#5257

Open

Add intermediate value captures (extends #4925) #5257

samanklesaria wants to merge 3 commits into google:main from samanklesaria:capture_intms

gemini-code-assist commented on 2026-02-16

samanklesaria force pushed from e94997bf to 195c1442 17 days ago

samanklesaria force pushed from 195c1442 to 86b57f21 17 days ago

samanklesaria requested a review from

cgarciae 17 days ago

samanklesaria force pushed from 79c10e01 to 24c63cd7 17 days ago

samanklesaria force pushed from a2202705 to 7b171c05 10 days ago

samanklesaria force pushed from 7b171c05 to fe5e00be 10 days ago

samanklesaria force pushed from fe5e00be to 95b4602d 6 days ago

samanklesaria changed the title ~~Add intermediate value captures based on hijax.Box (simplification of #4925)~~ Add intermediate value captures (extends #4925) 6 days ago

samanklesaria force pushed from 95b4602d to db322100 6 days ago

samanklesaria force pushed from db322100 to 4a185ab4 6 days ago

cgarciae commented on 2026-02-27

samanklesaria force pushed from f52dfca0 to f38d4270 6 days ago

samanklesaria force pushed from f38d4270 to a6fb4250 6 days ago

samanklesaria requested a review from

cgarciae 6 days ago

samanklesaria force pushed from a6fb4250 to 5f0d55d5 2 days ago

cgarciae commented on 2026-03-03

cgarciae commented on 2026-03-03

samanklesaria requested a review from

cgarciae 2 days ago

samanklesaria force pushed from 068bc8c0 to de2b7274 1 day ago

samanklesaria force pushed from de2b7274 to 04fa62f1 1 day ago

samanklesaria force pushed from 04fa62f1 to 9f1f49af 1 day ago

samanklesaria force pushed from 9f1f49af to 22bdf6b5 1 day ago

samanklesaria force pushed from 22bdf6b5 to 7153e452 1 day ago

samanklesaria force pushed from 7153e452 to d831a6bd 1 day ago

samanklesaria force pushed from d831a6bd to 41a29096 1 day ago

Add capture_intermediates API

d827f8d1

samanklesaria force pushed from 41a29096 to d827f8d1 1 day ago

Bracket graph mode calls

e063c9cd

Make Module.sow defer to existing attributes

1b9c7505

samanklesaria force pushed from c4c70860 to 1b9c7505 1 day ago

Reviewers

cgarciae

gemini-code-assist

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

flax Add intermediate value captures (extends #4925) #5257 Open

Add intermediate value captures (extends #4925) #5257

flax
Add intermediate value captures (extends #4925)
#5257

Open