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 2 years ago

chiamp marked this pull request as draft 2 years ago

chiamp changed the title ~~added wrt option to nnx.Optimizer.update~~ added `wrt` option to `nnx.Optimizer.update` 2 years ago

chiamp force pushed 2 years ago

chiamp force pushed 2 years ago

chiamp marked this pull request as ready for review 2 years ago

chiamp changed the title ~~added `wrt` option to `nnx.Optimizer.update`~~ added `wrt` option to `nnx.Optimizer` 2 years ago

chiamp requested a review from

cgarciae 2 years ago

cgarciae approved these changes on 2024-06-12

added wrt option to nnx.Optimizer.update

9a40b3fa

chiamp force pushed to 9a40b3fa 2 years ago

chiamp added pull ready

copybara-service merged 56216fa6 into main 2 years ago

chiamp deleted the optimizer branch 2 years ago

Reviewers

cgarciae

Assignees

chiamp

Labels

pull ready

Milestone

No milestone