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

Commit

1 year ago

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

References

Author

ywrt

Parents