Class CompiledGatedRMSNorm

constructor

• new CompiledGatedRMSNorm(weight: MlxArray, eps: number): CompiledGatedRMSNorm

  Parameters

  • weight: MlxArray
  • eps: number

  Returns CompiledGatedRMSNorm

Readonlyeps

Readonlyweight

forward

• forward(x: MlxArray, gate: MlxArray): MlxArray

  Parameters

  • x: MlxArray
  • gate: MlxArray

  Returns MlxArray