MCPcopy Create free account

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

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

⨍ Functions 1,187◇ Types & classes 308↳ Endpoints 13
Method__call__
( self, inputs: mx.array, cache=None, )
lora/models.py:242
Method__call__
( self, inputs: mx.array, cache=None, )
lora/models.py:269
Method__call__
(self, x)
normalizing_flow/flows.py:20
Method__call__
(self, x: mx.array)
normalizing_flow/flows.py:74
Method__call__
(self, *args: Any, **kwargs: Any)
clip/tokenizer.py:84
Method__call__
(self, images: List[Image])
clip/image_processor.py:37
Method__call__
(self, queries, keys, values, mask=None)
clip/model.py:115
Method__call__
(self, x: mx.array)
clip/model.py:145
Method__call__
(self, x: mx.array, mask: Optional[mx.array] = None)
clip/model.py:165
Method__call__
(self, x: mx.array)
clip/model.py:184
Method__call__
(self, x: mx.array)
clip/model.py:204
Method__call__
(self, x: mx.array)
clip/model.py:241
Method__call__
( self, x: mx.array, output_hidden_states: Optional[bool] = None, )
clip/model.py:271
Method__call__
( self, input_ids: Optional[mx.array] = None, pixel_values: Optional[mx.array] = None,
clip/model.py:314
Method__call__
(self, x)
speechcommands/kwt.py:27
Method__call__
(self, x)
speechcommands/kwt.py:47
Method__call__
(self, x)
speechcommands/kwt.py:61
Method__call__
(self, x)
speechcommands/kwt.py:143
Method__call__
(self, x, mask)
bert/model.py:32
Method__call__
(self, x, mask)
bert/model.py:54
Method__call__
( self, input_ids: mx.array, token_type_ids: mx.array = None )
bert/model.py:73
Method__call__
( self, input_ids: mx.array, token_type_ids: mx.array = None, attention_mask:
bert/model.py:103
Method__call__
(self, x: mx.array)
llava/llava.py:50
Method__call__
(self, input_ids: mx.array, pixel_values: mx.array, cache=None)
llava/llava.py:123
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llava/language.py:79
Method__call__
(self, x)
llava/language.py:118
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llava/language.py:135
Method__call__
( self, inputs: mx.array, cache=None, inputs_embeds=None, )
llava/language.py:161
Method__call__
( self, inputs: mx.array, cache=None, inputs_embeds=None, )
llava/language.py:198
Method__call__
(self, queries, keys, values, mask=None)
llava/vision.py:69
Method__call__
(self, x: mx.array)
llava/vision.py:98
Method__call__
(self, x: mx.array, mask: Optional[mx.array] = None)
llava/vision.py:115
Method__call__
(self, x: mx.array)
llava/vision.py:152
Method__call__
( self, x: mx.array, output_hidden_states: Optional[bool] = None, )
llava/vision.py:173
Method__call__
( self, x: mx.array, output_hidden_states: Optional[bool] = None )
llava/vision.py:202
Method__call__
(self, x)
cifar/resnet.py:26
Method__call__
(self, x)
cifar/resnet.py:57
Method__call__
(self, x)
cifar/resnet.py:96
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llms/mixtral/mixtral.py:47
Method__call__
(self, x)
llms/mixtral/mixtral.py:87
Method__call__
(self, x)
llms/mixtral/mixtral.py:99
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llms/mixtral/mixtral.py:131
Method__call__
( self, inputs: mx.array, cache=None, )
llms/mixtral/mixtral.py:156
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llms/gguf_llm/models.py:80
Method__call__
(self, x)
llms/gguf_llm/models.py:119
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llms/gguf_llm/models.py:136
Method__call__
( self, inputs: mx.array, cache=None, )
llms/gguf_llm/models.py:172
Method__call__
( self, inputs: mx.array, cache=None, )
llms/gguf_llm/models.py:199
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llms/llama/llama.py:51
Method__call__
(self, x)
llms/llama/llama.py:98
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llms/llama/llama.py:113
Method__call__
(self, x)
llms/llama/llama.py:136
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llms/mistral/mistral.py:47
Method__call__
(self, x)
llms/mistral/mistral.py:87
Method__call__
( self, x: mx.array, mask: Optional[mx.array] = None, cache: Optional[Tuple[mx
llms/mistral/mistral.py:102
Method__call__
( self, inputs: mx.array, cache=None, )
llms/mistral/mistral.py:127
Method__call__
Compute binned relative position bias
llms/speculative_decoding/model.py:70
Method__call__
( self, queries: mx.array, keys: mx.array, values: mx.array, mask: Opt
llms/speculative_decoding/model.py:101
Method__call__
(self, x)
llms/speculative_decoding/model.py:156
Method__call__
(self, x, mask)
llms/speculative_decoding/model.py:174
Method__call__
(self, x: mx.array)
llms/speculative_decoding/model.py:193
Method__call__
( self, x: mx.array, memory: mx.array, mask: mx.array, memory_mask: mx
llms/speculative_decoding/model.py:210
Method__call__
(self, x, memory, cache=None)
llms/speculative_decoding/model.py:248
Method__call__
(self, inputs)
llms/speculative_decoding/model.py:277
Method__call__
( self, inputs: mx.array, decoder_inputs: mx.array, )
llms/speculative_decoding/model.py:321
Method__call__
( self, result: dict, output_name: str, options: Optional[dict] = None, **kwargs )
whisper/mlx_whisper/writers.py:43
Method__call__
( self, x, xa=None, mask=None, kv_cache=None, )
whisper/mlx_whisper/whisper.py:49
Method__call__
(self, x, xa=None, mask=None, kv_cache=None)
whisper/mlx_whisper/whisper.py:107
Method__call__
(self, x)
whisper/mlx_whisper/whisper.py:139
Method__call__
x : mx.array, shape = (batch_size, <= n_ctx) the text tokens xa : mx.array, shape = (batch_size, n_audio_ctx, n_audio_sta
whisper/mlx_whisper/whisper.py:176
Method__call__
(self, mel, tokens)
whisper/mlx_whisper/whisper.py:254
Method__call__
(self, x, hidden=None, cell=None)
encodec/encodec.py:76
Method__call__
(self, hidden_states)
encodec/encodec.py:154
Method__call__
(self, hidden_states)
encodec/encodec.py:200
Method__call__
(self, hidden_states)
encodec/encodec.py:223
Method__call__
(self, hidden_states)
encodec/encodec.py:257
Method__call__
(self, hidden_states)
encodec/encodec.py:310
Method__call__
(self, hidden_states)
encodec/encodec.py:364
Method__call__
(self, x)
stable_diffusion/stable_diffusion/unet.py:27
Method__call__
(self, x, memory, attn_mask, memory_mask)
stable_diffusion/stable_diffusion/unet.py:62
Method__call__
(self, x, encoder_x, attn_mask, encoder_attn_mask)
stable_diffusion/stable_diffusion/unet.py:106
Method__call__
(self, x, temb=None)
stable_diffusion/stable_diffusion/unet.py:153
Method__call__
( self, x, encoder_x=None, temb=None, attn_mask=None, encoder_
stable_diffusion/stable_diffusion/unet.py:237
Method__call__
( self, x, timestep, encoder_x, attn_mask=None, encoder_attn_m
stable_diffusion/stable_diffusion/unet.py:403
Method__call__
(self, x, attn_mask=None)
stable_diffusion/stable_diffusion/clip.py:48
Method__call__
(self, x)
stable_diffusion/stable_diffusion/clip.py:87
Method__call__
(self, x)
stable_diffusion/stable_diffusion/vae.py:25
Method__call__
(self, x)
stable_diffusion/stable_diffusion/vae.py:79
Method__call__
(self, x)
stable_diffusion/stable_diffusion/vae.py:142
Method__call__
(self, x)
stable_diffusion/stable_diffusion/vae.py:209
Method__call__
(self, x, key=None)
stable_diffusion/stable_diffusion/vae.py:269
Method__call__
Compute binned relative position bias
musicgen/t5.py:105
Method__call__
( self, queries: mx.array, keys: mx.array, values: mx.array, mask: Opt
musicgen/t5.py:136
Method__call__
(self, x)
musicgen/t5.py:195
Method__call__
(self, x, mask)
musicgen/t5.py:213
Method__call__
(self, x: mx.array)
musicgen/t5.py:232
Method__call__
( self, x: mx.array, memory: mx.array, mask: mx.array, memory_mask: mx
musicgen/t5.py:249
Method__call__
(self, x, memory, mask, memory_mask, cache=None)
musicgen/t5.py:280
Method__call__
(self, inputs)
musicgen/t5.py:305
Method__call__
( self, inputs: mx.array, decoder_inputs: mx.array, )
musicgen/t5.py:346
← previousnext →601–700 of 1,187, ranked by callers