Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ml-explore/mlx
/ functions
Functions
4,946 in github.com/ml-explore/mlx
⨍
Functions
4,946
◇
Types & classes
770
↳
Endpoints
5
Function
slice_update_impl
mlx/backend/cpu/indexing.cpp:858
Function
slice_update_op_impl
mlx/backend/metal/kernels/indexing/scatter.h:70
Function
slow_conv_1D
mlx/backend/cpu/conv.cpp:21
Function
slow_conv_2D
mlx/backend/cpu/conv.cpp:110
Function
slow_conv_3D
mlx/backend/cpu/conv.cpp:358
Function
smooth_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
Function
softmax
mlx/ops.cpp:3810
Function
softmax
(axis, x)
benchmarks/python/comparative/bench_mlx.py:154
Function
softmax
(axis, x)
benchmarks/python/comparative/bench_torch.py:114
Function
softmax_fused
(axis, x)
benchmarks/python/comparative/bench_mlx.py:163
Function
softmax_fused
(axis, x)
benchmarks/python/comparative/bench_torch.py:124
Function
softmax_looped
mlx/backend/metal/kernels/softmax.h:101
Function
softmax_single_row
mlx/backend/metal/kernels/softmax.h:11
Function
softmin
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
Function
softplus
(x: mx.array)
benchmarks/python/comparative/bench_mlx.py:192
Function
softplus
(x)
benchmarks/python/comparative/bench_torch.py:172
Function
softsign
r"""Applies the Softsign function. Applies :math:`\frac{x}{1 + |x|}` element wise.
python/mlx/nn/layers/activations.py:102
Function
sort
(axis, x)
benchmarks/python/comparative/bench_mlx.py:321
Function
sort
(axis, x)
benchmarks/python/comparative/bench_torch.py:281
Method
sort
mlx/backend/metal/kernels/sort.h:158
Function
sparse
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
Function
special_mul
mlx/3rdparty/pocketfft.h:281
Method
special_mul
mlx/3rdparty/pocketfft.h:265
Method
split
mlx/distributed/distributed.cpp:67
Method
split
mlx/distributed/ring/ring.cpp:492
Method
split
mlx/distributed/mpi/mpi.cpp:377
Method
split
mlx/distributed/jaccl/jaccl.cpp:151
Method
split
mlx/distributed/nccl/nccl.cpp:287
Method
split
mlx/distributed/nccl/nccl.cpp:339
Function
start_capture
mlx/backend/metal/no_metal.cpp:16
Function
start_concurrent
mlx/backend/metal/device.h:87
Method
start_threads
mlx/threadpool.h:113
Method
state
The optimizer's state dictionary.
python/mlx/optimizers/optimizers.py:122
Method
state
(self)
python/mlx/optimizers/optimizers.py:209
Method
state
mlx/primitives.h:204
Method
state
mlx/primitives.h:224
Method
state
mlx/primitives.h:345
Method
state
mlx/primitives.h:372
Method
state
mlx/primitives.h:394
Method
state
mlx/primitives.h:415
Method
state
mlx/primitives.h:437
Method
state
mlx/primitives.h:480
Method
state
mlx/primitives.h:517
Method
state
mlx/primitives.h:546
Method
state
mlx/primitives.h:581
Method
state
mlx/primitives.h:604
Method
state
mlx/primitives.h:685
Method
state
mlx/primitives.h:759
Method
state
mlx/primitives.h:987
Method
state
mlx/primitives.h:1066
Method
state
mlx/primitives.h:1117
Method
state
mlx/primitives.h:1170
Method
state
mlx/primitives.h:1192
Method
state
mlx/primitives.h:1242
Method
state
mlx/primitives.h:1330
Method
state
mlx/primitives.h:1538
Method
state
mlx/primitives.h:1569
Method
state
mlx/primitives.h:1592
Method
state
mlx/primitives.h:1637
Method
state
mlx/primitives.h:1668
Method
state
mlx/primitives.h:1703
Method
state
mlx/primitives.h:1728
Method
state
mlx/primitives.h:1763
Method
state
mlx/primitives.h:1810
Method
state
mlx/primitives.h:1872
Method
state
mlx/primitives.h:1916
Method
state
mlx/primitives.h:1950
Method
state
mlx/primitives.h:2048
Method
state
mlx/primitives.h:2098
Method
state
mlx/primitives.h:2125
Method
state
mlx/primitives.h:2147
Method
state
mlx/primitives.h:2169
Method
state
mlx/primitives.h:2190
Method
state
mlx/primitives.h:2212
Method
state
mlx/primitives.h:2249
Method
state
mlx/primitives.h:2311
Method
state
mlx/primitives.h:2363
Method
state
mlx/primitives.h:2384
Method
state
mlx/primitives.h:2406
Method
state
mlx/primitives.h:2442
Method
state
mlx/primitives.h:2461
Method
state
mlx/primitives.h:2477
Method
state
mlx/primitives.h:2503
Method
state
mlx/primitives.h:2528
Method
state
mlx/fast_primitives.h:66
Method
state
mlx/fast_primitives.h:91
Method
state
mlx/fast_primitives.h:125
Method
state
mlx/fast_primitives.h:150
Method
state
mlx/fast_primitives.h:193
Method
state
mlx/fast_primitives.h:251
Method
state
mlx/fast_primitives.h:288
Method
state
mlx/fast_primitives.h:316
Method
state
mlx/fast_primitives.h:352
Method
state
mlx/fast_primitives.h:400
Function
std
mlx/ops.cpp:2249
Function
steel_gemm_splitk_axpby
mlx/backend/metal/matmul.cpp:501
Function
steel_gemm_splitk_axpby_nax
mlx/backend/metal/matmul.cpp:658
Function
steel_matmul_regular_axpby
mlx/backend/metal/matmul.cpp:341
Function
steel_matmul_regular_axpby_nax
mlx/backend/metal/matmul.cpp:176
Method
step
(self)
python/mlx/optimizers/optimizers.py:132
← previous
next →
3,901–4,000 of 4,946, ranked by callers