flax
relaxed `grads` kwarg for `Optimizer.update`
#3818

Merged

relaxed `grads` kwarg for `Optimizer.update` #3818

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

relaxed kwarg for

bb959624

chiamp assigned

chiamp 1 year ago

chiamp requested a review from

cgarciae 1 year ago

cgarciae approved these changes on 2024-04-04

chiamp added pull ready

copybara-service merged 6b03439c into main 1 year ago

chiamp deleted the optimizer branch 1 year ago

Reviewers

cgarciae

Assignees

chiamp

Labels

pull ready

Milestone

No milestone