flax
7814e6c5 - Makes `linen.Partitioned` compatible with `linen.WeightNorm`

Commit

110 days ago

Makes `linen.Partitioned` compatible with `linen.WeightNorm` PiperOrigin-RevId: 868199670

References

Author

jimlinntu

Committer

a-googler

Parents