MCPcopy Create free account

hub / github.com/ml-explore/mlx-lm / functions

Functions2,622 in github.com/ml-explore/mlx-lm

⨍ Functions 2,622◇ Types & classes 816↳ Endpoints 9
↓ 1 callersFunctiontrain
( model, optimizer, train_dataset, val_dataset=None, args: TrainingArgs = TrainingArgs(),
mlx_lm/tuner/trainer.py:218
↓ 1 callersFunctiontrain_model
( args, model: nn.Module, train_set, valid_set, training_callback: TrainingCallback = None
mlx_lm/lora.py:216
↓ 1 callersFunctiontranslate_weight_names
(name)
mlx_lm/gguf.py:103
↓ 1 callersMethodtrim
(self, n)
mlx_lm/models/cache.py:824
↓ 1 callersFunctionupdate_config
( model: nn.Module, config: Dict[str, Any], )
mlx_lm/quant/awq.py:513
↓ 1 callersMethodvalidate_model_parameters
Validate that the passed model parameters have correct types and values.
mlx_lm/server.py:1229
↓ 1 callersFunctionwired_limit
A context manager to temporarily change the wired limit. Note, the wired limit should not be changed during an async eval. If an async
mlx_lm/generate.py:230
↓ 1 callersFunctionwkv7_kernel
( r: mx.array, w: mx.array, k: mx.array, v: mx.array, a: mx.array, b: mx.array, st
mlx_lm/models/rwkv7.py:125
↓ 1 callersFunctionworker
()
benchmarks/server_benchmark.py:187
↓ 1 callersFunctionxielu
(x, alpha_p, alpha_n, beta, eps)
mlx_lm/models/activations.py:15
Method__call__
(self, tool_calls)
mlx_lm/server.py:74
Method__call__
(self, x)
mlx_lm/tuner/dora.py:111
Method__call__
(self, x)
mlx_lm/tuner/dora.py:198
Method__call__
(self, x)
mlx_lm/tuner/lora.py:95
Method__call__
(self, x, indices, sorted_indices=False)
mlx_lm/tuner/lora.py:181
Method__call__
(self, x)
mlx_lm/tuner/lora.py:276
Method__call__
(self, x, *args, **kwargs)
mlx_lm/quant/gptq.py:46
Method__call__
(self, x: mx.array, *args, **kwargs)
mlx_lm/quant/awq.py:431
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/telechat3.py:71
Method__call__
(self, x: mx.array)
mlx_lm/models/telechat3.py:116
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/telechat3.py:133
Method__call__
( self, inputs: mx.array, cache: Optional[Any] = None, input_embeddings: Optio
mlx_lm/models/telechat3.py:157
Method__call__
( self, inputs: mx.array, cache: Optional[Any] = None, input_embeddings: Optio
mlx_lm/models/telechat3.py:187
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/qwen2.py:57
Method__call__
(self, x)
mlx_lm/models/qwen2.py:94
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/qwen2.py:111
Method__call__
( self, inputs: mx.array, cache=None, input_embeddings: Optional[mx.array] = N
mlx_lm/models/qwen2.py:137
Method__call__
( self, inputs: mx.array, cache=None, input_embeddings: Optional[mx.array] = N
mlx_lm/models/qwen2.py:167
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/ernie4_5.py:54
Method__call__
(self, x)
mlx_lm/models/ernie4_5.py:90
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/ernie4_5.py:105
Method__call__
( self, inputs: mx.array, cache=None, )
mlx_lm/models/ernie4_5.py:124
Method__call__
( self, inputs: mx.array, cache=None, )
mlx_lm/models/ernie4_5.py:151
Method__call__
(self, x, gate)
mlx_lm/models/gpt_oss.py:66
Method__call__
(self, x: mx.array, mask: mx.array, cache=None)
mlx_lm/models/gpt_oss.py:106
Method__call__
(self, x: mx.array)
mlx_lm/models/gpt_oss.py:148
Method__call__
(self, x: mx.array, mask: mx.array, cache=None)
mlx_lm/models/gpt_oss.py:179
Method__call__
( self, inputs: mx.array, cache=None, input_embeddings: Optional[mx.array] = N
mlx_lm/models/gpt_oss.py:206
Method__call__
(self, inputs: mx.array, cache=None)
mlx_lm/models/gpt_oss.py:240
Method__call__
( self, inputs: mx.array, cache: Optional[Any] = None, )
mlx_lm/models/kimi_k25.py:33
Method__call__
( self, inputs: mx.array, cache: Optional[Any] = None, )
mlx_lm/models/kimi_k25.py:49
Method__call__
(self, x, mask=None, cache=None)
mlx_lm/models/phixtral.py:47
Method__call__
(self, x: mx.array)
mlx_lm/models/phixtral.py:99
Method__call__
(self, x, mask, cache)
mlx_lm/models/phixtral.py:122
Method__call__
(self, x, mask, cache)
mlx_lm/models/phixtral.py:135
Method__call__
(self, x)
mlx_lm/models/phixtral.py:150
Method__call__
(self, inputs)
mlx_lm/models/phixtral.py:160
Method__call__
( self, x: mx.array, mask: mx.array = None, cache=None, )
mlx_lm/models/phixtral.py:172
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
mlx_lm/models/cohere2.py:61
Method__call__
(self, x)
mlx_lm/models/cohere2.py:109
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
mlx_lm/models/cohere2.py:126
Method__call__
( self, inputs: mx.array, cache=None, )
mlx_lm/models/cohere2.py:157
Method__call__
( self, inputs: mx.array, cache=None, )
mlx_lm/models/cohere2.py:191
Method__call__
( self, hidden_states: mx.array, gate: mx.array | None = None )
mlx_lm/models/qwen3_next.py:71
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/qwen3_next.py:121
Method__call__
(self, x)
mlx_lm/models/qwen3_next.py:168
Method__call__
( self, inputs: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any
mlx_lm/models/qwen3_next.py:236
Method__call__
( self, x: mx.array, )
mlx_lm/models/qwen3_next.py:327
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/qwen3_next.py:377
Method__call__
( self, inputs: mx.array, cache: Optional[Any] = None, )
mlx_lm/models/qwen3_next.py:404
Method__call__
( self, inputs: mx.array, cache: Optional[Any] = None, )
mlx_lm/models/qwen3_next.py:433
Method__call__
(self, x)
mlx_lm/models/minimax.py:71
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/minimax.py:123
Method__call__
(self, x: mx.array)
mlx_lm/models/minimax.py:174
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/minimax.py:213
Method__call__
( self, inputs: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any
mlx_lm/models/minimax.py:235
Method__call__
( self, inputs: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any
mlx_lm/models/minimax.py:263
Method__call__
(self, x)
mlx_lm/models/recurrent_gemma.py:45
Method__call__
(self, x, cache=None)
mlx_lm/models/recurrent_gemma.py:89
Method__call__
( self, x: mx.array, cache=None, )
mlx_lm/models/recurrent_gemma.py:129
Method__call__
( self, x: mx.array, cache=None, mask=None, )
mlx_lm/models/recurrent_gemma.py:197
Method__call__
( self, x: mx.array, cache=None, mask=None, )
mlx_lm/models/recurrent_gemma.py:244
Method__call__
(self, x: mx.array)
mlx_lm/models/recurrent_gemma.py:281
Method__call__
( self, x: mx.array, cache=None, mask=None, )
mlx_lm/models/recurrent_gemma.py:342
Method__call__
( self, tokens, cache=None, )
mlx_lm/models/recurrent_gemma.py:391
Method__call__
(self, tokens: mx.array, cache=None)
mlx_lm/models/recurrent_gemma.py:421
Method__call__
(self, x, offset: int = 0)
mlx_lm/models/smollm3.py:32
Method__call__
( self, inputs: mx.array, cache=None, input_embeddings: Optional[mx.array] = N
mlx_lm/models/smollm3.py:52
Method__call__
(self, x: mx.array)
mlx_lm/models/gemma4_text.py:80
Method__call__
(self, x: mx.array)
mlx_lm/models/gemma4_text.py:113
Method__call__
(self, x: mx.array)
mlx_lm/models/gemma4_text.py:129
Method__call__
(self, x, gate)
mlx_lm/models/gemma4_text.py:149
Method__call__
( self, x: mx.array, top_k_indices: mx.array, top_k_weights: mx.array )
mlx_lm/models/gemma4_text.py:167
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/gemma4_text.py:231
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/gemma4_text.py:333
Method__call__
( self, inputs: mx.array = None, cache=None, input_embeddings: Optional[mx.arr
mlx_lm/models/gemma4_text.py:517
Method__call__
( self, inputs: mx.array, cache=None, input_embeddings: Optional[mx.array] = N
mlx_lm/models/gemma4_text.py:589
Method__call__
(self, x: mx.array)
mlx_lm/models/bailing_moe_linear.py:107
Method__call__
(self, x)
mlx_lm/models/bailing_moe_linear.py:133
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/bailing_moe_linear.py:169
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/bailing_moe_linear.py:278
Method__call__
(self, x: mx.array)
mlx_lm/models/bailing_moe_linear.py:387
Method__call__
(self, x: mx.array)
mlx_lm/models/bailing_moe_linear.py:424
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/bailing_moe_linear.py:459
Method__call__
( self, inputs: mx.array, cache: Optional[Any] = None, )
mlx_lm/models/bailing_moe_linear.py:486
Method__call__
( self, inputs: mx.array, cache: Optional[Any] = None, )
mlx_lm/models/bailing_moe_linear.py:519
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/lfm2_moe.py:80
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Any] = N
mlx_lm/models/lfm2_moe.py:135
Method__call__
(self, x)
mlx_lm/models/lfm2_moe.py:185
Method__call__
( self, x: mx.array, )
mlx_lm/models/lfm2_moe.py:205
← previousnext →601–700 of 2,622, ranked by callers