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