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
MethodLoad
mlx/primitives.h:1294
MethodLog
mlx/primitives.h:1319
FunctionLog1p
mlx/primitives.h:1352
FunctionLogAddExp
mlx/primitives.h:1407
MethodLogSumExp
mlx/primitives.h:1421
FunctionLogicalAnd
mlx/primitives.h:1379
FunctionLogicalNot
mlx/primitives.h:1365
FunctionLogicalOr
mlx/primitives.h:1393
MethodLoopedElemToLoc
mlx/backend/metal/kernels/utils.h:208
MethodLoopedElemToLoc
mlx/backend/metal/kernels/utils.h:257
MethodLoopedElemToLoc
mlx/backend/metal/kernels/utils.h:286
MethodMMATile
mlx/backend/metal/kernels/steel/gemm/mma.h:235
MethodMMATile
mlx/backend/metal/kernels/steel/attn/mma.h:258
MethodMPIGroup
mlx/distributed/mpi/mpi.cpp:348
MethodMPIWrapper
mlx/distributed/mpi/mpi.cpp:101
MethodMULPM
mlx/3rdparty/pocketfft.h:1569
MethodMaskedScatter
mlx/primitives.h:1961
MethodMatmul
mlx/primitives.h:1435
FunctionMaximum
mlx/primitives.h:1449
MethodMeshGroup
mlx/distributed/jaccl/lib/jaccl/mesh.cpp:9
MethodMeshImpl
mlx/distributed/jaccl/lib/jaccl/mesh_impl.h:18
MethodMetalAllocator
mlx/backend/metal/allocator.cpp:34
FunctionMinimum
mlx/primitives.h:1463
FunctionMultiply
mlx/primitives.h:1477
MethodNAXTile
mlx/backend/metal/kernels/steel/gemm/nax.h:564
MethodNAXTile
mlx/backend/metal/kernels/steel/attn/nax.h:564
FunctionNB_MODULE
python/src/mlx.cpp:28
FunctionNB_MODULE
examples/extensions/bindings.cpp:11
MethodNCCLComm
mlx/distributed/nccl/nccl.cpp:277
MethodNCCLGroup
mlx/distributed/nccl/nccl.cpp:305
FunctionNegative
mlx/primitives.h:1491
FunctionNotEqual
mlx/primitives.h:1505
MethodNumberOfElements
mlx/primitives.h:1519
MethodPad
mlx/primitives.h:1552
MethodParallelFileReader
mlx/io/load.h:61
MethodPartition
mlx/primitives.h:1581
MethodPathNode
mlx/einsum.cpp:40
FunctionPower
mlx/primitives.h:1603
FunctionPrimitive
mlx/primitives.h:117
MethodPrimitiveFactory
mlx/export.cpp:453
MethodPrimitiveSerializer
mlx/export.cpp:41
MethodPrintOptionsContext
python/src/print.cpp:19
MethodPyCheckpointedFun
python/src/transforms.cpp:592
MethodPyCompiledFun
python/src/transforms.cpp:426
MethodPyCustomFunction
python/src/transforms.cpp:683
MethodPyCustomKernelFunction
python/src/fast.cpp:23
MethodPyFileReader
python/src/load.cpp:100
MethodPyFileWriter
python/src/load.cpp:330
MethodPyFunctionExporter
python/src/export.cpp:73
MethodPyStreamContext
python/src/stream.cpp:20
MethodQQMatmul
mlx/primitives.h:1650
FunctionQRF
mlx/primitives.h:2419
MethodQuantize
mlx/fast_primitives.h:329
MethodQuantizedBlockLoader
mlx/backend/metal/kernels/quantized.h:605
MethodQuantizedBlockLoader
mlx/backend/metal/kernels/fp_quantized.h:183
MethodQuantizedBlockLoader
mlx/backend/metal/kernels/fp_quantized_nax.h:109
MethodQuantizedBlockLoader
mlx/backend/metal/kernels/quantized_nax.h:608
MethodQuantizedBlockLoader
mlx/backend/metal/kernels/quantized_nax.h:748
MethodQuantizedMatmul
mlx/primitives.h:1617
MethodRMSNorm
mlx/fast_primitives.h:41
MethodRMSNormVJP
mlx/fast_primitives.h:76
FunctionROT90
mlx/3rdparty/pocketfft.h:287
MethodROTX135
mlx/3rdparty/pocketfft.h:1131
MethodROTX45
mlx/3rdparty/pocketfft.h:1123
FunctionROTX90
mlx/3rdparty/pocketfft.h:289
MethodRandomBits
mlx/primitives.h:1719
MethodReal
mlx/primitives.h:1739
MethodRecv
mlx/distributed/primitives.h:116
MethodReduce
mlx/primitives.h:1777
MethodReduceScatter
mlx/distributed/primitives.h:133
MethodReductionPlan
mlx/backend/common/reduce.h:44
FunctionRemainder
mlx/primitives.h:955
MethodReshape
mlx/primitives.h:1753
MethodResidencySet
mlx/backend/metal/resident.cpp:8
MethodResidencySet
mlx/backend/metal/resident.h:16
MethodRetainGraph
mlx/transforms_impl.h:52
MethodRingGroup
mlx/distributed/ring/ring.cpp:383
MethodRingGroup
mlx/distributed/jaccl/lib/jaccl/ring.cpp:9
MethodRingImpl
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:15
FunctionRound
mlx/primitives.h:1821
MethodRowReduceArgs
mlx/backend/metal/reduce.cpp:36
MethodSVD
mlx/primitives.h:2432
MethodSVDWork
mlx/backend/cpu/svd.cpp:30
MethodSVDWork
mlx/backend/cpu/svd.cpp:113
MethodScaledDotProductAttention
mlx/fast_primitives.h:208
MethodScaledDotProductAttentionVJP
mlx/fast_primitives.h:265
MethodScan
mlx/primitives.h:1837
MethodScatter
mlx/primitives.h:1887
MethodScatterAxis
mlx/primitives.h:1929
MethodScheduler
mlx/scheduler.cpp:33
FunctionSegmentedMM
mlx/primitives.h:557
FunctionSelect
mlx/primitives.h:941
MethodSend
mlx/distributed/primitives.h:97
MethodShape2D
mlx/backend/metal/kernels/steel/attn/mma.h:27
MethodSharedBuffer
mlx/distributed/jaccl/lib/jaccl/rdma.h:105
MethodSharedBuffer
mlx/distributed/jaccl/lib/jaccl/rdma.cpp:74
MethodSideChannel
mlx/distributed/jaccl/lib/jaccl/rdma.h:272
MethodSideChannel
mlx/distributed/jaccl/lib/jaccl/rdma.cpp:297
FunctionSigmoid
mlx/primitives.h:1975
FunctionSign
mlx/primitives.h:1989
← previousnext →2,001–2,100 of 4,946, ranked by callers