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
Methodmma
mlx/backend/metal/kernels/steel/gemm/nax.h:393
Methodmma
mlx/backend/metal/kernels/steel/gemm/mma.h:181
Methodmma
Karatsuba MMA: 3 real MMAs per K-chunk */
mlx/backend/metal/kernels/steel/gemm/mma.h:840
Methodmma
mlx/backend/metal/kernels/steel/attn/nax.h:393
Methodmma
mlx/backend/metal/kernels/steel/attn/mma.h:171
Methodmodules
Return a list with all the modules in this instance. Returns: A list of :class:`mlx.nn.Module` instances.
python/mlx/nn/layers/base.py:435
Functionmse_loss
Computes the mean squared error loss. Args: predictions (array): The predicted values. targets (array): The target values.
python/mlx/nn/losses.py:211
Functionmtl_device
mlx/backend/metal/device.h:142
Functionmulti_one
tests/compile_tests.cpp:184
Functionmulti_three
tests/compile_tests.cpp:200
Functionmulti_two
tests/compile_tests.cpp:194
Functionmx_conv_1D
(a, b)
benchmarks/python/conv1d_bench.py:32
Functionmx_conv_2D
(a, b)
benchmarks/python/conv_bench.py:32
Functionmx_conv_2D
(a, b)
benchmarks/python/conv_unaligned_bench.py:26
Functionmx_conv_2D
(a, b)
benchmarks/python/conv2d_bench_cpu.py:27
Functionmx_conv_3D
(a, b, b_prime)
benchmarks/python/conv3d_bench.py:26
Functionmx_conv_3D
(a, b)
benchmarks/python/conv3d_bench_cpu.py:27
Functionmx_conv_3D
(a, b)
benchmarks/python/conv3d_transpose_bench_cpu.py:27
Functionmx_conv_transpose_2D
(a, b)
benchmarks/python/conv2d_transpose_bench_cpu.py:26
Functionmx_conv_transpose_2D
(a, b)
benchmarks/python/conv_transpose_bench.py:29
Methodmy_double
(params)
python/tests/test_autograd.py:649
Methodmy_exp_jvp
(x, dx)
python/tests/test_autograd.py:632
Methodmy_exp_vjp
(x, dx, ex)
python/tests/test_autograd.py:619
Methodmy_exp_vmap
(x, axis)
python/tests/test_autograd.py:636
Methodmy_fun
(tree)
python/tests/test_vmap.py:112
Methodname
mlx/primitives.h:462
Methodname
mlx/primitives.h:980
Methodname
mlx/primitives.h:1334
Methodname
mlx/primitives.h:1791
Methodname
mlx/primitives.h:1855
Methodname
mlx/primitives.h:1899
Methodname
mlx/primitives.h:1938
Methodname
mlx/primitives.h:2080
Methodname
mlx/primitives.h:2253
Methodname
mlx/fast_primitives.h:309
Methodname
mlx/primitives.cpp:6000
Methodname
mlx/distributed/primitives.h:141
Methodname
The name of primitive. */
examples/extensions/axpby/axpby.h:78
Methodnamed_modules
Return a list with all the modules in this instance and their name with dot notation. Returns: A list of tuples (str, :cl
python/mlx/nn/layers/base.py:445
Functionnan_aware_less
mlx/backend/cpu/sort.cpp:24
Functionnd_array_to_mlx_contiguous
python/src/convert.cpp:26
Functionnegative
(a)
benchmarks/python/single_ops.py:72
Methodnew_all_sum
(x, **kwargs)
python/tests/mlx_distributed_tests.py:16
Functionnew_stream
mlx/backend/no_gpu/eval.cpp:12
Functionnew_stream
mlx/backend/metal/eval.cpp:14
Functionnew_stream
mlx/backend/cuda/eval.cpp:22
Functionnext
Iteration helper */
mlx/backend/metal/kernels/steel/attn/loader.h:131
Functionnext
Iteration helper */
mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h:190
Functionnext
Iteration helper */
mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_l.h:134
Methodnext
python/src/array.cpp:82
Methodnext
mlx/backend/metal/kernels/quantized_nax.h:814
Methodnext
mlx/backend/metal/kernels/utils.h:259
Methodnext
mlx/backend/metal/kernels/utils.h:288
Methodnext
Iteration helper */
mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_n.h:313
Methodnext
Iteration helper */
mlx/backend/metal/kernels/steel/conv/loaders/loader_general.h:187
Methodnext
Iteration helper */
mlx/backend/metal/kernels/steel/conv/loaders/loader_general.h:361
Methodnext
Iteration helper */
mlx/backend/metal/kernels/steel/conv/loaders/loader_channel_l.h:941
Functionnll_loss
Computes the negative log likelihood loss. Args: inputs (array): The predicted distribution in log space. targets (array): T
python/mlx/nn/losses.py:236
Functionnone_or_list
(x)
benchmarks/python/comparative/bench_mlx.py:20
Functionnone_or_list
(x)
benchmarks/python/comparative/bench_torch.py:19
Functionnorm
mlx/linalg.cpp:165
Methodnp_fn
(*args)
python/tests/test_bf16.py:67
Methodnp_gather_mm
(a, b, lhs_indices=None, rhs_indices=None)
python/tests/test_blas.py:1049
Methodnp_softmax
(x, axis)
python/tests/test_ops.py:1807
Methodnp_transform
(x)
python/tests/test_bf16.py:58
Functionoffset
mlx/array.h:383
Methodones_fun
(x)
python/tests/test_compile.py:492
Functionoperator!
mlx/backend/cpu/simd/accelerate_simd.h:153
Functionoperator!
mlx/backend/cpu/simd/neon_fp16_simd.h:81
Functionoperator!=
mlx/array.h:180
Functionoperator!=
mlx/ops.h:419
Functionoperator!=
mlx/backend/cpu/simd/neon_fp16_simd.h:121
Methodoperator!=
mlx/small_vector.h:229
Methodoperator!=
mlx/backend/cpu/sort.cpp:65
Functionoperator%
mlx/ops.h:943
Functionoperator%
mlx/ops.cpp:2925
Functionoperator%
mlx/types/complex.h:59
Functionoperator%
mlx/backend/metal/kernels/complex.h:163
Functionoperator&
mlx/ops.cpp:6228
Functionoperator&&
mlx/ops.cpp:2813
Functionoperator&&
mlx/backend/cpu/simd/neon_fp16_simd.h:145
Functionoperator&&
mlx/backend/metal/kernels/steel/utils/integral_constant.h:83
Functionoperator()
mlx/export_impl.h:13
Functionoperator()
mlx/backend/cpu/unary_ops.h:63
Methodoperator()
python/src/transforms.cpp:576
Methodoperator()
python/src/transforms.cpp:618
Methodoperator()
python/src/transforms.cpp:642
Methodoperator()
python/src/transforms.cpp:711
Methodoperator()
python/src/transforms.cpp:745
Methodoperator()
python/src/transforms.cpp:785
Methodoperator()
python/src/transforms.cpp:840
Methodoperator()
python/src/export.cpp:87
Methodoperator()
python/src/fast.cpp:26
Methodoperator()
mlx/compile.cpp:557
Methodoperator()
mlx/backend/cpu/indexing.cpp:30
Methodoperator()
mlx/backend/cpu/indexing.cpp:36
Methodoperator()
mlx/backend/cpu/indexing.cpp:43
Methodoperator()
mlx/backend/cpu/indexing.cpp:50
Methodoperator()
mlx/backend/cpu/indexing.cpp:57
Methodoperator()
mlx/backend/cpu/unary_ops.h:105
← previousnext →3,301–3,400 of 4,946, ranked by callers