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
Methodfun
(x)
python/tests/ring_test_distributed.py:120
Methodfun
()
python/tests/test_export_import.py:30
Methodfun2
(x)
python/tests/test_compile.py:1197
Functionfun_creats_array
tests/compile_tests.cpp:70
Methodfunc
(x)
python/tests/test_autograd.py:495
Methodfunc
(x)
python/tests/test_ops.py:1133
Methodg
(x)
python/tests/test_fft.py:393
Functiongather
(x, idx)
benchmarks/python/gather_bench.py:11
Methodgather
(a, idx)
python/tests/test_vmap.py:401
Functiongather_axis
mlx/backend/cpu/indexing.cpp:258
Functiongather_impl
mlx/backend/metal/kernels/indexing/gather.h:8
Functiongather_mm
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.h:251
Functiongather_mm
(x, w1, w2, indices, sort)
benchmarks/python/gather_qmm_bench.py:53
Functiongather_mm
(x, w1, w2, indices, sort)
benchmarks/python/gather_mm_bench.py:45
Methodgather_mm_ref
(a, b, rhs)
python/tests/test_blas.py:1240
Functiongather_mm_rhs
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather.h:20
Functiongather_mm_rhs_nax
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_gather_nax.h:19
Functiongather_mm_simulate
(x, w, indices)
benchmarks/python/gather_qmm_bench.py:28
Functiongather_mm_simulate
(x, w, indices)
benchmarks/python/gather_mm_bench.py:28
Methodgather_mm_test
(a, b, rhs)
python/tests/test_blas.py:1244
Methodgather_qmm_ref
(x, w, s, b, lhs, rhs, trans, sort)
python/tests/test_quantized.py:1060
Methodgather_sort
(x, indices)
python/tests/test_quantized.py:958
Functiongaussian_nll_loss
r""" Computes the negative log likelihood loss for a Gaussian distribution. The loss is given by: .. math:: \frac{1}{2}\left(\lo
python/mlx/nn/losses.py:257
Functiongc_func_dealloc
python/src/mlx_func.cpp:53
Functiongc_func_get_doc
python/src/mlx_func.cpp:37
Functiongc_func_get_sig
python/src/mlx_func.cpp:41
Functiongc_func_getattro
python/src/mlx_func.cpp:72
Functiongc_func_tp_clear
python/src/mlx_func.cpp:31
Functiongc_func_tp_traverse
python/src/mlx_func.cpp:21
Functiongc_func_vectorcall
python/src/mlx_func.cpp:45
Methodgconv
(x, w)
python/tests/test_vmap.py:725
Functiongelu
(x)
benchmarks/python/compile_bench.py:12
Functiongemm
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.h:86
Functiongemm_epilogue
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_fused_nax.h:20
Methodgemm_loop
mlx/backend/metal/kernels/steel/gemm/gemm.h:79
Methodgemm_loop
mlx/backend/metal/kernels/steel/attn/attn.h:80
Functiongemm_loop_unaligned
mlx/backend/metal/kernels/quantized_utils.h:40
Functiongemm_nn_mlx
(a, b)
benchmarks/python/blas/bench_gemm.py:33
Functiongemm_nn_torch
(a, b)
benchmarks/python/blas/bench_gemm.py:70
Functiongemm_nt_mlx
(a, b)
benchmarks/python/blas/bench_gemm.py:42
Functiongemm_nt_torch
(a, b)
benchmarks/python/blas/bench_gemm.py:80
Functiongemm_splitk
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.h:21
Functiongemm_splitk_accum
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.h:172
Functiongemm_splitk_accum_axpby
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk.h:199
Functiongemm_splitk_nax
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_splitk_nax.h:23
Functiongemm_tn_mlx
(a, b)
benchmarks/python/blas/bench_gemm.py:51
Functiongemm_tn_torch
(a, b)
benchmarks/python/blas/bench_gemm.py:90
Functiongemm_tt_mlx
(a, b)
benchmarks/python/blas/bench_gemm.py:60
Functiongemm_tt_torch
(a, b)
benchmarks/python/blas/bench_gemm.py:100
Functiongemv_mlx
(m, v)
benchmarks/python/blas/bench_gemv.py:48
Functiongemv_t_masked
Batch ndim > 1 */
mlx/backend/metal/kernels/gemv_masked.h:744
Functiongemv_t_mlx
(m, v)
benchmarks/python/blas/bench_gemv.py:57
Functiongemv_t_torch
(m, v)
benchmarks/python/blas/bench_gemv.py:77
Functiongemv_torch
(m, v)
benchmarks/python/blas/bench_gemv.py:67
Functiongeneral_inv
mlx/backend/cpu/inverse.cpp:12
Functionget_active_memory
mlx/backend/no_gpu/allocator.cpp:106
Functionget_arange_kernel
mlx/backend/metal/nojit_kernels.cpp:9
Functionget_architecture_gen
mlx/backend/metal/device.h:149
Functionget_binary_kernel
mlx/backend/metal/nojit_kernels.cpp:25
Functionget_binary_two_kernel
mlx/backend/metal/nojit_kernels.cpp:34
Functionget_bytes_per_pack
mlx/backend/metal/kernels/quantized.h:23
Functionget_bytes_per_pack
mlx/backend/metal/kernels/quantized_nax.h:26
Functionget_cache_memory
No-ops for common allocator
mlx/backend/no_gpu/allocator.cpp:123
Methodget_command_encoder
mlx/backend/metal/device.cpp:473
Functionget_command_queue
mlx/backend/metal/device.h:97
Methodget_coord
mlx/backend/metal/kernels/steel/gemm/nax.h:45
Methodget_coord
mlx/backend/metal/kernels/steel/gemm/mma.h:49
Methodget_coord
mlx/backend/metal/kernels/steel/attn/nax.h:45
Methodget_coord
mlx/backend/metal/kernels/steel/attn/mma.h:71
Functionget_copy_kernel
mlx/backend/metal/nojit_kernels.cpp:51
Functionget_dynamic_copy_kernel
mlx/backend/metal/nojit_kernels.cpp:59
Functionget_fft_kernel
mlx/backend/metal/nojit_kernels.cpp:287
Methodget_function_
mlx/backend/metal/device.cpp:578
Functionget_gather_qmm_kernel
mlx/backend/metal/nojit_kernels.cpp:304
Functionget_gather_qmm_nax_kernel
mlx/backend/metal/nojit_kernels.cpp:395
Functionget_gemv_masked_kernel
mlx/backend/metal/nojit_kernels.cpp:229
Functionget_gflop_count
(B, M, N, K)
benchmarks/python/sdpa_bench.py:158
Functionget_grid_and_block_common
mlx/backend/common/utils.cpp:221
Methodget_kernel_
mlx/backend/metal/device.cpp:630
Methodget_linked_functions_
mlx/backend/metal/device.cpp:723
Functionget_logsumexp_kernel
mlx/backend/metal/nojit_kernels.cpp:75
Functionget_max_ops_mb_per_buffer
mlx/backend/metal/device.h:152
Functionget_mb_sort_kernel
mlx/backend/metal/nojit_kernels.cpp:103
Methodget_mem
()
python/tests/test_array.py:2223
Functionget_memory_limit
mlx/backend/no_gpu/allocator.cpp:118
Functionget_memory_limit
mlx/backend/metal/allocator.cpp:240
Functionget_memory_size
mlx/backend/no_gpu/apple_memory.h:9
Functionget_memory_size
mlx/backend/no_gpu/linux_memory.h:9
Functionget_pack_factor
mlx/backend/metal/kernels/quantized.h:18
Functionget_pack_factor
mlx/backend/metal/kernels/fp_quantized.h:21
Functionget_pack_factor
mlx/backend/metal/kernels/fp_quantized_nax.h:21
Functionget_pack_factor
mlx/backend/metal/kernels/quantized_nax.h:21
Functionget_peak_memory
mlx/backend/no_gpu/allocator.cpp:109
Functionget_peak_memory
mlx/backend/metal/allocator.cpp:255
Functionget_qmm_nax_kernel
mlx/backend/metal/nojit_kernels.cpp:387
Functionget_quantized_kernel
mlx/backend/metal/nojit_kernels.cpp:296
Functionget_reduce_init_kernel
mlx/backend/metal/nojit_kernels.cpp:113
Functionget_reduce_kernel
mlx/backend/metal/nojit_kernels.cpp:122
Functionget_scan_kernel
mlx/backend/metal/nojit_kernels.cpp:82
Methodget_slopes
(n: int)
python/mlx/nn/layers/positional_encoding.py:137
← previousnext →2,901–3,000 of 4,946, ranked by callers