PR #3983 added `wrt` option to `nnx.Optimizer`

added `wrt` option to `nnx.Optimizer` #3983

copybara-service merged 1 commit into google:main from chiamp:optimizer

chiamp assigned

chiamp 1 year ago

chiamp marked this pull request as draft 1 year ago

chiamp changed the title ~~added wrt option to nnx.Optimizer.update~~ added `wrt` option to `nnx.Optimizer.update` 1 year ago

chiamp force pushed 1 year ago

chiamp force pushed 1 year ago

chiamp marked this pull request as ready for review 1 year ago

chiamp changed the title ~~added `wrt` option to `nnx.Optimizer.update`~~ added `wrt` option to `nnx.Optimizer` 1 year ago

chiamp requested a review from

cgarciae 1 year ago

cgarciae approved these changes on 2024-06-12

added wrt option to nnx.Optimizer.update

9a40b3fa

chiamp force pushed to 9a40b3fa 1 year ago

chiamp added pull ready

copybara-service merged 56216fa6 into main 1 year ago

chiamp deleted the optimizer branch 1 year ago

Reviewers

cgarciae

Assignees

chiamp

Labels

pull ready

Milestone

No milestone