NormalizationΒΆ

A preprocessing layer that normalizes continuous features.

Abstract Signature:

Normalization(axis: int = -1, mean: float, variance: float)

PyTorch

API: β€”
Strategy: Custom / Partial

Keras

API: keras.layers.Normalization
Strategy: Direct Mapping

TensorFlow

API: tf.keras.layers.Normalization
Strategy: Direct Mapping

Apple MLX

API: β€”
Strategy: Custom / Partial

Flax NNX

API: flax.nnx.LayerNorm
Strategy: Direct Mapping

PaxML / Praxis

API: praxis.layers.LayerNorm
Strategy: Direct Mapping