Normalization layers implementation
This module provides implementations of various normalization techniques such as Layer Normalization, Batch Normalization, etc.