Class DenseEmbedding

Plain unquantized embedding — mlx nn.Embedding (+ as_linear tied head).

Index

Constructors

Properties

Methods

asLinear encode

Constructors

constructor

new DenseEmbedding(w: MlxArray): DenseEmbedding
Parameters
- w: MlxArray
Returns DenseEmbedding
- Defined in packages/inference/src/layers/dense-embedding.ts:17

Properties

`Readonly`w

`Readonly`wT

Methods

asLinear

asLinear(h: MlxArray): MlxArray
nn.Embedding.as_linear: x @ weight.T.
Parameters
- h: MlxArray
Returns MlxArray
- Defined in packages/inference/src/layers/dense-embedding.ts:26

encode

encode(ids: MlxArray): MlxArray
Parameters
- ids: MlxArray
Returns MlxArray
- Defined in packages/inference/src/layers/dense-embedding.ts:21