.. role:: hidden :class: hidden-section apex.normalization.fused_layer_norm =================================== .. automodule:: apex.normalization .. currentmodule:: apex.normalization .. FusedAdam ---------- .. autoclass:: FusedLayerNorm :members: .. autoclass:: FusedRMSNorm :members: