NativeLayerNormΒΆ

Native layer normalization.

Abstract Signature:

NativeLayerNorm(input: Tensor, normalized_shape: List[int], weight: Tensor, bias: Tensor, eps: float)

PyTorch

API: torch.native_layer_norm
Strategy: Direct Mapping

JAX (Core)

API: jax.nn.layer_norm
Strategy: Direct Mapping

Keras

API: keras.ops.layer_norm
Strategy: Direct Mapping