LayerNorm

class cuequivariance_jax.flax_linen.LayerNorm(

epsilon: float = 0.01,

parent: Union[flax.linen.module.Module,

flax.core.scope.Scope,

flax.linen.module._Sentinel,

NoneType] = <flax.linen.module._Sentinel object at 0x7f42d24e43e0>,

name: Optional[str] = None,

)