PR #4623 Implemented spectral norm in NNX

Implemented spectral norm in NNX #4623

copybara-service merged 2 commits into google:main from mattbahr:implement-spectral-and-instance-norm

mattbahr force pushed from 98a5d0a8 to d1b914f1 330 days ago

mattbahr force pushed from d1b914f1 to 32d8f2de 327 days ago

mattbahr force pushed from 32d8f2de to d40a1d9e 324 days ago

mattbahr force pushed from d40a1d9e to d53c9fe2 303 days ago

vfdev-5 commented on 2025-09-10

vfdev-5 force pushed from 7605b709 to fa2cc840 149 days ago

vfdev-5 marked this pull request as draft 149 days ago

vfdev-5 changed the title ~~Implement spectral and instance norm in NNX~~ Implemented spectral norm in NNX 149 days ago

vfdev-5 force pushed from fa2cc840 to d36a7874 149 days ago

vfdev-5 force pushed from d36a7874 to 83e73efa 145 days ago

vfdev-5 force pushed from 83e73efa to 4cb7c1d4 145 days ago

vfdev-5 marked this pull request as ready for review 145 days ago

vfdev-5 approved these changes on 2025-09-16

vfdev-5 force pushed from 4cb7c1d4 to eaeb0b8c 145 days ago

vfdev-5 force pushed from eaeb0b8c to 5d4eccd2 145 days ago

vfdev-5 commented on 2025-09-16

vfdev-5 commented on 2025-09-17

vfdev-5 force pushed from 5d4eccd2 to e5000435 143 days ago

vfdev-5 force pushed from e5000435 to e340b1d8 143 days ago

vfdev-5 force pushed from e340b1d8 to 164a7f22 139 days ago

vfdev-5 requested a review from

cgarciae 139 days ago

cgarciae approved these changes on 2025-09-29

cgarciae added pull ready

cgarciae commented on 2025-09-29

vfdev-5 commented on 2025-09-29

Improved nnx SpectralNorm implementation

77a8c4ca

vfdev-5 force pushed from 164a7f22 to 77a8c4ca 132 days ago

vfdev-5 requested a review from

vfdev-5 132 days ago

vfdev-5 approved these changes on 2025-09-29

vfdev-5 removed pull ready

vfdev-5 added pull ready

cgarciae approved these changes on 2025-09-29

vfdev-5 commented on 2025-10-01

Update flax/nnx/nn/normalization.py

4721a6e5

cgarciae removed pull ready

cgarciae added pull ready

copybara-service merged 8025b3dd into main 130 days ago

Reviewers

vfdev-5

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone

flax Implemented spectral norm in NNX #4623 Merged

Implemented spectral norm in NNX #4623

flax
Implemented spectral norm in NNX
#4623

Merged