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
↓ 20 callersFunctioncollapse_contiguous_dims
mlx/backend/common/utils.h:53
↓ 20 callersMethodfinfo
mlx/utils.cpp:314
↓ 20 callersFunctiongetenv
mlx/distributed/jaccl/lib/jaccl/jaccl.cpp:61
↓ 20 callersFunctionisinf
mlx/ops.cpp:1846
↓ 20 callersFunctionmean
mlx/ops.h:608
↓ 20 callersFunctionremove_index
mlx/backend/common/utils.h:233
↓ 20 callersFunctionvmap_binary_op
mlx/primitives.cpp:24
↓ 20 callersMethodwrite
mlx/io/load.h:156
↓ 19 callersMethodall_sum
mlx/distributed/mpi/mpi.cpp:389
↓ 19 callersFunctioncos
mlx/backend/cpu/simd/math.h:128
↓ 19 callersFunctionfn
(x)
benchmarks/python/synchronize_bench.py:28
↓ 19 callersFunctiongreater_equal
mlx/ops.cpp:1785
↓ 19 callersFunctionis_scalar
mlx/backend/common/compiled.h:47
↓ 19 callersFunctionmax
mlx/ops.cpp:2322
↓ 19 callersFunctionminimum
mlx/ops.cpp:2956
↓ 19 callersFunctionquantization_mode_to_string
mlx/primitives.cpp:3419
↓ 19 callersFunctionsin
mlx/backend/cpu/simd/math.h:119
↓ 19 callersFunctiontest_arg_reduce_small
tests/arg_reduce_tests.cpp:10
↓ 18 callersFunctioncategorical
mlx/random.cpp:395
↓ 18 callersMethoderase
mlx/backend/metal/resident.cpp:40
↓ 18 callersFunctionerfinv
mlx/ops.cpp:3163
↓ 18 callersFunctionget_block_dims
mlx/backend/metal/utils.cpp:61
↓ 18 callersFunctiongreater
mlx/ops.cpp:1777
↓ 18 callersFunctionlog
(verbose, *args, **kwargs)
python/mlx/_distributed_utils/common.py:121
↓ 18 callersFunctionresult_type
The type from promoting the arrays' types with one another. */
mlx/utils.h:89
↓ 18 callersFunctionrun
(dtype)
benchmarks/python/hadamard_bench.py:22
↓ 18 callersFunctionstop_gradient
mlx/ops.cpp:3172
↓ 17 callersMethodappend_ndim
mlx/backend/cuda/jit_module.h:56
↓ 17 callersFunctioncheck_transpose
mlx/backend/cuda/matmul.cpp:20
↓ 17 callersMethodclear
mlx/backend/metal/kernels/steel/gemm/nax.h:566
↓ 17 callersMethoddata
mlx/distributed/jaccl/lib/jaccl/rdma.h:127
↓ 17 callersFunctiondivmod
mlx/ops.cpp:2929
↓ 17 callersFunctiondtype_to_string
mlx/dtype_utils.cpp:7
↓ 17 callersFunctionensure_row_contiguous
mlx/backend/metal/quantized.cpp:53
↓ 17 callersFunctionexporter
mlx/export.cpp:891
↓ 17 callersFunctiongemm_loop
mlx/backend/metal/kernels/steel/gemm/gemm_nax.h:26
↓ 17 callersFunctionirfftn
mlx/fft.cpp:220
↓ 17 callersMethodreset
python/src/transforms.cpp:939
↓ 17 callersFunctionrfftn
mlx/fft.cpp:198
↓ 17 callersMethodsize
mlx/distributed/distributed.cpp:133
↓ 16 callersMethodclear
mlx/backend/cuda/lru_cache.h:70
↓ 16 callersFunctionenqueue
mlx/scheduler.h:123
↓ 16 callersFunctionget_2d_grid_dims
mlx/backend/metal/utils.cpp:66
↓ 16 callersMethodget_active_memory
mlx/backend/cuda/allocator.cpp:347
↓ 16 callersFunctionlog_error
(*args, **kwargs)
python/mlx/_distributed_utils/common.py:133
↓ 16 callersFunctionlogsumexp
mlx/ops.h:846
↓ 16 callersFunctionnew_stream
mlx/stream.cpp:66
↓ 16 callersMethodnext
mlx/backend/metal/kernels/quantized.h:671
↓ 16 callersMethodnext
mlx/backend/metal/kernels/fp_quantized.h:244
↓ 16 callersFunctionremainder
mlx/ops.cpp:2914
↓ 16 callersFunctionsign
mlx/ops.cpp:2791
↓ 16 callersFunctionsize_of
mlx/dtype.h:105
↓ 16 callersFunctionstack
Stack arrays along a new axis */
mlx/ops.cpp:1267
↓ 16 callersFunctiontree_equal
(fn, *args)
python/tests/test_optimizers.py:37
↓ 15 callersMethodclear
mlx/compile.cpp:374
↓ 15 callersFunctioncopy_cpu_inplace
mlx/backend/cpu/copy.cpp:298
↓ 15 callersFunctiondiagonal
mlx/ops.cpp:5937
↓ 15 callersFunctionerf
mlx/ops.cpp:3154
↓ 15 callersFunctioneye
mlx/ops.h:104
↓ 15 callersFunctionfree
mlx/allocator.h:58
↓ 15 callersFunctionifft
Compute the one-dimensional inverse Fourier Transform. */
mlx/fft.h:69
↓ 15 callersFunctionifftn
mlx/fft.cpp:176
↓ 15 callersMethodis_tracer
mlx/array.cpp:163
↓ 15 callersFunctionlog1p
mlx/ops.cpp:3127
↓ 15 callersFunctionlogaddexp
mlx/ops.cpp:3134
↓ 15 callersFunctionrope_orig
(x, dims, traditional, base, scale, offset, freqs=None)
python/tests/test_fast.py:10
↓ 15 callersFunctionstore
mlx/backend/cpu/simd/base_simd.h:48
↓ 15 callersMethodstore_result_slice
mlx/backend/metal/kernels/steel/gemm/mma.h:548
↓ 15 callersMethodupdate
(grads)
python/tests/test_optimizers.py:400
↓ 15 callersMethodwait
mlx/backend/cuda/fence.cpp:21
↓ 14 callersFunction_reduce
(loss: mx.array, reduction: Reduction = "none")
python/mlx/nn/losses.py:11
↓ 14 callersFunctioneinsum
mlx/einsum.cpp:851
↓ 14 callersMethodemplace
mlx/backend/cuda/lru_cache.h:84
↓ 14 callersFunctionis_index_scalar
python/src/indexing.cpp:19
↓ 14 callersFunctionlog
mlx/distributed/ring/ring.cpp:105
↓ 14 callersFunctionlogical_or
mlx/ops.cpp:2817
↓ 14 callersFunctionmlx_ref_attn
(q, k, v, scale=1.0, mask=None, sinks=None)
python/tests/test_fast_sdpa.py:11
↓ 14 callersFunctionpad
Pad an array with a constant value */
mlx/ops.cpp:1404
↓ 14 callersMethodquantize
(self)
python/mlx/nn/layers/quantized.py:369
↓ 14 callersMethodrelease
mlx/backend/metal/allocator.cpp:209
↓ 14 callersFunctionrepeat
array repeat with axis */
mlx/ops.cpp:1294
↓ 14 callersFunctionrsqrt
mlx/ops.cpp:3801
↓ 13 callersFunctionbernoulli
mlx/random.h:179
↓ 13 callersFunctionget_quantized_kernel_wrapped
mlx/backend/metal/quantized.cpp:20
↓ 13 callersFunctionid
A unique identifier for an array. */
mlx/array.h:222
↓ 13 callersMethodinit
mlx/array.h:588
↓ 13 callersFunctioninitializer
(a: mx.array)
python/mlx/nn/init.py:31
↓ 13 callersFunctionless_equal
mlx/ops.cpp:1807
↓ 13 callersFunctionload
mlx/backend/metal/kernels/fft/readwrite.h:100
↓ 13 callersFunctionlogical_and
mlx/ops.cpp:2803
↓ 13 callersFunctionput_along_axis
mlx/ops.cpp:3551
↓ 13 callersMethodrbegin
mlx/small_vector.h:262
↓ 13 callersFunctionrfft
Compute the one-dimensional Fourier Transform on a real input. */
mlx/fft.h:150
↓ 13 callersFunctionsave_gguf
mlx/io/gguf.cpp:294
↓ 13 callersFunctionscatter_add
mlx/ops.h:1224
↓ 13 callersMethodseed
mlx/random.cpp:16
↓ 13 callersFunctiontree_flatten
python/src/trees.cpp:236
↓ 13 callersMethodwait
mlx/array.cpp:144
↓ 12 callersFunctionadjust_matrix_offsets
mlx/backend/metal/kernels/fp_quantized.h:888
↓ 12 callersMethodbegin
mlx/distributed/jaccl/lib/jaccl/rdma.h:132
← previousnext →201–300 of 4,946, ranked by callers