GroupNorm ========= Applies Group Normalization. **Abstract Signature:** ``GroupNorm(input: Tensor, num_groups: int, weight: Optional[Tensor], bias: Optional[Tensor], eps: float = 1e-05)`` .. raw:: html

PyTorch

API: torch.group_norm
Strategy: Direct Mapping

JAX (Core)

API: flax.linen.GroupNorm
Strategy: Plugin (functional_layer_wrapper)

Keras

API: keras.layers.GroupNormalization
Strategy: Direct Mapping

TensorFlow

API: tf.nn.group_norm
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.GroupNorm
Strategy: Plugin (mlx_layer_wrapper)

Flax NNX

API: flax.nnx.GroupNorm
Strategy: Direct Mapping

PaxML / Praxis

API: paxml.layers.GroupNorm
Strategy: Direct Mapping