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

Methodoperator()

mlx/backend/cuda/lru_cache.h:179

Methodoperator()

mlx/distributed/reduction_ops.h:7

Methodoperator()

mlx/distributed/reduction_ops.h:18

Methodoperator()

mlx/distributed/reduction_ops.h:29

Methodoperator()

mlx/distributed/jaccl/lib/jaccl/reduction_ops.h:13

Methodoperator()

mlx/distributed/jaccl/lib/jaccl/reduction_ops.h:22

Methodoperator()

mlx/distributed/jaccl/lib/jaccl/reduction_ops.h:31

Methodoperator()

mlx/distributed/jaccl/lib/jaccl/reduction_ops.h:78

Methodoperator()

mlx/distributed/jaccl/lib/jaccl/reduction_ops.h:91

Methodoperator()

mlx/distributed/jaccl/lib/jaccl/reduction_ops.h:104

Functionoperator*

Functionoperator*

mlx/ops.cpp:2877

Functionoperator*

mlx/backend/metal/kernels/complex.h:145

Methodoperator*

mlx/array.cpp:345

Methodoperator*

mlx/3rdparty/pocketfft.h:253

Methodoperator*

mlx/backend/cpu/sort.cpp:52

Functionoperator+

Functionoperator+

mlx/ops.cpp:2845

Functionoperator+

mlx/backend/metal/kernels/complex.h:104

Methodoperator+

mlx/3rdparty/pocketfft.h:256

Methodoperator+

mlx/backend/cpu/sort.cpp:110

Methodoperator+

mlx/distributed/jaccl/lib/jaccl/types.h:112

Methodoperator+

mlx/distributed/jaccl/lib/jaccl/types.h:183

Functionoperator-

Functionoperator-

mlx/ops.cpp:2787

Functionoperator-

mlx/types/complex.h:77

Functionoperator-

mlx/types/fp16.h:172

Functionoperator-

mlx/types/bf16.h:125

Functionoperator-

mlx/backend/cpu/simd/accelerate_simd.h:137

Functionoperator-

mlx/backend/cpu/simd/neon_fp16_simd.h:86

Functionoperator-

mlx/backend/metal/kernels/complex.h:80

Methodoperator-

mlx/3rdparty/pocketfft.h:259

Methodoperator-

mlx/backend/cpu/sort.cpp:85

Functionoperator/

mlx/ops.cpp:2889

Functionoperator/

mlx/backend/metal/kernels/complex.h:149

Functionoperator<

Functionoperator<

mlx/device.h:29

Functionoperator<

mlx/stream.h:19

Functionoperator<

mlx/types/complex.h:73

Functionoperator<

mlx/backend/metal/kernels/complex.h:96

Functionoperator<

mlx/distributed/jaccl/lib/jaccl/types.h:195

Methodoperator<

mlx/backend/cpu/sort.cpp:69

Methodoperator<

mlx/distributed/jaccl/lib/jaccl/types.h:104

Methodoperator<

mlx/distributed/jaccl/lib/jaccl/types.h:175

Functionoperator<<

mlx/ops.cpp:6253

Functionoperator<=

Functionoperator<=

mlx/types/complex.h:69

Functionoperator<=

mlx/backend/metal/kernels/complex.h:92

Methodoperator<=

mlx/backend/cpu/sort.cpp:77

Functionoperator=

Special case so empty lists default to float32. */

Functionoperator==

mlx/array.h:177

Functionoperator==

Functionoperator==

mlx/device.h:34

Functionoperator==

mlx/backend/metal/kernels/complex.h:100

Methodoperator==

mlx/small_vector.h:222

Methodoperator==

mlx/backend/cpu/sort.cpp:61

Methodoperator==

mlx/backend/cuda/lru_cache.h:167

Functionoperator>

Functionoperator>

mlx/types/complex.h:55

Functionoperator>

mlx/backend/metal/kernels/complex.h:88

Functionoperator>

mlx/distributed/jaccl/lib/jaccl/types.h:200

Methodoperator>

mlx/backend/cpu/sort.cpp:73

Methodoperator>

mlx/distributed/jaccl/lib/jaccl/types.h:108

Methodoperator>

mlx/distributed/jaccl/lib/jaccl/types.h:179

Functionoperator>=

Functionoperator>=

mlx/types/complex.h:50

Functionoperator>=

mlx/backend/metal/kernels/complex.h:84

Methodoperator>=

mlx/backend/cpu/sort.cpp:81

Functionoperator>>

mlx/ops.cpp:6270

Methodoperator[]

mlx/backend/cpu/sort.cpp:56

Methodoperator[]

mlx/backend/cpu/simd/accelerate_simd.h:73

Methodoperator[]

mlx/backend/cpu/simd/neon_fp16_simd.h:44

Methodoperator[]

mlx/backend/cpu/simd/base_simd.h:33

Functionoperator^

mlx/ops.cpp:6242

Functionoperator|

mlx/ops.cpp:6235

Functionoperator||

mlx/ops.cpp:2827

Functionoperator||

mlx/backend/cpu/simd/neon_fp16_simd.h:132

Functionoperator||

mlx/backend/metal/kernels/steel/utils/integral_constant.h:74

Functionoperator~

mlx/ops.cpp:6285

Functionoperator~

mlx/backend/cpu/simd/accelerate_simd.h:142

Functionoperator~

mlx/backend/cpu/simd/base_simd.h:145

Functionorthogonal

r"""An initializer that returns an orthogonal matrix. Args: gain (float, optional): Scaling factor for the orthogonal matrix.

python/mlx/nn/init.py:400

mlx/backend/metal/kernels/logging.h:16

Functionout_of_bounds

mlx/backend/metal/kernels/fft/readwrite.h:94

python/tests/test_autograd.py:843

python/tests/test_vmap.py:650

python/tests/test_compile.py:289

python/tests/test_export_import.py:252

python/tests/test_autograd.py:668

Functionouter_fun

tests/compile_tests.cpp:88

Methodoutput_shape

mlx/primitives.cpp:866

Methodoutput_shapes

mlx/primitives.h:934

Methodoutput_shapes

mlx/primitives.h:1535

Methodoutput_shapes

mlx/primitives.cpp:211

Methodoutput_shapes

mlx/fast.cpp:932

Functionoutputs

The outputs of the array's primitive (i.e. this array and * its siblings) in the order the primitive expects. */

mlx/array.h:320

Functionoverwrite_descriptor

mlx/array.h:457

python/tests/test_vmap.py:733

Methodpage_size

mlx/backend/common/buffer_cache.h:105

Functionparse_address

* Parse a sockaddr from an ip and port provided as strings. */

mlx/distributed/utils.cpp:16

← previous next →3,501–3,600 of 4,946, ranked by callers