flax
Update optimizer.py to support masked variable from optax.
#4904

Merged

Update optimizer.py to support masked variable from optax. #4904

copybara-service merged 2 commits into google:main from ywrt:patch-1

Update optimizer.py to support masked variable from optax.

2f213b54

8bitmp3 requested a review from

cgarciae 167 days ago

8bitmp3 requested a review from

IvyZX 167 days ago

ywrt closed this 167 days ago

cgarciae reopened this 164 days ago

Merge branch 'main' of github.com:google/flax into patch-1

fb648ecb

vfdev-5 approved these changes on 2025-08-29

vfdev-5 added pull ready

copybara-service merged 49cfa78c into main 156 days ago

Reviewers

vfdev-5

cgarciae

IvyZX

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone