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
Functionbinary_op_dispatch_dims
mlx/backend/cpu/binary.h:109
Functionbinary_op_dispatch_dims
mlx/backend/cpu/binary_two.h:53
Functionbinary_ss
mlx/backend/metal/kernels/binary.h:4
Functionbinary_ss
mlx/backend/metal/kernels/binary_two.h:4
Functionbinary_sv
mlx/backend/metal/kernels/binary.h:13
Functionbinary_sv
mlx/backend/metal/kernels/binary_two.h:16
Functionbinary_sv2
mlx/backend/metal/kernels/binary.h:70
Functionbinary_sv2
mlx/backend/metal/kernels/binary_two.h:88
Functionbinary_vs
mlx/backend/metal/kernels/binary.h:32
Functionbinary_vs
mlx/backend/metal/kernels/binary_two.h:40
Functionbinary_vs2
mlx/backend/metal/kernels/binary.h:90
Functionbinary_vs2
mlx/backend/metal/kernels/binary_two.h:113
Functionbinary_vv
mlx/backend/metal/kernels/binary.h:51
Functionbinary_vv
mlx/backend/metal/kernels/binary_two.h:64
Functionbinary_vv2
mlx/backend/metal/kernels/binary.h:110
Functionbinary_vv2
mlx/backend/metal/kernels/binary_two.h:138
Methodbits_to_bfloat
mlx/types/limits.h:51
Methodbits_to_half
mlx/types/limits.h:25
Methodbitwise
(a, b)
python/tests/test_ops.py:2871
Functionblackman
mlx/ops.cpp:2522
Functionblock_masked_gemm
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.h:52
Methodblock_sort
mlx/backend/metal/kernels/sort.h:270
Methodblock_sort
mlx/backend/metal/kernels/sort.h:446
Functionblock_sort_nc
mlx/backend/metal/kernels/sort.h:374
Functionbluestein_fft
mlx/backend/metal/kernels/fft.h:374
Methodbool
mlx/backend/metal/kernels/gemv_masked.h:13
Methodbool
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_masked.h:14
Methodbuffer_info
python/src/buffer.h:64
Functionbuffer_size
mlx/array.h:359
Methodbuild_command
mlx/backend/cpu/jit_compiler.cpp:89
Methodbuild_extension
(self, ext: CMakeExtension)
setup.py:71
Methodbuild_extension
(self, ext: CMakeExtension)
python/mlx/extension.py:25
Methodbuild_library_
mlx/backend/metal/device.cpp:547
Methodcalc
mlx/3rdparty/pocketfft.h:302
Methodcall
(a)
python/tests/test_export_import.py:562
Methodcall_impl
python/src/transforms.cpp:630
Methodcall_impl
python/src/transforms.cpp:896
Methodcall_kernel
(a: mx.array, source)
python/tests/test_fast.py:959
Methodcallback
(args)
python/tests/test_export_import.py:513
Methodcapacity
mlx/small_vector.h:282
Methodcapacity
mlx/backend/cuda/lru_cache.h:45
Methodcat_constant
(x)
python/tests/test_vmap.py:563
Methodcat_fun
(x, y)
python/tests/test_vmap.py:560
Functioncelu
(x)
benchmarks/python/comparative/bench_mlx.py:234
Functioncelu
(x)
benchmarks/python/comparative/bench_torch.py:156
Methodcfftp
mlx/3rdparty/pocketfft.h:1537
Methodcheck
(x, axes)
python/tests/test_reduce.py:136
Functioncheck_cublas_error
mlx/backend/cuda/cublas_utils.cpp:83
Functioncheck_cuda_error
mlx/backend/cuda/utils.cpp:13
Functioncheck_cudnn_error
mlx/backend/cuda/cudnn_utils.cpp:66
Methodcheck_eigs_and_vecs
(A_np, kwargs={})
python/tests/test_double.py:235
Methodcheck_eigs_and_vecs
(A_np, kwargs={})
python/tests/test_linalg.py:354
Methodcheck_initialized
python/src/array.cpp:36
Methodcheck_jointly_gaussian
(data, mean, cov)
python/tests/test_random.py:171
Functioncheck_nvrtc_error
mlx/backend/cuda/jit_module.cpp:23
Methodcheck_slices
(arr_np, *idx_np)
python/tests/test_array.py:1058
Functioncholesky_impl
mlx/backend/cpu/cholesky.cpp:13
Functioncholesky_inv
mlx/linalg.cpp:401
Methodclear
Clear without reverting back to inline storage.
mlx/small_vector.h:438
Methodclear
mlx/backend/metal/kernels/steel/gemm/mma.h:237
Methodclear
mlx/backend/metal/kernels/steel/attn/mma.h:260
Methodclear
mlx/backend/common/buffer_cache.h:87
Functionclear_cache
mlx/backend/no_gpu/allocator.cpp:132
Functionclear_cache
mlx/backend/metal/allocator.cpp:264
Functionclear_streams
mlx/backend/no_gpu/eval.cpp:29
Functionclear_streams
mlx/backend/metal/eval.cpp:90
Functionclear_streams
mlx/backend/cuda/eval.cpp:83
Functionclip_grad_norm
Clips the global norm of the gradients. This function ensures that the global norm of the gradients does not exceed ``max_norm``. It scales d
python/mlx/optimizers/optimizers.py:951
Methodclosure
(y)
python/tests/test_compile.py:108
Methodcmplx
mlx/3rdparty/pocketfft.h:234
Functioncol_reduce_2pass
mlx/backend/metal/kernels/reduction/reduce_col.h:302
Functioncol_reduce_longcolumn
mlx/backend/metal/kernels/reduction/reduce_col.h:97
Functioncol_reduce_looped
mlx/backend/metal/kernels/reduction/reduce_col.h:163
Functioncol_reduce_small
mlx/backend/metal/kernels/reduction/reduce_col.h:4
Functioncollapse_contiguous_dims
mlx/backend/common/utils.cpp:20
Methodcommit
mlx/backend/cuda/worker.cpp:44
Methodcommunication_stream
mlx/distributed/ring/ring.cpp:465
Methodcommunication_stream
mlx/distributed/mpi/mpi.cpp:359
Methodcommunication_stream
mlx/distributed/jaccl/jaccl.cpp:66
Methodcommunication_stream
mlx/distributed/nccl/nccl.cpp:320
Methodcomp_twiddle
mlx/3rdparty/pocketfft.h:1512
Methodcomp_twiddle
mlx/3rdparty/pocketfft.h:2310
Functioncompile
mlx/compile.cpp:1098
Functioncompile_across_streams
tests/compile_tests.cpp:511
Functioncompile_available_for_device
mlx/backend/no_cpu/compiled.cpp:12
Functioncompile_broadcast_add
tests/compile_tests.cpp:722
Functioncompile_shapeless_not_ok
tests/compile_tests.cpp:685
Functioncompile_shapeless_ok
tests/compile_tests.cpp:690
Functioncompile_unary_inner
tests/compile_tests.cpp:605
Functioncompile_unary_outer
tests/compile_tests.cpp:610
Methodcomplex128_t
mlx/types/complex.h:17
Methodcomplex64_t
mlx/types/complex.h:36
Methodcomplex64_t
Constructors
mlx/backend/metal/kernels/complex.h:25
Functioncompute_dynamic_offset
mlx/backend/metal/slicing.cpp:45
Functionconcatenate
(axis, x, y)
benchmarks/python/comparative/bench_mlx.py:307
Functionconcatenate
(axis, x, y)
benchmarks/python/comparative/bench_torch.py:265
Methodcondition
mlx/backend/metal/kernels/atomic.h:193
Methodcondition
mlx/backend/metal/kernels/atomic.h:207
Methodcondition
mlx/backend/metal/kernels/atomic.h:220
Methodcondition
mlx/backend/metal/kernels/atomic.h:232
← previousnext →2,501–2,600 of 4,946, ranked by callers