MCPcopy Create free account

hub / github.com/ml-explore/mlx / functions

Functions4,946 in github.com/ml-explore/mlx

Methodcondition
mlx/backend/metal/kernels/atomic.h:243
Methodconnect
mlx/distributed/utils.cpp:163
Methodconnect
mlx/distributed/jaccl/lib/jaccl/tcp.cpp:163
Functionconst_for_loop
mlx/backend/metal/kernels/steel/utils/integral_constant.h:103
Methodconst_func
(a, b)
python/tests/test_vmap.py:541
Functionconstant
r"""An initializer that returns an array filled with ``value``. Args: value (float): The value to fill the array with. dtype (Dty
python/mlx/nn/init.py:9
Functioncontiguous_scan
mlx/backend/metal/kernels/scan.h:236
Functionconv1d
(x, y)
benchmarks/python/comparative/bench_mlx.py:120
Functionconv1d
(x, y)
benchmarks/python/comparative/bench_torch.py:71
Functionconv2d
(x, y)
benchmarks/python/comparative/bench_mlx.py:127
Functionconv2d
(x, y)
benchmarks/python/comparative/bench_torch.py:81
Methodconv_flip
(kernel)
python/tests/test_conv.py:920
Methodconv_general_pt
( inp, wt, stride, padding, kernel_dilation, input_dilation, groups, flip )
python/tests/test_conv.py:734
Methodconv_manual_flip
(kernel)
python/tests/test_conv.py:936
Methodconv_out_shape
mlx/primitives.cpp:1314
Methodconv_t
(w)
python/tests/test_conv.py:890
Functionconvert_vector
mlx/backend/cuda/cudnn_utils.h:49
Functioncopy
(x)
benchmarks/python/hadamard_bench.py:17
Methodcopy_and_norm
mlx/3rdparty/pocketfft.h:2210
Functioncopy_dims
mlx/backend/cpu/copy.cpp:33
Functioncopy_g
mlx/backend/metal/kernels/copy.h:113
Functioncopy_g_nd1
mlx/backend/metal/kernels/copy.h:78
Functioncopy_g_nd2
mlx/backend/metal/kernels/copy.h:88
Functioncopy_g_nd3
mlx/backend/metal/kernels/copy.h:100
Functioncopy_general
mlx/backend/cpu/copy.cpp:142
Functioncopy_general_general
mlx/backend/cpu/copy.cpp:57
Functioncopy_gg
mlx/backend/metal/kernels/copy.h:175
Functioncopy_gg_dynamic
mlx/backend/metal/kernels/copy.h:246
Functioncopy_gg_dynamic_nd1
mlx/backend/metal/kernels/copy.h:204
Functioncopy_gg_dynamic_nd2
mlx/backend/metal/kernels/copy.h:218
Functioncopy_gg_dynamic_nd3
mlx/backend/metal/kernels/copy.h:232
Functioncopy_gg_nd1
mlx/backend/metal/kernels/copy.h:139
Functioncopy_gg_nd2
mlx/backend/metal/kernels/copy.h:151
Functioncopy_gg_nd3
mlx/backend/metal/kernels/copy.h:163
Functioncopy_s
mlx/backend/metal/kernels/copy.h:4
Functioncopy_s2
mlx/backend/metal/kernels/copy.h:40
Functioncopy_single
mlx/backend/cpu/copy.cpp:16
Functioncopy_v
mlx/backend/metal/kernels/copy.h:22
Functioncopy_v2
mlx/backend/metal/kernels/copy.h:59
Functioncopy_vector
mlx/backend/cpu/copy.cpp:25
Methodcosine
(a, b)
python/tests/test_nn.py:1986
Functioncosine_decay
r"""Make a cosine decay scheduler. Args: init (float): Initial value. decay_steps (int): Number of steps to decay over. The decay
python/mlx/optimizers/schedulers.py:61
Functioncosine_similarity_loss
r""" Computes the cosine similarity between the two inputs. The cosine similarity loss is given by .. math:: \frac{x_1 \cdot x_
python/mlx/nn/losses.py:522
Methodcost_guess
mlx/3rdparty/pocketfft.h:383
Methodcount_prims
(outputs)
python/tests/test_compile.py:178
Methodcreate
mlx/distributed/nccl/nccl.cpp:280
MethodcreateTests
(self, *args, **kwargs)
python/tests/mlx_tests.py:26
Methodcreate_threads
mlx/3rdparty/pocketfft.h:684
Functioncross_entropy
Computes the cross entropy loss. Args: logits (array): The unnormalized logits. targets (array): The ground truth values. Th
python/mlx/nn/losses.py:23
Functioncross_entropy
(targets, x)
benchmarks/python/comparative/bench_mlx.py:255
Functioncross_entropy
(targets, x)
benchmarks/python/comparative/bench_torch.py:212
Functioncuda_kernel
mlx/backend/cuda/custom_kernel.cpp:144
Functioncumsum
(axis, x)
benchmarks/python/comparative/bench_mlx.py:314
Functioncumsum
(axis, x)
benchmarks/python/comparative/bench_torch.py:273
Functiondata
mlx/array.h:373
Methoddata
mlx/small_vector.h:233
Functiondata_size
The size (in elements) of the underlying buffer the array points to. * * This can be different than the actual size of the array if the array ha
mlx/array.h:348
Methoddealloc
mlx/3rdparty/pocketfft.h:194
Functiondecorator
(klass)
python/mlx/nn/layers/activations.py:12
Functiondelayload_helper
Called by system when failed to locate a lazy-loaded DLL.
mlx/backend/cuda/delayload.cpp:61
Functiondequantize
mlx/backend/metal/kernels/quantized.h:484
Functiondequantize
mlx/backend/metal/kernels/fp_quantized.h:139
Functiondequantize
mlx/backend/metal/kernels/fp_quantized_nax.h:63
Functiondequantize
mlx/backend/metal/kernels/quantized_nax.h:487
Functiondequantize_scale
mlx/backend/cpu/quantized.cpp:50
Functiondequantize_scale
mlx/backend/metal/kernels/fp_quantized.h:31
Functiondequantize_scale
mlx/backend/metal/kernels/fp_quantized_nax.h:31
Functiondeserialize
mlx/export.cpp:146
Functiondeserialize_primitive
mlx/export.cpp:305
Functiondeserialize_tuple
mlx/export.cpp:222
Functiondeserialize_variant
mlx/export.cpp:206
Functiondevice_count
mlx/backend/no_cpu/device_info.cpp:11
Functiondevice_count
mlx/backend/cpu/device_info.cpp:100
Functiondevice_count
mlx/backend/no_gpu/device_info.cpp:11
Functiondevice_count
mlx/backend/metal/device_info.cpp:15
Functiondispatch
mlx/backend/cpu/encoder.h:44
Functiondispatch_float_types
mlx/dtype_utils.h:69
Functiondispatch_gather
mlx/backend/cpu/indexing.cpp:150
Functiondispatch_gather_axis
mlx/backend/cpu/indexing.cpp:304
Functiondispatch_inexact_types
mlx/dtype_utils.h:80
Functiondispatch_int_float_types
mlx/dtype_utils.h:93
Functiondispatch_quantize
mlx/backend/cpu/quantized.cpp:1214
Functiondispatch_real_types
mlx/dtype_utils.h:106
Functiondispatch_scatter
mlx/backend/cpu/indexing.cpp:472
Functiondispatch_scatter_axis
mlx/backend/cpu/indexing.cpp:645
Functiondispatch_scatter_axis_op
mlx/backend/cpu/indexing.cpp:628
Functiondispatch_scatter_inds
mlx/backend/cpu/indexing.cpp:446
Functiondo_attention_bench
(f, q, k, v, scale, mask=None, transpose=False)
benchmarks/python/sdpa_bench.py:119
Functiondtype_from_str
(x)
benchmarks/python/comparative/bench_mlx.py:27
Functiondtype_from_str
(x)
benchmarks/python/comparative/bench_torch.py:26
Methoddtypes
(self)
python/tests/test_blas.py:14
Methoddummy_loss
(model, x, y)
python/tests/mlx_distributed_tests.py:171
Functioneig_impl
mlx/backend/cpu/eig.cpp:177
Functioneigh_impl
mlx/backend/cpu/eigh.cpp:143
Functioneinsum_attention
(x)
benchmarks/python/einsum_bench.py:64
Functionelem_to_loc_1
mlx/backend/metal/kernels/utils.h:131
Functionelem_to_loc_2
mlx/backend/metal/kernels/utils.h:136
Functionelem_to_loc_3
mlx/backend/metal/kernels/utils.h:141
Methodelems
mlx/backend/metal/kernels/steel/gemm/nax.h:631
Functionelu
(x)
benchmarks/python/comparative/bench_mlx.py:213
← previousnext →2,601–2,700 of 4,946, ranked by callers