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
Functionstep_decay
r"""Make a step decay scheduler. Args: init (float): Initial value. decay_rate (float): Multiplicative factor to decay by.
python/mlx/optimizers/schedulers.py:34
Functionstep_function
(x)
benchmarks/python/comparative/bench_mlx.py:336
Functionstep_function
(x)
benchmarks/python/comparative/bench_torch.py:298
Methodstop_and_wait
mlx/threadpool.h:97
Functionstop_capture
mlx/backend/metal/no_metal.cpp:17
Methodstore
mlx/backend/metal/kernels/steel/gemm/nax.h:192
Methodstore
mlx/backend/metal/kernels/steel/gemm/mma.h:101
Methodstore
mlx/backend/metal/kernels/steel/gemm/mma.h:289
Methodstore
mlx/backend/metal/kernels/steel/attn/nax.h:192
Methodstore
mlx/backend/metal/kernels/steel/attn/mma.h:126
Methodstore
mlx/backend/metal/kernels/steel/attn/mma.h:336
Methodstore_result
Store results from simdgroup_matrix results into device memory */
mlx/backend/metal/kernels/steel/gemm/mma.h:900
Methodstore_result_safe
mlx/backend/metal/kernels/steel/gemm/mma.h:950
Methodstore_result_slice
mlx/backend/metal/kernels/steel/gemm/mma.h:919
Methodstore_rows
mlx/backend/metal/kernels/steel/gemm/nax.h:232
Methodstore_rows
mlx/backend/metal/kernels/steel/gemm/nax.h:761
Methodstore_rows
mlx/backend/metal/kernels/steel/attn/nax.h:232
Methodstore_safe
mlx/backend/metal/kernels/steel/gemm/nax.h:277
Methodstore_safe
mlx/backend/metal/kernels/steel/gemm/mma.h:122
Methodstore_safe
mlx/backend/metal/kernels/steel/gemm/mma.h:356
Methodstore_safe
mlx/backend/metal/kernels/steel/attn/nax.h:277
Methodstore_safe
mlx/backend/metal/kernels/steel/attn/nax.h:778
Methodstore_safe
mlx/backend/metal/kernels/steel/attn/mma.h:147
Methodstore_safe
mlx/backend/metal/kernels/steel/attn/mma.h:403
Methodstore_slice
mlx/backend/metal/kernels/steel/gemm/nax.h:319
Methodstore_slice
mlx/backend/metal/kernels/steel/gemm/mma.h:155
Methodstore_slice
mlx/backend/metal/kernels/steel/gemm/mma.h:376
Methodstore_slice
mlx/backend/metal/kernels/steel/attn/nax.h:319
Methodstore_slice
mlx/backend/metal/kernels/steel/attn/nax.h:796
Functionstrided_scan
mlx/backend/metal/kernels/scan.h:392
Functionstrides
* Get the stride of the corresponding dimension. * * This function supports negative indexing and provides * bounds checking. */
mlx/array.h:138
Functionsum
mlx/backend/metal/kernels/steel/utils/integral_constant.h:118
Functionsum_and_add
(axis, x, y)
benchmarks/python/comparative/bench_mlx.py:147
Functionsum_and_add
(axis, x, y)
benchmarks/python/comparative/bench_torch.py:106
Functionsum_scatter
mlx/distributed/distributed.cpp:45
Methodsum_scatter
mlx/distributed/ring/ring.cpp:578
Methodsum_scatter
mlx/distributed/jaccl/jaccl.cpp:147
Methodsum_scatter
mlx/distributed/nccl/nccl.cpp:382
Methodsupports_bool_mask
mlx/backend/no_gpu/primitives.cpp:39
Methodsupports_bool_mask
mlx/backend/metal/scaled_dot_product_attention.cpp:639
Methodsupports_bool_mask
mlx/backend/cuda/scaled_dot_product_attention.cpp:570
Functionsvd_impl
mlx/backend/cpu/svd.cpp:189
Functionswap_endianness
mlx/backend/common/load.cpp:12
Functionswap_endianness
mlx/backend/cuda/load.cpp:13
Methodswizzle
mlx/backend/metal/kernels/steel/gemm/transforms.h:62
Methodswizzle
mlx/backend/metal/kernels/steel/attn/transforms.h:61
Functionsynchronize
mlx/backend/no_gpu/eval.cpp:25
Methodt
()
python/tests/test_array.py:2229
MethodtearDown
(self)
python/tests/mlx_tests.py:75
MethodtearDownClass
(cls)
python/tests/test_load.py:37
MethodtearDownClass
(cls)
python/tests/test_export_import.py:23
Methodtell
python/src/load.cpp:133
Methodtell
python/src/load.cpp:363
Methodtell
mlx/io/load.h:77
Methodterminate
Terminate or return the exit code.
python/mlx/_distributed_utils/launch.py:42
Functionternary_g
mlx/backend/metal/kernels/ternary.h:115
Functionternary_g_nd1
mlx/backend/metal/kernels/ternary.h:63
Functionternary_g_nd2
mlx/backend/metal/kernels/ternary.h:79
Functionternary_g_nd3
mlx/backend/metal/kernels/ternary.h:97
Functionternary_op
mlx/backend/cpu/ternary.h:124
Functionternary_op_dims
mlx/backend/cpu/ternary.h:12
Functionternary_op_dispatch_dims
mlx/backend/cpu/ternary.h:55
Functionternary_v
mlx/backend/metal/kernels/ternary.h:9
Functionternary_v2
mlx/backend/metal/kernels/ternary.h:38
Methodtest_abs
(self)
python/tests/test_ops.py:839
Methodtest_active_memory_count
(self)
python/tests/test_memory.py:61
Methodtest_adadelta
(self)
python/tests/test_optimizers.py:148
Methodtest_adafactor
(self)
python/tests/test_optimizers.py:269
Methodtest_adagrad
(self)
python/tests/test_optimizers.py:130
Methodtest_adam
(self)
python/tests/test_optimizers.py:173
Methodtest_adamw_matches_pytorch
(self)
python/tests/test_optimizers.py:207
Methodtest_add
(self)
python/tests/test_ops.py:181
Methodtest_add_numpy
(self)
python/tests/test_array.py:1994
Methodtest_addmm
(self)
python/tests/test_blas.py:602
Methodtest_addmm_grad
(self)
python/tests/test_blas.py:733
Methodtest_alibi
(self)
python/tests/test_nn.py:1134
Methodtest_all
(self)
python/tests/test_ops.py:1110
Methodtest_all_gather
(self)
python/tests/mlx_distributed_tests.py:309
Methodtest_all_gather_extra
(self)
python/tests/mpi_test_distributed.py:73
Methodtest_all_gather_extra
(self)
python/tests/ring_test_distributed.py:67
Methodtest_all_gather_split
(self)
python/tests/nccl_test_distributed.py:109
Methodtest_all_gather_vjp
(self)
python/tests/ring_test_distributed.py:119
Methodtest_all_reduce
(self)
python/tests/mlx_distributed_tests.py:53
Methodtest_all_reduce_extra
(self)
python/tests/mpi_test_distributed.py:31
Methodtest_all_reduce_extra
(self)
python/tests/ring_test_distributed.py:27
Methodtest_all_reduce_split
(self)
python/tests/nccl_test_distributed.py:68
Methodtest_allclose
(self)
python/tests/test_ops.py:1083
Methodtest_any
(self)
python/tests/test_ops.py:1121
Methodtest_api
(self)
python/tests/test_array.py:1620
Methodtest_arange_corner_cases_cast
(self)
python/tests/test_ops.py:1418
Methodtest_arange_inferred_dtype
(self)
python/tests/test_ops.py:1390
Methodtest_arange_overload_dispatch
(self)
python/tests/test_ops.py:1350
Methodtest_arg_reduce
(self)
python/tests/test_reduce.py:86
Functiontest_arg_reduce_small_bool
tests/arg_reduce_tests.cpp:125
Methodtest_arg_reduction_ops
(self)
python/tests/test_bf16.py:144
Methodtest_argmin_argmax
(self)
python/tests/test_ops.py:696
Methodtest_argpartition
(self)
python/tests/test_ops.py:2349
Methodtest_argsort_nan
(self)
python/tests/test_ops.py:3340
Methodtest_array_at
(self)
python/tests/test_array.py:1394
Methodtest_array_at_slice_update_extensive
(self)
python/tests/test_array.py:1472
← previousnext →4,001–4,100 of 4,946, ranked by callers