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 282 days ago

mattbahr force pushed from d1b914f1 to 32d8f2de 278 days ago

mattbahr force pushed from 32d8f2de to d40a1d9e 276 days ago

mattbahr force pushed from d40a1d9e to d53c9fe2 255 days ago

vfdev-5 commented on 2025-09-10

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

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

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

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

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

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

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

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

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

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

vfdev-5 commented on 2025-09-16

vfdev-5 commented on 2025-09-17

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

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

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

vfdev-5 requested a review from

cgarciae 91 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 83 days ago

vfdev-5 requested a review from

vfdev-5 83 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 81 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