unit\_scaling.functional.layer\_norm ==================================== .. currentmodule:: unit_scaling.functional .. autofunction:: layer_norm