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
↓ 1 callersFunctioninit_ops
python/src/ops.cpp:48
↓ 1 callersFunctioninit_print
python/src/print.cpp:30
↓ 1 callersFunctioninit_random
python/src/random.cpp:66
↓ 1 callersFunctioninit_reduce
mlx/backend/metal/reduce.cpp:289
↓ 1 callersMethodinit_safe
mlx/distributed/mpi/mpi.cpp:165
↓ 1 callersMethodinit_single
To be extended by the children classes to implement each optimizer's state initialization. Args: parameter (mx.array): A
python/mlx/optimizers/optimizers.py:75
↓ 1 callersFunctioninit_stream
python/src/stream.cpp:44
↓ 1 callersFunctioninit_transforms
python/src/transforms.cpp:1022
↓ 1 callersMethodinitialize
(self)
python/tests/test_nn.py:64
↓ 1 callersMethodinner
(x, w, y)
python/tests/test_fast.py:480
↓ 1 callersMethodinsert_graph_dependencies
mlx/backend/cuda/device.cpp:129
↓ 1 callersFunctionirfft2
Compute the two-dimensional inverse of `rfft2`. */
mlx/fft.h:200
↓ 1 callersFunctionis_available
mlx/distributed/distributed.cpp:105
↓ 1 callersMethodis_available
mlx/distributed/mpi/mpi.cpp:161
↓ 1 callersMethodis_available
mlx/distributed/jaccl/lib/jaccl/rdma.h:54
↓ 1 callersMethodis_big
mlx/small_vector.h:492
↓ 1 callersFunctionis_binary
mlx/compile.cpp:47
↓ 1 callersMethodis_equivalent
mlx/compile.cpp:173
↓ 1 callersFunctionis_fusable
mlx/compile.cpp:77
↓ 1 callersFunctionis_noop
mlx/compile.cpp:69
↓ 1 callersFunctionis_ternary
mlx/compile.cpp:61
↓ 1 callersFunctionis_unary
mlx/compile.cpp:26
↓ 1 callersMethodis_valid_mesh
mlx/distributed/jaccl/lib/jaccl/jaccl.cpp:109
↓ 1 callersFunctionis_windows
Check if running on Windows or Windows Subsystem for Linux
mlx/backend/cuda/allocator.cpp:32
↓ 1 callersFunctionis_zip_file
python/src/load.cpp:44
↓ 1 callersFunctionlaunch_jaccl
(parser, hosts, args, command)
python/mlx/_distributed_utils/launch.py:366
↓ 1 callersFunctionlaunch_mpi
(parser, hosts, args, command)
python/mlx/_distributed_utils/launch.py:418
↓ 1 callersFunctionlaunch_nccl
(parser, hosts, args, command)
python/mlx/_distributed_utils/launch.py:327
↓ 1 callersFunctionlaunch_ring
(parser, hosts, args, command)
python/mlx/_distributed_utils/launch.py:292
↓ 1 callersFunctionldexp_cexpf
mlx/backend/metal/kernels/cexpf.h:64
↓ 1 callersFunctionleaky_relu
r"""Applies the Leaky Rectified Linear Unit. Simply ``mx.maximum(negative_slope * x, x)``.
python/mlx/nn/layers/activations.py:57
↓ 1 callersFunctionload_arrays
mlx/io/gguf.cpp:214
↓ 1 callersFunctionload_cudnn
mlx/backend/cuda/delayload.cpp:38
↓ 1 callersFunctionload_default_library
mlx/backend/metal/device.cpp:163
↓ 1 callersFunctionload_device
mlx/backend/metal/device.cpp:50
↓ 1 callersFunctionload_library
mlx/backend/metal/device.cpp:205
↓ 1 callersFunctionload_metadata
mlx/io/gguf.cpp:203
↓ 1 callersFunctionload_module
mlx/backend/cuda/jit_module.cpp:346
↓ 1 callersFunctionload_nodes
* Load all addresses from the json hostfile. The hostfile is a list of * addresses in order of rank. For each rank there can be many addresses so *
mlx/distributed/ring/ring.cpp:311
↓ 1 callersMethodload_padded
mlx/backend/metal/kernels/fft/readwrite.h:318
↓ 1 callersFunctionload_safe
mlx/backend/metal/kernels/scan.h:181
↓ 1 callersMethodload_strided
mlx/backend/metal/kernels/fft/readwrite.h:224
↓ 1 callersFunctionload_unsafe
mlx/backend/metal/kernels/scan.h:168
↓ 1 callersFunctionlog1p
mlx/backend/cpu/simd/base_simd.h:112
↓ 1 callersFunctionlogsumexp
mlx/backend/metal/kernels/logsumexp.h:4
↓ 1 callersFunctionloss_fn
(w)
examples/python/logistic_regression.py:26
↓ 1 callersFunctionloss_fn
(w)
examples/python/linear_regression.py:26
↓ 1 callersFunctionloss_fn
(params, image)
benchmarks/python/conv3d_train_bench_cpu.py:48
↓ 1 callersFunctionloss_fn
(params, image)
benchmarks/python/conv2d_train_bench_cpu.py:48
↓ 1 callersFunctionmain
()
python/mlx/__main__.py:4
↓ 1 callersFunctionmain
()
benchmarks/python/conv3d_train_bench_cpu.py:128
↓ 1 callersFunctionmain
()
benchmarks/python/segmented_mm_bench.py:102
↓ 1 callersFunctionmain
()
benchmarks/python/conv2d_train_bench_cpu.py:128
↓ 1 callersFunctionmain
()
benchmarks/python/masked_scatter.py:208
↓ 1 callersFunctionmain
()
benchmarks/python/block_masked_mm_bench.py:88
↓ 1 callersFunctionmake_buffer
mlx/allocator.h:39
↓ 1 callersMethodmake_buffer
mlx/backend/metal/allocator.cpp:196
↓ 1 callersFunctionmake_connectivity_matrix
(tb_hosts, uuid_reverse_index)
python/mlx/_distributed_utils/config.py:265
↓ 1 callersMethodmake_kill_script
(pidfile)
python/mlx/_distributed_utils/launch.py:156
↓ 1 callersMethodmake_launch_script
(rank, cwd, files, env, command, is_local)
python/mlx/_distributed_utils/launch.py:107
↓ 1 callersFunctionmake_masks
Create block masks with given sparsity (fraction of blocks zeroed).
benchmarks/python/block_masked_mm_bench.py:26
↓ 1 callersFunctionmake_mx_conv_1D
(strides=1, padding=0, groups=1)
benchmarks/python/conv1d_bench.py:31
↓ 1 callersFunctionmake_mx_conv_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv_bench.py:31
↓ 1 callersFunctionmake_mx_conv_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv_unaligned_bench.py:25
↓ 1 callersFunctionmake_mx_conv_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv2d_bench_cpu.py:26
↓ 1 callersFunctionmake_mx_conv_3D
(strides=(1, 1, 1), padding=(0, 0, 0), groups=1)
benchmarks/python/conv3d_bench.py:25
↓ 1 callersFunctionmake_mx_conv_3D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv3d_bench_cpu.py:26
↓ 1 callersFunctionmake_mx_conv_3D
(strides=(1, 1, 1), padding=(0, 0, 0), groups=1)
benchmarks/python/conv3d_transpose_bench_cpu.py:26
↓ 1 callersFunctionmake_mx_conv_transpose_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv2d_transpose_bench_cpu.py:25
↓ 1 callersFunctionmake_mx_conv_transpose_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv_transpose_bench.py:28
↓ 1 callersFunctionmake_predicate
(positive_filter, negative_filter)
benchmarks/python/comparative/compare.py:47
↓ 1 callersFunctionmake_pt_conv_1D
(strides=1, padding=0, groups=1)
benchmarks/python/conv1d_bench.py:43
↓ 1 callersFunctionmake_pt_conv_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv_bench.py:43
↓ 1 callersFunctionmake_pt_conv_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv_unaligned_bench.py:37
↓ 1 callersFunctionmake_pt_conv_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv2d_bench_cpu.py:38
↓ 1 callersFunctionmake_pt_conv_3D
(strides=(1, 1, 1), padding=(0, 0, 0), groups=1)
benchmarks/python/conv3d_bench.py:37
↓ 1 callersFunctionmake_pt_conv_3D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv3d_bench_cpu.py:38
↓ 1 callersFunctionmake_pt_conv_3D
(strides=(1, 1, 1), padding=(0, 0, 0), groups=1)
benchmarks/python/conv3d_transpose_bench_cpu.py:40
↓ 1 callersFunctionmake_pt_conv_transpose_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv2d_transpose_bench_cpu.py:39
↓ 1 callersFunctionmake_pt_conv_transpose_2D
(strides=(1, 1), padding=(0, 0), groups=1)
benchmarks/python/conv_transpose_bench.py:42
↓ 1 callersFunctionmake_segments
(k, num_segments, pattern, seed)
benchmarks/python/segmented_mm_bench.py:24
↓ 1 callersMethodmalloc_async
mlx/backend/cuda/allocator.cpp:183
↓ 1 callersMethodmanaged_memory
mlx/backend/cuda/device.h:188
↓ 1 callersFunctionmax_occupancy_block_dim
mlx/backend/cuda/utils.h:14
↓ 1 callersFunctionmean
mlx/ops.cpp:2089
↓ 1 callersFunctionmedian
mlx/ops.h:625
↓ 1 callersMethodmemory_pools
mlx/backend/cuda/device.h:191
↓ 1 callersFunctionmetal_fast_synch
mlx/utils.h:166
↓ 1 callersFunctionmetal_kernel
mlx/backend/metal/no_metal.cpp:29
↓ 1 callersFunctionmlx_add_item
python/src/indexing.cpp:979
↓ 1 callersFunctionmlx_atomic_fetch_add_explicit
mlx/backend/metal/kernels/atomic.h:83
↓ 1 callersFunctionmlx_atomic_fetch_and_explicit
mlx/backend/metal/kernels/atomic.h:51
↓ 1 callersFunctionmlx_atomic_fetch_max_explicit
mlx/backend/metal/kernels/atomic.h:75
↓ 1 callersFunctionmlx_atomic_fetch_min_explicit
mlx/backend/metal/kernels/atomic.h:67
↓ 1 callersFunctionmlx_atomic_fetch_mul_explicit
mlx/backend/metal/kernels/atomic.h:91
↓ 1 callersFunctionmlx_atomic_fetch_or_explicit
mlx/backend/metal/kernels/atomic.h:59
↓ 1 callersFunctionmlx_divide_item
python/src/indexing.cpp:1033
↓ 1 callersFunctionmlx_gather_nd
python/src/indexing.cpp:136
↓ 1 callersFunctionmlx_get_item_int
python/src/indexing.cpp:124
↓ 1 callersFunctionmlx_get_item_nd
python/src/indexing.cpp:271
← previousnext →1,501–1,600 of 4,946, ranked by callers