Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ml-explore/mlx-lm
/ functions
Functions
2,622 in github.com/ml-explore/mlx-lm
⨍
Functions
2,622
◇
Types & classes
816
↳
Endpoints
9
Method
test_batch_generate_processor_tokens_match_prompt_on_first_step
(self)
tests/test_generate.py:405
Method
test_batch_generate_with_logits_processors
Test that batch_generate with logits_processors produces correct results.
tests/test_generate.py:362
Method
test_batch_generate_with_samplers
Test that batch_generate with logits_processors produces correct results.
tests/test_generate.py:441
Method
test_batch_generate_with_state_machines
Test that batch_generate with per-sequence state_machines stops on different tokens.
tests/test_generate.py:473
Method
test_batch_kv_cache
(self)
tests/test_prompt_cache.py:492
Method
test_batch_matches_single
(self)
tests/test_generate.py:182
Method
test_batch_max_kv_size_creates_rotating_cache
(self)
tests/test_generate.py:758
Method
test_batch_max_kv_size_limits_cache_growth
(self)
tests/test_generate.py:775
Method
test_batch_max_kv_size_none_creates_regular_cache
(self)
tests/test_generate.py:794
Method
test_batch_rotating_kv_cache
(self)
tests/test_prompt_cache.py:533
Method
test_batch_sliding_window
(self)
tests/test_generate.py:312
Method
test_batch_unique_max_toks
(self)
tests/test_generate.py:267
Method
test_bitnet
(self)
tests/test_models.py:444
Method
test_cache_copying
(self)
tests/test_prompt_cache.py:295
Method
test_cache_list
(self)
tests/test_prompt_cache.py:367
Method
test_cache_to_quantized
(self)
tests/test_prompt_cache.py:342
Method
test_cache_with_generate
(self)
tests/test_prompt_cache.py:183
Method
test_caching
(self)
tests/test_server.py:521
Method
test_causal_mask_padding
(self)
tests/test_models.py:140
Method
test_chat
(self)
tests/test_datsets.py:95
Method
test_cohere
(self)
tests/test_models.py:1121
Method
test_cohere2
(self)
tests/test_models.py:1869
Method
test_completions
(self)
tests/test_datsets.py:51
Method
test_completions_mask_prompt
(self)
tests/test_datsets.py:64
Method
test_convert
(self)
tests/test_utils.py:82
Method
test_convert_to_gguf
( self, mock_save_gguf, mock_from_pretrained, )
tests/test_gguf.py:27
Method
test_dbrx
(self)
tests/test_models.py:1132
Method
test_deepseek
(self)
tests/test_models.py:1338
Method
test_deepseek_v2
(self)
tests/test_models.py:1356
Method
test_deepseek_v3
(self)
tests/test_models.py:1388
Method
test_dora_dtype
(self)
tests/test_finetune.py:313
Method
test_dora_embedding
(self)
tests/test_finetune.py:146
Method
test_dora_from_linear
(self)
tests/test_finetune.py:252
Method
test_dora_m_parameter
(self)
tests/test_finetune.py:226
Method
test_dora_to_linear
(self)
tests/test_finetune.py:280
Method
test_evaluate_calls
(self)
tests/test_finetune.py:372
Method
test_evaluate_infinite_batches
(self)
tests/test_finetune.py:412
Method
test_exaone
(self)
tests/test_models.py:1852
Method
test_extend_cache_with_empty
(self)
tests/test_generate.py:707
Method
test_extend_with_empty_and_nonempty_batch_caches
Extending a batch cache when one side has keys=None should use the correct batch size for the placeholder, not the batch size from the
tests/test_prompt_cache.py:665
Method
test_falcon_h1
(self)
tests/test_models.py:1188
Method
test_fetch_empty_tokens_after_root_eviction
(self)
tests/test_server.py:654
Method
test_frequency_penalty
(self)
tests/test_sample_utils.py:134
Method
test_gated_delta
(self)
tests/test_models.py:3114
Method
test_gated_delta_masked
(self)
tests/test_models.py:3189
Method
test_gated_delta_precision
(self)
tests/test_models.py:3135
Method
test_gemma
(self)
tests/test_models.py:488
Method
test_gemma2
(self)
tests/test_models.py:1425
Method
test_gemma3_text
(self)
tests/test_models.py:1444
Method
test_gemma4
(self)
tests/test_tool_parsing.py:200
Method
test_gemma4_convert_then_load_keeps_language_model_prefix
(self)
tests/test_models.py:652
Method
test_gemma4_input_embeddings_reconstruct_per_layer_inputs
(self)
tests/test_models.py:1622
Method
test_gemma4_kv_shared_layers_omit_kv_projections
KV-shared layers must not create k_proj/v_proj/k_norm/v_norm so that models saved without redundant weights (e.g. via transformers sav
tests/test_models.py:1550
Method
test_gemma4_moe_router_quantizes_to_8bit
(self)
tests/test_models.py:794
Method
test_gemma4_quantized_embedding_preserves_lookup_scale
(self)
tests/test_models.py:1511
Method
test_gemma4_raw_hf_language_model_prefixes_model
(self)
tests/test_models.py:704
Method
test_gemma4_raw_hf_moe_expert_weights_split_for_switch_glu
(self)
tests/test_models.py:740
Method
test_gemma4_text
(self)
tests/test_models.py:1464
Method
test_generate
(self)
tests/test_generate.py:31
Method
test_generate_with_logit_bias
(self)
tests/test_generate.py:37
Method
test_generate_with_processor
(self)
tests/test_generate.py:66
Method
test_gpt2
(self)
tests/test_models.py:1201
Method
test_gpt_bigcode
(self)
tests/test_models.py:1664
Method
test_gpt_neox
(self)
tests/test_finetune.py:90
Method
test_gpt_neox
(self)
tests/test_models.py:1217
Method
test_gpt_oss
(self)
tests/test_models.py:1942
Method
test_handle_chat_completions
(self)
tests/test_server.py:210
Method
test_handle_chat_completions_with_content_fragments
(self)
tests/test_server.py:228
Method
test_handle_chat_completions_with_draft_model
(self)
tests/test_server.py:369
Method
test_handle_chat_completions_with_null_tool_content
(self)
tests/test_server.py:251
Method
test_handle_completions
(self)
tests/test_server.py:183
Method
test_handle_completions_with_draft_model
(self)
tests/test_server.py:347
Method
test_handle_models
(self)
tests/test_server.py:310
Method
test_hf
(self)
tests/test_datsets.py:114
Method
test_hunyuan
(self)
tests/test_models.py:1768
Method
test_hunyuan_v1_dense
(self)
tests/test_models.py:1800
Method
test_insert_empty_tokens_does_not_self_destruct
(self)
tests/test_server.py:642
Method
test_insert_trimmable_cache_removes_immediate_prefix
(self)
tests/test_server.py:630
Method
test_internlm2
(self)
tests/test_models.py:1294
Method
test_internlm3
(self)
tests/test_models.py:1885
Method
test_iquestloopcoder
(self)
tests/test_models.py:1902
Method
test_iterate_batches_ddp
(self)
tests/test_tuner_trainer.py:23
Method
test_join
(self)
tests/test_finetune.py:327
Method
test_js_div_loss
(self)
tests/test_losses.py:25
Method
test_js_div_loss_vjp
(self)
tests/test_losses.py:52
Method
test_keepalive_callback
Test keepalive callback sends SSE comments and handles errors
tests/test_server.py:472
Method
test_kimi_k2
(self)
tests/test_tool_parsing.py:278
Method
test_kl_div_loss
(self)
tests/test_losses.py:12
Method
test_kl_div_loss_vjp
(self)
tests/test_losses.py:38
Method
test_kv_cache
(self)
tests/test_models.py:22
Method
test_lfm2
(self)
tests/test_models.py:388
Method
test_lfm2_moe
(self)
tests/test_models.py:415
Method
test_llama
(self)
tests/test_finetune.py:31
Method
test_llama
(self)
tests/test_finetune.py:172
Method
test_llama
(self)
tests/test_models.py:371
Method
test_llama3_1
(self)
tests/test_models.py:1311
Method
test_llama_model_sliding_attention
(self)
tests/test_models.py:182
Method
test_load
(self)
tests/test_utils.py:30
Method
test_load_model_gemma4_with_per_layer_projection_quantization
(self)
tests/test_utils.py:127
Method
test_load_model_with_custom_get_classes
(self)
tests/test_utils.py:98
← previous
next →
2,401–2,500 of 2,622, ranked by callers