MCPcopy Create free account

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

Functions4,946 in github.com/ml-explore/mlx

⨍ Functions 4,946◇ Types & classes 770↳ Endpoints 5
Methodtest_multi_input_einsum
(self)
python/tests/test_einsum.py:214
Methodtest_multi_optimizer
(self)
python/tests/test_optimizers.py:558
Methodtest_multi_output_eval_during_transform
(self)
python/tests/test_eval.py:108
Methodtest_multi_output_leak
(self)
python/tests/test_array.py:1966
Methodtest_multiple_compile_same_capture
(self)
python/tests/test_compile.py:1208
Methodtest_multiply
(self)
python/tests/test_ops.py:250
Methodtest_multistream_deadlock
(self)
python/tests/test_eval.py:176
Methodtest_multivariate_normal
(self)
python/tests/test_random.py:112
Methodtest_muon
(self)
python/tests/test_optimizers.py:289
Methodtest_mxfp4_quantize_dequantize
(self)
python/tests/test_quantized.py:30
Methodtest_mxfp8_quantize_dequantize
(self)
python/tests/test_quantized.py:85
Methodtest_nan_propagation
(self)
python/tests/test_reduce.py:156
Methodtest_nan_propagation_complex64
(self)
python/tests/test_reduce.py:182
Methodtest_nan_to_num
(self)
python/tests/test_ops.py:2001
Methodtest_negative
(self)
python/tests/test_ops.py:847
Methodtest_newaxis_for_reshaping_arrays
(self)
python/tests/test_constants.py:33
Methodtest_nll_loss
(self)
python/tests/test_losses.py:254
Methodtest_non_contiguous
(self)
python/tests/test_load.py:391
Methodtest_non_multiples
(self)
python/tests/test_quantized.py:623
Methodtest_norm
(self)
python/tests/test_linalg.py:13
Methodtest_normal
(self)
python/tests/test_init.py:22
Methodtest_normal_and_laplace
(self)
python/tests/test_random.py:67
Methodtest_np_array_conversion_copies_by_default
(self)
python/tests/test_array.py:1703
Methodtest_numpy_conv
(self)
python/tests/test_conv.py:22
Methodtest_nvfp4_quantize_dequantize
(self)
python/tests/test_quantized.py:113
Methodtest_obj_inequality_array
(self)
python/tests/test_array.py:217
Methodtest_op_on_device
(self)
python/tests/test_device.py:52
Methodtest_op_on_stream
(self)
python/tests/test_device.py:96
Methodtest_ops
(npop, mlxop, x, y, atol, rtol)
python/tests/test_ops.py:1521
Methodtest_opt_einsum_test_cases
(self)
python/tests/test_einsum.py:220
Methodtest_optimizer_state
(self)
python/tests/test_optimizers.py:46
Methodtest_optimizers
(self)
python/tests/test_optimizers.py:54
Methodtest_orthogonal
(self)
python/tests/test_init.py:109
Methodtest_outer
(self)
python/tests/test_ops.py:2552
Methodtest_outputs_changing
(self)
python/tests/test_compile.py:1189
Methodtest_pad
(self)
python/tests/test_ops.py:1937
Methodtest_parameter_deletion
(self)
python/tests/test_nn.py:333
Methodtest_partition
(self)
python/tests/test_ops.py:2318
Methodtest_permutation
(self)
python/tests/test_random.py:328
Methodtest_pooling
(self)
python/tests/test_nn.py:1415
Methodtest_power_grad
(self)
python/tests/test_autograd.py:477
Methodtest_prelu
(self)
python/tests/test_nn.py:1070
Methodtest_prod
(self)
python/tests/test_ops.py:659
Methodtest_pseudo_inverse
(self)
python/tests/test_linalg.py:259
Methodtest_put_along_axis
(self)
python/tests/test_ops.py:1269
Methodtest_put_along_axis_grads
(self)
python/tests/test_autograd.py:764
Methodtest_qmm
(self)
python/tests/test_quantized.py:208
Methodtest_qmm_fp_type
(self)
python/tests/test_quantized.py:897
Methodtest_qmm_jvp
(self)
python/tests/test_quantized.py:275
Methodtest_qmm_shapes
(self)
python/tests/test_quantized.py:305
Methodtest_qmm_vjp
(self)
python/tests/test_quantized.py:245
Methodtest_qmv
(self)
python/tests/test_quantized.py:330
Methodtest_qmv_small_non_multiples
(self)
python/tests/test_quantized.py:697
Methodtest_qqmv
(self)
python/tests/test_quantized.py:176
Methodtest_qr_factorization
(self)
python/tests/test_linalg.py:95
Methodtest_quantize
(self)
python/tests/test_nn.py:205
Methodtest_quantize_dequantize
(self)
python/tests/test_quantized.py:11
Methodtest_quantize_freeze
(self)
python/tests/test_nn.py:245
Methodtest_quantize_strided
(self)
python/tests/test_quantized.py:1160
Methodtest_quantized_embedding
(self)
python/tests/test_nn.py:1974
Methodtest_quantized_sharded_linear_construction
(self)
python/tests/test_nn.py:252
Methodtest_qvm
(self)
python/tests/test_quantized.py:408
Methodtest_qvm_splitk
(self)
python/tests/test_quantized.py:435
Methodtest_radians
(self)
python/tests/test_ops.py:1050
Methodtest_randint
(self)
python/tests/test_random.py:202
Methodtest_real_imag
(self)
python/tests/test_array.py:2252
Methodtest_real_imag
(self)
python/tests/test_ops.py:3098
Methodtest_reciprocal
(self)
python/tests/test_ops.py:915
Methodtest_reduce_jvp
(self)
python/tests/test_autograd.py:953
Methodtest_reduction_ops
(self)
python/tests/test_bf16.py:127
Methodtest_reductions
(self)
python/tests/test_double.py:115
Methodtest_relu
(self)
python/tests/test_nn.py:944
Methodtest_relu6
(self)
python/tests/test_nn.py:979
Methodtest_remainder
(self)
python/tests/test_ops.py:300
Methodtest_repeat
(self)
python/tests/test_ops.py:2496
Methodtest_repeated_conv
(self)
python/tests/test_conv.py:1047
Methodtest_retain_graph
(self)
python/tests/test_eval.py:17
Methodtest_rfftfreq
(self)
python/tests/test_fft.py:270
Methodtest_rms_norm
(self)
python/tests/test_fast.py:388
Methodtest_rms_norm_grad
(self)
python/tests/test_fast.py:448
Methodtest_rmsprop
(self)
python/tests/test_optimizers.py:102
Methodtest_rnn
(self)
python/tests/test_nn.py:1900
Methodtest_roll
(self)
python/tests/test_ops.py:3059
Methodtest_roll_errors
(self)
python/tests/test_ops.py:3093
Methodtest_rope
(self)
python/tests/test_fast.py:72
Methodtest_rope
(self)
python/tests/test_nn.py:1119
Methodtest_rope_batch
(self)
python/tests/test_fast.py:319
Methodtest_rope_dims_validation
(self)
python/tests/test_fast.py:172
Methodtest_rope_grad
(self)
python/tests/test_fast.py:292
Methodtest_rope_with_freqs
(self)
python/tests/test_fast.py:207
Methodtest_rope_with_large_offset
(self)
python/tests/test_fast.py:368
Methodtest_round
(self)
python/tests/test_ops.py:537
Methodtest_rsqrt
(self)
python/tests/test_ops.py:909
Methodtest_save_and_load
(self)
python/tests/test_load.py:40
Methodtest_save_and_load_fs
(self)
python/tests/test_load.py:303
Methodtest_save_and_load_gguf
(self)
python/tests/test_load.py:131
Methodtest_save_and_load_gguf_metadata_arrays
(self)
python/tests/test_load.py:225
Methodtest_save_and_load_gguf_metadata_basic
(self)
python/tests/test_load.py:192
Methodtest_save_and_load_gguf_metadata_mixed
(self)
python/tests/test_load.py:261
Methodtest_save_and_load_safetensors
(self)
python/tests/test_load.py:90
← previousnext →4,501–4,600 of 4,946, ranked by callers