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
Method_extra_repr
(self)
python/mlx/nn/layers/dropout.py:26
Method_extra_repr
(self)
python/mlx/nn/layers/dropout.py:69
Method_extra_repr
(self)
python/mlx/nn/layers/dropout.py:118
Method_extra_repr
(self)
python/mlx/nn/layers/distributed.py:240
Method_extra_repr
(self)
python/mlx/nn/layers/distributed.py:324
Method_extra_repr
(self)
python/mlx/nn/layers/distributed.py:431
Method_extra_repr
(self)
python/mlx/nn/layers/distributed.py:566
Method_extra_repr
(self)
python/mlx/nn/layers/positional_encoding.py:43
Method_extra_repr
(self)
python/mlx/nn/layers/pooling.py:95
Method_extra_repr
(self)
python/mlx/nn/layers/convolution.py:67
Method_extra_repr
(self)
python/mlx/nn/layers/convolution.py:147
Method_extra_repr
(self)
python/mlx/nn/layers/convolution.py:220
Method_extra_repr
(self)
python/mlx/nn/layers/normalization.py:55
Method_extra_repr
(self)
python/mlx/nn/layers/normalization.py:104
Method_extra_repr
(self)
python/mlx/nn/layers/normalization.py:139
Method_extra_repr
(self)
python/mlx/nn/layers/normalization.py:193
Method_extra_repr
(self)
python/mlx/nn/layers/normalization.py:311
Method_extra_repr
(self)
python/mlx/nn/layers/embedding.py:28
Method_freeze_impl
(_, m)
python/mlx/nn/layers/base.py:499
Method_group_norm
(self, x)
python/mlx/nn/layers/normalization.py:215
Method_is_leaf_module
(m, k, v)
python/mlx/nn/layers/base.py:299
Function_linear_indices
(N, scale, align_corners, dim, ndims)
python/mlx/nn/layers/upsample.py:40
Function_make_activation_module
(f)
python/mlx/nn/layers/activations.py:11
Function_maybe_quantize
(path, m)
python/mlx/nn/layers/quantized.py:69
Function_power_of_two_formatter
(value, _position)
benchmarks/python/masked_scatter.py:62
Method_pytorch_compatible_group_norm
(self, x)
python/mlx/nn/layers/normalization.py:199
Function_qmm
mlx/backend/cpu/quantized.cpp:97
Function_qmm_dispatch_group
mlx/backend/cpu/quantized.cpp:310
Function_qmm_dispatch_transpose
mlx/backend/cpu/quantized.cpp:287
Function_qmm_dispatch_typed
mlx/backend/cpu/quantized.cpp:341
Function_qmm_t
mlx/backend/cpu/quantized.cpp:155
Function_qmm_t_simd
mlx/backend/cpu/quantized.cpp:240
Function_quant_matmul
(x, w, s, b, transpose, group_size, bits)
benchmarks/python/comparative/bench_mlx.py:63
Method_read
python/src/load.cpp:164
Method_set_training_mode
(self, mode: bool)
python/mlx/nn/layers/quantized.py:391
Function_shard_fn
(path, weight)
python/mlx/nn/layers/distributed.py:55
Method_test_logits_as_inputs
()
python/tests/test_losses.py:80
Method_test_probs_as_inputs
()
python/tests/test_losses.py:113
Method_test_tiny_probs_as_inputs
()
python/tests/test_losses.py:138
Function_thread_fn
(rank, *args, **kwargs)
python/mlx/_distributed_utils/launch.py:174
Method_unfreeze_impl
(_, m)
python/mlx/nn/layers/base.py:555
Functionabort_with_exception
mlx/utils.cpp:121
Methodaccept
mlx/distributed/utils.cpp:126
Functionadd3
tests/compile_tests.cpp:662
Methodadd_at_head
mlx/backend/common/buffer_cache.h:119
Functionadd_diff
tests/compile_tests.cpp:171
Methodadd_factor
mlx/3rdparty/pocketfft.h:839
Methodadd_factor
mlx/3rdparty/pocketfft.h:1565
Methodadd_kernel_node
mlx/backend/cuda/device.h:53
Methodadd_kernel_node_ex
mlx/backend/cuda/device.h:59
Functionadd_scalars
tests/compile_tests.cpp:112
Functionadd_temporaries
mlx/backend/cpu/encoder.h:32
Functionadd_temporary
Hold onto a temporary until any already scheduled tasks which use it as an input are complete.
mlx/backend/cpu/encoder.h:28
Functionadjust_matrix_offsets
mlx/backend/metal/kernels/quantized.h:1351
Functionadjust_matrix_offsets
mlx/backend/metal/kernels/quantized_nax.h:836
Functionaffine_gather_qmm_n
mlx/backend/metal/kernels/quantized.h:2174
Functionaffine_gather_qmm_n_nax
mlx/backend/metal/kernels/quantized_nax.h:1392
Functionaffine_gather_qmm_rhs
mlx/backend/metal/kernels/quantized.h:2245
Functionaffine_gather_qmm_rhs_nax
mlx/backend/metal/kernels/quantized_nax.h:1461
Functionaffine_gather_qmm_t
mlx/backend/metal/kernels/quantized.h:2093
Functionaffine_gather_qmm_t_nax
mlx/backend/metal/kernels/quantized_nax.h:1324
Functionaffine_gather_qmv
mlx/backend/metal/kernels/quantized.h:1962
Functionaffine_gather_qmv_fast
mlx/backend/metal/kernels/quantized.h:1900
Functionaffine_gather_qvm
mlx/backend/metal/kernels/quantized.h:2024
Functionaffine_qmm_n
mlx/backend/metal/kernels/quantized.h:1847
Functionaffine_qmm_n_nax
mlx/backend/metal/kernels/quantized_nax.h:1264
Functionaffine_qmm_t
mlx/backend/metal/kernels/quantized.h:1716
Functionaffine_qmm_t_nax
mlx/backend/metal/kernels/quantized_nax.h:1205
Functionaffine_qmm_t_splitk
mlx/backend/metal/kernels/quantized.h:1788
Functionaffine_qmv
mlx/backend/metal/kernels/quantized.h:1548
Functionaffine_qmv_fast
mlx/backend/metal/kernels/quantized.h:1496
Functionaffine_qmv_quad
mlx/backend/metal/kernels/quantized.h:1444
Functionaffine_qvm
mlx/backend/metal/kernels/quantized.h:1600
Functionaffine_qvm_split_k
mlx/backend/metal/kernels/quantized.h:1652
Methodaligned_allocator
mlx/3rdparty/pocketfft.h:605
Functionall_gather
mlx/distributed/distributed.cpp:33
Methodall_gather
mlx/distributed/ring/ring.cpp:496
Methodall_gather
mlx/distributed/jaccl/jaccl.cpp:117
Methodall_gather
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:299
Methodall_gather
mlx/distributed/jaccl/lib/jaccl/rdma.h:276
Methodall_gather
mlx/distributed/jaccl/lib/jaccl/mesh_impl.h:195
Methodall_gather
mlx/distributed/nccl/nccl.cpp:346
Methodall_gather_impl
mlx/distributed/ring/ring.cpp:756
Functionall_max
mlx/distributed/distributed.cpp:25
Methodall_max
mlx/distributed/ring/ring.cpp:482
Methodall_max
mlx/distributed/jaccl/jaccl.cpp:91
Methodall_max
mlx/distributed/jaccl/lib/jaccl/ring.cpp:139
Methodall_max
mlx/distributed/jaccl/lib/jaccl/mesh.cpp:152
Methodall_max
mlx/distributed/nccl/nccl.cpp:368
Functionall_min
mlx/distributed/distributed.cpp:29
Methodall_min
mlx/distributed/ring/ring.cpp:487
Methodall_min
mlx/distributed/jaccl/jaccl.cpp:104
Methodall_min
mlx/distributed/jaccl/lib/jaccl/ring.cpp:150
Methodall_min
mlx/distributed/jaccl/lib/jaccl/mesh.cpp:163
Methodall_min
mlx/distributed/nccl/nccl.cpp:375
Functionall_reduce
mlx/backend/metal/kernels/reduction/reduce_all.h:9
Methodall_reduce
mlx/distributed/ring/ring.cpp:584
Methodall_reduce
mlx/distributed/jaccl/lib/jaccl/ring.cpp:194
Methodall_reduce
mlx/distributed/jaccl/lib/jaccl/ring_impl.h:49
Methodall_reduce
mlx/distributed/jaccl/lib/jaccl/mesh.cpp:188
← previousnext →2,301–2,400 of 4,946, ranked by callers