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

Merged

Commits

Update optimizer.py to support masked variable from optax.

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

vfdev-5 committed 1 year ago

Loading