MCPcopy Create free account

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

Functions1,187 in github.com/ml-explore/mlx-examples

Functionnormalize
(x)
cifar/dataset.py:12
Functionnormalize
(x)
cvae/dataset.py:16
Methodnum_languages
(self)
whisper/mlx_whisper/torch_whisper.py:274
Methodnum_languages
(self)
whisper/mlx_whisper/whisper.py:262
Functionoptional_float
(string)
whisper/mlx_whisper/cli.py:19
Functionoptional_int
(string)
whisper/mlx_whisper/cli.py:16
Methodpad_id
(self)
llms/mistral/mistral.py:160
Methodpad_token
(self)
flux/flux/tokenizers.py:135
Functionpreprocess_audio
r""" Prepare inputs for the EnCodec model. Args: raw_audio (mx.array or List[mx.array]): The sequence or batch of sequenc
musicgen/encodec.py:704
Functionquantization_predicate
(name, m)
flux/generate_interactive.py:18
Functionquantization_predicate
(name, m)
flux/txt2image.py:28
Functionquick_gelu
A fast GELU approximation https://github.com/hendrycks/GELUs
clip/model.py:69
Methodrank
(self, tokens: List[List[List[int]]], sum_logprobs: List[List[float]])
whisper/mlx_whisper/decoding.py:174
Methodrearrange_kv_cache
Update the key-value cache according to the updated beams
whisper/mlx_whisper/decoding.py:144
Methodreplace_key
(key: str)
musicgen/t5.py:388
Methodreplace_key
(key: str)
t5/t5.py:388
Methodreset
Initialize any stateful variables for decoding a new sequence
whisper/mlx_whisper/decoding.py:192
Functionresnet110
(**kwargs)
cifar/resnet.py:122
Functionresnet1202
(**kwargs)
cifar/resnet.py:126
Functionresnet20
(**kwargs)
cifar/resnet.py:106
Functionresnet32
(**kwargs)
cifar/resnet.py:110
Functionresnet44
(**kwargs)
cifar/resnet.py:114
Functionresnet56
(**kwargs)
cifar/resnet.py:118
Methodsample
(logits)
llms/llama/llama.py:147
Methodsample
(logits)
llms/speculative_decoding/decoder.py:110
Methodsample_and_log_prob
( self, sample_shape: Union[int, Tuple[int, ...]], key: Optional[mx.array] = None )
normalizing_flow/distributions.py:27
Methodsampling_rate
(self)
encodec/encodec.py:620
Methodsampling_rate
(self)
musicgen/encodec.py:620
Methodsanitize
(weights)
llava/language.py:208
Methodsanitize
Remap CLIP .pth checkpoint keys to MLX model format. Handles both standard OpenCLIP naming and Wan2.1 HF naming. Extracts only visual
video/wan2.1/wan/clip.py:125
Methodsanitize
Remap PyTorch T5 keys to MLX format.
video/wan2.1/wan/t5.py:164
Methodsanitize
Remap PyTorch checkpoint keys to MLX model format.
video/wan2.1/wan/model.py:196
Functionsave_model
(save_dir: Union[str, Path], weights, tokenizer, config)
lora/utils.py:87
Methodsave_to_cache
(module, _, output)
whisper/mlx_whisper/torch_whisper.py:294
Methodscalar
(x)
flux/flux/flux.py:101
Methodscores
(logprobs, lengths)
whisper/mlx_whisper/decoding.py:175
MethodsetUpClass
(cls)
clip/test.py:34
MethodsetUpClass
(cls)
llava/test.py:32
MethodsetUpClass
(cls)
llava/test.py:79
MethodsetUpClass
(cls)
whisper/test.py:80
Methodset_alignment_heads
(self, dump: bytes)
whisper/mlx_whisper/torch_whisper.py:245
Methodset_timesteps
(self, num_inference_steps, shift=None)
video/wan2.1/wan/sampler.py:78
Methodsot
(self)
whisper/mlx_whisper/tokenizer.py:190
Methodsot_lm
(self)
whisper/mlx_whisper/tokenizer.py:194
Methodsot_prev
(self)
whisper/mlx_whisper/tokenizer.py:198
Methodsot_sequence_including_notimestamps
(self)
whisper/mlx_whisper/tokenizer.py:240
Functionspm_tokenizer
(metadata)
llms/gguf_llm/utils.py:5
Methodstate
(self)
musicgen/musicgen.py:67
Methodstep
(self, eps_pred, x_t, t, t_prev)
stable_diffusion/stable_diffusion/sampler.py:89
Methodstep
(self, model_output, timestep, sample)
video/wan2.1/wan/sampler.py:310
Methodstep_index
(self)
video/wan2.1/wan/sampler.py:75
Functionstr2bool
(string)
whisper/mlx_whisper/cli.py:22
Methodtest_clip_model
(self)
clip/test.py:112
Methodtest_decode_greedy
(self)
whisper/test.py:127
Methodtest_decode_lang
(self)
whisper/test.py:118
Methodtest_fp16
(self)
whisper/test.py:101
Methodtest_generate
(self)
llms/mistral/test.py:37
Methodtest_generated_tokens
(self)
llava/test.py:139
Methodtest_image_features
(self)
llava/test.py:37
Methodtest_image_processor
(self)
clip/test.py:38
Methodtest_load
(self)
whisper/test.py:444
Methodtest_mel_spec
(self)
whisper/test.py:457
Methodtest_merge_input_ids_with_image_features
(self)
llava/test.py:84
Methodtest_model
(self)
llms/mistral/test.py:11
Methodtest_pad
(self)
whisper/test.py:452
Methodtest_quantized_4bits
(self)
whisper/test.py:109
Methodtest_text_encoder
(self)
clip/test.py:61
Methodtest_text_tokenizer
(self)
clip/test.py:51
Methodtest_torch_mlx
(self)
whisper/test.py:86
Methodtest_transcribe
(self)
whisper/test.py:188
Methodtest_transcribe_alice
(self)
whisper/test.py:200
Methodtest_transcribe_word_level_timestamps_confidence_scores
(self)
whisper/test.py:312
Methodtest_vision_encoder
(self)
clip/test.py:79
Methodtimestamp_begin
(self)
whisper/mlx_whisper/tokenizer.py:210
Functiontiny_llama
(model_path, *, dtype: str)
llms/llama/convert.py:67
Functiontrain_step
(model, x, y)
speechcommands/main.py:60
Functiontrain_step
(model, inp, tgt)
cifar/main.py:38
Methodtraining_loss
( self, x_0: mx.array, t5_features: mx.array, clip_features: mx.array,
flux/flux/flux.py:195
Methodtranslate
(self)
whisper/mlx_whisper/tokenizer.py:186
Methodupdate
( self, tokens: mx.array, logits: mx.array, sum_logprobs: mx.array )
whisper/mlx_whisper/decoding.py:260
Functionupload_to_hub
(path: str, name: str, hf_path: str)
lora/utils.py:36
Functionwrite_all
( result: dict, file: TextIO, options: Optional[dict] = None, **kwargs )
whisper/mlx_whisper/writers.py:260
Methodwrite_result
( self, result: dict, file: TextIO, options: Optional[dict] = None, **kwargs )
whisper/mlx_whisper/writers.py:62
Methodwrite_result
( self, result: dict, file: TextIO, options: Optional[dict] = None, **kwargs )
whisper/mlx_whisper/writers.py:193
Methodwrite_result
( self, result: dict, file: TextIO, options: Optional[dict] = None, **kwargs )
whisper/mlx_whisper/writers.py:206
Methodwrite_result
( self, result: dict, file: TextIO, options: Optional[dict] = None, **kwargs )
whisper/mlx_whisper/writers.py:227
Methodwrite_result
( self, result: dict, file: TextIO, options: Optional[dict] = None, **kwargs )
whisper/mlx_whisper/writers.py:240
← previous1,101–1,187 of 1,187, ranked by callers