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_export_conv_shapeless
(self)
python/tests/test_export_import.py:349
Methodtest_export_import_custom_kernel
(self)
python/tests/test_export_import.py:541
Methodtest_export_import_multi_with_constants
(self)
python/tests/test_export_import.py:584
Methodtest_export_import_scatter_sum
(self)
python/tests/test_export_import.py:605
Methodtest_export_import_shapeless
(self)
python/tests/test_export_import.py:272
Methodtest_export_kwarg_ordering
(self)
python/tests/test_export_import.py:488
Methodtest_export_quantized_model
(self)
python/tests/test_export_import.py:466
Methodtest_export_random_sample
(self)
python/tests/test_export_import.py:134
Methodtest_export_scatter_gather
(self)
python/tests/test_export_import.py:290
Methodtest_export_variable_inputs
(self)
python/tests/test_export_import.py:203
Methodtest_export_with_callback
(self)
python/tests/test_export_import.py:501
Methodtest_export_with_kwargs
(self)
python/tests/test_export_import.py:152
Methodtest_eye
(self)
python/tests/test_ops.py:2369
Methodtest_failure
()
python/tests/test_threads.py:14
Methodtest_fast_transforms
(self)
python/tests/test_fast.py:701
Methodtest_fft
(self)
python/tests/test_fft.py:25
Methodtest_fft_big_powers_of_two
(self)
python/tests/test_fft.py:180
Methodtest_fft_contiguity
(self)
python/tests/test_fft.py:201
Methodtest_fft_exhaustive
(self)
python/tests/test_fft.py:172
Methodtest_fft_grads
(self)
python/tests/test_fft.py:360
Methodtest_fft_into_ifft
(self)
python/tests/test_fft.py:230
Methodtest_fft_large_numbers
(self)
python/tests/test_fft.py:188
Methodtest_fft_norm
(self)
python/tests/test_fft.py:105
Methodtest_fft_shared_mem
(self)
python/tests/test_fft.py:149
Methodtest_fft_throws
(self)
python/tests/test_fft.py:242
Methodtest_fftfreq
(self)
python/tests/test_fft.py:249
Methodtest_fftn
(self)
python/tests/test_fft.py:62
Methodtest_fftshift
(self)
python/tests/test_fft.py:291
Methodtest_fftshift_errors
(self)
python/tests/test_fft.py:347
Methodtest_finfo
(self)
python/tests/test_array.py:96
Methodtest_flatten
(self)
python/tests/test_ops.py:2409
Methodtest_flatten_unflatten_vjps
(self)
python/tests/test_autograd.py:702
Methodtest_floor
(self)
python/tests/test_ops.py:475
Methodtest_format
(self)
python/tests/test_array.py:2181
Methodtest_fp_qmv
(self)
python/tests/test_quantized.py:357
Methodtest_fp_qvm
(self)
python/tests/test_quantized.py:473
Methodtest_fp_vjp_scales_throws
(self)
python/tests/test_quantized.py:1132
Methodtest_fsdp_apply_gradients
(self)
python/tests/nccl_test_distributed.py:119
Methodtest_fsdp_ddp_apply_gradients
(self)
python/tests/nccl_test_distributed.py:199
Methodtest_fsdp_peak_memory
(self)
python/tests/nccl_test_distributed.py:300
Methodtest_full_ones_zeros
(self)
python/tests/test_ops.py:55
Methodtest_function_creates_array
(self)
python/tests/test_compile.py:160
Methodtest_gather_matmul
(self)
python/tests/test_blas.py:1048
Methodtest_gather_matmul_grad
(self)
python/tests/test_quantized.py:916
Methodtest_gather_matmul_grad
(self)
python/tests/test_blas.py:1193
Methodtest_gather_mm_sorted
(self)
python/tests/test_blas.py:1239
Methodtest_gather_mm_sorted_vjp
(self)
python/tests/test_blas.py:1267
Methodtest_gather_qmm
(self)
python/tests/test_quantized.py:756
Methodtest_gather_qmm_grad
(self)
python/tests/test_quantized.py:1059
Methodtest_gather_qmm_sorted
(self)
python/tests/test_quantized.py:945
Methodtest_gaussian_nll_loss
(self)
python/tests/test_losses.py:273
Methodtest_gelu
(self)
python/tests/test_nn.py:894
Methodtest_gemm
(self)
python/tests/test_double.py:158
Methodtest_gemv_gemm_same_precision
(self)
python/tests/test_blas.py:1356
Methodtest_get_and_set_item
(self)
python/tests/test_double.py:133
Methodtest_getitem_with_list
(self)
python/tests/test_array.py:2015
Methodtest_global_rng
(self)
python/tests/test_random.py:11
Methodtest_glorot_normal
(self)
python/tests/test_init.py:56
Methodtest_glorot_uniform
(self)
python/tests/test_init.py:65
Methodtest_glu
(self)
python/tests/test_nn.py:1091
Methodtest_grad
(self)
python/tests/test_autograd.py:73
Methodtest_grad
(slow, fast, args)
python/tests/test_fast_sdpa.py:619
Methodtest_grad_ids_pre_post
(self)
python/tests/test_autograd.py:869
Methodtest_grad_kwargs
(self)
python/tests/test_autograd.py:184
Methodtest_grad_of_module
(self)
python/tests/test_nn.py:264
Methodtest_grad_trees
(self)
python/tests/test_autograd.py:124
Methodtest_grad_with_copies
(self)
python/tests/test_autograd.py:858
Methodtest_grad_with_inplace_update
(self)
python/tests/test_autograd.py:890
Methodtest_group_norm
(self)
python/tests/test_nn.py:376
Methodtest_groups
(self)
python/tests/mpi_test_distributed.py:15
Methodtest_groups
(self)
python/tests/nccl_test_distributed.py:52
Methodtest_groups
(self)
python/tests/ring_test_distributed.py:15
Methodtest_gru
(self)
python/tests/test_nn.py:1927
Methodtest_gumbel
(self)
python/tests/test_random.py:292
Methodtest_hadamard
(self)
python/tests/test_ops.py:2936
Methodtest_hadamard_grad_vmap
(self)
python/tests/test_ops.py:3031
Methodtest_hamming_general
(self)
python/tests/test_ops.py:1484
Methodtest_hanning_general
(self)
python/tests/test_ops.py:1472
Methodtest_hard_shrink
(self)
python/tests/test_nn.py:1105
Methodtest_hard_tanh
(self)
python/tests/test_nn.py:1097
Methodtest_hardswish
(self)
python/tests/test_nn.py:1082
Methodtest_he_normal
(self)
python/tests/test_init.py:74
Methodtest_he_uniform
(self)
python/tests/test_init.py:83
Methodtest_hinge_loss
(self)
python/tests/test_losses.py:363
Methodtest_huber_loss
(self)
python/tests/test_losses.py:369
Methodtest_identity
(self)
python/tests/test_nn.py:357
Methodtest_identity
(self)
python/tests/test_init.py:46
Methodtest_ifftshift
(self)
python/tests/test_fft.py:319
Methodtest_iinfo
(self)
python/tests/test_array.py:110
Methodtest_indexing
(self)
python/tests/test_array.py:948
Methodtest_indexing_grad
(self)
python/tests/test_array.py:1124
Methodtest_inf_constant
(self)
python/tests/test_compile.py:755
Methodtest_init_from_array
(self)
python/tests/test_array.py:501
Methodtest_init_from_state
(self)
python/tests/test_optimizers.py:534
Methodtest_inner
(self)
python/tests/test_ops.py:2547
Methodtest_inplace
(self)
python/tests/test_array.py:1890
Methodtest_inplace_preserves_ids
(self)
python/tests/test_array.py:1942
Methodtest_instance_norm
(self)
python/tests/test_nn.py:412
Methodtest_int_type
(self)
python/tests/test_array.py:365
Methodtest_integer_power
(self)
python/tests/test_ops.py:3210
← previousnext →4,301–4,400 of 4,946, ranked by callers