NNX WeightNorm #4426
Labels
Priority: P2 - no schedule
Best effort response and resolution. We have no plan to work on this at the moment.
Status: pull requests welcome
We agree with the direction proposed, feel free to give it a shot and file a pull request
Is there an NNX implementation of the WeightNorm module as used in Linen? I can't seem to find it in the normalization layers implemented in NNX.
I'm in the process of writing a new model in NNX which would require them.
The text was updated successfully, but these errors were encountered: