Class LayerNorm

mlx nn.LayerNorm (weight + bias affine — starcoder2's norm).

Index

Constructors

Properties

bias eps weight

Methods

Constructors

constructor

new LayerNorm(weight: MlxArray, bias: MlxArray | null, eps: number): LayerNorm
Parameters
- weight: MlxArray
- bias: MlxArray | null
- eps: number
Returns LayerNorm
- Defined in packages/inference/src/layers/layer-norm.ts:15

Properties

`Readonly`bias

bias: MlxArray | null

`Readonly`eps

eps: number

`Readonly`weight

weight: MlxArray

Methods

forward

forward(x: MlxArray): MlxArray
Parameters
- x: MlxArray
Returns MlxArray
- Defined in packages/inference/src/layers/layer-norm.ts:21