Functions — github.com/ml-explore/mlx

CodeHub 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

Methodexport_with_callback

mlx/export.cpp:698

Functionextract_bits

mlx/backend/cpu/quantized.cpp:65

Functionextract_bits_simd

mlx/backend/cpu/quantized.cpp:215

python/tests/test_conv.py:208

python/tests/test_fft.py:389

python/tests/test_autograd.py:846

python/tests/test_vmap.py:653

python/tests/test_compile.py:1030

python/tests/test_array.py:493

python/tests/test_conv_transpose.py:181

python/tests/test_export_import.py:255

python/tests/test_export_import.py:364

python/tests/test_export_import.py:388

python/tests/test_export_import.py:412

python/tests/test_quantized.py:931

python/tests/test_blas.py:880

python/tests/test_autograd.py:441

python/tests/test_autograd.py:437

(x, qw, s, b, i1, i2)

python/tests/test_quantized.py:934

python/tests/test_blas.py:883

Methodfactorize

mlx/3rdparty/pocketfft.h:1475

Methodfactorize

mlx/3rdparty/pocketfft.h:2276

Functionfft_mlx

benchmarks/python/fft_bench.py:22

Functionfft_mps

benchmarks/python/fft_bench.py:30

python/tests/test_export_import.py:438

Methodfilter_and_add

python/tests/mlx_tests.py:43

Functionfinalize

mlx/backend/no_gpu/eval.cpp:21

Functionfinalize

mlx/backend/metal/eval.cpp:77

Functionflip_spatial_dims_inplace

mlx/backend/cpu/conv.cpp:837

python/tests/test_conv.py:955

python/tests/test_autograd.py:287

python/tests/test_eval.py:113

python/tests/test_vmap.py:907

python/tests/test_quantized.py:263

python/tests/test_compile.py:723

python/tests/test_export_import.py:491

python/tests/test_ops.py:2162

python/tests/test_conv.py:959

python/tests/test_export_import.py:338

benchmarks/python/conv3d_train_bench_cpu.py:98

benchmarks/python/conv2d_train_bench_cpu.py:98

Functionfour_step_fft

mlx/backend/metal/kernels/fft.h:443

Methodfp32_from_bits

mlx/backend/cpu/unary_ops.h:112

Methodfp32_to_bits

mlx/backend/cpu/unary_ops.h:116

mlx/backend/metal/kernels/fp4.h:4

mlx/backend/metal/kernels/fp8.h:5

mlx/backend/metal/kernels/fp8.h:52

Functionfp_bs_qmm_dispatch_mode

mlx/backend/cpu/quantized.cpp:806

Functionfp_bs_qmm_dispatch_typed

mlx/backend/cpu/quantized.cpp:847

Functionfp_extract_bits_simd

mlx/backend/cpu/quantized.cpp:536

Functionfp_gather_qmm_n

mlx/backend/metal/kernels/fp_quantized.h:1554

Functionfp_gather_qmm_n_nax

mlx/backend/metal/kernels/fp_quantized_nax.h:730

Functionfp_gather_qmm_rhs

mlx/backend/metal/kernels/fp_quantized.h:1621

Functionfp_gather_qmm_rhs_nax

mlx/backend/metal/kernels/fp_quantized_nax.h:798

Functionfp_gather_qmm_t

mlx/backend/metal/kernels/fp_quantized.h:1437

Functionfp_gather_qmm_t_nax

mlx/backend/metal/kernels/fp_quantized_nax.h:665

Functionfp_gather_qmv

mlx/backend/metal/kernels/fp_quantized.h:1332

Functionfp_gather_qmv_fast

mlx/backend/metal/kernels/fp_quantized.h:1283

Functionfp_gather_qvm

mlx/backend/metal/kernels/fp_quantized.h:1381

mlx/backend/cpu/quantized.cpp:450

Functionfp_qmm_dispatch_mode

mlx/backend/cpu/quantized.cpp:625

Functionfp_qmm_dispatch_transpose

mlx/backend/cpu/quantized.cpp:603

Functionfp_qmm_dispatch_typed

mlx/backend/cpu/quantized.cpp:656

Functionfp_qmm_n

mlx/backend/metal/kernels/fp_quantized.h:1234

Functionfp_qmm_n_impl

mlx/backend/metal/kernels/fp_quantized.h:760

Functionfp_qmm_n_impl

mlx/backend/metal/kernels/fp_quantized_nax.h:343

Functionfp_qmm_n_nax

mlx/backend/metal/kernels/fp_quantized_nax.h:606

Functionfp_qmm_t

mlx/backend/cpu/quantized.cpp:492

Functionfp_qmm_t

mlx/backend/metal/kernels/fp_quantized.h:1180

Functionfp_qmm_t_impl

mlx/backend/metal/kernels/fp_quantized.h:635

Functionfp_qmm_t_impl

mlx/backend/metal/kernels/fp_quantized_nax.h:199

Functionfp_qmm_t_nax

mlx/backend/metal/kernels/fp_quantized_nax.h:550

Functionfp_qmm_t_simd

mlx/backend/cpu/quantized.cpp:558

Functionfp_qmm_t_splitk

mlx/backend/metal/kernels/fp_quantized.h:1501

Functionfp_qmv_fast

mlx/backend/metal/kernels/fp_quantized.h:1012

Functionfp_qmv_fast_impl

mlx/backend/metal/kernels/fp_quantized.h:325

Functionfp_qmv_impl

mlx/backend/metal/kernels/fp_quantized.h:386

Functionfp_qmv_quad

mlx/backend/metal/kernels/fp_quantized.h:973

Functionfp_qmv_quad_impl

mlx/backend/metal/kernels/fp_quantized.h:263

Functionfp_quantize_dequantize

mlx/backend/metal/kernels/fp_quantized.h:1879

mlx/backend/metal/kernels/fp_quantized.h:1090

Functionfp_qvm_impl

mlx/backend/metal/kernels/fp_quantized.h:529

Functionfp_qvm_split_k

mlx/backend/metal/kernels/fp_quantized.h:1129

mlx/backend/no_gpu/allocator.cpp:90

mlx/backend/metal/allocator.cpp:172

Methodfree_async

mlx/backend/cuda/allocator.cpp:331

Methodfree_cuda_buffer

This must be called with mutex_ aquired

mlx/backend/cuda/allocator.cpp:322

Methodfree_storage

mlx/small_vector.h:473

python/src/small_vector.h:53

mlx/distributed/jaccl/lib/jaccl/jaccl.cpp:169

Methodfrom_linear

Create a :obj:`QuantizedLinear` layer from a :obj:`Linear` layer.

python/mlx/nn/layers/quantized.py:281

Methodfrom_linear

( cls, linear_layer: Module, *, segments: Union[int, list] = 1, group:

python/mlx/nn/layers/distributed.py:258

Methodfrom_linear

( cls, linear_layer: Module, *, segments: Union[int, list] = 1, group:

python/mlx/nn/layers/distributed.py:341

Methodfrom_python

python/src/small_vector.h:21

Methodfrom_quantized_linear

( cls, quantized_linear_layer: Module, *, segments: Union[int, list] = 1,

python/mlx/nn/layers/distributed.py:591

python/tests/test_autograd.py:165

python/tests/test_eval.py:18

python/tests/test_vmap.py:581

python/tests/test_compile.py:17

python/tests/test_array.py:1967

← previous next →2,801–2,900 of 4,946, ranked by callers