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
Function
scatter_add
mlx/ops.cpp:3673
Method
scatter_add
(a)
python/tests/test_vmap.py:459
Function
scatter_axis
mlx/backend/cpu/indexing.cpp:586
Method
scatter_fn
(x, m, src)
python/tests/test_vmap.py:816
Function
scatter_impl
mlx/backend/metal/kernels/indexing/scatter.h:15
Function
scatter_max
mlx/ops.cpp:3691
Function
scatter_min
mlx/ops.cpp:3700
Function
scatter_prod
mlx/ops.cpp:3682
Method
scatter_unsort
(x, inv_order, shape=None)
python/tests/test_quantized.py:965
Function
sdpa_mask
(*args)
benchmarks/python/sdpa_vector_bench.py:82
Function
sdpa_vector_2pass_1
mlx/backend/metal/kernels/sdpa_vector.h:180
Function
sdpa_vector_2pass_2
mlx/backend/metal/kernels/sdpa_vector.h:321
Function
search
(x)
benchmarks/python/comparative/compare.py:41
Method
seek
python/src/load.cpp:372
Method
seek
Warning: do not use this function from multiple threads as it advances the file descriptor
mlx/io/load.h:83
Function
segmented_mm
mlx/backend/cpu/masked_mm.cpp:57
Function
segmented_mm
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.h:19
Function
segmented_mm_nax
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented_nax.h:19
Method
segmented_mm_ref
(a, b, s)
python/tests/test_blas.py:1295
Function
selu
r"""Applies the Scaled Exponential Linear Unit. .. math:: \text{selu}(x) = \begin{cases} \lambda x & \text{if } x > 0 \\
python/mlx/nn/layers/activations.py:244
Function
selu
(x)
benchmarks/python/comparative/bench_mlx.py:343
Function
selu
(x)
benchmarks/python/comparative/bench_torch.py:306
Function
send
mlx/distributed/distributed.cpp:37
Method
send
mlx/distributed/ring/ring.cpp:144
Method
send
mlx/distributed/ring/ring.cpp:286
Method
send
mlx/distributed/mpi/mpi.cpp:449
Method
send
mlx/distributed/jaccl/jaccl.cpp:129
Method
send
mlx/distributed/jaccl/lib/jaccl/ring.cpp:169
Method
send
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:405
Method
send
mlx/distributed/jaccl/lib/jaccl/mesh.cpp:179
Method
send
mlx/distributed/jaccl/lib/jaccl/mesh_impl.h:275
Method
send
mlx/distributed/nccl/nccl.cpp:360
Method
send_impl
mlx/distributed/ring/ring.cpp:164
Method
send_to
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:540
Method
send_to
mlx/distributed/jaccl/lib/jaccl/mesh_impl.h:373
Function
serialize_primitive
mlx/export.cpp:270
Method
setUp
(self)
python/tests/mlx_tests.py:68
Method
setUpClass
(cls)
python/tests/test_load.py:30
Method
setUpClass
(cls)
python/tests/mpi_test_distributed.py:10
Method
setUpClass
(cls)
python/tests/nccl_test_distributed.py:12
Method
setUpClass
(cls)
python/tests/ring_test_distributed.py:10
Method
setUpClass
(cls)
python/tests/test_export_import.py:16
Function
set_bytes
mlx/backend/metal/device.h:74
Function
set_cache_limit
mlx/backend/no_gpu/allocator.cpp:126
Function
set_cache_limit
mlx/backend/metal/allocator.cpp:234
Function
set_compute_pipeline_state
mlx/backend/metal/device.h:59
Function
set_finfo_limits
mlx/utils.cpp:308
Function
set_iinfo_limits
mlx/utils.cpp:335
Function
set_memory_limit
mlx/backend/no_gpu/allocator.cpp:115
Function
set_memory_limit
mlx/backend/metal/allocator.cpp:237
Method
set_out
mlx/backend/cuda/gemms/cublas_gemm.cpp:121
Function
set_output_array
mlx/backend/cpu/encoder.h:24
Method
set_scales_ptrs
mlx/backend/cuda/quantized/cublas_qqmm.cpp:193
Function
set_siblings
mlx/array.h:313
Method
set_tensor_attrs
mlx/backend/cuda/cudnn_utils.cpp:157
Method
set_tensor_attrs_nchw
mlx/backend/cuda/cudnn_utils.cpp:182
Function
set_threadgroup_memory_length
mlx/backend/metal/device.h:83
Function
set_tracer
Mark the array as a tracer array (true) or not.
mlx/array.h:433
Function
set_vector_bytes
mlx/backend/metal/device.h:64
Function
set_wired_limit
mlx/backend/no_gpu/allocator.cpp:129
Function
set_wired_limit
mlx/backend/metal/allocator.cpp:243
Method
sharding
(path, weight)
python/tests/mlx_distributed_tests.py:281
Method
shutdown_locked
mlx/3rdparty/pocketfft.h:708
Function
sibling_position
The array's position in the sibling list. */
mlx/array.h:309
Method
signal
mlx/backend/no_gpu/event.cpp:35
Function
silu
r"""Applies the Sigmoid Linear Unit. Also known as Swish. Applies :math:`x \sigma(x)` element wise, where :math:`\sigma(\cdot)` is the logist
python/mlx/nn/layers/activations.py:135
Method
simd_exclusive_scan
mlx/backend/metal/kernels/scan.h:92
Method
simd_exclusive_scan
mlx/backend/metal/kernels/scan.h:138
Method
simd_exclusive_scan
mlx/backend/metal/kernels/scan.h:161
Method
simd_exclusive_scan_impl
mlx/backend/metal/kernels/scan.h:49
Method
simd_exclusive_scan_impl
mlx/backend/metal/kernels/scan.h:70
Function
simd_gemm
mlx/backend/cpu/gemms/simd_gemm.h:39
Method
simd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:40
Method
simd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:81
Method
simd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:123
Method
simd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:145
Method
simd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:167
Method
simd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:224
Method
simd_scan
mlx/backend/metal/kernels/scan.h:84
Method
simd_scan
mlx/backend/metal/kernels/scan.h:130
Method
simd_scan
mlx/backend/metal/kernels/scan.h:153
Method
simd_scan_impl
mlx/backend/metal/kernels/scan.h:45
Method
simd_scan_impl
mlx/backend/metal/kernels/scan.h:66
Function
simd_shuffle_up
mlx/backend/metal/kernels/utils.h:376
Method
simple_binary
(x, y)
python/tests/test_compile.py:220
Function
simple_max
mlx/distributed/mpi/mpi.cpp:61
Function
simple_min
mlx/distributed/mpi/mpi.cpp:81
Function
simple_sum
mlx/distributed/mpi/mpi.cpp:42
Method
simple_unary
(x)
python/tests/test_compile.py:211
Method
simple_unary_inner
(z)
python/tests/test_compile.py:243
Method
simple_unary_outer
(x)
python/tests/test_compile.py:239
Function
sincos
mlx/backend/cpu/simd/math.h:63
Method
sincos_2pibyn
mlx/3rdparty/pocketfft.h:337
Function
sine
(x)
benchmarks/python/distributed_bench.py:42
Method
size
mlx/backend/no_gpu/allocator.cpp:97
Method
size
mlx/distributed/distributed.cpp:63
Method
size
mlx/distributed/jaccl/jaccl.cpp:74
Method
size
mlx/distributed/jaccl/lib/jaccl/mesh.h:32
Function
slice_add
(a, b)
benchmarks/python/single_ops.py:26
Function
slice_update
(arguments)
benchmarks/python/slice_update_bench.py:11
← previous
next →
3,801–3,900 of 4,946, ranked by callers