Class TrellisFusedMLP

constructor

• new TrellisFusedMLP(
      gate: TrellisLinear,
      up: TrellisLinear,
      down: RowProjection,
      state?: LoraState,
  ): TrellisFusedMLP

  Parameters

  • gate: TrellisLinear
  • up: TrellisLinear
  • down: RowProjection
  • Optionalstate: LoraState

  Returns TrellisFusedMLP

Readonlydown

Readonlygate

Readonlyup

dispose

• dispose(): void

  Returns void

forward

• forward(x: MlxArray): MlxArray

  Parameters

  • x: MlxArray

  Returns MlxArray