flax
4721a6e5 - Update flax/nnx/nn/normalization.py

Commit

75 days ago

Update flax/nnx/nn/normalization.py

References

#4623 - Implemented spectral norm in NNX

Author

vfdev-5

vfdev-5

Parents

Loading