flax
2f213b54 - Update optimizer.py to support masked variable from optax.

Commit

137 days ago

Update optimizer.py to support masked variable from optax. see https://github.com/google/flax/issues/4901

References

Author

ywrt

Parents