Functions — github.com/ml-explore/mlx

CodeHub 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

benchmarks/python/comparative/bench_torch.py:148

mlx/small_vector.h:279

mlx/backend/metal/reduce.cpp:154

mlx/scheduler.cpp:39

Functionensure_row_contiguous_matrix

mlx/backend/cuda/quantized/quantized_utils.h:20

mlx/types/limits.h:36

mlx/types/limits.h:62

Functionequivalent_matmul

benchmarks/python/gather_qmm_bench.py:75

Functionequivalent_matmul

benchmarks/python/gather_mm_bench.py:65

mlx/small_vector.h:389

mlx/backend/cpu/simd/math.h:137

mlx/backend/cpu/simd/math.h:151

mlx/backend/no_gpu/eval.cpp:17

mlx/backend/metal/eval.cpp:30

mlx/backend/cuda/eval.cpp:35

mlx/transforms.cpp:55

mlx/fast_primitives.h:49

mlx/fast_primitives.h:82

mlx/fast_primitives.h:109

mlx/fast_primitives.h:141

mlx/fast_primitives.h:177

mlx/fast_primitives.h:233

mlx/fast_primitives.h:278

mlx/fast_primitives.h:391

mlx/backend/no_cpu/compiled.cpp:17

mlx/backend/cpu/logsumexp.cpp:80

mlx/backend/cpu/scan.cpp:245

mlx/backend/cpu/svd.cpp:270

mlx/backend/cpu/binary.cpp:17

mlx/backend/cpu/indexing.cpp:202

mlx/backend/cpu/sort.cpp:320

mlx/backend/cpu/fft.cpp:12

mlx/backend/cpu/eigh.cpp:187

mlx/backend/cpu/quantized.cpp:924

mlx/backend/cpu/hadamard.cpp:91

mlx/backend/cpu/cholesky.cpp:71

mlx/backend/cpu/unary.cpp:14

mlx/backend/cpu/compiled.cpp:288

mlx/backend/cpu/distributed.cpp:20

mlx/backend/cpu/softmax.cpp:117

mlx/backend/cpu/arg_reduce.cpp:66

mlx/backend/cpu/select.cpp:87

mlx/backend/cpu/qrf.cpp:131

mlx/backend/cpu/inverse.cpp:146

mlx/backend/cpu/primitives.cpp:79

mlx/backend/cpu/masked_mm.cpp:110

mlx/backend/cpu/reduce.cpp:449

mlx/backend/cpu/matmul.cpp:121

mlx/backend/cpu/luf.cpp:103

mlx/backend/cpu/conv.cpp:1293

mlx/backend/common/load.cpp:30

examples/extensions/axpby/axpby.cpp:124

mlx/transforms.cpp:56

mlx/backend/gpu/primitives.cpp:23

mlx/backend/metal/rope.cpp:14

mlx/backend/metal/logsumexp.cpp:14

mlx/backend/metal/scan.cpp:122

mlx/backend/metal/ternary.cpp:156

mlx/backend/metal/binary.cpp:237

mlx/backend/metal/scaled_dot_product_attention.cpp:643

mlx/backend/metal/indexing.cpp:57

mlx/backend/metal/sort.cpp:318

mlx/backend/metal/quantized.cpp:1387

mlx/backend/metal/hadamard.cpp:164

mlx/backend/metal/unary.cpp:146

mlx/backend/metal/compiled.cpp:257

mlx/backend/metal/distributed.cpp:17

mlx/backend/metal/softmax.cpp:15

mlx/backend/metal/custom_kernel.cpp:327

mlx/backend/metal/primitives.cpp:27

mlx/backend/metal/reduce.cpp:951

mlx/backend/metal/matmul.cpp:1225

mlx/backend/metal/normalization.cpp:17

mlx/backend/metal/conv.cpp:1283

mlx/backend/cuda/scaled_dot_product_attention.cpp:574

mlx/backend/cuda/indexing.cpp:62

mlx/backend/cuda/compiled.cpp:227

mlx/backend/cuda/custom_kernel.cpp:277

mlx/backend/cuda/matmul.cpp:143

mlx/backend/cuda/conv.cpp:259

mlx/backend/cuda/quantized/quantized.cpp:15

mlx/backend/cuda/quantized/qqmm.cpp:72

Evaluate primitive on GPU */

examples/extensions/axpby/axpby.cpp:153

mlx/3rdparty/pocketfft.h:2227

mlx/3rdparty/pocketfft.h:2490

mlx/3rdparty/pocketfft.h:2527

mlx/3rdparty/pocketfft.h:2547

mlx/3rdparty/pocketfft.h:2578

mlx/backend/cpu/jit_compiler.cpp:118

mlx/3rdparty/pocketfft.h:2433

mlx/backend/cuda/quantized/cublas_qqmm.cpp:209

mlx/backend/cuda/gemms/cublas_gemm.cpp:220

Methodexecute_matmul

mlx/backend/cuda/cublas_utils.cpp:169

python/src/print.cpp:25

Methodexit_status

Return a tuple (returncode, killed) for the command. It should be (None, None) while the command is running normally.

python/mlx/_distributed_utils/launch.py:32

Methodexit_status

python/mlx/_distributed_utils/launch.py:73

Functionexp_two

tests/compile_tests.cpp:147

Methodexpand_mask

(mask, block_size, Y, X)

python/tests/test_blas.py:850

Functionexponential_decay

r"""Make an exponential decay scheduler. Args: init (float): Initial value. decay_rate (float): Multiplicative factor to decay by

python/mlx/optimizers/schedulers.py:9

Functionexport_to_dot

mlx/graph_utils.cpp:105

← previous next →2,701–2,800 of 4,946, ranked by callers