MCPcopy Create free account

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

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

Methodtest_loglikelihood_rolling_processes_all_inputs
Test that loglikelihood_rolling processes all inputs correctly when batching.
tests/test_evaluate.py:24
Methodtest_lora_embedding
(self)
tests/test_finetune.py:112
Methodtest_lru
(self)
tests/test_server.py:571
Methodtest_lru_bytes
(self)
tests/test_server.py:665
Methodtest_make_logits_processors
(self)
tests/test_sample_utils.py:150
Methodtest_make_mask_with_cache
(self)
tests/test_prompt_cache.py:400
Methodtest_make_shards
(self)
tests/test_utils.py:44
Methodtest_make_state_machine_empty_tool_call_end
(self)
tests/test_server.py:283
Methodtest_malformed_config
(self)
tests/test_finetune.py:365
Methodtest_mamba
(self)
tests/test_models.py:1168
Methodtest_many_batches
(self)
tests/test_generate.py:216
Methodtest_mask_with_window
(self)
tests/test_models.py:164
Methodtest_merge_with_empty_caches
(self)
tests/test_prompt_cache.py:642
Methodtest_minicpm
(self)
tests/test_models.py:1147
Methodtest_minimax_m2
(self)
tests/test_tool_parsing.py:316
Methodtest_mixtral
(self)
tests/test_models.py:507
Methodtest_multi_token_match_preserves_order
(self)
tests/test_server.py:140
Methodtest_nemotron
(self)
tests/test_models.py:1680
Methodtest_no_system_prompt_in_messages
( self, mock_print, mock_input, mock_stream_generate, mock_make_prompt
tests/test_chat.py:119
Methodtest_non_zero_warmup
(self)
tests/test_finetune.py:354
Methodtest_olmo
(self)
tests/test_models.py:529
Methodtest_olmo2
(self)
tests/test_models.py:1832
Methodtest_openelm
(self)
tests/test_models.py:1236
Methodtest_parsers
(self)
tests/test_tool_parsing.py:19
Methodtest_phi2
(self)
tests/test_models.py:462
Methodtest_phi3
(self)
tests/test_models.py:471
Methodtest_phi3small
(self)
tests/test_models.py:1699
Methodtest_phimoe
(self)
tests/test_models.py:1719
Methodtest_plamo
(self)
tests/test_models.py:895
Methodtest_plamo2
(self)
tests/test_models.py:912
Methodtest_presence_penalty
(self)
tests/test_sample_utils.py:119
Methodtest_print_trainable_parameters
(self)
tests/test_tuner_utils.py:59
Methodtest_prompt_cache_with_draft_model
(self)
tests/test_server.py:424
Methodtest_quantize
(self)
tests/test_utils.py:62
Methodtest_quantized_print_trainable_parameters
(self)
tests/test_tuner_utils.py:22
Methodtest_quantized_sdpa
(self)
tests/test_models.py:306
Methodtest_qwen
(self)
tests/test_models.py:884
Methodtest_qwen2
(self)
tests/test_models.py:866
Methodtest_qwen2_moe
(self)
tests/test_models.py:845
Methodtest_qwen3
(self)
tests/test_models.py:577
Methodtest_qwen3_5_family_convert_then_load_norm_not_shift_twice
(self)
tests/test_models.py:599
Methodtest_qwen3_coder_single_quoted_params
(self)
tests/test_tool_parsing.py:157
Methodtest_qwen3_moe
(self)
tests/test_models.py:549
Methodtest_recurrent_gemma
(self)
tests/test_models.py:1744
Methodtest_remove_prompt_batch_updates_currently_processing
(self)
tests/test_generate.py:727
Methodtest_rope
(self)
tests/test_models.py:225
Methodtest_rotating_cache_mask
(self)
tests/test_prompt_cache.py:439
Methodtest_rotating_cache_updates
(self)
tests/test_prompt_cache.py:628
Methodtest_rotating_kv_cache
(self)
tests/test_models.py:42
Methodtest_rotating_kv_cache_chat_mode
(self)
tests/test_models.py:100
Methodtest_save_load
(self)
tests/test_prompt_cache.py:43
Methodtest_save_load_arrays_cache
(self)
tests/test_prompt_cache.py:170
Methodtest_save_load_batch_caches
(self)
tests/test_prompt_cache.py:605
Methodtest_save_load_cache_list
(self)
tests/test_prompt_cache.py:135
Methodtest_save_load_mixed_cache
(self)
tests/test_prompt_cache.py:101
Methodtest_save_load_quantized_cache
(self)
tests/test_prompt_cache.py:317
Methodtest_save_load_rotating_cache
(self)
tests/test_prompt_cache.py:64
Methodtest_setup_arg_parser_all_args
(self)
tests/test_chat.py:21
Methodtest_setup_arg_parser_system_prompt
(self)
tests/test_chat.py:10
Methodtest_shard
(self)
tests/model_parallel_tests.py:13
Methodtest_single_schedule
(self)
tests/test_finetune.py:343
Methodtest_single_tool_call_passes_body_with_open_and_close_crossings
(self)
tests/test_server.py:100
Methodtest_smollm3
(self)
tests/test_models.py:1925
Methodtest_special_tokens
(self)
tests/test_tokenizers.py:75
Methodtest_ssm
(self)
tests/test_models.py:3004
Methodtest_ssm_masked
(self)
tests/test_models.py:3035
Methodtest_ssm_right_pad
(self)
tests/test_models.py:3070
Methodtest_stablelm
(self)
tests/test_models.py:929
Methodtest_starcoder2
(self)
tests/test_models.py:969
Methodtest_step3p5
(self)
tests/test_models.py:985
Methodtest_step3p5_make_cache_uses_fallback_sliding_pattern
(self)
tests/test_models.py:1076
Methodtest_step3p5_make_cache_uses_rotating_for_sliding_layers
(self)
tests/test_models.py:1023
Methodtest_stream_generate_input_embeddings
(self)
tests/test_generate.py:119
Methodtest_stream_generate_input_embeddings_prefill
(self)
tests/test_generate.py:143
Methodtest_stream_generate_max_tokens
(self)
tests/test_generate.py:49
Methodtest_stream_generate_speculative
(self)
tests/test_generate.py:86
Methodtest_streaming_with_draft_model
(self)
tests/test_server.py:393
Methodtest_su_scaled_rope_no_mutation
(self)
tests/test_models.py:282
Methodtest_system_prompt_in_messages
( self, mock_print, mock_input, mock_stream_generate, mock_make_prompt
tests/test_chat.py:64
Methodtest_text
(self)
tests/test_datsets.py:38
Methodtest_thinking
(self)
tests/test_tokenizers.py:97
Methodtest_tokenizers
(self)
tests/test_tokenizers.py:54
Methodtest_tool_calling
(self)
tests/test_tokenizers.py:86
Methodtest_trim_cache
(self)
tests/test_prompt_cache.py:205
Methodtest_trim_cache_with_generate
(self)
tests/test_prompt_cache.py:265
Methodtest_window_mask_with_full_kv_cache
(self)
tests/test_prompt_cache.py:761
Methodtest_yarn_rope_no_mutation
(self)
tests/test_models.py:294
Methodthink_end
(self)
mlx_lm/tokenizer_utils.py:411
Methodthink_end_id
(self)
mlx_lm/tokenizer_utils.py:415
Methodthink_end_tokens
(self)
mlx_lm/tokenizer_utils.py:423
Methodthink_start
(self)
mlx_lm/tokenizer_utils.py:395
Methodthink_start_id
(self)
mlx_lm/tokenizer_utils.py:399
Methodthink_start_tokens
(self)
mlx_lm/tokenizer_utils.py:407
Methodto_lora
(self, r: int = 8, dropout: float = 0.0, scale: float = 20.0)
mlx_lm/models/afm7.py:117
Methodto_quantized
(self, group_size: int = 64, bits: int = 4)
mlx_lm/models/cache.py:551
Methodto_quantized
(self, group_size: int = 64, bits: int = 4)
mlx_lm/models/cache.py:1327
Methodto_quantized
( self, group_size: int, bits: int, mode: str = "affine", )
mlx_lm/models/mla.py:25
Methodto_remove
(k)
mlx_lm/models/llama4.py:292
Methodtokenizer_name
(self)
mlx_lm/evaluate.py:149
Methodtokens
(self)
mlx_lm/models/cache.py:1519
← previousnext →2,501–2,600 of 2,622, ranked by callers