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
Methodparse_h_string
(h_str)
python/tests/test_ops.py:2971
Functionpartition
mlx/backend/cpu/sort.cpp:218
Methodpass11
mlx/3rdparty/pocketfft.h:1253
Methodpass2
mlx/3rdparty/pocketfft.h:842
Methodpass3
mlx/3rdparty/pocketfft.h:889
Methodpass4
mlx/3rdparty/pocketfft.h:928
Methodpass5
mlx/3rdparty/pocketfft.h:1002
Methodpass7
mlx/3rdparty/pocketfft.h:1073
Methodpass8
mlx/3rdparty/pocketfft.h:1140
Methodpass_all
mlx/3rdparty/pocketfft.h:1422
Methodpassg
mlx/3rdparty/pocketfft.h:1313
Functionper_thread_row_reduce
mlx/backend/metal/kernels/reduction/reduce_row.h:19
Methodpocketfft_c
mlx/3rdparty/pocketfft.h:2471
Methodpocketfft_r
mlx/3rdparty/pocketfft.h:2508
Functionpositive_number
(x)
python/mlx/_distributed_utils/common.py:114
Methodpost_recv_all
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:597
Methodpost_recv_all
mlx/distributed/jaccl/lib/jaccl/mesh_impl.h:402
Methodpost_send_all
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:610
Methodpost_send_all
mlx/distributed/jaccl/lib/jaccl/mesh_impl.h:391
Functionprecompiled_cuda_kernel
mlx/backend/cuda/custom_kernel.cpp:244
Functionprelu
(x: mx.array)
benchmarks/python/comparative/bench_mlx.py:185
Functionprelu
(x: torch.Tensor)
benchmarks/python/comparative/bench_torch.py:188
Methodprepare_workspace
mlx/backend/cuda/cudnn_utils.cpp:151
Methodpreprocess_output
Preprocess the output of the command so that extra data can be capture or the format changed on the fly.
python/mlx/_distributed_utils/launch.py:37
Functionprimitive_id
A unique identifier for an arrays primitive. */
mlx/array.h:227
Functionprimitive_state
mlx/export.cpp:296
Functionprint_array
mlx/utils.cpp:242
Functionprint_complex_constant
mlx/backend/common/compiled.h:35
Functionprint_float_constant
mlx/backend/common/compiled.h:19
Functionprint_graph
mlx/graph_utils.cpp:62
Functionprint_int_constant
mlx/backend/common/compiled.h:30
Functionprint_subarray
mlx/utils.cpp:216
Methodprocess
Return the Popen object that refers to the current command.
python/mlx/_distributed_utils/launch.py:27
Methodprocess
(self)
python/mlx/_distributed_utils/launch.py:69
Functionprod
mlx/ops.cpp:2277
Methodpseudo_step_ddp
(grads, params, optimizer)
python/tests/nccl_test_distributed.py:316
Methodpseudo_step_fsdp
(grads, params, optimizer)
python/tests/nccl_test_distributed.py:322
Functionpt_conv_1D
(a, b)
benchmarks/python/conv1d_bench.py:45
Functionpt_conv_2D
(a, b)
benchmarks/python/conv_bench.py:45
Functionpt_conv_2D
(a, b)
benchmarks/python/conv_unaligned_bench.py:39
Functionpt_conv_2D
(a, b)
benchmarks/python/conv2d_bench_cpu.py:40
Functionpt_conv_3D
(a, b, b_prime)
benchmarks/python/conv3d_bench.py:39
Functionpt_conv_3D
(a, b)
benchmarks/python/conv3d_bench_cpu.py:40
Functionpt_conv_3D
(a, b)
benchmarks/python/conv3d_transpose_bench_cpu.py:42
Functionpt_conv_transpose_2D
(a, b)
benchmarks/python/conv2d_transpose_bench_cpu.py:41
Functionpt_conv_transpose_2D
(a, b)
benchmarks/python/conv_transpose_bench.py:44
Functionptr
Get the buffer pointer from the buffer
mlx/allocator.h:25
Functionpy_custom_function_tp_clear
python/src/transforms.cpp:1012
Functionpy_custom_function_tp_traverse
python/src/transforms.cpp:989
Functionpy_function_exporter_tp_traverse
python/src/export.cpp:98
Functionqdot
mlx/backend/metal/kernels/quantized.h:192
Functionqdot
mlx/backend/metal/kernels/fp_quantized.h:82
Functionqdot
mlx/backend/metal/kernels/quantized_nax.h:195
Functionqdot_safe
mlx/backend/metal/kernels/quantized.h:293
Functionqdot_safe
mlx/backend/metal/kernels/fp_quantized.h:103
Functionqdot_safe
mlx/backend/metal/kernels/quantized_nax.h:296
Functionqmm_n_impl
mlx/backend/metal/kernels/quantized.h:1221
Functionqmm_n_nax_tgp_impl
mlx/backend/metal/kernels/quantized_nax.h:1082
Functionqmm_t_impl
mlx/backend/metal/kernels/quantized.h:1094
Functionqmm_t_nax_tgp_impl
mlx/backend/metal/kernels/quantized_nax.h:938
Functionqmv_fast_impl
mlx/backend/metal/kernels/quantized.h:750
Functionqmv_impl
mlx/backend/metal/kernels/quantized.h:817
Functionqmv_quad_impl
mlx/backend/metal/kernels/quantized.h:693
Functionqouter
mlx/backend/metal/kernels/quantized.h:395
Functionqouter
mlx/backend/metal/kernels/fp_quantized.h:125
Functionqouter
mlx/backend/metal/kernels/quantized_nax.h:398
Functionqqmm_impl
mlx/backend/cuda/quantized/no_qqmm_impl.cpp:6
Functionqrf_impl
mlx/backend/cpu/qrf.cpp:12
Functionquantize
mlx/backend/cpu/quantized.cpp:1149
Functionqvm_impl
mlx/backend/metal/kernels/quantized.h:978
Methodradb2
mlx/3rdparty/pocketfft.h:1894
Methodradb3
mlx/3rdparty/pocketfft.h:1924
Methodradb4
mlx/3rdparty/pocketfft.h:1964
Methodradb5
mlx/3rdparty/pocketfft.h:2016
Methodradbg
mlx/3rdparty/pocketfft.h:2075
Functionrader_fft
mlx/backend/metal/kernels/fft.h:219
Methodradf2
mlx/3rdparty/pocketfft.h:1573
Methodradf3
mlx/3rdparty/pocketfft.h:1610
Methodradf4
mlx/3rdparty/pocketfft.h:1649
Methodradf5
mlx/3rdparty/pocketfft.h:1696
Methodradfg
mlx/3rdparty/pocketfft.h:1752
Functionradix10
mlx/backend/metal/kernels/fft/radix.h:173
Functionradix11
mlx/backend/metal/kernels/fft/radix.h:201
Functionradix12
mlx/backend/metal/kernels/fft/radix.h:238
Functionradix13
mlx/backend/metal/kernels/fft/radix.h:290
Functionradix2
mlx/backend/metal/kernels/fft/radix.h:36
Functionradix3
mlx/backend/metal/kernels/fft/radix.h:41
Functionradix7
mlx/backend/metal/kernels/fft/radix.h:122
Functionradix8
mlx/backend/metal/kernels/fft/radix.h:151
Functionradix_butterfly
mlx/backend/metal/kernels/fft.h:54
Functionradix_n_steps
mlx/backend/metal/kernels/fft.h:100
Methodralloc
mlx/3rdparty/pocketfft.h:187
Functionrandint
mlx/random.cpp:271
Methodrandom_grads
(primals, cotangents, outputs)
python/tests/test_autograd.py:659
Functionrandom_take
()
benchmarks/python/single_ops.py:98
Functionrandom_take
()
benchmarks/numpy/single_ops.py:29
Methodrank
mlx/distributed/distributed.cpp:59
Methodrank
mlx/distributed/ring/ring.cpp:469
Methodrank
mlx/distributed/mpi/mpi.cpp:363
Methodrank
mlx/distributed/jaccl/jaccl.cpp:70
← previousnext →3,601–3,700 of 4,946, ranked by callers