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 1 year ago

mattbahr force pushed from d1b914f1 to 32d8f2de 1 year ago

mattbahr force pushed from 32d8f2de to d40a1d9e 1 year ago

mattbahr force pushed from d40a1d9e to d53c9fe2 1 year ago

vfdev-5 commented on 2025-09-10

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

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

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

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

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

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

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

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

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

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

vfdev-5 commented on 2025-09-16

vfdev-5 commented on 2025-09-17

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

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

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

vfdev-5 requested a review from

cgarciae 246 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 238 days ago

vfdev-5 requested a review from

vfdev-5 238 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 236 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