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

Commit

160 days ago

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

References

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

Author

vfdev-5

vfdev-5

Parents

Loading