flax
Return updates PyTree from nnx.Optimizer.update
#5214

Merged

Return updates PyTree from nnx.Optimizer.update #5214

copybara-service merged 1 commit into google:main from anshuljoshi:optimizer-return-updates

Return updates PyTree from nnx.Optimizer.update

fb5184c3

gemini-code-assist commented on 2026-01-28

cgarciae approved these changes on 2026-03-31

cgarciae added pull ready

copybara-service merged 66bed3ab into main 11 days ago

anshuljoshi deleted the optimizer-return-updates branch 11 days ago

Reviewers

cgarciae

gemini-code-assist

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone