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
FunctionCommandEncoder
mlx/backend/cpu/encoder.h:16
MethodCommandEncoder
mlx/backend/metal/device.cpp:263
MethodCommandEncoder
mlx/backend/cuda/device.cpp:205
MethodCommandEncoder
mlx/backend/cuda/device.h:39
MethodCommonAllocator
mlx/backend/no_gpu/allocator.cpp:54
MethodCompiled
mlx/compile.cpp:81
MethodCompilerCache
mlx/compile.cpp:383
MethodConcatenate
mlx/primitives.h:674
MethodConcurrentContext
mlx/backend/metal/device.h:33
MethodConcurrentContext
mlx/backend/cuda/device.cpp:93
MethodConfig
mlx/distributed/jaccl/lib/jaccl/jaccl.cpp:76
FunctionConjugate
mlx/primitives.h:695
MethodConnection
mlx/distributed/jaccl/lib/jaccl/rdma.h:164
MethodConnection
mlx/distributed/jaccl/lib/jaccl/rdma.cpp:110
MethodContiguous
mlx/primitives.h:709
MethodContiguousIterator
mlx/backend/common/utils.h:167
MethodContraction
mlx/einsum.cpp:169
MethodConvParams
mlx/backend/cuda/conv/conv.h:26
MethodConvertFP8
mlx/fast_primitives.h:300
MethodConvolution
mlx/primitives.h:728
MethodCopy
mlx/primitives.h:791
FunctionCos
mlx/primitives.h:808
FunctionCosh
mlx/primitives.h:822
MethodCublasGemm
mlx/backend/cuda/gemms/cublas_gemm.cpp:38
MethodCublasMatmulBase
mlx/backend/cuda/cublas_utils.h:56
MethodCublasQQMM
mlx/backend/cuda/quantized/cublas_qqmm.cpp:40
MethodCudaAllocator
mlx/backend/cuda/allocator.cpp:160
MethodCudaEvent
mlx/backend/cuda/event.h:32
MethodCudaGraph
mlx/backend/cuda/utils.cpp:63
MethodCudaHandle
mlx/backend/cuda/cuda_utils.h:21
MethodCudaStream
mlx/backend/cuda/utils.cpp:77
MethodCustom
mlx/fast_primitives.h:15
MethodCustomKernel
mlx/fast_primitives.h:367
MethodCustomTransforms
mlx/primitives.h:836
MethodDLib
mlx/backend/cpu/compiled.cpp:24
MethodData
mlx/array.h:234
MethodDepends
mlx/primitives.h:888
MethodDevice
mlx/backend/metal/device.cpp:482
MethodDevice
mlx/backend/cuda/device.h:166
MethodDistPrimitive
mlx/distributed/primitives.h:13
MethodDivMod
mlx/primitives.h:923
FunctionDivide
mlx/primitives.h:909
MethodDnnGraph
mlx/backend/cuda/cudnn_utils.h:96
MethodDtype
mlx/dtype.h:52
MethodDynamicSlice
mlx/primitives.h:2112
MethodDynamicSliceUpdate
mlx/primitives.h:2136
MethodEig
mlx/primitives.h:2490
MethodEigWork
mlx/backend/cpu/eig.cpp:36
MethodEigWork
mlx/backend/cpu/eig.cpp:129
MethodEigh
mlx/primitives.h:2513
MethodEighWork
mlx/backend/cpu/eigh.cpp:32
MethodEighWork
mlx/backend/cpu/eigh.cpp:84
MethodEqual
mlx/primitives.h:969
FunctionErf
mlx/primitives.h:997
FunctionErfInv
mlx/primitives.h:1011
MethodEvent
mlx/event.h:14
MethodEvent
mlx/backend/no_gpu/event.cpp:17
MethodEvent
mlx/backend/metal/event.cpp:9
MethodExpandDims
mlx/primitives.h:1052
FunctionExpm1
mlx/primitives.h:1039
MethodFFT
mlx/primitives.h:1077
MethodFence
mlx/fence.h:29
MethodFence
mlx/backend/no_gpu/fence.cpp:18
MethodFence
mlx/backend/metal/fence.cpp:47
MethodFence
mlx/backend/cuda/fence.cpp:15
MethodFenceImpl
mlx/backend/metal/fence.cpp:10
MethodFileWriter
mlx/io/load.h:113
MethodFlatten
mlx/primitives.h:1104
FunctionFloor
mlx/primitives.h:1129
FunctionFull
mlx/primitives.h:1143
MethodFunction
mlx/export.cpp:533
MethodFunctionExporter
mlx/export.cpp:673
MethodFunctionTable
mlx/export.cpp:531
MethodGather
mlx/primitives.h:1157
MethodGatherAxis
mlx/primitives.h:1181
MethodGatherMM
mlx/primitives.h:527
MethodGatherQMM
mlx/primitives.h:1680
FunctionGreater
mlx/primitives.h:1202
FunctionGreaterEqual
mlx/primitives.h:1216
MethodHadamard
mlx/primitives.h:1230
MethodIBVWrapper
mlx/distributed/jaccl/lib/jaccl/rdma.cpp:36
FunctionImag
mlx/primitives.h:1252
MethodImportedFunction
mlx/export.cpp:1038
MethodInTracing
mlx/transforms_impl.h:25
MethodInnerFunction
python/src/transforms.cpp:604
MethodInnerFunction
python/src/transforms.cpp:694
MethodInnerJVPFunction
python/src/transforms.cpp:775
MethodInnerVJPFunction
python/src/transforms.cpp:728
MethodInnerVmapFunction
python/src/transforms.cpp:830
MethodInverse
mlx/primitives.h:2453
MethodJACCLGroup
mlx/distributed/jaccl/jaccl.cpp:63
MethodJitModule
mlx/backend/cuda/jit_module.h:97
MethodJitModule
mlx/backend/cuda/jit_module.cpp:376
MethodKeySequence
mlx/random.cpp:14
MethodLRUCache
mlx/backend/cuda/lru_cache.h:28
MethodLUF
mlx/primitives.h:2540
MethodLayerNorm
mlx/fast_primitives.h:101
MethodLayerNormVJP
mlx/fast_primitives.h:135
FunctionLess
mlx/primitives.h:1266
FunctionLessEqual
mlx/primitives.h:1280
← previousnext →1,901–2,000 of 4,946, ranked by callers