flax
49cfa78c - Merge pull request #4904 from ywrt:patch-1

Commit

157 days ago

Merge pull request #4904 from ywrt:patch-1 PiperOrigin-RevId: 802149667

References

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

Author

a-googler

a-googler

Parents

Loading