LayerNorm

class cuequivariance_jax.flax_linen.LayerNorm

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 0x7fd252a1e750>, name: Optional[str] = None)

__init__(

epsilon=0.01,

parent=<flax.linen.module._Sentinel object>,

name=None,

)

Parameters:

• epsilon (float)

• parent (Module | Scope | _Sentinel | None)

• name (str | None)

Return type:

None