MCPcopy Create free account

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

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

Functionslice_update_impl
mlx/backend/cpu/indexing.cpp:858
Functionslice_update_op_impl
mlx/backend/metal/kernels/indexing/scatter.h:70
Functionslow_conv_1D
mlx/backend/cpu/conv.cpp:21
Functionslow_conv_2D
mlx/backend/cpu/conv.cpp:110
Functionslow_conv_3D
mlx/backend/cpu/conv.cpp:358
Functionsmooth_l1_loss
r""" Computes the smooth L1 loss. The smooth L1 loss is a variant of the L1 loss which replaces the absolute difference with a squared di
python/mlx/nn/losses.py:339
Functionsoftmax
mlx/ops.cpp:3810
Functionsoftmax
(axis, x)
benchmarks/python/comparative/bench_mlx.py:154
Functionsoftmax
(axis, x)
benchmarks/python/comparative/bench_torch.py:114
Functionsoftmax_fused
(axis, x)
benchmarks/python/comparative/bench_mlx.py:163
Functionsoftmax_fused
(axis, x)
benchmarks/python/comparative/bench_torch.py:124
Functionsoftmax_looped
mlx/backend/metal/kernels/softmax.h:101
Functionsoftmax_single_row
mlx/backend/metal/kernels/softmax.h:11
Functionsoftmin
r"""Applies the Softmin function. Applies :math:`\frac{e^{-x_i}}{\sum_j e^{-x_j}}` element-wise.
python/mlx/nn/layers/activations.py:322
Functionsoftplus
(x: mx.array)
benchmarks/python/comparative/bench_mlx.py:192
Functionsoftplus
(x)
benchmarks/python/comparative/bench_torch.py:172
Functionsoftsign
r"""Applies the Softsign function. Applies :math:`\frac{x}{1 + |x|}` element wise.
python/mlx/nn/layers/activations.py:102
Functionsort
(axis, x)
benchmarks/python/comparative/bench_mlx.py:321
Functionsort
(axis, x)
benchmarks/python/comparative/bench_torch.py:281
Methodsort
mlx/backend/metal/kernels/sort.h:158
Functionsparse
r"""An initializer that returns a sparse matrix. Args: sparsity (float): The fraction of elements in each column to be set to zer
python/mlx/nn/init.py:353
Functionspecial_mul
mlx/3rdparty/pocketfft.h:281
Methodspecial_mul
mlx/3rdparty/pocketfft.h:265
Methodsplit
mlx/distributed/distributed.cpp:67
Methodsplit
mlx/distributed/ring/ring.cpp:492
Methodsplit
mlx/distributed/mpi/mpi.cpp:377
Methodsplit
mlx/distributed/jaccl/jaccl.cpp:151
Methodsplit
mlx/distributed/nccl/nccl.cpp:287
Methodsplit
mlx/distributed/nccl/nccl.cpp:339
Functionstart_capture
mlx/backend/metal/no_metal.cpp:16
Functionstart_concurrent
mlx/backend/metal/device.h:87
Methodstart_threads
mlx/threadpool.h:113
Methodstate
The optimizer's state dictionary.
python/mlx/optimizers/optimizers.py:122
Methodstate
(self)
python/mlx/optimizers/optimizers.py:209
Methodstate
mlx/primitives.h:204
Methodstate
mlx/primitives.h:224
Methodstate
mlx/primitives.h:345
Methodstate
mlx/primitives.h:372
Methodstate
mlx/primitives.h:394
Methodstate
mlx/primitives.h:415
Methodstate
mlx/primitives.h:437
Methodstate
mlx/primitives.h:480
Methodstate
mlx/primitives.h:517
Methodstate
mlx/primitives.h:546
Methodstate
mlx/primitives.h:581
Methodstate
mlx/primitives.h:604
Methodstate
mlx/primitives.h:685
Methodstate
mlx/primitives.h:759
Methodstate
mlx/primitives.h:987
Methodstate
mlx/primitives.h:1066
Methodstate
mlx/primitives.h:1117
Methodstate
mlx/primitives.h:1170
Methodstate
mlx/primitives.h:1192
Methodstate
mlx/primitives.h:1242
Methodstate
mlx/primitives.h:1330
Methodstate
mlx/primitives.h:1538
Methodstate
mlx/primitives.h:1569
Methodstate
mlx/primitives.h:1592
Methodstate
mlx/primitives.h:1637
Methodstate
mlx/primitives.h:1668
Methodstate
mlx/primitives.h:1703
Methodstate
mlx/primitives.h:1728
Methodstate
mlx/primitives.h:1763
Methodstate
mlx/primitives.h:1810
Methodstate
mlx/primitives.h:1872
Methodstate
mlx/primitives.h:1916
Methodstate
mlx/primitives.h:1950
Methodstate
mlx/primitives.h:2048
Methodstate
mlx/primitives.h:2098
Methodstate
mlx/primitives.h:2125
Methodstate
mlx/primitives.h:2147
Methodstate
mlx/primitives.h:2169
Methodstate
mlx/primitives.h:2190
Methodstate
mlx/primitives.h:2212
Methodstate
mlx/primitives.h:2249
Methodstate
mlx/primitives.h:2311
Methodstate
mlx/primitives.h:2363
Methodstate
mlx/primitives.h:2384
Methodstate
mlx/primitives.h:2406
Methodstate
mlx/primitives.h:2442
Methodstate
mlx/primitives.h:2461
Methodstate
mlx/primitives.h:2477
Methodstate
mlx/primitives.h:2503
Methodstate
mlx/primitives.h:2528
Methodstate
mlx/fast_primitives.h:66
Methodstate
mlx/fast_primitives.h:91
Methodstate
mlx/fast_primitives.h:125
Methodstate
mlx/fast_primitives.h:150
Methodstate
mlx/fast_primitives.h:193
Methodstate
mlx/fast_primitives.h:251
Methodstate
mlx/fast_primitives.h:288
Methodstate
mlx/fast_primitives.h:316
Methodstate
mlx/fast_primitives.h:352
Methodstate
mlx/fast_primitives.h:400
Functionstd
mlx/ops.cpp:2249
Functionsteel_gemm_splitk_axpby
mlx/backend/metal/matmul.cpp:501
Functionsteel_gemm_splitk_axpby_nax
mlx/backend/metal/matmul.cpp:658
Functionsteel_matmul_regular_axpby
mlx/backend/metal/matmul.cpp:341
Functionsteel_matmul_regular_axpby_nax
mlx/backend/metal/matmul.cpp:176
Methodstep
(self)
python/mlx/optimizers/optimizers.py:132
← previousnext →3,901–4,000 of 4,946, ranked by callers