flax
Makes `linen.Partitioned` compatible with `linen.WeightNorm`
#5234
Merged

Makes `linen.Partitioned` compatible with `linen.WeightNorm` #5234

copybara-service merged 1 commit into main from test_867835686
copybara-service
copybara-service copybara-service force pushed from fb84719f to df51c3bb 127 days ago
copybara-service copybara-service changed the title Allows `linen.Partitioned` to be supported in `linen.WeightNorm` Makes `linen.Partitioned` compatible with `linen.WeightNorm` 127 days ago
jimlinntu Makes `linen.Partitioned` compatible with `linen.WeightNorm`
7814e6c5
copybara-service copybara-service force pushed from df51c3bb to 7814e6c5 127 days ago
copybara-service copybara-service merged 7814e6c5 into main 127 days ago
copybara-service copybara-service deleted the test_867835686 branch 127 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone