Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ml-explore/mlx
/ functions
Functions
4,946 in github.com/ml-explore/mlx
⨍
Functions
4,946
◇
Types & classes
770
↳
Endpoints
5
Method
condition
mlx/backend/metal/kernels/atomic.h:243
Method
connect
mlx/distributed/utils.cpp:163
Method
connect
mlx/distributed/jaccl/lib/jaccl/tcp.cpp:163
Function
const_for_loop
mlx/backend/metal/kernels/steel/utils/integral_constant.h:103
Method
const_func
(a, b)
python/tests/test_vmap.py:541
Function
constant
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
Function
contiguous_scan
mlx/backend/metal/kernels/scan.h:236
Function
conv1d
(x, y)
benchmarks/python/comparative/bench_mlx.py:120
Function
conv1d
(x, y)
benchmarks/python/comparative/bench_torch.py:71
Function
conv2d
(x, y)
benchmarks/python/comparative/bench_mlx.py:127
Function
conv2d
(x, y)
benchmarks/python/comparative/bench_torch.py:81
Method
conv_flip
(kernel)
python/tests/test_conv.py:920
Method
conv_general_pt
( inp, wt, stride, padding, kernel_dilation, input_dilation, groups, flip )
python/tests/test_conv.py:734
Method
conv_manual_flip
(kernel)
python/tests/test_conv.py:936
Method
conv_out_shape
mlx/primitives.cpp:1314
Method
conv_t
(w)
python/tests/test_conv.py:890
Function
convert_vector
mlx/backend/cuda/cudnn_utils.h:49
Function
copy
(x)
benchmarks/python/hadamard_bench.py:17
Method
copy_and_norm
mlx/3rdparty/pocketfft.h:2210
Function
copy_dims
mlx/backend/cpu/copy.cpp:33
Function
copy_g
mlx/backend/metal/kernels/copy.h:113
Function
copy_g_nd1
mlx/backend/metal/kernels/copy.h:78
Function
copy_g_nd2
mlx/backend/metal/kernels/copy.h:88
Function
copy_g_nd3
mlx/backend/metal/kernels/copy.h:100
Function
copy_general
mlx/backend/cpu/copy.cpp:142
Function
copy_general_general
mlx/backend/cpu/copy.cpp:57
Function
copy_gg
mlx/backend/metal/kernels/copy.h:175
Function
copy_gg_dynamic
mlx/backend/metal/kernels/copy.h:246
Function
copy_gg_dynamic_nd1
mlx/backend/metal/kernels/copy.h:204
Function
copy_gg_dynamic_nd2
mlx/backend/metal/kernels/copy.h:218
Function
copy_gg_dynamic_nd3
mlx/backend/metal/kernels/copy.h:232
Function
copy_gg_nd1
mlx/backend/metal/kernels/copy.h:139
Function
copy_gg_nd2
mlx/backend/metal/kernels/copy.h:151
Function
copy_gg_nd3
mlx/backend/metal/kernels/copy.h:163
Function
copy_s
mlx/backend/metal/kernels/copy.h:4
Function
copy_s2
mlx/backend/metal/kernels/copy.h:40
Function
copy_single
mlx/backend/cpu/copy.cpp:16
Function
copy_v
mlx/backend/metal/kernels/copy.h:22
Function
copy_v2
mlx/backend/metal/kernels/copy.h:59
Function
copy_vector
mlx/backend/cpu/copy.cpp:25
Method
cosine
(a, b)
python/tests/test_nn.py:1986
Function
cosine_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
Function
cosine_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
Method
cost_guess
mlx/3rdparty/pocketfft.h:383
Method
count_prims
(outputs)
python/tests/test_compile.py:178
Method
create
mlx/distributed/nccl/nccl.cpp:280
Method
createTests
(self, *args, **kwargs)
python/tests/mlx_tests.py:26
Method
create_threads
mlx/3rdparty/pocketfft.h:684
Function
cross_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
Function
cross_entropy
(targets, x)
benchmarks/python/comparative/bench_mlx.py:255
Function
cross_entropy
(targets, x)
benchmarks/python/comparative/bench_torch.py:212
Function
cuda_kernel
mlx/backend/cuda/custom_kernel.cpp:144
Function
cumsum
(axis, x)
benchmarks/python/comparative/bench_mlx.py:314
Function
cumsum
(axis, x)
benchmarks/python/comparative/bench_torch.py:273
Function
data
mlx/array.h:373
Method
data
mlx/small_vector.h:233
Function
data_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
Method
dealloc
mlx/3rdparty/pocketfft.h:194
Function
decorator
(klass)
python/mlx/nn/layers/activations.py:12
Function
delayload_helper
Called by system when failed to locate a lazy-loaded DLL.
mlx/backend/cuda/delayload.cpp:61
Function
dequantize
mlx/backend/metal/kernels/quantized.h:484
Function
dequantize
mlx/backend/metal/kernels/fp_quantized.h:139
Function
dequantize
mlx/backend/metal/kernels/fp_quantized_nax.h:63
Function
dequantize
mlx/backend/metal/kernels/quantized_nax.h:487
Function
dequantize_scale
mlx/backend/cpu/quantized.cpp:50
Function
dequantize_scale
mlx/backend/metal/kernels/fp_quantized.h:31
Function
dequantize_scale
mlx/backend/metal/kernels/fp_quantized_nax.h:31
Function
deserialize
mlx/export.cpp:146
Function
deserialize_primitive
mlx/export.cpp:305
Function
deserialize_tuple
mlx/export.cpp:222
Function
deserialize_variant
mlx/export.cpp:206
Function
device_count
mlx/backend/no_cpu/device_info.cpp:11
Function
device_count
mlx/backend/cpu/device_info.cpp:100
Function
device_count
mlx/backend/no_gpu/device_info.cpp:11
Function
device_count
mlx/backend/metal/device_info.cpp:15
Function
dispatch
mlx/backend/cpu/encoder.h:44
Function
dispatch_float_types
mlx/dtype_utils.h:69
Function
dispatch_gather
mlx/backend/cpu/indexing.cpp:150
Function
dispatch_gather_axis
mlx/backend/cpu/indexing.cpp:304
Function
dispatch_inexact_types
mlx/dtype_utils.h:80
Function
dispatch_int_float_types
mlx/dtype_utils.h:93
Function
dispatch_quantize
mlx/backend/cpu/quantized.cpp:1214
Function
dispatch_real_types
mlx/dtype_utils.h:106
Function
dispatch_scatter
mlx/backend/cpu/indexing.cpp:472
Function
dispatch_scatter_axis
mlx/backend/cpu/indexing.cpp:645
Function
dispatch_scatter_axis_op
mlx/backend/cpu/indexing.cpp:628
Function
dispatch_scatter_inds
mlx/backend/cpu/indexing.cpp:446
Function
do_attention_bench
(f, q, k, v, scale, mask=None, transpose=False)
benchmarks/python/sdpa_bench.py:119
Function
dtype_from_str
(x)
benchmarks/python/comparative/bench_mlx.py:27
Function
dtype_from_str
(x)
benchmarks/python/comparative/bench_torch.py:26
Method
dtypes
(self)
python/tests/test_blas.py:14
Method
dummy_loss
(model, x, y)
python/tests/mlx_distributed_tests.py:171
Function
eig_impl
mlx/backend/cpu/eig.cpp:177
Function
eigh_impl
mlx/backend/cpu/eigh.cpp:143
Function
einsum_attention
(x)
benchmarks/python/einsum_bench.py:64
Function
elem_to_loc_1
mlx/backend/metal/kernels/utils.h:131
Function
elem_to_loc_2
mlx/backend/metal/kernels/utils.h:136
Function
elem_to_loc_3
mlx/backend/metal/kernels/utils.h:141
Method
elems
mlx/backend/metal/kernels/steel/gemm/nax.h:631
Function
elu
(x)
benchmarks/python/comparative/bench_mlx.py:213
← previous
next →
2,601–2,700 of 4,946, ranked by callers