WeightNorm

L2 weight normalization.

Abstract Signature:

WeightNorm(layer)

PyTorch

API: torch.nn.utils.weight_norm
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.keras.layers.WeightNormalization
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.WeightNorm
Strategy: Direct Mapping
Official Docs ↗