GroupNorm

Applies Group Normalization.

Abstract Signature:

GroupNorm(input: Tensor, num_groups: int, weight: Optional[Tensor], bias: Optional[Tensor], eps: float = 1e-05)

PyTorchJAX (Core)KerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.group_norm

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: flax.linen.GroupNorm

Strategy: Plugin (functional_layer_wrapper)

Keras

API: keras.layers.GroupNormalization

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.nn.group_norm

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.GroupNorm

Strategy: Plugin (mlx_layer_wrapper)

Flax NNX

API: flax.nnx.GroupNorm

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: paxml.layers.GroupNorm

Strategy: Direct Mapping

Official Docs ↗