MCPcopy Create free account

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

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

Functionscatter_add
mlx/ops.cpp:3673
Methodscatter_add
(a)
python/tests/test_vmap.py:459
Functionscatter_axis
mlx/backend/cpu/indexing.cpp:586
Methodscatter_fn
(x, m, src)
python/tests/test_vmap.py:816
Functionscatter_impl
mlx/backend/metal/kernels/indexing/scatter.h:15
Functionscatter_max
mlx/ops.cpp:3691
Functionscatter_min
mlx/ops.cpp:3700
Functionscatter_prod
mlx/ops.cpp:3682
Methodscatter_unsort
(x, inv_order, shape=None)
python/tests/test_quantized.py:965
Functionsdpa_mask
(*args)
benchmarks/python/sdpa_vector_bench.py:82
Functionsdpa_vector_2pass_1
mlx/backend/metal/kernels/sdpa_vector.h:180
Functionsdpa_vector_2pass_2
mlx/backend/metal/kernels/sdpa_vector.h:321
Functionsearch
(x)
benchmarks/python/comparative/compare.py:41
Methodseek
python/src/load.cpp:372
Methodseek
Warning: do not use this function from multiple threads as it advances the file descriptor
mlx/io/load.h:83
Functionsegmented_mm
mlx/backend/cpu/masked_mm.cpp:57
Functionsegmented_mm
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented.h:19
Functionsegmented_mm_nax
mlx/backend/metal/kernels/steel/gemm/kernels/steel_gemm_segmented_nax.h:19
Methodsegmented_mm_ref
(a, b, s)
python/tests/test_blas.py:1295
Functionselu
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
Functionselu
(x)
benchmarks/python/comparative/bench_mlx.py:343
Functionselu
(x)
benchmarks/python/comparative/bench_torch.py:306
Functionsend
mlx/distributed/distributed.cpp:37
Methodsend
mlx/distributed/ring/ring.cpp:144
Methodsend
mlx/distributed/ring/ring.cpp:286
Methodsend
mlx/distributed/mpi/mpi.cpp:449
Methodsend
mlx/distributed/jaccl/jaccl.cpp:129
Methodsend
mlx/distributed/jaccl/lib/jaccl/ring.cpp:169
Methodsend
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:405
Methodsend
mlx/distributed/jaccl/lib/jaccl/mesh.cpp:179
Methodsend
mlx/distributed/jaccl/lib/jaccl/mesh_impl.h:275
Methodsend
mlx/distributed/nccl/nccl.cpp:360
Methodsend_impl
mlx/distributed/ring/ring.cpp:164
Methodsend_to
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:540
Methodsend_to
mlx/distributed/jaccl/lib/jaccl/mesh_impl.h:373
Functionserialize_primitive
mlx/export.cpp:270
MethodsetUp
(self)
python/tests/mlx_tests.py:68
MethodsetUpClass
(cls)
python/tests/test_load.py:30
MethodsetUpClass
(cls)
python/tests/mpi_test_distributed.py:10
MethodsetUpClass
(cls)
python/tests/nccl_test_distributed.py:12
MethodsetUpClass
(cls)
python/tests/ring_test_distributed.py:10
MethodsetUpClass
(cls)
python/tests/test_export_import.py:16
Functionset_bytes
mlx/backend/metal/device.h:74
Functionset_cache_limit
mlx/backend/no_gpu/allocator.cpp:126
Functionset_cache_limit
mlx/backend/metal/allocator.cpp:234
Functionset_compute_pipeline_state
mlx/backend/metal/device.h:59
Functionset_finfo_limits
mlx/utils.cpp:308
Functionset_iinfo_limits
mlx/utils.cpp:335
Functionset_memory_limit
mlx/backend/no_gpu/allocator.cpp:115
Functionset_memory_limit
mlx/backend/metal/allocator.cpp:237
Methodset_out
mlx/backend/cuda/gemms/cublas_gemm.cpp:121
Functionset_output_array
mlx/backend/cpu/encoder.h:24
Methodset_scales_ptrs
mlx/backend/cuda/quantized/cublas_qqmm.cpp:193
Functionset_siblings
mlx/array.h:313
Methodset_tensor_attrs
mlx/backend/cuda/cudnn_utils.cpp:157
Methodset_tensor_attrs_nchw
mlx/backend/cuda/cudnn_utils.cpp:182
Functionset_threadgroup_memory_length
mlx/backend/metal/device.h:83
Functionset_tracer
Mark the array as a tracer array (true) or not.
mlx/array.h:433
Functionset_vector_bytes
mlx/backend/metal/device.h:64
Functionset_wired_limit
mlx/backend/no_gpu/allocator.cpp:129
Functionset_wired_limit
mlx/backend/metal/allocator.cpp:243
Methodsharding
(path, weight)
python/tests/mlx_distributed_tests.py:281
Methodshutdown_locked
mlx/3rdparty/pocketfft.h:708
Functionsibling_position
The array's position in the sibling list. */
mlx/array.h:309
Methodsignal
mlx/backend/no_gpu/event.cpp:35
Functionsilu
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
Methodsimd_exclusive_scan
mlx/backend/metal/kernels/scan.h:92
Methodsimd_exclusive_scan
mlx/backend/metal/kernels/scan.h:138
Methodsimd_exclusive_scan
mlx/backend/metal/kernels/scan.h:161
Methodsimd_exclusive_scan_impl
mlx/backend/metal/kernels/scan.h:49
Methodsimd_exclusive_scan_impl
mlx/backend/metal/kernels/scan.h:70
Functionsimd_gemm
mlx/backend/cpu/gemms/simd_gemm.h:39
Methodsimd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:40
Methodsimd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:81
Methodsimd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:123
Methodsimd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:145
Methodsimd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:167
Methodsimd_reduce_impl
mlx/backend/metal/kernels/reduction/ops.h:224
Methodsimd_scan
mlx/backend/metal/kernels/scan.h:84
Methodsimd_scan
mlx/backend/metal/kernels/scan.h:130
Methodsimd_scan
mlx/backend/metal/kernels/scan.h:153
Methodsimd_scan_impl
mlx/backend/metal/kernels/scan.h:45
Methodsimd_scan_impl
mlx/backend/metal/kernels/scan.h:66
Functionsimd_shuffle_up
mlx/backend/metal/kernels/utils.h:376
Methodsimple_binary
(x, y)
python/tests/test_compile.py:220
Functionsimple_max
mlx/distributed/mpi/mpi.cpp:61
Functionsimple_min
mlx/distributed/mpi/mpi.cpp:81
Functionsimple_sum
mlx/distributed/mpi/mpi.cpp:42
Methodsimple_unary
(x)
python/tests/test_compile.py:211
Methodsimple_unary_inner
(z)
python/tests/test_compile.py:243
Methodsimple_unary_outer
(x)
python/tests/test_compile.py:239
Functionsincos
mlx/backend/cpu/simd/math.h:63
Methodsincos_2pibyn
mlx/3rdparty/pocketfft.h:337
Functionsine
(x)
benchmarks/python/distributed_bench.py:42
Methodsize
mlx/backend/no_gpu/allocator.cpp:97
Methodsize
mlx/distributed/distributed.cpp:63
Methodsize
mlx/distributed/jaccl/jaccl.cpp:74
Methodsize
mlx/distributed/jaccl/lib/jaccl/mesh.h:32
Functionslice_add
(a, b)
benchmarks/python/single_ops.py:26
Functionslice_update
(arguments)
benchmarks/python/slice_update_bench.py:11
← previousnext →3,801–3,900 of 4,946, ranked by callers