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
MethodSimd
mlx/backend/cpu/simd/base_simd.h:27
MethodSimd<T, N>
mlx/backend/cpu/simd/accelerate_simd.h:60
MethodSimd<float16_t, N>
mlx/backend/cpu/simd/neon_fp16_simd.h:16
FunctionSin
mlx/primitives.h:2003
FunctionSinh
mlx/primitives.h:2017
MethodSlice
mlx/primitives.h:2031
MethodSliceUpdate
mlx/primitives.h:2062
MethodSmallSizePool
mlx/backend/cuda/allocator.h:42
MethodSmallSizePool
mlx/backend/cuda/allocator.cpp:104
MethodSmallVector
mlx/small_vector.h:103
MethodSocketTask
mlx/distributed/ring/ring.cpp:155
MethodSocketThread
mlx/distributed/ring/ring.cpp:130
MethodSoftmax
mlx/primitives.h:2157
MethodSort
mlx/primitives.h:2179
MethodSplit
mlx/primitives.h:2200
MethodSqrt
mlx/primitives.h:2239
FunctionSquare
mlx/primitives.h:2225
MethodSqueeze
mlx/primitives.h:2297
MethodStopGradient
mlx/primitives.h:2267
MethodStreamContext
mlx/utils.h:23
MethodStreamThread
mlx/scheduler.h:26
MethodStridedIterator
Constructors
mlx/backend/cpu/sort.cpp:43
MethodSubscript
mlx/einsum.cpp:25
FunctionSubtract
mlx/primitives.h:2283
MethodSynchronizer
mlx/transforms.cpp:53
MethodTCPSocket
mlx/distributed/utils.cpp:53
MethodTCPSocket
mlx/distributed/utils.h:36
MethodTCPSocket
mlx/distributed/jaccl/lib/jaccl/tcp.h:36
MethodTCPSocket
mlx/distributed/jaccl/lib/jaccl/tcp.cpp:53
MethodT_dct1
mlx/3rdparty/pocketfft.h:2544
MethodT_dst1
mlx/3rdparty/pocketfft.h:2575
FunctionTan
mlx/primitives.h:2322
FunctionTanh
mlx/primitives.h:2336
MethodThreadPool
mlx/threadpool.h:55
MethodTransformAdd
mlx/backend/metal/kernels/steel/gemm/transforms.h:27
MethodTransformAdd
mlx/backend/metal/kernels/steel/attn/transforms.h:27
MethodTransformAxpby
mlx/backend/metal/kernels/steel/gemm/transforms.h:43
MethodTransformAxpby
mlx/backend/metal/kernels/steel/attn/transforms.h:43
MethodTransformScale
mlx/backend/metal/kernels/steel/attn/kernels/steel_attention_nax.h:24
MethodTranspose
mlx/primitives.h:2395
MethodUnflatten
mlx/primitives.h:2350
MethodView
mlx/primitives.h:2375
MethodVisualStudioInfo
mlx/backend/cpu/jit_compiler.cpp:30
MethodWorker
mlx/backend/cuda/worker.h:22
MethodWorker
mlx/backend/cuda/worker.cpp:8
MethodZipFileWrapper
python/src/load.cpp:56
Method_MLX_BFloat16
Default constructor
mlx/types/bf16.h:26
Method_MLX_Float16
Default constructor
mlx/types/fp16.h:26
Method__bool__
(self)
python/mlx/_distributed_utils/config.py:30
Method__call__
(self, x)
python/mlx/nn/layers/containers.py:21
Method__call__
(self, x)
python/mlx/nn/layers/convolution_transpose.py:70
Method__call__
(self, x)
python/mlx/nn/layers/convolution_transpose.py:149
Method__call__
(self, x)
python/mlx/nn/layers/convolution_transpose.py:231
Method__call__
(self, x)
python/mlx/nn/layers/quantized.py:144
Method__call__
(self, x)
python/mlx/nn/layers/quantized.py:265
Method__call__
(self, x)
python/mlx/nn/layers/quantized.py:399
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/linear.py:22
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/linear.py:66
Method__call__
(self, x1: mx.array, x2: mx.array)
python/mlx/nn/layers/linear.py:159
Method__call__
(self, x, hidden=None)
python/mlx/nn/layers/recurrent.py:75
Method__call__
(self, x, hidden=None)
python/mlx/nn/layers/recurrent.py:156
Method__call__
(self, x, hidden=None, cell=None)
python/mlx/nn/layers/recurrent.py:262
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/upsample.py:250
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/dropout.py:29
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/dropout.py:72
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/dropout.py:121
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/distributed.py:246
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/distributed.py:330
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/distributed.py:440
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/distributed.py:574
Method__call__
(self, x, offset: int = 0)
python/mlx/nn/layers/positional_encoding.py:46
Method__call__
(self, x)
python/mlx/nn/layers/positional_encoding.py:101
Method__call__
(self, attention_scores, offset=0, mask=None)
python/mlx/nn/layers/positional_encoding.py:152
Method__call__
(self, x)
python/mlx/nn/layers/pooling.py:102
Method__call__
(self, x)
python/mlx/nn/layers/activations.py:355
Method__call__
(self, x)
python/mlx/nn/layers/activations.py:418
Method__call__
(self, x)
python/mlx/nn/layers/activations.py:436
Method__call__
(self, x)
python/mlx/nn/layers/activations.py:477
Method__call__
(self, x)
python/mlx/nn/layers/activations.py:496
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/activations.py:540
Method__call__
(self, x)
python/mlx/nn/layers/activations.py:581
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/activations.py:625
Method__call__
(self, x)
python/mlx/nn/layers/convolution.py:76
Method__call__
(self, x)
python/mlx/nn/layers/convolution.py:156
Method__call__
(self, x)
python/mlx/nn/layers/convolution.py:228
Method__call__
(self, queries, keys, values, mask=None)
python/mlx/nn/layers/transformer.py:79
Method__call__
(self, x, mask)
python/mlx/nn/layers/transformer.py:125
Method__call__
(self, x, mask)
python/mlx/nn/layers/transformer.py:175
Method__call__
(self, x, memory, x_mask, memory_mask)
python/mlx/nn/layers/transformer.py:207
Method__call__
(self, x, memory, x_mask, memory_mask)
python/mlx/nn/layers/transformer.py:266
Method__call__
(self, src, tgt, src_mask, tgt_mask, memory_mask)
python/mlx/nn/layers/transformer.py:352
Method__call__
(self, x: mx.array)
python/mlx/nn/layers/normalization.py:58
Method__call__
(self, x)
python/mlx/nn/layers/normalization.py:107
Method__call__
(self, x)
python/mlx/nn/layers/normalization.py:142
Method__call__
(self, x)
python/mlx/nn/layers/normalization.py:230
Method__call__
Forward pass of BatchNorm. Args: x (array): Input tensor. Returns: array: Normalized output tensor.
python/mlx/nn/layers/normalization.py:336
Method__call__
(self, x)
python/mlx/nn/layers/embedding.py:31
Method__call__
(self, parser, namespace, values, option_string=None)
python/mlx/_distributed_utils/common.py:107
Method__call__
(self, x)
python/tests/test_export_import.py:332
Method__call__
(self, x)
python/tests/test_export_import.py:358
← previousnext →2,101–2,200 of 4,946, ranked by callers