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

Functionlog_softmax

r"""Applies the Log Softmax function. Applies :math:`x + \log \sum_i e^{x_i}` element wise.

python/mlx/nn/layers/activations.py:66

Functionlogsumexp

mlx/ops.cpp:2708

Functionlogsumexp

mlx/backend/cpu/logsumexp.cpp:19

Functionlogsumexp

benchmarks/python/comparative/bench_mlx.py:265

Functionlogsumexp

benchmarks/python/comparative/bench_torch.py:220

Functionlogsumexp_looped

mlx/backend/metal/kernels/logsumexp.h:76

python/tests/test_optimizers.py:343

python/tests/test_compile.py:199

(kernel, inputs, strides, groups)

python/tests/test_conv.py:1134

python/tests/test_optimizers.py:215

python/tests/test_autograd.py:891

python/tests/test_nn.py:272

python/tests/test_compile.py:48

Methodloss_fn_tuple

python/tests/test_autograd.py:927

Methodloss_fn_vector

python/tests/test_autograd.py:931

mlx/types/limits.h:30

mlx/types/limits.h:56

Functionlu_factor

mlx/linalg.cpp:629

Functionluf_impl

mlx/backend/cpu/luf.cpp:14

python/mlx/_distributed_utils/config.py:568

python/mlx/_distributed_utils/launch.py:457

mlx/distributed/jaccl/lib/examples/minimal_env.cpp:4

mlx/distributed/jaccl/lib/examples/minimal_cfg.cpp:4

mlx/distributed/jaccl/lib/examples/allreduce_bench.cpp:91

tests/test_teardown.cpp:15

tests/tests.cpp:12

examples/cmake_project/example.cpp:9

examples/cpp/tutorial.cpp:96

examples/cpp/logistic_regression.cpp:15

examples/cpp/linear_regression.cpp:15

examples/cpp/metal_capture.cpp:10

examples/cpp/distributed.cpp:9

examples/export/train_mlp.cpp:8

examples/export/eval_mlp.cpp:8

benchmarks/cpp/compare_devices.cpp:25

benchmarks/cpp/single_ops.cpp:274

benchmarks/cpp/irregular_strides.cpp:188

benchmarks/cpp/autograd.cpp:36

Methodmake_addmm

python/tests/test_blas.py:737

Methodmake_and_update

python/tests/test_nn.py:342

Methodmake_arrays

mlx/array.cpp:42

Methodmake_jvp_function

python/src/transforms.cpp:965

Methodmake_mask

(tm_, tn_, batch, np_dtype)

python/tests/test_blas.py:917

Methodmake_model

python/tests/test_nn.py:89

Functionmake_op

mlx/backend/metal/indexing.cpp:42

Methodmake_ref_addmm

python/tests/test_blas.py:734

Methodmake_vjp_function

python/src/transforms.cpp:955

Methodmake_vmap_function

python/src/transforms.cpp:974

mlx/backend/no_gpu/allocator.cpp:79

mlx/backend/metal/allocator.cpp:96

Functionmargin_ranking_loss

r""" Calculate the margin ranking loss that loss given inputs :math:`x_1`, :math:`x_2` and a label :math:`y` (containing 1 or -1). The lo

python/mlx/nn/losses.py:558

Functionmasked_assign_impl

mlx/backend/metal/kernels/indexing/masked_scatter.h:8

Functionmasked_scatter_impl

mlx/backend/cpu/indexing.cpp:754

Functionmasked_scatter_mlx

(self_arr, mask_arr, src_arr)

benchmarks/python/masked_scatter.py:78

Functionmasked_scatter_torch

(self_tensor, mask_tensor, src_tensor)

benchmarks/python/masked_scatter.py:89

mlx/backend/metal/kernels/steel/gemm/mma.h:254

mlx/backend/metal/kernels/steel/attn/mma.h:277

mlx/export.cpp:599

benchmarks/python/comparative/bench_torch.py:63

Functionmatmul<bfloat16_t>

mlx/backend/cpu/gemms/bnns.cpp:180

Functionmatmul<bfloat16_t>

mlx/backend/cpu/gemms/simd_bf16.cpp:10

Functionmatmul<complex64_t>

mlx/backend/cpu/gemms/cblas.cpp:92

Functionmatmul<double>

mlx/backend/cpu/gemms/cblas.cpp:51

Functionmatmul<float16_t>

mlx/backend/cpu/gemms/bnns.cpp:145

Functionmatmul<float16_t>

mlx/backend/cpu/gemms/simd_fp16.cpp:10

Functionmatmul<float>

mlx/backend/cpu/gemms/cblas.cpp:10

Functionmatmul_dispatch

mlx/backend/cpu/matmul.cpp:15

Functionmatmul_square

benchmarks/python/comparative/bench_mlx.py:48

Functionmatmul_square

benchmarks/python/comparative/bench_torch.py:55

mlx/types/limits.h:59

Functionmax_scalars

tests/compile_tests.cpp:118

MethodmaybeInsertBarrier

mlx/backend/metal/device.cpp:345

Functionmb_block_merge

mlx/backend/metal/kernels/sort.h:614

Functionmb_block_partition

mlx/backend/metal/kernels/sort.h:561

Functionmb_block_sort

mlx/backend/metal/kernels/sort.h:517

mlx/ops.cpp:2122

Methodmerge_partition

mlx/backend/metal/kernels/sort.h:100

Methodmerge_partition

mlx/backend/metal/kernels/sort.h:484

Methodmerge_step

mlx/backend/metal/kernels/sort.h:126

Functionmetal_kernel

mlx/backend/metal/custom_kernel.cpp:175

Functionmid_slice_add

benchmarks/python/single_ops.py:34

r"""Applies the Mish function, element-wise. Mish: A Self Regularized Non-Monotonic Neural Activation Function. Reference: https://arxiv.org

python/mlx/nn/layers/activations.py:273

benchmarks/python/comparative/bench_mlx.py:199

(x: torch.Tensor)

benchmarks/python/comparative/bench_torch.py:196

Functionmlx_atomic_fetch_max_explicit<float>

mlx/backend/metal/kernels/atomic.h:132

Functionmlx_atomic_fetch_min_explicit<float>

mlx/backend/metal/kernels/atomic.h:117

Functionmlx_atomic_update_and_store

mlx/backend/metal/kernels/atomic.h:170

python/tests/test_bf16.py:63

Functionmlx_func

python/src/mlx_func.cpp:98

Functionmlx_fused_attn

(q, k, v, scale, mask)

benchmarks/python/sdpa_bench.py:104

Functionmlx_get_item

python/src/indexing.cpp:456

Functionmlx_load_helper

python/src/load.cpp:275

Functionmlx_ref_attn

(q, k, v, scale=1.0, mask=None)

benchmarks/python/sdpa_bench.py:58

Functionmlx_save_gguf_helper

python/src/load.cpp:513

Functionmlx_save_helper

python/src/load.cpp:404

Functionmlx_save_safetensor_helper

python/src/load.cpp:478

Functionmlx_set_item

python/src/indexing.cpp:951

Functionmlx_to_nd_array

python/src/convert.cpp:118

Functionmlx_to_nd_array_impl

python/src/convert.cpp:100

python/tests/test_quantized.py:1113

← previous next →3,201–3,300 of 4,946, ranked by callers