Class GemmaMlpWeights

Stored MLP projections. Each operation view names its activation literally.

Hierarchy (View Summary)

GegluMLP
- GemmaMlpWeights

Index

Constructors

Properties

compiled down gate up

Methods

Constructors

constructor

new GemmaMlpWeights(
    weights: Weights,
    config: ModelConfig,
    prefix: string,
): GemmaMlpWeights
Parameters
- weights: Weights
- config: ModelConfig
- prefix: string
Returns GemmaMlpWeights
Overrides GegluMLP.constructor
- Defined in packages/inference/src/models/gemma4/written-blocks.ts:35

Properties

`Readonly`compiled

compiled: boolean

`Readonly`down

down: QuantizedLinear

`Readonly`gate

gate: QuantizedLinear

`Readonly`up

up: QuantizedLinear

Methods

forward

forward(x: MlxArray): MlxArray
Parameters
- x: MlxArray
Returns MlxArray
Inherited from GegluMLP.forward
- Defined in packages/inference/src/layers/geglu.ts:74