From 36427f6126a1405d87da8b665731f82eaaab5cf0 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Sat, 10 Aug 2024 09:24:35 -0700 Subject: [PATCH] docs update --- docs/build/html/.buildinfo | 2 +- docs/build/html/_sources/dev/extensions.rst | 5 +- .../_sources/examples/llama-inference.rst | 2 +- docs/build/html/_sources/examples/mlp.rst | 2 +- docs/build/html/_sources/install.rst | 26 +- .../_autosummary/mlx.core.DtypeCategory.rst | 19 +- .../mlx.core.linalg.cholesky_inv.rst | 6 + .../_autosummary/mlx.core.linalg.tri_inv.rst | 6 + docs/build/html/_sources/python/linalg.rst | 2 + .../html/_static/documentation_options.js | 2 +- docs/build/html/annotated.html | 299 ++- docs/build/html/array_8h_source.html | 32 +- .../backend_2accelerate_2utils_8h_source.html | 4 +- .../build/html/backend_2metal_2device_8h.html | 4 - .../backend_2metal_2device_8h_source.html | 414 ++-- docs/build/html/backend_2metal_2utils_8h.html | 26 +- .../html/backend_2metal_2utils_8h_source.html | 226 +-- .../html/class_m_p_s_1_1_kernel-members.html | 92 - docs/build/html/class_m_p_s_1_1_kernel.html | 144 -- docs/build/html/class_m_p_s_1_1_kernel.png | Bin 656 -> 0 bytes .../html/class_m_p_s_1_1_matrix-members.html | 93 - docs/build/html/class_m_p_s_1_1_matrix.html | 183 -- docs/build/html/class_m_p_s_1_1_matrix.png | Bin 665 -> 0 bytes ...s_m_p_s_1_1_matrix_descriptor-members.html | 93 - .../class_m_p_s_1_1_matrix_descriptor.html | 221 --- .../class_m_p_s_1_1_matrix_descriptor.png | Bin 813 -> 0 bytes ...p_s_1_1_matrix_multiplication-members.html | 98 - ...class_m_p_s_1_1_matrix_multiplication.html | 318 --- .../class_m_p_s_1_1_matrix_multiplication.png | Bin 1041 -> 0 bytes ..._matrix_vector_multiplication-members.html | 93 - ..._p_s_1_1_matrix_vector_multiplication.html | 213 -- ...m_p_s_1_1_matrix_vector_multiplication.png | Bin 1145 -> 0 bytes .../html/class_m_p_s_1_1_vector-members.html | 93 - docs/build/html/class_m_p_s_1_1_vector.html | 183 -- docs/build/html/class_m_p_s_1_1_vector.png | Bin 658 -> 0 bytes ...s_m_p_s_1_1_vector_descriptor-members.html | 92 - .../class_m_p_s_1_1_vector_descriptor.html | 178 -- .../class_m_p_s_1_1_vector_descriptor.png | Bin 794 -> 0 bytes docs/build/html/classes.html | 10 +- .../classmlx_1_1core_1_1_inverse-members.html | 2 +- .../html/classmlx_1_1core_1_1_inverse.html | 19 +- .../html/classmlx_1_1core_1_1_primitive.html | 6 +- .../classmlx_1_1core_1_1_scatter-members.html | 2 +- .../html/classmlx_1_1core_1_1_scatter.html | 40 +- ...e_1_1fast_1_1_affine_quantize-members.html | 109 ++ ...x_1_1core_1_1fast_1_1_affine_quantize.html | 286 +++ ...lx_1_1core_1_1fast_1_1_affine_quantize.png | Bin 0 -> 1003 bytes .../classmlx_1_1core_1_1fast_1_1_custom.html | 15 +- .../classmlx_1_1core_1_1fast_1_1_custom.png | Bin 2664 -> 3025 bytes ...x_1_1core_1_1metal_1_1_device-members.html | 2 +- .../classmlx_1_1core_1_1metal_1_1_device.html | 14 +- docs/build/html/common_2arange_8h_source.html | 26 +- docs/build/html/common_2binary_8h_source.html | 26 +- .../html/common_2binary__two_8h_source.html | 26 +- .../html/common_2hadamard_8h_source.html | 2 +- docs/build/html/common_2unary_8h_source.html | 26 +- docs/build/html/compiled_8h_source.html | 4 +- docs/build/html/cpp/ops.html | 42 +- docs/build/html/dev/extensions.html | 15 +- docs/build/html/dev/metal_debugger.html | 10 +- .../dir_4336740ec0075891704443b417fef6cb.html | 2 + .../dir_6768c99e6145fb9510ccdb40db8ede25.html | 2 +- .../dir_70a37effa88bcbd6b791977fa1e64356.html | 2 + .../dir_d0c977ea65824390717cdb7efc36c157.html | 2 - docs/build/html/doxygen_crawl.html | 41 +- docs/build/html/dtype_8h.html | 6 - docs/build/html/dtype_8h_source.html | 269 ++- .../html/examples/linear_regression.html | 10 +- docs/build/html/examples/llama-inference.html | 12 +- docs/build/html/examples/mlp.html | 12 +- docs/build/html/fast_8h.html | 6 + docs/build/html/fast_8h_source.html | 26 +- docs/build/html/fast__primitives_8h.html | 2 + .../html/fast__primitives_8h_source.html | 44 +- docs/build/html/files.html | 146 +- docs/build/html/functions_a.html | 2 +- docs/build/html/functions_b.html | 2 + docs/build/html/functions_d.html | 4 +- docs/build/html/functions_e.html | 5 +- docs/build/html/functions_func_a.html | 2 +- docs/build/html/functions_func_d.html | 4 +- docs/build/html/functions_func_e.html | 5 +- docs/build/html/functions_func_i.html | 3 +- docs/build/html/functions_func_l.html | 6 +- docs/build/html/functions_func_m.html | 1 - docs/build/html/functions_func_o.html | 6 +- docs/build/html/functions_func_r.html | 3 +- docs/build/html/functions_func_s.html | 5 - docs/build/html/functions_func_v.html | 3 +- docs/build/html/functions_h.html | 4 + docs/build/html/functions_i.html | 4 +- docs/build/html/functions_l.html | 8 +- docs/build/html/functions_m.html | 1 - docs/build/html/functions_n.html | 1 + docs/build/html/functions_o.html | 4 +- docs/build/html/functions_r.html | 3 +- docs/build/html/functions_s.html | 5 - docs/build/html/functions_t.html | 8 +- docs/build/html/functions_v.html | 3 +- docs/build/html/functions_vars_b.html | 2 + ...a10604f9f12.html => functions_vars_h.html} | 25 +- docs/build/html/functions_vars_n.html | 1 + docs/build/html/functions_vars_t.html | 8 +- ...steel_2gemm_2gemm_8h.html => gemm_8h.html} | 2 +- ...emm_8h_source.html => gemm_8h_source.html} | 2 +- docs/build/html/genindex.html | 20 +- docs/build/html/globals.html | 2 - docs/build/html/globals_a.html | 3 + docs/build/html/globals_b.html | 3 + docs/build/html/globals_c.html | 2 + docs/build/html/globals_defs.html | 6 +- docs/build/html/globals_func_a.html | 3 + docs/build/html/globals_func_b.html | 3 + docs/build/html/globals_func_c.html | 2 + docs/build/html/globals_func_g.html | 2 + docs/build/html/globals_func_t.html | 1 + docs/build/html/globals_func_u.html | 1 + docs/build/html/globals_g.html | 3 + docs/build/html/globals_m.html | 4 +- docs/build/html/globals_n.html | 2 +- docs/build/html/globals_t.html | 1 + docs/build/html/globals_type.html | 2 +- docs/build/html/globals_u.html | 1 + docs/build/html/globals_vars.html | 1 + docs/build/html/group__ops.html | 58 +- docs/build/html/hierarchy.html | 482 +++-- docs/build/html/includes_8h.html | 2 + docs/build/html/includes_8h_source.html | 6 +- docs/build/html/index.html | 10 +- docs/build/html/install.html | 36 +- docs/build/html/jit_2gemv__masked_8h.html | 143 ++ .../html/jit_2gemv__masked_8h_source.html | 118 ++ docs/build/html/kernels_2gemv__masked_8h.html | 396 ++++ .../html/kernels_2gemv__masked_8h_source.html | 972 +++++++++ .../html/kernels_2softmax_8h_source.html | 8 +- docs/build/html/kernels_8h.html | 2 + docs/build/html/kernels_8h_source.html | 110 +- docs/build/html/linalg_8h.html | 4 + docs/build/html/linalg_8h_source.html | 10 +- docs/build/html/matmul_8h.html | 9 +- docs/build/html/matmul_8h_source.html | 94 +- docs/build/html/menudata.js | 7 +- docs/build/html/metal_2binary_8h.html | 16 +- docs/build/html/metal_2binary_8h_source.html | 20 +- .../build/html/metal_2kernels_2binary_8h.html | 123 ++ .../metal_2kernels_2binary_8h_source.html | 202 +- .../html/metal_2kernels_2binary__two_8h.html | 138 ++ ...metal_2kernels_2binary__two_8h_source.html | 239 ++- docs/build/html/metal_2kernels_2copy_8h.html | 72 + .../html/metal_2kernels_2copy_8h_source.html | 298 +-- .../html/metal_2kernels_2ternary_8h.html | 46 + .../metal_2kernels_2ternary_8h_source.html | 203 +- docs/build/html/metal_2kernels_2unary_8h.html | 36 + .../html/metal_2kernels_2unary_8h_source.html | 31 +- docs/build/html/metal_2reduce_8h_source.html | 4 +- docs/build/html/mps_2gemm_8h.html | 235 --- docs/build/html/mps_2gemm_8h_source.html | 572 ------ docs/build/html/namespace_m_p_s.html | 160 -- docs/build/html/namespace_m_t_l.html | 91 - .../html/namespace_m_t_l_1_1_private.html | 97 - ...namespace_m_t_l_1_1_private_1_1_class.html | 227 --- ...espace_m_t_l_1_1_private_1_1_selector.html | 439 ----- docs/build/html/namespacemembers.html | 46 +- docs/build/html/namespacemembers_a.html | 124 -- docs/build/html/namespacemembers_b.html | 4 +- docs/build/html/namespacemembers_c.html | 1 + docs/build/html/namespacemembers_d.html | 10 +- docs/build/html/namespacemembers_enum.html | 1 - docs/build/html/namespacemembers_eval.html | 5 - docs/build/html/namespacemembers_func.html | 45 +- docs/build/html/namespacemembers_func_a.html | 123 -- docs/build/html/namespacemembers_func_b.html | 4 +- docs/build/html/namespacemembers_func_c.html | 1 + docs/build/html/namespacemembers_func_d.html | 4 +- docs/build/html/namespacemembers_func_g.html | 6 +- docs/build/html/namespacemembers_func_i.html | 1 - docs/build/html/namespacemembers_func_m.html | 1 + docs/build/html/namespacemembers_func_p.html | 2 +- docs/build/html/namespacemembers_func_s.html | 1 + docs/build/html/namespacemembers_func_t.html | 2 + docs/build/html/namespacemembers_g.html | 6 +- docs/build/html/namespacemembers_i.html | 1 - docs/build/html/namespacemembers_m.html | 1 + docs/build/html/namespacemembers_p.html | 2 +- docs/build/html/namespacemembers_s.html | 1 + docs/build/html/namespacemembers_t.html | 2 + docs/build/html/namespacemembers_type.html | 1 - docs/build/html/namespacemlx_1_1core.html | 464 ++++- .../html/namespacemlx_1_1core_1_1fast.html | 121 ++ .../html/namespacemlx_1_1core_1_1linalg.html | 56 + .../html/namespacemlx_1_1core_1_1metal.html | 46 +- docs/build/html/namespaces.html | 95 +- docs/build/html/objects.inv | Bin 25204 -> 25261 bytes docs/build/html/ops_8h.html | 20 +- docs/build/html/ops_8h_source.html | 1740 +++++++++-------- docs/build/html/primitives_8h_source.html | 1028 +++++----- .../python/_autosummary/mlx.core.Device.html | 10 +- .../python/_autosummary/mlx.core.Dtype.html | 10 +- .../_autosummary/mlx.core.DtypeCategory.html | 41 +- .../python/_autosummary/mlx.core.Stream.html | 10 +- .../python/_autosummary/mlx.core.abs.html | 10 +- .../python/_autosummary/mlx.core.add.html | 10 +- .../python/_autosummary/mlx.core.addmm.html | 10 +- .../python/_autosummary/mlx.core.all.html | 10 +- .../_autosummary/mlx.core.allclose.html | 10 +- .../python/_autosummary/mlx.core.any.html | 10 +- .../python/_autosummary/mlx.core.arange.html | 10 +- .../python/_autosummary/mlx.core.arccos.html | 10 +- .../python/_autosummary/mlx.core.arccosh.html | 10 +- .../python/_autosummary/mlx.core.arcsin.html | 10 +- .../python/_autosummary/mlx.core.arcsinh.html | 10 +- .../python/_autosummary/mlx.core.arctan.html | 10 +- .../python/_autosummary/mlx.core.arctan2.html | 10 +- .../python/_autosummary/mlx.core.arctanh.html | 10 +- .../python/_autosummary/mlx.core.argmax.html | 10 +- .../python/_autosummary/mlx.core.argmin.html | 10 +- .../_autosummary/mlx.core.argpartition.html | 10 +- .../python/_autosummary/mlx.core.argsort.html | 10 +- .../python/_autosummary/mlx.core.array.T.html | 10 +- .../_autosummary/mlx.core.array.abs.html | 10 +- .../_autosummary/mlx.core.array.all.html | 10 +- .../_autosummary/mlx.core.array.any.html | 10 +- .../_autosummary/mlx.core.array.argmax.html | 10 +- .../_autosummary/mlx.core.array.argmin.html | 10 +- .../_autosummary/mlx.core.array.astype.html | 10 +- .../_autosummary/mlx.core.array.at.html | 10 +- .../_autosummary/mlx.core.array.conj.html | 10 +- .../_autosummary/mlx.core.array.cos.html | 10 +- .../_autosummary/mlx.core.array.cummax.html | 10 +- .../_autosummary/mlx.core.array.cummin.html | 10 +- .../_autosummary/mlx.core.array.cumprod.html | 10 +- .../_autosummary/mlx.core.array.cumsum.html | 10 +- .../_autosummary/mlx.core.array.diag.html | 10 +- .../_autosummary/mlx.core.array.diagonal.html | 10 +- .../_autosummary/mlx.core.array.dtype.html | 10 +- .../_autosummary/mlx.core.array.exp.html | 10 +- .../_autosummary/mlx.core.array.flatten.html | 10 +- .../python/_autosummary/mlx.core.array.html | 10 +- .../_autosummary/mlx.core.array.item.html | 10 +- .../_autosummary/mlx.core.array.itemsize.html | 10 +- .../_autosummary/mlx.core.array.log.html | 10 +- .../_autosummary/mlx.core.array.log10.html | 10 +- .../_autosummary/mlx.core.array.log1p.html | 10 +- .../_autosummary/mlx.core.array.log2.html | 10 +- .../mlx.core.array.logsumexp.html | 10 +- .../_autosummary/mlx.core.array.max.html | 10 +- .../_autosummary/mlx.core.array.mean.html | 10 +- .../_autosummary/mlx.core.array.min.html | 10 +- .../_autosummary/mlx.core.array.moveaxis.html | 10 +- .../_autosummary/mlx.core.array.nbytes.html | 10 +- .../_autosummary/mlx.core.array.ndim.html | 10 +- .../_autosummary/mlx.core.array.prod.html | 10 +- .../mlx.core.array.reciprocal.html | 10 +- .../_autosummary/mlx.core.array.reshape.html | 10 +- .../_autosummary/mlx.core.array.round.html | 10 +- .../_autosummary/mlx.core.array.rsqrt.html | 10 +- .../_autosummary/mlx.core.array.shape.html | 10 +- .../_autosummary/mlx.core.array.sin.html | 10 +- .../_autosummary/mlx.core.array.size.html | 10 +- .../_autosummary/mlx.core.array.split.html | 10 +- .../_autosummary/mlx.core.array.sqrt.html | 10 +- .../_autosummary/mlx.core.array.square.html | 10 +- .../_autosummary/mlx.core.array.squeeze.html | 10 +- .../_autosummary/mlx.core.array.sum.html | 10 +- .../_autosummary/mlx.core.array.swapaxes.html | 10 +- .../_autosummary/mlx.core.array.tolist.html | 10 +- .../mlx.core.array.transpose.html | 10 +- .../_autosummary/mlx.core.array.var.html | 10 +- .../_autosummary/mlx.core.array.view.html | 10 +- .../_autosummary/mlx.core.array_equal.html | 10 +- .../_autosummary/mlx.core.as_strided.html | 10 +- .../_autosummary/mlx.core.atleast_1d.html | 10 +- .../_autosummary/mlx.core.atleast_2d.html | 10 +- .../_autosummary/mlx.core.atleast_3d.html | 10 +- .../_autosummary/mlx.core.bitwise_and.html | 10 +- .../_autosummary/mlx.core.bitwise_or.html | 10 +- .../_autosummary/mlx.core.bitwise_xor.html | 10 +- .../mlx.core.block_masked_mm.html | 10 +- .../_autosummary/mlx.core.broadcast_to.html | 10 +- .../python/_autosummary/mlx.core.ceil.html | 10 +- .../python/_autosummary/mlx.core.clip.html | 10 +- .../python/_autosummary/mlx.core.compile.html | 10 +- .../_autosummary/mlx.core.concatenate.html | 10 +- .../python/_autosummary/mlx.core.conj.html | 10 +- .../_autosummary/mlx.core.conjugate.html | 10 +- .../python/_autosummary/mlx.core.conv1d.html | 10 +- .../python/_autosummary/mlx.core.conv2d.html | 10 +- .../_autosummary/mlx.core.conv_general.html | 10 +- .../_autosummary/mlx.core.convolve.html | 10 +- .../python/_autosummary/mlx.core.cos.html | 10 +- .../python/_autosummary/mlx.core.cosh.html | 10 +- .../python/_autosummary/mlx.core.cummax.html | 10 +- .../python/_autosummary/mlx.core.cummin.html | 10 +- .../python/_autosummary/mlx.core.cumprod.html | 10 +- .../python/_autosummary/mlx.core.cumsum.html | 10 +- .../mlx.core.custom_function.html | 10 +- .../_autosummary/mlx.core.default_device.html | 10 +- .../_autosummary/mlx.core.default_stream.html | 10 +- .../python/_autosummary/mlx.core.degrees.html | 10 +- .../_autosummary/mlx.core.dequantize.html | 10 +- .../python/_autosummary/mlx.core.diag.html | 10 +- .../_autosummary/mlx.core.diagonal.html | 10 +- .../mlx.core.disable_compile.html | 10 +- .../mlx.core.distributed.Group.html | 10 +- .../mlx.core.distributed.all_gather.html | 10 +- .../mlx.core.distributed.all_sum.html | 10 +- .../mlx.core.distributed.init.html | 10 +- .../mlx.core.distributed.is_available.html | 10 +- .../python/_autosummary/mlx.core.divide.html | 10 +- .../python/_autosummary/mlx.core.divmod.html | 10 +- .../python/_autosummary/mlx.core.einsum.html | 10 +- .../_autosummary/mlx.core.einsum_path.html | 10 +- .../_autosummary/mlx.core.enable_compile.html | 10 +- .../python/_autosummary/mlx.core.equal.html | 10 +- .../python/_autosummary/mlx.core.erf.html | 10 +- .../python/_autosummary/mlx.core.erfinv.html | 10 +- .../python/_autosummary/mlx.core.eval.html | 10 +- .../python/_autosummary/mlx.core.exp.html | 10 +- .../_autosummary/mlx.core.expand_dims.html | 10 +- .../python/_autosummary/mlx.core.expm1.html | 10 +- .../python/_autosummary/mlx.core.eye.html | 10 +- .../mlx.core.fast.layer_norm.html | 10 +- .../_autosummary/mlx.core.fast.rms_norm.html | 10 +- .../_autosummary/mlx.core.fast.rope.html | 10 +- ...ore.fast.scaled_dot_product_attention.html | 10 +- .../python/_autosummary/mlx.core.fft.fft.html | 10 +- .../_autosummary/mlx.core.fft.fft2.html | 10 +- .../_autosummary/mlx.core.fft.fftn.html | 10 +- .../_autosummary/mlx.core.fft.ifft.html | 10 +- .../_autosummary/mlx.core.fft.ifft2.html | 10 +- .../_autosummary/mlx.core.fft.ifftn.html | 10 +- .../_autosummary/mlx.core.fft.irfft.html | 10 +- .../_autosummary/mlx.core.fft.irfft2.html | 10 +- .../_autosummary/mlx.core.fft.irfftn.html | 10 +- .../_autosummary/mlx.core.fft.rfft.html | 10 +- .../_autosummary/mlx.core.fft.rfft2.html | 10 +- .../_autosummary/mlx.core.fft.rfftn.html | 10 +- .../python/_autosummary/mlx.core.flatten.html | 10 +- .../python/_autosummary/mlx.core.floor.html | 10 +- .../_autosummary/mlx.core.floor_divide.html | 10 +- .../python/_autosummary/mlx.core.full.html | 10 +- .../_autosummary/mlx.core.gather_mm.html | 10 +- .../_autosummary/mlx.core.gather_qmm.html | 10 +- .../python/_autosummary/mlx.core.grad.html | 10 +- .../python/_autosummary/mlx.core.greater.html | 10 +- .../_autosummary/mlx.core.greater_equal.html | 10 +- .../mlx.core.hadamard_transform.html | 10 +- .../_autosummary/mlx.core.identity.html | 10 +- .../python/_autosummary/mlx.core.inner.html | 10 +- .../python/_autosummary/mlx.core.isclose.html | 10 +- .../python/_autosummary/mlx.core.isinf.html | 10 +- .../python/_autosummary/mlx.core.isnan.html | 10 +- .../_autosummary/mlx.core.isneginf.html | 10 +- .../_autosummary/mlx.core.isposinf.html | 10 +- .../_autosummary/mlx.core.issubdtype.html | 10 +- .../python/_autosummary/mlx.core.jvp.html | 10 +- .../_autosummary/mlx.core.left_shift.html | 10 +- .../python/_autosummary/mlx.core.less.html | 10 +- .../_autosummary/mlx.core.less_equal.html | 10 +- .../mlx.core.linalg.cholesky.html | 18 +- .../mlx.core.linalg.cholesky_inv.html | 984 ++++++++++ .../_autosummary/mlx.core.linalg.inv.html | 16 +- .../_autosummary/mlx.core.linalg.norm.html | 16 +- .../_autosummary/mlx.core.linalg.qr.html | 16 +- .../_autosummary/mlx.core.linalg.svd.html | 10 +- .../_autosummary/mlx.core.linalg.tri_inv.html | 975 +++++++++ .../_autosummary/mlx.core.linspace.html | 10 +- .../python/_autosummary/mlx.core.load.html | 10 +- .../python/_autosummary/mlx.core.log.html | 10 +- .../python/_autosummary/mlx.core.log10.html | 10 +- .../python/_autosummary/mlx.core.log1p.html | 10 +- .../python/_autosummary/mlx.core.log2.html | 10 +- .../_autosummary/mlx.core.logaddexp.html | 10 +- .../_autosummary/mlx.core.logical_and.html | 10 +- .../_autosummary/mlx.core.logical_not.html | 10 +- .../_autosummary/mlx.core.logical_or.html | 10 +- .../_autosummary/mlx.core.logsumexp.html | 10 +- .../python/_autosummary/mlx.core.matmul.html | 10 +- .../python/_autosummary/mlx.core.max.html | 10 +- .../python/_autosummary/mlx.core.maximum.html | 10 +- .../python/_autosummary/mlx.core.mean.html | 10 +- .../_autosummary/mlx.core.meshgrid.html | 10 +- .../mlx.core.metal.clear_cache.html | 10 +- .../mlx.core.metal.device_info.html | 10 +- .../mlx.core.metal.get_active_memory.html | 10 +- .../mlx.core.metal.get_cache_memory.html | 10 +- .../mlx.core.metal.get_peak_memory.html | 10 +- .../mlx.core.metal.is_available.html | 10 +- .../mlx.core.metal.reset_peak_memory.html | 10 +- .../mlx.core.metal.set_cache_limit.html | 10 +- .../mlx.core.metal.set_memory_limit.html | 10 +- .../mlx.core.metal.start_capture.html | 10 +- .../mlx.core.metal.stop_capture.html | 10 +- .../python/_autosummary/mlx.core.min.html | 10 +- .../python/_autosummary/mlx.core.minimum.html | 10 +- .../_autosummary/mlx.core.moveaxis.html | 10 +- .../_autosummary/mlx.core.multiply.html | 10 +- .../_autosummary/mlx.core.nan_to_num.html | 10 +- .../_autosummary/mlx.core.negative.html | 10 +- .../_autosummary/mlx.core.new_stream.html | 10 +- .../_autosummary/mlx.core.not_equal.html | 10 +- .../python/_autosummary/mlx.core.ones.html | 10 +- .../_autosummary/mlx.core.ones_like.html | 10 +- .../python/_autosummary/mlx.core.outer.html | 10 +- .../python/_autosummary/mlx.core.pad.html | 15 +- .../_autosummary/mlx.core.partition.html | 10 +- .../python/_autosummary/mlx.core.power.html | 10 +- .../python/_autosummary/mlx.core.prod.html | 10 +- .../_autosummary/mlx.core.quantize.html | 10 +- .../mlx.core.quantized_matmul.html | 10 +- .../python/_autosummary/mlx.core.radians.html | 10 +- .../mlx.core.random.bernoulli.html | 10 +- .../mlx.core.random.categorical.html | 10 +- .../_autosummary/mlx.core.random.gumbel.html | 10 +- .../_autosummary/mlx.core.random.key.html | 10 +- .../_autosummary/mlx.core.random.laplace.html | 10 +- .../mlx.core.random.multivariate_normal.html | 10 +- .../_autosummary/mlx.core.random.normal.html | 10 +- .../_autosummary/mlx.core.random.randint.html | 10 +- .../_autosummary/mlx.core.random.seed.html | 10 +- .../_autosummary/mlx.core.random.split.html | 10 +- .../mlx.core.random.truncated_normal.html | 10 +- .../_autosummary/mlx.core.random.uniform.html | 10 +- .../_autosummary/mlx.core.reciprocal.html | 10 +- .../_autosummary/mlx.core.remainder.html | 10 +- .../python/_autosummary/mlx.core.repeat.html | 10 +- .../python/_autosummary/mlx.core.reshape.html | 10 +- .../_autosummary/mlx.core.right_shift.html | 10 +- .../python/_autosummary/mlx.core.round.html | 10 +- .../python/_autosummary/mlx.core.rsqrt.html | 10 +- .../python/_autosummary/mlx.core.save.html | 10 +- .../_autosummary/mlx.core.save_gguf.html | 10 +- .../mlx.core.save_safetensors.html | 10 +- .../python/_autosummary/mlx.core.savez.html | 10 +- .../mlx.core.savez_compressed.html | 10 +- .../mlx.core.set_default_device.html | 10 +- .../mlx.core.set_default_stream.html | 10 +- .../python/_autosummary/mlx.core.sigmoid.html | 10 +- .../python/_autosummary/mlx.core.sign.html | 10 +- .../python/_autosummary/mlx.core.sin.html | 10 +- .../python/_autosummary/mlx.core.sinh.html | 10 +- .../python/_autosummary/mlx.core.softmax.html | 10 +- .../python/_autosummary/mlx.core.sort.html | 10 +- .../python/_autosummary/mlx.core.split.html | 10 +- .../python/_autosummary/mlx.core.sqrt.html | 10 +- .../python/_autosummary/mlx.core.square.html | 10 +- .../python/_autosummary/mlx.core.squeeze.html | 10 +- .../python/_autosummary/mlx.core.stack.html | 10 +- .../python/_autosummary/mlx.core.std.html | 10 +- .../_autosummary/mlx.core.stop_gradient.html | 10 +- .../_autosummary/mlx.core.subtract.html | 10 +- .../python/_autosummary/mlx.core.sum.html | 10 +- .../_autosummary/mlx.core.swapaxes.html | 10 +- .../_autosummary/mlx.core.synchronize.html | 10 +- .../python/_autosummary/mlx.core.take.html | 10 +- .../mlx.core.take_along_axis.html | 10 +- .../python/_autosummary/mlx.core.tan.html | 10 +- .../python/_autosummary/mlx.core.tanh.html | 10 +- .../_autosummary/mlx.core.tensordot.html | 10 +- .../python/_autosummary/mlx.core.tile.html | 10 +- .../python/_autosummary/mlx.core.topk.html | 10 +- .../python/_autosummary/mlx.core.trace.html | 10 +- .../_autosummary/mlx.core.transpose.html | 10 +- .../python/_autosummary/mlx.core.tri.html | 10 +- .../python/_autosummary/mlx.core.tril.html | 10 +- .../python/_autosummary/mlx.core.triu.html | 10 +- .../_autosummary/mlx.core.value_and_grad.html | 10 +- .../python/_autosummary/mlx.core.var.html | 10 +- .../python/_autosummary/mlx.core.view.html | 10 +- .../python/_autosummary/mlx.core.vjp.html | 10 +- .../python/_autosummary/mlx.core.vmap.html | 10 +- .../python/_autosummary/mlx.core.where.html | 10 +- .../python/_autosummary/mlx.core.zeros.html | 10 +- .../_autosummary/mlx.core.zeros_like.html | 10 +- .../python/_autosummary/mlx.nn.quantize.html | 10 +- .../_autosummary/mlx.nn.value_and_grad.html | 10 +- .../mlx.optimizers.clip_grad_norm.html | 10 +- .../_autosummary/mlx.utils.tree_flatten.html | 10 +- .../_autosummary/mlx.utils.tree_map.html | 10 +- .../mlx.utils.tree_map_with_path.html | 10 +- .../_autosummary/mlx.utils.tree_reduce.html | 10 +- .../mlx.utils.tree_unflatten.html | 10 +- .../python/_autosummary/stream_class.html | 10 +- docs/build/html/python/array.html | 10 +- docs/build/html/python/data_types.html | 12 +- .../html/python/devices_and_streams.html | 10 +- docs/build/html/python/distributed.html | 10 +- docs/build/html/python/fast.html | 10 +- docs/build/html/python/fft.html | 10 +- docs/build/html/python/linalg.html | 20 +- docs/build/html/python/metal.html | 10 +- docs/build/html/python/nn.html | 10 +- .../python/nn/_autosummary/mlx.nn.ALiBi.html | 10 +- .../nn/_autosummary/mlx.nn.AvgPool1d.html | 10 +- .../nn/_autosummary/mlx.nn.AvgPool2d.html | 10 +- .../nn/_autosummary/mlx.nn.BatchNorm.html | 10 +- .../python/nn/_autosummary/mlx.nn.Conv1d.html | 10 +- .../python/nn/_autosummary/mlx.nn.Conv2d.html | 10 +- .../python/nn/_autosummary/mlx.nn.Conv3d.html | 10 +- .../nn/_autosummary/mlx.nn.Dropout.html | 10 +- .../nn/_autosummary/mlx.nn.Dropout2d.html | 10 +- .../nn/_autosummary/mlx.nn.Dropout3d.html | 10 +- .../nn/_autosummary/mlx.nn.Embedding.html | 10 +- .../python/nn/_autosummary/mlx.nn.GELU.html | 17 +- .../python/nn/_autosummary/mlx.nn.GLU.html | 10 +- .../python/nn/_autosummary/mlx.nn.GRU.html | 10 +- .../nn/_autosummary/mlx.nn.GroupNorm.html | 10 +- .../nn/_autosummary/mlx.nn.HardShrink.html | 10 +- .../nn/_autosummary/mlx.nn.HardTanh.html | 10 +- .../nn/_autosummary/mlx.nn.Hardswish.html | 10 +- .../nn/_autosummary/mlx.nn.InstanceNorm.html | 10 +- .../python/nn/_autosummary/mlx.nn.LSTM.html | 10 +- .../nn/_autosummary/mlx.nn.LayerNorm.html | 10 +- .../nn/_autosummary/mlx.nn.LeakyReLU.html | 10 +- .../python/nn/_autosummary/mlx.nn.Linear.html | 10 +- .../nn/_autosummary/mlx.nn.MaxPool1d.html | 10 +- .../nn/_autosummary/mlx.nn.MaxPool2d.html | 10 +- .../python/nn/_autosummary/mlx.nn.Mish.html | 10 +- .../nn/_autosummary/mlx.nn.Module.apply.html | 10 +- .../mlx.nn.Module.apply_to_modules.html | 10 +- .../_autosummary/mlx.nn.Module.children.html | 10 +- .../nn/_autosummary/mlx.nn.Module.eval.html | 10 +- .../mlx.nn.Module.filter_and_map.html | 10 +- .../nn/_autosummary/mlx.nn.Module.freeze.html | 10 +- .../mlx.nn.Module.leaf_modules.html | 10 +- .../mlx.nn.Module.load_weights.html | 10 +- .../_autosummary/mlx.nn.Module.modules.html | 10 +- .../mlx.nn.Module.named_modules.html | 10 +- .../mlx.nn.Module.parameters.html | 10 +- .../mlx.nn.Module.save_weights.html | 10 +- .../_autosummary/mlx.nn.Module.set_dtype.html | 10 +- .../nn/_autosummary/mlx.nn.Module.state.html | 10 +- .../nn/_autosummary/mlx.nn.Module.train.html | 10 +- .../mlx.nn.Module.trainable_parameters.html | 10 +- .../_autosummary/mlx.nn.Module.training.html | 10 +- .../_autosummary/mlx.nn.Module.unfreeze.html | 10 +- .../nn/_autosummary/mlx.nn.Module.update.html | 10 +- .../mlx.nn.Module.update_modules.html | 10 +- .../mlx.nn.MultiHeadAttention.html | 10 +- .../python/nn/_autosummary/mlx.nn.PReLU.html | 10 +- .../mlx.nn.QuantizedEmbedding.html | 10 +- .../_autosummary/mlx.nn.QuantizedLinear.html | 10 +- .../nn/_autosummary/mlx.nn.RMSNorm.html | 10 +- .../python/nn/_autosummary/mlx.nn.RNN.html | 10 +- .../python/nn/_autosummary/mlx.nn.ReLU.html | 10 +- .../python/nn/_autosummary/mlx.nn.ReLU6.html | 10 +- .../python/nn/_autosummary/mlx.nn.RoPE.html | 10 +- .../python/nn/_autosummary/mlx.nn.SELU.html | 10 +- .../nn/_autosummary/mlx.nn.Sequential.html | 10 +- .../python/nn/_autosummary/mlx.nn.SiLU.html | 10 +- .../mlx.nn.SinusoidalPositionalEncoding.html | 10 +- .../nn/_autosummary/mlx.nn.Softmax.html | 10 +- .../nn/_autosummary/mlx.nn.Softmin.html | 10 +- .../nn/_autosummary/mlx.nn.Softplus.html | 10 +- .../nn/_autosummary/mlx.nn.Softshrink.html | 10 +- .../nn/_autosummary/mlx.nn.Softsign.html | 10 +- .../python/nn/_autosummary/mlx.nn.Step.html | 10 +- .../python/nn/_autosummary/mlx.nn.Tanh.html | 10 +- .../nn/_autosummary/mlx.nn.Transformer.html | 10 +- .../nn/_autosummary/mlx.nn.Upsample.html | 10 +- .../nn/_autosummary/mlx.nn.init.constant.html | 10 +- .../mlx.nn.init.glorot_normal.html | 10 +- .../mlx.nn.init.glorot_uniform.html | 10 +- .../_autosummary/mlx.nn.init.he_normal.html | 10 +- .../_autosummary/mlx.nn.init.he_uniform.html | 10 +- .../nn/_autosummary/mlx.nn.init.identity.html | 10 +- .../nn/_autosummary/mlx.nn.init.normal.html | 10 +- .../nn/_autosummary/mlx.nn.init.uniform.html | 10 +- .../nn/_autosummary_functions/mlx.nn.elu.html | 10 +- .../_autosummary_functions/mlx.nn.gelu.html | 10 +- .../mlx.nn.gelu_approx.html | 10 +- .../mlx.nn.gelu_fast_approx.html | 10 +- .../nn/_autosummary_functions/mlx.nn.glu.html | 10 +- .../mlx.nn.hard_shrink.html | 10 +- .../mlx.nn.hard_tanh.html | 10 +- .../mlx.nn.hardswish.html | 10 +- .../mlx.nn.leaky_relu.html | 10 +- .../mlx.nn.log_sigmoid.html | 10 +- .../mlx.nn.log_softmax.html | 10 +- .../mlx.nn.losses.binary_cross_entropy.html | 10 +- .../mlx.nn.losses.cosine_similarity_loss.html | 10 +- .../mlx.nn.losses.cross_entropy.html | 10 +- .../mlx.nn.losses.gaussian_nll_loss.html | 10 +- .../mlx.nn.losses.hinge_loss.html | 10 +- .../mlx.nn.losses.huber_loss.html | 10 +- .../mlx.nn.losses.kl_div_loss.html | 10 +- .../mlx.nn.losses.l1_loss.html | 10 +- .../mlx.nn.losses.log_cosh_loss.html | 10 +- .../mlx.nn.losses.margin_ranking_loss.html | 10 +- .../mlx.nn.losses.mse_loss.html | 10 +- .../mlx.nn.losses.nll_loss.html | 10 +- .../mlx.nn.losses.smooth_l1_loss.html | 10 +- .../mlx.nn.losses.triplet_loss.html | 10 +- .../_autosummary_functions/mlx.nn.mish.html | 10 +- .../_autosummary_functions/mlx.nn.prelu.html | 10 +- .../_autosummary_functions/mlx.nn.relu.html | 10 +- .../_autosummary_functions/mlx.nn.relu6.html | 10 +- .../_autosummary_functions/mlx.nn.selu.html | 10 +- .../mlx.nn.sigmoid.html | 10 +- .../_autosummary_functions/mlx.nn.silu.html | 10 +- .../mlx.nn.softmax.html | 10 +- .../mlx.nn.softmin.html | 10 +- .../mlx.nn.softplus.html | 10 +- .../mlx.nn.softshrink.html | 10 +- .../_autosummary_functions/mlx.nn.step.html | 10 +- .../_autosummary_functions/mlx.nn.tanh.html | 10 +- docs/build/html/python/nn/functions.html | 10 +- docs/build/html/python/nn/init.html | 10 +- docs/build/html/python/nn/layers.html | 10 +- docs/build/html/python/nn/losses.html | 10 +- docs/build/html/python/nn/module.html | 10 +- docs/build/html/python/ops.html | 12 +- docs/build/html/python/optimizers.html | 10 +- .../_autosummary/mlx.optimizers.AdaDelta.html | 10 +- .../mlx.optimizers.Adafactor.html | 10 +- .../_autosummary/mlx.optimizers.Adagrad.html | 10 +- .../_autosummary/mlx.optimizers.Adam.html | 10 +- .../_autosummary/mlx.optimizers.AdamW.html | 10 +- .../_autosummary/mlx.optimizers.Adamax.html | 10 +- .../_autosummary/mlx.optimizers.Lion.html | 10 +- ....optimizers.Optimizer.apply_gradients.html | 10 +- .../mlx.optimizers.Optimizer.init.html | 10 +- .../mlx.optimizers.Optimizer.state.html | 10 +- .../mlx.optimizers.Optimizer.update.html | 10 +- .../_autosummary/mlx.optimizers.RMSprop.html | 10 +- .../_autosummary/mlx.optimizers.SGD.html | 10 +- .../mlx.optimizers.cosine_decay.html | 10 +- .../mlx.optimizers.exponential_decay.html | 10 +- .../mlx.optimizers.join_schedules.html | 10 +- .../mlx.optimizers.linear_schedule.html | 10 +- .../mlx.optimizers.step_decay.html | 10 +- .../python/optimizers/common_optimizers.html | 10 +- .../html/python/optimizers/optimizer.html | 10 +- .../html/python/optimizers/schedulers.html | 10 +- docs/build/html/python/random.html | 10 +- docs/build/html/python/transforms.html | 10 +- docs/build/html/python/tree_utils.html | 10 +- docs/build/html/quantized_8h.html | 131 +- docs/build/html/quantized_8h_source.html | 188 +- docs/build/html/random_8h_source.html | 6 +- docs/build/html/search.html | 10 +- docs/build/html/search/all_0.js | 8 +- docs/build/html/search/all_1.js | 173 +- docs/build/html/search/all_10.js | 35 +- docs/build/html/search/all_12.js | 9 +- docs/build/html/search/all_13.js | 296 ++- docs/build/html/search/all_14.js | 109 +- docs/build/html/search/all_15.js | 21 +- docs/build/html/search/all_16.js | 22 +- docs/build/html/search/all_2.js | 101 +- docs/build/html/search/all_3.js | 237 +-- docs/build/html/search/all_4.js | 109 +- docs/build/html/search/all_5.js | 73 +- docs/build/html/search/all_7.js | 156 +- docs/build/html/search/all_8.js | 6 +- docs/build/html/search/all_9.js | 45 +- docs/build/html/search/all_b.js | 17 +- docs/build/html/search/all_c.js | 6 +- docs/build/html/search/all_d.js | 203 +- docs/build/html/search/all_e.js | 53 +- docs/build/html/search/all_f.js | 2 +- docs/build/html/search/classes_1.js | 51 +- docs/build/html/search/classes_15.js | 8 +- docs/build/html/search/classes_7.js | 8 +- docs/build/html/search/classes_a.js | 7 +- docs/build/html/search/classes_c.js | 30 +- docs/build/html/search/defines_0.js | 4 +- docs/build/html/search/defines_8.js | 4 +- docs/build/html/search/enums_2.js | 3 +- docs/build/html/search/enumvalues_3.js | 7 +- docs/build/html/search/files_6.js | 7 +- docs/build/html/search/functions_0.js | 4 +- docs/build/html/search/functions_1.js | 135 +- docs/build/html/search/functions_10.js | 6 +- docs/build/html/search/functions_12.js | 7 +- docs/build/html/search/functions_13.js | 200 +- docs/build/html/search/functions_14.js | 47 +- docs/build/html/search/functions_15.js | 11 +- docs/build/html/search/functions_16.js | 13 +- docs/build/html/search/functions_2.js | 65 +- docs/build/html/search/functions_3.js | 157 +- docs/build/html/search/functions_4.js | 62 +- docs/build/html/search/functions_5.js | 57 +- docs/build/html/search/functions_7.js | 115 +- docs/build/html/search/functions_9.js | 39 +- docs/build/html/search/functions_c.js | 6 +- docs/build/html/search/functions_d.js | 14 +- docs/build/html/search/functions_f.js | 2 +- docs/build/html/search/namespaces_0.js | 7 +- docs/build/html/search/typedefs_2.js | 5 +- docs/build/html/search/typedefs_9.js | 2 +- docs/build/html/search/variables_1.js | 16 +- docs/build/html/search/variables_13.js | 30 +- docs/build/html/search/variables_6.js | 19 +- docs/build/html/search/variables_7.js | 6 +- docs/build/html/search/variables_d.js | 11 +- docs/build/html/searchindex.js | 2 +- docs/build/html/steel__gemm__masked_8h.html | 2 +- .../html/steel__gemm__masked_8h_source.html | 20 +- docs/build/html/struct___no_mask-members.html | 4 + docs/build/html/struct___no_mask.html | 121 +- .../html/struct_g_e_m_v_kernel-members.html | 99 + docs/build/html/struct_g_e_m_v_kernel.html | 546 ++++++ .../html/struct_g_e_m_v_t_kernel-members.html | 97 + docs/build/html/struct_g_e_m_v_t_kernel.html | 471 +++++ docs/build/html/struct_scale_op-members.html | 3 +- docs/build/html/struct_scale_op.html | 36 +- ..._1_1metal_1_1_command_encoder-members.html | 8 +- ..._1_1core_1_1metal_1_1_command_encoder.html | 32 - ...structmlx_1_1steel_1_1_g_e_m_m_kernel.html | 4 +- ...structmlx_1_1steel_1_1_loop_alignment.html | 4 +- docs/build/html/types_2complex_8h_source.html | 8 +- docs/build/html/usage/compile.html | 10 +- docs/build/html/usage/distributed.html | 10 +- .../build/html/usage/function_transforms.html | 10 +- docs/build/html/usage/indexing.html | 10 +- docs/build/html/usage/lazy_evaluation.html | 10 +- docs/build/html/usage/numpy.html | 10 +- docs/build/html/usage/quick_start.html | 10 +- docs/build/html/usage/saving_and_loading.html | 10 +- docs/build/html/usage/unified_memory.html | 10 +- docs/build/html/usage/using_streams.html | 10 +- docs/build/html/utils_8h.html | 2 - docs/build/html/utils_8h_source.html | 93 +- 724 files changed, 14529 insertions(+), 11046 deletions(-) create mode 100644 docs/build/html/_sources/python/_autosummary/mlx.core.linalg.cholesky_inv.rst create mode 100644 docs/build/html/_sources/python/_autosummary/mlx.core.linalg.tri_inv.rst delete mode 100644 docs/build/html/class_m_p_s_1_1_kernel-members.html delete mode 100644 docs/build/html/class_m_p_s_1_1_kernel.html delete mode 100644 docs/build/html/class_m_p_s_1_1_kernel.png delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix-members.html delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix.html delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix.png delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_descriptor-members.html delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_descriptor.html delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_descriptor.png delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_multiplication-members.html delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_multiplication.html delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_multiplication.png delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication-members.html delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication.html delete mode 100644 docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication.png delete mode 100644 docs/build/html/class_m_p_s_1_1_vector-members.html delete mode 100644 docs/build/html/class_m_p_s_1_1_vector.html delete mode 100644 docs/build/html/class_m_p_s_1_1_vector.png delete mode 100644 docs/build/html/class_m_p_s_1_1_vector_descriptor-members.html delete mode 100644 docs/build/html/class_m_p_s_1_1_vector_descriptor.html delete mode 100644 docs/build/html/class_m_p_s_1_1_vector_descriptor.png create mode 100644 docs/build/html/classmlx_1_1core_1_1fast_1_1_affine_quantize-members.html create mode 100644 docs/build/html/classmlx_1_1core_1_1fast_1_1_affine_quantize.html create mode 100644 docs/build/html/classmlx_1_1core_1_1fast_1_1_affine_quantize.png rename docs/build/html/{dir_1683daa6c50d5a1449f58a10604f9f12.html => functions_vars_h.html} (64%) rename docs/build/html/{kernels_2steel_2gemm_2gemm_8h.html => gemm_8h.html} (98%) rename docs/build/html/{kernels_2steel_2gemm_2gemm_8h_source.html => gemm_8h_source.html} (99%) create mode 100644 docs/build/html/jit_2gemv__masked_8h.html create mode 100644 docs/build/html/jit_2gemv__masked_8h_source.html create mode 100644 docs/build/html/kernels_2gemv__masked_8h.html create mode 100644 docs/build/html/kernels_2gemv__masked_8h_source.html delete mode 100644 docs/build/html/mps_2gemm_8h.html delete mode 100644 docs/build/html/mps_2gemm_8h_source.html delete mode 100644 docs/build/html/namespace_m_p_s.html delete mode 100644 docs/build/html/namespace_m_t_l.html delete mode 100644 docs/build/html/namespace_m_t_l_1_1_private.html delete mode 100644 docs/build/html/namespace_m_t_l_1_1_private_1_1_class.html delete mode 100644 docs/build/html/namespace_m_t_l_1_1_private_1_1_selector.html delete mode 100644 docs/build/html/namespacemembers_a.html delete mode 100644 docs/build/html/namespacemembers_func_a.html create mode 100644 docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html create mode 100644 docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html create mode 100644 docs/build/html/struct_g_e_m_v_kernel-members.html create mode 100644 docs/build/html/struct_g_e_m_v_kernel.html create mode 100644 docs/build/html/struct_g_e_m_v_t_kernel-members.html create mode 100644 docs/build/html/struct_g_e_m_v_t_kernel.html diff --git a/docs/build/html/.buildinfo b/docs/build/html/.buildinfo index 02b67f3136..e819c46104 100644 --- a/docs/build/html/.buildinfo +++ b/docs/build/html/.buildinfo @@ -1,4 +1,4 @@ # Sphinx build info version 1 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done. -config: 3b02955047d4e4b232f01bccda3ed898 +config: bbbe4e54ecfcc611156ee21a8f15e97b tags: 645f666f9bcd5a90fca523b33c5a78b7 diff --git a/docs/build/html/_sources/dev/extensions.rst b/docs/build/html/_sources/dev/extensions.rst index 9a2be90cd4..ecb418468f 100644 --- a/docs/build/html/_sources/dev/extensions.rst +++ b/docs/build/html/_sources/dev/extensions.rst @@ -486,9 +486,8 @@ below. std::ostringstream kname; kname << "axpby_" << "general_" << type_to_name(out); - // Make sure the metal library is available and look for it - // in the same folder as this executable if needed - d.register_library("mlx_ext", metal::get_colocated_mtllib_path); + // Make sure the metal library is available + d.register_library("mlx_ext"); // Make a kernel from this metal library auto kernel = d.get_kernel(kname.str(), "mlx_ext"); diff --git a/docs/build/html/_sources/examples/llama-inference.rst b/docs/build/html/_sources/examples/llama-inference.rst index 0e080146bc..7e06895e35 100644 --- a/docs/build/html/_sources/examples/llama-inference.rst +++ b/docs/build/html/_sources/examples/llama-inference.rst @@ -15,7 +15,7 @@ module to concisely define the model architecture. Attention layer ^^^^^^^^^^^^^^^^ -We will start with the llama attention layer which notably uses the RoPE +We will start with the Llama attention layer which notably uses the RoPE positional encoding. [1]_ In addition, our attention layer will optionally use a key/value cache that will be concatenated with the provided keys and values to support efficient inference. diff --git a/docs/build/html/_sources/examples/mlp.rst b/docs/build/html/_sources/examples/mlp.rst index 36890e95cf..3214af504e 100644 --- a/docs/build/html/_sources/examples/mlp.rst +++ b/docs/build/html/_sources/examples/mlp.rst @@ -64,7 +64,7 @@ set: Next, setup the problem parameters and load the data. To load the data, you need our `mnist data loader `_, which -we will import as `mnist`. +we will import as ``mnist``. .. code-block:: python diff --git a/docs/build/html/_sources/install.rst b/docs/build/html/_sources/install.rst index c2288e46df..c8cf5723bc 100644 --- a/docs/build/html/_sources/install.rst +++ b/docs/build/html/_sources/install.rst @@ -70,36 +70,36 @@ To build and install the MLX python library from source, first, clone MLX from git clone git@github.com:ml-explore/mlx.git mlx && cd mlx -Install `nanobind `_ with: - -.. code-block:: shell - - pip install git+https://github.com/wjakob/nanobind.git@2f04eac452a6d9142dedb957701bdb20125561e4 - Then simply build and install MLX using pip: .. code-block:: shell - env CMAKE_BUILD_PARALLEL_LEVEL="" pip install . + CMAKE_BUILD_PARALLEL_LEVEL="" pip install . -For developing use an editable install: +For developing, install the package with development dependencies, and use an +editable install: .. code-block:: shell - env CMAKE_BUILD_PARALLEL_LEVEL="" pip install -e . + CMAKE_BUILD_PARALLEL_LEVEL="" pip install -e ".[dev]" -To make sure the install is working run the tests with: +Once the development dependencies are installed, you can build faster with: + +.. code-block:: shell + + CMAKE_BUILD_PARALLEL_LEVEL="" python setup.py build_ext -j --inplace + +Run the tests with: .. code-block:: shell - pip install ".[testing]" python -m unittest discover python/tests -Optional: Install stubs to enable auto completions and type checking from your IDE: +Optional: Install stubs to enable auto completions and type checking from your +IDE: .. code-block:: shell - pip install ".[dev]" python setup.py generate_stubs C++ API diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.DtypeCategory.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.DtypeCategory.rst index ab504dd356..22664ef416 100644 --- a/docs/build/html/_sources/python/_autosummary/mlx.core.DtypeCategory.rst +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.DtypeCategory.rst @@ -9,14 +9,21 @@ .. automethod:: __init__ - .. rubric:: Methods + + + + + .. rubric:: Attributes .. autosummary:: - ~DtypeCategory.__init__ - - - - + ~DtypeCategory.complexfloating + ~DtypeCategory.floating + ~DtypeCategory.inexact + ~DtypeCategory.signedinteger + ~DtypeCategory.unsignedinteger + ~DtypeCategory.integer + ~DtypeCategory.number + ~DtypeCategory.generic \ No newline at end of file diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.cholesky_inv.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.cholesky_inv.rst new file mode 100644 index 0000000000..8e08050f84 --- /dev/null +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.cholesky_inv.rst @@ -0,0 +1,6 @@ +mlx.core.linalg.cholesky\_inv +============================= + +.. currentmodule:: mlx.core.linalg + +.. autofunction:: cholesky_inv \ No newline at end of file diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.tri_inv.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.tri_inv.rst new file mode 100644 index 0000000000..4f12fcc8ec --- /dev/null +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.tri_inv.rst @@ -0,0 +1,6 @@ +mlx.core.linalg.tri\_inv +======================== + +.. currentmodule:: mlx.core.linalg + +.. autofunction:: tri_inv \ No newline at end of file diff --git a/docs/build/html/_sources/python/linalg.rst b/docs/build/html/_sources/python/linalg.rst index 3c34cb3f79..e7fd5ecee3 100644 --- a/docs/build/html/_sources/python/linalg.rst +++ b/docs/build/html/_sources/python/linalg.rst @@ -9,7 +9,9 @@ Linear Algebra :toctree: _autosummary inv + tri_inv norm cholesky + cholesky_inv qr svd diff --git a/docs/build/html/_static/documentation_options.js b/docs/build/html/_static/documentation_options.js index d6799b6791..df3ade8a7b 100644 --- a/docs/build/html/_static/documentation_options.js +++ b/docs/build/html/_static/documentation_options.js @@ -1,5 +1,5 @@ const DOCUMENTATION_OPTIONS = { - VERSION: '0.16.1', + VERSION: '0.16.2', LANGUAGE: 'en', COLLAPSE_INDEX: false, BUILDER: 'html', diff --git a/docs/build/html/annotated.html b/docs/build/html/annotated.html index a846e55595..f07ab420b5 100644 --- a/docs/build/html/annotated.html +++ b/docs/build/html/annotated.html @@ -151,13 +151,14 @@ $(function() {  CDistPrimitive  CGroupA distributed::Group represents a group of independent mlx processes that can communicate  ►Nfast - CCustom - CLayerNorm - CLayerNormVJP - CRMSNorm - CRMSNormVJP - CRoPE - CScaledDotProductAttention + CAffineQuantize + CCustom + CLayerNorm + CLayerNormVJP + CRMSNorm + CRMSNormVJP + CRoPE + CScaledDotProductAttention  ►Nio  CFileReader  CFileWriter @@ -318,151 +319,145 @@ $(function() {  CTransformAdd  CTransformAxpby  CTransformNone - ►NMPS - CKernel - CMatrix - CMatrixDescriptor - CMatrixMultiplication - CMatrixVectorMultiplication - CVector - CVectorDescriptor - ►Npocketfft - ►Ndetail - ►Nthreading - Caligned_allocator - Cconcurrent_queue - Clatch - Cthread_pool - Cadd_vec - Cadd_vec< cmplx< T > > - Carr - Carr_info - Ccfftp - Ccmplx - Ccndarr - CExecC2C - CExecDcst - CExecHartley - CExecR2R - Cfftblue - Cmulti_iter - Cndarr - Cpocketfft_c - Cpocketfft_r - Crev_iter - Crfftp - Csimple_iter - Csincos_2pibyn - CT_dcst23 - CT_dcst4 - CT_dct1 - CT_dst1 - Cutil - CVLEN - CVTYPE - ►C_MLX_BFloat16 - Cbits_to_bfloat_struct - C_NoMask - CAbs - CAdd - CAnd - CArcCos - CArcCosh - CArcSin - CArcSinh - CArcTan - CArcTan2 - CArcTanh - CBitwiseAnd - CBitwiseOr - CBitwiseXor - CBlockMergeSort - Cbool4_or_uint - CCeil - Ccomplex64_t - CConjugate - CCos - CCosh - CCumMax - CCumMin - CCumProd - CCumProd< bool > - CCumSum - CDivide - CDivMod - CEqual - CErf - CErfInv - CExp - CExpm1 - CFloor - CFloorDivide - CGreater - CGreaterEqual - CIndices - CKernelMergeSort - CKernelMultiBlockMergeSort - CLeftShift - CLess - CLessEqual - CLessThan - CLimits - CLimits< bfloat16_t > - CLimits< bool > - CLimits< float > - CLimits< half > - CLimits< int16_t > - CLimits< int32_t > - CLimits< int64_t > - CLimits< int8_t > - CLimits< uint16_t > - CLimits< uint32_t > - CLimits< uint64_t > - CLimits< uint8_t > - CLog - CLog10 - CLog1p - CLog2 - CLogAddExp - CLogicalAnd - CLogicalNot - CLogicalOr - CMax - CMaximum - CMin - CMinimum - Cmlx_atomic - Cmlx_atomic< T, enable_if_t< is_metal_atomic< T > > > - CMLXConvParams - CMLXFastAttentionParams - CMLXScaledDotProductAttentionParams - CMultiply - CNaNEqual - CNegative - CNone - CNotEqual - COr - CPower - CProd - CQuantizedBlockLoader - CReadWriter - CRemainder - CRightShift - CRound - CRsqrt - CScaleOp - CSelect - CSigmoid - CSign - CSin - CSinh - CSqrt - CSquare - CSubtract - CSum - CTan - CTanh - CThreadSort + ►Npocketfft + ►Ndetail + ►Nthreading + Caligned_allocator + Cconcurrent_queue + Clatch + Cthread_pool + Cadd_vec + Cadd_vec< cmplx< T > > + Carr + Carr_info + Ccfftp + Ccmplx + Ccndarr + CExecC2C + CExecDcst + CExecHartley + CExecR2R + Cfftblue + Cmulti_iter + Cndarr + Cpocketfft_c + Cpocketfft_r + Crev_iter + Crfftp + Csimple_iter + Csincos_2pibyn + CT_dcst23 + CT_dcst4 + CT_dct1 + CT_dst1 + Cutil + CVLEN + CVTYPE + ►C_MLX_BFloat16 + Cbits_to_bfloat_struct + C_NoMask + CAbs + CAdd + CAnd + CArcCos + CArcCosh + CArcSin + CArcSinh + CArcTan + CArcTan2 + CArcTanh + CBitwiseAnd + CBitwiseOr + CBitwiseXor + CBlockMergeSort + Cbool4_or_uint + CCeil + Ccomplex64_t + CConjugate + CCos + CCosh + CCumMax + CCumMin + CCumProd + CCumProd< bool > + CCumSum + CDivide + CDivMod + CEqual + CErf + CErfInv + CExp + CExpm1 + CFloor + CFloorDivide + CGEMVKernel + CGEMVTKernelVector matrix multiplication + CGreater + CGreaterEqual + CIndices + CKernelMergeSort + CKernelMultiBlockMergeSort + CLeftShift + CLess + CLessEqual + CLessThan + CLimits + CLimits< bfloat16_t > + CLimits< bool > + CLimits< float > + CLimits< half > + CLimits< int16_t > + CLimits< int32_t > + CLimits< int64_t > + CLimits< int8_t > + CLimits< uint16_t > + CLimits< uint32_t > + CLimits< uint64_t > + CLimits< uint8_t > + CLog + CLog10 + CLog1p + CLog2 + CLogAddExp + CLogicalAnd + CLogicalNot + CLogicalOr + CMax + CMaximum + CMin + CMinimum + Cmlx_atomic + Cmlx_atomic< T, enable_if_t< is_metal_atomic< T > > > + CMLXConvParams + CMLXFastAttentionParams + CMLXScaledDotProductAttentionParams + CMultiply + CNaNEqual + CNegative + CNone + CNotEqual + COr + CPower + CProd + CQuantizedBlockLoader + CReadWriter + CRemainder + CRightShift + CRound + CRsqrt + CScaleOp + CSelect + CSigmoid + CSign + CSin + CSinh + CSqrt + CSquare + CSubtract + CSum + CTan + CTanh + CThreadSort diff --git a/docs/build/html/array_8h_source.html b/docs/build/html/array_8h_source.html index 68b0552273..ceb409e9c6 100644 --- a/docs/build/html/array_8h_source.html +++ b/docs/build/html/array_8h_source.html @@ -794,25 +794,25 @@ $(function() { codefold.init(0); });
mlx::core::allocator::free
void free(Buffer buffer)
mlx::core
Definition allocator.h:7
mlx::core::is_array_v
constexpr bool is_array_v
Definition array.h:559
-
mlx::core::bool_
constexpr Dtype bool_
Definition dtype.h:60
+
mlx::core::bool_
constexpr Dtype bool_
Definition dtype.h:58
mlx::core::deleter_t
std::function< void(allocator::Buffer)> deleter_t
Definition array.h:18
-
mlx::core::uint64
constexpr Dtype uint64
Definition dtype.h:65
-
mlx::core::uint16
constexpr Dtype uint16
Definition dtype.h:63
-
mlx::core::bfloat16
constexpr Dtype bfloat16
Definition dtype.h:74
-
mlx::core::int32
constexpr Dtype int32
Definition dtype.h:69
-
mlx::core::float32
constexpr Dtype float32
Definition dtype.h:73
-
mlx::core::int16
constexpr Dtype int16
Definition dtype.h:68
-
mlx::core::int8
constexpr Dtype int8
Definition dtype.h:67
-
mlx::core::int64
constexpr Dtype int64
Definition dtype.h:70
+
mlx::core::uint64
constexpr Dtype uint64
Definition dtype.h:63
+
mlx::core::uint16
constexpr Dtype uint16
Definition dtype.h:61
+
mlx::core::bfloat16
constexpr Dtype bfloat16
Definition dtype.h:72
+
mlx::core::int32
constexpr Dtype int32
Definition dtype.h:67
+
mlx::core::float32
constexpr Dtype float32
Definition dtype.h:71
+
mlx::core::int16
constexpr Dtype int16
Definition dtype.h:66
+
mlx::core::int8
constexpr Dtype int8
Definition dtype.h:65
+
mlx::core::int64
constexpr Dtype int64
Definition dtype.h:68
mlx::core::is_arrays_v
constexpr bool is_arrays_v
Definition array.h:563
-
mlx::core::uint8
constexpr Dtype uint8
Definition dtype.h:62
-
mlx::core::float16
constexpr Dtype float16
Definition dtype.h:72
-
mlx::core::uint32
constexpr Dtype uint32
Definition dtype.h:64
-
mlx::core::size_of
uint8_t size_of(const Dtype &t)
Definition dtype.h:95
+
mlx::core::uint8
constexpr Dtype uint8
Definition dtype.h:60
+
mlx::core::float16
constexpr Dtype float16
Definition dtype.h:70
+
mlx::core::uint32
constexpr Dtype uint32
Definition dtype.h:62
+
mlx::core::size_of
uint8_t size_of(const Dtype &t)
Definition dtype.h:93
mlx::core::enable_for_arrays_t
typename std::enable_if_t< is_arrays_v< T... > > enable_for_arrays_t
Definition array.h:566
-
mlx::core::complex64
constexpr Dtype complex64
Definition dtype.h:75
-
mlx::core::Dtype
Definition dtype.h:15
-
mlx::core::TypeToDtype
Definition dtype.h:102
+
mlx::core::complex64
constexpr Dtype complex64
Definition dtype.h:73
+
mlx::core::Dtype
Definition dtype.h:13
+
mlx::core::TypeToDtype
Definition dtype.h:100
mlx::core::array::ArrayIterator
Definition array.h:141
mlx::core::array::ArrayIterator::operator==
friend bool operator==(const ArrayIterator &a, const ArrayIterator &b)
Definition array.h:161
diff --git a/docs/build/html/backend_2accelerate_2utils_8h_source.html b/docs/build/html/backend_2accelerate_2utils_8h_source.html index b8057101e7..f5ce8d3f8f 100644 --- a/docs/build/html/backend_2accelerate_2utils_8h_source.html +++ b/docs/build/html/backend_2accelerate_2utils_8h_source.html @@ -117,8 +117,8 @@ $(function() { codefold.init(0); });
mlx::core
Definition allocator.h:7
mlx::core::to_bnns_dtype
BNNSDataType to_bnns_dtype(Dtype mlx_dtype)
Definition utils.h:10
mlx::core::kindof
Dtype::Kind kindof(const Dtype &t)
-
mlx::core::size_of
uint8_t size_of(const Dtype &t)
Definition dtype.h:95
-
mlx::core::Dtype
Definition dtype.h:15
+
mlx::core::size_of
uint8_t size_of(const Dtype &t)
Definition dtype.h:93
+
mlx::core::Dtype
Definition dtype.h:13
diff --git a/docs/build/html/backend_2metal_2device_8h.html b/docs/build/html/backend_2metal_2device_8h.html index 573f61f73a..9a49c74a9a 100644 --- a/docs/build/html/backend_2metal_2device_8h.html +++ b/docs/build/html/backend_2metal_2device_8h.html @@ -89,8 +89,6 @@ $(function() { #include <string>
#include <unordered_map>
#include <unordered_set>
-#include <dlfcn.h>
-#include <filesystem>
#include "mlx/array.h"
#include "mlx/device.h"
@@ -121,8 +119,6 @@ Typedefs - -

Functions

std::string mlx::core::metal::get_colocated_mtllib_path (const std::string &lib_name)
 
Device & mlx::core::metal::device (mlx::core::Device)
 
diff --git a/docs/build/html/backend_2metal_2device_8h_source.html b/docs/build/html/backend_2metal_2device_8h_source.html index 8a7e381018..13ee2fe02d 100644 --- a/docs/build/html/backend_2metal_2device_8h_source.html +++ b/docs/build/html/backend_2metal_2device_8h_source.html @@ -94,258 +94,190 @@ $(function() { codefold.init(0); });
9#include <unordered_map>
10#include <unordered_set>
11
-
12#include <dlfcn.h>
-
13#include <filesystem>
+
12#include "mlx/array.h"
+
13#include "mlx/device.h"
14
-
15#include "mlx/array.h"
-
16#include "mlx/device.h"
-
17
-
18namespace fs = std::filesystem;
+
15namespace mlx::core::metal {
+
16
+
17using MTLFCList =
+
18 std::vector<std::tuple<const void*, MTL::DataType, NS::UInteger>>;
19
-
20namespace mlx::core::metal {
-
21
-
-
22inline std::string get_colocated_mtllib_path(const std::string& lib_name) {
-
23 Dl_info info;
-
24 std::string mtllib_path;
-
25 std::string lib_ext = lib_name + ".metallib";
-
26
-
27 int success = dladdr((void*)get_colocated_mtllib_path, &info);
-
28 if (success) {
-
29 auto mtllib = fs::path(info.dli_fname).remove_filename() / lib_ext;
-
30 mtllib_path = mtllib.c_str();
-
31 }
-
32
-
33 return mtllib_path;
-
34}
+
+
20struct CommandEncoder {
+
21 CommandEncoder(MTL::CommandBuffer* cbuf);
+
22 CommandEncoder(const CommandEncoder&) = delete;
+
23 CommandEncoder& operator=(const CommandEncoder&) = delete;
+
24
+
+
25 struct ConcurrentContext {
+
+
26 ConcurrentContext(CommandEncoder& enc) : enc(enc) {
+
27 enc.concurrent = true;
+
28 }
+
+
+
29 ~ConcurrentContext() {
+
30 enc.concurrent = false;
+
31 enc.outputs.insert(
+
32 enc.concurrent_outputs.begin(), enc.concurrent_outputs.end());
+
33 enc.concurrent_outputs.clear();
+
34 }
35
-
36using MTLFCList =
-
37 std::vector<std::tuple<const void*, MTL::DataType, NS::UInteger>>;
-
38
-
-
39struct CommandEncoder {
+
36 private:
+
37 CommandEncoder& enc;
+
38 };
+
+
39
-
40 CommandEncoder(MTL::CommandBuffer* cbuf) : cbuf(cbuf) {
-
41 enc = cbuf->computeCommandEncoder(MTL::DispatchTypeConcurrent);
-
42 enc->retain();
-
43 };
+
40 MTL::ComputeCommandEncoder* operator->() {
+
41 return enc;
+
42 }
-
44 CommandEncoder(const CommandEncoder&) = delete;
-
45 CommandEncoder& operator=(const CommandEncoder&) = delete;
-
46
-
-
47 struct ConcurrentContext {
-
-
48 ConcurrentContext(CommandEncoder& enc) : enc(enc) {
-
49 enc.concurrent = true;
-
50 }
-
-
-
51 ~ConcurrentContext() {
-
52 enc.concurrent = false;
-
53 enc.outputs.insert(
-
54 enc.concurrent_outputs.begin(), enc.concurrent_outputs.end());
-
55 enc.concurrent_outputs.clear();
-
56 }
+
43
+
44 void set_input_array(const array& a, int idx, int64_t offset = 0);
+
45 void set_output_array(array& a, int idx, int64_t offset = 0);
+
46 void dispatchThreadgroups(MTL::Size grid_dims, MTL::Size group_dims);
+
47 void dispatchThreads(MTL::Size grid_dims, MTL::Size group_dims);
+
48
+
+
49 ConcurrentContext start_concurrent() {
+
50 return ConcurrentContext(*this);
+
51 }
+
52
+
53 ~CommandEncoder();
+
54
+
55 private:
+
56 void maybe_split();
57
-
58 private:
-
59 CommandEncoder& enc;
-
60 };
-
-
61
-
-
62 MTL::ComputeCommandEncoder* operator->() {
-
63 return enc;
-
64 }
+
58 int num_dispatches{0};
+
59 MTL::CommandBuffer* cbuf;
+
60 MTL::ComputeCommandEncoder* enc;
+
61 bool concurrent{false};
+
62 std::unordered_set<MTL::Resource*> outputs;
+
63 std::unordered_set<MTL::Resource*> concurrent_outputs;
+
64};
65
-
-
66 void set_input_array(const array& a, int idx, int64_t offset = 0) {
-
67 auto r_buf =
-
68 static_cast<MTL::Resource*>(const_cast<void*>(a.buffer().ptr()));
-
69 if (auto it = outputs.find(r_buf); it != outputs.end()) {
-
70 // Insert a barrier
-
71 enc->memoryBarrier(&r_buf, 1);
+
+
66class Device {
+
67 public:
+
68 Device();
+
69 Device(const Device&) = delete;
+
70 Device& operator=(const Device&) = delete;
+
71 ~Device();
72
-
73 // Remove the output
-
74 outputs.erase(it);
-
75 }
-
76 auto a_buf = static_cast<const MTL::Buffer*>(a.buffer().ptr());
-
77 auto base_offset = a.data<char>() -
-
78 static_cast<char*>(const_cast<MTL::Buffer*>(a_buf)->contents());
-
79 base_offset += offset;
-
80 enc->setBuffer(a_buf, base_offset, idx);
-
81 }
+
+
73 MTL::Device* mtl_device() {
+
74 return device_;
+
75 };
-
82
-
-
83 void set_output_array(array& a, int idx, int64_t offset = 0) {
-
84 // Add barriers before adding the output to the output set
-
85 set_input_array(a, idx, offset);
-
86 auto buf = static_cast<MTL::Resource*>(a.buffer().ptr());
-
87 if (concurrent) {
-
88 concurrent_outputs.insert(buf);
-
89 } else {
-
90 outputs.insert(buf);
-
91 }
-
92 }
-
-
93
-
94 void dispatchThreadgroups(MTL::Size grid_dims, MTL::Size group_dims);
-
95 void dispatchThreads(MTL::Size grid_dims, MTL::Size group_dims);
-
96
-
-
97 ConcurrentContext start_concurrent() {
-
98 return ConcurrentContext(*this);
-
99 }
-
-
100
-
-
101 ~CommandEncoder() {
-
102 enc->endEncoding();
-
103 enc->release();
-
104 }
-
-
105
-
106 private:
-
107 void maybe_split();
+
76
+
77 void new_queue(int index);
+
78 MTL::CommandBuffer* get_command_buffer(int index);
+
79 int get_command_buffer_ops(int index);
+
80 void increment_command_buffer_ops(int index);
+
81 void commit_command_buffer(int index);
+
82 CommandEncoder& get_command_encoder(int index);
+
83 void end_encoding(int index);
+
84
+
85 void register_library(
+
86 const std::string& lib_name,
+
87 const std::string& lib_path);
+
88
+
89 void register_library(const std::string& lib_name);
+
90
+
91 MTL::Library* get_library(const std::string& name);
+
92
+
93 MTL::Library* get_library(
+
94 const std::string& name,
+
95 const std::string& source_string,
+
96 bool cache = true);
+
97
+
98 MTL::Library* get_library(
+
99 const std::string& name,
+
100 const MTL::StitchedLibraryDescriptor* desc,
+
101 bool cache = true);
+
102
+
103 MTL::Function* get_function(
+
104 const std::string& base_name,
+
105 MTL::Library* mtl_lib,
+
106 const std::string& specialized_name = "",
+
107 const MTLFCList& func_consts = {});
108
-
109 int num_dispatches{0};
-
110 MTL::CommandBuffer* cbuf;
-
111 MTL::ComputeCommandEncoder* enc;
-
112 bool concurrent{false};
-
113 std::unordered_set<MTL::Resource*> outputs;
-
114 std::unordered_set<MTL::Resource*> concurrent_outputs;
-
115};
-
-
116
-
-
117class Device {
-
118 public:
-
119 Device();
-
120 Device(const Device&) = delete;
-
121 Device& operator=(const Device&) = delete;
-
122 ~Device();
-
123
-
-
124 MTL::Device* mtl_device() {
-
125 return device_;
-
126 };
-
-
127
-
128 void new_queue(int index);
-
129 MTL::CommandBuffer* get_command_buffer(int index);
-
130 int get_command_buffer_ops(int index);
-
131 void increment_command_buffer_ops(int index);
-
132 void commit_command_buffer(int index);
-
133 CommandEncoder& get_command_encoder(int index);
-
134 void end_encoding(int index);
-
135
-
136 void register_library(
-
137 const std::string& lib_name,
-
138 const std::string& lib_path);
-
139 void register_library(
-
140 const std::string& lib_name,
-
141 const std::function<std::string(const std::string&)>& lib_path_func =
-
142 get_colocated_mtllib_path);
-
143
-
144 MTL::Library* get_library(const std::string& name);
+
109 MTL::Function* get_function(
+
110 const std::string& base_name,
+
111 const std::string& lib_name = "mlx",
+
112 const std::string& specialized_name = "",
+
113 const MTLFCList& func_consts = {});
+
114
+
115 MTL::ComputePipelineState* get_kernel(
+
116 const std::string& base_name,
+
117 MTL::Library* mtl_lib,
+
118 const std::string& hash_name = "",
+
119 const MTLFCList& func_consts = {},
+
120 const std::vector<MTL::Function*>& linked_functions = {});
+
121
+
122 MTL::ComputePipelineState* get_kernel(
+
123 const std::string& base_name,
+
124 const std::string& lib_name = "mlx",
+
125 const std::string& hash_name = "",
+
126 const MTLFCList& func_consts = {},
+
127 const std::vector<MTL::Function*>& linked_functions = {});
+
128
+
129 MTL::ArgumentEncoder* argument_encoder(
+
130 const std::vector<MTL::ArgumentDescriptor*>& arg_descs) const;
+
131
+
132 private:
+
133 MTL::Library* get_library_cache_(const std::string& name);
+
134
+
135 MTL::Library* get_library_(const std::string& source_string);
+
136 MTL::Library* get_library_(const MTL::StitchedLibraryDescriptor* desc);
+
137
+
138 MTL::Function* get_function_(const std::string& name, MTL::Library* mtl_lib);
+
139
+
140 MTL::Function* get_function_(
+
141 const std::string& name,
+
142 const std::string& specialized_name,
+
143 const MTLFCList& func_consts,
+
144 MTL::Library* mtl_lib);
145
-
146 MTL::Library* get_library(
-
147 const std::string& name,
-
148 const std::string& source_string,
-
149 bool cache = true);
-
150
-
151 MTL::Library* get_library(
-
152 const std::string& name,
-
153 const MTL::StitchedLibraryDescriptor* desc,
-
154 bool cache = true);
-
155
-
156 MTL::Function* get_function(
-
157 const std::string& base_name,
-
158 MTL::Library* mtl_lib,
-
159 const std::string& specialized_name = "",
-
160 const MTLFCList& func_consts = {});
-
161
-
162 MTL::Function* get_function(
-
163 const std::string& base_name,
-
164 const std::string& lib_name = "mlx",
-
165 const std::string& specialized_name = "",
-
166 const MTLFCList& func_consts = {});
-
167
-
168 MTL::ComputePipelineState* get_kernel(
-
169 const std::string& base_name,
-
170 MTL::Library* mtl_lib,
-
171 const std::string& hash_name = "",
-
172 const MTLFCList& func_consts = {},
-
173 const std::vector<MTL::Function*>& linked_functions = {});
-
174
-
175 MTL::ComputePipelineState* get_kernel(
-
176 const std::string& base_name,
-
177 const std::string& lib_name = "mlx",
-
178 const std::string& hash_name = "",
-
179 const MTLFCList& func_consts = {},
-
180 const std::vector<MTL::Function*>& linked_functions = {});
-
181
-
182 MTL::ArgumentEncoder* argument_encoder(
-
183 const std::vector<MTL::ArgumentDescriptor*>& arg_descs) const;
-
184
-
185 private:
-
186 MTL::Library* get_library_cache_(const std::string& name);
-
187
-
188 MTL::Library* get_library_(const std::string& source_string);
-
189 MTL::Library* get_library_(const MTL::StitchedLibraryDescriptor* desc);
-
190
-
191 MTL::Function* get_function_(const std::string& name, MTL::Library* mtl_lib);
-
192
-
193 MTL::Function* get_function_(
-
194 const std::string& name,
-
195 const std::string& specialized_name,
-
196 const MTLFCList& func_consts,
-
197 MTL::Library* mtl_lib);
-
198
-
199 MTL::LinkedFunctions* get_linked_functions_(
-
200 const std::vector<MTL::Function*>& funcs);
-
201
-
202 MTL::ComputePipelineState* get_kernel_(
-
203 const std::string& name,
-
204 const MTL::Function* mtl_function);
-
205
-
206 MTL::ComputePipelineState* get_kernel_(
-
207 const std::string& name,
-
208 const MTL::Function* mtl_function,
-
209 const MTL::LinkedFunctions* linked_functions);
-
210
-
211 MTL::Device* device_;
-
212 std::unordered_map<int32_t, MTL::CommandQueue*> queue_map_;
-
213 std::unordered_map<int32_t, std::pair<int, MTL::CommandBuffer*>> buffer_map_;
-
214 std::unordered_map<int32_t, std::unique_ptr<CommandEncoder>> encoder_map_;
-
215 std::unordered_map<std::string, MTL::ComputePipelineState*> kernel_map_;
-
216 std::unordered_map<std::string, MTL::Library*> library_map_;
-
217 std::mutex mtx_;
-
218};
+
146 MTL::LinkedFunctions* get_linked_functions_(
+
147 const std::vector<MTL::Function*>& funcs);
+
148
+
149 MTL::ComputePipelineState* get_kernel_(
+
150 const std::string& name,
+
151 const MTL::Function* mtl_function);
+
152
+
153 MTL::ComputePipelineState* get_kernel_(
+
154 const std::string& name,
+
155 const MTL::Function* mtl_function,
+
156 const MTL::LinkedFunctions* linked_functions);
+
157
+
158 MTL::Device* device_;
+
159 std::unordered_map<int32_t, MTL::CommandQueue*> queue_map_;
+
160 std::unordered_map<int32_t, std::pair<int, MTL::CommandBuffer*>> buffer_map_;
+
161 std::unordered_map<int32_t, std::unique_ptr<CommandEncoder>> encoder_map_;
+
162 std::unordered_map<std::string, MTL::ComputePipelineState*> kernel_map_;
+
163 std::unordered_map<std::string, MTL::Library*> library_map_;
+
164 std::mutex mtx_;
+
165};
-
219
-
220Device& device(mlx::core::Device);
-
221
-
222} // namespace mlx::core::metal
+
166
+
167Device& device(mlx::core::Device);
+
168
+
169} // namespace mlx::core::metal
-
buf
MTL::Buffer * buf
Definition allocator.h:38
-
mlx::core::allocator::Buffer::ptr
const void * ptr() const
Definition allocator.h:23
mlx::core::array
Definition array.h:20
-
mlx::core::array::data
T * data()
Definition array.h:313
-
mlx::core::array::buffer
allocator::Buffer & buffer()
Definition array.h:299
-
mlx::core::metal::Device
Definition device.h:117
+
mlx::core::metal::Device
Definition device.h:66
mlx::core::metal::Device::get_command_buffer_ops
int get_command_buffer_ops(int index)
-
mlx::core::metal::Device::mtl_device
MTL::Device * mtl_device()
Definition device.h:124
+
mlx::core::metal::Device::mtl_device
MTL::Device * mtl_device()
Definition device.h:73
mlx::core::metal::Device::register_library
void register_library(const std::string &lib_name, const std::string &lib_path)
mlx::core::metal::Device::get_command_buffer
MTL::CommandBuffer * get_command_buffer(int index)
mlx::core::metal::Device::end_encoding
void end_encoding(int index)
mlx::core::metal::Device::get_kernel
MTL::ComputePipelineState * get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})
-
mlx::core::metal::Device::register_library
void register_library(const std::string &lib_name, const std::function< std::string(const std::string &)> &lib_path_func=get_colocated_mtllib_path)
mlx::core::metal::Device::argument_encoder
MTL::ArgumentEncoder * argument_encoder(const std::vector< MTL::ArgumentDescriptor * > &arg_descs) const
mlx::core::metal::Device::increment_command_buffer_ops
void increment_command_buffer_ops(int index)
mlx::core::metal::Device::new_queue
void new_queue(int index)
@@ -353,6 +285,7 @@ $(function() { codefold.init(0); });
mlx::core::metal::Device::get_library
MTL::Library * get_library(const std::string &name, const MTL::StitchedLibraryDescriptor *desc, bool cache=true)
mlx::core::metal::Device::commit_command_buffer
void commit_command_buffer(int index)
mlx::core::metal::Device::get_library
MTL::Library * get_library(const std::string &name, const std::string &source_string, bool cache=true)
+
mlx::core::metal::Device::register_library
void register_library(const std::string &lib_name)
mlx::core::metal::Device::get_function
MTL::Function * get_function(const std::string &base_name, MTL::Library *mtl_lib, const std::string &specialized_name="", const MTLFCList &func_consts={})
mlx::core::metal::Device::Device
Device(const Device &)=delete
mlx::core::metal::Device::get_function
MTL::Function * get_function(const std::string &base_name, const std::string &lib_name="mlx", const std::string &specialized_name="", const MTLFCList &func_consts={})
@@ -362,23 +295,22 @@ $(function() { codefold.init(0); });
mlx::core::metal::Device::get_command_encoder
CommandEncoder & get_command_encoder(int index)
mlx::core::metal
Definition allocator.h:12
-
mlx::core::metal::get_colocated_mtllib_path
std::string get_colocated_mtllib_path(const std::string &lib_name)
Definition device.h:22
-
mlx::core::metal::MTLFCList
std::vector< std::tuple< const void *, MTL::DataType, NS::UInteger > > MTLFCList
Definition device.h:36
+
mlx::core::metal::MTLFCList
std::vector< std::tuple< const void *, MTL::DataType, NS::UInteger > > MTLFCList
Definition device.h:17
mlx::core::metal::device
Device & device(mlx::core::Device)
mlx::core::Device
Definition device.h:7
- - -
mlx::core::metal::CommandEncoder::ConcurrentContext::ConcurrentContext
ConcurrentContext(CommandEncoder &enc)
Definition device.h:48
-
mlx::core::metal::CommandEncoder
Definition device.h:39
+ + +
mlx::core::metal::CommandEncoder::ConcurrentContext::ConcurrentContext
ConcurrentContext(CommandEncoder &enc)
Definition device.h:26
+
mlx::core::metal::CommandEncoder
Definition device.h:20
mlx::core::metal::CommandEncoder::dispatchThreads
void dispatchThreads(MTL::Size grid_dims, MTL::Size group_dims)
-
mlx::core::metal::CommandEncoder::CommandEncoder
CommandEncoder(MTL::CommandBuffer *cbuf)
Definition device.h:40
+
mlx::core::metal::CommandEncoder::CommandEncoder
CommandEncoder(MTL::CommandBuffer *cbuf)
mlx::core::metal::CommandEncoder::operator=
CommandEncoder & operator=(const CommandEncoder &)=delete
-
mlx::core::metal::CommandEncoder::start_concurrent
ConcurrentContext start_concurrent()
Definition device.h:97
-
mlx::core::metal::CommandEncoder::set_output_array
void set_output_array(array &a, int idx, int64_t offset=0)
Definition device.h:83
+
mlx::core::metal::CommandEncoder::start_concurrent
ConcurrentContext start_concurrent()
Definition device.h:49
+
mlx::core::metal::CommandEncoder::set_output_array
void set_output_array(array &a, int idx, int64_t offset=0)
mlx::core::metal::CommandEncoder::dispatchThreadgroups
void dispatchThreadgroups(MTL::Size grid_dims, MTL::Size group_dims)
-
mlx::core::metal::CommandEncoder::~CommandEncoder
~CommandEncoder()
Definition device.h:101
-
mlx::core::metal::CommandEncoder::operator->
MTL::ComputeCommandEncoder * operator->()
Definition device.h:62
-
mlx::core::metal::CommandEncoder::set_input_array
void set_input_array(const array &a, int idx, int64_t offset=0)
Definition device.h:66
+ +
mlx::core::metal::CommandEncoder::operator->
MTL::ComputeCommandEncoder * operator->()
Definition device.h:40
+
mlx::core::metal::CommandEncoder::set_input_array
void set_input_array(const array &a, int idx, int64_t offset=0)
mlx::core::metal::CommandEncoder::CommandEncoder
CommandEncoder(const CommandEncoder &)=delete
diff --git a/docs/build/html/backend_2metal_2utils_8h.html b/docs/build/html/backend_2metal_2utils_8h.html index 9a2da99d38..b01d6d0d53 100644 --- a/docs/build/html/backend_2metal_2utils_8h.html +++ b/docs/build/html/backend_2metal_2utils_8h.html @@ -76,7 +76,8 @@ $(function() {
-Namespaces
+Namespaces | +Functions
utils.h File Reference
@@ -92,6 +93,29 @@ Namespaces   namespace  mlx::core   + + + + + + + + + + + + + + + + + + + + + +

+Functions

template<typename T >
void mlx::core::set_vector_bytes (CommandEncoder &enc, const std::vector< T > &vec, size_t nelems, int idx)
 
template<typename T >
void mlx::core::set_vector_bytes (CommandEncoder &enc, const std::vector< T > &vec, int idx)
 
std::string mlx::core::type_to_name (const array &a)
 
MTL::Size mlx::core::get_block_dims (int dim0, int dim1, int dim2)
 
MTL::Size mlx::core::get_2d_grid_dims (const std::vector< int > &shape, const std::vector< size_t > &strides)
 
NS::String * mlx::core::make_string (std::ostringstream &os)
 
void mlx::core::debug_set_stream_queue_label (MTL::CommandQueue *queue, int index)
 
void mlx::core::debug_set_primitive_buffer_label (MTL::CommandBuffer *command_buffer, Primitive &primitive)
 
std::string mlx::core::get_primitive_string (Primitive *primitive)
 
diff --git a/docs/build/html/backend_2metal_2utils_8h_source.html b/docs/build/html/backend_2metal_2utils_8h_source.html index 7e3a1a80f4..015caa2fa0 100644 --- a/docs/build/html/backend_2metal_2utils_8h_source.html +++ b/docs/build/html/backend_2metal_2utils_8h_source.html @@ -93,155 +93,95 @@ $(function() { codefold.init(0); });
8
9namespace mlx::core {
10
-
11namespace {
+
11using metal::CommandEncoder;
12
-
13using metal::CommandEncoder;
-
14
-
15template <typename T>
-
16inline void set_vector_bytes(
-
17 CommandEncoder& enc,
-
18 const std::vector<T>& vec,
-
19 size_t nelems,
-
20 int idx) {
-
21 enc->setBytes(vec.data(), nelems * sizeof(T), idx);
-
22}
-
23
-
24template <typename T>
-
25inline void
-
26set_vector_bytes(CommandEncoder& enc, const std::vector<T>& vec, int idx) {
-
27 return set_vector_bytes(enc, vec, vec.size(), idx);
-
28}
+
13template <typename T>
+
+
14inline void set_vector_bytes(
+
15 CommandEncoder& enc,
+
16 const std::vector<T>& vec,
+
17 size_t nelems,
+
18 int idx) {
+
19 enc->setBytes(vec.data(), nelems * sizeof(T), idx);
+
20}
+
+
21
+
22template <typename T>
+
23inline void
+
+
24set_vector_bytes(CommandEncoder& enc, const std::vector<T>& vec, int idx) {
+
25 return set_vector_bytes(enc, vec, vec.size(), idx);
+
26}
+
+
27
+
28std::string type_to_name(const array& a);
29
-
30std::string type_to_name(const array& a) {
-
31 std::string tname;
-
32 switch (a.dtype()) {
-
33 case bool_:
-
34 tname = "bool_";
-
35 break;
-
36 case uint8:
-
37 tname = "uint8";
-
38 break;
-
39 case uint16:
-
40 tname = "uint16";
-
41 break;
-
42 case uint32:
-
43 tname = "uint32";
-
44 break;
-
45 case uint64:
-
46 tname = "uint64";
-
47 break;
-
48 case int8:
-
49 tname = "int8";
-
50 break;
-
51 case int16:
-
52 tname = "int16";
-
53 break;
-
54 case int32:
-
55 tname = "int32";
-
56 break;
-
57 case int64:
-
58 tname = "int64";
-
59 break;
-
60 case float16:
-
61 tname = "float16";
-
62 break;
-
63 case float32:
-
64 tname = "float32";
-
65 break;
-
66 case bfloat16:
-
67 tname = "bfloat16";
-
68 break;
-
69 case complex64:
-
70 tname = "complex64";
-
71 break;
-
72 }
-
73 return tname;
-
74}
-
75
-
76MTL::Size get_block_dims(int dim0, int dim1, int dim2) {
-
77 int pows[3] = {0, 0, 0};
-
78 int sum = 0;
-
79 while (true) {
-
80 int presum = sum;
-
81 // Check all the pows
-
82 if (dim0 >= (1 << (pows[0] + 1))) {
-
83 pows[0]++;
-
84 sum++;
-
85 }
-
86 if (sum == 10) {
-
87 break;
-
88 }
-
89 if (dim1 >= (1 << (pows[1] + 1))) {
-
90 pows[1]++;
-
91 sum++;
-
92 }
-
93 if (sum == 10) {
-
94 break;
-
95 }
-
96 if (dim2 >= (1 << (pows[2] + 1))) {
-
97 pows[2]++;
-
98 sum++;
-
99 }
-
100 if (sum == presum || sum == 10) {
-
101 break;
-
102 }
-
103 }
-
104 return MTL::Size{1ul << pows[0], 1ul << pows[1], 1ul << pows[2]};
-
105}
-
106
-
107inline NS::String* make_string(std::ostringstream& os) {
-
108 std::string string = os.str();
-
109 return NS::String::string(string.c_str(), NS::UTF8StringEncoding);
-
110}
-
111
-
112inline void debug_set_stream_queue_label(MTL::CommandQueue* queue, int index) {
-
113#ifdef MLX_METAL_DEBUG
-
114 std::ostringstream label;
-
115 label << "Stream " << index;
-
116 queue->setLabel(make_string(label));
-
117#endif
-
118}
-
119
-
120inline void debug_set_primitive_buffer_label(
-
121 MTL::CommandBuffer* command_buffer,
-
122 Primitive& primitive) {
-
123#ifdef MLX_METAL_DEBUG
-
124 std::ostringstream label;
-
125 if (auto cbuf_label = command_buffer->label(); cbuf_label) {
-
126 label << cbuf_label->utf8String();
-
127 }
-
128 primitive.print(label);
-
129 command_buffer->setLabel(make_string(label));
-
130#endif
-
131}
-
132
-
133std::string get_primitive_string(Primitive* primitive) {
-
134 std::ostringstream op_t;
-
135 primitive->print(op_t);
-
136 return op_t.str();
-
137}
-
138
-
139} // namespace
-
140
-
141} // namespace mlx::core
+
30// Compute the thread block dimensions which fit the given
+
31// input dimensions.
+
32// - The thread block dimensions will be powers of two
+
33// - The thread block size will be less than 1024
+
34MTL::Size get_block_dims(int dim0, int dim1, int dim2);
+
35
+
36// Computes a 2D grid where each element is < UINT_MAX
+
37// Assumes:
+
38// - overall size (product of non-broadcasted dimensions) is < UINT_MAX^2
+
39// - shape and strides correspond to a contiguous (no holes) but
+
40// possibly broadcasted array
+
41MTL::Size get_2d_grid_dims(
+
42 const std::vector<int>& shape,
+
43 const std::vector<size_t>& strides);
+
44
+
+
45inline NS::String* make_string(std::ostringstream& os) {
+
46 std::string string = os.str();
+
47 return NS::String::string(string.c_str(), NS::UTF8StringEncoding);
+
48}
+
+
49
+
+
50inline void debug_set_stream_queue_label(MTL::CommandQueue* queue, int index) {
+
51#ifdef MLX_METAL_DEBUG
+
52 std::ostringstream label;
+
53 label << "Stream " << index;
+
54 queue->setLabel(make_string(label));
+
55#endif
+
56}
+
+
57
+
+
58inline void debug_set_primitive_buffer_label(
+
59 MTL::CommandBuffer* command_buffer,
+
60 Primitive& primitive) {
+
61#ifdef MLX_METAL_DEBUG
+
62 std::ostringstream label;
+
63 if (auto cbuf_label = command_buffer->label(); cbuf_label) {
+
64 label << cbuf_label->utf8String();
+
65 }
+
66 primitive.print(label);
+
67 command_buffer->setLabel(make_string(label));
+
68#endif
+
69}
+
+
70
+
71std::string get_primitive_string(Primitive* primitive);
+
72
+
73} // namespace mlx::core
-
mlx::core::sum
array sum(const array &a, bool keepdims, StreamOrDevice s={})
Sums the elements of an array.
+
mlx::core::Primitive
Definition primitives.h:48
+
mlx::core::Primitive::print
virtual void print(std::ostream &os)=0
Print the primitive.
+
mlx::core::array
Definition array.h:20
mlx::core
Definition allocator.h:7
-
mlx::core::bool_
constexpr Dtype bool_
Definition dtype.h:60
-
mlx::core::uint64
constexpr Dtype uint64
Definition dtype.h:65
-
mlx::core::uint16
constexpr Dtype uint16
Definition dtype.h:63
-
mlx::core::bfloat16
constexpr Dtype bfloat16
Definition dtype.h:74
-
mlx::core::int32
constexpr Dtype int32
Definition dtype.h:69
-
mlx::core::float32
constexpr Dtype float32
Definition dtype.h:73
-
mlx::core::int16
constexpr Dtype int16
Definition dtype.h:68
-
mlx::core::int8
constexpr Dtype int8
Definition dtype.h:67
-
mlx::core::int64
constexpr Dtype int64
Definition dtype.h:70
-
mlx::core::uint8
constexpr Dtype uint8
Definition dtype.h:62
-
mlx::core::float16
constexpr Dtype float16
Definition dtype.h:72
-
mlx::core::uint32
constexpr Dtype uint32
Definition dtype.h:64
-
mlx::core::complex64
constexpr Dtype complex64
Definition dtype.h:75
+
mlx::core::debug_set_primitive_buffer_label
void debug_set_primitive_buffer_label(MTL::CommandBuffer *command_buffer, Primitive &primitive)
Definition utils.h:58
+
mlx::core::set_vector_bytes
void set_vector_bytes(CommandEncoder &enc, const std::vector< T > &vec, size_t nelems, int idx)
Definition utils.h:14
+
mlx::core::debug_set_stream_queue_label
void debug_set_stream_queue_label(MTL::CommandQueue *queue, int index)
Definition utils.h:50
+
mlx::core::get_block_dims
MTL::Size get_block_dims(int dim0, int dim1, int dim2)
+
mlx::core::get_2d_grid_dims
MTL::Size get_2d_grid_dims(const std::vector< int > &shape, const std::vector< size_t > &strides)
+
mlx::core::get_primitive_string
std::string get_primitive_string(Primitive *primitive)
+
mlx::core::make_string
NS::String * make_string(std::ostringstream &os)
Definition utils.h:45
+
mlx::core::type_to_name
std::string type_to_name(const array &a)
+
mlx::core::metal::CommandEncoder
Definition device.h:20
diff --git a/docs/build/html/class_m_p_s_1_1_kernel-members.html b/docs/build/html/class_m_p_s_1_1_kernel-members.html deleted file mode 100644 index f12b0a0507..0000000000 --- a/docs/build/html/class_m_p_s_1_1_kernel-members.html +++ /dev/null @@ -1,92 +0,0 @@ - - - - - - - -MLX: Member List - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
MPS::Kernel Member List
-
-
- -

This is the complete list of members for MPS::Kernel, including all inherited members.

- - - -
device() constMPS::Kernel
label() constMPS::Kernel
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_kernel.html b/docs/build/html/class_m_p_s_1_1_kernel.html deleted file mode 100644 index de2a0c5849..0000000000 --- a/docs/build/html/class_m_p_s_1_1_kernel.html +++ /dev/null @@ -1,144 +0,0 @@ - - - - - - - -MLX: MPS::Kernel Class Reference - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
-Public Member Functions | -List of all members
-
MPS::Kernel Class Reference
-
-
- -

#include <gemm.h>

-
-Inheritance diagram for MPS::Kernel:
-
-
- -
- - - - - - -

-Public Member Functions

NS::String * label () const
 
MTL::Device * device () const
 
-

Member Function Documentation

- -

◆ device()

- -
-
- - - - - - - -
_MTL_INLINE MTL::Device * MPS::Kernel::device () const
-
- -
-
- -

◆ label()

- -
-
- - - - - - - -
_MTL_INLINE NS::String * MPS::Kernel::label () const
-
- -
-
-
The documentation for this class was generated from the following file:
    -
  • mlx/backend/metal/mps/gemm.h
    • -
-
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_kernel.png b/docs/build/html/class_m_p_s_1_1_kernel.png deleted file mode 100644 index 0ed62c014d946e0ba1bb07f7be8739e94bc15dbd..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 656 zcmV;B0&o3^P)vTJr#LVva2S`&=-}Ys|Ns9r%~qrU000SeQchC<|NsC0|NsC0Hv*f~0006T zNkl3iy_{@y)J&m?)QLzDUCYYUR*>Xdc4ce8T(_rT9_Ajw@FQr7j}s8P-# zUE*4zJFh4+^^oPsOsqF;I~Xs+a+2;EDQi*}cQ(R(Xm)kF8<)E_QN7i@E4K5pUcN3a zTbJgTLtSoB4%5z+Gj|R><(?OJG}}ErJ+3ZmOI1e}+)VFM`u3~&@y9RKQ=P3iN6PK) zaaP0qJF(pH;`ca;Wh=F9${%HXt3rKU?t1!JUHack-#xbfT6el>XJmpn_IS(hP@gY* zoFqS@2>+j@eEkznlCSmQSo*(LlKhoEpK+4>tCEr=W0aI68KWnqh{)9i0Ks|yz~yxd z%E#jln75#OLf!!L9+cVR01&051b`SNB>==IDFGlxNeKWkN=g8TQBndxjFJ)nVw98s z5Tm37fEXnu0K_OM0U$>IFQtgcI_bIFWbDZ06zAs}pIQ05< zV17ZlESQGM{Si04qQGdEJ2^nNYl!)`sdy}?b7+fF*E$5mC&R#K{} q#waOORb!Nts;V)%4W)?4)%6eBqshFLMxG!50000 - - - - - - -MLX: Member List - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
MPS::Matrix Member List
-
-
- -

This is the complete list of members for MPS::Matrix, including all inherited members.

- - - - -
alloc()MPS::Matrixstatic
init(MTL::Buffer *buffer, MatrixDescriptor *descriptor)MPS::Matrix
init(const MTL::Buffer *buffer, MatrixDescriptor *descriptor)MPS::Matrix
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_matrix.html b/docs/build/html/class_m_p_s_1_1_matrix.html deleted file mode 100644 index 8fc4607bd6..0000000000 --- a/docs/build/html/class_m_p_s_1_1_matrix.html +++ /dev/null @@ -1,183 +0,0 @@ - - - - - - - -MLX: MPS::Matrix Class Reference - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
-Public Member Functions | -Static Public Member Functions | -List of all members
-
MPS::Matrix Class Reference
-
-
- -

#include <gemm.h>

-
-Inheritance diagram for MPS::Matrix:
-
-
- -
- - - - - - -

-Public Member Functions

Matrix * init (MTL::Buffer *buffer, MatrixDescriptor *descriptor)
 
Matrix * init (const MTL::Buffer *buffer, MatrixDescriptor *descriptor)
 
- - - -

-Static Public Member Functions

static class Matrix * alloc ()
 
-

Member Function Documentation

- -

◆ alloc()

- -
-
- - - - - -
- - - - - - - -
_MTL_INLINE Matrix * MPS::Matrix::alloc ()
- 		-static
-
- -
-
- -

◆ init() [1/2]

- -
-
- - - - - - - - - - - -
_MTL_INLINE Matrix * MPS::Matrix::init (const MTL::Buffer * buffer,
MatrixDescriptor * descriptor )
-
- -
-
- -

◆ init() [2/2]

- -
-
- - - - - - - - - - - -
_MTL_INLINE Matrix * MPS::Matrix::init (MTL::Buffer * buffer,
MatrixDescriptor * descriptor )
-
- -
-
-
The documentation for this class was generated from the following file:
    -
  • mlx/backend/metal/mps/gemm.h
    • -
-
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_matrix.png b/docs/build/html/class_m_p_s_1_1_matrix.png deleted file mode 100644 index a14d0bf98ab9ec89bc7677aba5189d2cf8871b16..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 665 zcmeAS@N?(olHy`uVBq!ia0vp^i-9~~^1IA5 zhU?xxts81ouD^b!S0}pqk9E7%mFer hwYP+;%euAlbBm9?DLuGqJ1`M3c)I$ztaD0e0swvcJA42D diff --git a/docs/build/html/class_m_p_s_1_1_matrix_descriptor-members.html b/docs/build/html/class_m_p_s_1_1_matrix_descriptor-members.html deleted file mode 100644 index dee3102efc..0000000000 --- a/docs/build/html/class_m_p_s_1_1_matrix_descriptor-members.html +++ /dev/null @@ -1,93 +0,0 @@ - - - - - - - -MLX: Member List - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
MPS::MatrixDescriptor Member List
-
-
- -

This is the complete list of members for MPS::MatrixDescriptor, including all inherited members.

- - - - -
matrixDescriptor(NS::UInteger rows, NS::UInteger columns, NS::UInteger rowBytes, NS::UInteger dataType)MPS::MatrixDescriptorstatic
matrixDescriptor(NS::UInteger rows, NS::UInteger columns, NS::UInteger matrices, NS::UInteger rowBytes, NS::UInteger matrixBytes, NS::UInteger dataType)MPS::MatrixDescriptorstatic
rows() constMPS::MatrixDescriptor
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_matrix_descriptor.html b/docs/build/html/class_m_p_s_1_1_matrix_descriptor.html deleted file mode 100644 index 1b88da36a8..0000000000 --- a/docs/build/html/class_m_p_s_1_1_matrix_descriptor.html +++ /dev/null @@ -1,221 +0,0 @@ - - - - - - - -MLX: MPS::MatrixDescriptor Class Reference - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
-Public Member Functions | -Static Public Member Functions | -List of all members
-
MPS::MatrixDescriptor Class Reference
-
-
- -

#include <gemm.h>

-
-Inheritance diagram for MPS::MatrixDescriptor:
-
-
- -
- - - - -

-Public Member Functions

NS::UInteger rows () const
 
- - - - - -

-Static Public Member Functions

static class MatrixDescriptor * matrixDescriptor (NS::UInteger rows, NS::UInteger columns, NS::UInteger rowBytes, NS::UInteger dataType)
 
static class MatrixDescriptor * matrixDescriptor (NS::UInteger rows, NS::UInteger columns, NS::UInteger matrices, NS::UInteger rowBytes, NS::UInteger matrixBytes, NS::UInteger dataType)
 
-

Member Function Documentation

- -

◆ matrixDescriptor() [1/2]

- -
-
- - - - - -
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
_MTL_INLINE MatrixDescriptor * MPS::MatrixDescriptor::matrixDescriptor (NS::UInteger rows,
NS::UInteger columns,
NS::UInteger matrices,
NS::UInteger rowBytes,
NS::UInteger matrixBytes,
NS::UInteger dataType )
- 		-static
-
- -
-
- -

◆ matrixDescriptor() [2/2]

- -
-
- - - - - -
- - - - - - - - - - - - - - - - - - - - - -
_MTL_INLINE MatrixDescriptor * MPS::MatrixDescriptor::matrixDescriptor (NS::UInteger rows,
NS::UInteger columns,
NS::UInteger rowBytes,
NS::UInteger dataType )
- 		-static
-
- -
-
- -

◆ rows()

- -
-
- - - - - - - -
_MTL_INLINE NS::UInteger MPS::MatrixDescriptor::rows () const
-
- -
-
-
The documentation for this class was generated from the following file:
    -
  • mlx/backend/metal/mps/gemm.h
    • -
-
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_matrix_descriptor.png b/docs/build/html/class_m_p_s_1_1_matrix_descriptor.png deleted file mode 100644 index 661b7f470ecb13762aefc4337df516e94f33e8bc..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 813 zcmeAS@N?(olHy`uVBq!ia0vp^CxAGBgBeI>Z$8xoq@)9ULR|m<{|{uoc=NTi|Il&^ z1I+@7>1SR%c<=xyZhAIs2~du+B*-tA0mugfbEer>7#Ns#c)B=-R4~4sdwbey1p$}t z=^=mri}wfW-EHW6Jo8Lq#F`~42idAGi^mH(aTs$ns{G$4#^Kqwgw<)1vSzc);m^Nv z7_S}ruNt{epI`B=j@#KY_wqbGg}hCg^kzlFo5y9Ii(~7xvo5+m&fI*q+-yHjk;~$^ zggrNGZ=G2byQNq7eD+rj?oIl3dLJjt&%AsBCzIL~p3;TYkSMb+v@$H>5*ZO8c?Mk^jwQbBF--#XO^9c3Yn|`xGL@oM_ zYRb9)(|7MWSk5WkHzRDh!p2u;-$$@A5!SRK#n zC8AF@RPnL9NJ=*w$NypSw7R~&`3uiH_xXAsH?QvyJE^eNv!1W}?IiK((v!ZpO%Kj4 zKj5nLQ*Qq`1zyKZ@x{l_MO}W!>shr}T0!+{tWqlT`@F(s%t2FwcxS45=9lcfaXHia zcaXA~jNw{8$47xPZvN*C65Vb7p|2$9H0!%^#~Qa1hp70RT~8mbaSPf~C#$m8-RfV? z=Z(KDCd9=2T{rd6EbskyCM+?RdV99+_H0%qPx%Wb3X@8_;r`mv*dlO< - - - - - - -MLX: Member List - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
MPS::MatrixMultiplication Member List
-
-
- -

This is the complete list of members for MPS::MatrixMultiplication, including all inherited members.

- - - - - - - - - -
alloc()MPS::MatrixMultiplicationstatic
encodeToCommandBuffer(MTL::CommandBuffer *commandBuffer, Matrix *leftMatrix, Matrix *rightMatrix, Matrix *resultMatrix)MPS::MatrixMultiplication
init(MTL::Device *device, bool transposeLeft, bool transposeRight, NS::UInteger resultRows, NS::UInteger resultColumns, NS::UInteger interiorColumns, double alpha, double beta)MPS::MatrixMultiplication
setBatchSize(NS::UInteger batchSize)MPS::MatrixMultiplication
setBatchStart(NS::UInteger batchStart)MPS::MatrixMultiplication
setLeftMatrixOrigin(MTL::Origin origin)MPS::MatrixMultiplication
setResultMatrixOrigin(MTL::Origin origin)MPS::MatrixMultiplication
setRightMatrixOrigin(MTL::Origin origin)MPS::MatrixMultiplication
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_matrix_multiplication.html b/docs/build/html/class_m_p_s_1_1_matrix_multiplication.html deleted file mode 100644 index f751a2bd4a..0000000000 --- a/docs/build/html/class_m_p_s_1_1_matrix_multiplication.html +++ /dev/null @@ -1,318 +0,0 @@ - - - - - - - -MLX: MPS::MatrixMultiplication Class Reference - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
-Public Member Functions | -Static Public Member Functions | -List of all members
-
MPS::MatrixMultiplication Class Reference
-
-
- -

#include <gemm.h>

-
-Inheritance diagram for MPS::MatrixMultiplication:
-
-
- -
- - - - - - - - - - - - - - - - -

-Public Member Functions

MatrixMultiplication * init (MTL::Device *device, bool transposeLeft, bool transposeRight, NS::UInteger resultRows, NS::UInteger resultColumns, NS::UInteger interiorColumns, double alpha, double beta)
 
void encodeToCommandBuffer (MTL::CommandBuffer *commandBuffer, Matrix *leftMatrix, Matrix *rightMatrix, Matrix *resultMatrix)
 
void setLeftMatrixOrigin (MTL::Origin origin)
 
void setRightMatrixOrigin (MTL::Origin origin)
 
void setResultMatrixOrigin (MTL::Origin origin)
 
void setBatchStart (NS::UInteger batchStart)
 
void setBatchSize (NS::UInteger batchSize)
 
- - - -

-Static Public Member Functions

static class MatrixMultiplication * alloc ()
 
-

Member Function Documentation

- -

◆ alloc()

- -
-
- - - - - -
- - - - - - - -
_MTL_INLINE MatrixMultiplication * MPS::MatrixMultiplication::alloc ()
- 		-static
-
- -
-
- -

◆ encodeToCommandBuffer()

- -
-
- - - - - - - - - - - - - - - - - - - - - -
_MTL_INLINE void MPS::MatrixMultiplication::encodeToCommandBuffer (MTL::CommandBuffer * commandBuffer,
Matrix * leftMatrix,
Matrix * rightMatrix,
Matrix * resultMatrix )
-
- -
-
- -

◆ init()

- -
-
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
_MTL_INLINE MatrixMultiplication * MPS::MatrixMultiplication::init (MTL::Device * device,
bool transposeLeft,
bool transposeRight,
NS::UInteger resultRows,
NS::UInteger resultColumns,
NS::UInteger interiorColumns,
double alpha,
double beta )
-
- -
-
- -

◆ setBatchSize()

- -
-
- - - - - - - -
_MTL_INLINE void MPS::MatrixMultiplication::setBatchSize (NS::UInteger batchSize)
-
- -
-
- -

◆ setBatchStart()

- -
-
- - - - - - - -
_MTL_INLINE void MPS::MatrixMultiplication::setBatchStart (NS::UInteger batchStart)
-
- -
-
- -

◆ setLeftMatrixOrigin()

- -
-
- - - - - - - -
_MTL_INLINE void MPS::MatrixMultiplication::setLeftMatrixOrigin (MTL::Origin origin)
-
- -
-
- -

◆ setResultMatrixOrigin()

- -
-
- - - - - - - -
_MTL_INLINE void MPS::MatrixMultiplication::setResultMatrixOrigin (MTL::Origin origin)
-
- -
-
- -

◆ setRightMatrixOrigin()

- -
-
- - - - - - - -
_MTL_INLINE void MPS::MatrixMultiplication::setRightMatrixOrigin (MTL::Origin origin)
-
- -
-
-
The documentation for this class was generated from the following file:
    -
  • mlx/backend/metal/mps/gemm.h
    • -
-
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_matrix_multiplication.png b/docs/build/html/class_m_p_s_1_1_matrix_multiplication.png deleted file mode 100644 index 7da3f69a96ea78b09a467274e22e6f2ac0159a49..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 1041 zcmeAS@N?(olHy`uVBq!ia0y~yV3Y>312~w0q@1)O2au8u@CkAK|NlRb`Qpvj(*8rs zEetdZB&MHv@!-J&pt$MTuq8k_&XOR%U-KE zqV+HT{U2W=Y!G3_JNstKUgOYJA0?Xv582&#Bq}KA>T-mU#ZyehGBD5MQZN$UygQQr z{QvV@ch~c$pD&)Y=b2H;?iq51-)~IV%KKa;xVZMQah%lj(#?16md%{BZ~uu|ch<~* zav|r0z|*#m$7ja(|Gjj!?7zg8-I3GI?`pf{745=Ea1F5R(qH~aSZ>f?JjwmtW;Z1y?Z zBJ=57vefTo)A#IMl=!Ay=W~|tSGAM2SEQdmyI{LS>d@zJH`MKpeLJ9SzPWv;f-k?{ zWr@3UdTT3scZz-IHMlZ+t;6@JcS@JO3pP(SJiqyzdAfP^3GcjbCgOS9WDWMc3KQj9 zy{6M<$H%P|9aew-UE5_SY|a-~p8WmF#a`LpGoEJM{gRt?ch8pPF;695pFJ77QmkK? zZ_{>#V(Yc$$IhPo5x?3zbG}Sg;kswlYvp&fFW8ds?EMvyeR(^L&2^lQv>X&2%6>;k{Wu6D|L$nFs5Z!D5=fbn@}^ zyE5Fa_tJ0oeNt6ha>wuWS^dJx*Zx@i_&a-3UEsMpHu~?v*6j;E-TG(luf{x;mu~<1 z1E)!QewqTrf5glueOa$*KIsWEuG|3*pR|9>7oUfRF0!cC17= diff --git a/docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication-members.html b/docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication-members.html deleted file mode 100644 index abb569f03f..0000000000 --- a/docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication-members.html +++ /dev/null @@ -1,93 +0,0 @@ - - - - - - - -MLX: Member List - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
MPS::MatrixVectorMultiplication Member List
-
-
- -

This is the complete list of members for MPS::MatrixVectorMultiplication, including all inherited members.

- - - - -
alloc()MPS::MatrixVectorMultiplicationstatic
encodeToCommandBuffer(MTL::CommandBuffer *commandBuffer, Matrix *inputMatrix, Vector *inputVector, Vector *resultVector)MPS::MatrixVectorMultiplication
init(MTL::Device *device, bool transpose, NS::UInteger rows, NS::UInteger columns, double alpha, double beta)MPS::MatrixVectorMultiplication
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication.html b/docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication.html deleted file mode 100644 index 074f379d7d..0000000000 --- a/docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication.html +++ /dev/null @@ -1,213 +0,0 @@ - - - - - - - -MLX: MPS::MatrixVectorMultiplication Class Reference - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
-Public Member Functions | -Static Public Member Functions | -List of all members
-
MPS::MatrixVectorMultiplication Class Reference
-
-
- -

#include <gemm.h>

-
-Inheritance diagram for MPS::MatrixVectorMultiplication:
-
-
- -
- - - - - - -

-Public Member Functions

MatrixVectorMultiplication * init (MTL::Device *device, bool transpose, NS::UInteger rows, NS::UInteger columns, double alpha, double beta)
 
void encodeToCommandBuffer (MTL::CommandBuffer *commandBuffer, Matrix *inputMatrix, Vector *inputVector, Vector *resultVector)
 
- - - -

-Static Public Member Functions

static class MatrixVectorMultiplication * alloc ()
 
-

Member Function Documentation

- -

◆ alloc()

- -
-
- - - - - -
- - - - - - - -
_MTL_INLINE MatrixVectorMultiplication * MPS::MatrixVectorMultiplication::alloc ()
- 		-static
-
- -
-
- -

◆ encodeToCommandBuffer()

- -
-
- - - - - - - - - - - - - - - - - - - - - -
_MTL_INLINE void MPS::MatrixVectorMultiplication::encodeToCommandBuffer (MTL::CommandBuffer * commandBuffer,
Matrix * inputMatrix,
Vector * inputVector,
Vector * resultVector )
-
- -
-
- -

◆ init()

- -
-
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
_MTL_INLINE MatrixVectorMultiplication * MPS::MatrixVectorMultiplication::init (MTL::Device * device,
bool transpose,
NS::UInteger rows,
NS::UInteger columns,
double alpha,
double beta )
-
- -
-
-
The documentation for this class was generated from the following file:
    -
  • mlx/backend/metal/mps/gemm.h
    • -
-
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication.png b/docs/build/html/class_m_p_s_1_1_matrix_vector_multiplication.png deleted file mode 100644 index 10e54c8bd6ac07f8af73d45b3c9f0584059b3703..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 1145 zcmeAS@N?(olHy`uVBq!ia0y~yU~~ks12~w0#Rfbr z?&fnV|Eu3qkjV{JttOS37aa?B-}!(4xA*_rie2`TpKR6h{U(1`=(Ep#u9J4lIxk!_YfGP$=Tf}* z^zt%!p*!*NcYXXmGv2mq{O_UC{!!)NiHixsx!=~;tn;t%EU2$HvEs0ED(pTkHnrdU z)|r2dwqLi=EH zwzJDGS2O*s2q z^@vvqE(TVQISb}z9Y46d<8@o(_l%##O~z|?2Ke1BZkOepH@AR;^S)=*Dc;h8e0^JO zs}S>AuYw%ce^>rJHT?JeOkv`tx&wDF2D9+Ti?dXJ{E&Eg)m5R5x$C(P?wHIIcKYw~ zBm;{dQ&y}~>6^}S(4afK=^>}Pez@TcZfB1VD|Py`9!0SG-;>(FdsF1f!J`)=e)W8L zyoHaA<$qN7`d151pIJKjM_ftCs>CXTqJ!q&ADyi6{xsvk1-Izfey-H$wB+)u-SYN5 z%V#F%oxho}{`%e-D)$$a|5iWgIX~8D>(TE*Ha~Mze)`oGoxH<$U#3}C*5i}(*R|RE z?7KZb)rSeHob2gR@JtI~@ibbj?UFc>GEYsyQYvSW@?PDm_xucOekhx5 zuYbFK|3>%sg)jfl`nluMvnSuKwWvS`Q7m^Y{xZgh5TMk zPyTc!P+IM$;7to_Pp{XyEY*LdgWej7uV_&^>BWQKMY|?h#0jw+*7>)TW%-k6&UH&q zrf)O0U*Fulc;lCuPmX8G{&DBn_NP<9#wz*zlbOIoV7_}pz`c5_8?j#>9p~RC{@l!c#7$uXYTQ6XH62@)9zAV; R7Fb9yc)I$ztaD0e0swr#1StRj diff --git a/docs/build/html/class_m_p_s_1_1_vector-members.html b/docs/build/html/class_m_p_s_1_1_vector-members.html deleted file mode 100644 index 31e25f93d1..0000000000 --- a/docs/build/html/class_m_p_s_1_1_vector-members.html +++ /dev/null @@ -1,93 +0,0 @@ - - - - - - - -MLX: Member List - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
MPS::Vector Member List
-
-
- -

This is the complete list of members for MPS::Vector, including all inherited members.

- - - - -
alloc()MPS::Vectorstatic
init(MTL::Buffer *buffer, VectorDescriptor *descriptor)MPS::Vector
init(const MTL::Buffer *buffer, VectorDescriptor *descriptor)MPS::Vector
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_vector.html b/docs/build/html/class_m_p_s_1_1_vector.html deleted file mode 100644 index 9e1039eadd..0000000000 --- a/docs/build/html/class_m_p_s_1_1_vector.html +++ /dev/null @@ -1,183 +0,0 @@ - - - - - - - -MLX: MPS::Vector Class Reference - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
-Public Member Functions | -Static Public Member Functions | -List of all members
-
MPS::Vector Class Reference
-
-
- -

#include <gemm.h>

-
-Inheritance diagram for MPS::Vector:
-
-
- -
- - - - - - -

-Public Member Functions

Vector * init (MTL::Buffer *buffer, VectorDescriptor *descriptor)
 
Vector * init (const MTL::Buffer *buffer, VectorDescriptor *descriptor)
 
- - - -

-Static Public Member Functions

static class Vector * alloc ()
 
-

Member Function Documentation

- -

◆ alloc()

- -
-
- - - - - -
- - - - - - - -
_MTL_INLINE Vector * MPS::Vector::alloc ()
- 		-static
-
- -
-
- -

◆ init() [1/2]

- -
-
- - - - - - - - - - - -
_MTL_INLINE Vector * MPS::Vector::init (const MTL::Buffer * buffer,
VectorDescriptor * descriptor )
-
- -
-
- -

◆ init() [2/2]

- -
-
- - - - - - - - - - - -
_MTL_INLINE Vector * MPS::Vector::init (MTL::Buffer * buffer,
VectorDescriptor * descriptor )
-
- -
-
-
The documentation for this class was generated from the following file:
    -
  • mlx/backend/metal/mps/gemm.h
    • -
-
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_vector.png b/docs/build/html/class_m_p_s_1_1_vector.png deleted file mode 100644 index e7adea3626eca42f23064dae57f6cb4ab2cb040c..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 658 zcmeAS@N?(olHy`uVBq!ia0vp^%YZn5gBeIJnfuQXNJ$6ygt-3y{~ySF@#br3|Doj; z2ATyD)6cwk@ZbSZ-1KbN5}+JsNswPK1CS2}=1jA%FfcI5db&7&J22n3nwf4dWEupw+8Nc>45vby~w`%)Bcv;dyWF6_t}3L6cP$ub36Rxo&;y z)w=iZu2oO^GgH2OORdMY^YPPiXPqzmskL_YyEUGlgu+d?d+hUDG3)$-yoq-o|DLn) z`;_H(-p3qQ*wCMG`bz8HT`GUmujoAACHY%o`?5amySX3gLak3_-Ti5CXHMOJy@NG* z*8=>CikDg5etkB0=MSG{ecAiF%2cl$vbFoY?6qF|yX~{|Q@6i=v~1tioOZ*EbKWbY zO=ILc)2_Z@DpHwcdiU9^{Z?Nhr}_2$U4CcIpPzgG-+$x1LUiNm6YF+MKRINfd3)Bk zJl6P^DmOp>+#Yb^*Ziw8T1_hFKRnF%mLE0A=J<2|xhndTXQ@cKW=%>F>0UPT=3B;w z3^|5^`hCuyKmOo4|NcUZ&i;d+y-hY>VpQn6!qT7-B+RhL!?bc3Xc^;B+76wJYbgD_8q&e!Ap-lGDZu-8p$xgiMxmkbV zvYS8uSD&5!SjHrF - - - - - - -MLX: Member List - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
MPS::VectorDescriptor Member List
-
-
- -

This is the complete list of members for MPS::VectorDescriptor, including all inherited members.

- - - -
vectorDescriptor(NS::UInteger length, NS::UInteger dataType)MPS::VectorDescriptorstatic
vectorDescriptor(NS::UInteger length, NS::UInteger vectors, NS::UInteger vectorBytes, NS::UInteger dataType)MPS::VectorDescriptorstatic
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_vector_descriptor.html b/docs/build/html/class_m_p_s_1_1_vector_descriptor.html deleted file mode 100644 index 681f3428ea..0000000000 --- a/docs/build/html/class_m_p_s_1_1_vector_descriptor.html +++ /dev/null @@ -1,178 +0,0 @@ - - - - - - - -MLX: MPS::VectorDescriptor Class Reference - - - - - - - - - - - -
-
- - - - - - -
-
MLX -
-
-
- - - - - - - - -
-
- - -
-
-
-
-
-
Loading...
-
Searching...
-
No Matches
-
-
-
-
- - -
-
-
-Static Public Member Functions | -List of all members
-
MPS::VectorDescriptor Class Reference
-
-
- -

#include <gemm.h>

-
-Inheritance diagram for MPS::VectorDescriptor:
-
-
- -
- - - - - - -

-Static Public Member Functions

static class VectorDescriptor * vectorDescriptor (NS::UInteger length, NS::UInteger dataType)
 
static class VectorDescriptor * vectorDescriptor (NS::UInteger length, NS::UInteger vectors, NS::UInteger vectorBytes, NS::UInteger dataType)
 
-

Member Function Documentation

- -

◆ vectorDescriptor() [1/2]

- -
-
- - - - - -
- - - - - - - - - - - -
_MTL_INLINE VectorDescriptor * MPS::VectorDescriptor::vectorDescriptor (NS::UInteger length,
NS::UInteger dataType )
- 		-static
-
- -
-
- -

◆ vectorDescriptor() [2/2]

- -
-
- - - - - -
- - - - - - - - - - - - - - - - - - - - - -
_MTL_INLINE VectorDescriptor * MPS::VectorDescriptor::vectorDescriptor (NS::UInteger length,
NS::UInteger vectors,
NS::UInteger vectorBytes,
NS::UInteger dataType )
- 		-static
-
- -
-
-
The documentation for this class was generated from the following file:
    -
  • mlx/backend/metal/mps/gemm.h
    • -
-
- -
-Generated by doxygen 1.10.0 -
- - diff --git a/docs/build/html/class_m_p_s_1_1_vector_descriptor.png b/docs/build/html/class_m_p_s_1_1_vector_descriptor.png deleted file mode 100644 index 00b2efd0e245b4a7485a5d74f370f82473449fd6..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 794 zcmV+#1LgdQP)vTJr#LVva2S`&=-}Ys|Ns9r%~qrU000SeQchC<|NsC0|NsC0Hv*f~0007| zNklN4wz)Gb z=|vn@4=FdvjE-X)pDT-E61&K9I+HKib0b7g3$o5nbC3*HW+J27QJ(gC8_JAp<$gXm zgYed?>8dO~{2l3i$8r>b6vMqNIU-IhHW@EiY%jhLPky(|1@%(yJy5N&3@dAxVE)SV9P43<1FR z^#XuFG!V<{H4e-`EDOpoFe9A8ZG@+ zF}B}FUl^QxRWH(5NouP0C`rBQWjWTmvP32`VJFK6i~Ko@ojM*@>Tp}3v?_<} z7E99@DoHo$jj8SNfn`k<=BwD`R^=s9!C}f`R-hd#cx@v=sDMR1eb#!yMRk;8lc@q#9cBz;XIu}IP*O%_S=r^zBo{
A
-
Abs
Abs (mlx::core)
Abs (mlx::core::detail)
AccumHelper (mlx::steel)
Add
Add (mlx::core)
Add (mlx::core::detail)
add_vec (pocketfft::detail)
add_vec< cmplx< T > > (pocketfft::detail)
AddMM (mlx::core)
aligned_allocator (pocketfft::detail::threading)
AllGather (mlx::core::distributed)
Allocator (mlx::core::allocator)
AllReduce (mlx::core::distributed)
And
Arange (mlx::core)
ArcCos
ArcCos (mlx::core)
ArcCos (mlx::core::detail)
ArcCosh
ArcCosh (mlx::core)
ArcCosh (mlx::core::detail)
ArcSin
ArcSin (mlx::core)
ArcSin (mlx::core::detail)
ArcSinh
ArcSinh (mlx::core)
ArcSinh (mlx::core::detail)
ArcTan
ArcTan (mlx::core)
ArcTan (mlx::core::detail)
ArcTan2
ArcTan2 (mlx::core)
ArcTan2 (mlx::core::detail)
ArcTanh
ArcTanh (mlx::core)
ArcTanh (mlx::core::detail)
ArgPartition (mlx::core)
ArgReduce (mlx::core)
ArgSort (mlx::core)
arr (pocketfft::detail)
arr_info (pocketfft::detail)
array (mlx::core)
array::ArrayIterator (mlx::core)
AsStrided (mlx::core)
AsType (mlx::core)
+
Abs
Abs (mlx::core)
Abs (mlx::core::detail)
AccumHelper (mlx::steel)
Add
Add (mlx::core)
Add (mlx::core::detail)
add_vec (pocketfft::detail)
add_vec< cmplx< T > > (pocketfft::detail)
AddMM (mlx::core)
AffineQuantize (mlx::core::fast)
aligned_allocator (pocketfft::detail::threading)
AllGather (mlx::core::distributed)
Allocator (mlx::core::allocator)
AllReduce (mlx::core::distributed)
And
Arange (mlx::core)
ArcCos
ArcCos (mlx::core)
ArcCos (mlx::core::detail)
ArcCosh
ArcCosh (mlx::core)
ArcCosh (mlx::core::detail)
ArcSin
ArcSin (mlx::core)
ArcSin (mlx::core::detail)
ArcSinh
ArcSinh (mlx::core)
ArcSinh (mlx::core::detail)
ArcTan
ArcTan (mlx::core)
ArcTan (mlx::core::detail)
ArcTan2
ArcTan2 (mlx::core)
ArcTan2 (mlx::core::detail)
ArcTanh
ArcTanh (mlx::core)
ArcTanh (mlx::core::detail)
ArgPartition (mlx::core)
ArgReduce (mlx::core)
ArgSort (mlx::core)
arr (pocketfft::detail)
arr_info (pocketfft::detail)
array (mlx::core)
array::ArrayIterator (mlx::core)
AsStrided (mlx::core)
AsType (mlx::core)
B
_MLX_BFloat16::bits_to_bfloat_struct
BitwiseAnd
BitwiseAnd (mlx::core::detail)
BitwiseBinary (mlx::core)
BitwiseOr
BitwiseOr (mlx::core::detail)
BitwiseXor
BitwiseXor (mlx::core::detail)
BlockLoader (mlx::steel)
BlockMaskedMM (mlx::core)
BlockMergeSort
BlockMMA (mlx::steel)
BlockSwizzle (mlx::steel)
bool4_or_uint
Broadcast (mlx::core)
Buffer (mlx::core::allocator)
@@ -96,7 +96,7 @@ $(function() {
FFT (mlx::core)
fftblue (pocketfft::detail)
FileReader (mlx::core::io)
FileWriter (mlx::core::io)
array::Flags (mlx::core)
Floor
Floor (mlx::core::detail)
Floor (mlx::core)
FloorDivide
Full (mlx::core)
G
-
Gather (mlx::core)
GatherMM (mlx::core)
GatherQMM (mlx::core)
GEMMAddMMParams (mlx::steel)
GEMMKernel (mlx::steel)
GEMMParams (mlx::steel)
GEMMSpiltKParams (mlx::steel)
Greater
Greater (mlx::core::detail)
Greater (mlx::core)
GreaterEqual
GreaterEqual (mlx::core::detail)
GreaterEqual (mlx::core)
Group (mlx::core::distributed)
+
Gather (mlx::core)
GatherMM (mlx::core)
GatherQMM (mlx::core)
GEMMAddMMParams (mlx::steel)
GEMMKernel (mlx::steel)
GEMMParams (mlx::steel)
GEMMSpiltKParams (mlx::steel)
GEMVKernel
GEMVTKernel
Greater
Greater (mlx::core::detail)
Greater (mlx::core)
GreaterEqual
GreaterEqual (mlx::core::detail)
GreaterEqual (mlx::core)
Group (mlx::core::distributed)
H
Hadamard (mlx::core)
@@ -105,13 +105,13 @@ $(function() {
ImplicitGemmConv2DParams (mlx::steel)
Indices
IntOrFloat (mlx::core::detail)
InTracing (mlx::core::detail)
Inverse (mlx::core)
K
-
Kernel (MPS)
KernelMergeSort
KernelMultiBlockMergeSort
KeySequence (mlx::core::random)
+
KernelMergeSort
KernelMultiBlockMergeSort
KeySequence (mlx::core::random)
L
latch (pocketfft::detail::threading)
LayerNorm (mlx::core::fast)
LayerNormVJP (mlx::core::fast)
LeftShift
LeftShift (mlx::core::detail)
Less
Less (mlx::core::detail)
Less (mlx::core)
LessEqual
LessEqual (mlx::core::detail)
LessEqual (mlx::core)
LessThan
Limits
Limits< bfloat16_t >
Limits< bool >
Limits< float >
Limits< half >
Limits< int16_t >
Limits< int32_t >
Limits< int64_t >
Limits< int8_t >
Limits< uint16_t >
Limits< uint32_t >
Limits< uint64_t >
Limits< uint8_t >
Load (mlx::core)
Log
Log (mlx::core::detail)
Log (mlx::core)
Log10
Log10 (mlx::core::detail)
Log1p
Log1p (mlx::core::detail)
Log1p (mlx::core)
Log2
Log2 (mlx::core::detail)
LogAddExp
LogAddExp (mlx::core::detail)
LogAddExp (mlx::core)
LogicalAnd
LogicalAnd (mlx::core::detail)
LogicalAnd (mlx::core)
LogicalNot
LogicalNot (mlx::core::detail)
LogicalNot (mlx::core)
LogicalOr
LogicalOr (mlx::core::detail)
LogicalOr (mlx::core)
LoopAlignment (mlx::steel)
M
-
Matmul (mlx::core)
Matrix (MPS)
MatrixDescriptor (MPS)
MatrixMultiplication (MPS)
MatrixVectorMultiplication (MPS)
Max
Maximum
Maximum (mlx::core::detail)
Maximum (mlx::core)
MetalAllocator (mlx::core::metal)
Min
Minimum
Minimum (mlx::core::detail)
Minimum (mlx::core)
mlx_atomic
mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >
MLXConvParams
MLXFastAttentionParams
MLXScaledDotProductAttentionParams
multi_iter (pocketfft::detail)
Multiply (mlx::core::detail)
Multiply (mlx::core)
Multiply
+
Matmul (mlx::core)
Max
Maximum
Maximum (mlx::core::detail)
Maximum (mlx::core)
MetalAllocator (mlx::core::metal)
Min
Minimum
Minimum (mlx::core::detail)
Minimum (mlx::core)
mlx_atomic
mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >
MLXConvParams
MLXFastAttentionParams
MLXScaledDotProductAttentionParams
multi_iter (pocketfft::detail)
Multiply (mlx::core::detail)
Multiply (mlx::core)
Multiply
N
NaNEqual (mlx::core::detail)
NaNEqual
ndarr (pocketfft::detail)
Negative (mlx::core::detail)
Negative (mlx::core)
Negative
NodeNamer (mlx::core)
None
NotEqual (mlx::core::detail)
NotEqual (mlx::core)
NotEqual
NumberOfElements (mlx::core)
@@ -138,7 +138,7 @@ $(function() {
UnaryPrimitive (mlx::core)
Uniform (mlx::core)
util (pocketfft::detail)
V
-
Vector (MPS)
VectorDescriptor (MPS)
View (mlx::core)
VLEN (pocketfft::detail)
VTYPE (pocketfft::detail)
+
View (mlx::core)
VLEN (pocketfft::detail)
VTYPE (pocketfft::detail)
W
Writer (mlx::core::io)
diff --git a/docs/build/html/classmlx_1_1core_1_1_inverse-members.html b/docs/build/html/classmlx_1_1core_1_1_inverse-members.html index 1fb2c948d4..60cb63ce88 100644 --- a/docs/build/html/classmlx_1_1core_1_1_inverse-members.html +++ b/docs/build/html/classmlx_1_1core_1_1_inverse-members.html @@ -86,7 +86,7 @@ $(function() { mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual eval_gpu(const std::vector< array > &inputs, array &output) overridemlx::core::Inversevirtual mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) overridemlx::core::UnaryPrimitiveinlinevirtual - Inverse(Stream stream)mlx::core::Inverseinlineexplicit + Inverse(Stream stream, bool tri, bool upper)mlx::core::Inverseinlineexplicit is_equivalent(const Primitive &other) constmlx::core::Primitiveinlinevirtual jvp(const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums)mlx::core::Primitivevirtual operator=(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive diff --git a/docs/build/html/classmlx_1_1core_1_1_inverse.html b/docs/build/html/classmlx_1_1core_1_1_inverse.html index ec23a10511..6785369533 100644 --- a/docs/build/html/classmlx_1_1core_1_1_inverse.html +++ b/docs/build/html/classmlx_1_1core_1_1_inverse.html @@ -96,8 +96,8 @@ Inheritance diagram for mlx::core::Inverse: - - + + @@ -160,8 +160,8 @@ Public Member Functions

Public Member Functions

 Inverse (Stream stream)
 
 Inverse (Stream stream, bool tri, bool upper)
 
void eval_cpu (const std::vector< array > &inputs, array &output) override
 
void eval_gpu (const std::vector< array > &inputs, array &output) override
 

Constructor & Destructor Documentation

- -

◆ Inverse()

+ +

◆ Inverse()

@@ -172,8 +172,17 @@ Public Member Functions mlx::core::Inverse::Inverse ( - Stream stream) + Stream stream, + + + + bool tri, + + + + + bool upper ) diff --git a/docs/build/html/classmlx_1_1core_1_1_primitive.html b/docs/build/html/classmlx_1_1core_1_1_primitive.html index caebf65cd4..4c0658e3de 100644 --- a/docs/build/html/classmlx_1_1core_1_1_primitive.html +++ b/docs/build/html/classmlx_1_1core_1_1_primitive.html @@ -305,7 +305,7 @@ Public Member Functions

A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the output arrays.

To avoid unnecessary allocations, the evaluation function is responsible for allocating space for the array.

-

Implemented in mlx::core::distributed::AllReduce, mlx::core::distributed::AllGather, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::UnaryPrimitive, mlx::core::Compiled, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::DivMod, mlx::core::Split, mlx::core::QRF, and mlx::core::SVD.

+

Implemented in mlx::core::distributed::AllReduce, mlx::core::distributed::AllGather, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::fast::AffineQuantize, mlx::core::UnaryPrimitive, mlx::core::Compiled, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::DivMod, mlx::core::Split, mlx::core::QRF, and mlx::core::SVD.

@@ -336,7 +336,7 @@ Public Member Functions
-

Implemented in mlx::core::distributed::DistPrimitive, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::UnaryPrimitive, mlx::core::Compiled, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::DivMod, mlx::core::Split, mlx::core::QRF, and mlx::core::SVD.

+

Implemented in mlx::core::distributed::DistPrimitive, mlx::core::fast::RMSNorm, mlx::core::fast::RMSNormVJP, mlx::core::fast::LayerNorm, mlx::core::fast::LayerNormVJP, mlx::core::fast::RoPE, mlx::core::fast::ScaledDotProductAttention, mlx::core::fast::AffineQuantize, mlx::core::UnaryPrimitive, mlx::core::Compiled, mlx::core::CustomTransforms, mlx::core::Depends, mlx::core::DivMod, mlx::core::Split, mlx::core::QRF, and mlx::core::SVD.

@@ -616,7 +616,7 @@ Public Member Functions

The primitive must know how to vectorize itself across the given axes.

The output is a pair containing the output arrays representing the vectorized computation and the axes which corresponds to the vectorized dimensions of each output.

-

Reimplemented in mlx::core::distributed::AllReduce, mlx::core::distributed::AllGather, mlx::core::fast::Custom, mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan, mlx::core::ArcTan2, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Divide, mlx::core::DivMod, mlx::core::Select, mlx::core::Remainder, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log, mlx::core::Log1p, mlx::core::LogicalNot, mlx::core::LogicalAnd, mlx::core::LogicalOr, mlx::core::LogAddExp, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::GatherQMM, mlx::core::RandomBits, mlx::core::Reshape, mlx::core::Reduce, mlx::core::Round, mlx::core::Scan, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Square, mlx::core::Sqrt, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Uniform, mlx::core::View, mlx::core::Transpose, mlx::core::SVD, mlx::core::Inverse, and mlx::core::Cholesky.

+

Reimplemented in mlx::core::distributed::AllReduce, mlx::core::distributed::AllGather, mlx::core::fast::Custom, mlx::core::Abs, mlx::core::Add, mlx::core::AddMM, mlx::core::ArcCos, mlx::core::ArcCosh, mlx::core::ArcSin, mlx::core::ArcSinh, mlx::core::ArcTan, mlx::core::ArcTan2, mlx::core::ArcTanh, mlx::core::ArgPartition, mlx::core::ArgReduce, mlx::core::ArgSort, mlx::core::AsType, mlx::core::BitwiseBinary, mlx::core::Broadcast, mlx::core::Ceil, mlx::core::Compiled, mlx::core::Concatenate, mlx::core::Conjugate, mlx::core::Copy, mlx::core::Cos, mlx::core::Cosh, mlx::core::CustomTransforms, mlx::core::Divide, mlx::core::DivMod, mlx::core::Select, mlx::core::Remainder, mlx::core::Equal, mlx::core::Erf, mlx::core::ErfInv, mlx::core::Exp, mlx::core::Expm1, mlx::core::FFT, mlx::core::Floor, mlx::core::Full, mlx::core::Gather, mlx::core::Greater, mlx::core::GreaterEqual, mlx::core::Hadamard, mlx::core::Less, mlx::core::LessEqual, mlx::core::Log, mlx::core::Log1p, mlx::core::LogicalNot, mlx::core::LogicalAnd, mlx::core::LogicalOr, mlx::core::LogAddExp, mlx::core::Matmul, mlx::core::Maximum, mlx::core::Minimum, mlx::core::Multiply, mlx::core::Negative, mlx::core::NotEqual, mlx::core::NumberOfElements, mlx::core::Pad, mlx::core::Partition, mlx::core::Power, mlx::core::QuantizedMatmul, mlx::core::GatherQMM, mlx::core::RandomBits, mlx::core::Reshape, mlx::core::Reduce, mlx::core::Round, mlx::core::Scan, mlx::core::Scatter, mlx::core::Sigmoid, mlx::core::Sign, mlx::core::Sin, mlx::core::Sinh, mlx::core::Slice, mlx::core::SliceUpdate, mlx::core::Softmax, mlx::core::Sort, mlx::core::Split, mlx::core::Square, mlx::core::Sqrt, mlx::core::StopGradient, mlx::core::Subtract, mlx::core::Tan, mlx::core::Tanh, mlx::core::Uniform, mlx::core::View, mlx::core::Transpose, mlx::core::SVD, mlx::core::Inverse, and mlx::core::Cholesky.

diff --git a/docs/build/html/classmlx_1_1core_1_1_scatter-members.html b/docs/build/html/classmlx_1_1core_1_1_scatter-members.html index f6db8360de..8a2c29aa9c 100644 --- a/docs/build/html/classmlx_1_1core_1_1_scatter-members.html +++ b/docs/build/html/classmlx_1_1core_1_1_scatter-members.html @@ -109,7 +109,7 @@ $(function() { UnaryPrimitive(const UnaryPrimitive &other)=deletemlx::core::UnaryPrimitive UnaryPrimitive(UnaryPrimitive &&other)=deletemlx::core::UnaryPrimitive vjp(const std::vector< array > &primals, const std::vector< array > &cotangents, const std::vector< int > &argnums, const std::vector< array > &outputs) overridemlx::core::Scattervirtual - vmap(const std::vector< array > &inputs, const std::vector< int > &axes)mlx::core::Primitivevirtual + vmap(const std::vector< array > &inputs, const std::vector< int > &axes) overridemlx::core::Scattervirtual ~Primitive()=defaultmlx::core::Primitivevirtual ~UnaryPrimitive()=defaultmlx::core::UnaryPrimitivevirtual diff --git a/docs/build/html/classmlx_1_1core_1_1_scatter.html b/docs/build/html/classmlx_1_1core_1_1_scatter.html index a46a7ff995..232701b6eb 100644 --- a/docs/build/html/classmlx_1_1core_1_1_scatter.html +++ b/docs/build/html/classmlx_1_1core_1_1_scatter.html @@ -116,6 +116,9 @@ Public Member Functions   void eval_gpu (const std::vector< array > &inputs, array &out) override   +virtual std::pair< std::vector< array >, std::vector< int > > vmap (const std::vector< array > &inputs, const std::vector< int > &axes) override + The primitive must know how to vectorize itself across the given axes.
+  std::vector< array > jvp (const std::vector< array > &primals, const std::vector< array > &tangents, const std::vector< int > &argnums) override  The Jacobian-vector product.
  @@ -156,9 +159,6 @@ Public Member Functions const Stream & stream ()  The stream the primitive will run on.
  -virtual std::pair< std::vector< array >, std::vector< int > > vmap (const std::vector< array > &inputs, const std::vector< int > &axes) - The primitive must know how to vectorize itself across the given axes.
-  virtual std::vector< std::vector< int > > output_shapes (const std::vector< array > &inputs)  Get the output shapes of the primitive.
  @@ -430,6 +430,40 @@ Public Member Functions

Reimplemented from mlx::core::Primitive.

+ + + +

◆ vmap()

+ +
+
+ + + + + +
+ + + + + + + + + + + +
virtual std::pair< std::vector< array >, std::vector< int > > mlx::core::Scatter::vmap (const std::vector< array > & inputs,
const std::vector< int > & axes )
+ 		+overridevirtual
+
+ +

The primitive must know how to vectorize itself across the given axes.

+

The output is a pair containing the output arrays representing the vectorized computation and the axes which corresponds to the vectorized dimensions of each output.

+ +

Reimplemented from mlx::core::Primitive.

+
The documentation for this class was generated from the following file: diff --git a/docs/build/html/functions_vars_t.html b/docs/build/html/functions_vars_t.html index 39aea5ff43..930102d13a 100644 --- a/docs/build/html/functions_vars_t.html +++ b/docs/build/html/functions_vars_t.html @@ -75,7 +75,7 @@ $(function() {

- t -

  • TCOLS : mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >
    • -
  • tgp_mem_size : mlx::steel::GEMMKernel< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, MN_aligned, K_aligned, AccumType, Epilogue >
    • +
  • tgp_mem_size : GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >, GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >, mlx::steel::GEMMKernel< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, MN_aligned, K_aligned, AccumType, Epilogue >
  • tgp_mem_size_a : mlx::steel::GEMMKernel< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, MN_aligned, K_aligned, AccumType, Epilogue >
  • tgp_mem_size_b : mlx::steel::GEMMKernel< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, MN_aligned, K_aligned, AccumType, Epilogue >
  • tgp_padding_a : mlx::steel::GEMMKernel< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, MN_aligned, K_aligned, AccumType, Epilogue >
    • @@ -84,16 +84,18 @@ $(function() {
  • thread : mlx::core::scheduler::StreamThread
  • thread_idx : mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >, mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, QuantizedBlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, group_size, bits >
  • threads_per_tg : ReadWriter< in_T, out_T, step, four_step_real >
    • +
  • threadsM : GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >, GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >
    • +
  • threadsN : GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >, GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >
  • tile_stride : mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >, QuantizedBlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, group_size, bits >
  • tile_stride_a : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • tile_stride_b : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • tiles_m : mlx::steel::GEMMParams, mlx::steel::GEMMSpiltKParams, mlx::steel::ImplicitGemmConv2DParams, MLXFastAttentionParams
  • tiles_n : mlx::steel::GEMMParams, mlx::steel::GEMMSpiltKParams, mlx::steel::ImplicitGemmConv2DParams, MLXFastAttentionParams
    • -
  • tm : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • TM : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
    • +
  • tm : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • TM_stride : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
    • -
  • TN : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • tn : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
    • +
  • TN : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • TN_stride : mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >
  • TROWS : mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >, mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >, mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >
  • type : mlx::core::Device, mlx::core::ReductionPlan, pocketfft::detail::ExecDcst
    • diff --git a/docs/build/html/kernels_2steel_2gemm_2gemm_8h.html b/docs/build/html/gemm_8h.html similarity index 98% rename from docs/build/html/kernels_2steel_2gemm_2gemm_8h.html rename to docs/build/html/gemm_8h.html index c870504162..6b0b8ae278 100644 --- a/docs/build/html/kernels_2steel_2gemm_2gemm_8h.html +++ b/docs/build/html/gemm_8h.html @@ -87,7 +87,7 @@ $(function() { #include "mlx/backend/metal/kernels/steel/gemm/transforms.h"
    #include "mlx/backend/metal/kernels/steel/utils.h"
    -

    Go to the source code of this file.

    +

    Go to the source code of this file.

    diff --git a/docs/build/html/kernels_2steel_2gemm_2gemm_8h_source.html b/docs/build/html/gemm_8h_source.html similarity index 99% rename from docs/build/html/kernels_2steel_2gemm_2gemm_8h_source.html rename to docs/build/html/gemm_8h_source.html index 7acb35fad7..c5ced7a99b 100644 --- a/docs/build/html/kernels_2steel_2gemm_2gemm_8h_source.html +++ b/docs/build/html/gemm_8h_source.html @@ -83,7 +83,7 @@ $(function() { codefold.init(0); });
    gemm.h
    -Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    +Go to the documentation of this file.
    1// Copyright © 2024 Apple Inc.
    2
    3#pragma once
    4
    diff --git a/docs/build/html/genindex.html b/docs/build/html/genindex.html index 5f7ae635e6..743f4444af 100644 --- a/docs/build/html/genindex.html +++ b/docs/build/html/genindex.html @@ -7,7 +7,7 @@ - Index — MLX 0.16.1 documentation + Index — MLX 0.16.2 documentation @@ -35,7 +35,7 @@ - + @@ -128,8 +128,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home +
    @@ -454,8 +454,10 @@
  • Linear Algebra @@ -1038,6 +1040,8 @@ document.write(`
  • children() (Module method)
  • cholesky() (in module mlx.core.linalg) +
    • +
  • cholesky_inv() (in module mlx.core.linalg)
  • clear_cache() (in module mlx.core.metal)
    • @@ -1078,11 +1082,11 @@ document.write(`
  • conv2d() (in module mlx.core)
  • conv3d (C++ function) -
    • -
  • Conv3d (class in mlx.nn)
    • + + diff --git a/docs/build/html/kernels_8h_source.html b/docs/build/html/kernels_8h_source.html index 8b9cba186a..867c607796 100644 --- a/docs/build/html/kernels_8h_source.html +++ b/docs/build/html/kernels_8h_source.html @@ -236,59 +236,74 @@ $(function() { codefold.init(0); });
    151 int n_channel_specialization,
    152 bool small_filter);
    153
    -
    154MTL::ComputePipelineState* get_steel_conv_general_kernel(
    +
    154MTL::ComputePipelineState* get_gemv_masked_kernel(
    155 metal::Device& d,
    156 const std::string& kernel_name,
    157 const array& out,
    -
    158 int bm,
    -
    159 int bn,
    -
    160 int bk,
    -
    161 int wm,
    -
    162 int wn);
    -
    163
    -
    164MTL::ComputePipelineState* get_fft_kernel(
    -
    165 metal::Device& d,
    -
    166 const std::string& kernel_name,
    -
    167 const std::string& hash_name,
    -
    168 const metal::MTLFCList& func_consts,
    -
    169 const std::string& template_def);
    -
    170
    -
    171MTL::ComputePipelineState* get_quantized_kernel(
    -
    172 metal::Device& d,
    -
    173 const std::string& kernel_name,
    -
    174 const std::string& template_def);
    -
    175
    -
    176// Create a GPU kernel template definition for JIT compilation
    -
    177template <typename... Args>
    -
    178std::string
    -
    -
    179get_template_definition(std::string name, std::string func, Args... args) {
    -
    180 std::ostringstream s;
    -
    181 s << func << "<";
    -
    182 bool first = true;
    -
    183 auto add_arg = [&s, &first](const auto& arg) {
    -
    184 if (!first) {
    -
    185 s << ", ";
    -
    186 }
    -
    187 first = false;
    -
    188 s << arg;
    -
    189 };
    -
    190 (add_arg(args), ...);
    -
    191 s << ">";
    -
    192 std::string base_string = R"(
    -
    193template [[host_name("{0}")]] [[kernel]] decltype({1}) {1};
    -
    194 )";
    -
    195 return fmt::format(base_string, name, s.str());
    -
    196}
    +
    158 const std::optional<array>& mask_out,
    +
    159 const std::optional<array>& mask_op,
    +
    160 bool transpose_mat,
    +
    161 int bm,
    +
    162 int bn,
    +
    163 int sm,
    +
    164 int sn,
    +
    165 int tm,
    +
    166 int tn,
    +
    167 bool contiguous);
    +
    168
    +
    169MTL::ComputePipelineState* get_steel_conv_general_kernel(
    +
    170 metal::Device& d,
    +
    171 const std::string& kernel_name,
    +
    172 const array& out,
    +
    173 int bm,
    +
    174 int bn,
    +
    175 int bk,
    +
    176 int wm,
    +
    177 int wn);
    +
    178
    +
    179MTL::ComputePipelineState* get_fft_kernel(
    +
    180 metal::Device& d,
    +
    181 const std::string& kernel_name,
    +
    182 const std::string& hash_name,
    +
    183 const metal::MTLFCList& func_consts,
    +
    184 const std::string& template_def);
    +
    185
    +
    186MTL::ComputePipelineState* get_quantized_kernel(
    +
    187 metal::Device& d,
    +
    188 const std::string& kernel_name,
    +
    189 const std::string& template_def);
    +
    190
    +
    191// Create a GPU kernel template definition for JIT compilation
    +
    192template <typename... Args>
    +
    193std::string
    +
    +
    194get_template_definition(std::string name, std::string func, Args... args) {
    +
    195 std::ostringstream s;
    +
    196 s << func << "<";
    +
    197 bool first = true;
    +
    198 auto add_arg = [&s, &first](const auto& arg) {
    +
    199 if (!first) {
    +
    200 s << ", ";
    +
    201 }
    +
    202 first = false;
    +
    203 s << arg;
    +
    204 };
    +
    205 (add_arg(args), ...);
    +
    206 s << ">";
    +
    207 std::string base_string = R"(
    +
    208template [[host_name("{0}")]] [[kernel]] decltype({1}) {1};
    +
    209 )";
    +
    210 return fmt::format(base_string, name, s.str());
    +
    211}
    -
    197
    -
    198} // namespace mlx::core
    +
    212
    +
    213} // namespace mlx::core
    mlx::core::array
    Definition array.h:20
    -
    mlx::core::metal::Device
    Definition device.h:117
    +
    mlx::core::metal::Device
    Definition device.h:66
    op
    Op op
    Definition binary.h:141
    -
    mlx::core::metal::MTLFCList
    std::vector< std::tuple< const void *, MTL::DataType, NS::UInteger > > MTLFCList
    Definition device.h:36
    +
    mlx::core::metal::MTLFCList
    std::vector< std::tuple< const void *, MTL::DataType, NS::UInteger > > MTLFCList
    Definition device.h:17
    mlx::core
    Definition allocator.h:7
    mlx::core::get_copy_kernel
    MTL::ComputePipelineState * get_copy_kernel(metal::Device &d, const std::string &kernel_name, const array &in, const array &out)
    mlx::core::get_unary_kernel
    MTL::ComputePipelineState * get_unary_kernel(metal::Device &d, const std::string &kernel_name, Dtype out_type, const std::string op)
    @@ -303,15 +318,16 @@ $(function() { codefold.init(0); });
    mlx::core::get_arange_kernel
    MTL::ComputePipelineState * get_arange_kernel(metal::Device &d, const std::string &kernel_name, const array &out)
    mlx::core::get_sort_kernel
    MTL::ComputePipelineState * get_sort_kernel(metal::Device &d, const std::string &kernel_name, const array &in, const array &out, int bn, int tn)
    mlx::core::get_steel_gemm_fused_kernel
    MTL::ComputePipelineState * get_steel_gemm_fused_kernel(metal::Device &d, const std::string &kernel_name, const std::string &hash_name, const metal::MTLFCList &func_consts, const array &out, bool transpose_a, bool transpose_b, int bm, int bn, int bk, int wm, int wn)
    +
    mlx::core::get_gemv_masked_kernel
    MTL::ComputePipelineState * get_gemv_masked_kernel(metal::Device &d, const std::string &kernel_name, const array &out, const std::optional< array > &mask_out, const std::optional< array > &mask_op, bool transpose_mat, int bm, int bn, int sm, int sn, int tm, int tn, bool contiguous)
    mlx::core::get_quantized_kernel
    MTL::ComputePipelineState * get_quantized_kernel(metal::Device &d, const std::string &kernel_name, const std::string &template_def)
    -
    mlx::core::get_template_definition
    std::string get_template_definition(std::string name, std::string func, Args... args)
    Definition kernels.h:179
    +
    mlx::core::get_template_definition
    std::string get_template_definition(std::string name, std::string func, Args... args)
    Definition kernels.h:194
    mlx::core::get_steel_gemm_masked_kernel
    MTL::ComputePipelineState * get_steel_gemm_masked_kernel(metal::Device &d, const std::string &kernel_name, const array &out, const std::optional< array > &mask_out, const std::optional< array > &mask_op, bool transpose_a, bool transpose_b, int bm, int bn, int bk, int wm, int wn, bool mn_aligned, bool k_aligned)
    mlx::core::get_steel_conv_general_kernel
    MTL::ComputePipelineState * get_steel_conv_general_kernel(metal::Device &d, const std::string &kernel_name, const array &out, int bm, int bn, int bk, int wm, int wn)
    mlx::core::get_steel_conv_kernel
    MTL::ComputePipelineState * get_steel_conv_kernel(metal::Device &d, const std::string &kernel_name, const array &out, int bm, int bn, int bk, int wm, int wn, int n_channel_specialization, bool small_filter)
    mlx::core::get_scan_kernel
    MTL::ComputePipelineState * get_scan_kernel(metal::Device &d, const std::string &kernel_name, bool reverse, bool inclusive, const std::string &reduce_type, const array &in, const array &out)
    mlx::core::get_steel_gemm_splitk_kernel
    MTL::ComputePipelineState * get_steel_gemm_splitk_kernel(metal::Device &d, const std::string &kernel_name, const array &in, const array &out, bool transpose_a, bool transpose_b, int bm, int bn, int bk, int wm, int wn, bool mn_aligned, bool k_aligned)
    mlx::core::get_mb_sort_kernel
    MTL::ComputePipelineState * get_mb_sort_kernel(metal::Device &d, const std::string &kernel_name, const array &in, const array &idx, int bn, int tn)
    -
    mlx::core::Dtype
    Definition dtype.h:15
    +
    mlx::core::Dtype
    Definition dtype.h:13
    diff --git a/docs/build/html/linalg_8h.html b/docs/build/html/linalg_8h.html index 10ab25b356..ecc063e1f1 100644 --- a/docs/build/html/linalg_8h.html +++ b/docs/build/html/linalg_8h.html @@ -119,8 +119,12 @@ Functions
    + + + +

    Classes

     
    MTL::ComputePipelineState * mlx::core::get_steel_conv_kernel (metal::Device &d, const std::string &kernel_name, const array &out, int bm, int bn, int bk, int wm, int wn, int n_channel_specialization, bool small_filter)
     
    MTL::ComputePipelineState * mlx::core::get_gemv_masked_kernel (metal::Device &d, const std::string &kernel_name, const array &out, const std::optional< array > &mask_out, const std::optional< array > &mask_op, bool transpose_mat, int bm, int bn, int sm, int sn, int tm, int tn, bool contiguous)
     
    MTL::ComputePipelineState * mlx::core::get_steel_conv_general_kernel (metal::Device &d, const std::string &kernel_name, const array &out, int bm, int bn, int bk, int wm, int wn)
     
    MTL::ComputePipelineState * mlx::core::get_fft_kernel (metal::Device &d, const std::string &kernel_name, const std::string &hash_name, const metal::MTLFCList &func_consts, const std::string &template_def)
     
    array mlx::core::linalg::inv (const array &a, StreamOrDevice s={})
     
    array mlx::core::linalg::tri_inv (const array &a, bool upper=false, StreamOrDevice s={})
     
    array mlx::core::linalg::cholesky (const array &a, bool upper=false, StreamOrDevice s={})
     
    array mlx::core::linalg::cholesky_inv (const array &a, bool upper=false, StreamOrDevice s={})
     
    diff --git a/docs/build/html/linalg_8h_source.html b/docs/build/html/linalg_8h_source.html index bf88ac6479..2c1c59b17b 100644 --- a/docs/build/html/linalg_8h_source.html +++ b/docs/build/html/linalg_8h_source.html @@ -147,9 +147,13 @@ $(function() { codefold.init(0); });
    66
    67array inv(const array& a, StreamOrDevice s = {});
    68
    -
    69array cholesky(const array& a, bool upper = false, StreamOrDevice s = {});
    +
    69array tri_inv(const array& a, bool upper = false, StreamOrDevice s = {});
    70
    -
    71} // namespace mlx::core::linalg
    +
    71array cholesky(const array& a, bool upper = false, StreamOrDevice s = {});
    +
    72
    +
    73array cholesky_inv(const array& a, bool upper = false, StreamOrDevice s = {});
    +
    74
    +
    75} // namespace mlx::core::linalg
    mlx::core::array
    Definition array.h:20
    @@ -157,9 +161,11 @@ $(function() { codefold.init(0); });
    mlx::core::linalg
    Definition linalg.h:12
    mlx::core::linalg::cholesky
    array cholesky(const array &a, bool upper=false, StreamOrDevice s={})
    mlx::core::linalg::svd
    std::vector< array > svd(const array &a, StreamOrDevice s={})
    +
    mlx::core::linalg::tri_inv
    array tri_inv(const array &a, bool upper=false, StreamOrDevice s={})
    mlx::core::linalg::norm
    array norm(const array &a, const double ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})
    Compute vector or matrix norms.
    mlx::core::linalg::inv
    array inv(const array &a, StreamOrDevice s={})
    mlx::core::linalg::qr
    std::pair< array, array > qr(const array &a, StreamOrDevice s={})
    +
    mlx::core::linalg::cholesky_inv
    array cholesky_inv(const array &a, bool upper=false, StreamOrDevice s={})
    mlx::core::StreamOrDevice
    std::variant< std::monostate, Stream, Device > StreamOrDevice
    Definition utils.h:14
    diff --git a/docs/build/html/matmul_8h.html b/docs/build/html/matmul_8h.html index cd4e2fab1d..4c0c73d329 100644 --- a/docs/build/html/matmul_8h.html +++ b/docs/build/html/matmul_8h.html @@ -81,14 +81,7 @@ $(function() {
    matmul.h File Reference
    -
    #include <algorithm>
    -#include <cassert>
    -#include <sstream>
    -#include "mlx/backend/metal/copy.h"
    -#include "mlx/backend/metal/device.h"
    -#include "mlx/backend/metal/mps/gemm.h"
    -#include "mlx/backend/metal/utils.h"
    -#include "mlx/utils.h"
    +
    #include "mlx/backend/metal/device.h"

    Go to the source code of this file.

    diff --git a/docs/build/html/matmul_8h_source.html b/docs/build/html/matmul_8h_source.html index a0d044bc99..faff1cc7ba 100644 --- a/docs/build/html/matmul_8h_source.html +++ b/docs/build/html/matmul_8h_source.html @@ -85,66 +85,54 @@ $(function() { codefold.init(0); });
    Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    2
    -
    3#include <algorithm>
    -
    4#include <cassert>
    -
    5#include <sstream>
    +
    3#include "mlx/backend/metal/device.h"
    +
    4
    +
    5namespace mlx::core {
    6
    -
    7#include "mlx/backend/metal/copy.h"
    -
    8#include "mlx/backend/metal/device.h"
    -
    9#include "mlx/backend/metal/mps/gemm.h"
    -
    10#include "mlx/backend/metal/utils.h"
    -
    11#include "mlx/utils.h"
    -
    12
    -
    13namespace mlx::core {
    -
    14
    -
    15void steel_matmul_conv_groups(
    -
    16 const Stream& s,
    -
    17 metal::Device& d,
    -
    18 const array& a,
    -
    19 const array& b,
    -
    20 array& out,
    -
    21 int M,
    -
    22 int N,
    -
    23 int K,
    -
    24 int lda,
    -
    25 int ldb,
    -
    26 int ldd,
    -
    27 bool transpose_a,
    -
    28 bool transpose_b,
    -
    29 int groups,
    -
    30 std::vector<array>& copies);
    -
    31
    -
    32void steel_matmul(
    -
    33 const Stream& s,
    -
    34 metal::Device& d,
    -
    35 const array& a,
    -
    36 const array& b,
    -
    37 array& out,
    -
    38 int M,
    -
    39 int N,
    -
    40 int K,
    -
    41 int batch_size_out,
    -
    42 int lda,
    -
    43 int ldb,
    -
    44 bool transpose_a,
    -
    45 bool transpose_b,
    -
    46 std::vector<array>& copies,
    -
    47 std::vector<int> batch_shape = {},
    -
    48 std::vector<size_t> A_batch_stride = {},
    -
    49 std::vector<size_t> B_batch_stride = {});
    -
    50
    -
    51} // namespace mlx::core
    +
    7void steel_matmul_conv_groups(
    +
    8 const Stream& s,
    +
    9 metal::Device& d,
    +
    10 const array& a,
    +
    11 const array& b,
    +
    12 array& out,
    +
    13 int M,
    +
    14 int N,
    +
    15 int K,
    +
    16 int lda,
    +
    17 int ldb,
    +
    18 int ldd,
    +
    19 bool transpose_a,
    +
    20 bool transpose_b,
    +
    21 int groups,
    +
    22 std::vector<array>& copies);
    +
    23
    +
    24void steel_matmul(
    +
    25 const Stream& s,
    +
    26 metal::Device& d,
    +
    27 const array& a,
    +
    28 const array& b,
    +
    29 array& out,
    +
    30 int M,
    +
    31 int N,
    +
    32 int K,
    +
    33 int batch_size_out,
    +
    34 int lda,
    +
    35 int ldb,
    +
    36 bool transpose_a,
    +
    37 bool transpose_b,
    +
    38 std::vector<array>& copies,
    +
    39 std::vector<int> batch_shape = {},
    +
    40 std::vector<size_t> A_batch_stride = {},
    +
    41 std::vector<size_t> B_batch_stride = {});
    +
    42
    +
    43} // namespace mlx::core
    -
    mlx::core::array
    Definition array.h:20
    -
    mlx::core::metal::Device
    Definition device.h:117
    - - +
    mlx::core::metal::Device
    Definition device.h:66
    mlx::core
    Definition allocator.h:7
    mlx::core::steel_matmul
    void steel_matmul(const Stream &s, metal::Device &d, const array &a, const array &b, array &out, int M, int N, int K, int batch_size_out, int lda, int ldb, bool transpose_a, bool transpose_b, std::vector< array > &copies, std::vector< int > batch_shape={}, std::vector< size_t > A_batch_stride={}, std::vector< size_t > B_batch_stride={})
    mlx::core::steel_matmul_conv_groups
    void steel_matmul_conv_groups(const Stream &s, metal::Device &d, const array &a, const array &b, array &out, int M, int N, int K, int lda, int ldb, int ldd, bool transpose_a, bool transpose_b, int groups, std::vector< array > &copies)
    mlx::core::Stream
    Definition stream.h:9
    -
    diff --git a/docs/build/html/menudata.js b/docs/build/html/menudata.js index 2bc8c49c14..ffaab557ff 100644 --- a/docs/build/html/menudata.js +++ b/docs/build/html/menudata.js @@ -29,8 +29,7 @@ var menudata={children:[ {text:"Namespace List",url:"namespaces.html"}, {text:"Namespace Members",url:"namespacemembers.html",children:[ {text:"All",url:"namespacemembers.html",children:[ -{text:"_",url:"namespacemembers.html#index__5F"}, -{text:"a",url:"namespacemembers_a.html#index_a"}, +{text:"a",url:"namespacemembers.html#index_a"}, {text:"b",url:"namespacemembers_b.html#index_b"}, {text:"c",url:"namespacemembers_c.html#index_c"}, {text:"d",url:"namespacemembers_d.html#index_d"}, @@ -55,8 +54,7 @@ var menudata={children:[ {text:"w",url:"namespacemembers_w.html#index_w"}, {text:"z",url:"namespacemembers_z.html#index_z"}]}, {text:"Functions",url:"namespacemembers_func.html",children:[ -{text:"_",url:"namespacemembers_func.html#index__5F"}, -{text:"a",url:"namespacemembers_func_a.html#index_a"}, +{text:"a",url:"namespacemembers_func.html#index_a"}, {text:"b",url:"namespacemembers_func_b.html#index_b"}, {text:"c",url:"namespacemembers_func_c.html#index_c"}, {text:"d",url:"namespacemembers_func_d.html#index_d"}, @@ -161,6 +159,7 @@ var menudata={children:[ {text:"e",url:"functions_vars_e.html#index_e"}, {text:"f",url:"functions_vars_f.html#index_f"}, {text:"g",url:"functions_vars_g.html#index_g"}, +{text:"h",url:"functions_vars_h.html#index_h"}, {text:"i",url:"functions_vars_i.html#index_i"}, {text:"j",url:"functions_vars_j.html#index_j"}, {text:"k",url:"functions_vars_k.html#index_k"}, diff --git a/docs/build/html/metal_2binary_8h.html b/docs/build/html/metal_2binary_8h.html index 0b84f0e0d9..efe203c554 100644 --- a/docs/build/html/metal_2binary_8h.html +++ b/docs/build/html/metal_2binary_8h.html @@ -94,14 +94,14 @@ Namespaces
    - - - - - - - - + + + + + + + +

    Functions

    void mlx::core::binary_op_gpu (const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s)
     
    void mlx::core::binary_op_gpu (const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)
     
    void mlx::core::binary_op_gpu_inplace (const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s)
     
    void mlx::core::binary_op_gpu_inplace (const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)
     
    void mlx::core::binary_op_gpu (const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)
     
    void mlx::core::binary_op_gpu (const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)
     
    void mlx::core::binary_op_gpu_inplace (const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)
     
    void mlx::core::binary_op_gpu_inplace (const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)
     
    diff --git a/docs/build/html/metal_2binary_8h_source.html b/docs/build/html/metal_2binary_8h_source.html index 1236572c87..87cf14e407 100644 --- a/docs/build/html/metal_2binary_8h_source.html +++ b/docs/build/html/metal_2binary_8h_source.html @@ -91,28 +91,28 @@ $(function() { codefold.init(0); });
    6
    7namespace mlx::core {
    8
    -
    9void binary_op_gpu(
    +
    9void binary_op_gpu(
    10 const std::vector<array>& inputs,
    11 std::vector<array>& outputs,
    -
    12 const std::string op,
    +
    12 const std::string& op,
    13 const Stream& s);
    14
    -
    15void binary_op_gpu(
    +
    15void binary_op_gpu(
    16 const std::vector<array>& inputs,
    17 array& out,
    -
    18 const std::string op,
    +
    18 const std::string& op,
    19 const Stream& s);
    20
    -
    21void binary_op_gpu_inplace(
    +
    21void binary_op_gpu_inplace(
    22 const std::vector<array>& inputs,
    23 std::vector<array>& outputs,
    -
    24 const std::string op,
    +
    24 const std::string& op,
    25 const Stream& s);
    26
    -
    27void binary_op_gpu_inplace(
    +
    27void binary_op_gpu_inplace(
    28 const std::vector<array>& inputs,
    29 array& out,
    -
    30 const std::string op,
    +
    30 const std::string& op,
    31 const Stream& s);
    32
    33} // namespace mlx::core
    @@ -120,8 +120,8 @@ $(function() { codefold.init(0); });
    mlx::core::array
    Definition array.h:20
    op
    Op op
    Definition binary.h:141
    mlx::core
    Definition allocator.h:7
    -
    mlx::core::binary_op_gpu
    void binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s)
    -
    mlx::core::binary_op_gpu_inplace
    void binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s)
    +
    mlx::core::binary_op_gpu_inplace
    void binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)
    +
    mlx::core::binary_op_gpu
    void binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)
    mlx::core::Stream
    Definition stream.h:9
    diff --git a/docs/build/html/metal_2kernels_2binary_8h.html b/docs/build/html/metal_2kernels_2binary_8h.html index f2bb6e00f8..8a0c2c8cc8 100644 --- a/docs/build/html/metal_2kernels_2binary_8h.html +++ b/docs/build/html/metal_2kernels_2binary_8h.html @@ -97,6 +97,15 @@ Functions template<typename T , typename U , typename Op > void binary_vv (device const T *a, device const T *b, device U *c, uint index)   +template<typename T , typename U , typename Op > +void binary_sv2 (device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim) +  +template<typename T , typename U , typename Op > +void binary_vs2 (device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim) +  +template<typename T , typename U , typename Op > +void binary_vv2 (device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim) +  template<typename T , typename U , typename Op > void binary_g_nd1 (device const T *a, device const T *b, device U *c, constant const size_t &a_stride, constant const size_t &b_stride, uint index)   @@ -428,6 +437,44 @@ template<typename T , typename U , typename Op >
    +
    + + +

    ◆ binary_sv2()

    + +
    +
    +
    +template<typename T , typename U , typename Op >
    + + + + + + + + + + + + + + + + + + + + + + + + + + +
    void binary_sv2 (device const T * a,
    device const T * b,
    device U * c,
    uint2 index,
    uint2 grid_dim )
    +
    +
    @@ -461,6 +508,44 @@ template<typename T , typename U , typename Op >
    +
    + + +

    ◆ binary_vs2()

    + +
    +
    +
    +template<typename T , typename U , typename Op >
    + + + + + + + + + + + + + + + + + + + + + + + + + + +
    void binary_vs2 (device const T * a,
    device const T * b,
    device U * c,
    uint2 index,
    uint2 grid_dim )
    +
    +
    @@ -494,6 +579,44 @@ template<typename T , typename U , typename Op >
    +
    + + +

    ◆ binary_vv2()

    + +
    +
    +
    +template<typename T , typename U , typename Op >
    + + + + + + + + + + + + + + + + + + + + + + + + + + +
    void binary_vv2 (device const T * a,
    device const T * b,
    device U * c,
    uint2 index,
    uint2 grid_dim )
    +
    +
    diff --git a/docs/build/html/metal_2kernels_2binary_8h_source.html b/docs/build/html/metal_2kernels_2binary_8h_source.html index 3610537e3c..faadf60ebf 100644 --- a/docs/build/html/metal_2kernels_2binary_8h_source.html +++ b/docs/build/html/metal_2kernels_2binary_8h_source.html @@ -131,101 +131,143 @@ $(function() { codefold.init(0); });
    38
    39template <typename T, typename U, typename Op>
    -
    40[[kernel]] void binary_g_nd1(
    +
    40[[kernel]] void binary_sv2(
    41 device const T* a,
    42 device const T* b,
    43 device U* c,
    -
    44 constant const size_t& a_stride,
    -
    45 constant const size_t& b_stride,
    -
    46 uint index [[thread_position_in_grid]]) {
    -
    47 auto a_idx = elem_to_loc_1(index, a_stride);
    -
    48 auto b_idx = elem_to_loc_1(index, b_stride);
    -
    49 c[index] = Op()(a[a_idx], b[b_idx]);
    -
    50}
    +
    44 uint2 index [[thread_position_in_grid]],
    +
    45 uint2 grid_dim [[threads_per_grid]]) {
    +
    46 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    47 c[offset] = Op()(a[0], b[offset]);
    +
    48}
    -
    51
    -
    52template <typename T, typename U, typename Op>
    -
    -
    53[[kernel]] void binary_g_nd2(
    -
    54 device const T* a,
    -
    55 device const T* b,
    -
    56 device U* c,
    -
    57 constant const size_t a_strides[2],
    -
    58 constant const size_t b_strides[2],
    -
    59 uint2 index [[thread_position_in_grid]],
    -
    60 uint2 grid_dim [[threads_per_grid]]) {
    -
    61 auto a_idx = elem_to_loc_2(index, a_strides);
    -
    62 auto b_idx = elem_to_loc_2(index, b_strides);
    -
    63 size_t out_idx = index.x + (size_t)grid_dim.x * index.y;
    -
    64 c[out_idx] = Op()(a[a_idx], b[b_idx]);
    -
    65}
    +
    49
    +
    50template <typename T, typename U, typename Op>
    +
    +
    51[[kernel]] void binary_vs2(
    +
    52 device const T* a,
    +
    53 device const T* b,
    +
    54 device U* c,
    +
    55 uint2 index [[thread_position_in_grid]],
    +
    56 uint2 grid_dim [[threads_per_grid]]) {
    +
    57 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    58 c[offset] = Op()(a[offset], b[0]);
    +
    59}
    -
    66
    -
    67template <typename T, typename U, typename Op>
    -
    -
    68[[kernel]] void binary_g_nd3(
    -
    69 device const T* a,
    -
    70 device const T* b,
    -
    71 device U* c,
    -
    72 constant const size_t a_strides[3],
    -
    73 constant const size_t b_strides[3],
    -
    74 uint3 index [[thread_position_in_grid]],
    -
    75 uint3 grid_dim [[threads_per_grid]]) {
    -
    76 auto a_idx = elem_to_loc_3(index, a_strides);
    -
    77 auto b_idx = elem_to_loc_3(index, b_strides);
    -
    78 size_t out_idx =
    -
    79 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    -
    80 c[out_idx] = Op()(a[a_idx], b[b_idx]);
    -
    81}
    +
    60
    +
    61template <typename T, typename U, typename Op>
    +
    +
    62[[kernel]] void binary_vv2(
    +
    63 device const T* a,
    +
    64 device const T* b,
    +
    65 device U* c,
    +
    66 uint2 index [[thread_position_in_grid]],
    +
    67 uint2 grid_dim [[threads_per_grid]]) {
    +
    68 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    69 c[offset] = Op()(a[offset], b[offset]);
    +
    70}
    -
    82
    -
    83template <typename T, typename U, typename Op, int DIM>
    -
    -
    84[[kernel]] void binary_g_nd(
    -
    85 device const T* a,
    -
    86 device const T* b,
    -
    87 device U* c,
    -
    88 constant const int shape[DIM],
    -
    89 constant const size_t a_strides[DIM],
    -
    90 constant const size_t b_strides[DIM],
    -
    91 uint3 index [[thread_position_in_grid]],
    -
    92 uint3 grid_dim [[threads_per_grid]]) {
    -
    93 auto idx = elem_to_loc_2_nd<DIM>(index, shape, a_strides, b_strides);
    -
    94 size_t out_idx =
    -
    95 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    -
    96 c[out_idx] = Op()(a[idx.x], b[idx.y]);
    -
    97}
    +
    71
    +
    72template <typename T, typename U, typename Op>
    +
    +
    73[[kernel]] void binary_g_nd1(
    +
    74 device const T* a,
    +
    75 device const T* b,
    +
    76 device U* c,
    +
    77 constant const size_t& a_stride,
    +
    78 constant const size_t& b_stride,
    +
    79 uint index [[thread_position_in_grid]]) {
    +
    80 auto a_idx = elem_to_loc_1(index, a_stride);
    +
    81 auto b_idx = elem_to_loc_1(index, b_stride);
    +
    82 c[index] = Op()(a[a_idx], b[b_idx]);
    +
    83}
    -
    98
    -
    99template <typename T, typename U, typename Op>
    -
    -
    100[[kernel]] void binary_g(
    -
    101 device const T* a,
    -
    102 device const T* b,
    -
    103 device U* c,
    -
    104 constant const int* shape,
    -
    105 constant const size_t* a_strides,
    -
    106 constant const size_t* b_strides,
    -
    107 constant const int& ndim,
    -
    108 uint3 index [[thread_position_in_grid]],
    -
    109 uint3 grid_dim [[threads_per_grid]]) {
    -
    110 auto idx = elem_to_loc_2_nd(index, shape, a_strides, b_strides, ndim);
    -
    111 size_t out_idx = index.x + grid_dim.x * (index.y + grid_dim.y * index.z);
    -
    112 c[out_idx] = Op()(a[idx.x], b[idx.y]);
    -
    113}
    +
    84
    +
    85template <typename T, typename U, typename Op>
    +
    +
    86[[kernel]] void binary_g_nd2(
    +
    87 device const T* a,
    +
    88 device const T* b,
    +
    89 device U* c,
    +
    90 constant const size_t a_strides[2],
    +
    91 constant const size_t b_strides[2],
    +
    92 uint2 index [[thread_position_in_grid]],
    +
    93 uint2 grid_dim [[threads_per_grid]]) {
    +
    94 auto a_idx = elem_to_loc_2(index, a_strides);
    +
    95 auto b_idx = elem_to_loc_2(index, b_strides);
    +
    96 size_t out_idx = index.x + (size_t)grid_dim.x * index.y;
    +
    97 c[out_idx] = Op()(a[a_idx], b[b_idx]);
    +
    98}
    +
    +
    99
    +
    100template <typename T, typename U, typename Op>
    +
    +
    101[[kernel]] void binary_g_nd3(
    +
    102 device const T* a,
    +
    103 device const T* b,
    +
    104 device U* c,
    +
    105 constant const size_t a_strides[3],
    +
    106 constant const size_t b_strides[3],
    +
    107 uint3 index [[thread_position_in_grid]],
    +
    108 uint3 grid_dim [[threads_per_grid]]) {
    +
    109 auto a_idx = elem_to_loc_3(index, a_strides);
    +
    110 auto b_idx = elem_to_loc_3(index, b_strides);
    +
    111 size_t out_idx =
    +
    112 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    +
    113 c[out_idx] = Op()(a[a_idx], b[b_idx]);
    +
    114}
    +
    +
    115
    +
    116template <typename T, typename U, typename Op, int DIM>
    +
    +
    117[[kernel]] void binary_g_nd(
    +
    118 device const T* a,
    +
    119 device const T* b,
    +
    120 device U* c,
    +
    121 constant const int shape[DIM],
    +
    122 constant const size_t a_strides[DIM],
    +
    123 constant const size_t b_strides[DIM],
    +
    124 uint3 index [[thread_position_in_grid]],
    +
    125 uint3 grid_dim [[threads_per_grid]]) {
    +
    126 auto idx = elem_to_loc_2_nd<DIM>(index, shape, a_strides, b_strides);
    +
    127 size_t out_idx =
    +
    128 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    +
    129 c[out_idx] = Op()(a[idx.x], b[idx.y]);
    +
    130}
    +
    +
    131
    +
    132template <typename T, typename U, typename Op>
    +
    +
    133[[kernel]] void binary_g(
    +
    134 device const T* a,
    +
    135 device const T* b,
    +
    136 device U* c,
    +
    137 constant const int* shape,
    +
    138 constant const size_t* a_strides,
    +
    139 constant const size_t* b_strides,
    +
    140 constant const int& ndim,
    +
    141 uint3 index [[thread_position_in_grid]],
    +
    142 uint3 grid_dim [[threads_per_grid]]) {
    +
    143 auto idx = elem_to_loc_2_nd(index, shape, a_strides, b_strides, ndim);
    +
    144 size_t out_idx = index.x + grid_dim.x * (index.y + grid_dim.y * index.z);
    +
    145 c[out_idx] = Op()(a[idx.x], b[idx.y]);
    +
    146}
    elem_to_loc_1
    METAL_FUNC stride_t elem_to_loc_1(uint elem, constant const stride_t &stride)
    Definition utils.h:123
    elem_to_loc_3
    METAL_FUNC stride_t elem_to_loc_3(uint3 elem, constant const stride_t strides[3])
    Definition utils.h:135
    elem_to_loc_2_nd
    METAL_FUNC uint2 elem_to_loc_2_nd(uint3 elem, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, int ndim)
    Definition utils.h:200
    elem_to_loc_2
    METAL_FUNC stride_t elem_to_loc_2(uint2 elem, constant const stride_t strides[2])
    Definition utils.h:129
    +
    binary_vv2
    void binary_vv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim)
    Definition binary.h:62
    binary_ss
    void binary_ss(device const T *a, device const T *b, device U *c, uint index)
    Definition binary.h:4
    binary_sv
    void binary_sv(device const T *a, device const T *b, device U *c, uint index)
    Definition binary.h:13
    -
    binary_g_nd
    void binary_g_nd(device const T *a, device const T *b, device U *c, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], uint3 index, uint3 grid_dim)
    Definition binary.h:84
    +
    binary_vs2
    void binary_vs2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim)
    Definition binary.h:51
    +
    binary_g_nd
    void binary_g_nd(device const T *a, device const T *b, device U *c, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], uint3 index, uint3 grid_dim)
    Definition binary.h:117
    binary_vs
    void binary_vs(device const T *a, device const T *b, device U *c, uint index)
    Definition binary.h:22
    -
    binary_g_nd1
    void binary_g_nd1(device const T *a, device const T *b, device U *c, constant const size_t &a_stride, constant const size_t &b_stride, uint index)
    Definition binary.h:40
    -
    binary_g
    void binary_g(device const T *a, device const T *b, device U *c, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
    Definition binary.h:100
    -
    binary_g_nd2
    void binary_g_nd2(device const T *a, device const T *b, device U *c, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim)
    Definition binary.h:53
    -
    binary_g_nd3
    void binary_g_nd3(device const T *a, device const T *b, device U *c, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim)
    Definition binary.h:68
    +
    binary_g_nd1
    void binary_g_nd1(device const T *a, device const T *b, device U *c, constant const size_t &a_stride, constant const size_t &b_stride, uint index)
    Definition binary.h:73
    +
    binary_g
    void binary_g(device const T *a, device const T *b, device U *c, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
    Definition binary.h:133
    +
    binary_g_nd2
    void binary_g_nd2(device const T *a, device const T *b, device U *c, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim)
    Definition binary.h:86
    +
    binary_sv2
    void binary_sv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim)
    Definition binary.h:40
    +
    binary_g_nd3
    void binary_g_nd3(device const T *a, device const T *b, device U *c, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim)
    Definition binary.h:101
    binary_vv
    void binary_vv(device const T *a, device const T *b, device U *c, uint index)
    Definition binary.h:31
    diff --git a/docs/build/html/metal_2kernels_2binary__two_8h.html b/docs/build/html/metal_2kernels_2binary__two_8h.html index 31d36595de..ac2be5e834 100644 --- a/docs/build/html/metal_2kernels_2binary__two_8h.html +++ b/docs/build/html/metal_2kernels_2binary__two_8h.html @@ -97,6 +97,15 @@ Functions template<typename T , typename U , typename Op > void binary_vv (device const T *a, device const T *b, device U *c, device U *d, uint index)   +template<typename T , typename U , typename Op > +void binary_sv2 (device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim) +  +template<typename T , typename U , typename Op > +void binary_vs2 (device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim) +  +template<typename T , typename U , typename Op > +void binary_vv2 (device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim) +  template<typename T , typename U , typename Op > void binary_g_nd1 (device const T *a, device const T *b, device U *c, device U *d, constant const size_t &a_stride, constant const size_t &b_stride, uint index)   @@ -463,6 +472,49 @@ template<typename T , typename U , typename Op >
    +
    + + +

    ◆ binary_sv2()

    + +
    +
    +
    +template<typename T , typename U , typename Op >
    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
    void binary_sv2 (device const T * a,
    device const T * b,
    device U * c,
    device U * d,
    uint2 index,
    uint2 grid_dim )
    +
    +
    @@ -501,6 +553,49 @@ template<typename T , typename U , typename Op >
    +
    + + +

    ◆ binary_vs2()

    + +
    +
    +
    +template<typename T , typename U , typename Op >
    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
    void binary_vs2 (device const T * a,
    device const T * b,
    device U * c,
    device U * d,
    uint2 index,
    uint2 grid_dim )
    +
    +
    @@ -539,6 +634,49 @@ template<typename T , typename U , typename Op >
    +
    + + +

    ◆ binary_vv2()

    + +
    +
    +
    +template<typename T , typename U , typename Op >
    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
    void binary_vv2 (device const T * a,
    device const T * b,
    device U * c,
    device U * d,
    uint2 index,
    uint2 grid_dim )
    +
    +
    diff --git a/docs/build/html/metal_2kernels_2binary__two_8h_source.html b/docs/build/html/metal_2kernels_2binary__two_8h_source.html index 5b1bf1f91f..71216338f7 100644 --- a/docs/build/html/metal_2kernels_2binary__two_8h_source.html +++ b/docs/build/html/metal_2kernels_2binary__two_8h_source.html @@ -143,116 +143,167 @@ $(function() { codefold.init(0); });
    50
    51template <typename T, typename U, typename Op>
    -
    52[[kernel]] void binary_g_nd1(
    +
    52[[kernel]] void binary_sv2(
    53 device const T* a,
    54 device const T* b,
    55 device U* c,
    56 device U* d,
    -
    57 constant const size_t& a_stride,
    -
    58 constant const size_t& b_stride,
    -
    59 uint index [[thread_position_in_grid]]) {
    -
    60 auto a_idx = elem_to_loc_1(index, a_stride);
    -
    61 auto b_idx = elem_to_loc_1(index, b_stride);
    -
    62 auto out = Op()(a[a_idx], b[b_idx]);
    -
    63 c[index] = out[0];
    -
    64 d[index] = out[1];
    -
    65}
    +
    57 uint2 index [[thread_position_in_grid]],
    +
    58 uint2 grid_dim [[threads_per_grid]]) {
    +
    59 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    60 auto out = Op()(a[0], b[offset]);
    +
    61 c[offset] = out[0];
    +
    62 d[offset] = out[1];
    +
    63}
    -
    66
    -
    67template <typename T, typename U, typename Op>
    -
    -
    68[[kernel]] void binary_g_nd2(
    -
    69 device const T* a,
    -
    70 device const T* b,
    -
    71 device U* c,
    -
    72 device U* d,
    -
    73 constant const size_t a_strides[2],
    -
    74 constant const size_t b_strides[2],
    -
    75 uint2 index [[thread_position_in_grid]],
    -
    76 uint2 grid_dim [[threads_per_grid]]) {
    -
    77 auto a_idx = elem_to_loc_2(index, a_strides);
    -
    78 auto b_idx = elem_to_loc_2(index, b_strides);
    -
    79 size_t out_idx = index.x + (size_t)grid_dim.x * index.y;
    -
    80 auto out = Op()(a[a_idx], b[b_idx]);
    -
    81 c[out_idx] = out[0];
    -
    82 d[out_idx] = out[1];
    -
    83}
    +
    64
    +
    65template <typename T, typename U, typename Op>
    +
    +
    66[[kernel]] void binary_vs2(
    +
    67 device const T* a,
    +
    68 device const T* b,
    +
    69 device U* c,
    +
    70 device U* d,
    +
    71 uint2 index [[thread_position_in_grid]],
    +
    72 uint2 grid_dim [[threads_per_grid]]) {
    +
    73 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    74 auto out = Op()(a[offset], b[0]);
    +
    75 c[offset] = out[0];
    +
    76 d[offset] = out[1];
    +
    77}
    -
    84
    -
    85template <typename T, typename U, typename Op>
    -
    -
    86[[kernel]] void binary_g_nd3(
    -
    87 device const T* a,
    -
    88 device const T* b,
    -
    89 device U* c,
    -
    90 device U* d,
    -
    91 constant const size_t a_strides[3],
    -
    92 constant const size_t b_strides[3],
    -
    93 uint3 index [[thread_position_in_grid]],
    -
    94 uint3 grid_dim [[threads_per_grid]]) {
    -
    95 auto a_idx = elem_to_loc_3(index, a_strides);
    -
    96 auto b_idx = elem_to_loc_3(index, b_strides);
    -
    97 size_t out_idx =
    -
    98 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    -
    99 auto out = Op()(a[a_idx], b[b_idx]);
    -
    100 c[out_idx] = out[0];
    -
    101 d[out_idx] = out[1];
    -
    102}
    +
    78
    +
    79template <typename T, typename U, typename Op>
    +
    +
    80[[kernel]] void binary_vv2(
    +
    81 device const T* a,
    +
    82 device const T* b,
    +
    83 device U* c,
    +
    84 device U* d,
    +
    85 uint2 index [[thread_position_in_grid]],
    +
    86 uint2 grid_dim [[threads_per_grid]]) {
    +
    87 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    88 auto out = Op()(a[offset], b[offset]);
    +
    89 c[offset] = out[0];
    +
    90 d[offset] = out[1];
    +
    91}
    -
    103
    -
    104template <typename T, typename U, typename Op, int DIM>
    -
    -
    105[[kernel]] void binary_g_nd(
    -
    106 device const T* a,
    -
    107 device const T* b,
    -
    108 device U* c,
    -
    109 device U* d,
    -
    110 constant const int shape[DIM],
    -
    111 constant const size_t a_strides[DIM],
    -
    112 constant const size_t b_strides[DIM],
    -
    113 uint3 index [[thread_position_in_grid]],
    -
    114 uint3 grid_dim [[threads_per_grid]]) {
    -
    115 auto idx = elem_to_loc_2_nd<DIM>(index, shape, a_strides, b_strides);
    -
    116 size_t out_idx =
    -
    117 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    -
    118 auto out = Op()(a[idx.x], b[idx.y]);
    -
    119 c[out_idx] = out[0];
    -
    120 d[out_idx] = out[1];
    -
    121}
    +
    92
    +
    93template <typename T, typename U, typename Op>
    +
    +
    94[[kernel]] void binary_g_nd1(
    +
    95 device const T* a,
    +
    96 device const T* b,
    +
    97 device U* c,
    +
    98 device U* d,
    +
    99 constant const size_t& a_stride,
    +
    100 constant const size_t& b_stride,
    +
    101 uint index [[thread_position_in_grid]]) {
    +
    102 auto a_idx = elem_to_loc_1(index, a_stride);
    +
    103 auto b_idx = elem_to_loc_1(index, b_stride);
    +
    104 auto out = Op()(a[a_idx], b[b_idx]);
    +
    105 c[index] = out[0];
    +
    106 d[index] = out[1];
    +
    107}
    -
    122
    -
    123template <typename T, typename U, typename Op>
    -
    -
    124[[kernel]] void binary_g(
    -
    125 device const T* a,
    -
    126 device const T* b,
    -
    127 device U* c,
    -
    128 device U* d,
    -
    129 constant const int* shape,
    -
    130 constant const size_t* a_strides,
    -
    131 constant const size_t* b_strides,
    -
    132 constant const int& ndim,
    -
    133 uint3 index [[thread_position_in_grid]],
    -
    134 uint3 grid_dim [[threads_per_grid]]) {
    -
    135 auto idx = elem_to_loc_2_nd(index, shape, a_strides, b_strides, ndim);
    -
    136 size_t out_idx = index.x + grid_dim.x * (index.y + grid_dim.y * index.z);
    -
    137 auto out = Op()(a[idx.x], b[idx.y]);
    -
    138 c[out_idx] = out[0];
    -
    139 d[out_idx] = out[1];
    -
    140}
    +
    108
    +
    109template <typename T, typename U, typename Op>
    +
    +
    110[[kernel]] void binary_g_nd2(
    +
    111 device const T* a,
    +
    112 device const T* b,
    +
    113 device U* c,
    +
    114 device U* d,
    +
    115 constant const size_t a_strides[2],
    +
    116 constant const size_t b_strides[2],
    +
    117 uint2 index [[thread_position_in_grid]],
    +
    118 uint2 grid_dim [[threads_per_grid]]) {
    +
    119 auto a_idx = elem_to_loc_2(index, a_strides);
    +
    120 auto b_idx = elem_to_loc_2(index, b_strides);
    +
    121 size_t out_idx = index.x + (size_t)grid_dim.x * index.y;
    +
    122 auto out = Op()(a[a_idx], b[b_idx]);
    +
    123 c[out_idx] = out[0];
    +
    124 d[out_idx] = out[1];
    +
    125}
    +
    +
    126
    +
    127template <typename T, typename U, typename Op>
    +
    +
    128[[kernel]] void binary_g_nd3(
    +
    129 device const T* a,
    +
    130 device const T* b,
    +
    131 device U* c,
    +
    132 device U* d,
    +
    133 constant const size_t a_strides[3],
    +
    134 constant const size_t b_strides[3],
    +
    135 uint3 index [[thread_position_in_grid]],
    +
    136 uint3 grid_dim [[threads_per_grid]]) {
    +
    137 auto a_idx = elem_to_loc_3(index, a_strides);
    +
    138 auto b_idx = elem_to_loc_3(index, b_strides);
    +
    139 size_t out_idx =
    +
    140 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    +
    141 auto out = Op()(a[a_idx], b[b_idx]);
    +
    142 c[out_idx] = out[0];
    +
    143 d[out_idx] = out[1];
    +
    144}
    +
    +
    145
    +
    146template <typename T, typename U, typename Op, int DIM>
    +
    +
    147[[kernel]] void binary_g_nd(
    +
    148 device const T* a,
    +
    149 device const T* b,
    +
    150 device U* c,
    +
    151 device U* d,
    +
    152 constant const int shape[DIM],
    +
    153 constant const size_t a_strides[DIM],
    +
    154 constant const size_t b_strides[DIM],
    +
    155 uint3 index [[thread_position_in_grid]],
    +
    156 uint3 grid_dim [[threads_per_grid]]) {
    +
    157 auto idx = elem_to_loc_2_nd<DIM>(index, shape, a_strides, b_strides);
    +
    158 size_t out_idx =
    +
    159 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    +
    160 auto out = Op()(a[idx.x], b[idx.y]);
    +
    161 c[out_idx] = out[0];
    +
    162 d[out_idx] = out[1];
    +
    163}
    +
    +
    164
    +
    165template <typename T, typename U, typename Op>
    +
    +
    166[[kernel]] void binary_g(
    +
    167 device const T* a,
    +
    168 device const T* b,
    +
    169 device U* c,
    +
    170 device U* d,
    +
    171 constant const int* shape,
    +
    172 constant const size_t* a_strides,
    +
    173 constant const size_t* b_strides,
    +
    174 constant const int& ndim,
    +
    175 uint3 index [[thread_position_in_grid]],
    +
    176 uint3 grid_dim [[threads_per_grid]]) {
    +
    177 auto idx = elem_to_loc_2_nd(index, shape, a_strides, b_strides, ndim);
    +
    178 size_t out_idx = index.x + grid_dim.x * (index.y + grid_dim.y * index.z);
    +
    179 auto out = Op()(a[idx.x], b[idx.y]);
    +
    180 c[out_idx] = out[0];
    +
    181 d[out_idx] = out[1];
    +
    182}
    elem_to_loc_1
    METAL_FUNC stride_t elem_to_loc_1(uint elem, constant const stride_t &stride)
    Definition utils.h:123
    elem_to_loc_3
    METAL_FUNC stride_t elem_to_loc_3(uint3 elem, constant const stride_t strides[3])
    Definition utils.h:135
    elem_to_loc_2_nd
    METAL_FUNC uint2 elem_to_loc_2_nd(uint3 elem, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, int ndim)
    Definition utils.h:200
    elem_to_loc_2
    METAL_FUNC stride_t elem_to_loc_2(uint2 elem, constant const stride_t strides[2])
    Definition utils.h:129
    -
    binary_g_nd2
    void binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim)
    Definition binary_two.h:68
    +
    binary_g_nd2
    void binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim)
    Definition binary_two.h:110
    +
    binary_sv2
    void binary_sv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim)
    Definition binary_two.h:52
    binary_vs
    void binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index)
    Definition binary_two.h:28
    -
    binary_g_nd3
    void binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim)
    Definition binary_two.h:86
    -
    binary_g_nd
    void binary_g_nd(device const T *a, device const T *b, device U *c, device U *d, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], uint3 index, uint3 grid_dim)
    Definition binary_two.h:105
    +
    binary_vv2
    void binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim)
    Definition binary_two.h:80
    +
    binary_vs2
    void binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim)
    Definition binary_two.h:66
    +
    binary_g_nd3
    void binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim)
    Definition binary_two.h:128
    +
    binary_g_nd
    void binary_g_nd(device const T *a, device const T *b, device U *c, device U *d, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], uint3 index, uint3 grid_dim)
    Definition binary_two.h:147
    binary_sv
    void binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index)
    Definition binary_two.h:16
    binary_vv
    void binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index)
    Definition binary_two.h:40
    -
    binary_g
    void binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
    Definition binary_two.h:124
    -
    binary_g_nd1
    void binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const size_t &a_stride, constant const size_t &b_stride, uint index)
    Definition binary_two.h:52
    +
    binary_g
    void binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
    Definition binary_two.h:166
    +
    binary_g_nd1
    void binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const size_t &a_stride, constant const size_t &b_stride, uint index)
    Definition binary_two.h:94
    binary_ss
    void binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index)
    Definition binary_two.h:4
    diff --git a/docs/build/html/metal_2kernels_2copy_8h.html b/docs/build/html/metal_2kernels_2copy_8h.html index 15d04e834b..bf840126e2 100644 --- a/docs/build/html/metal_2kernels_2copy_8h.html +++ b/docs/build/html/metal_2kernels_2copy_8h.html @@ -91,6 +91,12 @@ Functions template<typename T , typename U > void copy_v (device const T *src, device U *dst, uint index)   +template<typename T , typename U > +void copy_s2 (device const T *src, device U *dst, uint2 index, uint2 grid_dim) +  +template<typename T , typename U > +void copy_v2 (device const T *src, device U *dst, uint2 index, uint2 grid_dim) +  template<typename T , typename U > void copy_g_nd1 (device const T *src, device U *dst, constant const int64_t &src_stride, uint index)   @@ -554,6 +560,39 @@ template<typename T , typename U >
    +
    + + +

    ◆ copy_s2()

    + +
    +
    +
    +template<typename T , typename U >
    + + + + + + + + + + + + + + + + + + + + + +
    void copy_s2 (device const T * src,
    device U * dst,
    uint2 index,
    uint2 grid_dim )
    +
    +
    @@ -582,6 +621,39 @@ template<typename T , typename U >
    +
    + + +

    ◆ copy_v2()

    + +
    +
    +
    +template<typename T , typename U >
    + + + + + + + + + + + + + + + + + + + + + +
    void copy_v2 (device const T * src,
    device U * dst,
    uint2 index,
    uint2 grid_dim )
    +
    +
    diff --git a/docs/build/html/metal_2kernels_2copy_8h_source.html b/docs/build/html/metal_2kernels_2copy_8h_source.html index b3cca38144..2ee6df35cf 100644 --- a/docs/build/html/metal_2kernels_2copy_8h_source.html +++ b/docs/build/html/metal_2kernels_2copy_8h_source.html @@ -107,165 +107,191 @@ $(function() { codefold.init(0); });
    18
    19template <typename T, typename U>
    -
    20[[kernel]] void copy_g_nd1(
    +
    20[[kernel]] void copy_s2(
    21 device const T* src [[buffer(0)]],
    22 device U* dst [[buffer(1)]],
    -
    23 constant const int64_t& src_stride [[buffer(3)]],
    -
    24 uint index [[thread_position_in_grid]]) {
    -
    25 auto src_idx = elem_to_loc_1(index, src_stride);
    -
    26 dst[index] = static_cast<U>(src[src_idx]);
    +
    23 uint2 index [[thread_position_in_grid]],
    +
    24 uint2 grid_dim [[threads_per_grid]]) {
    +
    25 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    26 dst[offset] = static_cast<U>(src[0]);
    27}
    28
    29template <typename T, typename U>
    -
    30[[kernel]] void copy_g_nd2(
    +
    30[[kernel]] void copy_v2(
    31 device const T* src [[buffer(0)]],
    32 device U* dst [[buffer(1)]],
    -
    33 constant const int64_t* src_strides [[buffer(3)]],
    -
    34 uint2 index [[thread_position_in_grid]],
    -
    35 uint2 grid_dim [[threads_per_grid]]) {
    -
    36 auto src_idx = elem_to_loc_2(index, src_strides);
    -
    37 int64_t dst_idx = index.x + (int64_t)grid_dim.x * index.y;
    -
    38 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    39}
    +
    33 uint2 index [[thread_position_in_grid]],
    +
    34 uint2 grid_dim [[threads_per_grid]]) {
    +
    35 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    36 dst[offset] = static_cast<U>(src[offset]);
    +
    37}
    -
    40
    -
    41template <typename T, typename U>
    -
    -
    42[[kernel]] void copy_g_nd3(
    -
    43 device const T* src [[buffer(0)]],
    -
    44 device U* dst [[buffer(1)]],
    -
    45 constant const int64_t* src_strides [[buffer(3)]],
    -
    46 uint3 index [[thread_position_in_grid]],
    -
    47 uint3 grid_dim [[threads_per_grid]]) {
    -
    48 auto src_idx = elem_to_loc_3(index, src_strides);
    -
    49 int64_t dst_idx =
    -
    50 index.x + (int64_t)grid_dim.x * (index.y + (int64_t)grid_dim.y * index.z);
    -
    51 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    52}
    +
    38
    +
    39template <typename T, typename U>
    +
    +
    40[[kernel]] void copy_g_nd1(
    +
    41 device const T* src [[buffer(0)]],
    +
    42 device U* dst [[buffer(1)]],
    +
    43 constant const int64_t& src_stride [[buffer(3)]],
    +
    44 uint index [[thread_position_in_grid]]) {
    +
    45 auto src_idx = elem_to_loc_1(index, src_stride);
    +
    46 dst[index] = static_cast<U>(src[src_idx]);
    +
    47}
    -
    53
    -
    54template <typename T, typename U, int DIM>
    -
    -
    55[[kernel]] void copy_g_nd(
    -
    56 device const T* src [[buffer(0)]],
    -
    57 device U* dst [[buffer(1)]],
    -
    58 constant const int* src_shape [[buffer(2)]],
    -
    59 constant const int64_t* src_strides [[buffer(3)]],
    -
    60 uint3 index [[thread_position_in_grid]],
    -
    61 uint3 grid_dim [[threads_per_grid]]) {
    -
    62 auto src_idx = elem_to_loc_nd<DIM>(index, src_shape, src_strides);
    -
    63 int64_t dst_idx =
    -
    64 index.x + (int64_t)grid_dim.x * (index.y + (int64_t)grid_dim.y * index.z);
    -
    65 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    66}
    +
    48
    +
    49template <typename T, typename U>
    +
    +
    50[[kernel]] void copy_g_nd2(
    +
    51 device const T* src [[buffer(0)]],
    +
    52 device U* dst [[buffer(1)]],
    +
    53 constant const int64_t* src_strides [[buffer(3)]],
    +
    54 uint2 index [[thread_position_in_grid]],
    +
    55 uint2 grid_dim [[threads_per_grid]]) {
    +
    56 auto src_idx = elem_to_loc_2(index, src_strides);
    +
    57 int64_t dst_idx = index.x + (int64_t)grid_dim.x * index.y;
    +
    58 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    59}
    -
    67
    -
    68template <typename T, typename U>
    -
    -
    69[[kernel]] void copy_g(
    -
    70 device const T* src [[buffer(0)]],
    -
    71 device U* dst [[buffer(1)]],
    -
    72 constant const int* src_shape [[buffer(2)]],
    -
    73 constant const int64_t* src_strides [[buffer(3)]],
    -
    74 constant const int& ndim [[buffer(5)]],
    -
    75 uint3 index [[thread_position_in_grid]],
    -
    76 uint3 grid_dim [[threads_per_grid]]) {
    -
    77 auto src_idx = elem_to_loc(index, src_shape, src_strides, ndim);
    -
    78 int64_t dst_idx =
    -
    79 index.x + (int64_t)grid_dim.x * (index.y + (int64_t)grid_dim.y * index.z);
    -
    80 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    81}
    +
    60
    +
    61template <typename T, typename U>
    +
    +
    62[[kernel]] void copy_g_nd3(
    +
    63 device const T* src [[buffer(0)]],
    +
    64 device U* dst [[buffer(1)]],
    +
    65 constant const int64_t* src_strides [[buffer(3)]],
    +
    66 uint3 index [[thread_position_in_grid]],
    +
    67 uint3 grid_dim [[threads_per_grid]]) {
    +
    68 auto src_idx = elem_to_loc_3(index, src_strides);
    +
    69 int64_t dst_idx =
    +
    70 index.x + (int64_t)grid_dim.x * (index.y + (int64_t)grid_dim.y * index.z);
    +
    71 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    72}
    -
    82
    -
    83template <typename T, typename U>
    -
    -
    84[[kernel]] void copy_gg_nd1(
    -
    85 device const T* src [[buffer(0)]],
    -
    86 device U* dst [[buffer(1)]],
    -
    87 constant const int64_t& src_stride [[buffer(3)]],
    -
    88 constant const int64_t& dst_stride [[buffer(4)]],
    -
    89 uint index [[thread_position_in_grid]]) {
    -
    90 auto src_idx = elem_to_loc_1(index, src_stride);
    -
    91 auto dst_idx = elem_to_loc_1(index, dst_stride);
    -
    92 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    93}
    +
    73
    +
    74template <typename T, typename U, int DIM>
    +
    +
    75[[kernel]] void copy_g_nd(
    +
    76 device const T* src [[buffer(0)]],
    +
    77 device U* dst [[buffer(1)]],
    +
    78 constant const int* src_shape [[buffer(2)]],
    +
    79 constant const int64_t* src_strides [[buffer(3)]],
    +
    80 uint3 index [[thread_position_in_grid]],
    +
    81 uint3 grid_dim [[threads_per_grid]]) {
    +
    82 auto src_idx = elem_to_loc_nd<DIM>(index, src_shape, src_strides);
    +
    83 int64_t dst_idx =
    +
    84 index.x + (int64_t)grid_dim.x * (index.y + (int64_t)grid_dim.y * index.z);
    +
    85 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    86}
    -
    94
    -
    95template <typename T, typename U>
    -
    -
    96[[kernel]] void copy_gg_nd2(
    -
    97 device const T* src [[buffer(0)]],
    -
    98 device U* dst [[buffer(1)]],
    -
    99 constant const int64_t* src_strides [[buffer(3)]],
    -
    100 constant const int64_t* dst_strides [[buffer(4)]],
    -
    101 uint2 index [[thread_position_in_grid]]) {
    -
    102 auto src_idx = elem_to_loc_2(index, src_strides);
    -
    103 auto dst_idx = elem_to_loc_2(index, dst_strides);
    -
    104 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    105}
    +
    87
    +
    88template <typename T, typename U>
    +
    +
    89[[kernel]] void copy_g(
    +
    90 device const T* src [[buffer(0)]],
    +
    91 device U* dst [[buffer(1)]],
    +
    92 constant const int* src_shape [[buffer(2)]],
    +
    93 constant const int64_t* src_strides [[buffer(3)]],
    +
    94 constant const int& ndim [[buffer(5)]],
    +
    95 uint3 index [[thread_position_in_grid]],
    +
    96 uint3 grid_dim [[threads_per_grid]]) {
    +
    97 auto src_idx = elem_to_loc(index, src_shape, src_strides, ndim);
    +
    98 int64_t dst_idx =
    +
    99 index.x + (int64_t)grid_dim.x * (index.y + (int64_t)grid_dim.y * index.z);
    +
    100 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    101}
    -
    106
    -
    107template <typename T, typename U>
    -
    -
    108[[kernel]] void copy_gg_nd3(
    -
    109 device const T* src [[buffer(0)]],
    -
    110 device U* dst [[buffer(1)]],
    -
    111 constant const int64_t* src_strides [[buffer(3)]],
    -
    112 constant const int64_t* dst_strides [[buffer(4)]],
    -
    113 uint3 index [[thread_position_in_grid]]) {
    -
    114 auto src_idx = elem_to_loc_3(index, src_strides);
    -
    115 auto dst_idx = elem_to_loc_3(index, dst_strides);
    -
    116 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    117}
    +
    102
    +
    103template <typename T, typename U>
    +
    +
    104[[kernel]] void copy_gg_nd1(
    +
    105 device const T* src [[buffer(0)]],
    +
    106 device U* dst [[buffer(1)]],
    +
    107 constant const int64_t& src_stride [[buffer(3)]],
    +
    108 constant const int64_t& dst_stride [[buffer(4)]],
    +
    109 uint index [[thread_position_in_grid]]) {
    +
    110 auto src_idx = elem_to_loc_1(index, src_stride);
    +
    111 auto dst_idx = elem_to_loc_1(index, dst_stride);
    +
    112 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    113}
    -
    118
    -
    119template <typename T, typename U, int DIM>
    -
    -
    120[[kernel]] void copy_gg_nd(
    -
    121 device const T* src [[buffer(0)]],
    -
    122 device U* dst [[buffer(1)]],
    -
    123 constant const int* src_shape [[buffer(2)]],
    -
    124 constant const int64_t* src_strides [[buffer(3)]],
    -
    125 constant const int64_t* dst_strides [[buffer(4)]],
    -
    126 uint3 index [[thread_position_in_grid]]) {
    -
    127 auto src_idx = elem_to_loc_nd<DIM>(index, src_shape, src_strides);
    -
    128 auto dst_idx = elem_to_loc_nd<DIM>(index, src_shape, dst_strides);
    -
    129 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    130}
    +
    114
    +
    115template <typename T, typename U>
    +
    +
    116[[kernel]] void copy_gg_nd2(
    +
    117 device const T* src [[buffer(0)]],
    +
    118 device U* dst [[buffer(1)]],
    +
    119 constant const int64_t* src_strides [[buffer(3)]],
    +
    120 constant const int64_t* dst_strides [[buffer(4)]],
    +
    121 uint2 index [[thread_position_in_grid]]) {
    +
    122 auto src_idx = elem_to_loc_2(index, src_strides);
    +
    123 auto dst_idx = elem_to_loc_2(index, dst_strides);
    +
    124 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    125}
    -
    131
    -
    132template <typename T, typename U>
    -
    -
    133[[kernel]] void copy_gg(
    -
    134 device const T* src [[buffer(0)]],
    -
    135 device U* dst [[buffer(1)]],
    -
    136 constant const int* src_shape [[buffer(2)]],
    -
    137 constant const int64_t* src_strides [[buffer(3)]],
    -
    138 constant const int64_t* dst_strides [[buffer(4)]],
    -
    139 constant const int& ndim [[buffer(5)]],
    -
    140 uint3 index [[thread_position_in_grid]]) {
    -
    141 auto src_idx = elem_to_loc(index, src_shape, src_strides, ndim);
    -
    142 auto dst_idx = elem_to_loc(index, src_shape, dst_strides, ndim);
    -
    143 dst[dst_idx] = static_cast<U>(src[src_idx]);
    -
    144}
    +
    126
    +
    127template <typename T, typename U>
    +
    +
    128[[kernel]] void copy_gg_nd3(
    +
    129 device const T* src [[buffer(0)]],
    +
    130 device U* dst [[buffer(1)]],
    +
    131 constant const int64_t* src_strides [[buffer(3)]],
    +
    132 constant const int64_t* dst_strides [[buffer(4)]],
    +
    133 uint3 index [[thread_position_in_grid]]) {
    +
    134 auto src_idx = elem_to_loc_3(index, src_strides);
    +
    135 auto dst_idx = elem_to_loc_3(index, dst_strides);
    +
    136 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    137}
    +
    +
    138
    +
    139template <typename T, typename U, int DIM>
    +
    +
    140[[kernel]] void copy_gg_nd(
    +
    141 device const T* src [[buffer(0)]],
    +
    142 device U* dst [[buffer(1)]],
    +
    143 constant const int* src_shape [[buffer(2)]],
    +
    144 constant const int64_t* src_strides [[buffer(3)]],
    +
    145 constant const int64_t* dst_strides [[buffer(4)]],
    +
    146 uint3 index [[thread_position_in_grid]]) {
    +
    147 auto src_idx = elem_to_loc_nd<DIM>(index, src_shape, src_strides);
    +
    148 auto dst_idx = elem_to_loc_nd<DIM>(index, src_shape, dst_strides);
    +
    149 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    150}
    +
    +
    151
    +
    152template <typename T, typename U>
    +
    +
    153[[kernel]] void copy_gg(
    +
    154 device const T* src [[buffer(0)]],
    +
    155 device U* dst [[buffer(1)]],
    +
    156 constant const int* src_shape [[buffer(2)]],
    +
    157 constant const int64_t* src_strides [[buffer(3)]],
    +
    158 constant const int64_t* dst_strides [[buffer(4)]],
    +
    159 constant const int& ndim [[buffer(5)]],
    +
    160 uint3 index [[thread_position_in_grid]]) {
    +
    161 auto src_idx = elem_to_loc(index, src_shape, src_strides, ndim);
    +
    162 auto dst_idx = elem_to_loc(index, src_shape, dst_strides, ndim);
    +
    163 dst[dst_idx] = static_cast<U>(src[src_idx]);
    +
    164}
    elem_to_loc_1
    METAL_FUNC stride_t elem_to_loc_1(uint elem, constant const stride_t &stride)
    Definition utils.h:123
    elem_to_loc_3
    METAL_FUNC stride_t elem_to_loc_3(uint3 elem, constant const stride_t strides[3])
    Definition utils.h:135
    elem_to_loc
    METAL_FUNC stride_t elem_to_loc(uint elem, device const int *shape, device const stride_t *strides, int ndim)
    Definition utils.h:77
    elem_to_loc_2
    METAL_FUNC stride_t elem_to_loc_2(uint2 elem, constant const stride_t strides[2])
    Definition utils.h:129
    -
    copy_g_nd
    void copy_g_nd(device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, uint3 index, uint3 grid_dim)
    Definition copy.h:55
    -
    copy_g
    void copy_g(device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
    Definition copy.h:69
    -
    copy_gg_nd
    void copy_gg_nd(device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint3 index)
    Definition copy.h:120
    -
    copy_gg_nd1
    void copy_gg_nd1(device const T *src, device U *dst, constant const int64_t &src_stride, constant const int64_t &dst_stride, uint index)
    Definition copy.h:84
    -
    copy_gg_nd2
    void copy_gg_nd2(device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint2 index)
    Definition copy.h:96
    -
    copy_gg_nd3
    void copy_gg_nd3(device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint3 index)
    Definition copy.h:108
    -
    copy_g_nd3
    void copy_g_nd3(device const T *src, device U *dst, constant const int64_t *src_strides, uint3 index, uint3 grid_dim)
    Definition copy.h:42
    -
    copy_gg
    void copy_gg(device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int64_t *dst_strides, constant const int &ndim, uint3 index)
    Definition copy.h:133
    -
    copy_g_nd1
    void copy_g_nd1(device const T *src, device U *dst, constant const int64_t &src_stride, uint index)
    Definition copy.h:20
    +
    copy_g_nd
    void copy_g_nd(device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, uint3 index, uint3 grid_dim)
    Definition copy.h:75
    +
    copy_g
    void copy_g(device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
    Definition copy.h:89
    +
    copy_gg_nd
    void copy_gg_nd(device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint3 index)
    Definition copy.h:140
    +
    copy_gg_nd1
    void copy_gg_nd1(device const T *src, device U *dst, constant const int64_t &src_stride, constant const int64_t &dst_stride, uint index)
    Definition copy.h:104
    +
    copy_gg_nd2
    void copy_gg_nd2(device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint2 index)
    Definition copy.h:116
    +
    copy_gg_nd3
    void copy_gg_nd3(device const T *src, device U *dst, constant const int64_t *src_strides, constant const int64_t *dst_strides, uint3 index)
    Definition copy.h:128
    +
    copy_s2
    void copy_s2(device const T *src, device U *dst, uint2 index, uint2 grid_dim)
    Definition copy.h:20
    +
    copy_g_nd3
    void copy_g_nd3(device const T *src, device U *dst, constant const int64_t *src_strides, uint3 index, uint3 grid_dim)
    Definition copy.h:62
    +
    copy_gg
    void copy_gg(device const T *src, device U *dst, constant const int *src_shape, constant const int64_t *src_strides, constant const int64_t *dst_strides, constant const int &ndim, uint3 index)
    Definition copy.h:153
    +
    copy_g_nd1
    void copy_g_nd1(device const T *src, device U *dst, constant const int64_t &src_stride, uint index)
    Definition copy.h:40
    copy_v
    void copy_v(device const T *src, device U *dst, uint index)
    Definition copy.h:12
    -
    copy_g_nd2
    void copy_g_nd2(device const T *src, device U *dst, constant const int64_t *src_strides, uint2 index, uint2 grid_dim)
    Definition copy.h:30
    +
    copy_v2
    void copy_v2(device const T *src, device U *dst, uint2 index, uint2 grid_dim)
    Definition copy.h:30
    +
    copy_g_nd2
    void copy_g_nd2(device const T *src, device U *dst, constant const int64_t *src_strides, uint2 index, uint2 grid_dim)
    Definition copy.h:50
    copy_s
    void copy_s(device const T *src, device U *dst, uint index)
    Definition copy.h:4
    diff --git a/docs/build/html/metal_2kernels_2ternary_8h.html b/docs/build/html/metal_2kernels_2ternary_8h.html index 54e9abffb5..45f8bb751f 100644 --- a/docs/build/html/metal_2kernels_2ternary_8h.html +++ b/docs/build/html/metal_2kernels_2ternary_8h.html @@ -88,6 +88,9 @@ Functions template<typename T , typename Op > void ternary_v (device const bool *a, device const T *b, device const T *c, device T *d, uint index)   +template<typename T , typename Op > +void ternary_v2 (device const bool *a, device const T *b, device const T *c, device T *d, uint2 index, uint2 grid_dim) +  template<typename T , typename Op > void ternary_g_nd1 (device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t &a_strides, constant const size_t &b_strides, constant const size_t &c_strides, uint index)   @@ -441,6 +444,49 @@ template<typename T , typename Op >
    +
    +
    + +

    ◆ ternary_v2()

    + +
    +
    +
    +template<typename T , typename Op >
    + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
    void ternary_v2 (device const bool * a,
    device const T * b,
    device const T * c,
    device T * d,
    uint2 index,
    uint2 grid_dim )
    +
    +
    diff --git a/docs/build/html/metal_2kernels_2ternary_8h_source.html b/docs/build/html/metal_2kernels_2ternary_8h_source.html index 5e0059e2da..707d3d9dd4 100644 --- a/docs/build/html/metal_2kernels_2ternary_8h_source.html +++ b/docs/build/html/metal_2kernels_2ternary_8h_source.html @@ -99,114 +99,129 @@ $(function() { codefold.init(0); });
    12
    13template <typename T, typename Op>
    -
    14[[kernel]] void ternary_g_nd1(
    +
    14[[kernel]] void ternary_v2(
    15 device const bool* a,
    16 device const T* b,
    17 device const T* c,
    18 device T* d,
    -
    19 constant const size_t& a_strides,
    -
    20 constant const size_t& b_strides,
    -
    21 constant const size_t& c_strides,
    -
    22 uint index [[thread_position_in_grid]]) {
    -
    23 auto a_idx = elem_to_loc_1(index, a_strides);
    -
    24 auto b_idx = elem_to_loc_1(index, b_strides);
    -
    25 auto c_idx = elem_to_loc_1(index, c_strides);
    -
    26 d[index] = Op()(a[a_idx], b[b_idx], c[c_idx]);
    -
    27}
    +
    19 uint2 index [[thread_position_in_grid]],
    +
    20 uint2 grid_dim [[threads_per_grid]]) {
    +
    21 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    22 d[offset] = Op()(a[offset], b[offset], c[offset]);
    +
    23}
    -
    28
    -
    29template <typename T, typename Op>
    -
    -
    30[[kernel]] void ternary_g_nd2(
    -
    31 device const bool* a,
    -
    32 device const T* b,
    -
    33 device const T* c,
    -
    34 device T* d,
    -
    35 constant const size_t a_strides[2],
    -
    36 constant const size_t b_strides[2],
    -
    37 constant const size_t c_strides[2],
    -
    38 uint2 index [[thread_position_in_grid]],
    -
    39 uint2 grid_dim [[threads_per_grid]]) {
    -
    40 auto a_idx = elem_to_loc_2(index, a_strides);
    -
    41 auto b_idx = elem_to_loc_2(index, b_strides);
    -
    42 auto c_idx = elem_to_loc_2(index, c_strides);
    -
    43 size_t out_idx = index.x + (size_t)grid_dim.x * index.y;
    -
    44 d[out_idx] = Op()(a[a_idx], b[b_idx], c[c_idx]);
    -
    45}
    +
    24
    +
    25template <typename T, typename Op>
    +
    +
    26[[kernel]] void ternary_g_nd1(
    +
    27 device const bool* a,
    +
    28 device const T* b,
    +
    29 device const T* c,
    +
    30 device T* d,
    +
    31 constant const size_t& a_strides,
    +
    32 constant const size_t& b_strides,
    +
    33 constant const size_t& c_strides,
    +
    34 uint index [[thread_position_in_grid]]) {
    +
    35 auto a_idx = elem_to_loc_1(index, a_strides);
    +
    36 auto b_idx = elem_to_loc_1(index, b_strides);
    +
    37 auto c_idx = elem_to_loc_1(index, c_strides);
    +
    38 d[index] = Op()(a[a_idx], b[b_idx], c[c_idx]);
    +
    39}
    -
    46
    -
    47template <typename T, typename Op>
    -
    -
    48[[kernel]] void ternary_g_nd3(
    -
    49 device const bool* a,
    -
    50 device const T* b,
    -
    51 device const T* c,
    -
    52 device T* d,
    -
    53 constant const size_t a_strides[3],
    -
    54 constant const size_t b_strides[3],
    -
    55 constant const size_t c_strides[3],
    -
    56 uint3 index [[thread_position_in_grid]],
    -
    57 uint3 grid_dim [[threads_per_grid]]) {
    -
    58 auto a_idx = elem_to_loc_3(index, a_strides);
    -
    59 auto b_idx = elem_to_loc_3(index, b_strides);
    -
    60 auto c_idx = elem_to_loc_3(index, c_strides);
    -
    61 size_t out_idx =
    -
    62 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    -
    63 d[out_idx] = Op()(a[a_idx], b[b_idx], c[c_idx]);
    -
    64}
    +
    40
    +
    41template <typename T, typename Op>
    +
    +
    42[[kernel]] void ternary_g_nd2(
    +
    43 device const bool* a,
    +
    44 device const T* b,
    +
    45 device const T* c,
    +
    46 device T* d,
    +
    47 constant const size_t a_strides[2],
    +
    48 constant const size_t b_strides[2],
    +
    49 constant const size_t c_strides[2],
    +
    50 uint2 index [[thread_position_in_grid]],
    +
    51 uint2 grid_dim [[threads_per_grid]]) {
    +
    52 auto a_idx = elem_to_loc_2(index, a_strides);
    +
    53 auto b_idx = elem_to_loc_2(index, b_strides);
    +
    54 auto c_idx = elem_to_loc_2(index, c_strides);
    +
    55 size_t out_idx = index.x + (size_t)grid_dim.x * index.y;
    +
    56 d[out_idx] = Op()(a[a_idx], b[b_idx], c[c_idx]);
    +
    57}
    -
    65
    -
    66template <typename T, typename Op, int DIM>
    -
    -
    67[[kernel]] void ternary_g_nd(
    -
    68 device const bool* a,
    -
    69 device const T* b,
    -
    70 device const T* c,
    -
    71 device T* d,
    -
    72 constant const int shape[DIM],
    -
    73 constant const size_t a_strides[DIM],
    -
    74 constant const size_t b_strides[DIM],
    -
    75 constant const size_t c_strides[DIM],
    -
    76 uint3 index [[thread_position_in_grid]],
    -
    77 uint3 grid_dim [[threads_per_grid]]) {
    -
    78 auto idx =
    -
    79 elem_to_loc_3_nd<DIM>(index, shape, a_strides, b_strides, c_strides);
    -
    80 size_t out_idx =
    -
    81 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    -
    82 d[out_idx] = Op()(a[idx.x], b[idx.y], c[idx.z]);
    -
    83}
    +
    58
    +
    59template <typename T, typename Op>
    +
    +
    60[[kernel]] void ternary_g_nd3(
    +
    61 device const bool* a,
    +
    62 device const T* b,
    +
    63 device const T* c,
    +
    64 device T* d,
    +
    65 constant const size_t a_strides[3],
    +
    66 constant const size_t b_strides[3],
    +
    67 constant const size_t c_strides[3],
    +
    68 uint3 index [[thread_position_in_grid]],
    +
    69 uint3 grid_dim [[threads_per_grid]]) {
    +
    70 auto a_idx = elem_to_loc_3(index, a_strides);
    +
    71 auto b_idx = elem_to_loc_3(index, b_strides);
    +
    72 auto c_idx = elem_to_loc_3(index, c_strides);
    +
    73 size_t out_idx =
    +
    74 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    +
    75 d[out_idx] = Op()(a[a_idx], b[b_idx], c[c_idx]);
    +
    76}
    -
    84
    -
    85template <typename T, typename Op>
    -
    -
    86[[kernel]] void ternary_g(
    -
    87 device const bool* a,
    -
    88 device const T* b,
    -
    89 device const T* c,
    -
    90 device T* d,
    -
    91 constant const int* shape,
    -
    92 constant const size_t* a_strides,
    -
    93 constant const size_t* b_strides,
    -
    94 constant const size_t* c_strides,
    -
    95 constant const int& ndim,
    -
    96 uint3 index [[thread_position_in_grid]],
    -
    97 uint3 grid_dim [[threads_per_grid]]) {
    -
    98 auto idx =
    -
    99 elem_to_loc_3_nd(index, shape, a_strides, b_strides, c_strides, ndim);
    -
    100 size_t out_idx = index.x + grid_dim.x * (index.y + grid_dim.y * index.z);
    -
    101 d[out_idx] = Op()(a[idx.x], b[idx.y], c[idx.z]);
    -
    102}
    +
    77
    +
    78template <typename T, typename Op, int DIM>
    +
    +
    79[[kernel]] void ternary_g_nd(
    +
    80 device const bool* a,
    +
    81 device const T* b,
    +
    82 device const T* c,
    +
    83 device T* d,
    +
    84 constant const int shape[DIM],
    +
    85 constant const size_t a_strides[DIM],
    +
    86 constant const size_t b_strides[DIM],
    +
    87 constant const size_t c_strides[DIM],
    +
    88 uint3 index [[thread_position_in_grid]],
    +
    89 uint3 grid_dim [[threads_per_grid]]) {
    +
    90 auto idx =
    +
    91 elem_to_loc_3_nd<DIM>(index, shape, a_strides, b_strides, c_strides);
    +
    92 size_t out_idx =
    +
    93 index.x + (size_t)grid_dim.x * (index.y + (size_t)grid_dim.y * index.z);
    +
    94 d[out_idx] = Op()(a[idx.x], b[idx.y], c[idx.z]);
    +
    95}
    +
    +
    96
    +
    97template <typename T, typename Op>
    +
    +
    98[[kernel]] void ternary_g(
    +
    99 device const bool* a,
    +
    100 device const T* b,
    +
    101 device const T* c,
    +
    102 device T* d,
    +
    103 constant const int* shape,
    +
    104 constant const size_t* a_strides,
    +
    105 constant const size_t* b_strides,
    +
    106 constant const size_t* c_strides,
    +
    107 constant const int& ndim,
    +
    108 uint3 index [[thread_position_in_grid]],
    +
    109 uint3 grid_dim [[threads_per_grid]]) {
    +
    110 auto idx =
    +
    111 elem_to_loc_3_nd(index, shape, a_strides, b_strides, c_strides, ndim);
    +
    112 size_t out_idx = index.x + grid_dim.x * (index.y + grid_dim.y * index.z);
    +
    113 d[out_idx] = Op()(a[idx.x], b[idx.y], c[idx.z]);
    +
    114}
    elem_to_loc_1
    METAL_FUNC stride_t elem_to_loc_1(uint elem, constant const stride_t &stride)
    Definition utils.h:123
    elem_to_loc_3
    METAL_FUNC stride_t elem_to_loc_3(uint3 elem, constant const stride_t strides[3])
    Definition utils.h:135
    elem_to_loc_3_nd
    METAL_FUNC uint3 elem_to_loc_3_nd(uint3 elem, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const size_t *c_strides, int ndim)
    Definition utils.h:220
    elem_to_loc_2
    METAL_FUNC stride_t elem_to_loc_2(uint2 elem, constant const stride_t strides[2])
    Definition utils.h:129
    -
    ternary_g_nd3
    void ternary_g_nd3(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t a_strides[3], constant const size_t b_strides[3], constant const size_t c_strides[3], uint3 index, uint3 grid_dim)
    Definition ternary.h:48
    -
    ternary_g_nd1
    void ternary_g_nd1(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t &a_strides, constant const size_t &b_strides, constant const size_t &c_strides, uint index)
    Definition ternary.h:14
    +
    ternary_g_nd3
    void ternary_g_nd3(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t a_strides[3], constant const size_t b_strides[3], constant const size_t c_strides[3], uint3 index, uint3 grid_dim)
    Definition ternary.h:60
    +
    ternary_g_nd1
    void ternary_g_nd1(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t &a_strides, constant const size_t &b_strides, constant const size_t &c_strides, uint index)
    Definition ternary.h:26
    +
    ternary_v2
    void ternary_v2(device const bool *a, device const T *b, device const T *c, device T *d, uint2 index, uint2 grid_dim)
    Definition ternary.h:14
    ternary_v
    void ternary_v(device const bool *a, device const T *b, device const T *c, device T *d, uint index)
    Definition ternary.h:4
    -
    ternary_g_nd
    void ternary_g_nd(device const bool *a, device const T *b, device const T *c, device T *d, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], constant const size_t c_strides[DIM], uint3 index, uint3 grid_dim)
    Definition ternary.h:67
    -
    ternary_g
    void ternary_g(device const bool *a, device const T *b, device const T *c, device T *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const size_t *c_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
    Definition ternary.h:86
    -
    ternary_g_nd2
    void ternary_g_nd2(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t a_strides[2], constant const size_t b_strides[2], constant const size_t c_strides[2], uint2 index, uint2 grid_dim)
    Definition ternary.h:30
    +
    ternary_g_nd
    void ternary_g_nd(device const bool *a, device const T *b, device const T *c, device T *d, constant const int shape[DIM], constant const size_t a_strides[DIM], constant const size_t b_strides[DIM], constant const size_t c_strides[DIM], uint3 index, uint3 grid_dim)
    Definition ternary.h:79
    +
    ternary_g
    void ternary_g(device const bool *a, device const T *b, device const T *c, device T *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const size_t *c_strides, constant const int &ndim, uint3 index, uint3 grid_dim)
    Definition ternary.h:98
    +
    ternary_g_nd2
    void ternary_g_nd2(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t a_strides[2], constant const size_t b_strides[2], constant const size_t c_strides[2], uint2 index, uint2 grid_dim)
    Definition ternary.h:42
    diff --git a/docs/build/html/metal_2kernels_2unary_8h.html b/docs/build/html/metal_2kernels_2unary_8h.html index 0afe1ddf12..291da68a57 100644 --- a/docs/build/html/metal_2kernels_2unary_8h.html +++ b/docs/build/html/metal_2kernels_2unary_8h.html @@ -88,6 +88,9 @@ Functions template<typename T , typename Op > void unary_v (device const T *in, device T *out, uint index)   +template<typename T , typename Op > +void unary_v2 (device const T *in, device T *out, uint2 index, uint2 grid_dim) +  template<typename T , typename Op > void unary_g (device const T *in, device T *out, device const int *in_shape, device const size_t *in_strides, device const int &ndim, uint index)   @@ -162,6 +165,39 @@ template<typename T , typename Op >
    +
    +
    + +

    ◆ unary_v2()

    + +
    +
    +
    +template<typename T , typename Op >
    + + + + + + + + + + + + + + + + + + + + + +
    void unary_v2 (device const T * in,
    device T * out,
    uint2 index,
    uint2 grid_dim )
    +
    +
    diff --git a/docs/build/html/metal_2kernels_2unary_8h_source.html b/docs/build/html/metal_2kernels_2unary_8h_source.html index 141892d1f9..e379752e91 100644 --- a/docs/build/html/metal_2kernels_2unary_8h_source.html +++ b/docs/build/html/metal_2kernels_2unary_8h_source.html @@ -97,19 +97,32 @@ $(function() { codefold.init(0); });
    10
    11template <typename T, typename Op>
    -
    12[[kernel]] void unary_g(
    +
    12[[kernel]] void unary_v2(
    13 device const T* in,
    14 device T* out,
    -
    15 device const int* in_shape,
    -
    16 device const size_t* in_strides,
    -
    17 device const int& ndim,
    -
    18 uint index [[thread_position_in_grid]]) {
    -
    19 auto idx = elem_to_loc(index, in_shape, in_strides, ndim);
    -
    20 out[index] = Op()(in[idx]);
    -
    21}
    +
    15 uint2 index [[thread_position_in_grid]],
    +
    16 uint2 grid_dim [[threads_per_grid]]) {
    +
    17 size_t offset = index.x + grid_dim.x * size_t(index.y);
    +
    18 out[offset] = Op()(in[offset]);
    +
    19}
    +
    +
    20
    +
    21template <typename T, typename Op>
    +
    +
    22[[kernel]] void unary_g(
    +
    23 device const T* in,
    +
    24 device T* out,
    +
    25 device const int* in_shape,
    +
    26 device const size_t* in_strides,
    +
    27 device const int& ndim,
    +
    28 uint index [[thread_position_in_grid]]) {
    +
    29 auto idx = elem_to_loc(index, in_shape, in_strides, ndim);
    +
    30 out[index] = Op()(in[idx]);
    +
    31}
    elem_to_loc
    METAL_FUNC stride_t elem_to_loc(uint elem, device const int *shape, device const stride_t *strides, int ndim)
    Definition utils.h:77
    -
    unary_g
    void unary_g(device const T *in, device T *out, device const int *in_shape, device const size_t *in_strides, device const int &ndim, uint index)
    Definition unary.h:12
    +
    unary_g
    void unary_g(device const T *in, device T *out, device const int *in_shape, device const size_t *in_strides, device const int &ndim, uint index)
    Definition unary.h:22
    +
    unary_v2
    void unary_v2(device const T *in, device T *out, uint2 index, uint2 grid_dim)
    Definition unary.h:12
    unary_v
    void unary_v(device const T *in, device T *out, uint index)
    Definition unary.h:4
    diff --git a/docs/build/html/metal_2reduce_8h_source.html b/docs/build/html/metal_2reduce_8h_source.html index 0b477ee3e2..5e95b42c31 100644 --- a/docs/build/html/metal_2reduce_8h_source.html +++ b/docs/build/html/metal_2reduce_8h_source.html @@ -126,7 +126,7 @@ $(function() { codefold.init(0); });
    41} // namespace mlx::core
    mlx::core::array
    Definition array.h:20
    -
    mlx::core::metal::Device
    Definition device.h:117
    +
    mlx::core::metal::Device
    Definition device.h:66
    mlx::core
    Definition allocator.h:7
    mlx::core::all_reduce_dispatch
    void all_reduce_dispatch(const array &in, array &out, const std::string &op_name, CommandEncoder &compute_encoder, metal::Device &d, const Stream &s)
    @@ -135,7 +135,7 @@ $(function() { codefold.init(0); });
    mlx::core::ReductionPlan
    Definition reduce.h:39
    mlx::core::Stream
    Definition stream.h:9
    -
    mlx::core::metal::CommandEncoder
    Definition device.h:39
    +
    mlx::core::metal::CommandEncoder
    Definition device.h:20
    diff --git a/docs/build/html/mps_2gemm_8h.html b/docs/build/html/mps_2gemm_8h.html deleted file mode 100644 index f264c755db..0000000000 --- a/docs/build/html/mps_2gemm_8h.html +++ /dev/null @@ -1,235 +0,0 @@ - - - - - - - -MLX: mlx/backend/metal/mps/gemm.h File Reference - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - - -
    -
    -
    -Classes | -Namespaces | -Macros | -Typedefs | -Enumerations | -Functions
    -
    gemm.h File Reference
    -
    -
    -
    #include <Metal/Metal.hpp>
    -
    -

    Go to the source code of this file.

    - - - - - - - - - - - - - - - - -

    -Classes

    class  MPS::MatrixDescriptor
     
    class  MPS::Matrix
     
    class  MPS::Kernel
     
    class  MPS::MatrixMultiplication
     
    class  MPS::VectorDescriptor
     
    class  MPS::Vector
     
    class  MPS::MatrixVectorMultiplication
     
    - - - - - - - - - - - -

    -Namespaces

    namespace  MTL
     
    namespace  MTL::Private
     
    namespace  MTL::Private::Class
     
    namespace  MTL::Private::Selector
     
    namespace  MPS
     
    - - - - - -

    -Macros

    #define _MPS_PRIVATE_CLS(symbol)   (MTL::Private::Class::s_k##symbol)
     
    #define _MPS_PRIVATE_SEL(accessor)   (MTL::Private::Selector::s_k##accessor)
     
    - - - -

    -Typedefs

    typedef enum MPS::DataType MPS::DataType
     
    - - - -

    -Enumerations

    enum  MPS::DataType : uint32_t {
    -  MPS::DataTypeFloatBit = 0x10000000 -, MPS::DataTypeAlternateEncodingBit = 0x80000000 -, MPS::DataTypeFloat16 = DataTypeFloatBit | 16 -, MPS::DataTypeFloat32 = DataTypeFloatBit | 32 -,
    -  MPS::DataTypeBFloat16 = DataTypeAlternateEncodingBit | DataTypeFloat16 -
    - }
     
    - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

    -Functions

     MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixDescriptor)
     
     MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrix)
     
     MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSVectorDescriptor)
     
     MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSVector)
     
     MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSKernel)
     
     MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixMultiplication)
     
     MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixVectorMultiplication)
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (matrixDescriptorWithRows_columns_rowBytes_dataType, "matrixDescriptorWithRows:columns:rowBytes:dataType:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (matrixDescriptorWithRows_columns_matrices_rowBytes_matrixBytes_dataType, "matrixDescriptorWithRows:columns:matrices:rowBytes:matrixBytes:dataType:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (rows, "rows")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithBuffer_descriptor, "initWithBuffer:descriptor:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithDevice_, "initWithDevice:transposeLeft:transposeRight:" "resultRows:resultColumns:interiorColumns:alpha:beta:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (encodeToCommandBuffer_leftMatrix_rightMatrix_resultMatrix, "encodeToCommandBuffer:leftMatrix:rightMatrix:resultMatrix:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setLeftMatrixOrigin_, "setLeftMatrixOrigin:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setRightMatrixOrigin_, "setRightMatrixOrigin:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setResultMatrixOrigin_, "setResultMatrixOrigin:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setBatchStart_, "setBatchStart:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setBatchSize_, "setBatchSize:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (vectorDescriptorWithLength_dataType, "vectorDescriptorWithLength:dataType:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (vectorDescriptorWithLength_vectors_vectorBytes_dataType, "vectorDescriptorWithLength:vectors:vectorBytes:dataType:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithDevice_transpose_rows_columns_alpha_beta, "initWithDevice:transpose:rows:columns:alpha:beta:")
     
     MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (encodeToCommandBuffer_inputMatrix_inputVector_resultVector, "encodeToCommandBuffer:inputMatrix:inputVector:resultVector:")
     
    -

    Macro Definition Documentation

    - -

    ◆ _MPS_PRIVATE_CLS

    - -
    -
    - - - - - - - -
    #define _MPS_PRIVATE_CLS( symbol)   (MTL::Private::Class::s_k##symbol)
    -
    - -
    -
    - -

    ◆ _MPS_PRIVATE_SEL

    - -
    -
    - - - - - - - -
    #define _MPS_PRIVATE_SEL( accessor)   (MTL::Private::Selector::s_k##accessor)
    -
    - -
    -
    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/mps_2gemm_8h_source.html b/docs/build/html/mps_2gemm_8h_source.html deleted file mode 100644 index e991428e28..0000000000 --- a/docs/build/html/mps_2gemm_8h_source.html +++ /dev/null @@ -1,572 +0,0 @@ - - - - - - - -MLX: mlx/backend/metal/mps/gemm.h Source File - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - - - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - - -
    -
    -
    gemm.h
    -
    -
    -Go to the documentation of this file.
    1// Copyright © 2023 Apple Inc.
    -
    2
    -
    3#pragma once
    -
    4
    -
    5#include <Metal/Metal.hpp>
    -
    6
    -
    7#define _MPS_PRIVATE_CLS(symbol) (MTL::Private::Class::s_k##symbol)
    -
    8#define _MPS_PRIVATE_SEL(accessor) (MTL::Private::Selector::s_k##accessor)
    -
    9
    -
    -
    10namespace MTL::Private::Class {
    -
    11_MTL_PRIVATE_DEF_CLS(MPSMatrixDescriptor);
    -
    12_MTL_PRIVATE_DEF_CLS(MPSMatrix);
    -
    13_MTL_PRIVATE_DEF_CLS(MPSVectorDescriptor);
    -
    14_MTL_PRIVATE_DEF_CLS(MPSVector);
    -
    15_MTL_PRIVATE_DEF_CLS(MPSKernel);
    -
    16_MTL_PRIVATE_DEF_CLS(MPSMatrixMultiplication);
    -
    17_MTL_PRIVATE_DEF_CLS(MPSMatrixVectorMultiplication);
    -
    18} // namespace MTL::Private::Class
    -
    -
    19
    -
    -
    20namespace MTL::Private::Selector {
    -
    21_MTL_PRIVATE_DEF_SEL(
    -
    22 matrixDescriptorWithRows_columns_rowBytes_dataType,
    -
    23 "matrixDescriptorWithRows:columns:rowBytes:dataType:");
    -
    24_MTL_PRIVATE_DEF_SEL(
    -
    25 matrixDescriptorWithRows_columns_matrices_rowBytes_matrixBytes_dataType,
    -
    26 "matrixDescriptorWithRows:columns:matrices:rowBytes:matrixBytes:dataType:");
    -
    27_MTL_PRIVATE_DEF_SEL(rows, "rows");
    -
    28_MTL_PRIVATE_DEF_SEL(initWithBuffer_descriptor, "initWithBuffer:descriptor:");
    -
    29_MTL_PRIVATE_DEF_SEL(
    -
    30 initWithDevice_,
    -
    31 "initWithDevice:transposeLeft:transposeRight:"
    -
    32 "resultRows:resultColumns:interiorColumns:alpha:beta:");
    -
    33_MTL_PRIVATE_DEF_SEL(
    -
    34 encodeToCommandBuffer_leftMatrix_rightMatrix_resultMatrix,
    -
    35 "encodeToCommandBuffer:leftMatrix:rightMatrix:resultMatrix:");
    -
    36_MTL_PRIVATE_DEF_SEL(setLeftMatrixOrigin_, "setLeftMatrixOrigin:");
    -
    37_MTL_PRIVATE_DEF_SEL(setRightMatrixOrigin_, "setRightMatrixOrigin:");
    -
    38_MTL_PRIVATE_DEF_SEL(setResultMatrixOrigin_, "setResultMatrixOrigin:");
    -
    39_MTL_PRIVATE_DEF_SEL(setBatchStart_, "setBatchStart:");
    -
    40_MTL_PRIVATE_DEF_SEL(setBatchSize_, "setBatchSize:");
    -
    41_MTL_PRIVATE_DEF_SEL(
    -
    42 vectorDescriptorWithLength_dataType,
    -
    43 "vectorDescriptorWithLength:dataType:");
    -
    44_MTL_PRIVATE_DEF_SEL(
    -
    45 vectorDescriptorWithLength_vectors_vectorBytes_dataType,
    -
    46 "vectorDescriptorWithLength:vectors:vectorBytes:dataType:");
    -
    47_MTL_PRIVATE_DEF_SEL(
    -
    48 initWithDevice_transpose_rows_columns_alpha_beta,
    -
    49 "initWithDevice:transpose:rows:columns:alpha:beta:");
    -
    50_MTL_PRIVATE_DEF_SEL(
    -
    51 encodeToCommandBuffer_inputMatrix_inputVector_resultVector,
    -
    52 "encodeToCommandBuffer:inputMatrix:inputVector:resultVector:");
    -
    53} // namespace MTL::Private::Selector
    -
    -
    54
    -
    -
    55namespace MPS {
    -
    56
    - -
    64
    -
    -
    65class MatrixDescriptor : public NS::Copying<MatrixDescriptor> {
    -
    66 public:
    -
    67 static class MatrixDescriptor* matrixDescriptor(
    -
    68 NS::UInteger rows,
    -
    69 NS::UInteger columns,
    -
    70 NS::UInteger rowBytes,
    -
    71 NS::UInteger dataType);
    -
    72 static class MatrixDescriptor* matrixDescriptor(
    -
    73 NS::UInteger rows,
    -
    74 NS::UInteger columns,
    -
    75 NS::UInteger matrices,
    -
    76 NS::UInteger rowBytes,
    -
    77 NS::UInteger matrixBytes,
    -
    78 NS::UInteger dataType);
    -
    79 NS::UInteger rows() const;
    -
    80};
    -
    -
    81
    -
    -
    82class Matrix : public NS::Referencing<Matrix> {
    -
    83 public:
    -
    84 static class Matrix* alloc();
    -
    85 Matrix* init(MTL::Buffer* buffer, MatrixDescriptor* descriptor);
    -
    86 Matrix* init(const MTL::Buffer* buffer, MatrixDescriptor* descriptor);
    -
    87};
    -
    -
    88
    -
    -
    89class Kernel : public NS::Referencing<Kernel> {
    -
    90 public:
    -
    91 NS::String* label() const;
    -
    92 MTL::Device* device() const;
    -
    93};
    -
    -
    94
    -
    -
    95class MatrixMultiplication
    -
    96 : public NS::Referencing<MatrixMultiplication, Kernel> {
    -
    97 public:
    -
    98 static class MatrixMultiplication* alloc();
    -
    99
    -
    100 MatrixMultiplication* init(
    -
    101 MTL::Device* device,
    -
    102 bool transposeLeft,
    -
    103 bool transposeRight,
    -
    104 NS::UInteger resultRows,
    -
    105 NS::UInteger resultColumns,
    -
    106 NS::UInteger interiorColumns,
    -
    107 double alpha,
    -
    108 double beta);
    -
    109
    -
    110 void encodeToCommandBuffer(
    -
    111 MTL::CommandBuffer* commandBuffer,
    -
    112 Matrix* leftMatrix,
    -
    113 Matrix* rightMatrix,
    -
    114 Matrix* resultMatrix);
    -
    115
    -
    116 void setLeftMatrixOrigin(MTL::Origin origin);
    -
    117 void setRightMatrixOrigin(MTL::Origin origin);
    -
    118 void setResultMatrixOrigin(MTL::Origin origin);
    -
    119 void setBatchStart(NS::UInteger batchStart);
    -
    120 void setBatchSize(NS::UInteger batchSize);
    -
    121};
    -
    -
    122
    -
    -
    123class VectorDescriptor : public NS::Copying<VectorDescriptor> {
    -
    124 public:
    -
    125 static class VectorDescriptor* vectorDescriptor(
    -
    126 NS::UInteger length,
    -
    127 NS::UInteger dataType);
    -
    128 static class VectorDescriptor* vectorDescriptor(
    -
    129 NS::UInteger length,
    -
    130 NS::UInteger vectors,
    -
    131 NS::UInteger vectorBytes,
    -
    132 NS::UInteger dataType);
    -
    133};
    -
    -
    134
    -
    -
    135class Vector : public NS::Referencing<Vector> {
    -
    136 public:
    -
    137 static class Vector* alloc();
    -
    138 Vector* init(MTL::Buffer* buffer, VectorDescriptor* descriptor);
    -
    139 Vector* init(const MTL::Buffer* buffer, VectorDescriptor* descriptor);
    -
    140};
    -
    -
    141
    -
    -
    142class MatrixVectorMultiplication
    -
    143 : public NS::Referencing<MatrixVectorMultiplication, Kernel> {
    -
    144 public:
    -
    145 static class MatrixVectorMultiplication* alloc();
    -
    146
    -
    147 MatrixVectorMultiplication* init(
    -
    148 MTL::Device* device,
    -
    149 bool transpose,
    -
    150 NS::UInteger rows,
    -
    151 NS::UInteger columns,
    -
    152 double alpha,
    -
    153 double beta);
    -
    154
    -
    155 void encodeToCommandBuffer(
    -
    156 MTL::CommandBuffer* commandBuffer,
    -
    157 Matrix* inputMatrix,
    -
    158 Vector* inputVector,
    -
    159 Vector* resultVector);
    -
    160};
    -
    -
    161
    -
    -
    162_MTL_INLINE MatrixDescriptor* MatrixDescriptor::matrixDescriptor(
    -
    163 NS::UInteger rows,
    -
    164 NS::UInteger columns,
    -
    165 NS::UInteger rowBytes,
    -
    166 NS::UInteger dataType) {
    -
    167 return Object::sendMessage<MatrixDescriptor*>(
    -
    168 _MPS_PRIVATE_CLS(MPSMatrixDescriptor),
    -
    169 _MPS_PRIVATE_SEL(matrixDescriptorWithRows_columns_rowBytes_dataType),
    -
    170 rows,
    -
    171 columns,
    -
    172 rowBytes,
    -
    173 dataType);
    -
    174}
    -
    -
    175
    -
    -
    176_MTL_INLINE MatrixDescriptor* MatrixDescriptor::matrixDescriptor(
    -
    177 NS::UInteger rows,
    -
    178 NS::UInteger columns,
    -
    179 NS::UInteger matrices,
    -
    180 NS::UInteger rowBytes,
    -
    181 NS::UInteger matrixBytes,
    -
    182 NS::UInteger dataType) {
    -
    183 return Object::sendMessage<MatrixDescriptor*>(
    -
    184 _MPS_PRIVATE_CLS(MPSMatrixDescriptor),
    -
    185 _MPS_PRIVATE_SEL(
    -
    186 matrixDescriptorWithRows_columns_matrices_rowBytes_matrixBytes_dataType),
    -
    187 rows,
    -
    188 columns,
    -
    189 matrices,
    -
    190 rowBytes,
    -
    191 matrixBytes,
    -
    192 dataType);
    -
    193}
    -
    -
    194
    -
    -
    195_MTL_INLINE NS::UInteger MatrixDescriptor::rows() const {
    -
    196 return Object::sendMessage<NS::UInteger>(this, _MPS_PRIVATE_SEL(rows));
    -
    197}
    -
    -
    198
    -
    -
    199_MTL_INLINE Matrix* Matrix::alloc() {
    -
    200 return NS::Object::alloc<Matrix>(_MPS_PRIVATE_CLS(MPSMatrix));
    -
    201}
    -
    -
    202
    -
    -
    203_MTL_INLINE Matrix* Matrix::init(
    -
    204 MTL::Buffer* buffer,
    -
    205 MatrixDescriptor* descriptor) {
    -
    206 return Object::sendMessage<Matrix*>(
    -
    207 this, _MPS_PRIVATE_SEL(initWithBuffer_descriptor), buffer, descriptor);
    -
    208}
    -
    -
    209
    -
    -
    210_MTL_INLINE Matrix* Matrix::init(
    -
    211 const MTL::Buffer* buffer,
    -
    212 MatrixDescriptor* descriptor) {
    -
    213 return init(const_cast<MTL::Buffer*>(buffer), descriptor);
    -
    214}
    -
    -
    215
    -
    -
    216_MTL_INLINE NS::String* Kernel::label() const {
    -
    217 return Object::sendMessage<NS::String*>(this, _MPS_PRIVATE_SEL(label));
    -
    218}
    -
    -
    219
    -
    -
    220_MTL_INLINE MTL::Device* Kernel::device() const {
    -
    221 return Object::sendMessage<MTL::Device*>(this, _MPS_PRIVATE_SEL(device));
    -
    222}
    -
    -
    223
    -
    -
    224_MTL_INLINE MatrixMultiplication* MatrixMultiplication::alloc() {
    -
    225 return NS::Object::alloc<MatrixMultiplication>(
    -
    226 _MPS_PRIVATE_CLS(MPSMatrixMultiplication));
    -
    227}
    -
    -
    228
    -
    -
    229_MTL_INLINE MatrixMultiplication* MatrixMultiplication::init(
    -
    230 MTL::Device* device,
    -
    231 bool transposeLeft,
    -
    232 bool transposeRight,
    -
    233 NS::UInteger resultRows,
    -
    234 NS::UInteger resultColumns,
    -
    235 NS::UInteger interiorColumns,
    -
    236 double alpha,
    -
    237 double beta) {
    -
    238 return Object::sendMessage<MatrixMultiplication*>(
    -
    239 this,
    -
    240 _MPS_PRIVATE_SEL(initWithDevice_),
    -
    241 device,
    -
    242 transposeLeft,
    -
    243 transposeRight,
    -
    244 resultRows,
    -
    245 resultColumns,
    -
    246 interiorColumns,
    -
    247 alpha,
    -
    248 beta);
    -
    249}
    -
    -
    250
    -
    -
    251_MTL_INLINE void MatrixMultiplication::encodeToCommandBuffer(
    -
    252 MTL::CommandBuffer* commandBuffer,
    -
    253 Matrix* leftMatrix,
    -
    254 Matrix* rightMatrix,
    -
    255 Matrix* resultMatrix) {
    -
    256 return Object::sendMessage<void>(
    -
    257 this,
    -
    258 _MPS_PRIVATE_SEL(
    -
    259 encodeToCommandBuffer_leftMatrix_rightMatrix_resultMatrix),
    -
    260 commandBuffer,
    -
    261 leftMatrix,
    -
    262 rightMatrix,
    -
    263 resultMatrix);
    -
    264}
    -
    -
    265
    -
    -
    266_MTL_INLINE void MatrixMultiplication::setLeftMatrixOrigin(MTL::Origin origin) {
    -
    267 Object::sendMessage<void>(
    -
    268 this, _MPS_PRIVATE_SEL(setLeftMatrixOrigin_), origin);
    -
    269}
    -
    -
    270
    -
    -
    271_MTL_INLINE void MatrixMultiplication::setRightMatrixOrigin(
    -
    272 MTL::Origin origin) {
    -
    273 Object::sendMessage<void>(
    -
    274 this, _MPS_PRIVATE_SEL(setRightMatrixOrigin_), origin);
    -
    275}
    -
    -
    276
    -
    -
    277_MTL_INLINE void MatrixMultiplication::setResultMatrixOrigin(
    -
    278 MTL::Origin origin) {
    -
    279 Object::sendMessage<void>(
    -
    280 this, _MPS_PRIVATE_SEL(setResultMatrixOrigin_), origin);
    -
    281}
    -
    -
    282
    -
    -
    283_MTL_INLINE void MatrixMultiplication::setBatchStart(NS::UInteger batchStart) {
    -
    284 Object::sendMessage<void>(this, _MPS_PRIVATE_SEL(setBatchStart_), batchStart);
    -
    285}
    -
    -
    286
    -
    -
    287_MTL_INLINE void MatrixMultiplication::setBatchSize(NS::UInteger batchSize) {
    -
    288 Object::sendMessage<void>(this, _MPS_PRIVATE_SEL(setBatchSize_), batchSize);
    -
    289}
    -
    -
    290
    -
    -
    291_MTL_INLINE VectorDescriptor* VectorDescriptor::vectorDescriptor(
    -
    292 NS::UInteger length,
    -
    293 NS::UInteger dataType) {
    -
    294 return Object::sendMessage<VectorDescriptor*>(
    -
    295 _MPS_PRIVATE_CLS(MPSVectorDescriptor),
    -
    296 _MPS_PRIVATE_SEL(vectorDescriptorWithLength_dataType),
    -
    297 length,
    -
    298 dataType);
    -
    299}
    -
    -
    300
    -
    -
    301_MTL_INLINE VectorDescriptor* VectorDescriptor::vectorDescriptor(
    -
    302 NS::UInteger length,
    -
    303 NS::UInteger vectors,
    -
    304 NS::UInteger vectorBytes,
    -
    305 NS::UInteger dataType) {
    -
    306 return Object::sendMessage<VectorDescriptor*>(
    -
    307 _MPS_PRIVATE_CLS(MPSVectorDescriptor),
    -
    308 _MPS_PRIVATE_SEL(vectorDescriptorWithLength_vectors_vectorBytes_dataType),
    -
    309 length,
    -
    310 vectors,
    -
    311 vectorBytes,
    -
    312 dataType);
    -
    313}
    -
    -
    314
    -
    -
    315_MTL_INLINE Vector* Vector::alloc() {
    -
    316 return NS::Object::alloc<Vector>(_MPS_PRIVATE_CLS(MPSVector));
    -
    317}
    -
    -
    318
    -
    -
    319_MTL_INLINE Vector* Vector::init(
    -
    320 MTL::Buffer* buffer,
    -
    321 VectorDescriptor* descriptor) {
    -
    322 return Object::sendMessage<Vector*>(
    -
    323 this, _MPS_PRIVATE_SEL(initWithBuffer_descriptor), buffer, descriptor);
    -
    324}
    -
    -
    325
    -
    -
    326_MTL_INLINE Vector* Vector::init(
    -
    327 const MTL::Buffer* buffer,
    -
    328 VectorDescriptor* descriptor) {
    -
    329 return init(const_cast<MTL::Buffer*>(buffer), descriptor);
    -
    330}
    -
    -
    331
    -
    -
    332_MTL_INLINE MatrixVectorMultiplication* MatrixVectorMultiplication::alloc() {
    -
    333 return NS::Object::alloc<MatrixVectorMultiplication>(
    -
    334 _MPS_PRIVATE_CLS(MPSMatrixVectorMultiplication));
    -
    335}
    -
    -
    336
    -
    -
    337_MTL_INLINE MatrixVectorMultiplication* MatrixVectorMultiplication::init(
    -
    338 MTL::Device* device,
    -
    339 bool transpose,
    -
    340 NS::UInteger rows,
    -
    341 NS::UInteger columns,
    -
    342 double alpha,
    -
    343 double beta) {
    -
    344 return Object::sendMessage<MatrixVectorMultiplication*>(
    -
    345 this,
    -
    346 _MPS_PRIVATE_SEL(initWithDevice_transpose_rows_columns_alpha_beta),
    -
    347 device,
    -
    348 transpose,
    -
    349 rows,
    -
    350 columns,
    -
    351 alpha,
    -
    352 beta);
    -
    353}
    -
    -
    354
    -
    -
    355_MTL_INLINE void MatrixVectorMultiplication::encodeToCommandBuffer(
    -
    356 MTL::CommandBuffer* commandBuffer,
    -
    357 Matrix* inputMatrix,
    -
    358 Vector* inputVector,
    -
    359 Vector* resultVector) {
    -
    360 return Object::sendMessage<void>(
    -
    361 this,
    -
    362 _MPS_PRIVATE_SEL(
    -
    363 encodeToCommandBuffer_inputMatrix_inputVector_resultVector),
    -
    364 commandBuffer,
    -
    365 inputMatrix,
    -
    366 inputVector,
    -
    367 resultVector);
    -
    368}
    -
    -
    369
    -
    370} // namespace MPS
    -
    -
    MPS::Kernel
    Definition gemm.h:89
    -
    MPS::Kernel::device
    MTL::Device * device() const
    Definition gemm.h:220
    -
    MPS::Kernel::label
    NS::String * label() const
    Definition gemm.h:216
    -
    MPS::MatrixDescriptor
    Definition gemm.h:65
    -
    MPS::MatrixDescriptor::rows
    NS::UInteger rows() const
    Definition gemm.h:195
    -
    MPS::MatrixDescriptor::matrixDescriptor
    static class MatrixDescriptor * matrixDescriptor(NS::UInteger rows, NS::UInteger columns, NS::UInteger rowBytes, NS::UInteger dataType)
    Definition gemm.h:162
    -
    MPS::Matrix
    Definition gemm.h:82
    -
    MPS::Matrix::init
    Matrix * init(MTL::Buffer *buffer, MatrixDescriptor *descriptor)
    Definition gemm.h:203
    -
    MPS::Matrix::alloc
    static class Matrix * alloc()
    Definition gemm.h:199
    -
    MPS::MatrixMultiplication
    Definition gemm.h:96
    -
    MPS::MatrixMultiplication::init
    MatrixMultiplication * init(MTL::Device *device, bool transposeLeft, bool transposeRight, NS::UInteger resultRows, NS::UInteger resultColumns, NS::UInteger interiorColumns, double alpha, double beta)
    Definition gemm.h:229
    -
    MPS::MatrixMultiplication::setBatchStart
    void setBatchStart(NS::UInteger batchStart)
    Definition gemm.h:283
    -
    MPS::MatrixMultiplication::setResultMatrixOrigin
    void setResultMatrixOrigin(MTL::Origin origin)
    Definition gemm.h:277
    -
    MPS::MatrixMultiplication::setLeftMatrixOrigin
    void setLeftMatrixOrigin(MTL::Origin origin)
    Definition gemm.h:266
    -
    MPS::MatrixMultiplication::alloc
    static class MatrixMultiplication * alloc()
    Definition gemm.h:224
    -
    MPS::MatrixMultiplication::setBatchSize
    void setBatchSize(NS::UInteger batchSize)
    Definition gemm.h:287
    -
    MPS::MatrixMultiplication::encodeToCommandBuffer
    void encodeToCommandBuffer(MTL::CommandBuffer *commandBuffer, Matrix *leftMatrix, Matrix *rightMatrix, Matrix *resultMatrix)
    Definition gemm.h:251
    -
    MPS::MatrixMultiplication::setRightMatrixOrigin
    void setRightMatrixOrigin(MTL::Origin origin)
    Definition gemm.h:271
    -
    MPS::MatrixVectorMultiplication
    Definition gemm.h:143
    -
    MPS::MatrixVectorMultiplication::init
    MatrixVectorMultiplication * init(MTL::Device *device, bool transpose, NS::UInteger rows, NS::UInteger columns, double alpha, double beta)
    Definition gemm.h:337
    -
    MPS::MatrixVectorMultiplication::encodeToCommandBuffer
    void encodeToCommandBuffer(MTL::CommandBuffer *commandBuffer, Matrix *inputMatrix, Vector *inputVector, Vector *resultVector)
    Definition gemm.h:355
    -
    MPS::MatrixVectorMultiplication::alloc
    static class MatrixVectorMultiplication * alloc()
    Definition gemm.h:332
    -
    MPS::VectorDescriptor
    Definition gemm.h:123
    -
    MPS::VectorDescriptor::vectorDescriptor
    static class VectorDescriptor * vectorDescriptor(NS::UInteger length, NS::UInteger dataType)
    Definition gemm.h:291
    -
    MPS::Vector
    Definition gemm.h:135
    -
    MPS::Vector::init
    Vector * init(MTL::Buffer *buffer, VectorDescriptor *descriptor)
    Definition gemm.h:319
    -
    MPS::Vector::alloc
    static class Vector * alloc()
    Definition gemm.h:315
    -
    _MPS_PRIVATE_SEL
    #define _MPS_PRIVATE_SEL(accessor)
    Definition gemm.h:8
    -
    _MPS_PRIVATE_CLS
    #define _MPS_PRIVATE_CLS(symbol)
    Definition gemm.h:7
    -
    MPS
    Definition gemm.h:55
    -
    MPS::DataType
    DataType
    Definition gemm.h:57
    -
    MPS::DataTypeFloat32
    @ DataTypeFloat32
    Definition gemm.h:61
    -
    MPS::DataTypeFloatBit
    @ DataTypeFloatBit
    Definition gemm.h:58
    -
    MPS::DataTypeBFloat16
    @ DataTypeBFloat16
    Definition gemm.h:62
    -
    MPS::DataTypeFloat16
    @ DataTypeFloat16
    Definition gemm.h:60
    -
    MPS::DataTypeAlternateEncodingBit
    @ DataTypeAlternateEncodingBit
    Definition gemm.h:59
    -
    MTL::Private::Class
    Definition gemm.h:10
    -
    MTL::Private::Class::_MTL_PRIVATE_DEF_CLS
    _MTL_PRIVATE_DEF_CLS(MPSMatrixDescriptor)
    -
    MTL::Private::Selector
    Definition gemm.h:20
    -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL
    _MTL_PRIVATE_DEF_SEL(matrixDescriptorWithRows_columns_rowBytes_dataType, "matrixDescriptorWithRows:columns:rowBytes:dataType:")
    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/namespace_m_p_s.html b/docs/build/html/namespace_m_p_s.html deleted file mode 100644 index b28781ded5..0000000000 --- a/docs/build/html/namespace_m_p_s.html +++ /dev/null @@ -1,160 +0,0 @@ - - - - - - - -MLX: MPS Namespace Reference - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - -
    -
    -
    -Classes | -Typedefs | -Enumerations
    -
    MPS Namespace Reference
    -
    -
    - - - - - - - - - - - - - - - - -

    -Classes

    class  Kernel
     
    class  Matrix
     
    class  MatrixDescriptor
     
    class  MatrixMultiplication
     
    class  MatrixVectorMultiplication
     
    class  Vector
     
    class  VectorDescriptor
     
    - - - -

    -Typedefs

    typedef enum MPS::DataType DataType
     
    - - - -

    -Enumerations

    enum  DataType : uint32_t {
    -  DataTypeFloatBit = 0x10000000 -, DataTypeAlternateEncodingBit = 0x80000000 -, DataTypeFloat16 = DataTypeFloatBit | 16 -, DataTypeFloat32 = DataTypeFloatBit | 32 -,
    -  DataTypeBFloat16 = DataTypeAlternateEncodingBit | DataTypeFloat16 -
    - }
     
    -

    Typedef Documentation

    - -

    ◆ DataType

    - -
    -
    - - - - -
    typedef enum MPS::DataType MPS::DataType
    -
    - -
    -
    -

    Enumeration Type Documentation

    - -

    ◆ DataType

    - -
    -
    - - - - -
    enum MPS::DataType : uint32_t
    -
    - - - - - - -
    Enumerator
    DataTypeFloatBit 
    DataTypeAlternateEncodingBit 
    DataTypeFloat16 
    DataTypeFloat32 
    DataTypeBFloat16 
    - -
    -
    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/namespace_m_t_l.html b/docs/build/html/namespace_m_t_l.html deleted file mode 100644 index 1ac55b0597..0000000000 --- a/docs/build/html/namespace_m_t_l.html +++ /dev/null @@ -1,91 +0,0 @@ - - - - - - - -MLX: MTL Namespace Reference - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - -
    -
    -
    -Namespaces
    -
    MTL Namespace Reference
    -
    -
    - - - - -

    -Namespaces

    namespace  Private
     
    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/namespace_m_t_l_1_1_private.html b/docs/build/html/namespace_m_t_l_1_1_private.html deleted file mode 100644 index f2a17676cb..0000000000 --- a/docs/build/html/namespace_m_t_l_1_1_private.html +++ /dev/null @@ -1,97 +0,0 @@ - - - - - - - -MLX: MTL::Private Namespace Reference - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - - -
    -
    -
    -Namespaces
    -
    MTL::Private Namespace Reference
    -
    -
    - - - - - - -

    -Namespaces

    namespace  Class
     
    namespace  Selector
     
    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/namespace_m_t_l_1_1_private_1_1_class.html b/docs/build/html/namespace_m_t_l_1_1_private_1_1_class.html deleted file mode 100644 index 99e9f9c46a..0000000000 --- a/docs/build/html/namespace_m_t_l_1_1_private_1_1_class.html +++ /dev/null @@ -1,227 +0,0 @@ - - - - - - - -MLX: MTL::Private::Class Namespace Reference - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - - -
    -
    -
    -Functions
    -
    MTL::Private::Class Namespace Reference
    -
    -
    - - - - - - - - - - - - - - - - -

    -Functions

     _MTL_PRIVATE_DEF_CLS (MPSMatrixDescriptor)
     
     _MTL_PRIVATE_DEF_CLS (MPSMatrix)
     
     _MTL_PRIVATE_DEF_CLS (MPSVectorDescriptor)
     
     _MTL_PRIVATE_DEF_CLS (MPSVector)
     
     _MTL_PRIVATE_DEF_CLS (MPSKernel)
     
     _MTL_PRIVATE_DEF_CLS (MPSMatrixMultiplication)
     
     _MTL_PRIVATE_DEF_CLS (MPSMatrixVectorMultiplication)
     
    -

    Function Documentation

    - -

    ◆ _MTL_PRIVATE_DEF_CLS() [1/7]

    - -
    -
    - - - - - - - -
    MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSKernel )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_CLS() [2/7]

    - -
    -
    - - - - - - - -
    MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrix )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_CLS() [3/7]

    - -
    -
    - - - - - - - -
    MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixDescriptor )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_CLS() [4/7]

    - -
    -
    - - - - - - - -
    MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixMultiplication )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_CLS() [5/7]

    - -
    -
    - - - - - - - -
    MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSMatrixVectorMultiplication )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_CLS() [6/7]

    - -
    -
    - - - - - - - -
    MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSVector )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_CLS() [7/7]

    - -
    -
    - - - - - - - -
    MTL::Private::Class::_MTL_PRIVATE_DEF_CLS (MPSVectorDescriptor )
    -
    - -
    -
    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/namespace_m_t_l_1_1_private_1_1_selector.html b/docs/build/html/namespace_m_t_l_1_1_private_1_1_selector.html deleted file mode 100644 index a3efee9d0b..0000000000 --- a/docs/build/html/namespace_m_t_l_1_1_private_1_1_selector.html +++ /dev/null @@ -1,439 +0,0 @@ - - - - - - - -MLX: MTL::Private::Selector Namespace Reference - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - - -
    -
    -
    -Functions
    -
    MTL::Private::Selector Namespace Reference
    -
    -
    - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

    -Functions

     _MTL_PRIVATE_DEF_SEL (matrixDescriptorWithRows_columns_rowBytes_dataType, "matrixDescriptorWithRows:columns:rowBytes:dataType:")
     
     _MTL_PRIVATE_DEF_SEL (matrixDescriptorWithRows_columns_matrices_rowBytes_matrixBytes_dataType, "matrixDescriptorWithRows:columns:matrices:rowBytes:matrixBytes:dataType:")
     
     _MTL_PRIVATE_DEF_SEL (rows, "rows")
     
     _MTL_PRIVATE_DEF_SEL (initWithBuffer_descriptor, "initWithBuffer:descriptor:")
     
     _MTL_PRIVATE_DEF_SEL (initWithDevice_, "initWithDevice:transposeLeft:transposeRight:" "resultRows:resultColumns:interiorColumns:alpha:beta:")
     
     _MTL_PRIVATE_DEF_SEL (encodeToCommandBuffer_leftMatrix_rightMatrix_resultMatrix, "encodeToCommandBuffer:leftMatrix:rightMatrix:resultMatrix:")
     
     _MTL_PRIVATE_DEF_SEL (setLeftMatrixOrigin_, "setLeftMatrixOrigin:")
     
     _MTL_PRIVATE_DEF_SEL (setRightMatrixOrigin_, "setRightMatrixOrigin:")
     
     _MTL_PRIVATE_DEF_SEL (setResultMatrixOrigin_, "setResultMatrixOrigin:")
     
     _MTL_PRIVATE_DEF_SEL (setBatchStart_, "setBatchStart:")
     
     _MTL_PRIVATE_DEF_SEL (setBatchSize_, "setBatchSize:")
     
     _MTL_PRIVATE_DEF_SEL (vectorDescriptorWithLength_dataType, "vectorDescriptorWithLength:dataType:")
     
     _MTL_PRIVATE_DEF_SEL (vectorDescriptorWithLength_vectors_vectorBytes_dataType, "vectorDescriptorWithLength:vectors:vectorBytes:dataType:")
     
     _MTL_PRIVATE_DEF_SEL (initWithDevice_transpose_rows_columns_alpha_beta, "initWithDevice:transpose:rows:columns:alpha:beta:")
     
     _MTL_PRIVATE_DEF_SEL (encodeToCommandBuffer_inputMatrix_inputVector_resultVector, "encodeToCommandBuffer:inputMatrix:inputVector:resultVector:")
     
    -

    Function Documentation

    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [1/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (encodeToCommandBuffer_inputMatrix_inputVector_resultVector ,
    "encodeToCommandBuffer:inputMatrix:inputVector:resultVector:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [2/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (encodeToCommandBuffer_leftMatrix_rightMatrix_resultMatrix ,
    "encodeToCommandBuffer:leftMatrix:rightMatrix:resultMatrix:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [3/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithBuffer_descriptor ,
    "initWithBuffer:descriptor:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [4/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithDevice_ ,
    "initWithDevice:transposeLeft:transposeRight:" "resultRows:resultColumns:interiorColumns:alpha:beta:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [5/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (initWithDevice_transpose_rows_columns_alpha_beta ,
    "initWithDevice:transpose:rows:columns:alpha:beta:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [6/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (matrixDescriptorWithRows_columns_matrices_rowBytes_matrixBytes_dataType ,
    "matrixDescriptorWithRows:columns:matrices:rowBytes:matrixBytes:dataType:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [7/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (matrixDescriptorWithRows_columns_rowBytes_dataType ,
    "matrixDescriptorWithRows:columns:rowBytes:dataType:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [8/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (rows ,
    "rows"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [9/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setBatchSize_ ,
    "setBatchSize:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [10/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setBatchStart_ ,
    "setBatchStart:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [11/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setLeftMatrixOrigin_ ,
    "setLeftMatrixOrigin:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [12/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setResultMatrixOrigin_ ,
    "setResultMatrixOrigin:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [13/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (setRightMatrixOrigin_ ,
    "setRightMatrixOrigin:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [14/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (vectorDescriptorWithLength_dataType ,
    "vectorDescriptorWithLength:dataType:"  )
    -
    - -
    -
    - -

    ◆ _MTL_PRIVATE_DEF_SEL() [15/15]

    - -
    -
    - - - - - - - - - - - -
    MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL (vectorDescriptorWithLength_vectors_vectorBytes_dataType ,
    "vectorDescriptorWithLength:vectors:vectorBytes:dataType:"  )
    -
    - -
    -
    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/namespacemembers.html b/docs/build/html/namespacemembers.html index e3075eacf9..6b0ac0cace 100644 --- a/docs/build/html/namespacemembers.html +++ b/docs/build/html/namespacemembers.html @@ -73,9 +73,49 @@ $(function() {
    Here is a list of all namespace members with links to the namespace documentation for each member:
    -

    - _ -

    diff --git a/docs/build/html/namespacemembers_a.html b/docs/build/html/namespacemembers_a.html deleted file mode 100644 index 337149156d..0000000000 --- a/docs/build/html/namespacemembers_a.html +++ /dev/null @@ -1,124 +0,0 @@ - - - - - - - -MLX: Namespace Members - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - -
    - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - -
    -
    Here is a list of all namespace members with links to the namespace documentation for each member:
    - -

    - a -

    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/namespacemembers_b.html b/docs/build/html/namespacemembers_b.html index 24a73680f9..7392c31cfe 100644 --- a/docs/build/html/namespacemembers_b.html +++ b/docs/build/html/namespacemembers_b.html @@ -79,8 +79,8 @@ $(function() {
  • bfloat16 : mlx::core
  • bfloat16_t : mlx::core
  • binary() : mlx::core::metal
    • -
  • binary_op_gpu() : mlx::core
    • -
  • binary_op_gpu_inplace() : mlx::core
    • +
  • binary_op_gpu() : mlx::core
    • +
  • binary_op_gpu_inplace() : mlx::core
  • binary_ops() : mlx::core::metal
  • binary_two() : mlx::core::metal
  • bits() : mlx::core::random
    • diff --git a/docs/build/html/namespacemembers_c.html b/docs/build/html/namespacemembers_c.html index a2be847233..dff3139eb5 100644 --- a/docs/build/html/namespacemembers_c.html +++ b/docs/build/html/namespacemembers_c.html @@ -84,6 +84,7 @@ $(function() {
  • check_shape_dim() : mlx::core
  • checkpoint : mlx::core
  • cholesky() : mlx::core::linalg
    • +
  • cholesky_inv() : mlx::core::linalg
  • clear_cache() : mlx::core::metal
  • clip() : mlx::core
  • collapse_contiguous_dims() : mlx::core
    • diff --git a/docs/build/html/namespacemembers_d.html b/docs/build/html/namespacemembers_d.html index 34f1413880..d55554e5df 100644 --- a/docs/build/html/namespacemembers_d.html +++ b/docs/build/html/namespacemembers_d.html @@ -74,13 +74,9 @@ $(function() {
    Here is a list of all namespace members with links to the namespace documentation for each member:

    - d -

    diff --git a/docs/build/html/namespacemembers_enum.html b/docs/build/html/namespacemembers_enum.html index ca5e914409..cd9598a274 100644 --- a/docs/build/html/namespacemembers_enum.html +++ b/docs/build/html/namespacemembers_enum.html @@ -74,7 +74,6 @@ $(function() {
    Here is a list of all namespace enums with links to the namespace documentation for each enum:
    diff --git a/docs/build/html/namespacemembers_eval.html b/docs/build/html/namespacemembers_eval.html index 417bd48a83..2dd42ff375 100644 --- a/docs/build/html/namespacemembers_eval.html +++ b/docs/build/html/namespacemembers_eval.html @@ -75,11 +75,6 @@ $(function() {
  • ContiguousAllReduce : mlx::core
  • ContiguousReduce : mlx::core
  • ContiguousStridedReduce : mlx::core
    • -
  • DataTypeAlternateEncodingBit : MPS
    • -
  • DataTypeBFloat16 : MPS
    • -
  • DataTypeFloat16 : MPS
    • -
  • DataTypeFloat32 : MPS
    • -
  • DataTypeFloatBit : MPS
  • GeneralContiguousReduce : mlx::core
  • GeneralReduce : mlx::core
  • GeneralStridedReduce : mlx::core
    • diff --git a/docs/build/html/namespacemembers_func.html b/docs/build/html/namespacemembers_func.html index 58c784a672..04114ecf90 100644 --- a/docs/build/html/namespacemembers_func.html +++ b/docs/build/html/namespacemembers_func.html @@ -73,9 +73,48 @@ $(function() {
    Here is a list of all namespace functions with links to the namespace documentation for each function:
    -

    - _ -

    diff --git a/docs/build/html/namespacemembers_func_a.html b/docs/build/html/namespacemembers_func_a.html deleted file mode 100644 index f50d428a42..0000000000 --- a/docs/build/html/namespacemembers_func_a.html +++ /dev/null @@ -1,123 +0,0 @@ - - - - - - - -MLX: Namespace Members - - - - - - - - - - - -
    -
    - - - - - - -
    -
    MLX -
    -
    -
    - - - - - - - -
    - -
    -
    - - -
    -
    -
    -
    -
    -
    Loading...
    -
    Searching...
    -
    No Matches
    -
    -
    -
    -
    - -
    -
    Here is a list of all namespace functions with links to the namespace documentation for each function:
    - -

    - a -

    -
    - -
    -Generated by doxygen 1.10.0 -
    - - diff --git a/docs/build/html/namespacemembers_func_b.html b/docs/build/html/namespacemembers_func_b.html index 1a677bac55..f23f31f8bc 100644 --- a/docs/build/html/namespacemembers_func_b.html +++ b/docs/build/html/namespacemembers_func_b.html @@ -76,8 +76,8 @@ $(function() {

    - b -

    • bernoulli() : mlx::core::random
    • binary() : mlx::core::metal
      • -
    • binary_op_gpu() : mlx::core
      • -
    • binary_op_gpu_inplace() : mlx::core
      • +
    • binary_op_gpu() : mlx::core
      • +
    • binary_op_gpu_inplace() : mlx::core
    • binary_ops() : mlx::core::metal
    • binary_two() : mlx::core::metal
    • bits() : mlx::core::random
      • diff --git a/docs/build/html/namespacemembers_func_c.html b/docs/build/html/namespacemembers_func_c.html index 35861765ce..a699671ffd 100644 --- a/docs/build/html/namespacemembers_func_c.html +++ b/docs/build/html/namespacemembers_func_c.html @@ -81,6 +81,7 @@ $(function() {
    • check_contiguity() : mlx::core
    • check_shape_dim() : mlx::core
    • cholesky() : mlx::core::linalg
      • +
    • cholesky_inv() : mlx::core::linalg
    • clear_cache() : mlx::core::metal
    • clip() : mlx::core
    • collapse_contiguous_dims() : mlx::core
      • diff --git a/docs/build/html/namespacemembers_func_d.html b/docs/build/html/namespacemembers_func_d.html index 69815108ed..764ae04403 100644 --- a/docs/build/html/namespacemembers_func_d.html +++ b/docs/build/html/namespacemembers_func_d.html @@ -75,6 +75,8 @@ $(function() {

      - d -

      diff --git a/docs/build/html/namespacemembers_func_g.html b/docs/build/html/namespacemembers_func_g.html index 59e8726930..dab191043b 100644 --- a/docs/build/html/namespacemembers_func_g.html +++ b/docs/build/html/namespacemembers_func_g.html @@ -78,21 +78,25 @@ $(function() {
    • gather_mm() : mlx::core
    • gather_qmm() : mlx::core
    • gemm() : mlx::core::metal
      • +
    • gemv_masked() : mlx::core::metal
    • general_c2r() : pocketfft::detail
    • general_nd() : pocketfft::detail
    • general_r2c() : pocketfft::detail
      • +
    • get_2d_grid_dims() : mlx::core
    • get_active_memory() : mlx::core::metal
    • get_arange_kernel() : mlx::core
    • get_binary_kernel() : mlx::core
    • get_binary_two_kernel() : mlx::core
      • +
    • get_block_dims() : mlx::core
    • get_cache_memory() : mlx::core::metal
      • -
    • get_colocated_mtllib_path() : mlx::core::metal
    • get_copy_kernel() : mlx::core
    • get_fft_kernel() : mlx::core
      • +
    • get_gemv_masked_kernel() : mlx::core
    • get_mb_sort_kernel() : mlx::core
    • get_peak_memory() : mlx::core::metal
    • get_plan() : pocketfft::detail
    • get_pool() : pocketfft::detail::threading
      • +
    • get_primitive_string() : mlx::core
    • get_quantized_kernel() : mlx::core
    • get_reduce_init_kernel() : mlx::core
    • get_reduce_kernel() : mlx::core
      • diff --git a/docs/build/html/namespacemembers_func_i.html b/docs/build/html/namespacemembers_func_i.html index 9e9d91a91a..fd5a6478fa 100644 --- a/docs/build/html/namespacemembers_func_i.html +++ b/docs/build/html/namespacemembers_func_i.html @@ -85,7 +85,6 @@ $(function() {
    • irfft2() : mlx::core::fft
    • irfftn() : mlx::core::fft
    • is_available() : mlx::core::distributed, mlx::core::metal
      • -
    • is_big_endian() : mlx::core
    • is_power_of_2() : mlx::core
    • is_same_shape() : mlx::core
    • is_scalar() : mlx::core
      • diff --git a/docs/build/html/namespacemembers_func_m.html b/docs/build/html/namespacemembers_func_m.html index d75ac56195..9e5e8f5e9d 100644 --- a/docs/build/html/namespacemembers_func_m.html +++ b/docs/build/html/namespacemembers_func_m.html @@ -75,6 +75,7 @@ $(function() {

      - m -

      • make_contiguous_strides() : mlx::core
        • +
      • make_string() : mlx::core
      • make_synchronize_task() : mlx::core::metal
      • make_task() : mlx::core::metal
      • malloc() : mlx::core::allocator
        • diff --git a/docs/build/html/namespacemembers_func_p.html b/docs/build/html/namespacemembers_func_p.html index 00740a69fa..14e0e2cefa 100644 --- a/docs/build/html/namespacemembers_func_p.html +++ b/docs/build/html/namespacemembers_func_p.html @@ -74,7 +74,7 @@ $(function() {
        Here is a list of all namespace functions with links to the namespace documentation for each function:

        - p -

        diff --git a/docs/build/html/namespacemembers_g.html b/docs/build/html/namespacemembers_g.html index 8e7fd0ebdb..b21199d586 100644 --- a/docs/build/html/namespacemembers_g.html +++ b/docs/build/html/namespacemembers_g.html @@ -78,6 +78,7 @@ $(function() {
      • gather_mm() : mlx::core
      • gather_qmm() : mlx::core
      • gemm() : mlx::core::metal
        • +
      • gemv_masked() : mlx::core::metal
      • general_c2r() : pocketfft::detail
      • general_nd() : pocketfft::detail
      • general_r2c() : pocketfft::detail
        • @@ -85,18 +86,21 @@ $(function() {
      • GeneralReduce : mlx::core
      • GeneralStridedReduce : mlx::core
      • generic : mlx::core
        • +
      • get_2d_grid_dims() : mlx::core
      • get_active_memory() : mlx::core::metal
      • get_arange_kernel() : mlx::core
      • get_binary_kernel() : mlx::core
      • get_binary_two_kernel() : mlx::core
        • +
      • get_block_dims() : mlx::core
      • get_cache_memory() : mlx::core::metal
        • -
      • get_colocated_mtllib_path() : mlx::core::metal
      • get_copy_kernel() : mlx::core
      • get_fft_kernel() : mlx::core
        • +
      • get_gemv_masked_kernel() : mlx::core
      • get_mb_sort_kernel() : mlx::core
      • get_peak_memory() : mlx::core::metal
      • get_plan() : pocketfft::detail
      • get_pool() : pocketfft::detail::threading
        • +
      • get_primitive_string() : mlx::core
      • get_quantized_kernel() : mlx::core
      • get_reduce_init_kernel() : mlx::core
      • get_reduce_kernel() : mlx::core
        • diff --git a/docs/build/html/namespacemembers_i.html b/docs/build/html/namespacemembers_i.html index 0e09e532ff..8822ea26a6 100644 --- a/docs/build/html/namespacemembers_i.html +++ b/docs/build/html/namespacemembers_i.html @@ -93,7 +93,6 @@ $(function() {
      • is_array_v : mlx::core
      • is_arrays_v : mlx::core
      • is_available() : mlx::core::distributed, mlx::core::metal
        • -
      • is_big_endian() : mlx::core
      • is_power_of_2() : mlx::core
      • is_same_shape() : mlx::core
      • is_scalar() : mlx::core
        • diff --git a/docs/build/html/namespacemembers_m.html b/docs/build/html/namespacemembers_m.html index 354a1f0f03..8bb44d3d88 100644 --- a/docs/build/html/namespacemembers_m.html +++ b/docs/build/html/namespacemembers_m.html @@ -75,6 +75,7 @@ $(function() {

        - m -

        • make_contiguous_strides() : mlx::core
          • +
        • make_string() : mlx::core
        • make_synchronize_task() : mlx::core::metal
        • make_task() : mlx::core::metal
        • malloc() : mlx::core::allocator
          • diff --git a/docs/build/html/namespacemembers_p.html b/docs/build/html/namespacemembers_p.html index cd69cc29b2..8348c45bab 100644 --- a/docs/build/html/namespacemembers_p.html +++ b/docs/build/html/namespacemembers_p.html @@ -74,7 +74,7 @@ $(function() {
          Here is a list of all namespace members with links to the namespace documentation for each member:

          - p -

          diff --git a/docs/build/html/namespacemembers_type.html b/docs/build/html/namespacemembers_type.html index 9d45cf3e12..f51ee9f706 100644 --- a/docs/build/html/namespacemembers_type.html +++ b/docs/build/html/namespacemembers_type.html @@ -74,7 +74,6 @@ $(function() {
          Here is a list of all namespace typedefs with links to the namespace documentation for each typedef:
          • add_vec_t : pocketfft::detail
          • bfloat16_t : mlx::core
            • -
          • DataType : MPS
          • deleter_t : mlx::core
          • enable_for_arrays_t : mlx::core
          • float16_t : mlx::core
            • diff --git a/docs/build/html/namespacemlx_1_1core.html b/docs/build/html/namespacemlx_1_1core.html index ef47cec4c0..fb9af74f63 100644 --- a/docs/build/html/namespacemlx_1_1core.html +++ b/docs/build/html/namespacemlx_1_1core.html @@ -463,14 +463,14 @@ Functions template<typename stride_t > auto check_contiguity (const std::vector< int > &shape, const std::vector< stride_t > &strides)   -void binary_op_gpu (const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s) -  -void binary_op_gpu (const std::vector< array > &inputs, array &out, const std::string op, const Stream &s) -  -void binary_op_gpu_inplace (const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s) -  -void binary_op_gpu_inplace (const std::vector< array > &inputs, array &out, const std::string op, const Stream &s) -  +void binary_op_gpu (const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s) +  +void binary_op_gpu (const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s) +  +void binary_op_gpu_inplace (const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s) +  +void binary_op_gpu_inplace (const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s) +  template<typename stride_t > void copy_gpu_inplace (const array &in, array &out, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s)   @@ -516,6 +516,8 @@ Functions   MTL::ComputePipelineState * get_steel_conv_kernel (metal::Device &d, const std::string &kernel_name, const array &out, int bm, int bn, int bk, int wm, int wn, int n_channel_specialization, bool small_filter)   +MTL::ComputePipelineState * get_gemv_masked_kernel (metal::Device &d, const std::string &kernel_name, const array &out, const std::optional< array > &mask_out, const std::optional< array > &mask_op, bool transpose_mat, int bm, int bn, int sm, int sn, int tm, int tn, bool contiguous) +  MTL::ComputePipelineState * get_steel_conv_general_kernel (metal::Device &d, const std::string &kernel_name, const array &out, int bm, int bn, int bk, int wm, int wn)   MTL::ComputePipelineState * get_fft_kernel (metal::Device &d, const std::string &kernel_name, const std::string &hash_name, const metal::MTLFCList &func_consts, const std::string &template_def) @@ -549,6 +551,26 @@ Functions   void unary_op_gpu_inplace (const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)   +template<typename T > +void set_vector_bytes (CommandEncoder &enc, const std::vector< T > &vec, size_t nelems, int idx) +  +template<typename T > +void set_vector_bytes (CommandEncoder &enc, const std::vector< T > &vec, int idx) +  +std::string type_to_name (const array &a) +  +MTL::Size get_block_dims (int dim0, int dim1, int dim2) +  +MTL::Size get_2d_grid_dims (const std::vector< int > &shape, const std::vector< size_t > &strides) +  +NS::String * make_string (std::ostringstream &os) +  +void debug_set_stream_queue_label (MTL::CommandQueue *queue, int index) +  +void debug_set_primitive_buffer_label (MTL::CommandBuffer *command_buffer, Primitive &primitive) +  +std::string get_primitive_string (Primitive *primitive) +  void disable_compile ()  Globally disable compilation.
              @@ -580,10 +602,6 @@ Functions   Dtype::Kind kindof (const Dtype &t)   -std::string dtype_to_array_protocol (const Dtype &t) -  -Dtype dtype_from_array_protocol (std::string_view t) -  std::pair< std::vector< std::vector< int > >, std::string > einsum_path (const std::string &subscripts, const std::vector< array > &operands)   array einsum (const std::string &subscripts, const std::vector< array > &operands, StreamOrDevice s={}) @@ -790,16 +808,16 @@ Functions array moveaxis (const array &a, int source, int destination, StreamOrDevice s={})  Move an axis of an array.
              -array pad (const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), StreamOrDevice s={}) - Pad an array with a constant value.
            -  -array pad (const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), StreamOrDevice s={}) - Pad an array with a constant value along all axes.
            -  -array pad (const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), StreamOrDevice s={}) -  -array pad (const array &a, int pad_width, const array &pad_value=array(0), StreamOrDevice s={}) -  +array pad (const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={}) + Pad an array with a constant value.
            +  +array pad (const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={}) + Pad an array with a constant value along all axes.
            +  +array pad (const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={}) +  +array pad (const array &a, int pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={}) +  array transpose (const array &a, StreamOrDevice s={})  Permutes the dimensions in reverse order.
              @@ -2209,8 +2227,6 @@ Functions int check_shape_dim (const T dim)  Returns the shape dimension if it's within allowed range.
              -bool is_big_endian () -  int normalize_axis (int axis, int ndim)  Returns the axis normalized to be in the range [0, ndim).
              @@ -2652,8 +2668,8 @@ template<typename... T> - -

            ◆ binary_op_gpu() [1/2]

            + +

            ◆ binary_op_gpu() [1/2]

            @@ -2671,7 +2687,7 @@ template<typename... T>
            - const std::string op, + const std::string & op, @@ -2683,8 +2699,8 @@ template<typename... T>
            - -

            ◆ binary_op_gpu() [2/2]

            + +

            ◆ binary_op_gpu() [2/2]

            @@ -2702,7 +2718,7 @@ template<typename... T>
            - const std::string op, + const std::string & op, @@ -2714,8 +2730,8 @@ template<typename... T>
            - -

            ◆ binary_op_gpu_inplace() [1/2]

            + +

            ◆ binary_op_gpu_inplace() [1/2]

            @@ -2733,7 +2749,7 @@ template<typename... T>
            - const std::string op, + const std::string & op, @@ -2745,8 +2761,8 @@ template<typename... T>
            - -

            ◆ binary_op_gpu_inplace() [2/2]

            + +

            ◆ binary_op_gpu_inplace() [2/2]

            @@ -2764,7 +2780,7 @@ template<typename... T>
            - const std::string op, + const std::string & op, @@ -3346,6 +3362,64 @@ template<typename stride_t >
            +
            + + +

            ◆ debug_set_primitive_buffer_label()

            + +
            +
            + + + + + +
            + + + + + + + + + + + +
            void mlx::core::debug_set_primitive_buffer_label (MTL::CommandBuffer * command_buffer,
            Primitive & primitive )
            +             		+inline
            +
            + +
            +
            + +

            ◆ debug_set_stream_queue_label()

            + +
            +
            + + + + + +
            + + + + + + + + + + + +
            void mlx::core::debug_set_stream_queue_label (MTL::CommandQueue * queue,
            int index )
            +             		+inline
            +
            +
            @@ -3427,40 +3501,6 @@ template<typename stride_t >

            Globally disable compilation.

            Setting the environment variable MLX_DISABLE_COMPILE can also be used to disable compilation.

            - - - -

            ◆ dtype_from_array_protocol()

            - -
            -
            - - - - - - - -
            Dtype mlx::core::dtype_from_array_protocol (std::string_view t)
            -
            - -
            -
            - -

            ◆ dtype_to_array_protocol()

            - -
            -
            - - - - - - - -
            std::string mlx::core::dtype_to_array_protocol (const Dtype & t)
            -
            -
            @@ -3673,6 +3713,27 @@ template<typename... Arrays, typename = enable_for_arrays_t<Arrays...>
            +
            + + +

            ◆ get_2d_grid_dims()

            + +
            +
            + + + + + + + + + + + +
            MTL::Size mlx::core::get_2d_grid_dims (const std::vector< int > & shape,
            const std::vector< size_t > & strides )
            +
            +
            @@ -3771,6 +3832,32 @@ template<typename... Arrays, typename = enable_for_arrays_t<Arrays...>
            +
            + + +

            ◆ get_block_dims()

            + +
            +
            + + + + + + + + + + + + + + + + +
            MTL::Size mlx::core::get_block_dims (int dim0,
            int dim1,
            int dim2 )
            +
            +
            @@ -3838,6 +3925,82 @@ template<typename... Arrays, typename = enable_for_arrays_t<Arrays...>
            +
            + + +

            ◆ get_gemv_masked_kernel()

            + +
            +
            + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
            MTL::ComputePipelineState * mlx::core::get_gemv_masked_kernel (metal::Device & d,
            const std::string & kernel_name,
            const array & out,
            const std::optional< array > & mask_out,
            const std::optional< array > & mask_op,
            bool transpose_mat,
            int bm,
            int bn,
            int sm,
            int sn,
            int tm,
            int tn,
            bool contiguous )
            +
            +
            @@ -3879,6 +4042,23 @@ template<typename... Arrays, typename = enable_for_arrays_t<Arrays...>
            +
            + + +

            ◆ get_primitive_string()

            + +
            +
            + + + + + + + +
            std::string mlx::core::get_primitive_string (Primitive * primitive)
            +
            +
            @@ -4711,31 +4891,6 @@ template<typename... Args>
            -
            - - -

            ◆ is_big_endian()

            - -
            -
            - - - - - -
            - - - - - - - -
            bool mlx::core::is_big_endian ()
            -             		-inline
            -
            -
            @@ -5118,6 +5273,31 @@ template<typename stride_t >
            +
            + + +

            ◆ make_string()

            + +
            +
            + + + + + +
            + + + + + + + +
            NS::String * mlx::core::make_string (std::ostringstream & os)
            +             		+inline
            +
            +
            @@ -16528,6 +16708,83 @@ template<typename T , typename U , typename OpS , typename OpC , typename Op

            Make the stream the default for its device.

            + + + +

            ◆ set_vector_bytes() [1/2]

            + +
            +
            +
            +template<typename T >
            + + + + + +
            + + + + + + + + + + + + + + + + +
            void mlx::core::set_vector_bytes (CommandEncoder & enc,
            const std::vector< T > & vec,
            int idx )
            +             		+inline
            +
            + +
            +
            + +

            ◆ set_vector_bytes() [2/2]

            + +
            +
            +
            +template<typename T >
            + + + + + +
            + + + + + + + + + + + + + + + + + + + + + +
            void mlx::core::set_vector_bytes (CommandEncoder & enc,
            const std::vector< T > & vec,
            size_t nelems,
            int idx )
            +             		+inline
            +
            +
            @@ -17004,6 +17261,23 @@ template<typename T , typename U , typename OpS , typename OpC , typename Op
            +
            + + +

            ◆ type_to_name()

            + +
            +
            + + + + + + + +
            std::string mlx::core::type_to_name (const array & a)
            +
            +
            diff --git a/docs/build/html/namespacemlx_1_1core_1_1fast.html b/docs/build/html/namespacemlx_1_1core_1_1fast.html index 4a27785365..b874c22307 100644 --- a/docs/build/html/namespacemlx_1_1core_1_1fast.html +++ b/docs/build/html/namespacemlx_1_1core_1_1fast.html @@ -84,6 +84,8 @@ $(function() { + + @@ -110,8 +112,127 @@ Functions + + + + + +

            Classes

            class  AffineQuantize
             
            class  Custom
             
            class  LayerNorm
            array scaled_dot_product_attention (const array &queries, const array &keys, const array &values, const float scale, const std::optional< array > &mask=std::nullopt, StreamOrDevice s={})
             Computes: O = softmax(Q @ K.T) @ V.
             
            std::tuple< array, array, array > affine_quantize (const array &w, int group_size=64, int bits=4, StreamOrDevice s={})
             
            array affine_quantize (const array &w, const array &scales, const array &biases, int group_size=64, int bits=4, StreamOrDevice s={})
             
            array affine_dequantize (const array &w, const array &scales, const array &biases, int group_size=64, int bits=4, StreamOrDevice s={})
             

            Function Documentation

            + +

            ◆ affine_dequantize()

            + +
            +
            + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
            array mlx::core::fast::affine_dequantize (const array & w,
            const array & scales,
            const array & biases,
            int group_size = 64,
            int bits = 4,
            StreamOrDevice s = {} )
            +
            + +
            +
            + +

            ◆ affine_quantize() [1/2]

            + +
            +
            + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
            array mlx::core::fast::affine_quantize (const array & w,
            const array & scales,
            const array & biases,
            int group_size = 64,
            int bits = 4,
            StreamOrDevice s = {} )
            +
            + +
            +
            + +

            ◆ affine_quantize() [2/2]

            + +
            +
            + + + + + + + + + + + + + + + + + + + + + +
            std::tuple< array, array, array > mlx::core::fast::affine_quantize (const array & w,
            int group_size = 64,
            int bits = 4,
            StreamOrDevice s = {} )
            +
            + +
            +

            ◆ layer_norm()

            diff --git a/docs/build/html/namespacemlx_1_1core_1_1linalg.html b/docs/build/html/namespacemlx_1_1core_1_1linalg.html index 8520bbfdcc..40eb7e672a 100644 --- a/docs/build/html/namespacemlx_1_1core_1_1linalg.html +++ b/docs/build/html/namespacemlx_1_1core_1_1linalg.html @@ -102,8 +102,12 @@ Functions   array inv (const array &a, StreamOrDevice s={})   +array tri_inv (const array &a, bool upper=false, StreamOrDevice s={}) +  array cholesky (const array &a, bool upper=false, StreamOrDevice s={})   +array cholesky_inv (const array &a, bool upper=false, StreamOrDevice s={}) + 

            Function Documentation

            @@ -130,6 +134,32 @@ Functions
            +
            + + +

            ◆ cholesky_inv()

            + +
            +
            + + + + + + + + + + + + + + + + +
            array mlx::core::linalg::cholesky_inv (const array & a,
            bool upper = false,
            StreamOrDevice s = {} )
            +
            +
            @@ -431,6 +461,32 @@ Functions
            +
            + + +

            ◆ tri_inv()

            + +
            +
            + + + + + + + + + + + + + + + + +
            array mlx::core::linalg::tri_inv (const array & a,
            bool upper = false,
            StreamOrDevice s = {} )
            +
            +
            diff --git a/docs/build/html/namespacemlx_1_1core_1_1metal.html b/docs/build/html/namespacemlx_1_1core_1_1metal.html index fa8ca447dd..6b25727f88 100644 --- a/docs/build/html/namespacemlx_1_1core_1_1metal.html +++ b/docs/build/html/namespacemlx_1_1core_1_1metal.html @@ -102,8 +102,6 @@ Typedefs Functions MetalAllocator & allocator ()   -std::string get_colocated_mtllib_path (const std::string &lib_name) -  Device & device (mlx::core::Device)   const char * utils () @@ -160,6 +158,8 @@ Functions   const char * steel_conv_general ()   +const char * gemv_masked () +  bool is_available ()   size_t get_active_memory () @@ -435,6 +435,23 @@ Variables
            +
            + + +

            ◆ gemv_masked()

            + +
            +
            + + + + + + + +
            const char * mlx::core::metal::gemv_masked ()
            +
            +
            @@ -469,31 +486,6 @@ Variables
            -
            - - -

            ◆ get_colocated_mtllib_path()

            - -
            -
            - - - - - -
            - - - - - - - -
            std::string mlx::core::metal::get_colocated_mtllib_path (const std::string & lib_name)
            -             		-inline
            -
            -
            diff --git a/docs/build/html/namespaces.html b/docs/build/html/namespaces.html index 8cfc9efdcb..9763764c06 100644 --- a/docs/build/html/namespaces.html +++ b/docs/build/html/namespaces.html @@ -154,13 +154,14 @@ $(function() {  CDistPrimitive  CGroupA distributed::Group represents a group of independent mlx processes that can communicate  ►Nfast - CCustom - CLayerNorm - CLayerNormVJP - CRMSNorm - CRMSNormVJP - CRoPE - CScaledDotProductAttention + CAffineQuantize + CCustom + CLayerNorm + CLayerNormVJP + CRMSNorm + CRMSNormVJP + CRoPE + CScaledDotProductAttention  Nfft  ►Nio  CFileReader @@ -323,52 +324,40 @@ $(function() {  CTransformAdd  CTransformAxpby  CTransformNone - ▼NMPS - CKernel - CMatrix - CMatrixDescriptor - CMatrixMultiplication - CMatrixVectorMultiplication - CVector - CVectorDescriptor - ▼NMTL - ►NPrivate - NClass - NSelector - ▼Npocketfft - ►Ndetail - ►Nthreading - Caligned_allocator - Cconcurrent_queue - Clatch - Cthread_pool - Cadd_vec - Cadd_vec< cmplx< T > > - Carr - Carr_info - Ccfftp - Ccmplx - Ccndarr - CExecC2C - CExecDcst - CExecHartley - CExecR2R - Cfftblue - Cmulti_iter - Cndarr - Cpocketfft_c - Cpocketfft_r - Crev_iter - Crfftp - Csimple_iter - Csincos_2pibyn - CT_dcst23 - CT_dcst4 - CT_dct1 - CT_dst1 - Cutil - CVLEN - CVTYPE + ▼Npocketfft + ►Ndetail + ►Nthreading + Caligned_allocator + Cconcurrent_queue + Clatch + Cthread_pool + Cadd_vec + Cadd_vec< cmplx< T > > + Carr + Carr_info + Ccfftp + Ccmplx + Ccndarr + CExecC2C + CExecDcst + CExecHartley + CExecR2R + Cfftblue + Cmulti_iter + Cndarr + Cpocketfft_c + Cpocketfft_r + Crev_iter + Crfftp + Csimple_iter + Csincos_2pibyn + CT_dcst23 + CT_dcst4 + CT_dct1 + CT_dst1 + Cutil + CVLEN + CVTYPE diff --git a/docs/build/html/objects.inv b/docs/build/html/objects.inv index 16ea7e782006a6efbddc1dce058b19fe4af9df02..7abac4b34507de0d7d4dabcd91ffc6c27801c471 100644 GIT binary patch literal 25261 zcmZsCWmFu&)-CQbxWnKQ+zIY30|a+>hv0$1-JQT7!GpWITW}37!96%H_rCY%TkreR z)xD~^>YT3H=j>g*iW;clVq@#%4YYOgvUKurcJl#xS-QE~Iy(V5QK*4xZq9a=<{qDc z%1S!_9<}~&)n_0(D+e#@e;R1oSOVQF9ZhYWEG*rC&Q?GV8(Vjvm92v%@Ly@}?C9cV z>F#c60rYgYb+QKfI@p@AqIj=ucIh{44A-$fP8;_jUuH#Q>2O{U3<=Qp>nC@S;afT- zVa&{XETC@!I)dp?y5yJ+M^y1=3AokZH4B5K?WmaDdDNZ)bX%$>+k@4fvdI#*tV_X* zKDwP=i!-7bvQL}UB4;0sPdn%9N4@l#@6QQ|ns`sU2$v@ci36gQ6hrRQgQun1R}U|j zV`S6%>PZ9s>}|M&M_Q|UuP~e#;o~L342H0pteC$1I;mDGG!CZiA0MZpoR9g)grNEc zX6B-Xs7*I_fax-tRL4rw7%gOc2n~sKT$m?^0@$bc74JMwtZ^9pc=-ZJ_I2qcY z5iBQ5#_j1cE!(D~jPV_dNkviHoGwI<+;r1KkzDtmObweTD0Sf%49MNas%xFfS@m;c z(y7-o)O9jT_mi71zq*Ko#~!!d5}mm9d`33>F77Xs^i~2KI#F6w#LYB_DR?yF$;L0r z;~*Zas3`(vh>xMOmx0_XI<3IE>FY{&mcL}iu?|=#$ArfXYe7V!$J1&rSjk_+rVc1G z&QMO%PwI!-Oa!!wSJwDZK{?56!C3D)-vpv*6Y)XT@&V}}8-chRTK05kjw1gLVeQo> z$A$DM8@dy{+WA)Om^PZFn>Ch--^9->mD@MOhl*w_sJziyDHTSIb(1i;)$zKu6QA1S zOJcNtY5s`W>$em@oGoPdRjWb7mY}$HPUM20H5|hFb6-J^wfd0BmvtlPX6X7ZC^l_w zdO(A51g44(nkuvk@+KBKZ2~;rSzBc-Cx9GQWi2Sz#$D_vTv?;>__ra$PRxVfVb z#w#<~k~7#&Gg7aV&3#&UlmG&t+zFC--oCI6z&x`n&M z03Nz^I;AU5#qr65Z7SBI_8EMjjOm0n7S3&iRilS2uLO`uVVdI@L5+o@)@tV^g+KMTdclD0d z{hlB7Lx+w_(wK(p&XP0a_t+ERcmJgNlv7I?Ckw*^gO%J@x9|El_&0$XUyihCZnuU? z-v5-|QWPy@rgvwfSfdyqrUiR;psDpl91#hx0%uHXJ!_a zV+3gRWcEoIhe}ru*R!ogOa+f(f}N|XDP0Hhl>R_z6_LI2ePT??sR4$d%!67b);F#U zTadqva6?2?&k}oXGi`IWk&v834kTR<^HkDfdVhd78dFmq^s2CL*(b+>@;JjZd{W%P z)2^tozelQ;G2rUXjpEC!I&rIP4QLQJacw7H9=~t6U$?bi2F%;GI6z4ejw@;1}+CzHM|y$anBf6K-{fohoe~-u~)-Yj*PZ zefx=EpZcWRBx^Su>rJsZP*>tK~2~SBFmSu5` z+6)qms~&)2?Map;djt!4VB3EwB6D#Wf8;WH(9qfzmM#N!eRc8o@z2E3iBIR-`(`es zAqVi*j^pc|YzvX$0@6_uLk0|?(9@{RqGPYRRbZniYtWcyZ3Y`=$EVu0R9>5vVUxIM zN3`bNTjCIw=iFhFm@aQ0(#kojE?N=#rv3qgb@1Sj!3H=)rUjWsr2Y%=R|XrN&u1My zWI5ZA@J8&1GPzHt=t2_;Iz=4*gD3P>cP>>~7Jt}@e%>~t{DOa$=4(O`Y6 z4*7~r-O>Q>kdOD52TV|{fyu(RAgr%;rgY0AIE*Lnb@cP{BlH8#Wb+wO<75FAcfZKBA_H~j&+Ciw3H+3u zP>LcLG3^Y8J4!<1yx)li`DQc5Fi*C3bD6sw0=32_`+D%8rZ!suoJ&+h;$zZy0P(9IGVSXApDyANo^=EffnXzTB6kqx* z0`kLd--lx097A~5EstLtj15TpBw3HReO~Wh5m!(n+pOH!+O#)k@!R-oWIi) z(>;Q((UEQOZeMtgTMC8)>G|YNYpW#|TS{j6Aj7g9&e8lW`z7d=-`(rd^+4y1n{miJ zBro>1y5DX;Pp{s-d2)X@4}7aBvNTMRAXj{v`dg^F)eg?@k%!;; z>WJ6$@B8D&KpmxKh7A(^mlBXa4tq`1pKi&U5V22aC2;;CFi$?^I7gyKRKHd>zpbyY zcf(C7$ZU4E(kt~~h%IkEXggTm*Mi+ey*@I)^)y;H5Kl3fJ5<~uEVX50OpMWD*oR4I z=gtdDcQa)79 zb&wonX0Kg0M*9<<^stv3MT->`RlM8taJ>=VrFD|pXjPI9eKF=b(gXiDZBlV$7 z2wiP$9;0PBI%L>SMj-jLTN^WPN^8-EU%jT^L+s^afZyKOwmBSOVa`&y(H>F?`<0>%aoFhJ0 zk^_$2SUf8CytWyPQm7S3JYg=OF`i*a-GniGKCkHi;k~vCdh5#Za1B4ex&_JY11*ia zF`VAA&V@o)!jaVOq07m)(BT1OU9L|7tf8V@s~cm|Co3a zI~f?W3S_p^(2G@al$b=OmM`2Mxg^I{R7@glz)rlVV3vH%mSZ%%(*JH(5#}S@5_%1OSxNc%uM8rJ)6|a+-j;U z_z|#TUaXZLcqd4KVE{K+F<8L zRE|P1y8bWP5WJt55oG$fbMxlof6MvFhpr;T!yQU2`G^xMYZoR~LT8jlM}_@TefcbU z@l%DyXl{)=&Rtzoc{;fCw10wU8&%!m#7!gjtUA4ow#UROE~{Ix@v}dtLhRb7`goW~ z9P`R@cA?%|yvEtw;wKcn(VD`b6)yZv^={OBK#+UFc$mM6#PFtb#Vc0 z#Y1QOS@L4bbcv^cTntUk9M1K~F8;{E=I77aoVce3RI{Erar%!DNM$6XTIx`@Q$N!@ z#K;tN6{9-6@xQPwtt@@BytOW8;emuwVm9PqpG7h z3Aal22B4WysowUY`ABl3kxU&130JRL{F$e{mCKlDnZF(%@l7pLHI*O^eW1nZS}Q05XmG=Oa=?k#1kcz zA>Lo684LSMI_O~{9Ocle>{AFgowB@N%HK3|=ha&RfCRpNgv zkVD94mJRZ!F_%Mg40mA1cv8B_fnY!hSj46e3NZ&n0V86OyADl+t1`NWt)P#CkHNH3 z1k9BG-urO))%;=0?JoE_o36+Jt<>n8t0|F7?l z+P;-#>aZtK7b`Cz$=ljHqJgYc#{c%btl2jY&n5RC2k1hy95v#pOi=t#qF6&L1bTWX za^{A>$DhU#EZAgj!DqF5KG}#?qa^hcDz?0cXRcFpiutTrvh5b+f2e$3^V87{{Yl42 z@TGF-KLM^?x!GC~M3PENh#!ls+NOA9wxCSk9=4zPq&y?Lt#ti zP?N--ugre*xWxJWKl^yj{}=nx&HKC@Y6ro`4HmHVctt+qyyw?KQ?Ok7=&P-z+&Qoh ztyU>NRR6OQI1@awgw}L!6#n_6^-`L6OeOb`5J#^a4{eg0j0|i?n*Fd(`mr;WZ3rcm zUJ`W7DW>H^%z~>~Qj&OVE=S)n;~1_i5fgW*{CQIOP|lxy!9^Uun* z+|9|QPj!52W5NeVAW5%M-UREK;CKer)C{7x%(t3iIj4kJ0jVA-VUhm7E&;6K-8Nm* znn>C`nfmBP49gw<9qYZuWivxPsctVf^L&--1^c^UFQ<~Js2;1*k3C6tVx9(z2>E00 zcXN2}{Fg9I-MQGP+HaE41*abx)2Ex-WgjEn1Qj2v3Xo{sZkOxv8r`a`TzcU0&#j7S zM1nLzj9fz;W8HLF(y&3&jseCO!W&%fLk$F*JIn`sC_{z>Q`Mk}#Ra3z#tBs~E*fQt z99!r#3@%A9T!}sO#6fW79Gy3&W_y^a11nWd$p4MB1~c{`C-)tP31`;#IdAiWsJ8Fk zRzxOF^v#+i0X>&Oixh>r$X>;o`}N<=&*u!=bqJVe66nUw5VxfV2 z9KX$DVf*Q)l~UVzp-j|fpX4t=ACjD6qO^gG%S5C9e*Dl{s2GouGJt6;B){i_z;`EA zmrn4hktQd-waoILMC0D*j5|ZLL;?cvM0y%g@OA_=gQhmy|53@kzXI@wDu!@n2H2Xl zq%!7Ko}m2Gjbwe(LZ@9R^NFE4OsN32Ub)f-FVV(G*>1K!nmc<#x+u>7Mq9AZfN_8sR9 z*8=5?>Q7QtP*r78tj5Kil;lD5-vHHND1R;8UhZ#w zKD(*+sBMZ&Hp%7jg}Q&zu1g7&@uDcwWiDlCxxXeARcW!9<}k+chIVv&s{az)_+tzE z5BT7{|LG&#Exxg3V*eY_oTb8!m`IV5%aP<=roX6EgN}-I*6$~Qz+9)X+?o2jQ`?k@ zjGxh~GJj7~SamMgazC6IjvC@FsGp3(JqpxopWoeO!#Xq5E@f@-C;kY{3kr@T%(GZw?IrxdUu-gyx1OM({0l<%mL{<%ybiX~rnosQXN0`_w8TzqrhBV8l z0M=~3*s*U|0d7xoc;KeqE&TK3A0NAxSIP(E_E7{(Az?NY%u)V+4DL=%-K%tD`q7RW z5PY0-@q>B@uedDaAK5aQmeB1RJJ@!(IiB!`fpdqAh0ip=aIn^n7;=I^(2l@&apP$1 zdR{i8mQ7LX8L&TFHcQn&yik_P?4m_Ia^<~$?HsSk(07Wb^A8{ILTst>M z6(3faT7P^kG13qzk6%x;961(Bp8M*@8{zc~Q*vI>^QQB!j0Dbnn34%5HAe5J8Lv0+ z-8KT>8t(|{RN4cj;;(pJ+yq*2${pw={~?P8m#Xq1(G4U=S><#XGb67J-S8*8q0gO* zn2Eh<>H`+?aBu}R=IsNaIzeodgbQ`1J8P!_;=4&x z)RTt}?x5W@YFs;-f>V(A`FtKzO_oXe0WZ$KDNJ%3^iS~6y1-IN{64)RnZYs*KXHvCfZ+=sE6ksdp9#swDM0v4!|uWSRoTUv#ifneMOkD;=@EhH z`w;spQm(x4q#lCAl&q=j!jMCFuQ+uEgX=*ZT&wPVTB?!kCi-f|kQY~(yrd1_!NO|R z#O%A}F}rLzf$3nLjZvtF4TW;F-i-$+2F z0|uqYOyw1=)$i#1$>*}{ z^bHAF)B`rJfaQX% zrtF#$fX9^iGihtasB)04^gXzC2*OD|DNL#~`oIURMq8C~TSdGG{W#)ePT*HZJC1W~ zh~C9Y_f(UOmSVLf(qdf+Q}h@soE)Ohxz_)2TE&#LL7#@kHIe|9Ru0;M9L&|`d-b2d z`K{1!Z@xOgyubTt;R`vNI9jJ$VUIN-i_|DzS@OUjEh*-78Jnien*F-x5HmM-P&cPf zv<8iiY-_{jE%?cjvt)ni8@b;)bFXA+s&-ctS**&D%0!Rwc4%XNF#UKE)Y97RxJ}v1 zxLnPXLxzm7-B&ZecT$$NrVX=H8--dJYpOs45*X3%Dl+jFCMtRb&K0ISJxv=WDpf1K{{y1V}7F=`M>O&G_ z>IIC1rumR^wwCQ$v3{hc0dt}|^^QVU9iVCG&6Q3bmO}*MytP~91%-?$BQh**GkIDl z`H5UbdmNcdkhXxZgL@Vu1}Ms=&m^dYJR*C6F7@*Fr(E+d!|J#_m^ghqs-E*cz(#kyWu11wyh*PTSlS%xvZ{MT0P zuxEy0I;I%Q*SK&UCo;CLZ5mB%nJ7$XI#?)P+?NZ`#UTdB`SrO5EOSgG<1q;)hZAr4 zN^6cVxW30>?$3}f-4Gi&Bs4)3jQT7cK#8{*q9?$`JcG$4CO(cNV9icU7PUZ634|Wd zf6CcM@)qohl5rZGtUV|f%aEY92zR(p8J1w8q~7N_Z9SowX&?N{ATU>Sh)!A*BIDO5 zswGQn5uOX;+xs&An=&q3Q{5agbw1*)Noy58^b%#RvU^{T$9$+ejbxBvXU~M+oW>bq z^fkoXl?^Oo(}aVJj6!9_-=q~k1WSaoEhA<>pj`cX&;#N&AEjQyU`)gTLn;v9 zAEGaP9CgxLnf&k9PB$xPluQ7f8AqVvQ~BNJ%z!|tJ~3Z1B_I9@*agC3Eh$tjb{Oc= zX9OV16;lb~8+z134)kH;ko^+MALxA7rIIvuF&`U-e>@y=#5bK? zv#Ng{y)~%e$qdYE5gdpK=A68^F~Dvo$RVOOP}Ri5sAdpe+>U)mFMKNUlqHl~gUlSG z>5$iNIhz!(N1^Dd6T_i06${TrswS4*WKTdWO1BTfg;M-$*2WyNF~ZX?h`?l|t;w^J zWG*4{p9+GpeS)hxwEkgEKq9AakH_ruteI0yPsW0{d69|6>!%7dcnwS6#7=LS-Jnp6 z6QZmd>wspFA}Z`RGZ>+;Opp28f!<)l!8-^d0rFQDPcdT`<&&6dGD`qSU<Gvp@JQSkF0S_gv@JeSR7Xfd?T6IsqwAb{>uYvjqT4Wd@tg%W&zQJ)B8sd0 z$qeDv3gTncYKD-@Sosv&d3ZQ^=4M(gUBOfOtxpP-AO0NELG%aHDm<=#e>&u-m! zOcn^Q{HWJOex-@g12DfY;{+y**DRPj~+&*BjvU|0teP(0oF&3R25T_Usk2cAnYzk zKx%B|9EPk8YqqfJC0AMoU18zm21$dUPw7jF0;6Y@>SC-yI}QaUV?u(l^9w6zLVNW4b;T znaZ z0o+d0O}CbEzkjBF@nEfaM7V5uu;jTq(cdi=Yg%uJ#IM>u*b!u^C=@#XD}Zrtw1hUZ zrl{{0e%dZ+%Qf`i0Wrhy{bYmi4(y&3gJPIt4wt}iu8!Q0IcdYlFXN- zFK9hyzmA~)G#!a2*4h=j0q@BqRVLyC%n%I@c(C=#P8r4gnCFpunYJ#hCVzeX7m70) zQkPu9-BXuwQkTTSseO~TUsck6Lp7{AsS@E=VsSz%Bu70c`eMheC-L-t#1Fl0!Yw$I zhm_E`HveR+3?ZCE?;=jV4x&+!^myF5*3dS7wq0p!ogozgtih0T5Zlv80Wu8LDEf!xE_gzD{FM*P`HwHibtpjHPridlv7jbUcb2Gk5S3@j5MrpbiN{OG^I`NploPHqVN;MOz zh0QwFd{6~up82e%FRap0#~+=+_%i&?UTFOP265{z&iS{)CWfINL8EUEO@SE!x4JrL zXvNEYnxk^fShKL3@o&_RHt-r(7m>PcNOEe z_wW2EESm^JT4H#f0OWm>{7EY5Mm8F}tp@BuDCugU2%bl>30kY%z!HlKLPiqa^iOGz zV2-|gzj1%hZ#y(GEa+>ep#TdxZ>1k}3zs`0dth+{`*^cEBR%97TXh&urx1IYXeU9h zU@p%ed&!B~Oc&-;_{kBfV(w9!zEVymeDPscZ4#c=?S%^+Tm3Q!cevNbt^4{%4F1Tc zt_iUt8m;&JVSeVLegsq=0bqb5>#ZRSGUkCgsb#9xH;_g!`H*IMchWu>LiB6QiN(Ko zTYalX^b>XIoo3rFHK3{^{x|d%DpC##oy*Wt8nM{>>%|+1A48fiQo(fjj%Ir_S|8lg z)q!@fZJu@#rH)bLQcbOt-5jRL@5Gsvx&tONnA7=#oFJ;YYREa`41A9ZZ^M( zq`Ij`ZUe#Z@}4*WvJSW!=U6u72TG3^;Haqc8{wL}jGysb1~kf}8RpIW3p=Lk>mAHn zmF~3T0U~~B{AYeX)QXnvicTWL(-JH3?#4RkQj zC!Y3Mf?6^>rEiF=eqfOdIb5)K7mIi@6vbLc;OPqy=;Az-XWHAdCd=$oi)u4sw9NtGJj=OYr{@cBgo&8zdE#b)O!)hl)a4d>(~CdhU5?iQ;LoHUfRfqJYd zqKPPCOrI|>q(M+K`<5-!_PdwD&P`}9q*5#-DmEcP`yKq&YdR5y{T-p>(hH29L?CT}Xed~IF zZrc6v^a#h5;58`xS6l)Yxy&nIQA1mVDpA;S81nV5<7Hfatm zj0V&`O_-?z7#_^Y=XJo#l zEo00~s?)3%`hCq0lPQx}RW6P9&|D^p$+m>+u0G3iWl?SC63#pPtgM!h*v{{haLH!9 zCei>(qC#kC?(u63Yoy=z=zSfGk}ed=!mAM)n>G%q5Q;~k0;+tYNA9tOxY#_k0uWnw zwplyo_dJ(64NT58LpXEo7o$Z@Z3Iw5XH7bUVXzh9t9EnMVsfnoeVlGxOj|!<0_7Zzm<+@@j8I z2g+&C_uSh=dFBh>esZa1c5(e(cP_CJVvgf0zAmx#XO0sszK-C2?d-6sWjR7_Vk>QF zvj)R|vFK734A~ULF?dH~HJ_2Ing>$jy>KNSMKSuEz8@`EvusSCN7nqs|vTF8;V>j zsa(_iIut$NpWT>Qs01=X#MmXXrV)YWzw;D?m+%4AHM}CKgeeAL2-Y z?&oFWHSrS_Qx;Ju7w5vBh*v;gtDtP)9YFouwh+u8Oh5n&sq5M5OisJBWNL_(PL1$ z<&QYL-zm4g>53pXB_d$krK*xpzR5w4C3tvO8;&kKXE%_8px$!tcrgq-&x7dFk0ZMM ztv#qi$iMiC$M71KLNU}$M;zc>9u7ww$`Kt~+g?S!YeP9K$9lk(V^#sdz$~DUDfoCJ zeCUzwYZ`_L{@p_OeOdJM@2)=hLue&#%jaokB7g63S+q2GT6T+ddD&)$QOC|{uyJ_eP186PfXl41A0NYhe?rBS<>4Da=m4*t2>ElesbH}$16?I0-meDEG_&SB1v zf!e&I-dp(*V2L>~D)8A&RG8NzM<@iD_SyVpJ|v{$mtW@WUf|<5Uz=AxegYg~Qt~3v z%vQ{l!H>Yir!AS2Z>nymeC1PDtatGfgl{w9WB!eQ3*7C7BYCRFQo@|gJY0rb5%$*S zC1G<-??~sjouK@)g<#)U zm3IC%?Z$M*=1bOOztAzp;Y;M|&x>9?@W<;4PN@KgQ`*13<&|~pJ3H3-xXV8h-0!7< z1E+48I|%sY>+!yIm$|)Y9{c2KtE%zyG7|NGjs5%u;?PObg&h1EFKQC+yHe45kl zhYsG8y_ecoj^AT$*RR&sx`q@h4l14EL9;2?zII{nw;0C}d{<|9pD+<;pO$8uEPwr> zh7C@y33y%Tdn)j`dA3U`-u+zRhsG23HPDmsf+9PA$|_6y%p2}qi_M6W);(~c7y$;< zw9ffuZi@C#8YJgY^=eQ4$1##PjTrAqKTN=qbB#XS{Y$y>JN4_+bJz3jVgR0p=c4BH zDTPWsD`w5NcI5q`Q$Ep;Ep(;ZKcuzXca581vP<#J9><;EW!5#{ghc%ojarySRx1C~-*JMzTL zOqtcb-mW_T9Mm#Y{`u@xiZofiR>*&8`*{D{8qyKb_rb`<35o(E3!emJa~;k&8yQn< zUSYfx;lt+TS4#if7v=HmH|oK|!x?FZ30nY3JGzz=WVk(ldVNy}us6uL^6Gmr{&;>* z8@_^^%^wVD$iCMQbog#!0gWA{3!LUY{4oQ4)()R8PiD0BoBlW#weJpZVvnGcX17yW z3YiDeaYm#_J$7eEQ_4$o&KahKT?9Yr$}pmPXBX2%`X|8$;YyUg5b^1v6k+&c ziM|@<{|G#B?%fitYJiqwdJmx zijahQPNWT8t;l)2Itwi9#d1P0go*|bN4m@}$UZ19-@;#bfK3nDG|5!lWLts(rMcV$AYQ=ig3Oy7z0C4a) zfv-dLnSeyK*$xr~Z?V#jaw@{o-}$S&4&n)wPu6Xo>!JJ^eYvc})($K8 z9!+fD_HHla+B#Mm=9$txg(3b8aAt{JEaWBjg@!PS(PDZEZe<`f2n;b+;zC53iE*VH z#Ut{NT(-<2;l%PmMunXT#-Q+aO#}v@^X;M451Di-5se99Sth!(;R^v~HC)Ly-JviW zq3rP`ifBy%LGo@A>=Nw!{Lt9U?ga`shwS7&7-;>b>CZ{|Pvs<5q1(MC{6!qCf+l!m zS9t#P_mHg-M(%I)UamzcM^8yerkOF)$H;7yv*4^F8P`!VLfItC!x0d+UldX=eNeLE zCZUkLtV@L8B+fs8@Pj4|BWQw0%16}6KQm^Y!J!df6IMI>QsObTdMB1LyS(-uEJXG0 ze>QRC>PyO&-fjAg!$;+X71 zjw+yOkc~nqksqDe&psl?wJg3wmnpf0SrBF+w9h{7V=7ZX-FwN5i>qtJs3V~^a#T*O zBO$fxn}k`-K}&vwd(`R;V3*#ew?gKwQBp$))PyqEB$O5Jt<46B-uE5S=)&y~5Ym(( z9_P*KnT#r!(-125!QLBMF7-wzcQV6Y@}u=S!zQG<(-5L%u%gMcm(WMi`_l&*R*Q?Z!a$!_=9LXyKPY{Mky4d;yW`B7T?%qG9MO()r#Y+rtid zm_9ohWI`a~#z&9@R3ca!Oo=BULQ44Pr9!jI4Vc9j!ivm9MPQ|v>9v%MKNdl=N46G` zpC^1yfR#)+b(K})*$2gwsQE{J4{#BL-*L*N1C)-Gqn!9t@!#8m;i=<5ut!+Al|Yzv z3NZt;I@M%sigKdIKf#!$R@s)e#{&>#JdzS*_zh;cr0n1^AWazPd%F=-2)4)9D_Y@Y zMB|Xp4EbsA*rRRm<3@toDjkC1St)nC3{&bX|^mKFvyaI7ElBie4}D$Q545{tPYU1 z-X^gQCV5ozo@Emr?uw+ArANxcL1H6{>$PbZJyJ_xhPq4_OHJai;QfMzCQQ07OqdJ2 zD+tZoX!O0oCW0= zT}cm%F*2jkHpr2N7zWMjaQJaIxg%5NRzflvV<91uTPnSG!*(X&vZG_IP}#O85*c3b znR-_e4Y)#k458NJKH_!=d7sb*SO*Peae(RJlqk`06~IWBeIRLVC>>x;(trRi6pKoN z-n})*Wd%_Y++YnIam*gepd4UR21;$m%G6131EdA(|MQeM)Ra&N<5VrBaX*D$Pdl-QC7cPQ}>^&7bHS=z)rHB_NMpXusJBb9wB80{t=}`SbCQ+B~UYGzi z{i%OO9SKhw0=7@49XAXajVLBv=9WP0|3X~eM0SNN!J`N-JO&aE|Fj!Q-G|LS14KF%iEgOs*6C9S(E}11h8CxU=Mga;-6k7%fL=7S$ardm_7zWfs!zFnSM6&NuDN*7hoRUeAA7}asi8VLqu#kbJl_ah>HlQ4; zX2d(JsJ3EUf_2K=j<}Bbg+MWhac5UnnKPuC;jE2G$(0vZ?MUAg|%Rw>=3K9qK+#ly2NsVLRj? z`a$h!JSi3cDLaEOt`3|b{UNuC)L}#-&+_UXcRRJyL9(n9biM4UP(Pjl2;FdHhqZvZbZ!S7ZP=ClK4It zT5a7W5f9jl3Y)SOj|ba9+9!hJWXS0;2U=t3AUIez z1S{$V04=4m%W+&`M+uLNFVW)B2b$zWVu;L+4e7=$_o0P3h=7aXB9w-?verd2=MTB+ zN2E^`FxjbHhc~k^OdMfWQ#2(9vGA!)*oU@qqb3D#J(Ta!CUU9j4iTbx=8Av78FTJX z$|q*SQ?5wz0YE6wl-H0_a2#O~DZW!mIT@frMj)^-R8Trg zYUxyl<2DjuBJqSYu1r1J2D}o6W&3hV85&M>U+}B{c%mp1#T(z9nJ@xCOLLWfayvUA z32llK=8Lx>X@?&y9k&Q0>OqU(LPSVChmfPzF-i_)1>32XARSBq-;dc!Z$cV;nEiS7 zyUR2=|JziBO7Vj_))q9qGMpV;JY(yYoUtU2X2J94&?8fPF&8XJ=}2z5-pz1VC}f0( z)Do;#$gH{~3aFVRiDZyO-CA;Y#>T}8P4baYxxsvqhuLoWrL=D zSX%t;rBL%jiP3yo6GSouPa;XSID-=$Hl!{cpe+5NMx;Eajh58;(7sW2kbKOpnDV!w zQSnSH#LoOdXs%Q%`9$$$rB|TEwkN%a8xvKyE0UgG;t$eTX>Z=h0 z$MEg=PiA;;GTRvvZ0h=)MTFz5QvXnGBp_!kCBT+fsZ|4f4o>8N!vVvF=!`|E(L;YQ zqvMo)8QxU#nE>Dg(MQ55iUsqde3^(u3UlR_hs6j)qz6z^Bbp}lm%$mzIbIy70N}U} zMHItQ+zvU5g(nJu%I1g$94-ks#{ihSC^DpS;Mz1c4PlrGp+ylxF_TV$K^d~>sHV{N z;3($=oT?dyaB`o^_6;1ds5c~(RwOn6W|@z24X$M~LXFwAEjpJILsCJZ5Xy`|#W~Z_r$K-b>u=ig1s)9@* zBvCR+fZ5&P>e-;FAw?nx3)=uw&Sb@88^xP_wA?9A%qD{!B?Z1VH3QF1ZXB z5e=%X**@_p*c;%-BnLW4Iq%xA0k-=zFcCmI5{E%R^_O%)w|8b}bfH5!6&fmb38PFf zItp+jmdf)HRyU_kParrDc^Y>Glbx(O2cDf+3Sz#$>kIWyGRrm)GWEmTGxT#?`_ ztuPdYXb3@m^`zNlQ#ZE6t`Y1oMvhR1I1)1d^2Sdda1b7I(q*wz?*ceq5Ae`uM70cy z4Fi&fg5~Fro88jrtA|pElbv0cTt-bY(SG)UAtc^7%nKvs5Cssigh>W17o1z9W@!fQffHnE&qf%5W2H6& zx-i5;NZ}%vJyHl9NSLJ_cpFr$c#D9iLQR4RpM*&FAQip}5`DoU8bbxD$tpG1Eg-Cb z{*Tj)IBMkfj!S)`5%aAfRShlxAukG=t&54KvUxg3*|VU@|i- zShMpx@ng`5`B^L=##X+&M`0|$*)s^e03&5dlsN?KH&4L%ArAP+B6d^c@*4Cf<_C=U zU^Zg#99ifv81O`_meD=rep&(+H3SC;LP7x@T=|X^fD#x85~v$wZ3;JrQBv(f6KcK- zVr)O-YUW+JX~<;{0HZ932~pmEtmX*?1d8e~j}!Tsz*$<7kqrC*LDkMl=m21bk}&YO z!_+*9!1Yb4P7sB$ntSFjvO$ERwyg3RGWP{jBskX--9rW6RtMKW)N@U zdH^)7Fin+D<;H3+6xi4(P!przstJLtO$ts1&A>-inFGJranx1~mv}HLWY|sMDxPLu z0*(|XG2=Y&saoC%p}#kE#WLNNugoGH|J&n?o=g zr%~#u*&(D$MZriz7l?QOk&in59YHAI2hC1E&X9+WA>j4F9|r$SZI8%eSH;8}hT3P3$7$O%=CBAjD#| zubitn14hFjO(|1r>&SMNW#FceP?;OwBOH1HQ~=bA-bF88SK1r8F- z^VEb=49zNqfR2a|oLh*%dWtzz#2l=j5Cj`dr4R#)Lb$SOr44c z$C_no+d|-HYI-~NAZ&!7)u%$STl;8ir|Tp#i~yw3__)df!!quSlrlm9Prz4 zBRW@SrND({V6N3}S_{LkjdHsvJ;{fhX%g z)e^O_8Qcck7$6+A6c%J@0s~YUkpvo;8kwNE(2OBBe+Z9<83QvLDWeE+6!=uoHfT%( z0^1@0_Shc)9ELi7a82Uq7~Tc8z6T;=!8) zH#&*Pp=z(TpVE|t4pj@M;0ZzG4%S23a_K!NK4xRS~13Zf|19ra_=WGqC;;QB!Cg>NYgygl-JsMVR_X%_z-WmT}=H z35x@DqgP5C2sL4JCw_tj%N($TYLxOlbt*mOnI9+0Wcv`c;ARhmaJgz<)KNR$9r!`i zMx!u>kqLc#PzYrjSrWuxdNOq{Ped7q!lx`u-2e-wE@FUjGE!47P@_(kIX*a3;GIE7 zz=F|Oonc85LLhbm4^JH+$zX^Iv&2baBp8Q2nSr}P{WJ)?G;`G!0*D*q1cm~fam*aX zoAMTMsLp-E&`z=tI$0=u*FlZ}5dC_-2gaL0un0qmIuxDwP!%NOS(G6)bKMY{m0IP1 z;v!)XgLQ+}7P1s#{xn8{fo}}KWui`KsOT<(HX)*brBEelh#NOEbzcntvV=MjhyD;| z;Qgf#5C_169G$?+V~(S4{!m9K2#I{P>nZTn6(UZAoZHTP2O=+Ue83)peQYj-v~{0PQ!0(ID|tmmx=VwHy=s zVW!rC)pb-bq&UI}Aht*{B9L?2nOY`=IG90f2T_zG)Wro58wVUBJam~w5obP&V=zKO zogfqr*bBjlx`2Wpz71`uh2^-moqPDmVcwOs(`(`EvCc9=MUI;{f^GK?xI7;H5`3eG!n;V%=J+Cl}6 zlsfbjIWUL>vBMys0*l6xv+kL?sv}TysxY2AfX1O)D(g=YCk`g(&u;e!pKnj@Tp_IP z!ES!4o60r!7+>C9(Eff&+uYGEUp1dX(p{Te6o2l1-RSS|cTV>0)#T^v*Y8!j-N@cv zo!(r2{rC3qtM;qNCZh8M({Wxo})BCqtfYhNKI?$`cP`}nCNKD=?e{lKNMWX>OxJ!_{v zqItP&TE4p6sXf%^a;rAT81B_RsPDQDuBSeJdAhtfH?Ok!vQB$Z^`ZH(X+J}uSNQy) zi6-GwoACWkTws1s4_+0;sPKQh{zaTBsKj4H{i&=zD1qCj`b-6NP0)fA#nVmXR$|5J zaQvccPtA+lefq^Sv_Mr~b@}?2)8XxxrZNAs-a#S$wIT=6DmJ)1|M~%dYn64b&bR(* zz)kl0F@R3ROw`!h2JOD$Odoq-WhK6A@2c4$vk=@ixFYg2YamoHGgzJF#x^4)tR&3hfuckgYeP+r1CLQ6O8F|^4rW9~&lPdGXL zw#f=?t!YNb%X2G2Z127a>1nNc`Xwc*zu$z6%+r0pNhwkObXQ2vET_HebIKaaTI(5Y zx9y28{U|x;gtV++5S&;nIh+ zBq>N}DdD*d;pJ1CIMVJ{Hec)C*^uY8z&2hnt*{(^p~1(v-P(sH-2teAO@ zwaiLH&k8SWIA^$4M!(9|sj39aAuU5Y8aF{L|LwJ%7IH_>?#5Zlz~8*HA}>z`w%i*T zJ+XRYMJqp6*r=?L(bK9oHnejJ+FZ=mg)GsgOBzg6W3c-1RQW+8r`L9L`r-0GbbIXVC}MW2?Tr~E{YvEqOe|c-8!ZWFmb&|PH)l-DXt<6G;_IsM%7#+ zpkY*xr`RNdNDesUB>WD8zZ()vYxkMFM4Pok?5d%Dz0hi<5V=OE zT^+Pp7i1?{mxt>XX9VG$+<^J+?YIhDWaPB3HypPY_+G51!2L@*e(=8qIiUGlF#~`5 za*BUj2}EFgPCsC5o=#CgR8!QL8Ks-ZfbjBSf)lgmtXXZIUS~$HFdNpE?N*g}E(uj9 z5sPj0|{_jXbRh!yFnhA{GP>aY(+iF?VR8u#z*6(Rfm27D29N@0#RH}MgbZi#Su zVN9t;c0!{`Bij|Yo_WIRay=0B+V~owh1`&5lJ8ZUo2N|&U`(q{vMKGJbB@tPB_A-W zhm$-9#1P&lAMzZg25^xVm{!*&@A0$`ZFn|E=oLt^Mn=onyqTaaW*NbIu|9cWZvt_X zeY$BriwQnaW%d+R)+ea4N%*{Y#xOwgAqhO3;CaRr)$_2Vo+k?mv14MmOJZ_o1SB8@ z-|+k(F*FyMA>HN8B<;b)5FXqi)xn)0`GV*@Uj|*~%S1oPh^)VBj<{}$va+{FJ-Hrj z@2!Oj)j`fs;3B1?zr3>{FA*!nR|zd?d9*2T$;e(n6D#YD3iVE`-ZWe3P$AYNl1{;~ z^b?9_wffIqy*7PciT0@#m51f>^kkkDTY+UO^lTQzm}AQ^*B4}NRgyWnD04$u=5~dd zqf0Y46lZQ(o)x-4D?^D^c12p@%d|2TYGqlf6~0(2Q@K`l1&c0ML$TVGD!x!nWolWZ z_!2c0s7-ky_h0LGUR&+C&h{l@-L6SJped);>Y5}QWLVY}(ot?~LbU9JI5egb_nA&p zN5$GX1k4KLRiejreQrhg5zqM1MEP-&vm*9!u1ehJfk#seg4Brlo^$#EPxI7@3c1PS zPD)Q}v(aNmIDsd8(`>rQFmCKnw5P@}!MHBad%&5i?GHI#1bV!e#}jNtfvDdmb)QE) zn;?<6dlSy-`$W>>)`VQ65uFkovRPqE1p#hu%_i3}JoFUC24d$ym#fV4dpcrp&Z7g@cLyBy(f-9&n1w--~K zQX<%-9Pmgv&44t?HxUh+jWIsonZ!}5b;mC)4ql>; zTxt$mP9L*`6ohlka=}$%$kjfdB+MDvqxST6WQL@dH+Hl|S)*w|#y|^i>}V0Q12+)^ z>C`*{{>-fDw5;Vh*#oH$CwqsZP2QtwA5QUnMv#0SFnpeb&ku+d;&b{Tje0skg*oD@ zIovyatoI<~Zd>IX!O{v0nA*ovY&E!#Y_G08WrXbng0?3);P1V+QKa2u2IFd0A{FUK zY8C0cw^3v(>J4Fa{wy{Jall+cX^__{HArQvT2r9&HS3BS73(Il)$1t8hxuxCq>Wm2 z(%2|9Yj@62RI85tj+{1qCFDazG?3S7HArQnn!MKcK0hFKRcWQQYW33CD%ECTwpLwn zqgvfWR(jQ>Qkgl{rwZGs?qu~EbyL`>Qyyn)a}zZvb4xIx3u4#!aLf4QesM^E8{*sa zLmG=A9(%!s;c5%JODzuSUhm@`VGGUzeYW%CakhdKo4VW-h?2aC^cZh1COE~Gs3Fdt z@hs5iJ3pRc`-I(oeiIomezMwj4#5O+>h3DhXS_TgCoBuZCU=#%N24B1Gk`_>AW$@o zgodkmK0$cGEPpkr`$VV7rLQAme&jfq9yw0Uj^M9yHQnQCd5P<>{Ddg9CUu`kJ)0oG zoYT=9(wRP?gFZ=NI7eZ3isF)rZ;;_{U(mpQc|ER0MR-mM`(Aub-{U+#ouVpudH9Zs zswhwUeU9L~7aWk^zqZnXwO8=xnA?ShqIyj_39NM4o(}5s3rICuDZK{mJ1g=J;tQ4; zMD(=q!isZGj(f~(^u3s#`~1?1zj0EYNtHuIH9Jo6jX(pT*GrSoO2>I>jcI})SO4Ti zjb%TJ>FLiet@x!t+`!S1(KD+zcC@HVlDa|$;^}LW&ME8a)SPDLGn->yUawC-m6d9z zB1i3yxfjzba(+3%kH7!Tc=Ma>oo|&m5cTz&3PE|67%-lnPZ1sk;s)$C`G872oZ`XV zuBN}h&g$yAD&Ov~J0m(WeahRL30l-$dAeKjEbhm{-HfNZ6VKu{JV_8ys!H6WP>-e= zHrwqfL(w!68m{KK1)(IFs6`>mO5%7sf;#%n>uD{1M^XQ-;rrc!q&f1YzR89d$NGXg z>dxy4ErdEMKkFF&P_(bfAc4Iu$T^Xr=!h;hb0jj~9!kWj);WEjwS78CRbY6EAxSaC zfFT&-Ps8sp_`4y&#F|S|LKtq%CuXLVZY6r>u!u0#O_IbAbrV z?xY5kop)2@IIFF-)Hc@nMC*z$ME=eKeXjH4DK63W<7naYQ|NT?7~}dy}pb_Zi^P1Ve?Hqij<5IjYGS zN&Hp9rb~n^uMe)P96^;2Nb*VeJqCY2B-l?foG=WhA;W2$L4D}Ex#e|(I&gO{)?>YY zImKTgNUH(U#FP{2CTGnZ&eOLyKmTUGbrjO}@qe4xF8X{R2z*G~CxA!e44-xa;Q}|Q zdmPQPagyIfg}4r~N!=r9o>`FKR?u%EI=b;S^rAqmqIQzsxH=EVc<{Y+AauuR*+Vft&UopVX8A=`{K1a ziPGldhJ3p*PlwpKPp?JlB>$m298N{fNO#|jx(P@F)LanMApDiwK3$=|0_1wn$6xGE{K0f<`Dpq(svqtyEgzBdkUZXy$Ft$_kv#s8$DiW)u_}+{ zskSFN3*0fCpV?7iO7AM7qw_5D=Hm%{E#YKg{&b*%^Ss#F6JAB`2(M2k2>!Qy?QYQx#;zjwT$iU)go1qd@~gmf>Z`~-;pOQRVe8=8vo-1OshURT_@U-R;eEPkU4H(_VZ>j{37Y7EQaE z{Fd{S=l~e?>7=n!jqUP1uS9`X=_0681;fS6(?UAx?k&yZ2wA>Y7oVvJRX+=Z z5!b15pTbh9%CNp~Bc@fN>Mr$y+uf zw^T>%uQS}+W+|yvy*h)}6S};`g;8$@A=F!o@aexV2QFt!Ee&Pw=!dUub-~7TJwd$| zR#D!sZ{Fde<+H#Y<@wnZ)i1vwXny(Sp5m8Zrr=SHXV7*mnvO~m;V7a3M>IuJzWoWo z0jcjP;K2mLJJb~CyZ#jCyXh&;cjDto^>|Elb(6VgyFQ-c%u%_@`?<=ebCnk-S>~MV z- zGWF1cO(N%=Z!!Z$X*2KKB-EcnnUEOrb&jK$WNU$#d{C+HS>VA018xp|Fn?|_cmDoz zX}ds-o*kikzT5K&+J7A-IUM&07xE%;M|gQS!Pb0vkd@6eJ7Qh&PVA2N?u`xkc3)Oy zZ9eU6Ejv`>?lnRe^Fe2OVWT#^6VGFaJ29_VcEVUBJAo*UotRk^H$klih$0yIB!Z3D zkiP$IgfWy^g>Nc(`(mG0=EedaFYW7xoaF^g>`&9wTCqEB_-#$OoyocFlWwuB8*)5W z<>$HNXYqj!=jS+8)E=%!_32;?mFm2-*KB6Z^eYBYFFWP`AijoV{;J<9FId~_Qley_>xIZhSZ zaA;I~sLE)7q{EU@$nR)gsMoA2)%P}vx+;%f=BZ|c>#1r?hgF@aZ^sCzgG0V26%5cD z_2L2L^?*9{^(oOASg6n^OZ%Z>-jb7)IVm>9e*b6*53qFF4iaHC-sQzsJgVvqM5p8J zEn2w~71hJaT|uZ~1UfzCdU-l=FJ=0=96>1kSa5@=%sF zG?0h8$vu!x#rqUkMe8a*?nV8mnq3&3lHK>#DjsI~&4<3re870i$GML3ao{5#QSR|v z-T^I|HJVEs4R|AKVmULrvY%av$?gO3hc5QO#vL-Bnd!Wr=^T^k`H_w;%E3lB(snV( zC_1D!E(_@~>=ua3vI7N1mWuQ!^Cv~5e)&jFEgxx+td@54a`)WkMV_Jb+&P%R^hnn7sV#AgDQ3y>L6-t>pd=f8}J~9Omwb?$tQ}19SWCeD1ROyDQiXhz7xP z*w6^70a>SXyMad!(7rn#+-&~72$n}3Cp>Kv|MMs+a>sN#!mgFOo^L;kUw#k8M-R~d zg_9q}z9yp_RtVa|_m`J=tun%D?dRIC5GK9aw|K=2_axu|OiFbxO|VBiJfzRTq|9N4 zCjWj+m46#``S%QEv?DG`-l)!P^tHtMS-*u3A+6Qr=g#&$ORTG{I3EzXTwW*Vt7nAN z>TFCsJ0pg1WEReZ$|}ihJxx2X=K)PR_+54TkmK?vnd3UVJ<9sF0 z09xJ7?~p3J``eH6)d69xD)1bs#(7$v3#PpZWCn`{w3_zPAXQaS%%hcm!_X>td#%xp zm4`!FY%D$PVe*#ohvxCP3x54`oZ07!M^WF9L1+@K|ga+mOnTba1 z*XHkwNPXFnrz5|;Y(=lz_vB4nU5upeNYaX5US7n7c>OQ+`&CENJy9R~(dzoA|F~!? z$+MbmSLZF&>AuT;;H&IOlCAjjFjWiFK7WGePZ#NT&7r| zSNrqzoA%43t(>}Azjtlv-HU6VfgkPVOzmsKT~C9TQ&&B;0++dmjlQKC6Wvc$+?7}( zqP|7)xwo(U{>^35;+*c>JlZsS^81&KI*)uJ{S!Uf1eY0%-@m;6>H6@DRh$m@OXZ!% zmohG%p-m~Gx_tf1>G1YT({le=@4ycIRUJ@k?}=(9Kb~^-VbgrDbO_sd*@FMKdhq&P zllpI`oQqec&NKuwbhwe;rE8KyTuqbJx6zYS{F=MUG_TJvvg!Wk>&cpAn|_P!b)bHm z{ZsS!sVgF#`{|Ehh F5~78fwz&WR literal 25204 zcmZs>bx@qavo4G~i))a@ZE<&Z*Wkh3-2;oeyE_C49-PJ9C1`?saEHLn@1FBjeYei} z>#gaYx4LV3y61U%h6#S-QE~Iy(V5P^f@vZq9a=<{kn- zWhLEzL!JLp2>{tx*m+pk|4FoLEP-y8j;6Lw7M5;6XDgtGjjcP-%GSXW`0s1(?C9cV z>F#c60rYgYb+QKfIoO)9p!lqB^%yj7@-(qL)qg3D{>6#G*|^#Lh2$tN%4uV(%iO@p zQIS+wRTCP<1h0; z3XimQuh+q>=x5pMt$P0RkCx}1KO097y6mU>J_(JV`>~(8?d?Vb(ea%oqvSU5nKRPfczrBUrQZWI}zWqfSXDoWT!Bq4L_f+&; z+cj(^QBwikWQOT+Nl`rDTh$dEqT0YT|} zOWs}FJWwW(G`TG_;jtUEvWV`)KNqTp zn6X5+c6-Hs`wSzJ^pQC+R4ejO8rI>WFn^I~ zfY3WQ$6Nzc1TZ4TI1#i!sEjcS4FwbjAl#;cY2ffeQpG(ebtaVJYz_CHavgKgFCwj3 zt-us{qKwwyQ)k*($`IJv9vr(2Ne-MSdr1nMfOE6{MR{m&3iyp3CpKJCv;q6U=|1N4F|&4f4tLSZ~n0s4FFs}5# zku83?9}uU~Ew>hZ&q2jgQjI#ItP<2Q!|aRb+gJVVkqLYFGGOuTZt*!$-cCvJi*h*Q zEWC4XFWK%VHB;HC$LZQCnb*GgL*11{ESdhW`^T4rewC+LRn#&%lgk5Gyalpr(5jBh-ren0l<9kgZ} z4#$jxC2Oay*J>#@jr~YEwfh=O#*M?IYe$;CA4g7yL}BDP#?@1}4u&XwyhOp&nro1j zbFF>(Q~EsUi_U)k(#k#k&X?1MVG;Ejsjm0lv35B`uDEl$^wx-5{~<>6_Qy{DbC~*x zc$=+~mr2*W+Suwn$yne=pgNgqyzdTZ?5(6s*|G)dH^Cg~Oa6{{T?Xp_opBBZcccBJ zysv2!48fQOj!IlWd=)k)UnAjWFCVcb_QF>B)@q9lnZ2C4bQ8?8RlDi^0UrIt_v+v$ zg?(*#`9Z~}S^AOFa$fFEMNQmA61B`dS9cB+KalF=?X)#!Gykb;Cjn^UzWIK`)_w(Z z(SDb~KTtljp-|VPk?X2?K1OG$oMZ+S_jD0#w0qfSJzNmi#;d}7sn_UmcItt9=GKyM z)rl)RpsVimlSZ#2tF!#tlV9Z8tjnO0MQHDRJKn@#1BXN&I>_FY%(qqa9!2*d5AVM5 z*qq5O^fI1_(R|VQv(`~=H8rY1LkSPQjya8Efcfig4IWbNqU&tP-T?oz*!PI?bUh;` zRDYtQoDM!V()@tQLW(`cP#+vWqB@6$Xj{(&LyOFz6mQJ-esK&?m$ z045!|r+Ys{$Y)lRXZ(W>zyVTQ1R%{#wJY?rQ!11`la7*zX6GVZ^0~)yMbA*rZ@uZe zfoj(Tpd*ZKmgo0G-MLtscQ8%-9eZC}L9Wp?q8%DR`QtLVobC0;SEDZtTf4id&c$-% zYEST!cm(Ewum>+u15BA`*Rk_taB{qjQyy>Nbxk%5+6)-Zm-XE{qvZB8kTPV*%|x(y zJ+~VV+t;_8C{Gvd!h@}QyH?S+>w;#n#8gGSM2(#Q#WXlJe z$-#4!T9$HT;x=}|QnOE?3=Kx+Y|=k$(JEGBGe;Zil~xRP-8HC$rdpZh45x;f`D{FL z#dfI-;IRXrHLxl}i_?SCXl>+}OcP>Suty*|Is=4T0}6!ZtOjIp&h|6PZ;+=vj{ze7 zN=OzGT$L{F-I$_P)*qf~H|zYH9rX9t%PvtLdDH~;4``XqYIw!^L^=(gL37p4)Px`V zd*R*D_VZ3=zrPmZNtv)P>V-A=>J50^!+B>pAg^ammlauj4B$Q62DX76lwp*!olakP zp2=`*b9MrC2~|NcblwNK0mY9gZ9gJiVX$s3jk!45sV}}A|FS!77i^!qGIQNO+6k)} zSw7Jq&fC1eHrv$Z@@OM#(S?68!KINgQZSdZAlfTxzAqZSnXB-&im1pWFuVLd1Y>Ov z9eWURwrPR=DMdNdgtig=KLL<_c32SZ`d8tzoNUn;@Wb8$xW3KKj@n{s1cF3QmEX_sZyL$zSV*+ha59}|-VYk8MtC^oAmY?dMn zvz;l{6V0uO!KX>rsr^IW27%@PV|??h+3^_C04xtjKwH_GE6z{0Q;mcNE5a$gb@!=1 ziHswCt#4V}S+ch;NIwUBWu_Xx*DFvMHz35M^oNRK?==noP2vBry6KMm{l}}9VEQji ztzt&#+9=k7{ciuWwmav{e8fby7>7apwGk2Fx-g! zJAn1&u8!&HIadoah*w74KS0H<0m(%rT2HP8YIX&V;4H*h=?AFmJs&Ouy4m5LUh~0B z$xOI{)O26WuFclb*mzkc!Hff6(`&}?BrPGFeYs8O$5YOxn^*wFJNUYZfjE`PCPi`G zjDh42*B=^qw|SIs!(H{v5zaXcP;I%c#$*#0G5a7-ambGCqG_7}Fr>LPc=!eazKEQu zu-X=;BL#D0oVc>HMAHOm{`p)nrZeI^I;ku_k?wnXAFEIs{w1v=89oV_E?2Ny_CA19 ztP!}~Kw%RC+#$hasgVgblNamdu36nrS%Ct$n-Ie@WrT+VfAifM>BI6F9%40^$*&5} z=Ekq$`I>Wa2PitRm26>`_6|EuC-*=_>c&fEk$gk75Rl}XD#xH-bs*JkM5ujE3w2ja z7?5dmR7^OB=Nr-|k!DB;%V}>Lr(rgKq2EtI_{6l^Fu7<-W6_9TziH4%?B!{Q-^tOj zH43A96Ze{Iz#q69%|1WpU2V`B1VG+ z>j3476+;bzHhI!7wB>JPn8#+!9;JI;+w{h%R0^LwVJ@NnujhUwyB=&m7L0b6p(6cx zn}>-qWI?nP$)Rb!&mwSfj7O1KE0P7w-a{Ghql!q4w&aEB736>9UwO z8+Cs8cj)VWf68QegtZ0Gr9L-SDcZ7ecp|gYh0V>a`OTT3+oY;iKlNa^f?&)Rtgfpj zi)_{@52@I1{~z#ehS@qPY_`p%NhH{!Tm3)4cMC~u=)AT3zv9p7&!TDOwWJ&TAFRp^ z>6rJ57OOBI9I49J-)0HdJPu~NT|_vrLbcWpYI{=LQ%&3 zt4x5gKgD|HOa&kIP$Yzmiu>iR^!KaFHYHFdAMoFVV%X*XB@~Q4t>eOsKHTa=8R5)W!=rT=DK2TFUDX>$8Da?rlX4ryM)Yf~oq94w^ob!@BHV z!4`oa4*9rsrltg#C>-;eYBmwbEnbUSUfDB>e$@98sNZ0QAfEMuAbG*%#u<4w9YlC0 z{O~&By;S@Y?#AfAjmsHUukNuCU zL@YI4zS=UV!(0QVlVE|2rVMQ9=c%)Z=I?c*u`4p##DSh7NhfIaxdCua8nWl`SNL$d z6j$jcM;~k#P8vFyy|O%rT=5H)|b4JZ~ zD#{uo#Fosaz=@MIl@G$vf(Ay$ji)pWYGfk0jb+VGBAl7_EOA@G7a>gGgNhycu_y4y zw+qqNdHjvhf{SUgV{p`a7AFj#+3n$!t(wX?Qib!f&l{)SFgOK!oM~T>u_f0;mDnDp zrA*&3^0M>)+j~yI|KUxlcxf90Ef!n2aAX>Ds0CkNUt!L|Fu00o0?uNV?iPIBu;|+a zySR_OW_iTb4DD~&ytXj=pLMq&YnHxw5=H+c@WGeN3yR`Ul>xr(Ss1&Z2mDvCc8H&9 z|CeB8>Iwc<>41?#vrJH8Ytxkc_!!{6I6+%8t>%odUw8LgGZC)+Y1nU#YQ<*27P`hk z#6^PIrWs0uFO^S&lf+y&h1z7RdNdADO69fKD_ERDICq`4QWQ1dNc_J$V!Sk(=KC-; zj8*Gb0D}SC$p`!O{x=4wI86@NG-jB)4`K`3fMzRlsN{0~A_pR1sgN>03ZtgG+?u%@ zUGuDCgrh`g{H3G7l=7k6wga)PWK-8-V}CO4zm|psnX4Z0Ym@s!f|YW~ zZh)Hg%_De!n2Y#Ot=q9jXuwokR1s6eP&Gqg>LY}Y<)w;#zqcd5WoCkjeqa&CMZQcU_U2vY0O!V&Gfn- z2kATDp8i4EVgvA3yPG4 zB{3s!D+JUJZS^X$OK8v;rrgV-b+A)BBjlf5ON-}RaLM0uiPR?<>0n$bCSv@!eh1?} z$#fR5{1$UXQh4|+v1enaF`>Hr6}I)}#nnW+dy>4Hf3A6V?0Wng=Zr4$=Z zDaylVBi0<89sHg0$+X0YM#?;PxDi9@SH2UrYvs=&gyKVPW~;W*6re$_Sgn-H{?Pty zlX|9`ox}*SBF4(EedjlIE=@dgiTp6>umHl%i!Y55;zChE@I4Wos7wOAnTpnPU1g`A zcJAh?LdG&Ax}o3?-#LVo8@ISWH8Qz1Um89x>j(wiw zUysWU+XIec3k~fMl!le33z(r!?4`F7W8-xo4E{%eysSKRFRV;lJZ+%p@IyByzoPTF zwIDAGc2@*3%kXCX_Fz?69FG58jvnZ-~+=?VuEGbU-&{5I^=3wSvx{~g%h{0r! z(X%9-ACAYhs-5_$wF>xZ+Z_!9`L*k)iS$5XZ=!-5p*sUpt6vnGu@9EkCFM2nHDF|o zzX-a$q^4DD$pp9)4rGYeLR=ldF5G41}+kz4JJZPmIiArflC3nUJ^!AR!AgVZqeAE z$rOXnw7yoCEv34(?!_*#2t`E$$kn-TS*7pH+eP8eR1D-KAwb)LVGG$!=ze zYZJ2m>9v4Q`W9clpbp8#)zt{SN~#Izq_C>R?Vjy7uIBolB+We813B1FVTF0FxJ6^Q zuYbaPPauy&Az4uF{KUg209=Z4D9Ymg1qWnI;TU{taH$Xl)fMUL_R}af&{}zmv>;ao zurL*wiI+n$y){0&?F&dv(K?i#>%*#8ykbqQotqzyu@Q@g*_X8bNLl%1r_cH9$9?yo zhO7>K8e+)N9loNsSfd0{ly=kiriEjy55t$;zs^zTkA?DO zv#m&~W8oN&htlRGAv!eo^rgPh6d6pJlN+UFBWpPJ+)&^Nluhi;2%|0vrx*Qm4lfV1_vmSPaW+z-cd*T8fj9X3P+kUqXmtL;Cz6l(?T8pEp_~f zdGTH+-QyxvDI$AMw!efH;xcDuyQ(Vasr=_8SQ!zuz%n>|m zj}xF54+XxTu(4+$`#Tob<8L7M$8$4aG}8J}9-a|2CV%2zy{ z#*cD5bCH^NFcX>aM?}s=zslC+eDw};d%F?d!l8h=dn#YkJ}BhfU((w3;7fX$EJPd^ z9QYvI3;3}%w$?E%14G6K#qK}UhDyvtt9hp8u#_|K8p0J|u>8ek-KgwP&9OBfa(Vhy z#b~}_y+ytSc(^_dNmi1JGUm;jqNM&D!WIEzs)X*w2Q&pB;>`smg4ZsWfgY@%16<_l z;!x_}z!Hqwe-O`=JBDqS6zGfT>s+{hE7%l!`PAkUD+P7HN5W(4SCNt_mS%3^#Sg+D zGp9w4-n5tq3S|kT8u)&&p(rZY6gAw_(=4gUYVaDJ2g7zxRGV&0D!E2!NFgJ@c3<-? zRjd>TJR0wkoSlxH9mWzZ`7;v+Vdm;w2pO6baY@QN)^je?^R;$d{PNVnhySc%@vu&= z7WE?H>brH12CAy?ZuveutSlLV67TOBw6$$OIsX87YJNknDNf6V!u1gfG$_-6Gup^s zqNeX{6Rc*KdoU>R(PH3%XPVvHQ{RclPs6i_4Nv1p4c3-_;3?8j!3a2!Xvd$WeZ~9; z7-B@B)Nc=9?E^bdkspk>MGvD<}{Gh|k2D_7@iv7$E6IvG| zs{{(8N;#uo4uSNKwAD}@O{FS9$00Z#{+S$pV%@?893zpZ**_+ubf=p0@Y|<)$5_is z{MG0xBQ1yfVQ|Y@l6XSr_vNTXk)e_4e@*irchkON&gV$TJ_=&mtb>zo@MHa4!*or_x?Nh~6V00cd9B_h|b zC%Rf}xZ*X9*y-hou^jC1K(f0Y9~;`Ope*L6lk^L>%o+Zq2H0HktcBD|L4(pv#{;du z%#)=p4E?+YxNRo(_%e@?3yVe0x3R5*g;2tH)kzsQ0d{7@byHP=R`d-yRp97BvHC)y z&X!~huSuG41I3L{jiU}uPZNVM^g&0#bjMVSf0ZoxTM@$Vc%9=ZVF$6!YdxXnQX?at zSQK|`>)z*{F?l@TYFK_JGVrO3!&cc&3M00Nlv1w?+C2P-3-NIO`a@3?Ep373V)~w< zfop!*F#;o`+-vh#o@?3EeVErLGLr+RRyrRc>2pUr^n6kc_sTcVi9s z_;|4AHS2eGT&4UP@t7N;Xjp8%m3o%@PfJFPFoQKyiz(1y@nB)gCrs?ks#Sx5GN+yOJV`P-V@ z(Y;iza+g)~);%X`g$B@h|0{itiWJ77UXia~`vQzg=JEjL_i&NsaAl|1=yQCYXXXkH z!;Ss+f-yJ8^dmpU#a8Y5vuKBneLu=Y9p8e9Chde)qTWAoY!QnWF=LN-PF_uTbI^`g z0K+dT@csjit{EqIZ274Pd8|DsC(xq$6hT^&D{u$jC+-voRfrAc-0|l_cRovAz~Mmz z2ax0%l)E(2s1Pq2Tfi{HxM-p1cDygrGVq@>+BF>EwZve0w%$P{a7+^Y>(dyVs(OLOzW;X(-51 zxR^|sbKbD`e*7-eN|Kg;*tk5d2V0P6apNNaD;3tvW3m5rV z;YR_>BXDKGxjemNRojf4NOX%#rfqy_b;M;G=+VLI7h&BOAwy84%VK2Mqnd-5BS|76i+^87IDQ%=B9GjSQ; zwiWLzB_mtHEd^KCXAHyhHqU++wfs2j`ViBo3;t@}v^mnz9Y9zs+?K!^!dn80>Kx7r zwu5goMey?!FJcfMIpWBC9FTT={Y_+OSvIRo!0l#-qJk_qW6TTQ98_0c)#1a%ONGRo z-JWsP%eTD}lm?Y5mCusMc8_(Zg6N4^^PoFPXQ2_rN~zaygO;MV?6 zQiDLQWfYDZtR20rX9tffCTn=^q7?zq8Lg?D(IKxs^Cyc)86Uq!h50kFiox-r-+lWN^)j>4|BSl;Q1`)TR9eu+wQ+EYtb z;FVc`i~G}abips?&q%-(Il7QvR1^DW#w?+Wt^%ef zPns)YF;B9)K#U7vfy5OD?#O%MjyP-D%YUNr-%wwV~4YfaN+`x`nlBKIOWv#TiY?Ws*E zxlxb>qZPzP5=#3oqA%omvXl=}>Dx4b2CdNZBzndgzgH3-gfq%A7^#cT&P!cO8?)cc zOn^No6*4kSb-~2PUJhi?6~Y{>eLus%j7va&>Iq|XcZq>KT(((@uXeL$&JC}>E@;x{ zxB;qo(nc(}Y%=O!Xs_Nr*&Pt3)QAD#B4Pg3m93sfq&yul=q01@1SmUr53%Eu!`vld z3%#Jn%);0{{FqSNa+tG}0JNL#Km+a9i?U{lIcX`x!=3(a_SG-)lHPrJblg*rdP=GW z^7_h->7*dFyaN%@ha|QF;t|<_C-W@wvs2xCV)wk}U;t8Tb4?UrYBZFg>1mN-FAncu zh2)%(1LpNrRrF($M7DxVvItWPs?Q86NE>Qwu%O+^6^|(7FJ1LomyuB#`P94|sJj;hHCm90+^+ zgPC9WD}0TmM@*#01^WgttCdB;zLFy5qZ$8kXfTw)Vd<0BpkJw-rZc10x;yt+abP9Z za{|fiD{=%~kLR6mqr6gwY*8tG#+5J}Db5z2zfKJ+Q@&NZ+-VZ)sFrKh-r%<%-oA~Q zM;v9Z0S}C>tB*JO8T#G?8)$yQ1-P9(Tka_vKdrv#PpVW`Y#y~iEW z@Qt8(?l)cpm?u4E0=lEN?|~vorDzd_sGZ){5QAW7Q}AeZUvG z8OLJK+2dfRp_`Si7$A%9lOPqP`zy zT}W2a8J=^Y_5ZuTSjhsE*0>h&DEblv3-3@im;!m_y$DK#vqx?IVmik_@q^+b!q#@S zXO_PW+owRb-3X;0kWHxpyi11#+3X(pB%Lpb^5G{8(6kKgMXmxp2RO1=g79{!G%H+CL z5XqiVsaC*ka9(q5BS$&0fR=Oh; zt$Atr>DsE93BEfm>_u$m$9#kj6p1 zskeQK$~a8Belg8qib^;fH2$UfL9dD-E!*NvPQy=SgIr`V;S3|9v^YAS3_Z|B&uii@ zDyA%=P%XBI9UHmv3tmFDQOr8$OmbIE8$Io6&~cYcPWI;#JevM=sYPHG)^fjF|9g>s zjp#a{O7=iErkk&hjvd^jtyG;EK)6;C?|qJ}xD}@G12F(2|2GAS;$M{Xpa($Qj4xJ2 z*B)0=OrYC3uk*n9*pe3{)@?w=9hK>@G^=YXoifAw@VSoN_B z4flOuEHU6WW_a%pEpr(70bFSn8_i`5R}V~0>dc$QiY%t8LCNDt`Qxvm5q~e`UCJB9 ziZGaEF7Ibgd!8b;MH9XD9*y5`Z;rU$?`M6BZs&hY-g8g5qeSG`v!l1P2h87@o?ag* zr;pPDu8SD(j{MS2?nag7(!QnA^@R2H`qJH6{zjUG(g%Bg{g@{s32TLTeDUyRFcrD3 zO8>d&VJftjX5)oId{~ZAX1Q0GsJ+I_!z#uvv3T>YBThndobW{zsF(Un(w3+zeb{~= zk{(f4e`Y-S^`x=b;D~67o}FZ)kc?!r_19n=+2i2HqU2uRNJrNE_7&k@@zKC6n)k=h zsgD(n5C05<9{K5kE;T_G%{x(h+S~bGkrUO~lsfISstCQ&EB)EK#PQCm80aoJ^X>|v zUC>wKesEf7HB>r8s?2N-nS?M?H`#k>ov16rh0|9K0`Pi5Qad9eD2N~a7L0Z6J3BT8y30QiJnyB41x?2Y^%3wP zHywQKt#W%^I`qxU)KINkTC6$OI$yJI+kUg3ADk#D)L&rS*qGy*kIM10c>lugs-e&Yw|Hg(Q6Ue_I-S zE)KkTu}coy7ntxz!wdHg@?^LmUoZS)mHqk72ku>kRY#q{J!q){0mj&SgTr!Rn&u?E zsPIYkYEQo67+IWJjOS(m#&gB_yCB^2YqjD#)!Xw+&)e;CAl{v)w${uUxk^bj;QQB3 zWZZ)@UeS-=%u25(Bn@0^wq5XoBEg+sH~M>$XN^C6f*!tgIq9|FaBnGB<>bXL<~OhQ z_<#vi34+PHtiK=p`18wi*0>So7yerHj*_yA|JUkOa(~LEkh}@7*YcF;kJ$d*S?j*m zLjzy#*7J=X%DKwZnBDHmou6lWWEW%zAC)CjzSD{m%TQ5Z2~=_*Af zTCA7G2SS3O)+HMO#Qt54k0N+Vn57yCl;FUIGtJw9m`PyMDIrhFWtGDzu(@{gDI$$$ zU;ZWB4HqEtuv2%a8j{~lkl45kxx4BNqLZdr z2}KosIjsIfO%|=2CaH;<4=f$yNPQ5fC@8GV#u!1@Cyc~`OfG=Di3ZW>ok$`~N;O%d zjK&NA%2UczGRSXUoM2Ehmy5)#^^mIJ%x+wy3a*4fD2a?|!K+*Yl0fiaz~Me_MEB}_ z{{mPKrQ=hl2!33r9$C3@&TW{qIzp-IHuW4d5*?O!2GLH?{(u-kGxbC25>@df8$pgM z_K1m#tI6Dxy8;#9Z&If`?r0Fw0|&AOzF-j}cH9t=#mqGS>R?(hxadpX0RXpvU{5J& z()~E1UeTSmOy~xi7@+``ROD(GDE>n1G(&F@JK&S8k>W2jL2+a}r3Wdl+!g3Nf+Dkl zUIB&4C=LLjQuo75!c1IT{_VXBiFDefV$cl66uU=a#{Iv8X>4cC?np|20vDo zA!4oL43S1Nk|BXh{URGoDye*IRx`*G0;>QM;epYORCJTnq_l1q>Dwq^C9wh9MUa81L_g58dGl!$PS*?#8g-F((Y8N|C5UAV=6VM=&?KfULq;30)`0^Nb58 zMxZgWDDxi`7hdy)nkIm)>O*Ul*&=(%B~Zup#!S*~2ZTZsp!#l;tj9PM`?0|SB8sFc zI0HC8V*^~`Id|DJc1Q>?U@DN_hJ->6({JOzfcF0RTm;Lr2t#JX^n*?F|M~LhMAvRisB0&T-;bQ90UN4T=9UL5% z7JFjy2v)Z=g0Uh%8Kfcsb1e;J3a#sKti&FhY{Jgx8N`BKic=nh?GmMy0*RMk4U2+8 z9H9V)08rw|ccplfdqX~j6k@3#!&5W~Ugt@%b754?=YapggE@>9G8ph+8u>8=JO#DwEdx2L?~KGH15==xLHimT?jHp!O1HUU^*GVSEB$0FPsr1R5h;bH92!JmaVW9G{ zhV{f0+|xw5?a%B39BfgWlaRsk5Lj`qVE9m*EP(~zakHyVVSuR>A+}4XA**+i2~jxK zgyo!&8(^M|h9M-F4yml>jPjFXnq(%pNC;6@5MvV(IS?9jqJAii#=j&^T_9Gg(2OJl zMf{9*nvAf`h=Zi(7Bf|WR#hAs+Kos>I^?RSatk-0MyC>tgqQ*TnL28WACbH#kqUd8 zDw&vcNH>rGErPrWXE%e|ucuTduGs%F2+RK+6X9Ro@*1F@mOxZe-<_n0?qC=a2r4MY zT0p%)zNMIrk>ElM|!G*>Wz#u0k;U#q? zaPeSn4(jHKP2uimV?@Mt1jj9dj<~3*^CT51QfT-?MKH_Q<|GRk7SPdwKEWavsY!y{ z#BfP_gj2-2d~7+}EL8nIDAIc&(cw^8hY6~IC;-O^^JDl;S#Uo+VkuNe)zB}y z2LpG=CMDiz$x%Xy#*c)^HIBV2gGr45T^x2bH%>Z2(f054#P+!!1h|A&3Q`QTsZvkt z#LCjm>B8IOF_TOl2J7y{}sTXK$z4ZgpjVqLgMD4 z)N@`}vxOUsNv0f;_|rnMA1OGLvOkGiZ7n?-f^;w3MO7&-_Cw;9LI4l;& zvT+*dgXi)_F{>+~S1*E_^Cb1aGxQB}Az}it``>W*2#fPXL!C~>u0{`B2a-&XS6}ES z@WVHwbkKFG90D0cJw3$fXnU@FK}VG-Z-Ho#V<&qjsxK6Bk~9 znqV@QL*$41pquVb-QDgNq~Jnqu=mZf)Dj38O8|_pAyiREKxIV+L9O7+^a>q`=L!1C zC7Iy9!{Q_#WXI(WpiN;NpOtM=0T#pourvIq1DG$xQYt=Zu4L{1Zp+y{*hzGeN}XVc zcI9O(Y9bq=#SrEAEky2=5aI|h@J4KO$sD>Vf7neOXUZNMJh+Qs z(m-+HUE&qpAM2uSMDKp2dCUh&o-D2QNT7(I+#U+92{}J@1A=uED#^kamfSgP$hRkN z3<25zY_xTUw26#lH5r_Ykv!=*^(bE;{cX|%S{!iIzAyxbID84Ccw9nm$5QQSN>(K$ z_!VR8Q{I_S#5KJ+D3o5%di$N7<#S)Mi7!GgZo9CXa0ty%<_w+`GbV}J7#j)*M)4E8 z)`KJAoG`y0Hex}pAF|ZIHsTJX=xhiSi&OAQk(PSqhbG~e6KsZw>Lb?x=VyuX@09{s zf+_KAjHDt+2S7t2!UiF@@AL;F7B@`ZvZ?F= z<@gaxq4#i{#G>DZ${X)|gnow-2frOAiT_M2GNm5tB?__JUy2vQieh?Of-Yo)DHlo^ zcfmo<82dx#nkMPLw@oi-iz~TG6ykm=%{7qyhd!2yF|&forfV*pl5l(r6UOw)E+37* zNNW3Aa45nuQUtn0Fq|P=W*J2b7m^w!BCAS-!An$IQ=4Op`wO4B9tpCgCu z@THD}$RH%6u1GmyO6Wi#vZB46k&IxYP!!MhD+VKsavU3Nj{^H|(AM!#DaWX|*mHva z0xWKe=8oMAA*@HSRD`a@mf~qymD#AJZn%P&r|@yEWy??mum}b=7c5c(uDO#WU=JiD zzAU~?Ac-Kryf{}}jj+%GdKTnrZ4p3WYQo@=fW6F5FFFU$NnS z&pSc{eF-y_{@VVlkpLt*6T>Td5ABYjP&m|LgU)UGLw+ebshM-}FERVncp*2E;9>qa zO6aF%t_b!#37?%%90k$_m(_SQ_7;{gbFwjMfyA+lPh;Vt5}P25KmQs zN&#{=+*c!rERb?9>ln{V;unF#S=dQ(A{JKxgOMjDg3A?0#ZeV!32v%=C{(sdFJu9t#lK{WsOEDpzxU=0@H5KM47Pk&Vs z{0e=`7e_T`sEOE#GgA&z*=IxT1`ZLPUO;-&v#B;k_}7vNC1;GMZZWYaRW`?n;ASJkJP9+MfOshOZKe7R z%N6oO9DlAK6yy0D)ge#&Ne%*@y6*q#k)#a!g4JG#2{$OZz?m^)jWXz35Dt=@u^(~^ z!bcG3m`EB}g{$L16~fqD=kp|Skd17N2an43|Gdrikfi9Q0zAHHIl0Q@6rqu*(|NS- zr7q3(P{-Tkw7dO9QUF2fbVirLlSEY^00G=njtFDoj%?L#;GJ0lNq$wKcyOv@Go@Q5 z{p|wWzoQn>jfGhTFp$Q#{g)b$Rm`OdCh4@=MWlfLE_j`+iNzi#Ikx~7vNQpe@IB_W z#7_ci$B^AF#i#ap&=1>%GUlD(f*9^7JM_3>iq9peb4t)ps7Rqfd5qu+^ABuwzwz@wgg&R2;|ZPZoqLLIHgIL zUW7XCI68$42F7NqYF?DZb>iC2-q6b}u0nrC+uVZ_fI|#4LzG}d+PZ~9&JbrqYf{Ix z;4MjF4?!8#;hHxH^!|Yb?fge*e zUF6)a#3}GgZG+*cFcT#)VD`|Bgzk!016&q_8sv;Iw>xq822?WODMAeeFweTSt2*3r zZ4lVSK&5ohA+vd<}3Hh{)kDu6bz=h$TcsFd8QTd+6vk@inyv50ZQTx3# zF?r<*Vz7yw(Ky#got*>~bBnM@rzLV^?_xS)AOgXn`_M_49$bN{vqR|BO(c&vLqkE> z+(I8cGCFY-)4s|6$dIPp?ScbPp8B>)9y_3uO=)`c^r zxbyxkzZE>eEzl8h7OE{V(@6&x@CX$l?=oHVxtwIcfu0**yb#5RRSJ9%vI%ocz2s?| zbY!xyk4f%^IECPb!4U=kNG%%Uva^P%CA0~Cm|)HT&*PAqN}K89H0XeAFc2F=6-7d9 zKruoHnx+LXqd`~2B6PH07@^n~D#;XPhCPxmjdqN~TqvYJVUpR5D~@C|1jhYljAr4c zV(enbja;C7&)SWH1BB^Xu@DBQjtx$voxK?vgNDxuB?wLj zSQ2J55`t}iPn!F>AR~zH>yUM;LVxv`Z*-dLcS z26ZJ;gj}catDyYn5Lf`JW)*hW`Y*sLppBALZr7$SJh@$elQ_vq~ z@3qTA^`q!;B3%g2#+p=x&dCyU7Iq}Z@n}XNNErTBI`B*1{F+*(7U0I&gae<3;D!y{ zA0va36qBJ~RwSMEHj=qTdxs0nwwA}Hcok78%efEN94y)TL~!Eq8217Y9Ww5+t46kIJ}2PPh!a-FmAGIy!*j?+ z(ryaSmu4ew##FLOfVcM!q7#AJC%V7QlB8x5wqOlygtDFmLZJUo03{*X-i(wcQSK11 z-z)*=hd3TjdF-ah6*1^f)DI}}!ED43hSAWWFyQf6E>?TUEwBVEY6uPx1YR6Exbz(} z07_sWNT604$?HOo(*YV>uhgAsmy3d7Q}4 zInL6OoaEpK@Jw!Ygbn~^Cru`gAq%hc z%glja>^O2mgiAaa6>{vpZ5g#vF9Ao2k(e?T_?cWp387!Wz=xb#&E%X?n0oS<48$fW z^Ei!@G~pb^t1!W81{RG(2{?ChGA@9@4%{jijATN1ii}X?J`xz10*^A6%k=;#_mo5D z^0-g2&R!;`*(13dI?utSf^H7+XPic#9b8oEHl0*HL%@$U#i0Y7Lr?=gxz zbPO@75B@OtXL1ul9=kG(WiS+nF^59K`DdBj8z<-gAsUlAD}tO$e7Qm+_eJ1*4S9l; zgBRsO>Cwmwz#jQL;MAe;e{P;3!Q{Lz*jk?Ya^s(u$W;p%^1>{Wa{)Loxsaa|8K%J4 z24;dn%n}EwJZlrX$aqu6tq};(SZ)PpautSVVUVU7l`GiDc9!ShrjSsY8{Z=gdID4c z)QjVSe;nfiF$w`|3a$YZ58=>D@HSasrdO z5a%bpT;}uS!l4IIlUz1sUgmp&g9P(D62XWG%_@a}j_?qiTZq7X3M~+N1rm&c4srz- z`alRd;}G_PvCLi9lUpZ|U=Sn54wMoYB5*NyCb#NE38fB8BIFq{3MIqjLXIc5Si6C2 zYT%DbLkmt8cadN)K*qVtJvWvcbitE#p#93FqtJ!sfdnJcBY-9r@Faj(C-MUrlu{U* zWE(_k7Jvq_z@g-0u+=hPgd9UN$!b|)wuk8(y~$jjwfLdKFX3FMT78wbchjFNym za;H0tm?=+lFJN`0^HZg;k4nij0vFxA;g}5uCV8lrR<~#Ns2b@5hd%)yb zDe#9W*nYVIHUrl+TtgL-kOdIONf)%;1}@{N#0`-{9b+_sdJE;T z6Yi#9W}TctTnzg8-~!^ThFo5j=l|0rg^&>9G9NmGoTFl%ydnu%0T7aN@Go-K9^I7J z%w%qq%VW7*&bT7MQYYj*^>`*%>bSgw%k>CnEQcv;?_B*C$yT%Z*;Smw50d!3}jh zax%HS+E25ThE68eI>D{VNeSJ{8nu5QRvvTppo= zVb}MZ%m;6U(NuR#97dHePn;x1 zf-&euId~YEp9X=K=B`|*hqy6Lzc~l!lW0Hr^FG6t69HiNX=-2Z- zu(uR~MX>(z9CPABRgj$JQI6Eibwg-Ya+L#$i-bW81_@eQNK=UU(-;W`zA*%si981( zqq`j11djrmLY1T;Zm7%UeKi0`lgx=2^oKA9hb@JGH~=2xfC7#gbsTx~hdlK_NaV}C zMS(A`5OE^p+;;9e5P5;G0wxShT6-ZRqhWgw0!fg&Z@{AkJ1cjQGgt0NadUY{DR<-8 z2z{YDf$8w%IYb`}xtC@h5)1%hxzQ`~nGCOePhJtha>s*k4;don zVGPbSv;{ehk%eHuq5Y;{z!N`pDRK~&%Q3MZ=5j4qUPlF^fWw^t!hs|w967h0%VkoC zgDJ#z5coJ;-ZcQRaljzLLzh_;G3wJe1|!7fIX&)xz2J<HPtQkVJq#SvdR}K}JL6j7N!xF)emxCP(Qx>>^ zT#!tWO@sI`_=#Y*eR(n?h5=aChX++0L(9O8PT!0*(rJ*=ZD~@;nd85`qj~!Rg7s(!>yF1`r&A^-moqPDmUxxm^I~hUOf4 zc9=MUJQD*xEsQED7;HH~3jQ{7;Vu|pvS0*l6wv+lXPT_ccl zsxY2AfX1L(O6yM&Ck|%k&u;e!pKnj@T_G&*!ES%bo5~gU7+>C9(Ef2r+uqSGUbUY> zQeB%|mVfSl-RK|icR}{u)#T^v*Y8!j-N@cuo!(x4{g3YPtM04FI->J8lh0J`9nbbzC4lJH=*Ujr&i);Df`p)^{ahRlKoam-@UqWcQ<1HWxwTFOO=cu(|v|Qt?>Cn6K%q$F5&wfzrg$;AG|8dN#Xx^{R=H|r=2KOFkOH?)_2~-gnxF$I%cq;jt;EXH;rK<>p4u0;`}E6aXo0f6n(Fm0r^DMX zZDamtvx7qXTTKqUQ*3m5{`CU@*DC8>op1crfSc^~V*tI1nW?e2E!us>=|1*AtD1k; z-BsW3%Ld6iuEE0@9{Cvna~hd?$ReK^ko-UzzOCvnII@eBiuCesnw(_9m=e+sh~UW# z{Mo-Zf@f?9&)6*d;ea7H9L5xf!wkv&s4M(!hbBBb zB+*XEG@qr0pt1zX$3*bZgvUb?c}gbFgldPR+9|1KOK&T`WoFK_y?hnz_2A(=531p9 z4~m8-AI|a|N!=99;EONDLSxSBt664I)eVMJrXTakr*l-7uT7Z|UA{o|`u>>#$#?IO zH1AbJ-@P}XLU{=n2_@aU$IvCijJX#HHR0^|+a)WpwWb&yFVBq#vAz2yq^33M>6etK z{(ckEGSBz@Hl;xM(|sW|vzYg;&nX)$Yolf~>HhF%`5Tf?iYOW73o}lErW6`Rqb^G2 zV{8NNgw(X{D^q541MY;h%w`Q3QXcDy;i;vRj$6JnB!@S6Zg``khF2!ctRPz?uM|NJDdD*Z;pJ1CIMV)CHeajXnUEI^ z$2MpwqNJ5COgTmIJGbDKVp?wb(u99sVfzwDC|TjTA))KwDP2cL>N=*hM+@4cmiA~! z>)}~F9jRx?32*RS_bR7j@7|iIfFFu=M$D|hT4pJtW`!3foEEN?)~_;ks%lOPNK4U< z#!XPke|v4Fh1e0ayK$D%@VD=b$g5L{E%#PNO)TFS(Ta~XHY#gn)U@)A39U^*>xG0X3A zA5jZ~_i}3f%Ucr_o=J@)2MHxDJU1dN_$MqQh^Q&yg#qV=L(fAF{+*19wRkC3 z3&s(H?p0(+dVMj&+3YzH&BjG=UPXqS*B54-9-h9wRc*%k*@E=5mi6;2?Y^QPHU}a;aJE5g_!!6y>F?jps8Cmg zQP~`t%ACI^G$il6nr5cFFgH&!@qkG_nqk-%<}(*E@sI%?%`jAlW3e|M$Y&NGAU%nuI$XCnBM9&0M$C6_r&VA) zC#QY0VYt1(_hJJD?q8bmga6IR0nOiuY52RBbNt(yBLd@d`VnLMbdCz5nzF&nDBVOx zgqIgHoR~Fd#cK2XIx~8OS+llmwyMkuK{9y~0qF`2_5Jvu89`?Z~ zaW6SmYZq_X}8j$LabRNm4Z|0XB5vH`JcIZUHYLC-BTkf56k7L$vh*r z63bSq*$j%Yz?Ng7F37^DBnxy=7Mijw%nGwWmu8_U&cd)fYjlCuni8$einPX;X{{^N z+OSk>e6iNLa;?n@7G18IVl^vOe4*;f)UZhLC8{e>lk!CFzgF+OHrjKY?<>T*U7LDD zQ_Zc_wMjO}u&ghnqTHB-X!!|oXiO#^GM(m*igj}cm=(yY#DMGi+=%cap7EoO^5ZOL zP0Zt5nRv(pkLDOSX%O>0=kz0<_NfsSa+AlMl$zFLqsNXg0#Dec-E@;<+}N?`K#g&N zXOzk zj-*6vsc%z{NXog5^pJ#aahd4^##HU&IksbkEV@o$#MV8YV-u$gcW&z@GGe^EnCATL z*?bBE(hgYQ$u#_)=gmg!3WUFP6B&@*Ud(X{fnXD2#3RH!1JWelL=N)<%TlnPC}0 zL`?}VW;w-?Al9BQ0%N}R@eEsq@XKW4AxkyqmtjsOj?W~{o<_u-LsXqWG&p;RJ9Vf! zZ)kAR5Ou~-ak_B+Tp^?&jA7=-uM%Uf?)fZXLCGGqiM1y)CcV5dqb;ijO$#y_T6kkd ziwK8rA{x@Uz!-l>RCheo@L=eHG>4P<2-_teP<0RIcs?UYK93ka&%zf6#FFSa{g_5R zouR^< zA73FA=}9UT>Ag2mWXqckp-u73HwS*e+`6cdS1L71WvW_Rp!S;e#kGp{6PfCD8dIgZ zELdyR7uTxQPh_N5Lu#3sV{@vpjp|NTtx-RPnL5?cTAQ1wMwwfJ8C?+Dq{n-tXE#Vg z0^AVar61E64Dr|-9W-}2nBC%VkQZ4W_Xt~X78tUfA5XKDq}(*srbLwFO=Q4$dojbw zw?qzc=8R{7A>aA&9NQ=Ce(Rgai1CxrK5GajkaJa6i6P_V`7~iwB6fAl!~+`nXr2Kq z;s=hRX(bd~?eiJJ6K1`qO+6$!&93zv5%VL*(e%i1ZgvEJcd72;Qo~zIkJTqcp|z=p zMDp1T3Fcsm;#A7~kredN0nNbyvtt97RBVF`e}{q^_RH&OEo#CFQkwVTbNT`2`RN>0 z$*RM58L!LgwBHv9&U?WT`Tc7nEogV=e1W-6XDq7Lq?f=*m+k2wzc+wXqmxo=(7iJv z|G>XsnL$KN3onc~_vE<8%tqgfskzTDjrdz9<(bq4R8+gg65j|E5Nf>?35|5LQ)@~S z1iAdDC>t#MSxilTerd!nIO58Qo{XAVzA>Xk-3rtf(h$$z3v^D|)Tj3NHnVJwLwU76 z!&FA9oyr2WA>>|6t;qT13_t$DGwoGprk9-6{6OUAUTOs8Sz^R^em+NdJm7E>)!-HM5!|I zfI>c+XV`4FryNDoN+`J6=LUp=Lu{~6XRjqUSA#3+^ma4?? zlw*=|jsZh3#-E4ZVeofjf|)~>qRJc0qU<^Skf(b(M};ulif^jSe^3SIJ*wPB1Ah@U z{wRhYgV6_fU!?M;NRvAvvA092?}aqF5mGdSelRDJsfRT3*&GSNsx%{u(#%;CxdbA2 ztQQ;+nB7T@D0}bb$Z=L%X{k-D^C@d;LJ|2p3kWzW8D~51m?5^VT_uKGm*+Etm^%}i@l6TM?EMH}yv$9yOgv7mHW{$E= zJ>;loXC(1==jv|FHM}pksS5;EF(N5u;rAH){g_}s%W%RloW=~NX$JXa>-JjH4eEs1 zz1V>D{^cBhjUcT@OfyqXsH=+=mln@oSN!~&`PNZLJH-EOX1nO~fgtc9@sI!>O*4Gj zafA!prXFy#&!$O!=QZN~!#4GRq| zr(qB6eMns1Cfg=9zW75ceA7gmM61yW21@~Ejq?rSi1t|Oro;H13Atgfyq8dtPW8|G zWSdV#vopRMN++vR;+Dchotm?gnZqy=QYz5BGGRWIb`^h>Q}SQmDplyhxeeWu`7?O^ zv;&*+?k+0SeUrBL_1VAlKRU}uLpJj0qe_L_%TDq~`2G65{CyQ~Coe_Y={p^jIO9}j z!0yFsc@m||#|`;*MVtz;cY$1o)Jy(DbvT^Lf|CBe8+rAO0;s(ar$zW1xqYiaedopX zVvE21?fNC%?);b{@%dm`oBrpb671?`u52j$@BJYRqd76O3#VhlU<&g5qfw+ zPf6&_5&o4lboYyT8=n8!2W@sT|Kn8%;v z`LV8!)u}NjIt$z}ou8ReVM^~RqN4K*^XB6TeI?;+Vg7W;IAcY*H7C4^+!0=%&Jh0n zr=S02&Lvad(Z~mL3@Cyj{yhAz82DcY@W0N&%UgJHX6Pz%Pjz`ZL-?PhX76@&QGPII zyvf{A-X0rpzE@B4%}9*+-uGtvCS=8@LLli{kr7|}bcXOBrD#Y=Ij$P}Qg^id7cC{(|d&0}pIl|7tGiPhl-&3`Z zW_aF^&sdl0DssnmeQHMN;R!t@;p}Mk2G!@@47=N#`JVQsVy8X-jvUozbqu8L8GA1f z%<{m^q5UoFl;{Cy_35QCQ;qJ5J*#<%R_P+BQswf_Oa;wN#PdQb>i)IF(+GL7mp6UM z2vt1`gArG$a-YIbsamtXZz85tqV6yCPowNP$*r4cUs$C`|Gl9)6=9oCT<&GGP^TxX zRHygeOdZ>TnJ5!fag^jB&!<>waVM?P$@H^89R$>X%;-G{5|EPw~qybMUCfGio~qO-Ck)a1_yqBbpi$@Av6XNn(7iv%r{Y*3*)yj{B3q zaY>00nS5x#CXn;aH<=NmFqwC56ROXl%t(y+ddE?8vJFQ}KFHMfEbw540XK&}nm;#~ zJAZ$aBfLDEVQW9NN2_+49kH%>Cw9ks_r`>L zyRYi1(VuoUmL02c_Zp##{-CqDuvVMFiRUTA9k(l%9oH7gjw6a=$1RKECaBZ^Q3Mm; z05B07(+@ueFoj}O__~reFLt{!*B1D6XCFgcXy2Y|? z$njWLpKZy{;v*f-&vB~AJzS6K)591m)q82KnPtrkD+W<7JNbqRMWx@RMn=#s!rv{QaIGXq1cle z2I!4^@qqGbK$ZIHl;{jBROqv%{ZKJ)$w`QUl$&zDf3$>0SSoEtiLe^)igL>zRrLm< z(((2dt=uW&^~1_tL6pi}-(r>fu$SMUdsMp~ib}oSYpi~I_fg%>2r9kKuh2Sy@dbOZ zdswqRj#9sV0<4zsa$A&}nv@@)={~Bf0!^i{A{kn5@o-PRKqxg7cD;B3TR3F`Z7rWT zR8<2FnI;bKJpRe z9^3McXwj^ZEp0U7jf{zfWp>3dy8@Hl2jUM^?17CtL}8ifVwmXylWF@%M-}B@BOGD6 z7^D>)(`%Q7)EH(9M0(kg0uxI`Y83lP5usi_5_8K(3M8YY9ktv&w|ZR+oQUD5D&4QSw@KZlkXu-p}eSdS3$mQ}nVXvMR zQmL~xwRJ``2uS->RbA2)g6C{oz(PN>%TZAobk0+l#N__KFUGl$s9GAXSx7%%hcm!%!-Cd#%uo zR)<4XZVWx`aq^DwhxYNb3x54`n&JBAz^hjvUW$B9>07!E^WF9L1+@E`gaYOKnTb~H z*Y@v=NORedry{?->_o5I_ry(HUW_E~NK%SlUS7mGfBi4{`&CcVJ&_;#QR@1q|F~!? z$g`Ynm**|z>AuT;;H&IOkS+i7I8_JJJ%57dPZ#NT?W6XpqMy%?+MATbt3h=P)zeo9 zXxpn20lkUt?jgEo=i2MuzXiQt4Ju(sIj9EU=~aZ>*Hp;UeYa1S4oc-dMF(+y_H>r| zSNHSvo9@e`ot(T{e{gN--HU6Vfgj!FOx zBELoQd9bhi{>^35;+*c>Ji0V{^81&KJdb=L!xKHq1eY0%-@m;6>H6@DRh|y_OXa=C zmohG&p-oAmrh5I$>G1YT+j9Te?7$BFO&(C|?uqIoKb{KuVbgxFbPU^j*@6GJeDL~R zoBHpkg7H_n&NK$obhwG$rD~F6Tt$;rx6zZ3{EEBE6tB-PqV4|Y>&e<=n_-I`bf96I z{Zsq+sV_Cviia8wyXQ#9KedmZE;0j+uf}$4Chp8_YbLG+_j*U@fL+&fK>F#`{|8&u FQl4uG+!O!+ diff --git a/docs/build/html/ops_8h.html b/docs/build/html/ops_8h.html index 3928d178d5..550981121d 100644 --- a/docs/build/html/ops_8h.html +++ b/docs/build/html/ops_8h.html @@ -260,16 +260,16 @@ Functions array mlx::core::moveaxis (const array &a, int source, int destination, StreamOrDevice s={})  Move an axis of an array.
              -array mlx::core::pad (const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), StreamOrDevice s={}) - Pad an array with a constant value.
            -  -array mlx::core::pad (const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), StreamOrDevice s={}) - Pad an array with a constant value along all axes.
            -  -array mlx::core::pad (const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), StreamOrDevice s={}) -  -array mlx::core::pad (const array &a, int pad_width, const array &pad_value=array(0), StreamOrDevice s={}) -  +array mlx::core::pad (const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={}) + Pad an array with a constant value.
            +  +array mlx::core::pad (const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={}) + Pad an array with a constant value along all axes.
            +  +array mlx::core::pad (const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={}) +  +array mlx::core::pad (const array &a, int pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={}) +  array mlx::core::transpose (const array &a, StreamOrDevice s={})  Permutes the dimensions in reverse order.
              diff --git a/docs/build/html/ops_8h_source.html b/docs/build/html/ops_8h_source.html index 08917ed839..235345f749 100644 --- a/docs/build/html/ops_8h_source.html +++ b/docs/build/html/ops_8h_source.html @@ -322,974 +322,978 @@ $(function() { codefold.init(0); });
            256 int destination,
            257 StreamOrDevice s = {});
            258
            -
            260array pad(
            +
            260array pad(
            261 const array& a,
            262 const std::vector<int>& axes,
            263 const std::vector<int>& low_pad_size,
            264 const std::vector<int>& high_pad_size,
            265 const array& pad_value = array(0),
            -
            266 StreamOrDevice s = {});
            -
            267
            -
            269array pad(
            -
            270 const array& a,
            -
            271 const std::vector<std::pair<int, int>>& pad_width,
            -
            272 const array& pad_value = array(0),
            -
            273 StreamOrDevice s = {});
            -
            274array pad(
            -
            275 const array& a,
            -
            276 const std::pair<int, int>& pad_width,
            -
            277 const array& pad_value = array(0),
            -
            278 StreamOrDevice s = {});
            -
            279array pad(
            -
            280 const array& a,
            -
            281 int pad_width,
            -
            282 const array& pad_value = array(0),
            -
            283 StreamOrDevice s = {});
            -
            284
            -
            286array transpose(const array& a, StreamOrDevice s = {});
            -
            287
            -
            289array broadcast_to(
            -
            290 const array& a,
            -
            291 const std::vector<int>& shape,
            -
            292 StreamOrDevice s = {});
            -
            293
            -
            295std::vector<array> broadcast_arrays(
            -
            296 const std::vector<array>& inputs,
            -
            297 StreamOrDevice s = {});
            -
            298
            -
            300array equal(const array& a, const array& b, StreamOrDevice s = {});
            -
            -
            301inline array operator==(const array& a, const array& b) {
            -
            302 return equal(a, b);
            -
            303}
            -
            -
            304template <typename T>
            +
            266 const std::string mode = "constant",
            +
            267 StreamOrDevice s = {});
            +
            268
            +
            270array pad(
            +
            271 const array& a,
            +
            272 const std::vector<std::pair<int, int>>& pad_width,
            +
            273 const array& pad_value = array(0),
            +
            274 const std::string mode = "constant",
            +
            275 StreamOrDevice s = {});
            +
            276array pad(
            +
            277 const array& a,
            +
            278 const std::pair<int, int>& pad_width,
            +
            279 const array& pad_value = array(0),
            +
            280 const std::string mode = "constant",
            +
            281 StreamOrDevice s = {});
            +
            282array pad(
            +
            283 const array& a,
            +
            284 int pad_width,
            +
            285 const array& pad_value = array(0),
            +
            286 const std::string mode = "constant",
            +
            287 StreamOrDevice s = {});
            +
            288
            +
            290array transpose(const array& a, StreamOrDevice s = {});
            +
            291
            +
            293array broadcast_to(
            +
            294 const array& a,
            +
            295 const std::vector<int>& shape,
            +
            296 StreamOrDevice s = {});
            +
            297
            +
            299std::vector<array> broadcast_arrays(
            +
            300 const std::vector<array>& inputs,
            +
            301 StreamOrDevice s = {});
            +
            302
            +
            304array equal(const array& a, const array& b, StreamOrDevice s = {});
            -
            305array operator==(T a, const array& b) {
            -
            306 return equal(array(a), b);
            +
            305inline array operator==(const array& a, const array& b) {
            +
            306 return equal(a, b);
            307}
            308template <typename T>
            -
            309array operator==(const array& a, T b) {
            -
            310 return equal(a, array(b));
            +
            309array operator==(T a, const array& b) {
            +
            310 return equal(array(a), b);
            311}
            -
            312
            -
            314array not_equal(const array& a, const array& b, StreamOrDevice s = {});
            -
            -
            315inline array operator!=(const array& a, const array& b) {
            -
            316 return not_equal(a, b);
            -
            317}
            +
            312template <typename T>
            +
            +
            313array operator==(const array& a, T b) {
            +
            314 return equal(a, array(b));
            +
            315}
            -
            318template <typename T>
            +
            316
            +
            318array not_equal(const array& a, const array& b, StreamOrDevice s = {});
            -
            319array operator!=(T a, const array& b) {
            -
            320 return not_equal(array(a), b);
            +
            319inline array operator!=(const array& a, const array& b) {
            +
            320 return not_equal(a, b);
            321}
            322template <typename T>
            -
            323array operator!=(const array& a, T b) {
            -
            324 return not_equal(a, array(b));
            +
            323array operator!=(T a, const array& b) {
            +
            324 return not_equal(array(a), b);
            325}
            -
            326
            -
            328array greater(const array& a, const array& b, StreamOrDevice s = {});
            -
            -
            329inline array operator>(const array& a, const array& b) {
            -
            330 return greater(a, b);
            -
            331}
            +
            326template <typename T>
            +
            +
            327array operator!=(const array& a, T b) {
            +
            328 return not_equal(a, array(b));
            +
            329}
            -
            332template <typename T>
            +
            330
            +
            332array greater(const array& a, const array& b, StreamOrDevice s = {});
            -
            333array operator>(T a, const array& b) {
            -
            334 return greater(array(a), b);
            +
            333inline array operator>(const array& a, const array& b) {
            +
            334 return greater(a, b);
            335}
            336template <typename T>
            -
            337array operator>(const array& a, T b) {
            -
            338 return greater(a, array(b));
            +
            337array operator>(T a, const array& b) {
            +
            338 return greater(array(a), b);
            339}
            -
            340
            -
            342array greater_equal(const array& a, const array& b, StreamOrDevice s = {});
            -
            -
            343inline array operator>=(const array& a, const array& b) {
            -
            344 return greater_equal(a, b);
            -
            345}
            +
            340template <typename T>
            +
            +
            341array operator>(const array& a, T b) {
            +
            342 return greater(a, array(b));
            +
            343}
            -
            346template <typename T>
            +
            344
            +
            346array greater_equal(const array& a, const array& b, StreamOrDevice s = {});
            -
            347array operator>=(T a, const array& b) {
            -
            348 return greater_equal(array(a), b);
            +
            347inline array operator>=(const array& a, const array& b) {
            +
            348 return greater_equal(a, b);
            349}
            350template <typename T>
            -
            351array operator>=(const array& a, T b) {
            -
            352 return greater_equal(a, array(b));
            +
            351array operator>=(T a, const array& b) {
            +
            352 return greater_equal(array(a), b);
            353}
            -
            354
            -
            356array less(const array& a, const array& b, StreamOrDevice s = {});
            -
            -
            357inline array operator<(const array& a, const array& b) {
            -
            358 return less(a, b);
            -
            359}
            +
            354template <typename T>
            +
            +
            355array operator>=(const array& a, T b) {
            +
            356 return greater_equal(a, array(b));
            +
            357}
            -
            360template <typename T>
            +
            358
            +
            360array less(const array& a, const array& b, StreamOrDevice s = {});
            -
            361array operator<(T a, const array& b) {
            -
            362 return less(array(a), b);
            +
            361inline array operator<(const array& a, const array& b) {
            +
            362 return less(a, b);
            363}
            364template <typename T>
            -
            365array operator<(const array& a, T b) {
            -
            366 return less(a, array(b));
            +
            365array operator<(T a, const array& b) {
            +
            366 return less(array(a), b);
            367}
            -
            368
            -
            370array less_equal(const array& a, const array& b, StreamOrDevice s = {});
            -
            -
            371inline array operator<=(const array& a, const array& b) {
            -
            372 return less_equal(a, b);
            -
            373}
            +
            368template <typename T>
            +
            +
            369array operator<(const array& a, T b) {
            +
            370 return less(a, array(b));
            +
            371}
            -
            374template <typename T>
            +
            372
            +
            374array less_equal(const array& a, const array& b, StreamOrDevice s = {});
            -
            375array operator<=(T a, const array& b) {
            -
            376 return less_equal(array(a), b);
            +
            375inline array operator<=(const array& a, const array& b) {
            +
            376 return less_equal(a, b);
            377}
            378template <typename T>
            -
            379array operator<=(const array& a, T b) {
            -
            380 return less_equal(a, array(b));
            +
            379array operator<=(T a, const array& b) {
            +
            380 return less_equal(array(a), b);
            381}
            -
            382
            -
            384array array_equal(
            -
            385 const array& a,
            -
            386 const array& b,
            -
            387 bool equal_nan,
            -
            388 StreamOrDevice s = {});
            -
            389inline array
            -
            -
            390array_equal(const array& a, const array& b, StreamOrDevice s = {}) {
            -
            391 return array_equal(a, b, false, s);
            -
            392}
            +
            382template <typename T>
            +
            +
            383array operator<=(const array& a, T b) {
            +
            384 return less_equal(a, array(b));
            +
            385}
            +
            +
            386
            +
            388array array_equal(
            +
            389 const array& a,
            +
            390 const array& b,
            +
            391 bool equal_nan,
            +
            392 StreamOrDevice s = {});
            +
            393inline array
            +
            +
            394array_equal(const array& a, const array& b, StreamOrDevice s = {}) {
            +
            395 return array_equal(a, b, false, s);
            +
            396}
            -
            393
            -
            394array isnan(const array& a, StreamOrDevice s = {});
            -
            395
            -
            396array isinf(const array& a, StreamOrDevice s = {});
            397
            -
            398array isposinf(const array& a, StreamOrDevice s = {});
            +
            398array isnan(const array& a, StreamOrDevice s = {});
            399
            -
            400array isneginf(const array& a, StreamOrDevice s = {});
            +
            400array isinf(const array& a, StreamOrDevice s = {});
            401
            -
            403array where(
            -
            404 const array& condition,
            -
            405 const array& x,
            -
            406 const array& y,
            -
            407 StreamOrDevice s = {});
            -
            408
            -
            410array nan_to_num(
            -
            411 const array& a,
            -
            412 float nan = 0.0f,
            -
            413 const std::optional<float>& posinf = std::nullopt,
            -
            414 const std::optional<float>& neginf = std::nullopt,
            -
            415 StreamOrDevice s = {});
            -
            416
            -
            418array all(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            419inline array all(const array& a, StreamOrDevice s = {}) {
            -
            420 return all(a, false, to_stream(s));
            -
            421}
            +
            402array isposinf(const array& a, StreamOrDevice s = {});
            +
            403
            +
            404array isneginf(const array& a, StreamOrDevice s = {});
            +
            405
            +
            407array where(
            +
            408 const array& condition,
            +
            409 const array& x,
            +
            410 const array& y,
            +
            411 StreamOrDevice s = {});
            +
            412
            +
            414array nan_to_num(
            +
            415 const array& a,
            +
            416 float nan = 0.0f,
            +
            417 const std::optional<float>& posinf = std::nullopt,
            +
            418 const std::optional<float>& neginf = std::nullopt,
            +
            419 StreamOrDevice s = {});
            +
            420
            +
            422array all(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            423inline array all(const array& a, StreamOrDevice s = {}) {
            +
            424 return all(a, false, to_stream(s));
            +
            425}
            -
            422
            -
            424array allclose(
            -
            425 const array& a,
            -
            426 const array& b,
            -
            427 double rtol = 1e-5,
            -
            428 double atol = 1e-8,
            -
            429 bool equal_nan = false,
            -
            430 StreamOrDevice s = {});
            -
            431
            -
            434array isclose(
            -
            435 const array& a,
            -
            436 const array& b,
            -
            437 double rtol = 1e-5,
            -
            438 double atol = 1e-8,
            -
            439 bool equal_nan = false,
            -
            440 StreamOrDevice s = {});
            -
            441
            -
            446array all(
            -
            447 const array& a,
            -
            448 const std::vector<int>& axes,
            -
            449 bool keepdims = false,
            -
            450 StreamOrDevice s = {});
            -
            451
            -
            456array all(
            -
            457 const array& a,
            -
            458 int axis,
            -
            459 bool keepdims = false,
            -
            460 StreamOrDevice s = {});
            -
            461
            -
            463array any(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            464inline array any(const array& a, StreamOrDevice s = {}) {
            -
            465 return any(a, false, to_stream(s));
            -
            466}
            +
            426
            +
            428array allclose(
            +
            429 const array& a,
            +
            430 const array& b,
            +
            431 double rtol = 1e-5,
            +
            432 double atol = 1e-8,
            +
            433 bool equal_nan = false,
            +
            434 StreamOrDevice s = {});
            +
            435
            +
            438array isclose(
            +
            439 const array& a,
            +
            440 const array& b,
            +
            441 double rtol = 1e-5,
            +
            442 double atol = 1e-8,
            +
            443 bool equal_nan = false,
            +
            444 StreamOrDevice s = {});
            +
            445
            +
            450array all(
            +
            451 const array& a,
            +
            452 const std::vector<int>& axes,
            +
            453 bool keepdims = false,
            +
            454 StreamOrDevice s = {});
            +
            455
            +
            460array all(
            +
            461 const array& a,
            +
            462 int axis,
            +
            463 bool keepdims = false,
            +
            464 StreamOrDevice s = {});
            +
            465
            +
            467array any(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            468inline array any(const array& a, StreamOrDevice s = {}) {
            +
            469 return any(a, false, to_stream(s));
            +
            470}
            -
            467
            -
            472array any(
            -
            473 const array& a,
            -
            474 const std::vector<int>& axes,
            -
            475 bool keepdims = false,
            -
            476 StreamOrDevice s = {});
            -
            477
            -
            482array any(
            -
            483 const array& a,
            -
            484 int axis,
            -
            485 bool keepdims = false,
            -
            486 StreamOrDevice s = {});
            -
            487
            -
            489array sum(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            490inline array sum(const array& a, StreamOrDevice s = {}) {
            -
            491 return sum(a, false, to_stream(s));
            -
            492}
            +
            471
            +
            476array any(
            +
            477 const array& a,
            +
            478 const std::vector<int>& axes,
            +
            479 bool keepdims = false,
            +
            480 StreamOrDevice s = {});
            +
            481
            +
            486array any(
            +
            487 const array& a,
            +
            488 int axis,
            +
            489 bool keepdims = false,
            +
            490 StreamOrDevice s = {});
            +
            491
            +
            493array sum(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            494inline array sum(const array& a, StreamOrDevice s = {}) {
            +
            495 return sum(a, false, to_stream(s));
            +
            496}
            -
            493
            -
            495array sum(
            -
            496 const array& a,
            -
            497 const std::vector<int>& axes,
            -
            498 bool keepdims = false,
            -
            499 StreamOrDevice s = {});
            -
            500
            -
            502array sum(
            -
            503 const array& a,
            -
            504 int axis,
            -
            505 bool keepdims = false,
            -
            506 StreamOrDevice s = {});
            -
            507
            -
            509array mean(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            510inline array mean(const array& a, StreamOrDevice s = {}) {
            -
            511 return mean(a, false, to_stream(s));
            -
            512}
            +
            497
            +
            499array sum(
            +
            500 const array& a,
            +
            501 const std::vector<int>& axes,
            +
            502 bool keepdims = false,
            +
            503 StreamOrDevice s = {});
            +
            504
            +
            506array sum(
            +
            507 const array& a,
            +
            508 int axis,
            +
            509 bool keepdims = false,
            +
            510 StreamOrDevice s = {});
            +
            511
            +
            513array mean(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            514inline array mean(const array& a, StreamOrDevice s = {}) {
            +
            515 return mean(a, false, to_stream(s));
            +
            516}
            -
            513
            -
            515array mean(
            -
            516 const array& a,
            -
            517 const std::vector<int>& axes,
            -
            518 bool keepdims = false,
            -
            519 StreamOrDevice s = {});
            -
            520
            -
            522array mean(
            -
            523 const array& a,
            -
            524 int axis,
            -
            525 bool keepdims = false,
            -
            526 StreamOrDevice s = {});
            -
            527
            -
            529array var(const array& a, bool keepdims, int ddof = 0, StreamOrDevice s = {});
            -
            -
            530inline array var(const array& a, StreamOrDevice s = {}) {
            -
            531 return var(a, false, 0, to_stream(s));
            -
            532}
            +
            517
            +
            519array mean(
            +
            520 const array& a,
            +
            521 const std::vector<int>& axes,
            +
            522 bool keepdims = false,
            +
            523 StreamOrDevice s = {});
            +
            524
            +
            526array mean(
            +
            527 const array& a,
            +
            528 int axis,
            +
            529 bool keepdims = false,
            +
            530 StreamOrDevice s = {});
            +
            531
            +
            533array var(const array& a, bool keepdims, int ddof = 0, StreamOrDevice s = {});
            +
            +
            534inline array var(const array& a, StreamOrDevice s = {}) {
            +
            535 return var(a, false, 0, to_stream(s));
            +
            536}
            -
            533
            -
            536array var(
            -
            537 const array& a,
            -
            538 const std::vector<int>& axes,
            -
            539 bool keepdims = false,
            -
            540 int ddof = 0,
            -
            541 StreamOrDevice s = {});
            -
            542
            -
            545array var(
            -
            546 const array& a,
            -
            547 int axis,
            -
            548 bool keepdims = false,
            -
            549 int ddof = 0,
            -
            550 StreamOrDevice s = {});
            -
            551
            -
            553array std(const array& a, bool keepdims, int ddof = 0, StreamOrDevice s = {});
            -
            -
            554inline array std(const array& a, StreamOrDevice s = {}) {
            -
            555 return std(a, false, 0, to_stream(s));
            -
            556}
            +
            537
            +
            540array var(
            +
            541 const array& a,
            +
            542 const std::vector<int>& axes,
            +
            543 bool keepdims = false,
            +
            544 int ddof = 0,
            +
            545 StreamOrDevice s = {});
            +
            546
            +
            549array var(
            +
            550 const array& a,
            +
            551 int axis,
            +
            552 bool keepdims = false,
            +
            553 int ddof = 0,
            +
            554 StreamOrDevice s = {});
            +
            555
            +
            557array std(const array& a, bool keepdims, int ddof = 0, StreamOrDevice s = {});
            +
            +
            558inline array std(const array& a, StreamOrDevice s = {}) {
            +
            559 return std(a, false, 0, to_stream(s));
            +
            560}
            -
            557
            -
            560array std(
            -
            561 const array& a,
            -
            562 const std::vector<int>& axes,
            -
            563 bool keepdims = false,
            -
            564 int ddof = 0,
            -
            565 StreamOrDevice s = {});
            -
            566
            -
            569array std(
            -
            570 const array& a,
            -
            571 int axis,
            -
            572 bool keepdims = false,
            -
            573 int ddof = 0,
            -
            574 StreamOrDevice s = {});
            -
            575
            -
            577array prod(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            578inline array prod(const array& a, StreamOrDevice s = {}) {
            -
            579 return prod(a, false, to_stream(s));
            -
            580}
            +
            561
            +
            564array std(
            +
            565 const array& a,
            +
            566 const std::vector<int>& axes,
            +
            567 bool keepdims = false,
            +
            568 int ddof = 0,
            +
            569 StreamOrDevice s = {});
            +
            570
            +
            573array std(
            +
            574 const array& a,
            +
            575 int axis,
            +
            576 bool keepdims = false,
            +
            577 int ddof = 0,
            +
            578 StreamOrDevice s = {});
            +
            579
            +
            581array prod(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            582inline array prod(const array& a, StreamOrDevice s = {}) {
            +
            583 return prod(a, false, to_stream(s));
            +
            584}
            -
            581
            -
            583array prod(
            -
            584 const array& a,
            -
            585 const std::vector<int>& axes,
            -
            586 bool keepdims = false,
            -
            587 StreamOrDevice s = {});
            -
            588
            -
            590array prod(
            -
            591 const array& a,
            -
            592 int axis,
            -
            593 bool keepdims = false,
            -
            594 StreamOrDevice s = {});
            -
            595
            -
            597array max(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            598inline array max(const array& a, StreamOrDevice s = {}) {
            -
            599 return max(a, false, to_stream(s));
            -
            600}
            +
            585
            +
            587array prod(
            +
            588 const array& a,
            +
            589 const std::vector<int>& axes,
            +
            590 bool keepdims = false,
            +
            591 StreamOrDevice s = {});
            +
            592
            +
            594array prod(
            +
            595 const array& a,
            +
            596 int axis,
            +
            597 bool keepdims = false,
            +
            598 StreamOrDevice s = {});
            +
            599
            +
            601array max(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            602inline array max(const array& a, StreamOrDevice s = {}) {
            +
            603 return max(a, false, to_stream(s));
            +
            604}
            -
            601
            -
            603array max(
            -
            604 const array& a,
            -
            605 const std::vector<int>& axes,
            -
            606 bool keepdims = false,
            -
            607 StreamOrDevice s = {});
            -
            608
            -
            610array max(
            -
            611 const array& a,
            -
            612 int axis,
            -
            613 bool keepdims = false,
            -
            614 StreamOrDevice s = {});
            -
            615
            -
            617array min(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            618inline array min(const array& a, StreamOrDevice s = {}) {
            -
            619 return min(a, false, to_stream(s));
            -
            620}
            +
            605
            +
            607array max(
            +
            608 const array& a,
            +
            609 const std::vector<int>& axes,
            +
            610 bool keepdims = false,
            +
            611 StreamOrDevice s = {});
            +
            612
            +
            614array max(
            +
            615 const array& a,
            +
            616 int axis,
            +
            617 bool keepdims = false,
            +
            618 StreamOrDevice s = {});
            +
            619
            +
            621array min(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            622inline array min(const array& a, StreamOrDevice s = {}) {
            +
            623 return min(a, false, to_stream(s));
            +
            624}
            -
            621
            -
            623array min(
            -
            624 const array& a,
            -
            625 const std::vector<int>& axes,
            -
            626 bool keepdims = false,
            -
            627 StreamOrDevice s = {});
            -
            628
            -
            630array min(
            -
            631 const array& a,
            -
            632 int axis,
            -
            633 bool keepdims = false,
            -
            634 StreamOrDevice s = {});
            -
            635
            -
            637array argmin(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            638inline array argmin(const array& a, StreamOrDevice s = {}) {
            -
            639 return argmin(a, false, s);
            -
            640}
            +
            625
            +
            627array min(
            +
            628 const array& a,
            +
            629 const std::vector<int>& axes,
            +
            630 bool keepdims = false,
            +
            631 StreamOrDevice s = {});
            +
            632
            +
            634array min(
            +
            635 const array& a,
            +
            636 int axis,
            +
            637 bool keepdims = false,
            +
            638 StreamOrDevice s = {});
            +
            639
            +
            641array argmin(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            642inline array argmin(const array& a, StreamOrDevice s = {}) {
            +
            643 return argmin(a, false, s);
            +
            644}
            -
            641
            -
            643array argmin(
            -
            644 const array& a,
            -
            645 int axis,
            -
            646 bool keepdims = false,
            -
            647 StreamOrDevice s = {});
            -
            648
            -
            650array argmax(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            651inline array argmax(const array& a, StreamOrDevice s = {}) {
            -
            652 return argmax(a, false, s);
            -
            653}
            +
            645
            +
            647array argmin(
            +
            648 const array& a,
            +
            649 int axis,
            +
            650 bool keepdims = false,
            +
            651 StreamOrDevice s = {});
            +
            652
            +
            654array argmax(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            655inline array argmax(const array& a, StreamOrDevice s = {}) {
            +
            656 return argmax(a, false, s);
            +
            657}
            -
            654
            -
            656array argmax(
            -
            657 const array& a,
            -
            658 int axis,
            -
            659 bool keepdims = false,
            -
            660 StreamOrDevice s = {});
            -
            661
            -
            663array sort(const array& a, StreamOrDevice s = {});
            -
            664
            -
            666array sort(const array& a, int axis, StreamOrDevice s = {});
            -
            667
            -
            669array argsort(const array& a, StreamOrDevice s = {});
            -
            670
            -
            672array argsort(const array& a, int axis, StreamOrDevice s = {});
            -
            673
            -
            678array partition(const array& a, int kth, StreamOrDevice s = {});
            -
            679
            -
            684array partition(const array& a, int kth, int axis, StreamOrDevice s = {});
            -
            685
            -
            690array argpartition(const array& a, int kth, StreamOrDevice s = {});
            -
            691
            -
            696array argpartition(const array& a, int kth, int axis, StreamOrDevice s = {});
            -
            697
            -
            699array topk(const array& a, int k, StreamOrDevice s = {});
            -
            700
            -
            702array topk(const array& a, int k, int axis, StreamOrDevice s = {});
            -
            703
            -
            705array logsumexp(const array& a, bool keepdims, StreamOrDevice s = {});
            -
            -
            706inline array logsumexp(const array& a, StreamOrDevice s = {}) {
            -
            707 return logsumexp(a, false, to_stream(s));
            -
            708}
            +
            658
            +
            660array argmax(
            +
            661 const array& a,
            +
            662 int axis,
            +
            663 bool keepdims = false,
            +
            664 StreamOrDevice s = {});
            +
            665
            +
            667array sort(const array& a, StreamOrDevice s = {});
            +
            668
            +
            670array sort(const array& a, int axis, StreamOrDevice s = {});
            +
            671
            +
            673array argsort(const array& a, StreamOrDevice s = {});
            +
            674
            +
            676array argsort(const array& a, int axis, StreamOrDevice s = {});
            +
            677
            +
            682array partition(const array& a, int kth, StreamOrDevice s = {});
            +
            683
            +
            688array partition(const array& a, int kth, int axis, StreamOrDevice s = {});
            +
            689
            +
            694array argpartition(const array& a, int kth, StreamOrDevice s = {});
            +
            695
            +
            700array argpartition(const array& a, int kth, int axis, StreamOrDevice s = {});
            +
            701
            +
            703array topk(const array& a, int k, StreamOrDevice s = {});
            +
            704
            +
            706array topk(const array& a, int k, int axis, StreamOrDevice s = {});
            +
            707
            +
            709array logsumexp(const array& a, bool keepdims, StreamOrDevice s = {});
            +
            +
            710inline array logsumexp(const array& a, StreamOrDevice s = {}) {
            +
            711 return logsumexp(a, false, to_stream(s));
            +
            712}
            -
            709
            -
            711array logsumexp(
            -
            712 const array& a,
            -
            713 const std::vector<int>& axes,
            -
            714 bool keepdims = false,
            -
            715 StreamOrDevice s = {});
            -
            716
            -
            718array logsumexp(
            -
            719 const array& a,
            -
            720 int axis,
            -
            721 bool keepdims = false,
            -
            722 StreamOrDevice s = {});
            -
            723
            -
            725array abs(const array& a, StreamOrDevice s = {});
            -
            726
            -
            728array negative(const array& a, StreamOrDevice s = {});
            -
            729array operator-(const array& a);
            +
            713
            +
            715array logsumexp(
            +
            716 const array& a,
            +
            717 const std::vector<int>& axes,
            +
            718 bool keepdims = false,
            +
            719 StreamOrDevice s = {});
            +
            720
            +
            722array logsumexp(
            +
            723 const array& a,
            +
            724 int axis,
            +
            725 bool keepdims = false,
            +
            726 StreamOrDevice s = {});
            +
            727
            +
            729array abs(const array& a, StreamOrDevice s = {});
            730
            -
            732array sign(const array& a, StreamOrDevice s = {});
            -
            733
            -
            735array logical_not(const array& a, StreamOrDevice s = {});
            -
            736
            -
            738array logical_and(const array& a, const array& b, StreamOrDevice s = {});
            -
            739array operator&&(const array& a, const array& b);
            +
            732array negative(const array& a, StreamOrDevice s = {});
            +
            733array operator-(const array& a);
            +
            734
            +
            736array sign(const array& a, StreamOrDevice s = {});
            +
            737
            +
            739array logical_not(const array& a, StreamOrDevice s = {});
            740
            -
            742array logical_or(const array& a, const array& b, StreamOrDevice s = {});
            -
            743array operator||(const array& a, const array& b);
            +
            742array logical_and(const array& a, const array& b, StreamOrDevice s = {});
            +
            743array operator&&(const array& a, const array& b);
            744
            -
            746array reciprocal(const array& a, StreamOrDevice s = {});
            -
            747
            -
            749array add(const array& a, const array& b, StreamOrDevice s = {});
            -
            750array operator+(const array& a, const array& b);
            -
            751template <typename T>
            -
            -
            752array operator+(T a, const array& b) {
            -
            753 return add(array(a), b);
            -
            754}
            -
            +
            746array logical_or(const array& a, const array& b, StreamOrDevice s = {});
            +
            747array operator||(const array& a, const array& b);
            +
            748
            +
            750array reciprocal(const array& a, StreamOrDevice s = {});
            +
            751
            +
            753array add(const array& a, const array& b, StreamOrDevice s = {});
            +
            754array operator+(const array& a, const array& b);
            755template <typename T>
            -
            756array operator+(const array& a, T b) {
            -
            757 return add(a, array(b));
            +
            756array operator+(T a, const array& b) {
            +
            757 return add(array(a), b);
            758}
            -
            759
            -
            761array subtract(const array& a, const array& b, StreamOrDevice s = {});
            -
            762array operator-(const array& a, const array& b);
            -
            763template <typename T>
            -
            -
            764array operator-(T a, const array& b) {
            -
            765 return subtract(array(a), b);
            -
            766}
            +
            759template <typename T>
            +
            +
            760array operator+(const array& a, T b) {
            +
            761 return add(a, array(b));
            +
            762}
            +
            763
            +
            765array subtract(const array& a, const array& b, StreamOrDevice s = {});
            +
            766array operator-(const array& a, const array& b);
            767template <typename T>
            -
            768array operator-(const array& a, T b) {
            -
            769 return subtract(a, array(b));
            +
            768array operator-(T a, const array& b) {
            +
            769 return subtract(array(a), b);
            770}
            -
            771
            -
            773array multiply(const array& a, const array& b, StreamOrDevice s = {});
            -
            774array operator*(const array& a, const array& b);
            -
            775template <typename T>
            -
            -
            776array operator*(T a, const array& b) {
            -
            777 return multiply(array(a), b);
            -
            778}
            +
            771template <typename T>
            +
            +
            772array operator-(const array& a, T b) {
            +
            773 return subtract(a, array(b));
            +
            774}
            +
            775
            +
            777array multiply(const array& a, const array& b, StreamOrDevice s = {});
            +
            778array operator*(const array& a, const array& b);
            779template <typename T>
            -
            780array operator*(const array& a, T b) {
            -
            781 return multiply(a, array(b));
            +
            780array operator*(T a, const array& b) {
            +
            781 return multiply(array(a), b);
            782}
            -
            783
            -
            785array divide(const array& a, const array& b, StreamOrDevice s = {});
            -
            786array operator/(const array& a, const array& b);
            -
            787array operator/(double a, const array& b);
            -
            788array operator/(const array& a, double b);
            -
            789
            -
            791std::vector<array>
            -
            792divmod(const array& a, const array& b, StreamOrDevice s = {});
            -
            793
            -
            795array floor_divide(const array& a, const array& b, StreamOrDevice s = {});
            -
            796
            -
            798array remainder(const array& a, const array& b, StreamOrDevice s = {});
            -
            799array operator%(const array& a, const array& b);
            -
            800template <typename T>
            -
            -
            801array operator%(T a, const array& b) {
            -
            802 return remainder(array(a), b);
            -
            803}
            +
            783template <typename T>
            +
            +
            784array operator*(const array& a, T b) {
            +
            785 return multiply(a, array(b));
            +
            786}
            +
            787
            +
            789array divide(const array& a, const array& b, StreamOrDevice s = {});
            +
            790array operator/(const array& a, const array& b);
            +
            791array operator/(double a, const array& b);
            +
            792array operator/(const array& a, double b);
            +
            793
            +
            795std::vector<array>
            +
            796divmod(const array& a, const array& b, StreamOrDevice s = {});
            +
            797
            +
            799array floor_divide(const array& a, const array& b, StreamOrDevice s = {});
            +
            800
            +
            802array remainder(const array& a, const array& b, StreamOrDevice s = {});
            +
            803array operator%(const array& a, const array& b);
            804template <typename T>
            -
            805array operator%(const array& a, T b) {
            -
            806 return remainder(a, array(b));
            +
            805array operator%(T a, const array& b) {
            +
            806 return remainder(array(a), b);
            807}
            -
            808
            -
            810array maximum(const array& a, const array& b, StreamOrDevice s = {});
            -
            811
            -
            813array minimum(const array& a, const array& b, StreamOrDevice s = {});
            -
            814
            -
            816array floor(const array& a, StreamOrDevice s = {});
            -
            817
            -
            819array ceil(const array& a, StreamOrDevice s = {});
            -
            820
            -
            822array square(const array& a, StreamOrDevice s = {});
            -
            823
            -
            825array exp(const array& a, StreamOrDevice s = {});
            -
            826
            -
            828array sin(const array& a, StreamOrDevice s = {});
            -
            829
            -
            831array cos(const array& a, StreamOrDevice s = {});
            -
            832
            -
            834array tan(const array& a, StreamOrDevice s = {});
            -
            835
            -
            837array arcsin(const array& a, StreamOrDevice s = {});
            -
            838
            -
            840array arccos(const array& a, StreamOrDevice s = {});
            -
            841
            -
            843array arctan(const array& a, StreamOrDevice s = {});
            -
            844
            -
            846array arctan2(const array& a, const array& b, StreamOrDevice s = {});
            -
            847
            -
            849array sinh(const array& a, StreamOrDevice s = {});
            -
            850
            -
            852array cosh(const array& a, StreamOrDevice s = {});
            -
            853
            -
            855array tanh(const array& a, StreamOrDevice s = {});
            -
            856
            -
            858array arcsinh(const array& a, StreamOrDevice s = {});
            -
            859
            -
            861array arccosh(const array& a, StreamOrDevice s = {});
            -
            862
            -
            864array arctanh(const array& a, StreamOrDevice s = {});
            -
            865
            -
            867array degrees(const array& a, StreamOrDevice s = {});
            -
            868
            -
            870array radians(const array& a, StreamOrDevice s = {});
            -
            871
            -
            873array log(const array& a, StreamOrDevice s = {});
            -
            874
            -
            876array log2(const array& a, StreamOrDevice s = {});
            -
            877
            -
            879array log10(const array& a, StreamOrDevice s = {});
            -
            880
            -
            882array log1p(const array& a, StreamOrDevice s = {});
            -
            883
            -
            885array logaddexp(const array& a, const array& b, StreamOrDevice s = {});
            -
            886
            -
            888array sigmoid(const array& a, StreamOrDevice s = {});
            -
            889
            -
            891array erf(const array& a, StreamOrDevice s = {});
            -
            892
            -
            894array erfinv(const array& a, StreamOrDevice s = {});
            -
            895
            -
            897array expm1(const array& a, StreamOrDevice s = {});
            -
            898
            -
            900array stop_gradient(const array& a, StreamOrDevice s = {});
            -
            901
            -
            903array round(const array& a, int decimals, StreamOrDevice s = {});
            -
            -
            904inline array round(const array& a, StreamOrDevice s = {}) {
            -
            905 return round(a, 0, s);
            -
            906}
            +
            808template <typename T>
            +
            +
            809array operator%(const array& a, T b) {
            +
            810 return remainder(a, array(b));
            +
            811}
            -
            907
            -
            909array matmul(const array& a, const array& b, StreamOrDevice s = {});
            -
            910
            -
            912array gather(
            -
            913 const array& a,
            -
            914 const std::vector<array>& indices,
            -
            915 const std::vector<int>& axes,
            -
            916 const std::vector<int>& slice_sizes,
            -
            917 StreamOrDevice s = {});
            -
            -
            918inline array gather(
            -
            919 const array& a,
            -
            920 const array& indices,
            -
            921 int axis,
            -
            922 const std::vector<int>& slice_sizes,
            -
            923 StreamOrDevice s = {}) {
            -
            924 return gather(a, {indices}, std::vector<int>{axis}, slice_sizes, s);
            -
            925}
            +
            812
            +
            814array maximum(const array& a, const array& b, StreamOrDevice s = {});
            +
            815
            +
            817array minimum(const array& a, const array& b, StreamOrDevice s = {});
            +
            818
            +
            820array floor(const array& a, StreamOrDevice s = {});
            +
            821
            +
            823array ceil(const array& a, StreamOrDevice s = {});
            +
            824
            +
            826array square(const array& a, StreamOrDevice s = {});
            +
            827
            +
            829array exp(const array& a, StreamOrDevice s = {});
            +
            830
            +
            832array sin(const array& a, StreamOrDevice s = {});
            +
            833
            +
            835array cos(const array& a, StreamOrDevice s = {});
            +
            836
            +
            838array tan(const array& a, StreamOrDevice s = {});
            +
            839
            +
            841array arcsin(const array& a, StreamOrDevice s = {});
            +
            842
            +
            844array arccos(const array& a, StreamOrDevice s = {});
            +
            845
            +
            847array arctan(const array& a, StreamOrDevice s = {});
            +
            848
            +
            850array arctan2(const array& a, const array& b, StreamOrDevice s = {});
            +
            851
            +
            853array sinh(const array& a, StreamOrDevice s = {});
            +
            854
            +
            856array cosh(const array& a, StreamOrDevice s = {});
            +
            857
            +
            859array tanh(const array& a, StreamOrDevice s = {});
            +
            860
            +
            862array arcsinh(const array& a, StreamOrDevice s = {});
            +
            863
            +
            865array arccosh(const array& a, StreamOrDevice s = {});
            +
            866
            +
            868array arctanh(const array& a, StreamOrDevice s = {});
            +
            869
            +
            871array degrees(const array& a, StreamOrDevice s = {});
            +
            872
            +
            874array radians(const array& a, StreamOrDevice s = {});
            +
            875
            +
            877array log(const array& a, StreamOrDevice s = {});
            +
            878
            +
            880array log2(const array& a, StreamOrDevice s = {});
            +
            881
            +
            883array log10(const array& a, StreamOrDevice s = {});
            +
            884
            +
            886array log1p(const array& a, StreamOrDevice s = {});
            +
            887
            +
            889array logaddexp(const array& a, const array& b, StreamOrDevice s = {});
            +
            890
            +
            892array sigmoid(const array& a, StreamOrDevice s = {});
            +
            893
            +
            895array erf(const array& a, StreamOrDevice s = {});
            +
            896
            +
            898array erfinv(const array& a, StreamOrDevice s = {});
            +
            899
            +
            901array expm1(const array& a, StreamOrDevice s = {});
            +
            902
            +
            904array stop_gradient(const array& a, StreamOrDevice s = {});
            +
            905
            +
            907array round(const array& a, int decimals, StreamOrDevice s = {});
            +
            +
            908inline array round(const array& a, StreamOrDevice s = {}) {
            +
            909 return round(a, 0, s);
            +
            910}
            -
            926
            -
            928array take(
            -
            929 const array& a,
            -
            930 const array& indices,
            -
            931 int axis,
            -
            932 StreamOrDevice s = {});
            -
            933
            -
            935array take(const array& a, const array& indices, StreamOrDevice s = {});
            -
            936
            -
            938array take_along_axis(
            -
            939 const array& a,
            -
            940 const array& indices,
            -
            941 int axis,
            -
            942 StreamOrDevice s = {});
            -
            943
            -
            1042array scatter(
            -
            1043 const array& a,
            -
            1044 const std::vector<array>& indices,
            -
            1045 const array& updates,
            -
            1046 const std::vector<int>& axes,
            -
            1047 StreamOrDevice s = {});
            -
            -
            1048inline array scatter(
            -
            1049 const array& a,
            -
            1050 const array& indices,
            -
            1051 const array& updates,
            -
            1052 int axis,
            -
            1053 StreamOrDevice s = {}) {
            -
            1054 return scatter(a, {indices}, updates, std::vector<int>{axis}, s);
            -
            1055}
            +
            911
            +
            913array matmul(const array& a, const array& b, StreamOrDevice s = {});
            +
            914
            +
            916array gather(
            +
            917 const array& a,
            +
            918 const std::vector<array>& indices,
            +
            919 const std::vector<int>& axes,
            +
            920 const std::vector<int>& slice_sizes,
            +
            921 StreamOrDevice s = {});
            +
            +
            922inline array gather(
            +
            923 const array& a,
            +
            924 const array& indices,
            +
            925 int axis,
            +
            926 const std::vector<int>& slice_sizes,
            +
            927 StreamOrDevice s = {}) {
            +
            928 return gather(a, {indices}, std::vector<int>{axis}, slice_sizes, s);
            +
            929}
            -
            1056
            -
            1058array scatter_add(
            -
            1059 const array& a,
            -
            1060 const std::vector<array>& indices,
            -
            1061 const array& updates,
            -
            1062 const std::vector<int>& axes,
            -
            1063 StreamOrDevice s = {});
            -
            -
            1064inline array scatter_add(
            -
            1065 const array& a,
            -
            1066 const array& indices,
            -
            1067 const array& updates,
            -
            1068 int axis,
            -
            1069 StreamOrDevice s = {}) {
            -
            1070 return scatter_add(a, {indices}, updates, std::vector<int>{axis}, s);
            -
            1071}
            +
            930
            +
            932array take(
            +
            933 const array& a,
            +
            934 const array& indices,
            +
            935 int axis,
            +
            936 StreamOrDevice s = {});
            +
            937
            +
            939array take(const array& a, const array& indices, StreamOrDevice s = {});
            +
            940
            +
            942array take_along_axis(
            +
            943 const array& a,
            +
            944 const array& indices,
            +
            945 int axis,
            +
            946 StreamOrDevice s = {});
            +
            947
            +
            1046array scatter(
            +
            1047 const array& a,
            +
            1048 const std::vector<array>& indices,
            +
            1049 const array& updates,
            +
            1050 const std::vector<int>& axes,
            +
            1051 StreamOrDevice s = {});
            +
            +
            1052inline array scatter(
            +
            1053 const array& a,
            +
            1054 const array& indices,
            +
            1055 const array& updates,
            +
            1056 int axis,
            +
            1057 StreamOrDevice s = {}) {
            +
            1058 return scatter(a, {indices}, updates, std::vector<int>{axis}, s);
            +
            1059}
            -
            1072
            -
            1074array scatter_prod(
            -
            1075 const array& a,
            -
            1076 const std::vector<array>& indices,
            -
            1077 const array& updates,
            -
            1078 const std::vector<int>& axes,
            -
            1079 StreamOrDevice s = {});
            -
            -
            1080inline array scatter_prod(
            -
            1081 const array& a,
            -
            1082 const array& indices,
            -
            1083 const array& updates,
            -
            1084 int axis,
            -
            1085 StreamOrDevice s = {}) {
            -
            1086 return scatter_prod(a, {indices}, updates, std::vector<int>{axis}, s);
            -
            1087}
            +
            1060
            +
            1062array scatter_add(
            +
            1063 const array& a,
            +
            1064 const std::vector<array>& indices,
            +
            1065 const array& updates,
            +
            1066 const std::vector<int>& axes,
            +
            1067 StreamOrDevice s = {});
            +
            +
            1068inline array scatter_add(
            +
            1069 const array& a,
            +
            1070 const array& indices,
            +
            1071 const array& updates,
            +
            1072 int axis,
            +
            1073 StreamOrDevice s = {}) {
            +
            1074 return scatter_add(a, {indices}, updates, std::vector<int>{axis}, s);
            +
            1075}
            -
            1088
            -
            1090array scatter_max(
            -
            1091 const array& a,
            -
            1092 const std::vector<array>& indices,
            -
            1093 const array& updates,
            -
            1094 const std::vector<int>& axes,
            -
            1095 StreamOrDevice s = {});
            -
            -
            1096inline array scatter_max(
            -
            1097 const array& a,
            -
            1098 const array& indices,
            -
            1099 const array& updates,
            -
            1100 int axis,
            -
            1101 StreamOrDevice s = {}) {
            -
            1102 return scatter_max(a, {indices}, updates, std::vector<int>{axis}, s);
            -
            1103}
            +
            1076
            +
            1078array scatter_prod(
            +
            1079 const array& a,
            +
            1080 const std::vector<array>& indices,
            +
            1081 const array& updates,
            +
            1082 const std::vector<int>& axes,
            +
            1083 StreamOrDevice s = {});
            +
            +
            1084inline array scatter_prod(
            +
            1085 const array& a,
            +
            1086 const array& indices,
            +
            1087 const array& updates,
            +
            1088 int axis,
            +
            1089 StreamOrDevice s = {}) {
            +
            1090 return scatter_prod(a, {indices}, updates, std::vector<int>{axis}, s);
            +
            1091}
            -
            1105array scatter_min(
            -
            1106 const array& a,
            -
            1107 const std::vector<array>& indices,
            -
            1108 const array& updates,
            -
            1109 const std::vector<int>& axes,
            -
            1110 StreamOrDevice s = {});
            -
            -
            1111inline array scatter_min(
            -
            1112 const array& a,
            -
            1113 const array& indices,
            -
            1114 const array& updates,
            -
            1115 int axis,
            -
            1116 StreamOrDevice s = {}) {
            -
            1117 return scatter_min(a, {indices}, updates, std::vector<int>{axis}, s);
            -
            1118}
            +
            1092
            +
            1094array scatter_max(
            +
            1095 const array& a,
            +
            1096 const std::vector<array>& indices,
            +
            1097 const array& updates,
            +
            1098 const std::vector<int>& axes,
            +
            1099 StreamOrDevice s = {});
            +
            +
            1100inline array scatter_max(
            +
            1101 const array& a,
            +
            1102 const array& indices,
            +
            1103 const array& updates,
            +
            1104 int axis,
            +
            1105 StreamOrDevice s = {}) {
            +
            1106 return scatter_max(a, {indices}, updates, std::vector<int>{axis}, s);
            +
            1107}
            -
            1119
            -
            1121array sqrt(const array& a, StreamOrDevice s = {});
            -
            1122
            -
            1124array rsqrt(const array& a, StreamOrDevice s = {});
            -
            1125
            -
            1127array softmax(
            -
            1128 const array& a,
            -
            1129 const std::vector<int>& axes,
            -
            1130 bool precise = false,
            -
            1131 StreamOrDevice s = {});
            -
            1132
            -
            1134array softmax(const array& a, bool precise = false, StreamOrDevice s = {});
            -
            1135
            -
            1137inline array
            -
            -
            1138softmax(const array& a, int axis, bool precise = false, StreamOrDevice s = {}) {
            -
            1139 return softmax(a, std::vector<int>{axis}, precise, s);
            -
            1140}
            +
            1109array scatter_min(
            +
            1110 const array& a,
            +
            1111 const std::vector<array>& indices,
            +
            1112 const array& updates,
            +
            1113 const std::vector<int>& axes,
            +
            1114 StreamOrDevice s = {});
            +
            +
            1115inline array scatter_min(
            +
            1116 const array& a,
            +
            1117 const array& indices,
            +
            1118 const array& updates,
            +
            1119 int axis,
            +
            1120 StreamOrDevice s = {}) {
            +
            1121 return scatter_min(a, {indices}, updates, std::vector<int>{axis}, s);
            +
            1122}
            -
            1141
            -
            1143array power(const array& a, const array& b, StreamOrDevice s = {});
            -
            1144
            -
            1146array cumsum(
            -
            1147 const array& a,
            -
            1148 int axis,
            -
            1149 bool reverse = false,
            -
            1150 bool inclusive = true,
            -
            1151 StreamOrDevice s = {});
            -
            1152
            -
            1154array cumprod(
            -
            1155 const array& a,
            -
            1156 int axis,
            -
            1157 bool reverse = false,
            -
            1158 bool inclusive = true,
            -
            1159 StreamOrDevice s = {});
            -
            1160
            -
            1162array cummax(
            -
            1163 const array& a,
            -
            1164 int axis,
            -
            1165 bool reverse = false,
            -
            1166 bool inclusive = true,
            -
            1167 StreamOrDevice s = {});
            -
            1168
            -
            1170array cummin(
            -
            1171 const array& a,
            -
            1172 int axis,
            -
            1173 bool reverse = false,
            -
            1174 bool inclusive = true,
            -
            1175 StreamOrDevice s = {});
            -
            1176
            -
            1178array conv_general(
            -
            1179 array input,
            -
            1180 array weight,
            -
            1181 std::vector<int> stride = {},
            -
            1182 std::vector<int> padding_lo = {},
            -
            1183 std::vector<int> padding_hi = {},
            -
            1184 std::vector<int> kernel_dilation = {},
            -
            1185 std::vector<int> input_dilation = {},
            -
            1186 int groups = 1,
            -
            1187 bool flip = false,
            -
            1188 StreamOrDevice s = {});
            -
            1189
            -
            -
            1191inline array conv_general(
            -
            1192 const array& input,
            -
            1193 const array& weight,
            -
            1194 std::vector<int> stride = {},
            -
            1195 std::vector<int> padding = {},
            -
            1196 std::vector<int> kernel_dilation = {},
            -
            1197 std::vector<int> input_dilation = {},
            -
            1198 int groups = 1,
            -
            1199 bool flip = false,
            -
            1200 StreamOrDevice s = {}) {
            -
            1201 return conv_general(
            -
            1202 /* const array& input = */ input,
            -
            1203 /* const array& weight = */ weight,
            -
            1204 /* std::vector<int> stride = */ stride,
            -
            1205 /* std::vector<int> padding_lo = */ padding,
            -
            1206 /* std::vector<int> padding_hi = */ padding,
            -
            1207 /* std::vector<int> kernel_dilation = */ kernel_dilation,
            -
            1208 /* std::vector<int> input_dilation = */ input_dilation,
            -
            1209 /* int groups = */ groups,
            -
            1210 /* bool flip = */ flip,
            -
            1211 /* StreamOrDevice s = */ s);
            -
            1212}
            +
            1123
            +
            1125array sqrt(const array& a, StreamOrDevice s = {});
            +
            1126
            +
            1128array rsqrt(const array& a, StreamOrDevice s = {});
            +
            1129
            +
            1131array softmax(
            +
            1132 const array& a,
            +
            1133 const std::vector<int>& axes,
            +
            1134 bool precise = false,
            +
            1135 StreamOrDevice s = {});
            +
            1136
            +
            1138array softmax(const array& a, bool precise = false, StreamOrDevice s = {});
            +
            1139
            +
            1141inline array
            +
            +
            1142softmax(const array& a, int axis, bool precise = false, StreamOrDevice s = {}) {
            +
            1143 return softmax(a, std::vector<int>{axis}, precise, s);
            +
            1144}
            -
            1213
            -
            1215array conv1d(
            -
            1216 const array& input,
            -
            1217 const array& weight,
            -
            1218 int stride = 1,
            -
            1219 int padding = 0,
            -
            1220 int dilation = 1,
            -
            1221 int groups = 1,
            -
            1222 StreamOrDevice s = {});
            -
            1223
            -
            1225array conv2d(
            -
            1226 const array& input,
            -
            1227 const array& weight,
            -
            1228 const std::pair<int, int>& stride = {1, 1},
            -
            1229 const std::pair<int, int>& padding = {0, 0},
            -
            1230 const std::pair<int, int>& dilation = {1, 1},
            -
            1231 int groups = 1,
            -
            1232 StreamOrDevice s = {});
            -
            1233
            -
            1235array conv3d(
            -
            1236 const array& input,
            -
            1237 const array& weight,
            -
            1238 const std::tuple<int, int, int>& stride = {1, 1, 1},
            -
            1239 const std::tuple<int, int, int>& padding = {0, 0, 0},
            -
            1240 const std::tuple<int, int, int>& dilation = {1, 1, 1},
            -
            1241 int groups = 1,
            -
            1242 StreamOrDevice s = {});
            -
            1243
            -
            1245array quantized_matmul(
            -
            1246 const array& x,
            -
            1247 const array& w,
            -
            1248 const array& scales,
            -
            1249 const array& biases,
            -
            1250 bool transpose = true,
            -
            1251 int group_size = 64,
            -
            1252 int bits = 4,
            -
            1253 StreamOrDevice s = {});
            -
            1254
            -
            1256std::tuple<array, array, array> quantize(
            -
            1257 const array& w,
            -
            1258 int group_size = 64,
            -
            1259 int bits = 4,
            -
            1260 StreamOrDevice s = {});
            -
            1261
            -
            1263array dequantize(
            -
            1264 const array& w,
            -
            1265 const array& scales,
            -
            1266 const array& biases,
            -
            1267 int group_size = 64,
            -
            1268 int bits = 4,
            -
            1269 StreamOrDevice s = {});
            -
            1270
            -
            1272array gather_qmm(
            -
            1273 const array& x,
            -
            1274 const array& w,
            -
            1275 const array& scales,
            -
            1276 const array& biases,
            -
            1277 std::optional<array> lhs_indices = std::nullopt,
            -
            1278 std::optional<array> rhs_indices = std::nullopt,
            -
            1279 bool transpose = true,
            -
            1280 int group_size = 64,
            -
            1281 int bits = 4,
            -
            1282 StreamOrDevice s = {});
            -
            1283
            -
            1285array tensordot(
            -
            1286 const array& a,
            -
            1287 const array& b,
            -
            1288 const int axis = 2,
            -
            1289 StreamOrDevice s = {});
            -
            1290
            -
            1291array tensordot(
            -
            1292 const array& a,
            -
            1293 const array& b,
            -
            1294 const std::vector<int>& axes_a,
            -
            1295 const std::vector<int>& axes_b,
            -
            1296 StreamOrDevice s = {});
            -
            1297
            -
            1299array outer(const array& a, const array& b, StreamOrDevice s = {});
            -
            1300
            -
            1302array inner(const array& a, const array& b, StreamOrDevice s = {});
            -
            1303
            -
            1305array addmm(
            -
            1306 array c,
            -
            1307 array a,
            -
            1308 array b,
            -
            1309 const float& alpha = 1.f,
            -
            1310 const float& beta = 1.f,
            -
            1311 StreamOrDevice s = {});
            -
            1312
            -
            1314array block_masked_mm(
            -
            1315 array a,
            -
            1316 array b,
            -
            1317 int block_size,
            -
            1318 std::optional<array> mask_out = std::nullopt,
            -
            1319 std::optional<array> mask_lhs = std::nullopt,
            -
            1320 std::optional<array> mask_rhs = std::nullopt,
            -
            1321 StreamOrDevice s = {});
            -
            1322
            -
            1324array gather_mm(
            -
            1325 array a,
            -
            1326 array b,
            -
            1327 std::optional<array> lhs_indices = std::nullopt,
            -
            1328 std::optional<array> rhs_indices = std::nullopt,
            -
            1329 StreamOrDevice s = {});
            -
            1330
            -
            1332array diagonal(
            -
            1333 const array& a,
            -
            1334 int offset = 0,
            -
            1335 int axis1 = 0,
            -
            1336 int axis2 = 1,
            -
            1337 StreamOrDevice s = {});
            -
            1338
            -
            1340array diag(const array& a, int k = 0, StreamOrDevice s = {});
            -
            1341
            -
            1343array trace(
            -
            1344 const array& a,
            -
            1345 int offset,
            -
            1346 int axis1,
            -
            1347 int axis2,
            -
            1348 Dtype dtype,
            -
            1349 StreamOrDevice s = {});
            -
            1350array trace(
            -
            1351 const array& a,
            -
            1352 int offset,
            -
            1353 int axis1,
            -
            1354 int axis2,
            -
            1355 StreamOrDevice s = {});
            -
            1356array trace(const array& a, StreamOrDevice s = {});
            -
            1357
            -
            1363std::vector<array> depends(
            -
            1364 const std::vector<array>& inputs,
            -
            1365 const std::vector<array>& dependencies);
            -
            1366
            -
            1368array atleast_1d(const array& a, StreamOrDevice s = {});
            -
            1369std::vector<array> atleast_1d(
            -
            1370 const std::vector<array>& a,
            -
            1371 StreamOrDevice s = {});
            -
            1372array atleast_2d(const array& a, StreamOrDevice s = {});
            -
            1373std::vector<array> atleast_2d(
            +
            1145
            +
            1147array power(const array& a, const array& b, StreamOrDevice s = {});
            +
            1148
            +
            1150array cumsum(
            +
            1151 const array& a,
            +
            1152 int axis,
            +
            1153 bool reverse = false,
            +
            1154 bool inclusive = true,
            +
            1155 StreamOrDevice s = {});
            +
            1156
            +
            1158array cumprod(
            +
            1159 const array& a,
            +
            1160 int axis,
            +
            1161 bool reverse = false,
            +
            1162 bool inclusive = true,
            +
            1163 StreamOrDevice s = {});
            +
            1164
            +
            1166array cummax(
            +
            1167 const array& a,
            +
            1168 int axis,
            +
            1169 bool reverse = false,
            +
            1170 bool inclusive = true,
            +
            1171 StreamOrDevice s = {});
            +
            1172
            +
            1174array cummin(
            +
            1175 const array& a,
            +
            1176 int axis,
            +
            1177 bool reverse = false,
            +
            1178 bool inclusive = true,
            +
            1179 StreamOrDevice s = {});
            +
            1180
            +
            1182array conv_general(
            +
            1183 array input,
            +
            1184 array weight,
            +
            1185 std::vector<int> stride = {},
            +
            1186 std::vector<int> padding_lo = {},
            +
            1187 std::vector<int> padding_hi = {},
            +
            1188 std::vector<int> kernel_dilation = {},
            +
            1189 std::vector<int> input_dilation = {},
            +
            1190 int groups = 1,
            +
            1191 bool flip = false,
            +
            1192 StreamOrDevice s = {});
            +
            1193
            +
            +
            1195inline array conv_general(
            +
            1196 const array& input,
            +
            1197 const array& weight,
            +
            1198 std::vector<int> stride = {},
            +
            1199 std::vector<int> padding = {},
            +
            1200 std::vector<int> kernel_dilation = {},
            +
            1201 std::vector<int> input_dilation = {},
            +
            1202 int groups = 1,
            +
            1203 bool flip = false,
            +
            1204 StreamOrDevice s = {}) {
            +
            1205 return conv_general(
            +
            1206 /* const array& input = */ input,
            +
            1207 /* const array& weight = */ weight,
            +
            1208 /* std::vector<int> stride = */ stride,
            +
            1209 /* std::vector<int> padding_lo = */ padding,
            +
            1210 /* std::vector<int> padding_hi = */ padding,
            +
            1211 /* std::vector<int> kernel_dilation = */ kernel_dilation,
            +
            1212 /* std::vector<int> input_dilation = */ input_dilation,
            +
            1213 /* int groups = */ groups,
            +
            1214 /* bool flip = */ flip,
            +
            1215 /* StreamOrDevice s = */ s);
            +
            1216}
            +
            +
            1217
            +
            1219array conv1d(
            +
            1220 const array& input,
            +
            1221 const array& weight,
            +
            1222 int stride = 1,
            +
            1223 int padding = 0,
            +
            1224 int dilation = 1,
            +
            1225 int groups = 1,
            +
            1226 StreamOrDevice s = {});
            +
            1227
            +
            1229array conv2d(
            +
            1230 const array& input,
            +
            1231 const array& weight,
            +
            1232 const std::pair<int, int>& stride = {1, 1},
            +
            1233 const std::pair<int, int>& padding = {0, 0},
            +
            1234 const std::pair<int, int>& dilation = {1, 1},
            +
            1235 int groups = 1,
            +
            1236 StreamOrDevice s = {});
            +
            1237
            +
            1239array conv3d(
            +
            1240 const array& input,
            +
            1241 const array& weight,
            +
            1242 const std::tuple<int, int, int>& stride = {1, 1, 1},
            +
            1243 const std::tuple<int, int, int>& padding = {0, 0, 0},
            +
            1244 const std::tuple<int, int, int>& dilation = {1, 1, 1},
            +
            1245 int groups = 1,
            +
            1246 StreamOrDevice s = {});
            +
            1247
            +
            1249array quantized_matmul(
            +
            1250 const array& x,
            +
            1251 const array& w,
            +
            1252 const array& scales,
            +
            1253 const array& biases,
            +
            1254 bool transpose = true,
            +
            1255 int group_size = 64,
            +
            1256 int bits = 4,
            +
            1257 StreamOrDevice s = {});
            +
            1258
            +
            1260std::tuple<array, array, array> quantize(
            +
            1261 const array& w,
            +
            1262 int group_size = 64,
            +
            1263 int bits = 4,
            +
            1264 StreamOrDevice s = {});
            +
            1265
            +
            1267array dequantize(
            +
            1268 const array& w,
            +
            1269 const array& scales,
            +
            1270 const array& biases,
            +
            1271 int group_size = 64,
            +
            1272 int bits = 4,
            +
            1273 StreamOrDevice s = {});
            +
            1274
            +
            1276array gather_qmm(
            +
            1277 const array& x,
            +
            1278 const array& w,
            +
            1279 const array& scales,
            +
            1280 const array& biases,
            +
            1281 std::optional<array> lhs_indices = std::nullopt,
            +
            1282 std::optional<array> rhs_indices = std::nullopt,
            +
            1283 bool transpose = true,
            +
            1284 int group_size = 64,
            +
            1285 int bits = 4,
            +
            1286 StreamOrDevice s = {});
            +
            1287
            +
            1289array tensordot(
            +
            1290 const array& a,
            +
            1291 const array& b,
            +
            1292 const int axis = 2,
            +
            1293 StreamOrDevice s = {});
            +
            1294
            +
            1295array tensordot(
            +
            1296 const array& a,
            +
            1297 const array& b,
            +
            1298 const std::vector<int>& axes_a,
            +
            1299 const std::vector<int>& axes_b,
            +
            1300 StreamOrDevice s = {});
            +
            1301
            +
            1303array outer(const array& a, const array& b, StreamOrDevice s = {});
            +
            1304
            +
            1306array inner(const array& a, const array& b, StreamOrDevice s = {});
            +
            1307
            +
            1309array addmm(
            +
            1310 array c,
            +
            1311 array a,
            +
            1312 array b,
            +
            1313 const float& alpha = 1.f,
            +
            1314 const float& beta = 1.f,
            +
            1315 StreamOrDevice s = {});
            +
            1316
            +
            1318array block_masked_mm(
            +
            1319 array a,
            +
            1320 array b,
            +
            1321 int block_size,
            +
            1322 std::optional<array> mask_out = std::nullopt,
            +
            1323 std::optional<array> mask_lhs = std::nullopt,
            +
            1324 std::optional<array> mask_rhs = std::nullopt,
            +
            1325 StreamOrDevice s = {});
            +
            1326
            +
            1328array gather_mm(
            +
            1329 array a,
            +
            1330 array b,
            +
            1331 std::optional<array> lhs_indices = std::nullopt,
            +
            1332 std::optional<array> rhs_indices = std::nullopt,
            +
            1333 StreamOrDevice s = {});
            +
            1334
            +
            1336array diagonal(
            +
            1337 const array& a,
            +
            1338 int offset = 0,
            +
            1339 int axis1 = 0,
            +
            1340 int axis2 = 1,
            +
            1341 StreamOrDevice s = {});
            +
            1342
            +
            1344array diag(const array& a, int k = 0, StreamOrDevice s = {});
            +
            1345
            +
            1347array trace(
            +
            1348 const array& a,
            +
            1349 int offset,
            +
            1350 int axis1,
            +
            1351 int axis2,
            +
            1352 Dtype dtype,
            +
            1353 StreamOrDevice s = {});
            +
            1354array trace(
            +
            1355 const array& a,
            +
            1356 int offset,
            +
            1357 int axis1,
            +
            1358 int axis2,
            +
            1359 StreamOrDevice s = {});
            +
            1360array trace(const array& a, StreamOrDevice s = {});
            +
            1361
            +
            1367std::vector<array> depends(
            +
            1368 const std::vector<array>& inputs,
            +
            1369 const std::vector<array>& dependencies);
            +
            1370
            +
            1372array atleast_1d(const array& a, StreamOrDevice s = {});
            +
            1373std::vector<array> atleast_1d(
            1374 const std::vector<array>& a,
            1375 StreamOrDevice s = {});
            -
            1376array atleast_3d(const array& a, StreamOrDevice s = {});
            -
            1377std::vector<array> atleast_3d(
            +
            1376array atleast_2d(const array& a, StreamOrDevice s = {});
            +
            1377std::vector<array> atleast_2d(
            1378 const std::vector<array>& a,
            1379 StreamOrDevice s = {});
            -
            1380
            -
            1385array number_of_elements(
            -
            1386 const array& a,
            -
            1387 std::vector<int> axes,
            -
            1388 bool inverted,
            -
            1389 Dtype dtype = int32,
            -
            1390 StreamOrDevice s = {});
            -
            1391
            -
            1392array conjugate(const array& a, StreamOrDevice s = {});
            -
            1393
            -
            1395array bitwise_and(const array& a, const array& b, StreamOrDevice s = {});
            -
            1396array operator&(const array& a, const array& b);
            +
            1380array atleast_3d(const array& a, StreamOrDevice s = {});
            +
            1381std::vector<array> atleast_3d(
            +
            1382 const std::vector<array>& a,
            +
            1383 StreamOrDevice s = {});
            +
            1384
            +
            1389array number_of_elements(
            +
            1390 const array& a,
            +
            1391 std::vector<int> axes,
            +
            1392 bool inverted,
            +
            1393 Dtype dtype = int32,
            +
            1394 StreamOrDevice s = {});
            +
            1395
            +
            1396array conjugate(const array& a, StreamOrDevice s = {});
            1397
            -
            1399array bitwise_or(const array& a, const array& b, StreamOrDevice s = {});
            -
            1400array operator|(const array& a, const array& b);
            +
            1399array bitwise_and(const array& a, const array& b, StreamOrDevice s = {});
            +
            1400array operator&(const array& a, const array& b);
            1401
            -
            1403array bitwise_xor(const array& a, const array& b, StreamOrDevice s = {});
            -
            1404array operator^(const array& a, const array& b);
            +
            1403array bitwise_or(const array& a, const array& b, StreamOrDevice s = {});
            +
            1404array operator|(const array& a, const array& b);
            1405
            -
            1407array left_shift(const array& a, const array& b, StreamOrDevice s = {});
            -
            1408array operator<<(const array& a, const array& b);
            +
            1407array bitwise_xor(const array& a, const array& b, StreamOrDevice s = {});
            +
            1408array operator^(const array& a, const array& b);
            1409
            -
            1411array right_shift(const array& a, const array& b, StreamOrDevice s = {});
            -
            1412array operator>>(const array& a, const array& b);
            +
            1411array left_shift(const array& a, const array& b, StreamOrDevice s = {});
            +
            1412array operator<<(const array& a, const array& b);
            1413
            -
            1414array view(const array& a, const Dtype& dtype, StreamOrDevice s = {});
            -
            1417} // namespace mlx::core
            +
            1415array right_shift(const array& a, const array& b, StreamOrDevice s = {});
            +
            1416array operator>>(const array& a, const array& b);
            +
            1417
            +
            1418array view(const array& a, const Dtype& dtype, StreamOrDevice s = {});
            +
            1421} // namespace mlx::core
            mlx::core::array
            Definition array.h:20
            @@ -1337,7 +1341,7 @@ $(function() { codefold.init(0); });
            mlx::core::scatter_prod
            array scatter_prod(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})
            Scatter and prod updates to given indices.
            mlx::core::slice_update
            array slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})
            Update a slice from the source array.
            mlx::core::cos
            array cos(const array &a, StreamOrDevice s={})
            Cosine of the elements of an array.
            -
            mlx::core::operator>=
            array operator>=(const array &a, const array &b)
            Definition ops.h:343
            +
            mlx::core::operator>=
            array operator>=(const array &a, const array &b)
            Definition ops.h:347
            mlx::core::degrees
            array degrees(const array &a, StreamOrDevice s={})
            Convert the elements of an array from Radians to Degrees.
            mlx::core::all
            array all(const array &a, bool keepdims, StreamOrDevice s={})
            True if all elements in the array are true (or non-zero).
            mlx::core::tan
            array tan(const array &a, StreamOrDevice s={})
            Tangent of the elements of an array.
            @@ -1347,10 +1351,9 @@ $(function() { codefold.init(0); });
            mlx::core::minimum
            array minimum(const array &a, const array &b, StreamOrDevice s={})
            Element-wise minimum between two arrays.
            mlx::core::prod
            array prod(const array &a, bool keepdims, StreamOrDevice s={})
            The product of all elements of the array.
            mlx::core::atleast_3d
            array atleast_3d(const array &a, StreamOrDevice s={})
            -
            mlx::core::operator<=
            array operator<=(const array &a, const array &b)
            Definition ops.h:371
            +
            mlx::core::operator<=
            array operator<=(const array &a, const array &b)
            Definition ops.h:375
            mlx::core::reciprocal
            array reciprocal(const array &a, StreamOrDevice s={})
            The reciprocal (1/x) of the elements in an array.
            mlx::core::tri
            array tri(int n, int m, int k, Dtype type, StreamOrDevice s={})
            -
            mlx::core::pad
            array pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), StreamOrDevice s={})
            Pad an array with a constant value.
            mlx::core::nan_to_num
            array nan_to_num(const array &a, float nan=0.0f, const std::optional< float > &posinf=std::nullopt, const std::optional< float > &neginf=std::nullopt, StreamOrDevice s={})
            Replace NaN and infinities with finite numbers.
            mlx::core::flatten
            array flatten(const array &a, int start_axis, int end_axis=-1, StreamOrDevice s={})
            Flatten the dimensions in the range [start_axis, end_axis] .
            mlx::core::isclose
            array isclose(const array &a, const array &b, double rtol=1e-5, double atol=1e-8, bool equal_nan=false, StreamOrDevice s={})
            Returns a boolean array where two arrays are element-wise equal within the specified tolerance.
            @@ -1374,7 +1377,7 @@ $(function() { codefold.init(0); });
            mlx::core::greater_equal
            array greater_equal(const array &a, const array &b, StreamOrDevice s={})
            Returns bool array with (a >= b) element-wise.
            mlx::core::expand_dims
            array expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})
            Add a singleton dimension at the given axes.
            mlx::core::conv2d
            array conv2d(const array &input, const array &weight, const std::pair< int, int > &stride={1, 1}, const std::pair< int, int > &padding={0, 0}, const std::pair< int, int > &dilation={1, 1}, int groups=1, StreamOrDevice s={})
            2D convolution with a filter
            -
            mlx::core::operator>
            array operator>(const array &a, const array &b)
            Definition ops.h:329
            +
            mlx::core::operator>
            array operator>(const array &a, const array &b)
            Definition ops.h:333
            mlx::core::bitwise_and
            array bitwise_and(const array &a, const array &b, StreamOrDevice s={})
            Bitwise and.
            mlx::core::split
            std::vector< array > split(const array &a, int num_splits, int axis, StreamOrDevice s={})
            Split an array into sub-arrays along a given axis.
            mlx::core::matmul
            array matmul(const array &a, const array &b, StreamOrDevice s={})
            Matrix-matrix multiplication.
            @@ -1390,6 +1393,7 @@ $(function() { codefold.init(0); });
            mlx::core::softmax
            array softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})
            Softmax of an array.
            mlx::core::sort
            array sort(const array &a, StreamOrDevice s={})
            Returns a sorted copy of the flattened array.
            mlx::core::max
            array max(const array &a, bool keepdims, StreamOrDevice s={})
            The maximum of all elements of the array.
            +
            mlx::core::pad
            array pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})
            Pad an array with a constant value.
            mlx::core::addmm
            array addmm(array c, array a, array b, const float &alpha=1.f, const float &beta=1.f, StreamOrDevice s={})
            Compute D = beta * C + alpha * (A @ B)
            mlx::core::tril
            array tril(array x, int k=0, StreamOrDevice s={})
            mlx::core::any
            array any(const array &a, bool keepdims, StreamOrDevice s={})
            True if any elements in the array are true (or non-zero).
            @@ -1452,7 +1456,7 @@ $(function() { codefold.init(0); });
            mlx::core::sin
            array sin(const array &a, StreamOrDevice s={})
            Sine of the elements of an array.
            mlx::core::operator&&
            array operator&&(const array &a, const array &b)
            mlx::core::cummax
            array cummax(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})
            Cumulative max of an array.
            -
            mlx::core::operator<
            array operator<(const array &a, const array &b)
            Definition ops.h:357
            +
            mlx::core::operator<
            array operator<(const array &a, const array &b)
            Definition ops.h:361
            mlx::core::atleast_2d
            array atleast_2d(const array &a, StreamOrDevice s={})
            mlx::core::operator/
            array operator/(const array &a, const array &b)
            mlx::core::allclose
            array allclose(const array &a, const array &b, double rtol=1e-5, double atol=1e-8, bool equal_nan=false, StreamOrDevice s={})
            True if the two arrays are equal within the specified tolerance.
            @@ -1470,13 +1474,13 @@ $(function() { codefold.init(0); });
            mlx::core::arange
            void arange(const std::vector< array > &inputs, array &out, double start, double step)
            Definition arange.h:24
            mlx::core::to_stream
            Stream to_stream(StreamOrDevice s)
            mlx::core::copy
            void copy(const array &src, array &dst, CopyType ctype)
            -
            mlx::core::int32
            constexpr Dtype int32
            Definition dtype.h:69
            -
            mlx::core::float32
            constexpr Dtype float32
            Definition dtype.h:73
            +
            mlx::core::int32
            constexpr Dtype int32
            Definition dtype.h:67
            +
            mlx::core::float32
            constexpr Dtype float32
            Definition dtype.h:71
            mlx::core::operator==
            bool operator==(const Device &lhs, const Device &rhs)
            mlx::core::operator!=
            bool operator!=(const Device &lhs, const Device &rhs)
            mlx::core::StreamOrDevice
            std::variant< std::monostate, Stream, Device > StreamOrDevice
            Definition utils.h:14
            -
            mlx::core::Dtype
            Definition dtype.h:15
            +
            mlx::core::Dtype
            Definition dtype.h:13
            diff --git a/docs/build/html/primitives_8h_source.html b/docs/build/html/primitives_8h_source.html index cd061f2e6f..733b23803f 100644 --- a/docs/build/html/primitives_8h_source.html +++ b/docs/build/html/primitives_8h_source.html @@ -2013,514 +2013,521 @@ $(function() { codefold.init(0); });
            1713 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            1714 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            1715
            -
            1716 DEFINE_GRADS();
            -
            -
            1717 void print(std::ostream& os) override {
            -
            1718 os << "Scatter";
            -
            1719 switch (reduce_type_) {
            -
            1720 case Sum:
            -
            1721 os << " Sum";
            -
            1722 break;
            -
            1723 case Prod:
            -
            1724 os << " Prod";
            -
            1725 break;
            -
            1726 case Min:
            -
            1727 os << " Min";
            -
            1728 break;
            -
            1729 case Max:
            -
            1730 os << " Max";
            -
            1731 break;
            -
            1732 case None:
            +
            1716 DEFINE_VMAP();
            +
            1717 DEFINE_GRADS();
            +
            1718
            +
            +
            1719 void print(std::ostream& os) override {
            +
            1720 os << "Scatter";
            +
            1721 switch (reduce_type_) {
            +
            1722 case Sum:
            +
            1723 os << " Sum";
            +
            1724 break;
            +
            1725 case Prod:
            +
            1726 os << " Prod";
            +
            1727 break;
            +
            1728 case Min:
            +
            1729 os << " Min";
            +
            1730 break;
            +
            1731 case Max:
            +
            1732 os << " Max";
            1733 break;
            -
            1734 }
            -
            1735 }
            +
            1734 case None:
            +
            1735 break;
            +
            1736 }
            +
            1737 }
            -
            1736 bool is_equivalent(const Primitive& other) const override;
            -
            1737
            -
            1738 private:
            -
            1739 void eval(const std::vector<array>& inputs, array& out);
            -
            1740 ReduceType reduce_type_;
            -
            1741 std::vector<int> axes_;
            -
            1742};
            +
            1738 bool is_equivalent(const Primitive& other) const override;
            +
            1739
            +
            1740 private:
            +
            1741 void eval(const std::vector<array>& inputs, array& out);
            +
            1742 ReduceType reduce_type_;
            +
            1743 std::vector<int> axes_;
            +
            1744};
            -
            1743
            -
            -
            1744class Sigmoid : public UnaryPrimitive {
            -
            1745 public:
            -
            1746 explicit Sigmoid(Stream stream) : UnaryPrimitive(stream) {}
            -
            1747
            -
            1748 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1749 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1750
            -
            1751 DEFINE_VMAP()
            -
            1752 DEFINE_GRADS()
            -
            1753 DEFINE_PRINT(Sigmoid)
            -
            1754 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            1755 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1756
            -
            1757 private:
            -
            1758 void eval(const std::vector<array>& inputs, array& out);
            -
            1759};
            +
            1745
            +
            +
            1746class Sigmoid : public UnaryPrimitive {
            +
            1747 public:
            +
            1748 explicit Sigmoid(Stream stream) : UnaryPrimitive(stream) {}
            +
            1749
            +
            1750 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1751 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1752
            +
            1753 DEFINE_VMAP()
            +
            1754 DEFINE_GRADS()
            +
            1755 DEFINE_PRINT(Sigmoid)
            +
            1756 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            1757 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            1758
            +
            1759 private:
            +
            1760 void eval(const std::vector<array>& inputs, array& out);
            +
            1761};
            -
            1760
            -
            -
            1761class Sign : public UnaryPrimitive {
            -
            1762 public:
            -
            1763 explicit Sign(Stream stream) : UnaryPrimitive(stream) {}
            -
            1764
            -
            1765 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1766 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1767
            -
            1768 DEFINE_VMAP()
            -
            1769 DEFINE_GRADS()
            -
            1770 DEFINE_PRINT(Sign)
            -
            1771 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            1772 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1773
            -
            1774 private:
            -
            1775 void eval(const std::vector<array>& inputs, array& out);
            -
            1776};
            +
            1762
            +
            +
            1763class Sign : public UnaryPrimitive {
            +
            1764 public:
            +
            1765 explicit Sign(Stream stream) : UnaryPrimitive(stream) {}
            +
            1766
            +
            1767 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1768 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1769
            +
            1770 DEFINE_VMAP()
            +
            1771 DEFINE_GRADS()
            +
            1772 DEFINE_PRINT(Sign)
            +
            1773 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            1774 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            1775
            +
            1776 private:
            +
            1777 void eval(const std::vector<array>& inputs, array& out);
            +
            1778};
            -
            1777
            -
            -
            1778class Sin : public UnaryPrimitive {
            -
            1779 public:
            -
            1780 explicit Sin(Stream stream) : UnaryPrimitive(stream) {}
            -
            1781
            -
            1782 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1783 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1784
            -
            1785 DEFINE_VMAP()
            -
            1786 DEFINE_GRADS()
            -
            1787 DEFINE_PRINT(Sin)
            -
            1788 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            1789 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1790
            -
            1791 private:
            -
            1792 void eval(const std::vector<array>& inputs, array& out);
            -
            1793};
            +
            1779
            +
            +
            1780class Sin : public UnaryPrimitive {
            +
            1781 public:
            +
            1782 explicit Sin(Stream stream) : UnaryPrimitive(stream) {}
            +
            1783
            +
            1784 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1785 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1786
            +
            1787 DEFINE_VMAP()
            +
            1788 DEFINE_GRADS()
            +
            1789 DEFINE_PRINT(Sin)
            +
            1790 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            1791 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            1792
            +
            1793 private:
            +
            1794 void eval(const std::vector<array>& inputs, array& out);
            +
            1795};
            -
            1794
            -
            -
            1795class Sinh : public UnaryPrimitive {
            -
            1796 public:
            -
            1797 explicit Sinh(Stream stream) : UnaryPrimitive(stream) {}
            -
            1798
            -
            1799 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1800 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1801
            -
            1802 DEFINE_VMAP()
            -
            1803 DEFINE_GRADS()
            -
            1804 DEFINE_PRINT(Sinh)
            -
            1805 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            1806 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1807
            -
            1808 private:
            -
            1809 void eval(const std::vector<array>& inputs, array& out);
            -
            1810};
            +
            1796
            +
            +
            1797class Sinh : public UnaryPrimitive {
            +
            1798 public:
            +
            1799 explicit Sinh(Stream stream) : UnaryPrimitive(stream) {}
            +
            1800
            +
            1801 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1802 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1803
            +
            1804 DEFINE_VMAP()
            +
            1805 DEFINE_GRADS()
            +
            1806 DEFINE_PRINT(Sinh)
            +
            1807 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            1808 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            1809
            +
            1810 private:
            +
            1811 void eval(const std::vector<array>& inputs, array& out);
            +
            1812};
            -
            1811
            -
            -
            1812class Slice : public UnaryPrimitive {
            -
            1813 public:
            -
            -
            1814 explicit Slice(
            -
            1815 Stream stream,
            -
            1816 const std::vector<int>& start_indices,
            -
            1817 const std::vector<int>& end_indices,
            -
            1818 const std::vector<int>& strides)
            -
            1819 : UnaryPrimitive(stream),
            -
            1820 start_indices_(start_indices),
            -
            1821 end_indices_(end_indices),
            -
            1822 strides_(strides) {}
            +
            1813
            +
            +
            1814class Slice : public UnaryPrimitive {
            +
            1815 public:
            +
            +
            1816 explicit Slice(
            +
            1817 Stream stream,
            +
            1818 const std::vector<int>& start_indices,
            +
            1819 const std::vector<int>& end_indices,
            +
            1820 const std::vector<int>& strides)
            +
            1821 : UnaryPrimitive(stream),
            +
            1822 start_indices_(start_indices),
            +
            1823 end_indices_(end_indices),
            +
            1824 strides_(strides) {}
            -
            1823
            -
            1824 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1825 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1826
            -
            1827 DEFINE_VMAP()
            -
            1828 DEFINE_GRADS()
            -
            1829 DEFINE_PRINT(Slice)
            -
            1830 bool is_equivalent(const Primitive& other) const override;
            -
            1831
            -
            1832 private:
            -
            1833 std::vector<int> start_indices_;
            -
            1834 std::vector<int> end_indices_;
            -
            1835 std::vector<int> strides_;
            -
            1836
            -
            1837 void eval(const std::vector<array>& inputs, array& out);
            -
            1838};
            +
            1825
            +
            1826 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1827 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1828
            +
            1829 DEFINE_VMAP()
            +
            1830 DEFINE_GRADS()
            +
            1831 DEFINE_PRINT(Slice)
            +
            1832 bool is_equivalent(const Primitive& other) const override;
            +
            1833
            +
            1834 private:
            +
            1835 std::vector<int> start_indices_;
            +
            1836 std::vector<int> end_indices_;
            +
            1837 std::vector<int> strides_;
            +
            1838
            +
            1839 void eval(const std::vector<array>& inputs, array& out);
            +
            1840};
            -
            1839
            -
            -
            1840class SliceUpdate : public UnaryPrimitive {
            -
            1841 public:
            -
            -
            1842 explicit SliceUpdate(
            -
            1843 Stream stream,
            -
            1844 const std::vector<int>& start_indices,
            -
            1845 const std::vector<int>& end_indices,
            -
            1846 const std::vector<int>& strides)
            -
            1847 : UnaryPrimitive(stream),
            -
            1848 start_indices_(start_indices),
            -
            1849 end_indices_(end_indices),
            -
            1850 strides_(strides) {}
            +
            1841
            +
            +
            1842class SliceUpdate : public UnaryPrimitive {
            +
            1843 public:
            +
            +
            1844 explicit SliceUpdate(
            +
            1845 Stream stream,
            +
            1846 const std::vector<int>& start_indices,
            +
            1847 const std::vector<int>& end_indices,
            +
            1848 const std::vector<int>& strides)
            +
            1849 : UnaryPrimitive(stream),
            +
            1850 start_indices_(start_indices),
            +
            1851 end_indices_(end_indices),
            +
            1852 strides_(strides) {}
            -
            1851
            -
            1852 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1853 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1854
            -
            1855 DEFINE_VMAP()
            -
            1856 DEFINE_GRADS()
            -
            1857 DEFINE_PRINT(SliceUpdate)
            -
            1858 bool is_equivalent(const Primitive& other) const override;
            -
            1859
            -
            1860 private:
            -
            1861 std::vector<int> start_indices_;
            -
            1862 std::vector<int> end_indices_;
            -
            1863 std::vector<int> strides_;
            -
            1864
            -
            1865 void eval(const std::vector<array>& inputs, array& out);
            +
            1853
            +
            1854 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1855 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1856
            +
            1857 DEFINE_VMAP()
            +
            1858 DEFINE_GRADS()
            +
            1859 DEFINE_PRINT(SliceUpdate)
            +
            1860 bool is_equivalent(const Primitive& other) const override;
            +
            1861
            +
            1862 private:
            +
            1863 std::vector<int> start_indices_;
            +
            1864 std::vector<int> end_indices_;
            +
            1865 std::vector<int> strides_;
            1866
            -
            1867 std::tuple<int64_t, std::vector<int64_t>> prepare_slice(const array& in);
            -
            1868};
            +
            1867 void eval(const std::vector<array>& inputs, array& out);
            +
            1868
            +
            1869 std::tuple<int64_t, std::vector<int64_t>> prepare_slice(const array& in);
            +
            1870};
            -
            1869
            -
            -
            1870class Softmax : public UnaryPrimitive {
            -
            1871 public:
            -
            -
            1872 explicit Softmax(Stream stream, bool precise)
            -
            1873 : UnaryPrimitive(stream), precise_(precise) {}
            +
            1871
            +
            +
            1872class Softmax : public UnaryPrimitive {
            +
            1873 public:
            +
            +
            1874 explicit Softmax(Stream stream, bool precise)
            +
            1875 : UnaryPrimitive(stream), precise_(precise) {}
            -
            1874
            -
            1875 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1876 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1877
            -
            1878 DEFINE_VMAP()
            -
            1879 DEFINE_GRADS()
            -
            1880 DEFINE_PRINT(Softmax)
            -
            1881 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1882
            -
            1883 bool is_equivalent(const Primitive& other) const override;
            +
            1876
            +
            1877 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1878 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1879
            +
            1880 DEFINE_VMAP()
            +
            1881 DEFINE_GRADS()
            +
            1882 DEFINE_PRINT(Softmax)
            +
            1883 DEFINE_INPUT_OUTPUT_SHAPE()
            1884
            -
            1885 private:
            -
            1886 void eval(const std::vector<array>& inputs, array& out);
            -
            1887 bool precise_;
            -
            1888};
            +
            1885 bool is_equivalent(const Primitive& other) const override;
            +
            1886
            +
            1887 private:
            +
            1888 void eval(const std::vector<array>& inputs, array& out);
            +
            1889 bool precise_;
            +
            1890};
            -
            1889
            -
            -
            1890class Sort : public UnaryPrimitive {
            -
            1891 public:
            -
            -
            1892 explicit Sort(Stream stream, int axis)
            -
            1893 : UnaryPrimitive(stream), axis_(axis) {}
            +
            1891
            +
            +
            1892class Sort : public UnaryPrimitive {
            +
            1893 public:
            +
            +
            1894 explicit Sort(Stream stream, int axis)
            +
            1895 : UnaryPrimitive(stream), axis_(axis) {}
            -
            1894
            -
            1895 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1896 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1897
            -
            1898 DEFINE_VMAP()
            -
            1899 DEFINE_GRADS()
            -
            1900 DEFINE_PRINT(Sort)
            -
            1901 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1902 bool is_equivalent(const Primitive& other) const override;
            -
            1903
            -
            1904 private:
            -
            1905 int axis_;
            -
            1906
            -
            1907 void eval(const std::vector<array>& inputs, array& out);
            -
            1908};
            +
            1896
            +
            1897 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1898 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1899
            +
            1900 DEFINE_VMAP()
            +
            1901 DEFINE_GRADS()
            +
            1902 DEFINE_PRINT(Sort)
            +
            1903 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            1904 bool is_equivalent(const Primitive& other) const override;
            +
            1905
            +
            1906 private:
            +
            1907 int axis_;
            +
            1908
            +
            1909 void eval(const std::vector<array>& inputs, array& out);
            +
            1910};
            -
            1909
            -
            -
            1910class Split : public Primitive {
            -
            1911 public:
            -
            -
            1912 explicit Split(Stream stream, const std::vector<int>& indices, int axis)
            -
            1913 : Primitive(stream), indices_(indices), axis_(axis) {}
            +
            1911
            +
            +
            1912class Split : public Primitive {
            +
            1913 public:
            +
            +
            1914 explicit Split(Stream stream, const std::vector<int>& indices, int axis)
            +
            1915 : Primitive(stream), indices_(indices), axis_(axis) {}
            -
            1914
            -
            1915 void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            -
            1916 override;
            -
            1917 void eval_gpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            +
            1916
            +
            1917 void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            1918 override;
            -
            1919
            -
            1920 DEFINE_VMAP()
            -
            1921 DEFINE_GRADS()
            -
            1922 DEFINE_PRINT(Split)
            -
            1923 bool is_equivalent(const Primitive& other) const override;
            -
            1924
            -
            1925 private:
            -
            1926 void eval(const std::vector<array>& inputs, std::vector<array>& outputs);
            -
            1927
            -
            1928 std::vector<int> indices_;
            -
            1929 int axis_;
            -
            1930};
            +
            1919 void eval_gpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            +
            1920 override;
            +
            1921
            +
            1922 DEFINE_VMAP()
            +
            1923 DEFINE_GRADS()
            +
            1924 DEFINE_PRINT(Split)
            +
            1925 bool is_equivalent(const Primitive& other) const override;
            +
            1926
            +
            1927 private:
            +
            1928 void eval(const std::vector<array>& inputs, std::vector<array>& outputs);
            +
            1929
            +
            1930 std::vector<int> indices_;
            +
            1931 int axis_;
            +
            1932};
            -
            1931
            -
            -
            1932class Square : public UnaryPrimitive {
            -
            1933 public:
            -
            1934 explicit Square(Stream stream) : UnaryPrimitive(stream) {}
            -
            1935
            -
            1936 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1937 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1938
            -
            1939 DEFINE_VMAP()
            -
            1940 DEFINE_GRADS()
            -
            1941 DEFINE_PRINT(Square)
            -
            1942 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            1943 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1944
            -
            1945 private:
            -
            1946 void eval(const std::vector<array>& inputs, array& out);
            -
            1947};
            +
            1933
            +
            +
            1934class Square : public UnaryPrimitive {
            +
            1935 public:
            +
            1936 explicit Square(Stream stream) : UnaryPrimitive(stream) {}
            +
            1937
            +
            1938 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1939 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1940
            +
            1941 DEFINE_VMAP()
            +
            1942 DEFINE_GRADS()
            +
            1943 DEFINE_PRINT(Square)
            +
            1944 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            1945 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            1946
            +
            1947 private:
            +
            1948 void eval(const std::vector<array>& inputs, array& out);
            +
            1949};
            -
            1948
            -
            -
            1949class Sqrt : public UnaryPrimitive {
            -
            1950 public:
            -
            -
            1951 explicit Sqrt(Stream stream, bool recip = false)
            -
            1952 : UnaryPrimitive(stream), recip_(recip) {}
            +
            1950
            +
            +
            1951class Sqrt : public UnaryPrimitive {
            +
            1952 public:
            +
            +
            1953 explicit Sqrt(Stream stream, bool recip = false)
            +
            1954 : UnaryPrimitive(stream), recip_(recip) {}
            -
            1953
            -
            1954 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1955 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1956
            -
            1957 DEFINE_VMAP()
            -
            1958 DEFINE_GRADS()
            -
            1959 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1960 bool is_equivalent(const Primitive& other) const override;
            -
            1961
            -
            -
            1962 void print(std::ostream& os) override {
            -
            1963 if (recip_) {
            -
            1964 os << "Rsqrt";
            -
            1965 } else {
            -
            1966 os << "Sqrt";
            -
            1967 }
            -
            1968 }
            +
            1955
            +
            1956 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1957 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1958
            +
            1959 DEFINE_VMAP()
            +
            1960 DEFINE_GRADS()
            +
            1961 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            1962 bool is_equivalent(const Primitive& other) const override;
            +
            1963
            +
            +
            1964 void print(std::ostream& os) override {
            +
            1965 if (recip_) {
            +
            1966 os << "Rsqrt";
            +
            1967 } else {
            +
            1968 os << "Sqrt";
            +
            1969 }
            +
            1970 }
            -
            1969
            -
            1970 private:
            -
            1971 void eval(const std::vector<array>& inputs, array& out);
            -
            1972 bool recip_;
            -
            1973};
            +
            1971
            +
            1972 private:
            +
            1973 void eval(const std::vector<array>& inputs, array& out);
            +
            1974 bool recip_;
            +
            1975};
            -
            1974
            -
            -
            1975class StopGradient : public UnaryPrimitive {
            -
            1976 public:
            -
            1977 explicit StopGradient(Stream stream) : UnaryPrimitive(stream) {}
            -
            1978
            -
            1979 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1980 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1981
            -
            1982 DEFINE_VMAP()
            -
            1983 DEFINE_PRINT(StopGradient)
            -
            1984 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            1985 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            1986
            -
            1987 private:
            -
            1988 void eval(const std::vector<array>& inputs, array& out);
            -
            1989};
            +
            1976
            +
            +
            1977class StopGradient : public UnaryPrimitive {
            +
            1978 public:
            +
            1979 explicit StopGradient(Stream stream) : UnaryPrimitive(stream) {}
            +
            1980
            +
            1981 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1982 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1983
            +
            1984 DEFINE_VMAP()
            +
            1985 DEFINE_PRINT(StopGradient)
            +
            1986 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            1987 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            1988
            +
            1989 private:
            +
            1990 void eval(const std::vector<array>& inputs, array& out);
            +
            1991};
            -
            1990
            -
            -
            1991class Subtract : public UnaryPrimitive {
            -
            1992 public:
            -
            1993 explicit Subtract(Stream stream) : UnaryPrimitive(stream) {}
            -
            1994
            -
            1995 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            1996 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            1997
            -
            1998 DEFINE_VMAP()
            -
            1999 DEFINE_GRADS()
            -
            2000 DEFINE_PRINT(Subtract)
            -
            2001 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            2002 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            2003
            -
            2004 private:
            -
            2005 void eval(const std::vector<array>& inputs, array& out);
            -
            2006};
            +
            1992
            +
            +
            1993class Subtract : public UnaryPrimitive {
            +
            1994 public:
            +
            1995 explicit Subtract(Stream stream) : UnaryPrimitive(stream) {}
            +
            1996
            +
            1997 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            1998 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            1999
            +
            2000 DEFINE_VMAP()
            +
            2001 DEFINE_GRADS()
            +
            2002 DEFINE_PRINT(Subtract)
            +
            2003 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            2004 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            2005
            +
            2006 private:
            +
            2007 void eval(const std::vector<array>& inputs, array& out);
            +
            2008};
            -
            2007
            -
            -
            2008class Tan : public UnaryPrimitive {
            -
            2009 public:
            -
            2010 explicit Tan(Stream stream) : UnaryPrimitive(stream) {}
            -
            2011
            -
            2012 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            2013 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            2014
            -
            2015 DEFINE_VMAP()
            -
            2016 DEFINE_GRADS()
            -
            2017 DEFINE_PRINT(Tan)
            -
            2018 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            2019 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            2020
            -
            2021 private:
            -
            2022 void eval(const std::vector<array>& inputs, array& out);
            -
            2023};
            +
            2009
            +
            +
            2010class Tan : public UnaryPrimitive {
            +
            2011 public:
            +
            2012 explicit Tan(Stream stream) : UnaryPrimitive(stream) {}
            +
            2013
            +
            2014 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            2015 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            2016
            +
            2017 DEFINE_VMAP()
            +
            2018 DEFINE_GRADS()
            +
            2019 DEFINE_PRINT(Tan)
            +
            2020 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            2021 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            2022
            +
            2023 private:
            +
            2024 void eval(const std::vector<array>& inputs, array& out);
            +
            2025};
            -
            2024
            -
            -
            2025class Tanh : public UnaryPrimitive {
            -
            2026 public:
            -
            2027 explicit Tanh(Stream stream) : UnaryPrimitive(stream) {}
            -
            2028
            -
            2029 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            2030 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            2031
            -
            2032 DEFINE_VMAP()
            -
            2033 DEFINE_GRADS()
            -
            2034 DEFINE_PRINT(Tanh)
            -
            2035 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            2036 DEFINE_INPUT_OUTPUT_SHAPE()
            -
            2037
            -
            2038 private:
            -
            2039 void eval(const std::vector<array>& inputs, array& out);
            -
            2040};
            +
            2026
            +
            +
            2027class Tanh : public UnaryPrimitive {
            +
            2028 public:
            +
            2029 explicit Tanh(Stream stream) : UnaryPrimitive(stream) {}
            +
            2030
            +
            2031 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            2032 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            2033
            +
            2034 DEFINE_VMAP()
            +
            2035 DEFINE_GRADS()
            +
            2036 DEFINE_PRINT(Tanh)
            +
            2037 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            2038 DEFINE_INPUT_OUTPUT_SHAPE()
            +
            2039
            +
            2040 private:
            +
            2041 void eval(const std::vector<array>& inputs, array& out);
            +
            2042};
            -
            2041
            -
            -
            2042class Uniform : public UnaryPrimitive {
            -
            2043 public:
            -
            2044 explicit Uniform(Stream stream) : UnaryPrimitive(stream) {}
            -
            2045
            -
            2046 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            2047 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            2048
            -
            2049 DEFINE_VMAP()
            -
            2050 DEFINE_PRINT(Uniform)
            -
            2051 DEFINE_DEFAULT_IS_EQUIVALENT()
            -
            2052
            -
            2053 private:
            -
            2054 void eval(const std::vector<array>& inputs, array& out);
            -
            2055};
            +
            2043
            +
            +
            2044class Uniform : public UnaryPrimitive {
            +
            2045 public:
            +
            2046 explicit Uniform(Stream stream) : UnaryPrimitive(stream) {}
            +
            2047
            +
            2048 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            2049 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            2050
            +
            2051 DEFINE_VMAP()
            +
            2052 DEFINE_PRINT(Uniform)
            +
            2053 DEFINE_DEFAULT_IS_EQUIVALENT()
            +
            2054
            +
            2055 private:
            +
            2056 void eval(const std::vector<array>& inputs, array& out);
            +
            2057};
            -
            2056
            -
            -
            2057class View : public UnaryPrimitive {
            -
            2058 public:
            -
            -
            2059 explicit View(Stream stream, Dtype dtype)
            -
            2060 : UnaryPrimitive(stream), dtype_(dtype) {}
            +
            2058
            +
            +
            2059class View : public UnaryPrimitive {
            +
            2060 public:
            +
            +
            2061 explicit View(Stream stream, Dtype dtype)
            +
            2062 : UnaryPrimitive(stream), dtype_(dtype) {}
            -
            2061
            -
            2062 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            2063 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            2064
            -
            2065 DEFINE_VMAP()
            -
            2066 void print(std::ostream& os) override;
            -
            2067 bool is_equivalent(const Primitive& other) const override;
            -
            2068
            -
            2069 private:
            -
            2070 Dtype dtype_;
            -
            2071};
            +
            2063
            +
            2064 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            2065 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            2066
            +
            2067 DEFINE_VMAP()
            +
            2068 void print(std::ostream& os) override;
            +
            2069 bool is_equivalent(const Primitive& other) const override;
            +
            2070
            +
            2071 private:
            +
            2072 Dtype dtype_;
            +
            2073};
            -
            2072
            -
            -
            2073class Transpose : public UnaryPrimitive {
            -
            2074 public:
            -
            -
            2075 explicit Transpose(Stream stream, const std::vector<int>& axes)
            -
            2076 : UnaryPrimitive(stream), axes_(axes) {}
            +
            2074
            +
            +
            2075class Transpose : public UnaryPrimitive {
            +
            2076 public:
            +
            +
            2077 explicit Transpose(Stream stream, const std::vector<int>& axes)
            +
            2078 : UnaryPrimitive(stream), axes_(axes) {}
            -
            2077
            -
            2078 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            2079 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            2080
            -
            2081 DEFINE_VMAP()
            -
            2082 DEFINE_GRADS()
            -
            2083 DEFINE_PRINT(Transpose)
            -
            2084 bool is_equivalent(const Primitive& other) const override;
            -
            2085
            -
            2086 private:
            -
            2087 std::vector<int> axes_;
            -
            2088
            -
            2089 void eval(const std::vector<array>& inputs, array& out);
            -
            2090};
            +
            2079
            +
            2080 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            2081 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            2082
            +
            2083 DEFINE_VMAP()
            +
            2084 DEFINE_GRADS()
            +
            2085 DEFINE_PRINT(Transpose)
            +
            2086 bool is_equivalent(const Primitive& other) const override;
            +
            2087
            +
            2088 private:
            +
            2089 std::vector<int> axes_;
            +
            2090
            +
            2091 void eval(const std::vector<array>& inputs, array& out);
            +
            2092};
            -
            2091
            -
            2092/* QR Factorization primitive. */
            -
            -
            2093class QRF : public Primitive {
            -
            2094 public:
            -
            2095 explicit QRF(Stream stream) : Primitive(stream) {}
            -
            2096
            -
            2097 void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            -
            2098 override;
            -
            2099 void eval_gpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            +
            2093
            +
            2094/* QR Factorization primitive. */
            +
            +
            2095class QRF : public Primitive {
            +
            2096 public:
            +
            2097 explicit QRF(Stream stream) : Primitive(stream) {}
            +
            2098
            +
            2099 void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            2100 override;
            -
            2101
            -
            2102 DEFINE_PRINT(QRF)
            +
            2101 void eval_gpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            +
            2102 override;
            2103
            -
            2104 private:
            -
            2105 void eval(const std::vector<array>& inputs, std::vector<array>& outputs);
            -
            2106};
            +
            2104 DEFINE_PRINT(QRF)
            +
            2105
            +
            2106 private:
            +
            2107 void eval(const std::vector<array>& inputs, std::vector<array>& outputs);
            +
            2108};
            -
            2107
            -
            2108/* SVD primitive. */
            -
            -
            2109class SVD : public Primitive {
            -
            2110 public:
            -
            2111 explicit SVD(Stream stream) : Primitive(stream) {}
            -
            2112
            -
            2113 void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            -
            2114 override;
            -
            2115 void eval_gpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            +
            2109
            +
            2110/* SVD primitive. */
            +
            +
            2111class SVD : public Primitive {
            +
            2112 public:
            +
            2113 explicit SVD(Stream stream) : Primitive(stream) {}
            +
            2114
            +
            2115 void eval_cpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            2116 override;
            -
            2117
            -
            2118 DEFINE_VMAP()
            -
            2119 DEFINE_PRINT(SVD)
            -
            2120
            -
            2121 private:
            -
            2122 void eval(const std::vector<array>& inputs, std::vector<array>& outputs);
            -
            2123};
            +
            2117 void eval_gpu(const std::vector<array>& inputs, std::vector<array>& outputs)
            +
            2118 override;
            +
            2119
            +
            2120 DEFINE_VMAP()
            +
            2121 DEFINE_PRINT(SVD)
            +
            2122
            +
            2123 private:
            +
            2124 void eval(const std::vector<array>& inputs, std::vector<array>& outputs);
            +
            2125};
            +
            +
            2126
            +
            2127/* Matrix inversion primitive. */
            +
            +
            2128class Inverse : public UnaryPrimitive {
            +
            2129 public:
            +
            +
            2130 explicit Inverse(Stream stream, bool tri, bool upper)
            +
            2131 : UnaryPrimitive(stream), tri_(tri), upper_(upper) {}
            -
            2124
            -
            2125/* Matrix inversion primitive. */
            -
            -
            2126class Inverse : public UnaryPrimitive {
            -
            2127 public:
            -
            2128 explicit Inverse(Stream stream) : UnaryPrimitive(stream) {}
            -
            2129
            -
            2130 void eval_cpu(const std::vector<array>& inputs, array& output) override;
            -
            2131 void eval_gpu(const std::vector<array>& inputs, array& output) override;
            2132
            -
            2133 DEFINE_VMAP()
            -
            2134 DEFINE_PRINT(Inverse)
            +
            2133 void eval_cpu(const std::vector<array>& inputs, array& output) override;
            +
            2134 void eval_gpu(const std::vector<array>& inputs, array& output) override;
            2135
            -
            2136 private:
            -
            2137 void eval(const std::vector<array>& inputs, array& output);
            -
            2138};
            -
            -
            2139
            -
            -
            2140class Cholesky : public UnaryPrimitive {
            -
            2141 public:
            -
            -
            2142 explicit Cholesky(Stream stream, bool upper)
            -
            2143 : UnaryPrimitive(stream), upper_(upper) {}
            +
            2136 DEFINE_VMAP()
            +
            2137 DEFINE_PRINT(Inverse)
            +
            2138
            +
            2139 private:
            +
            2140 void eval(const std::vector<array>& inputs, array& output);
            +
            2141 bool tri_;
            +
            2142 bool upper_;
            +
            2143};
            2144
            -
            2145 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            -
            2146 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            -
            2147
            -
            2148 DEFINE_VMAP()
            -
            2149 DEFINE_PRINT(Cholesky)
            -
            2150
            -
            2151 private:
            -
            2152 void eval(const std::vector<array>& inputs, array& output);
            -
            2153 bool upper_;
            -
            2154};
            +
            +
            2145class Cholesky : public UnaryPrimitive {
            +
            2146 public:
            +
            +
            2147 explicit Cholesky(Stream stream, bool upper)
            +
            2148 : UnaryPrimitive(stream), upper_(upper) {}
            +
            2149
            +
            2150 void eval_cpu(const std::vector<array>& inputs, array& out) override;
            +
            2151 void eval_gpu(const std::vector<array>& inputs, array& out) override;
            +
            2152
            +
            2153 DEFINE_VMAP()
            +
            2154 DEFINE_PRINT(Cholesky)
            2155
            -
            2156} // namespace mlx::core
            +
            2156 private:
            +
            2157 void eval(const std::vector<array>& inputs, array& output);
            +
            2158 bool upper_;
            +
            2159};
            +
            +
            2160
            +
            2161} // namespace mlx::core
            mlx::core::Abs
            Definition primitives.h:155
            mlx::core::Abs::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            @@ -2612,9 +2619,9 @@ $(function() { codefold.init(0); });
            mlx::core::Ceil::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Ceil::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Ceil::Ceil
            Ceil(Stream stream)
            Definition primitives.h:547
            -
            mlx::core::Cholesky
            Definition primitives.h:2140
            +
            mlx::core::Cholesky
            Definition primitives.h:2145
            mlx::core::Cholesky::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Cholesky::Cholesky
            Cholesky(Stream stream, bool upper)
            Definition primitives.h:2142
            +
            mlx::core::Cholesky::Cholesky
            Cholesky(Stream stream, bool upper)
            Definition primitives.h:2147
            mlx::core::Cholesky::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Compiled
            Definition primitives.h:562
            mlx::core::Compiled::Compiled
            Compiled(Stream stream, std::vector< array > inputs, std::vector< array > outputs, std::vector< array > tape, std::unordered_set< uintptr_t > constant_ids)
            @@ -2719,9 +2726,9 @@ $(function() { codefold.init(0); });
            mlx::core::Hadamard::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Hadamard::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Hadamard::Hadamard
            Hadamard(Stream stream, float scale)
            Definition primitives.h:1088
            -
            mlx::core::Inverse
            Definition primitives.h:2126
            +
            mlx::core::Inverse
            Definition primitives.h:2128
            mlx::core::Inverse::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &output) override
            -
            mlx::core::Inverse::Inverse
            Inverse(Stream stream)
            Definition primitives.h:2128
            +
            mlx::core::Inverse::Inverse
            Inverse(Stream stream, bool tri, bool upper)
            Definition primitives.h:2130
            mlx::core::Inverse::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &output) override
            mlx::core::LessEqual
            Definition primitives.h:1124
            mlx::core::LessEqual::LessEqual
            LessEqual(Stream stream)
            Definition primitives.h:1126
            @@ -2818,8 +2825,8 @@ $(function() { codefold.init(0); });
            mlx::core::Primitive::eval_gpu
            virtual void eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs)=0
            mlx::core::Primitive::print
            virtual void print(std::ostream &os)=0
            Print the primitive.
            mlx::core::Primitive::Primitive
            Primitive(Stream stream)
            Definition primitives.h:50
            -
            mlx::core::QRF
            Definition primitives.h:2093
            -
            mlx::core::QRF::QRF
            QRF(Stream stream)
            Definition primitives.h:2095
            +
            mlx::core::QRF
            Definition primitives.h:2095
            +
            mlx::core::QRF::QRF
            QRF(Stream stream)
            Definition primitives.h:2097
            mlx::core::QRF::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override
            A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the out...
            mlx::core::QRF::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override
            mlx::core::QuantizedMatmul
            Definition primitives.h:1483
            @@ -2849,10 +2856,10 @@ $(function() { codefold.init(0); });
            mlx::core::Round::Round
            Round(Stream stream)
            Definition primitives.h:1638
            mlx::core::Round::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Round::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::SVD
            Definition primitives.h:2109
            +
            mlx::core::SVD
            Definition primitives.h:2111
            mlx::core::SVD::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override
            A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the out...
            mlx::core::SVD::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override
            -
            mlx::core::SVD::SVD
            SVD(Stream stream)
            Definition primitives.h:2111
            +
            mlx::core::SVD::SVD
            SVD(Stream stream)
            Definition primitives.h:2113
            mlx::core::Scan
            Definition primitives.h:1653
            mlx::core::Scan::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Scan::ReduceType
            ReduceType
            Definition primitives.h:1655
            @@ -2865,75 +2872,75 @@ $(function() { codefold.init(0); });
            mlx::core::Scatter::ReduceType
            ReduceType
            Definition primitives.h:1705
            mlx::core::Scatter::Max
            @ Max
            Definition primitives.h:1705
            mlx::core::Scatter::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Scatter::print
            void print(std::ostream &os) override
            Print the primitive.
            Definition primitives.h:1717
            +
            mlx::core::Scatter::print
            void print(std::ostream &os) override
            Print the primitive.
            Definition primitives.h:1719
            mlx::core::Scatter::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Scatter::Scatter
            Scatter(Stream stream, ReduceType reduce_type, const std::vector< int > &axes)
            Definition primitives.h:1707
            mlx::core::Select
            Definition primitives.h:844
            mlx::core::Select::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Select::Select
            Select(Stream stream)
            Definition primitives.h:846
            mlx::core::Select::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Sigmoid
            Definition primitives.h:1744
            -
            mlx::core::Sigmoid::Sigmoid
            Sigmoid(Stream stream)
            Definition primitives.h:1746
            +
            mlx::core::Sigmoid
            Definition primitives.h:1746
            +
            mlx::core::Sigmoid::Sigmoid
            Sigmoid(Stream stream)
            Definition primitives.h:1748
            mlx::core::Sigmoid::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Sigmoid::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Sign
            Definition primitives.h:1761
            +
            mlx::core::Sign
            Definition primitives.h:1763
            mlx::core::Sign::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Sign::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Sign::Sign
            Sign(Stream stream)
            Definition primitives.h:1763
            -
            mlx::core::Sin
            Definition primitives.h:1778
            -
            mlx::core::Sin::Sin
            Sin(Stream stream)
            Definition primitives.h:1780
            +
            mlx::core::Sign::Sign
            Sign(Stream stream)
            Definition primitives.h:1765
            +
            mlx::core::Sin
            Definition primitives.h:1780
            +
            mlx::core::Sin::Sin
            Sin(Stream stream)
            Definition primitives.h:1782
            mlx::core::Sin::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Sin::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Sinh
            Definition primitives.h:1795
            -
            mlx::core::Sinh::Sinh
            Sinh(Stream stream)
            Definition primitives.h:1797
            +
            mlx::core::Sinh
            Definition primitives.h:1797
            +
            mlx::core::Sinh::Sinh
            Sinh(Stream stream)
            Definition primitives.h:1799
            mlx::core::Sinh::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Sinh::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Slice
            Definition primitives.h:1812
            +
            mlx::core::Slice
            Definition primitives.h:1814
            mlx::core::Slice::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Slice::Slice
            Slice(Stream stream, const std::vector< int > &start_indices, const std::vector< int > &end_indices, const std::vector< int > &strides)
            Definition primitives.h:1814
            +
            mlx::core::Slice::Slice
            Slice(Stream stream, const std::vector< int > &start_indices, const std::vector< int > &end_indices, const std::vector< int > &strides)
            Definition primitives.h:1816
            mlx::core::Slice::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::SliceUpdate
            Definition primitives.h:1840
            -
            mlx::core::SliceUpdate::SliceUpdate
            SliceUpdate(Stream stream, const std::vector< int > &start_indices, const std::vector< int > &end_indices, const std::vector< int > &strides)
            Definition primitives.h:1842
            +
            mlx::core::SliceUpdate
            Definition primitives.h:1842
            +
            mlx::core::SliceUpdate::SliceUpdate
            SliceUpdate(Stream stream, const std::vector< int > &start_indices, const std::vector< int > &end_indices, const std::vector< int > &strides)
            Definition primitives.h:1844
            mlx::core::SliceUpdate::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::SliceUpdate::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Softmax
            Definition primitives.h:1870
            +
            mlx::core::Softmax
            Definition primitives.h:1872
            mlx::core::Softmax::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Softmax::Softmax
            Softmax(Stream stream, bool precise)
            Definition primitives.h:1872
            +
            mlx::core::Softmax::Softmax
            Softmax(Stream stream, bool precise)
            Definition primitives.h:1874
            mlx::core::Softmax::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Sort
            Definition primitives.h:1890
            +
            mlx::core::Sort
            Definition primitives.h:1892
            mlx::core::Sort::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Sort::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Sort::Sort
            Sort(Stream stream, int axis)
            Definition primitives.h:1892
            -
            mlx::core::Split
            Definition primitives.h:1910
            +
            mlx::core::Sort::Sort
            Sort(Stream stream, int axis)
            Definition primitives.h:1894
            +
            mlx::core::Split
            Definition primitives.h:1912
            mlx::core::Split::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override
            -
            mlx::core::Split::Split
            Split(Stream stream, const std::vector< int > &indices, int axis)
            Definition primitives.h:1912
            +
            mlx::core::Split::Split
            Split(Stream stream, const std::vector< int > &indices, int axis)
            Definition primitives.h:1914
            mlx::core::Split::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override
            A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the out...
            -
            mlx::core::Sqrt
            Definition primitives.h:1949
            +
            mlx::core::Sqrt
            Definition primitives.h:1951
            mlx::core::Sqrt::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Sqrt::Sqrt
            Sqrt(Stream stream, bool recip=false)
            Definition primitives.h:1951
            +
            mlx::core::Sqrt::Sqrt
            Sqrt(Stream stream, bool recip=false)
            Definition primitives.h:1953
            mlx::core::Sqrt::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Square
            Definition primitives.h:1932
            +
            mlx::core::Square
            Definition primitives.h:1934
            mlx::core::Square::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Square::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Square::Square
            Square(Stream stream)
            Definition primitives.h:1934
            -
            mlx::core::StopGradient
            Definition primitives.h:1975
            +
            mlx::core::Square::Square
            Square(Stream stream)
            Definition primitives.h:1936
            +
            mlx::core::StopGradient
            Definition primitives.h:1977
            mlx::core::StopGradient::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::StopGradient::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::StopGradient::StopGradient
            StopGradient(Stream stream)
            Definition primitives.h:1977
            -
            mlx::core::Subtract
            Definition primitives.h:1991
            +
            mlx::core::StopGradient::StopGradient
            StopGradient(Stream stream)
            Definition primitives.h:1979
            +
            mlx::core::Subtract
            Definition primitives.h:1993
            mlx::core::Subtract::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Subtract::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Subtract::Subtract
            Subtract(Stream stream)
            Definition primitives.h:1993
            -
            mlx::core::Tan
            Definition primitives.h:2008
            -
            mlx::core::Tan::Tan
            Tan(Stream stream)
            Definition primitives.h:2010
            +
            mlx::core::Subtract::Subtract
            Subtract(Stream stream)
            Definition primitives.h:1995
            +
            mlx::core::Tan
            Definition primitives.h:2010
            +
            mlx::core::Tan::Tan
            Tan(Stream stream)
            Definition primitives.h:2012
            mlx::core::Tan::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Tan::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Tanh
            Definition primitives.h:2025
            +
            mlx::core::Tanh
            Definition primitives.h:2027
            mlx::core::Tanh::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Tanh::Tanh
            Tanh(Stream stream)
            Definition primitives.h:2027
            +
            mlx::core::Tanh::Tanh
            Tanh(Stream stream)
            Definition primitives.h:2029
            mlx::core::Tanh::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Transpose
            Definition primitives.h:2073
            -
            mlx::core::Transpose::Transpose
            Transpose(Stream stream, const std::vector< int > &axes)
            Definition primitives.h:2075
            +
            mlx::core::Transpose
            Definition primitives.h:2075
            +
            mlx::core::Transpose::Transpose
            Transpose(Stream stream, const std::vector< int > &axes)
            Definition primitives.h:2077
            mlx::core::Transpose::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Transpose::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::UnaryPrimitive
            Definition primitives.h:127
            @@ -2947,18 +2954,19 @@ $(function() { codefold.init(0); });
            mlx::core::UnaryPrimitive::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override
            A primitive must know how to evaluate itself on the CPU/GPU for the given inputs and populate the out...
            Definition primitives.h:137
            mlx::core::UnaryPrimitive::operator=
            UnaryPrimitive & operator=(UnaryPrimitive &&other)=delete
            mlx::core::UnaryPrimitive::~UnaryPrimitive
            virtual ~UnaryPrimitive()=default
            -
            mlx::core::Uniform
            Definition primitives.h:2042
            +
            mlx::core::Uniform
            Definition primitives.h:2044
            mlx::core::Uniform::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            mlx::core::Uniform::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::Uniform::Uniform
            Uniform(Stream stream)
            Definition primitives.h:2044
            -
            mlx::core::View
            Definition primitives.h:2057
            +
            mlx::core::Uniform::Uniform
            Uniform(Stream stream)
            Definition primitives.h:2046
            +
            mlx::core::View
            Definition primitives.h:2059
            mlx::core::View::eval_cpu
            void eval_cpu(const std::vector< array > &inputs, array &out) override
            -
            mlx::core::View::View
            View(Stream stream, Dtype dtype)
            Definition primitives.h:2059
            +
            mlx::core::View::View
            View(Stream stream, Dtype dtype)
            Definition primitives.h:2061
            mlx::core::View::eval_gpu
            void eval_gpu(const std::vector< array > &inputs, array &out) override
            mlx::core::array
            Definition array.h:20
            op
            Op op
            Definition binary.h:141
            mlx::core::std
            array std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})
            Computes the standard deviation of the elements of an array.
            +
            mlx::core::tri
            array tri(int n, int m, int k, Dtype type, StreamOrDevice s={})
            mlx::core::transpose
            array transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})
            Permutes the dimensions according to the given axes.
            mlx::core
            Definition allocator.h:7
            @@ -2982,7 +2990,7 @@ $(function() { codefold.init(0); });
            RightShift
            Definition binary_ops.h:277
            Sum
            Definition ops.h:99
            mlx::core::Device
            Definition device.h:7
            -
            mlx::core::Dtype
            Definition dtype.h:15
            +
            mlx::core::Dtype
            Definition dtype.h:13
            mlx::core::Stream
            Definition stream.h:9
            mlx::core::Stream::device
            Device device
            Definition stream.h:11
            diff --git a/docs/build/html/python/_autosummary/mlx.core.Device.html b/docs/build/html/python/_autosummary/mlx.core.Device.html index f01a48434d..52a1bc5971 100644 --- a/docs/build/html/python/_autosummary/mlx.core.Device.html +++ b/docs/build/html/python/_autosummary/mlx.core.Device.html @@ -8,7 +8,7 @@ - mlx.core.Device — MLX 0.16.1 documentation + mlx.core.Device — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home +
            @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.Dtype.html b/docs/build/html/python/_autosummary/mlx.core.Dtype.html index d0dc8a5475..52a262941c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.Dtype.html +++ b/docs/build/html/python/_autosummary/mlx.core.Dtype.html @@ -8,7 +8,7 @@ - mlx.core.Dtype — MLX 0.16.1 documentation + mlx.core.Dtype — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home +
            • @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html b/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html index 77f80e27fc..beb7a6fe5e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html +++ b/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html @@ -8,7 +8,7 @@ - mlx.core.DtypeCategory — MLX 0.16.1 documentation + mlx.core.DtypeCategory — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home +
            • @@ -457,8 +457,10 @@
          • Linear Algebra @@ -849,7 +851,7 @@ document.write(`

            mlx.core.DtypeCategory#

            -class DtypeCategory#
            +class DtypeCategory(value)#

            Type to hold categories of dtypes.

            @@ -900,13 +902,34 @@ document.write(`

            See also issubdtype().

            -__init__(*args, **kwargs)#
            +__init__()#
            -

            Methods

            +

            Attributes

            - + + + + + + + + + + + + + + + + + + + + + + diff --git a/docs/build/html/python/_autosummary/mlx.core.Stream.html b/docs/build/html/python/_autosummary/mlx.core.Stream.html index c686f1133a..89e79731d2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.Stream.html +++ b/docs/build/html/python/_autosummary/mlx.core.Stream.html @@ -8,7 +8,7 @@ - mlx.core.stream — MLX 0.16.1 documentation + mlx.core.stream — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.abs.html b/docs/build/html/python/_autosummary/mlx.core.abs.html index 56718a7668..630ea61040 100644 --- a/docs/build/html/python/_autosummary/mlx.core.abs.html +++ b/docs/build/html/python/_autosummary/mlx.core.abs.html @@ -8,7 +8,7 @@ - mlx.core.abs — MLX 0.16.1 documentation + mlx.core.abs — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.add.html b/docs/build/html/python/_autosummary/mlx.core.add.html index 9e2ad809f4..5ecdcc3588 100644 --- a/docs/build/html/python/_autosummary/mlx.core.add.html +++ b/docs/build/html/python/_autosummary/mlx.core.add.html @@ -8,7 +8,7 @@ - mlx.core.add — MLX 0.16.1 documentation + mlx.core.add — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.addmm.html b/docs/build/html/python/_autosummary/mlx.core.addmm.html index 9ebdc17cb8..216247879e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.addmm.html +++ b/docs/build/html/python/_autosummary/mlx.core.addmm.html @@ -8,7 +8,7 @@ - mlx.core.addmm — MLX 0.16.1 documentation + mlx.core.addmm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.all.html b/docs/build/html/python/_autosummary/mlx.core.all.html index 252fe28de2..809b059ffd 100644 --- a/docs/build/html/python/_autosummary/mlx.core.all.html +++ b/docs/build/html/python/_autosummary/mlx.core.all.html @@ -8,7 +8,7 @@ - mlx.core.all — MLX 0.16.1 documentation + mlx.core.all — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.allclose.html b/docs/build/html/python/_autosummary/mlx.core.allclose.html index 26a634c2a7..3f4ecaadaf 100644 --- a/docs/build/html/python/_autosummary/mlx.core.allclose.html +++ b/docs/build/html/python/_autosummary/mlx.core.allclose.html @@ -8,7 +8,7 @@ - mlx.core.allclose — MLX 0.16.1 documentation + mlx.core.allclose — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.any.html b/docs/build/html/python/_autosummary/mlx.core.any.html index 04f7386ccf..4f9dd8a44e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.any.html +++ b/docs/build/html/python/_autosummary/mlx.core.any.html @@ -8,7 +8,7 @@ - mlx.core.any — MLX 0.16.1 documentation + mlx.core.any — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.arange.html b/docs/build/html/python/_autosummary/mlx.core.arange.html index b635342dff..fca10be478 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arange.html +++ b/docs/build/html/python/_autosummary/mlx.core.arange.html @@ -8,7 +8,7 @@ - mlx.core.arange — MLX 0.16.1 documentation + mlx.core.arange — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.arccos.html b/docs/build/html/python/_autosummary/mlx.core.arccos.html index 19a9978c5b..60471dd117 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arccos.html +++ b/docs/build/html/python/_autosummary/mlx.core.arccos.html @@ -8,7 +8,7 @@ - mlx.core.arccos — MLX 0.16.1 documentation + mlx.core.arccos — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.arccosh.html b/docs/build/html/python/_autosummary/mlx.core.arccosh.html index ea3d8e6602..dc8426cc8a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arccosh.html +++ b/docs/build/html/python/_autosummary/mlx.core.arccosh.html @@ -8,7 +8,7 @@ - mlx.core.arccosh — MLX 0.16.1 documentation + mlx.core.arccosh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsin.html b/docs/build/html/python/_autosummary/mlx.core.arcsin.html index a4c035edb3..ece202a540 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arcsin.html +++ b/docs/build/html/python/_autosummary/mlx.core.arcsin.html @@ -8,7 +8,7 @@ - mlx.core.arcsin — MLX 0.16.1 documentation + mlx.core.arcsin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsinh.html b/docs/build/html/python/_autosummary/mlx.core.arcsinh.html index 055311582d..2b248b3f31 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arcsinh.html +++ b/docs/build/html/python/_autosummary/mlx.core.arcsinh.html @@ -8,7 +8,7 @@ - mlx.core.arcsinh — MLX 0.16.1 documentation + mlx.core.arcsinh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan.html b/docs/build/html/python/_autosummary/mlx.core.arctan.html index 745e6def37..e8c19ed42b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arctan.html +++ b/docs/build/html/python/_autosummary/mlx.core.arctan.html @@ -8,7 +8,7 @@ - mlx.core.arctan — MLX 0.16.1 documentation + mlx.core.arctan — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan2.html b/docs/build/html/python/_autosummary/mlx.core.arctan2.html index 73cd68f784..b2f0489215 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arctan2.html +++ b/docs/build/html/python/_autosummary/mlx.core.arctan2.html @@ -8,7 +8,7 @@ - mlx.core.arctan2 — MLX 0.16.1 documentation + mlx.core.arctan2 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.arctanh.html b/docs/build/html/python/_autosummary/mlx.core.arctanh.html index 3f9a8ece18..4592c82a5d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.arctanh.html +++ b/docs/build/html/python/_autosummary/mlx.core.arctanh.html @@ -8,7 +8,7 @@ - mlx.core.arctanh — MLX 0.16.1 documentation + mlx.core.arctanh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.argmax.html b/docs/build/html/python/_autosummary/mlx.core.argmax.html index 63c198c10f..00e6e3f4d8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.argmax.html +++ b/docs/build/html/python/_autosummary/mlx.core.argmax.html @@ -8,7 +8,7 @@ - mlx.core.argmax — MLX 0.16.1 documentation + mlx.core.argmax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.argmin.html b/docs/build/html/python/_autosummary/mlx.core.argmin.html index 5f0ae28cd5..4a8f047ce9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.argmin.html +++ b/docs/build/html/python/_autosummary/mlx.core.argmin.html @@ -8,7 +8,7 @@ - mlx.core.argmin — MLX 0.16.1 documentation + mlx.core.argmin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.argpartition.html b/docs/build/html/python/_autosummary/mlx.core.argpartition.html index ac19f10a8e..3d8bf535a0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.argpartition.html +++ b/docs/build/html/python/_autosummary/mlx.core.argpartition.html @@ -8,7 +8,7 @@ - mlx.core.argpartition — MLX 0.16.1 documentation + mlx.core.argpartition — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.argsort.html b/docs/build/html/python/_autosummary/mlx.core.argsort.html index 71dcecc2ac..da73580042 100644 --- a/docs/build/html/python/_autosummary/mlx.core.argsort.html +++ b/docs/build/html/python/_autosummary/mlx.core.argsort.html @@ -8,7 +8,7 @@ - mlx.core.argsort — MLX 0.16.1 documentation + mlx.core.argsort — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.T.html b/docs/build/html/python/_autosummary/mlx.core.array.T.html index 872e4509e2..5513ca6e55 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.T.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.T.html @@ -8,7 +8,7 @@ - mlx.core.array.T — MLX 0.16.1 documentation + mlx.core.array.T — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.abs.html b/docs/build/html/python/_autosummary/mlx.core.array.abs.html index 632124d026..85e5045952 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.abs.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.abs.html @@ -8,7 +8,7 @@ - mlx.core.array.abs — MLX 0.16.1 documentation + mlx.core.array.abs — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.all.html b/docs/build/html/python/_autosummary/mlx.core.array.all.html index a30ebb2e36..91ba69c094 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.all.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.all.html @@ -8,7 +8,7 @@ - mlx.core.array.all — MLX 0.16.1 documentation + mlx.core.array.all — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.any.html b/docs/build/html/python/_autosummary/mlx.core.array.any.html index dc0e76884e..a08fddcecf 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.any.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.any.html @@ -8,7 +8,7 @@ - mlx.core.array.any — MLX 0.16.1 documentation + mlx.core.array.any — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.argmax.html b/docs/build/html/python/_autosummary/mlx.core.array.argmax.html index bcb09c603b..a4a24e6c9c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.argmax.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.argmax.html @@ -8,7 +8,7 @@ - mlx.core.array.argmax — MLX 0.16.1 documentation + mlx.core.array.argmax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.argmin.html b/docs/build/html/python/_autosummary/mlx.core.array.argmin.html index 0d508b34ba..3fe4e3a3b4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.argmin.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.argmin.html @@ -8,7 +8,7 @@ - mlx.core.array.argmin — MLX 0.16.1 documentation + mlx.core.array.argmin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.astype.html b/docs/build/html/python/_autosummary/mlx.core.array.astype.html index 57b35b4b03..4d2d4807c1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.astype.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.astype.html @@ -8,7 +8,7 @@ - mlx.core.array.astype — MLX 0.16.1 documentation + mlx.core.array.astype — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.at.html b/docs/build/html/python/_autosummary/mlx.core.array.at.html index fafad459f4..1b059e2615 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.at.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.at.html @@ -8,7 +8,7 @@ - mlx.core.array.at — MLX 0.16.1 documentation + mlx.core.array.at — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.conj.html b/docs/build/html/python/_autosummary/mlx.core.array.conj.html index e202884e89..afc31beee3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.conj.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.conj.html @@ -8,7 +8,7 @@ - mlx.core.array.conj — MLX 0.16.1 documentation + mlx.core.array.conj — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cos.html b/docs/build/html/python/_autosummary/mlx.core.array.cos.html index 9a22449280..751f037814 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cos.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cos.html @@ -8,7 +8,7 @@ - mlx.core.array.cos — MLX 0.16.1 documentation + mlx.core.array.cos — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cummax.html b/docs/build/html/python/_autosummary/mlx.core.array.cummax.html index 8355275cca..bf611c89b7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cummax.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cummax.html @@ -8,7 +8,7 @@ - mlx.core.array.cummax — MLX 0.16.1 documentation + mlx.core.array.cummax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cummin.html b/docs/build/html/python/_autosummary/mlx.core.array.cummin.html index aa3fb6eccb..884bdf4390 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cummin.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cummin.html @@ -8,7 +8,7 @@ - mlx.core.array.cummin — MLX 0.16.1 documentation + mlx.core.array.cummin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html b/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html index 2ba2ab0a10..d7883152e0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html @@ -8,7 +8,7 @@ - mlx.core.array.cumprod — MLX 0.16.1 documentation + mlx.core.array.cumprod — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html b/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html index d48650f473..fb8eeca83b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html @@ -8,7 +8,7 @@ - mlx.core.array.cumsum — MLX 0.16.1 documentation + mlx.core.array.cumsum — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.diag.html b/docs/build/html/python/_autosummary/mlx.core.array.diag.html index d34450194f..245f6f0c5f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.diag.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.diag.html @@ -8,7 +8,7 @@ - mlx.core.array.diag — MLX 0.16.1 documentation + mlx.core.array.diag — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html b/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html index 0d03f7d24f..89a6813e6e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html @@ -8,7 +8,7 @@ - mlx.core.array.diagonal — MLX 0.16.1 documentation + mlx.core.array.diagonal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.dtype.html b/docs/build/html/python/_autosummary/mlx.core.array.dtype.html index af73d2a869..c4525d3c6f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.dtype.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.dtype.html @@ -8,7 +8,7 @@ - mlx.core.array.dtype — MLX 0.16.1 documentation + mlx.core.array.dtype — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.exp.html b/docs/build/html/python/_autosummary/mlx.core.array.exp.html index 68b9e8c480..e7090ecb38 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.exp.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.exp.html @@ -8,7 +8,7 @@ - mlx.core.array.exp — MLX 0.16.1 documentation + mlx.core.array.exp — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.flatten.html b/docs/build/html/python/_autosummary/mlx.core.array.flatten.html index 2bb84828c9..68d9da2287 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.flatten.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.flatten.html @@ -8,7 +8,7 @@ - mlx.core.array.flatten — MLX 0.16.1 documentation + mlx.core.array.flatten — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.html b/docs/build/html/python/_autosummary/mlx.core.array.html index 6b1402dfc4..eee2edf179 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.html @@ -8,7 +8,7 @@ - mlx.core.array — MLX 0.16.1 documentation + mlx.core.array — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.item.html b/docs/build/html/python/_autosummary/mlx.core.array.item.html index 0a0a062df8..2fc5f642cf 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.item.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.item.html @@ -8,7 +8,7 @@ - mlx.core.array.item — MLX 0.16.1 documentation + mlx.core.array.item — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html b/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html index dd434a0e5f..fa263453f5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html @@ -8,7 +8,7 @@ - mlx.core.array.itemsize — MLX 0.16.1 documentation + mlx.core.array.itemsize — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log.html b/docs/build/html/python/_autosummary/mlx.core.array.log.html index b7177e450e..23ff1ab62e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.log.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.log.html @@ -8,7 +8,7 @@ - mlx.core.array.log — MLX 0.16.1 documentation + mlx.core.array.log — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log10.html b/docs/build/html/python/_autosummary/mlx.core.array.log10.html index 6a592025bd..9edfed222e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.log10.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.log10.html @@ -8,7 +8,7 @@ - mlx.core.array.log10 — MLX 0.16.1 documentation + mlx.core.array.log10 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log1p.html b/docs/build/html/python/_autosummary/mlx.core.array.log1p.html index 9f01802890..ede5a82713 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.log1p.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.log1p.html @@ -8,7 +8,7 @@ - mlx.core.array.log1p — MLX 0.16.1 documentation + mlx.core.array.log1p — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log2.html b/docs/build/html/python/_autosummary/mlx.core.array.log2.html index 05e25fa39f..1ccda02fc0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.log2.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.log2.html @@ -8,7 +8,7 @@ - mlx.core.array.log2 — MLX 0.16.1 documentation + mlx.core.array.log2 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html b/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html index ee072eb713..f088b0c6d2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html @@ -8,7 +8,7 @@ - mlx.core.array.logsumexp — MLX 0.16.1 documentation + mlx.core.array.logsumexp — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.max.html b/docs/build/html/python/_autosummary/mlx.core.array.max.html index 60b0d8803c..2741975346 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.max.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.max.html @@ -8,7 +8,7 @@ - mlx.core.array.max — MLX 0.16.1 documentation + mlx.core.array.max — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.mean.html b/docs/build/html/python/_autosummary/mlx.core.array.mean.html index fc9b234fb2..c06751738f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.mean.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.mean.html @@ -8,7 +8,7 @@ - mlx.core.array.mean — MLX 0.16.1 documentation + mlx.core.array.mean — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.min.html b/docs/build/html/python/_autosummary/mlx.core.array.min.html index 9dfa565d94..f4010ad74f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.min.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.min.html @@ -8,7 +8,7 @@ - mlx.core.array.min — MLX 0.16.1 documentation + mlx.core.array.min — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html b/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html index f9c08d4e53..3cc457526c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html @@ -8,7 +8,7 @@ - mlx.core.array.moveaxis — MLX 0.16.1 documentation + mlx.core.array.moveaxis — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html b/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html index e69493bc18..39f68fe354 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html @@ -8,7 +8,7 @@ - mlx.core.array.nbytes — MLX 0.16.1 documentation + mlx.core.array.nbytes — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.ndim.html b/docs/build/html/python/_autosummary/mlx.core.array.ndim.html index 5f49d97d6f..514088a3c4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.ndim.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.ndim.html @@ -8,7 +8,7 @@ - mlx.core.array.ndim — MLX 0.16.1 documentation + mlx.core.array.ndim — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.prod.html b/docs/build/html/python/_autosummary/mlx.core.array.prod.html index d1aaa4da04..6b574fd0ab 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.prod.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.prod.html @@ -8,7 +8,7 @@ - mlx.core.array.prod — MLX 0.16.1 documentation + mlx.core.array.prod — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html b/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html index 555f42b38c..91f5dfa0af 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html @@ -8,7 +8,7 @@ - mlx.core.array.reciprocal — MLX 0.16.1 documentation + mlx.core.array.reciprocal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.reshape.html b/docs/build/html/python/_autosummary/mlx.core.array.reshape.html index 2352651533..74fbd0e94a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.reshape.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.reshape.html @@ -8,7 +8,7 @@ - mlx.core.array.reshape — MLX 0.16.1 documentation + mlx.core.array.reshape — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.round.html b/docs/build/html/python/_autosummary/mlx.core.array.round.html index bf694cda40..94487fb630 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.round.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.round.html @@ -8,7 +8,7 @@ - mlx.core.array.round — MLX 0.16.1 documentation + mlx.core.array.round — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html b/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html index ef54ad6b80..fdf0b65728 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html @@ -8,7 +8,7 @@ - mlx.core.array.rsqrt — MLX 0.16.1 documentation + mlx.core.array.rsqrt — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.shape.html b/docs/build/html/python/_autosummary/mlx.core.array.shape.html index a2f677a6c0..27f25d2c68 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.shape.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.shape.html @@ -8,7 +8,7 @@ - mlx.core.array.shape — MLX 0.16.1 documentation + mlx.core.array.shape — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sin.html b/docs/build/html/python/_autosummary/mlx.core.array.sin.html index 8aee0638c8..22a3ff3f2c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.sin.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.sin.html @@ -8,7 +8,7 @@ - mlx.core.array.sin — MLX 0.16.1 documentation + mlx.core.array.sin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.size.html b/docs/build/html/python/_autosummary/mlx.core.array.size.html index 5f447c7e54..6ce480d493 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.size.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.size.html @@ -8,7 +8,7 @@ - mlx.core.array.size — MLX 0.16.1 documentation + mlx.core.array.size — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.split.html b/docs/build/html/python/_autosummary/mlx.core.array.split.html index d9564129a8..8cea74eeb6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.split.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.split.html @@ -8,7 +8,7 @@ - mlx.core.array.split — MLX 0.16.1 documentation + mlx.core.array.split — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html b/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html index b702172f78..589487271f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html @@ -8,7 +8,7 @@ - mlx.core.array.sqrt — MLX 0.16.1 documentation + mlx.core.array.sqrt — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.square.html b/docs/build/html/python/_autosummary/mlx.core.array.square.html index b087ee92e4..286950f170 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.square.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.square.html @@ -8,7 +8,7 @@ - mlx.core.array.square — MLX 0.16.1 documentation + mlx.core.array.square — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html b/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html index 7ce553bf90..c727bb1f90 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html @@ -8,7 +8,7 @@ - mlx.core.array.squeeze — MLX 0.16.1 documentation + mlx.core.array.squeeze — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sum.html b/docs/build/html/python/_autosummary/mlx.core.array.sum.html index ba947b71ef..435dc542b6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.sum.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.sum.html @@ -8,7 +8,7 @@ - mlx.core.array.sum — MLX 0.16.1 documentation + mlx.core.array.sum — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html b/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html index 9bd63f45cc..e4da3856d1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html @@ -8,7 +8,7 @@ - mlx.core.array.swapaxes — MLX 0.16.1 documentation + mlx.core.array.swapaxes — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.tolist.html b/docs/build/html/python/_autosummary/mlx.core.array.tolist.html index d9eb275d82..dbc3e71fbd 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.tolist.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.tolist.html @@ -8,7 +8,7 @@ - mlx.core.array.tolist — MLX 0.16.1 documentation + mlx.core.array.tolist — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.transpose.html b/docs/build/html/python/_autosummary/mlx.core.array.transpose.html index 5c825776d4..dbeb1343c7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.transpose.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.transpose.html @@ -8,7 +8,7 @@ - mlx.core.array.transpose — MLX 0.16.1 documentation + mlx.core.array.transpose — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.var.html b/docs/build/html/python/_autosummary/mlx.core.array.var.html index fbcfb302e0..287d8baed3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.var.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.var.html @@ -8,7 +8,7 @@ - mlx.core.array.var — MLX 0.16.1 documentation + mlx.core.array.var — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array.view.html b/docs/build/html/python/_autosummary/mlx.core.array.view.html index d01f75ef6a..d540c61794 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.view.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.view.html @@ -8,7 +8,7 @@ - mlx.core.array.view — MLX 0.16.1 documentation + mlx.core.array.view — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.array_equal.html b/docs/build/html/python/_autosummary/mlx.core.array_equal.html index e230633f3c..6c6b283f98 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array_equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.array_equal.html @@ -8,7 +8,7 @@ - mlx.core.array_equal — MLX 0.16.1 documentation + mlx.core.array_equal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.as_strided.html b/docs/build/html/python/_autosummary/mlx.core.as_strided.html index 7a95e96f8f..91dc8b8f90 100644 --- a/docs/build/html/python/_autosummary/mlx.core.as_strided.html +++ b/docs/build/html/python/_autosummary/mlx.core.as_strided.html @@ -8,7 +8,7 @@ - mlx.core.as_strided — MLX 0.16.1 documentation + mlx.core.as_strided — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html index 49f35c0ae9..5f4ae9420f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html +++ b/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html @@ -8,7 +8,7 @@ - mlx.core.atleast_1d — MLX 0.16.1 documentation + mlx.core.atleast_1d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html index aec628a4c6..e74c684b4b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html +++ b/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html @@ -8,7 +8,7 @@ - mlx.core.atleast_2d — MLX 0.16.1 documentation + mlx.core.atleast_2d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html index 4e0c9a9c8e..bd6027915e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html +++ b/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html @@ -8,7 +8,7 @@ - mlx.core.atleast_3d — MLX 0.16.1 documentation + mlx.core.atleast_3d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html index 232760d85a..d3595a0b10 100644 --- a/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html +++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html @@ -8,7 +8,7 @@ - mlx.core.bitwise_and — MLX 0.16.1 documentation + mlx.core.bitwise_and — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html index 7cb7ccc43a..fd1d8bc5a2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html +++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html @@ -8,7 +8,7 @@ - mlx.core.bitwise_or — MLX 0.16.1 documentation + mlx.core.bitwise_or — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html index 6c1e7f840b..4f6b80423c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html +++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html @@ -8,7 +8,7 @@ - mlx.core.bitwise_xor — MLX 0.16.1 documentation + mlx.core.bitwise_xor — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html b/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html index 38f5fc8c69..eec4e48f52 100644 --- a/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html +++ b/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html @@ -8,7 +8,7 @@ - mlx.core.block_masked_mm — MLX 0.16.1 documentation + mlx.core.block_masked_mm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html b/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html index e195780f3f..e547ce84a5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html +++ b/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html @@ -8,7 +8,7 @@ - mlx.core.broadcast_to — MLX 0.16.1 documentation + mlx.core.broadcast_to — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.ceil.html b/docs/build/html/python/_autosummary/mlx.core.ceil.html index 303647e190..9b95481447 100644 --- a/docs/build/html/python/_autosummary/mlx.core.ceil.html +++ b/docs/build/html/python/_autosummary/mlx.core.ceil.html @@ -8,7 +8,7 @@ - mlx.core.ceil — MLX 0.16.1 documentation + mlx.core.ceil — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.clip.html b/docs/build/html/python/_autosummary/mlx.core.clip.html index 866979a34b..c611724210 100644 --- a/docs/build/html/python/_autosummary/mlx.core.clip.html +++ b/docs/build/html/python/_autosummary/mlx.core.clip.html @@ -8,7 +8,7 @@ - mlx.core.clip — MLX 0.16.1 documentation + mlx.core.clip — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.compile.html b/docs/build/html/python/_autosummary/mlx.core.compile.html index bca6bb2c2e..e06e383c52 100644 --- a/docs/build/html/python/_autosummary/mlx.core.compile.html +++ b/docs/build/html/python/_autosummary/mlx.core.compile.html @@ -8,7 +8,7 @@ - mlx.core.compile — MLX 0.16.1 documentation + mlx.core.compile — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.concatenate.html b/docs/build/html/python/_autosummary/mlx.core.concatenate.html index 88f4b6c377..b790125092 100644 --- a/docs/build/html/python/_autosummary/mlx.core.concatenate.html +++ b/docs/build/html/python/_autosummary/mlx.core.concatenate.html @@ -8,7 +8,7 @@ - mlx.core.concatenate — MLX 0.16.1 documentation + mlx.core.concatenate — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.conj.html b/docs/build/html/python/_autosummary/mlx.core.conj.html index 26ebae2f8a..aaf50f0a81 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conj.html +++ b/docs/build/html/python/_autosummary/mlx.core.conj.html @@ -8,7 +8,7 @@ - mlx.core.conj — MLX 0.16.1 documentation + mlx.core.conj — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.conjugate.html b/docs/build/html/python/_autosummary/mlx.core.conjugate.html index 9a0959b1b3..9261dcb419 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conjugate.html +++ b/docs/build/html/python/_autosummary/mlx.core.conjugate.html @@ -8,7 +8,7 @@ - mlx.core.conjugate — MLX 0.16.1 documentation + mlx.core.conjugate — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.conv1d.html b/docs/build/html/python/_autosummary/mlx.core.conv1d.html index 493d455d93..a4cf3521c7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv1d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv1d.html @@ -8,7 +8,7 @@ - mlx.core.conv1d — MLX 0.16.1 documentation + mlx.core.conv1d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.conv2d.html b/docs/build/html/python/_autosummary/mlx.core.conv2d.html index f4bf1351d1..de4e6d5c29 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv2d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv2d.html @@ -8,7 +8,7 @@ - mlx.core.conv2d — MLX 0.16.1 documentation + mlx.core.conv2d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_general.html b/docs/build/html/python/_autosummary/mlx.core.conv_general.html index 629625594c..a22c1ad7e3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv_general.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv_general.html @@ -8,7 +8,7 @@ - mlx.core.conv_general — MLX 0.16.1 documentation + mlx.core.conv_general — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.convolve.html b/docs/build/html/python/_autosummary/mlx.core.convolve.html index 19a1870e0e..d14158f502 100644 --- a/docs/build/html/python/_autosummary/mlx.core.convolve.html +++ b/docs/build/html/python/_autosummary/mlx.core.convolve.html @@ -8,7 +8,7 @@ - mlx.core.convolve — MLX 0.16.1 documentation + mlx.core.convolve — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.cos.html b/docs/build/html/python/_autosummary/mlx.core.cos.html index fdd1b84581..6643cc08ca 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cos.html +++ b/docs/build/html/python/_autosummary/mlx.core.cos.html @@ -8,7 +8,7 @@ - mlx.core.cos — MLX 0.16.1 documentation + mlx.core.cos — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.cosh.html b/docs/build/html/python/_autosummary/mlx.core.cosh.html index 2ba021b08d..0de0c4f9cf 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cosh.html +++ b/docs/build/html/python/_autosummary/mlx.core.cosh.html @@ -8,7 +8,7 @@ - mlx.core.cosh — MLX 0.16.1 documentation + mlx.core.cosh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.cummax.html b/docs/build/html/python/_autosummary/mlx.core.cummax.html index ebcb72beb2..e2ac768663 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cummax.html +++ b/docs/build/html/python/_autosummary/mlx.core.cummax.html @@ -8,7 +8,7 @@ - mlx.core.cummax — MLX 0.16.1 documentation + mlx.core.cummax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.cummin.html b/docs/build/html/python/_autosummary/mlx.core.cummin.html index 2a8d10b1d2..743131d9ce 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cummin.html +++ b/docs/build/html/python/_autosummary/mlx.core.cummin.html @@ -8,7 +8,7 @@ - mlx.core.cummin — MLX 0.16.1 documentation + mlx.core.cummin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.cumprod.html b/docs/build/html/python/_autosummary/mlx.core.cumprod.html index 820c196931..cd16066348 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cumprod.html +++ b/docs/build/html/python/_autosummary/mlx.core.cumprod.html @@ -8,7 +8,7 @@ - mlx.core.cumprod — MLX 0.16.1 documentation + mlx.core.cumprod — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.cumsum.html b/docs/build/html/python/_autosummary/mlx.core.cumsum.html index 8d6bbe454c..c404ce61c8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cumsum.html +++ b/docs/build/html/python/_autosummary/mlx.core.cumsum.html @@ -8,7 +8,7 @@ - mlx.core.cumsum — MLX 0.16.1 documentation + mlx.core.cumsum — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.custom_function.html b/docs/build/html/python/_autosummary/mlx.core.custom_function.html index aceedc4a96..3e8a43a62e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.custom_function.html +++ b/docs/build/html/python/_autosummary/mlx.core.custom_function.html @@ -8,7 +8,7 @@ - mlx.core.custom_function — MLX 0.16.1 documentation + mlx.core.custom_function — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.default_device.html b/docs/build/html/python/_autosummary/mlx.core.default_device.html index 289ea85128..c68a5784f2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.default_device.html +++ b/docs/build/html/python/_autosummary/mlx.core.default_device.html @@ -8,7 +8,7 @@ - mlx.core.default_device — MLX 0.16.1 documentation + mlx.core.default_device — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.default_stream.html b/docs/build/html/python/_autosummary/mlx.core.default_stream.html index 3785f39609..6ddefdd992 100644 --- a/docs/build/html/python/_autosummary/mlx.core.default_stream.html +++ b/docs/build/html/python/_autosummary/mlx.core.default_stream.html @@ -8,7 +8,7 @@ - mlx.core.default_stream — MLX 0.16.1 documentation + mlx.core.default_stream — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.degrees.html b/docs/build/html/python/_autosummary/mlx.core.degrees.html index 7e43552d09..661b7d51af 100644 --- a/docs/build/html/python/_autosummary/mlx.core.degrees.html +++ b/docs/build/html/python/_autosummary/mlx.core.degrees.html @@ -8,7 +8,7 @@ - mlx.core.degrees — MLX 0.16.1 documentation + mlx.core.degrees — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.dequantize.html b/docs/build/html/python/_autosummary/mlx.core.dequantize.html index 7b90b59e1c..e95f07cbab 100644 --- a/docs/build/html/python/_autosummary/mlx.core.dequantize.html +++ b/docs/build/html/python/_autosummary/mlx.core.dequantize.html @@ -8,7 +8,7 @@ - mlx.core.dequantize — MLX 0.16.1 documentation + mlx.core.dequantize — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.diag.html b/docs/build/html/python/_autosummary/mlx.core.diag.html index 38d5c5cef8..6fa05ccae1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.diag.html +++ b/docs/build/html/python/_autosummary/mlx.core.diag.html @@ -8,7 +8,7 @@ - mlx.core.diag — MLX 0.16.1 documentation + mlx.core.diag — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.diagonal.html b/docs/build/html/python/_autosummary/mlx.core.diagonal.html index ee0b125d6d..ad262fae4a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.diagonal.html +++ b/docs/build/html/python/_autosummary/mlx.core.diagonal.html @@ -8,7 +8,7 @@ - mlx.core.diagonal — MLX 0.16.1 documentation + mlx.core.diagonal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.disable_compile.html b/docs/build/html/python/_autosummary/mlx.core.disable_compile.html index 724855bc01..336b4c8abe 100644 --- a/docs/build/html/python/_autosummary/mlx.core.disable_compile.html +++ b/docs/build/html/python/_autosummary/mlx.core.disable_compile.html @@ -8,7 +8,7 @@ - mlx.core.disable_compile — MLX 0.16.1 documentation + mlx.core.disable_compile — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html b/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html index dfc8231fe5..641c594384 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html @@ -8,7 +8,7 @@ - mlx.core.distributed.Group — MLX 0.16.1 documentation + mlx.core.distributed.Group — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html b/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html index c4b0a295d6..2286d14c58 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html @@ -8,7 +8,7 @@ - mlx.core.distributed.all_gather — MLX 0.16.1 documentation + mlx.core.distributed.all_gather — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html b/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html index 7d7e525735..deae412ca4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html @@ -8,7 +8,7 @@ - mlx.core.distributed.all_sum — MLX 0.16.1 documentation + mlx.core.distributed.all_sum — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.init.html b/docs/build/html/python/_autosummary/mlx.core.distributed.init.html index 71b0b064a0..bd345cbedf 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.init.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.init.html @@ -8,7 +8,7 @@ - mlx.core.distributed.init — MLX 0.16.1 documentation + mlx.core.distributed.init — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html b/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html index b1f637a72d..8f1b70e22d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html @@ -8,7 +8,7 @@ - mlx.core.distributed.is_available — MLX 0.16.1 documentation + mlx.core.distributed.is_available — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.divide.html b/docs/build/html/python/_autosummary/mlx.core.divide.html index 3c07cbe644..a98f3c789b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.divide.html +++ b/docs/build/html/python/_autosummary/mlx.core.divide.html @@ -8,7 +8,7 @@ - mlx.core.divide — MLX 0.16.1 documentation + mlx.core.divide — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.divmod.html b/docs/build/html/python/_autosummary/mlx.core.divmod.html index fc4f3a1af6..8b889cc2c6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.divmod.html +++ b/docs/build/html/python/_autosummary/mlx.core.divmod.html @@ -8,7 +8,7 @@ - mlx.core.divmod — MLX 0.16.1 documentation + mlx.core.divmod — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.einsum.html b/docs/build/html/python/_autosummary/mlx.core.einsum.html index f52dfa57cb..1978c048f4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.einsum.html +++ b/docs/build/html/python/_autosummary/mlx.core.einsum.html @@ -8,7 +8,7 @@ - mlx.core.einsum — MLX 0.16.1 documentation + mlx.core.einsum — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.einsum_path.html b/docs/build/html/python/_autosummary/mlx.core.einsum_path.html index d4415064e7..eab54376a3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.einsum_path.html +++ b/docs/build/html/python/_autosummary/mlx.core.einsum_path.html @@ -8,7 +8,7 @@ - mlx.core.einsum_path — MLX 0.16.1 documentation + mlx.core.einsum_path — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.enable_compile.html b/docs/build/html/python/_autosummary/mlx.core.enable_compile.html index 7bc670d3f9..58e8cd4018 100644 --- a/docs/build/html/python/_autosummary/mlx.core.enable_compile.html +++ b/docs/build/html/python/_autosummary/mlx.core.enable_compile.html @@ -8,7 +8,7 @@ - mlx.core.enable_compile — MLX 0.16.1 documentation + mlx.core.enable_compile — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.equal.html b/docs/build/html/python/_autosummary/mlx.core.equal.html index 510ef15d30..1c6fdeec8f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.equal.html @@ -8,7 +8,7 @@ - mlx.core.equal — MLX 0.16.1 documentation + mlx.core.equal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.erf.html b/docs/build/html/python/_autosummary/mlx.core.erf.html index b14b4a620a..52c2badf3c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.erf.html +++ b/docs/build/html/python/_autosummary/mlx.core.erf.html @@ -8,7 +8,7 @@ - mlx.core.erf — MLX 0.16.1 documentation + mlx.core.erf — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.erfinv.html b/docs/build/html/python/_autosummary/mlx.core.erfinv.html index 308765781a..390575dd4d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.erfinv.html +++ b/docs/build/html/python/_autosummary/mlx.core.erfinv.html @@ -8,7 +8,7 @@ - mlx.core.erfinv — MLX 0.16.1 documentation + mlx.core.erfinv — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.eval.html b/docs/build/html/python/_autosummary/mlx.core.eval.html index 04ed64a0fe..e7d34a3fe7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.eval.html +++ b/docs/build/html/python/_autosummary/mlx.core.eval.html @@ -8,7 +8,7 @@ - mlx.core.eval — MLX 0.16.1 documentation + mlx.core.eval — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.exp.html b/docs/build/html/python/_autosummary/mlx.core.exp.html index ca92951434..b20d6276a8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.exp.html +++ b/docs/build/html/python/_autosummary/mlx.core.exp.html @@ -8,7 +8,7 @@ - mlx.core.exp — MLX 0.16.1 documentation + mlx.core.exp — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.expand_dims.html b/docs/build/html/python/_autosummary/mlx.core.expand_dims.html index 90e85091f1..fd586a58c1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.expand_dims.html +++ b/docs/build/html/python/_autosummary/mlx.core.expand_dims.html @@ -8,7 +8,7 @@ - mlx.core.expand_dims — MLX 0.16.1 documentation + mlx.core.expand_dims — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.expm1.html b/docs/build/html/python/_autosummary/mlx.core.expm1.html index a3a456b39a..5a46b64d0f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.expm1.html +++ b/docs/build/html/python/_autosummary/mlx.core.expm1.html @@ -8,7 +8,7 @@ - mlx.core.expm1 — MLX 0.16.1 documentation + mlx.core.expm1 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.eye.html b/docs/build/html/python/_autosummary/mlx.core.eye.html index e3f467cd6e..904945fe5f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.eye.html +++ b/docs/build/html/python/_autosummary/mlx.core.eye.html @@ -8,7 +8,7 @@ - mlx.core.eye — MLX 0.16.1 documentation + mlx.core.eye — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.layer_norm.html b/docs/build/html/python/_autosummary/mlx.core.fast.layer_norm.html index 57a9852ad7..ec953f465c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.layer_norm.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.layer_norm.html @@ -8,7 +8,7 @@ - mlx.core.fast.layer_norm — MLX 0.16.1 documentation + mlx.core.fast.layer_norm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.rms_norm.html b/docs/build/html/python/_autosummary/mlx.core.fast.rms_norm.html index baacc6193b..94beae23b4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.rms_norm.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.rms_norm.html @@ -8,7 +8,7 @@ - mlx.core.fast.rms_norm — MLX 0.16.1 documentation + mlx.core.fast.rms_norm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.rope.html b/docs/build/html/python/_autosummary/mlx.core.fast.rope.html index 1e4c80dc0b..389fd49891 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.rope.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.rope.html @@ -8,7 +8,7 @@ - mlx.core.fast.rope — MLX 0.16.1 documentation + mlx.core.fast.rope — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html b/docs/build/html/python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html index 72c3cebe15..47a7266798 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.scaled_dot_product_attention.html @@ -8,7 +8,7 @@ - mlx.core.fast.scaled_dot_product_attention — MLX 0.16.1 documentation + mlx.core.fast.scaled_dot_product_attention — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.fft.html b/docs/build/html/python/_autosummary/mlx.core.fft.fft.html index 28f608ccb2..736dd6200f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.fft.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.fft.html @@ -8,7 +8,7 @@ - mlx.core.fft.fft — MLX 0.16.1 documentation + mlx.core.fft.fft — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.fft2.html b/docs/build/html/python/_autosummary/mlx.core.fft.fft2.html index 44c5f8f283..863be49f32 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.fft2.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.fft2.html @@ -8,7 +8,7 @@ - mlx.core.fft.fft2 — MLX 0.16.1 documentation + mlx.core.fft.fft2 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.fftn.html b/docs/build/html/python/_autosummary/mlx.core.fft.fftn.html index f01397a481..c9ced9e4ac 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.fftn.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.fftn.html @@ -8,7 +8,7 @@ - mlx.core.fft.fftn — MLX 0.16.1 documentation + mlx.core.fft.fftn — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.ifft.html b/docs/build/html/python/_autosummary/mlx.core.fft.ifft.html index b026d6658d..84f28aa3f9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.ifft.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.ifft.html @@ -8,7 +8,7 @@ - mlx.core.fft.ifft — MLX 0.16.1 documentation + mlx.core.fft.ifft — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.ifft2.html b/docs/build/html/python/_autosummary/mlx.core.fft.ifft2.html index 5665771271..2335c33502 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.ifft2.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.ifft2.html @@ -8,7 +8,7 @@ - mlx.core.fft.ifft2 — MLX 0.16.1 documentation + mlx.core.fft.ifft2 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.ifftn.html b/docs/build/html/python/_autosummary/mlx.core.fft.ifftn.html index b9d6886f90..0af33d28e2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.ifftn.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.ifftn.html @@ -8,7 +8,7 @@ - mlx.core.fft.ifftn — MLX 0.16.1 documentation + mlx.core.fft.ifftn — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.irfft.html b/docs/build/html/python/_autosummary/mlx.core.fft.irfft.html index 0185644329..7f2f1f2777 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.irfft.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.irfft.html @@ -8,7 +8,7 @@ - mlx.core.fft.irfft — MLX 0.16.1 documentation + mlx.core.fft.irfft — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.irfft2.html b/docs/build/html/python/_autosummary/mlx.core.fft.irfft2.html index e043005df9..77d6ad26f0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.irfft2.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.irfft2.html @@ -8,7 +8,7 @@ - mlx.core.fft.irfft2 — MLX 0.16.1 documentation + mlx.core.fft.irfft2 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.irfftn.html b/docs/build/html/python/_autosummary/mlx.core.fft.irfftn.html index 90a12dfda3..8e1f88aebb 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.irfftn.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.irfftn.html @@ -8,7 +8,7 @@ - mlx.core.fft.irfftn — MLX 0.16.1 documentation + mlx.core.fft.irfftn — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.rfft.html b/docs/build/html/python/_autosummary/mlx.core.fft.rfft.html index e366f0c32a..e6f457d2f4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.rfft.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.rfft.html @@ -8,7 +8,7 @@ - mlx.core.fft.rfft — MLX 0.16.1 documentation + mlx.core.fft.rfft — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.rfft2.html b/docs/build/html/python/_autosummary/mlx.core.fft.rfft2.html index 93dee08af1..89ab060263 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.rfft2.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.rfft2.html @@ -8,7 +8,7 @@ - mlx.core.fft.rfft2 — MLX 0.16.1 documentation + mlx.core.fft.rfft2 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.rfftn.html b/docs/build/html/python/_autosummary/mlx.core.fft.rfftn.html index 3b038606af..21355f16b7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.rfftn.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.rfftn.html @@ -8,7 +8,7 @@ - mlx.core.fft.rfftn — MLX 0.16.1 documentation + mlx.core.fft.rfftn — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.flatten.html b/docs/build/html/python/_autosummary/mlx.core.flatten.html index ab5ff39e59..6e5878e2b4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.flatten.html +++ b/docs/build/html/python/_autosummary/mlx.core.flatten.html @@ -8,7 +8,7 @@ - mlx.core.flatten — MLX 0.16.1 documentation + mlx.core.flatten — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.floor.html b/docs/build/html/python/_autosummary/mlx.core.floor.html index 53e7413dfd..e8acfa9953 100644 --- a/docs/build/html/python/_autosummary/mlx.core.floor.html +++ b/docs/build/html/python/_autosummary/mlx.core.floor.html @@ -8,7 +8,7 @@ - mlx.core.floor — MLX 0.16.1 documentation + mlx.core.floor — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.floor_divide.html b/docs/build/html/python/_autosummary/mlx.core.floor_divide.html index eb399569fc..481e298b40 100644 --- a/docs/build/html/python/_autosummary/mlx.core.floor_divide.html +++ b/docs/build/html/python/_autosummary/mlx.core.floor_divide.html @@ -8,7 +8,7 @@ - mlx.core.floor_divide — MLX 0.16.1 documentation + mlx.core.floor_divide — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.full.html b/docs/build/html/python/_autosummary/mlx.core.full.html index 3dbcf7ff8d..c8ab7913f0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.full.html +++ b/docs/build/html/python/_autosummary/mlx.core.full.html @@ -8,7 +8,7 @@ - mlx.core.full — MLX 0.16.1 documentation + mlx.core.full — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.gather_mm.html b/docs/build/html/python/_autosummary/mlx.core.gather_mm.html index 40a749acf5..d1cb1350ea 100644 --- a/docs/build/html/python/_autosummary/mlx.core.gather_mm.html +++ b/docs/build/html/python/_autosummary/mlx.core.gather_mm.html @@ -8,7 +8,7 @@ - mlx.core.gather_mm — MLX 0.16.1 documentation + mlx.core.gather_mm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.gather_qmm.html b/docs/build/html/python/_autosummary/mlx.core.gather_qmm.html index 10339eebd2..7b967d2141 100644 --- a/docs/build/html/python/_autosummary/mlx.core.gather_qmm.html +++ b/docs/build/html/python/_autosummary/mlx.core.gather_qmm.html @@ -8,7 +8,7 @@ - mlx.core.gather_qmm — MLX 0.16.1 documentation + mlx.core.gather_qmm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.grad.html b/docs/build/html/python/_autosummary/mlx.core.grad.html index 3280cce3ff..7d63a80c45 100644 --- a/docs/build/html/python/_autosummary/mlx.core.grad.html +++ b/docs/build/html/python/_autosummary/mlx.core.grad.html @@ -8,7 +8,7 @@ - mlx.core.grad — MLX 0.16.1 documentation + mlx.core.grad — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.greater.html b/docs/build/html/python/_autosummary/mlx.core.greater.html index 839d448e21..b00d023a00 100644 --- a/docs/build/html/python/_autosummary/mlx.core.greater.html +++ b/docs/build/html/python/_autosummary/mlx.core.greater.html @@ -8,7 +8,7 @@ - mlx.core.greater — MLX 0.16.1 documentation + mlx.core.greater — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.greater_equal.html b/docs/build/html/python/_autosummary/mlx.core.greater_equal.html index a7414d144c..3037b0f75d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.greater_equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.greater_equal.html @@ -8,7 +8,7 @@ - mlx.core.greater_equal — MLX 0.16.1 documentation + mlx.core.greater_equal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html b/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html index bc2e802807..c8ae11fbd2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html +++ b/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html @@ -8,7 +8,7 @@ - mlx.core.hadamard_transform — MLX 0.16.1 documentation + mlx.core.hadamard_transform — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.identity.html b/docs/build/html/python/_autosummary/mlx.core.identity.html index c387f8b7ed..4006ffba25 100644 --- a/docs/build/html/python/_autosummary/mlx.core.identity.html +++ b/docs/build/html/python/_autosummary/mlx.core.identity.html @@ -8,7 +8,7 @@ - mlx.core.identity — MLX 0.16.1 documentation + mlx.core.identity — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.inner.html b/docs/build/html/python/_autosummary/mlx.core.inner.html index f13e4d11f5..f7d17ce5e5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.inner.html +++ b/docs/build/html/python/_autosummary/mlx.core.inner.html @@ -8,7 +8,7 @@ - mlx.core.inner — MLX 0.16.1 documentation + mlx.core.inner — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.isclose.html b/docs/build/html/python/_autosummary/mlx.core.isclose.html index abb0566bc0..1e676a6a7a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isclose.html +++ b/docs/build/html/python/_autosummary/mlx.core.isclose.html @@ -8,7 +8,7 @@ - mlx.core.isclose — MLX 0.16.1 documentation + mlx.core.isclose — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.isinf.html b/docs/build/html/python/_autosummary/mlx.core.isinf.html index f465f25bc2..8054be5608 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isinf.html +++ b/docs/build/html/python/_autosummary/mlx.core.isinf.html @@ -8,7 +8,7 @@ - mlx.core.isinf — MLX 0.16.1 documentation + mlx.core.isinf — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.isnan.html b/docs/build/html/python/_autosummary/mlx.core.isnan.html index 997249e91c..14df8e6c0a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isnan.html +++ b/docs/build/html/python/_autosummary/mlx.core.isnan.html @@ -8,7 +8,7 @@ - mlx.core.isnan — MLX 0.16.1 documentation + mlx.core.isnan — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.isneginf.html b/docs/build/html/python/_autosummary/mlx.core.isneginf.html index c64637e42d..e7091a78dd 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isneginf.html +++ b/docs/build/html/python/_autosummary/mlx.core.isneginf.html @@ -8,7 +8,7 @@ - mlx.core.isneginf — MLX 0.16.1 documentation + mlx.core.isneginf — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.isposinf.html b/docs/build/html/python/_autosummary/mlx.core.isposinf.html index 0944c9da5e..f805e1e9bd 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isposinf.html +++ b/docs/build/html/python/_autosummary/mlx.core.isposinf.html @@ -8,7 +8,7 @@ - mlx.core.isposinf — MLX 0.16.1 documentation + mlx.core.isposinf — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.issubdtype.html b/docs/build/html/python/_autosummary/mlx.core.issubdtype.html index bc506704a4..3db7875dff 100644 --- a/docs/build/html/python/_autosummary/mlx.core.issubdtype.html +++ b/docs/build/html/python/_autosummary/mlx.core.issubdtype.html @@ -8,7 +8,7 @@ - mlx.core.issubdtype — MLX 0.16.1 documentation + mlx.core.issubdtype — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.jvp.html b/docs/build/html/python/_autosummary/mlx.core.jvp.html index 70f90d7f40..313cde1376 100644 --- a/docs/build/html/python/_autosummary/mlx.core.jvp.html +++ b/docs/build/html/python/_autosummary/mlx.core.jvp.html @@ -8,7 +8,7 @@ - mlx.core.jvp — MLX 0.16.1 documentation + mlx.core.jvp — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.left_shift.html b/docs/build/html/python/_autosummary/mlx.core.left_shift.html index 7011b396c5..6b047422f3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.left_shift.html +++ b/docs/build/html/python/_autosummary/mlx.core.left_shift.html @@ -8,7 +8,7 @@ - mlx.core.left_shift — MLX 0.16.1 documentation + mlx.core.left_shift — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.less.html b/docs/build/html/python/_autosummary/mlx.core.less.html index 41310b195b..95df89d101 100644 --- a/docs/build/html/python/_autosummary/mlx.core.less.html +++ b/docs/build/html/python/_autosummary/mlx.core.less.html @@ -8,7 +8,7 @@ - mlx.core.less — MLX 0.16.1 documentation + mlx.core.less — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.less_equal.html b/docs/build/html/python/_autosummary/mlx.core.less_equal.html index fc300c15de..4a6a4cce36 100644 --- a/docs/build/html/python/_autosummary/mlx.core.less_equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.less_equal.html @@ -8,7 +8,7 @@ - mlx.core.less_equal — MLX 0.16.1 documentation + mlx.core.less_equal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html index cf49cfa376..17b1783e0b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html @@ -8,7 +8,7 @@ - mlx.core.linalg.cholesky — MLX 0.16.1 documentation + mlx.core.linalg.cholesky — MLX 0.16.2 documentation @@ -36,14 +36,14 @@ - + - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra @@ -863,7 +865,7 @@ in which case the default stream of the default device is used.

            • Returns:
            -

            If upper = False, it returns a lower trinagular L matrix such +

            If upper = False, it returns a lower triangular L matrix such that dot(L, L.T) = a. If upper = True, it returns an upper triangular U matrix such that dot(U.T, U) = a.

            @@ -896,11 +898,11 @@ that dot(L,

            next

            -

            mlx.core.linalg.qr

            +

            mlx.core.linalg.cholesky_inv

             diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html new file mode 100644 index 0000000000..7596dfb885 --- /dev/null +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html @@ -0,0 +1,984 @@ + + + + + + + + + + + mlx.core.linalg.cholesky_inv — MLX 0.16.2 documentation + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
            + + + + + + + + + + + +
            +
            +
            +
            + + + Ctrl+K +
            +
            + + + +
            +
            + +
            + + + +
            + + + + +
            + +
            +
            + + + +
            +
            + +
            +
            +
            + + +
            +
            + + + + +
            + +
            + + + +
            + +
            +
            + +
            +
            + +
            + +
            + +
            + + +
            + +
            + +
            + + + + + + + + + + + + + + + + +
            + + +
            + + + + + + + + + + + + + +
            + +
            + +
            +
            + + + +
            +

            mlx.core.linalg.cholesky_inv

            + +
            +
            + +
            +

            Contents

            +
            + +
            +
            +
            + + + + +
            + +
            +

            mlx.core.linalg.cholesky_inv#

            +
            +
            +cholesky_inv(L: array, upper: bool = False, *, stream: None | Stream | Device = None) → array#
            +

            Compute the inverse of a real symmetric positive semi-definite matrix using it’s Cholesky decomposition L.

            +

            Let A be a real symmetric positive semi-definite matrix and L its Cholesky definition such that:

            +
            +\[\begin{aligned} + \mathbf{A} = \mathbf{L}\mathbf{L}^T +\end{aligned}\]
            +

            This function computes \(\mathbf{A}^{-1}\).

            +

            This function supports arrays with at least 2 dimensions. When the input +has more than two dimensions, the Cholesky inverse is computed for each matrix +in the last two dimensions of L.

            +

            If the input matrix is not a triangular matrix behaviour is undefined.

            +
            +
            Parameters:
            +
              +

            • L (array) – Input array.

              • +

            • upper (bool, optional) – If True, return the upper triangular Cholesky factor. +If False, return the lower triangular Cholesky factor. Default: False.

              • +

            • stream (Stream, optional) – Stream or device. Defaults to None +in which case the default stream of the default device is used.

              • +
            +
            +
            Returns:
            +

            \(A^{-1}\) where \(\mathbf{A} = \mathbf{L}\mathbf{L}^T\).

            +
            +
            Return type:
            +

            array

            +
            +
            +
            + +
            + + +
            + + + + + + + + +
            + + + +
            + +
            +
            + Contents +
            +
            + +
            + + +
            +
            + +
            + +
            + +

            +By MLX Contributors +

            + +
            + +
            + + +

            + + © Copyright 2023, MLX Contributors. +
            + +

            + +
            + +
            + +
            + +
            + +
            + +
            +
            + + +
            +
            +
            + + + + + +
            +
            + + \ No newline at end of file diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html index 29a300697d..8cfbcb2411 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html @@ -8,7 +8,7 @@ - mlx.core.linalg.inv — MLX 0.16.1 documentation + mlx.core.linalg.inv — MLX 0.16.2 documentation @@ -36,14 +36,14 @@ - + - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra @@ -891,11 +893,11 @@ in which case the default stream of the default device is used.

            • next

            -

            mlx.core.linalg.norm

            +

            mlx.core.linalg.tri_inv

             diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html b/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html index b41acbd92e..998090955f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html @@ -8,7 +8,7 @@ - mlx.core.linalg.norm — MLX 0.16.1 documentation + mlx.core.linalg.norm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -45,7 +45,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra @@ -1021,12 +1023,12 @@ Baltimore, MD, Johns Hopkins University Press, 1985, pg. 15

            previous

            -

            mlx.core.linalg.inv

            +

            mlx.core.linalg.tri_inv

             - mlx.core.linalg.qr — MLX 0.16.1 documentation + mlx.core.linalg.qr — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -44,7 +44,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home +
            @@ -457,8 +457,10 @@
          • Linear Algebra @@ -893,12 +895,12 @@ in which case the default stream of the default device is used.

            • previous

            -

            mlx.core.linalg.cholesky

            +

            mlx.core.linalg.cholesky_inv

             - mlx.core.linalg.svd — MLX 0.16.1 documentation + mlx.core.linalg.svd — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home +
            @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html new file mode 100644 index 0000000000..b0f02099f4 --- /dev/null +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html @@ -0,0 +1,975 @@ + + + + + + + + + + + mlx.core.linalg.tri_inv — MLX 0.16.2 documentation + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
            + + + + + + + + + + + +
            +
            +
            +
            + + + Ctrl+K +
            +
            + + + +
            +
            + +
            + + + +
            + + + + +
            + +
            +
            + + + +
            +
            + +
            +
            +
            + + +
            +
            + + + + +
            + +
            + + + +
            + +
            +
            + +
            +
            + +
            + +
            + +
            + + +
            + +
            + +
            + + + + + + + + + + + + + + + + +
            + + +
            + + + + + + + + + + + + + +
            + +
            + +
            +
            + + + +
            +

            mlx.core.linalg.tri_inv

            + +
            +
            + +
            +

            Contents

            +
            + +
            +
            +
            + + + + +
            + +
            +

            mlx.core.linalg.tri_inv#

            +
            +
            +tri_inv(a: array, upper: bool = False, *, stream: None | Stream | Device = None) → array#
            +

            Compute the inverse of a triangular square matrix.

            +

            This function supports arrays with at least 2 dimensions. When the input +has more than two dimensions, the inverse is computed for each matrix +in the last two dimensions of a.

            +
            +
            Parameters:
            +
              +

            • a (array) – Input array.

              • +

            • upper (array) – Whether the array is upper or lower triangular. Defaults to False.

              • +

            • stream (Stream, optional) – Stream or device. Defaults to None +in which case the default stream of the default device is used.

              • +
            +
            +
            Returns:
            +

            ainv such that dot(a, ainv) = dot(ainv, a) = eye(a.shape[0])

            +
            +
            Return type:
            +

            array

            +
            +
            +
            + +
            + + +
            + + + + + + + + +
            + + + +
            + +
            +
            + Contents +
            +
            + +
            + + +
            +
            + +
            + +
            + +

            +By MLX Contributors +

            + +
            + +
            + + +

            + + © Copyright 2023, MLX Contributors. +
            + +

            + +
            + +
            + +
            + +
            + +
            + +
            +
            + + +
            +
            +
            + + + + + +
            +
            + + \ No newline at end of file diff --git a/docs/build/html/python/_autosummary/mlx.core.linspace.html b/docs/build/html/python/_autosummary/mlx.core.linspace.html index b17ade6342..4d46708f43 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linspace.html +++ b/docs/build/html/python/_autosummary/mlx.core.linspace.html @@ -8,7 +8,7 @@ - mlx.core.linspace — MLX 0.16.1 documentation + mlx.core.linspace — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.load.html b/docs/build/html/python/_autosummary/mlx.core.load.html index 0abeaaaacf..cbd78b4e07 100644 --- a/docs/build/html/python/_autosummary/mlx.core.load.html +++ b/docs/build/html/python/_autosummary/mlx.core.load.html @@ -8,7 +8,7 @@ - mlx.core.load — MLX 0.16.1 documentation + mlx.core.load — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.log.html b/docs/build/html/python/_autosummary/mlx.core.log.html index 2ccfa05db7..fff3d4c843 100644 --- a/docs/build/html/python/_autosummary/mlx.core.log.html +++ b/docs/build/html/python/_autosummary/mlx.core.log.html @@ -8,7 +8,7 @@ - mlx.core.log — MLX 0.16.1 documentation + mlx.core.log — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.log10.html b/docs/build/html/python/_autosummary/mlx.core.log10.html index 99f4242852..f86e90d819 100644 --- a/docs/build/html/python/_autosummary/mlx.core.log10.html +++ b/docs/build/html/python/_autosummary/mlx.core.log10.html @@ -8,7 +8,7 @@ - mlx.core.log10 — MLX 0.16.1 documentation + mlx.core.log10 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.log1p.html b/docs/build/html/python/_autosummary/mlx.core.log1p.html index e3117d7fd1..4ddacad11f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.log1p.html +++ b/docs/build/html/python/_autosummary/mlx.core.log1p.html @@ -8,7 +8,7 @@ - mlx.core.log1p — MLX 0.16.1 documentation + mlx.core.log1p — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.log2.html b/docs/build/html/python/_autosummary/mlx.core.log2.html index e9fa8907ce..67f4d702b3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.log2.html +++ b/docs/build/html/python/_autosummary/mlx.core.log2.html @@ -8,7 +8,7 @@ - mlx.core.log2 — MLX 0.16.1 documentation + mlx.core.log2 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.logaddexp.html b/docs/build/html/python/_autosummary/mlx.core.logaddexp.html index 7527b25c02..6c19e2fe67 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logaddexp.html +++ b/docs/build/html/python/_autosummary/mlx.core.logaddexp.html @@ -8,7 +8,7 @@ - mlx.core.logaddexp — MLX 0.16.1 documentation + mlx.core.logaddexp — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_and.html b/docs/build/html/python/_autosummary/mlx.core.logical_and.html index 3b5a7260ed..1de3312705 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logical_and.html +++ b/docs/build/html/python/_autosummary/mlx.core.logical_and.html @@ -8,7 +8,7 @@ - mlx.core.logical_and — MLX 0.16.1 documentation + mlx.core.logical_and — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_not.html b/docs/build/html/python/_autosummary/mlx.core.logical_not.html index 31c4799287..dd4eb4fdae 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logical_not.html +++ b/docs/build/html/python/_autosummary/mlx.core.logical_not.html @@ -8,7 +8,7 @@ - mlx.core.logical_not — MLX 0.16.1 documentation + mlx.core.logical_not — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_or.html b/docs/build/html/python/_autosummary/mlx.core.logical_or.html index e68445d693..4c36f2cd3b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logical_or.html +++ b/docs/build/html/python/_autosummary/mlx.core.logical_or.html @@ -8,7 +8,7 @@ - mlx.core.logical_or — MLX 0.16.1 documentation + mlx.core.logical_or — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.logsumexp.html b/docs/build/html/python/_autosummary/mlx.core.logsumexp.html index 19467af450..ed4ba418a8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logsumexp.html +++ b/docs/build/html/python/_autosummary/mlx.core.logsumexp.html @@ -8,7 +8,7 @@ - mlx.core.logsumexp — MLX 0.16.1 documentation + mlx.core.logsumexp — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.matmul.html b/docs/build/html/python/_autosummary/mlx.core.matmul.html index 1e9c1e39db..e47101971e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.matmul.html +++ b/docs/build/html/python/_autosummary/mlx.core.matmul.html @@ -8,7 +8,7 @@ - mlx.core.matmul — MLX 0.16.1 documentation + mlx.core.matmul — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.max.html b/docs/build/html/python/_autosummary/mlx.core.max.html index bd9e77e161..be196f10e1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.max.html +++ b/docs/build/html/python/_autosummary/mlx.core.max.html @@ -8,7 +8,7 @@ - mlx.core.max — MLX 0.16.1 documentation + mlx.core.max — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.maximum.html b/docs/build/html/python/_autosummary/mlx.core.maximum.html index 13a9880c1b..d3e4a9600d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.maximum.html +++ b/docs/build/html/python/_autosummary/mlx.core.maximum.html @@ -8,7 +8,7 @@ - mlx.core.maximum — MLX 0.16.1 documentation + mlx.core.maximum — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.mean.html b/docs/build/html/python/_autosummary/mlx.core.mean.html index 6b82fec83e..e95f461470 100644 --- a/docs/build/html/python/_autosummary/mlx.core.mean.html +++ b/docs/build/html/python/_autosummary/mlx.core.mean.html @@ -8,7 +8,7 @@ - mlx.core.mean — MLX 0.16.1 documentation + mlx.core.mean — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.meshgrid.html b/docs/build/html/python/_autosummary/mlx.core.meshgrid.html index 4e61d06eb5..b1a62418f9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.meshgrid.html +++ b/docs/build/html/python/_autosummary/mlx.core.meshgrid.html @@ -8,7 +8,7 @@ - mlx.core.meshgrid — MLX 0.16.1 documentation + mlx.core.meshgrid — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html b/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html index f53bed0637..2dbdac15d5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html @@ -8,7 +8,7 @@ - mlx.core.metal.clear_cache — MLX 0.16.1 documentation + mlx.core.metal.clear_cache — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html b/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html index cdcd971dce..1652f3d66a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html @@ -8,7 +8,7 @@ - mlx.core.metal.device_info — MLX 0.16.1 documentation + mlx.core.metal.device_info — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html index e08e8d56b9..366ecd3cca 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html @@ -8,7 +8,7 @@ - mlx.core.metal.get_active_memory — MLX 0.16.1 documentation + mlx.core.metal.get_active_memory — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html index a3421dd443..788146f6d6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html @@ -8,7 +8,7 @@ - mlx.core.metal.get_cache_memory — MLX 0.16.1 documentation + mlx.core.metal.get_cache_memory — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html index 3b24187866..ae84bb689b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html @@ -8,7 +8,7 @@ - mlx.core.metal.get_peak_memory — MLX 0.16.1 documentation + mlx.core.metal.get_peak_memory — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html b/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html index 1740be943b..7bf92fcce2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html @@ -8,7 +8,7 @@ - mlx.core.metal.is_available — MLX 0.16.1 documentation + mlx.core.metal.is_available — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html index 939a2aa28e..1241947169 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html @@ -8,7 +8,7 @@ - mlx.core.metal.reset_peak_memory — MLX 0.16.1 documentation + mlx.core.metal.reset_peak_memory — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html b/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html index 5319360df9..a083659bf7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html @@ -8,7 +8,7 @@ - mlx.core.metal.set_cache_limit — MLX 0.16.1 documentation + mlx.core.metal.set_cache_limit — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html b/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html index 06fb95c62d..1fe743c9b3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html @@ -8,7 +8,7 @@ - mlx.core.metal.set_memory_limit — MLX 0.16.1 documentation + mlx.core.metal.set_memory_limit — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html b/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html index 553dc9c482..6ca2b6ebcc 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html @@ -8,7 +8,7 @@ - mlx.core.metal.start_capture — MLX 0.16.1 documentation + mlx.core.metal.start_capture — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html b/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html index 81467e0266..946d117940 100644 --- a/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html +++ b/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html @@ -8,7 +8,7 @@ - mlx.core.metal.stop_capture — MLX 0.16.1 documentation + mlx.core.metal.stop_capture — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.min.html b/docs/build/html/python/_autosummary/mlx.core.min.html index 0d79265c68..48d7ba5765 100644 --- a/docs/build/html/python/_autosummary/mlx.core.min.html +++ b/docs/build/html/python/_autosummary/mlx.core.min.html @@ -8,7 +8,7 @@ - mlx.core.min — MLX 0.16.1 documentation + mlx.core.min — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.minimum.html b/docs/build/html/python/_autosummary/mlx.core.minimum.html index 0aa78ae88d..9576feecff 100644 --- a/docs/build/html/python/_autosummary/mlx.core.minimum.html +++ b/docs/build/html/python/_autosummary/mlx.core.minimum.html @@ -8,7 +8,7 @@ - mlx.core.minimum — MLX 0.16.1 documentation + mlx.core.minimum — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.moveaxis.html b/docs/build/html/python/_autosummary/mlx.core.moveaxis.html index fab8c49880..4b82f8a489 100644 --- a/docs/build/html/python/_autosummary/mlx.core.moveaxis.html +++ b/docs/build/html/python/_autosummary/mlx.core.moveaxis.html @@ -8,7 +8,7 @@ - mlx.core.moveaxis — MLX 0.16.1 documentation + mlx.core.moveaxis — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.multiply.html b/docs/build/html/python/_autosummary/mlx.core.multiply.html index 54faffcbcc..cb364c2e1d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.multiply.html +++ b/docs/build/html/python/_autosummary/mlx.core.multiply.html @@ -8,7 +8,7 @@ - mlx.core.multiply — MLX 0.16.1 documentation + mlx.core.multiply — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html b/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html index 047801765a..ae413ad6dc 100644 --- a/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html +++ b/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html @@ -8,7 +8,7 @@ - mlx.core.nan_to_num — MLX 0.16.1 documentation + mlx.core.nan_to_num — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.negative.html b/docs/build/html/python/_autosummary/mlx.core.negative.html index 25f9a98800..bb002333ea 100644 --- a/docs/build/html/python/_autosummary/mlx.core.negative.html +++ b/docs/build/html/python/_autosummary/mlx.core.negative.html @@ -8,7 +8,7 @@ - mlx.core.negative — MLX 0.16.1 documentation + mlx.core.negative — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.new_stream.html b/docs/build/html/python/_autosummary/mlx.core.new_stream.html index eea5a60b3c..410395bc7c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.new_stream.html +++ b/docs/build/html/python/_autosummary/mlx.core.new_stream.html @@ -8,7 +8,7 @@ - mlx.core.new_stream — MLX 0.16.1 documentation + mlx.core.new_stream — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.not_equal.html b/docs/build/html/python/_autosummary/mlx.core.not_equal.html index 7089fee9c7..45e3f756fa 100644 --- a/docs/build/html/python/_autosummary/mlx.core.not_equal.html +++ b/docs/build/html/python/_autosummary/mlx.core.not_equal.html @@ -8,7 +8,7 @@ - mlx.core.not_equal — MLX 0.16.1 documentation + mlx.core.not_equal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.ones.html b/docs/build/html/python/_autosummary/mlx.core.ones.html index 0363830dcd..7e61084c8e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.ones.html +++ b/docs/build/html/python/_autosummary/mlx.core.ones.html @@ -8,7 +8,7 @@ - mlx.core.ones — MLX 0.16.1 documentation + mlx.core.ones — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.ones_like.html b/docs/build/html/python/_autosummary/mlx.core.ones_like.html index 69340a1b21..6ba43ab03d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.ones_like.html +++ b/docs/build/html/python/_autosummary/mlx.core.ones_like.html @@ -8,7 +8,7 @@ - mlx.core.ones_like — MLX 0.16.1 documentation + mlx.core.ones_like — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.outer.html b/docs/build/html/python/_autosummary/mlx.core.outer.html index 98d9d0ccb3..4ffa9ba778 100644 --- a/docs/build/html/python/_autosummary/mlx.core.outer.html +++ b/docs/build/html/python/_autosummary/mlx.core.outer.html @@ -8,7 +8,7 @@ - mlx.core.outer — MLX 0.16.1 documentation + mlx.core.outer — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.pad.html b/docs/build/html/python/_autosummary/mlx.core.pad.html index bb0ebb315a..ed6070ce57 100644 --- a/docs/build/html/python/_autosummary/mlx.core.pad.html +++ b/docs/build/html/python/_autosummary/mlx.core.pad.html @@ -8,7 +8,7 @@ - mlx.core.pad — MLX 0.16.1 documentation + mlx.core.pad — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra @@ -846,7 +848,7 @@ document.write(`

            mlx.core.pad#

            -pad(a: array, pad_width: int | Tuple[int] | Tuple[int, int] | List[Tuple[int, int]], constant_values: scalar | array = 0, *, stream: None | Stream | Device = None) → array#
            +pad(a: array, pad_width: int | Tuple[int] | Tuple[int, int] | List[Tuple[int, int]], mode: Literal['constant', 'edge'] = 'constant', constant_values: scalar | array = 0, *, stream: None | Stream | Device = None) → array#

            Pad an array with a constant value

            Parameters:
            @@ -858,6 +860,9 @@ values to add to the edges of each axis:(before_i, after_i) are all the same. If a single integer or tuple with a single integer is passed then all axes are extended by the same number on each side.

            • +

          • mode – Padding mode. One of the following strings: +“constant” (default): Pads with a constant value. +“edge”: Pads with the edge values of array.

          • constant_value (array or scalar, optional) – Optional constant value to pad the edges of the array with.

            • diff --git a/docs/build/html/python/_autosummary/mlx.core.partition.html b/docs/build/html/python/_autosummary/mlx.core.partition.html index 966c1904f2..39ab895446 100644 --- a/docs/build/html/python/_autosummary/mlx.core.partition.html +++ b/docs/build/html/python/_autosummary/mlx.core.partition.html @@ -8,7 +8,7 @@ - mlx.core.partition — MLX 0.16.1 documentation + mlx.core.partition — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.power.html b/docs/build/html/python/_autosummary/mlx.core.power.html index d5873250d3..1de917a9d9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.power.html +++ b/docs/build/html/python/_autosummary/mlx.core.power.html @@ -8,7 +8,7 @@ - mlx.core.power — MLX 0.16.1 documentation + mlx.core.power — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.prod.html b/docs/build/html/python/_autosummary/mlx.core.prod.html index c933eddc67..28f19a3f96 100644 --- a/docs/build/html/python/_autosummary/mlx.core.prod.html +++ b/docs/build/html/python/_autosummary/mlx.core.prod.html @@ -8,7 +8,7 @@ - mlx.core.prod — MLX 0.16.1 documentation + mlx.core.prod — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.quantize.html b/docs/build/html/python/_autosummary/mlx.core.quantize.html index 3f45531158..9005ecf2cb 100644 --- a/docs/build/html/python/_autosummary/mlx.core.quantize.html +++ b/docs/build/html/python/_autosummary/mlx.core.quantize.html @@ -8,7 +8,7 @@ - mlx.core.quantize — MLX 0.16.1 documentation + mlx.core.quantize — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html b/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html index eee8bf0a2e..766c6ef804 100644 --- a/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html +++ b/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html @@ -8,7 +8,7 @@ - mlx.core.quantized_matmul — MLX 0.16.1 documentation + mlx.core.quantized_matmul — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.radians.html b/docs/build/html/python/_autosummary/mlx.core.radians.html index 6b1e3aff8a..48091a3d97 100644 --- a/docs/build/html/python/_autosummary/mlx.core.radians.html +++ b/docs/build/html/python/_autosummary/mlx.core.radians.html @@ -8,7 +8,7 @@ - mlx.core.radians — MLX 0.16.1 documentation + mlx.core.radians — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html b/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html index 5dc0aefb1e..beb508486b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html @@ -8,7 +8,7 @@ - mlx.core.random.bernoulli — MLX 0.16.1 documentation + mlx.core.random.bernoulli — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.categorical.html b/docs/build/html/python/_autosummary/mlx.core.random.categorical.html index 269220e587..218f8152c7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.categorical.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.categorical.html @@ -8,7 +8,7 @@ - mlx.core.random.categorical — MLX 0.16.1 documentation + mlx.core.random.categorical — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html b/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html index c63c24c728..94f63047ca 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html @@ -8,7 +8,7 @@ - mlx.core.random.gumbel — MLX 0.16.1 documentation + mlx.core.random.gumbel — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.key.html b/docs/build/html/python/_autosummary/mlx.core.random.key.html index 2c1832af69..084402cb99 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.key.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.key.html @@ -8,7 +8,7 @@ - mlx.core.random.key — MLX 0.16.1 documentation + mlx.core.random.key — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.laplace.html b/docs/build/html/python/_autosummary/mlx.core.random.laplace.html index ce6b7ae408..7b0dd9ce2c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.laplace.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.laplace.html @@ -8,7 +8,7 @@ - mlx.core.random.laplace — MLX 0.16.1 documentation + mlx.core.random.laplace — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.multivariate_normal.html b/docs/build/html/python/_autosummary/mlx.core.random.multivariate_normal.html index 6550183858..703d9215c2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.multivariate_normal.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.multivariate_normal.html @@ -8,7 +8,7 @@ - mlx.core.random.multivariate_normal — MLX 0.16.1 documentation + mlx.core.random.multivariate_normal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.normal.html b/docs/build/html/python/_autosummary/mlx.core.random.normal.html index e30c9b9b7a..38f6fd1db5 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.normal.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.normal.html @@ -8,7 +8,7 @@ - mlx.core.random.normal — MLX 0.16.1 documentation + mlx.core.random.normal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.randint.html b/docs/build/html/python/_autosummary/mlx.core.random.randint.html index 31bea48547..8945d157ee 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.randint.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.randint.html @@ -8,7 +8,7 @@ - mlx.core.random.randint — MLX 0.16.1 documentation + mlx.core.random.randint — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.seed.html b/docs/build/html/python/_autosummary/mlx.core.random.seed.html index 28fff028d9..8fa7808115 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.seed.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.seed.html @@ -8,7 +8,7 @@ - mlx.core.random.seed — MLX 0.16.1 documentation + mlx.core.random.seed — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.split.html b/docs/build/html/python/_autosummary/mlx.core.random.split.html index e8d4143568..abbd9f7fb8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.split.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.split.html @@ -8,7 +8,7 @@ - mlx.core.random.split — MLX 0.16.1 documentation + mlx.core.random.split — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html b/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html index 70190e1117..9b624b8461 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html @@ -8,7 +8,7 @@ - mlx.core.random.truncated_normal — MLX 0.16.1 documentation + mlx.core.random.truncated_normal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.random.uniform.html b/docs/build/html/python/_autosummary/mlx.core.random.uniform.html index 846475fe45..9e4b8bc6a4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.uniform.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.uniform.html @@ -8,7 +8,7 @@ - mlx.core.random.uniform — MLX 0.16.1 documentation + mlx.core.random.uniform — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.reciprocal.html b/docs/build/html/python/_autosummary/mlx.core.reciprocal.html index 5edee69d83..c75a7940be 100644 --- a/docs/build/html/python/_autosummary/mlx.core.reciprocal.html +++ b/docs/build/html/python/_autosummary/mlx.core.reciprocal.html @@ -8,7 +8,7 @@ - mlx.core.reciprocal — MLX 0.16.1 documentation + mlx.core.reciprocal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.remainder.html b/docs/build/html/python/_autosummary/mlx.core.remainder.html index 8873222fff..f1d6652717 100644 --- a/docs/build/html/python/_autosummary/mlx.core.remainder.html +++ b/docs/build/html/python/_autosummary/mlx.core.remainder.html @@ -8,7 +8,7 @@ - mlx.core.remainder — MLX 0.16.1 documentation + mlx.core.remainder — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.repeat.html b/docs/build/html/python/_autosummary/mlx.core.repeat.html index b8af5997ed..f0c2b6a114 100644 --- a/docs/build/html/python/_autosummary/mlx.core.repeat.html +++ b/docs/build/html/python/_autosummary/mlx.core.repeat.html @@ -8,7 +8,7 @@ - mlx.core.repeat — MLX 0.16.1 documentation + mlx.core.repeat — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.reshape.html b/docs/build/html/python/_autosummary/mlx.core.reshape.html index 3331984d23..a83dccd2fc 100644 --- a/docs/build/html/python/_autosummary/mlx.core.reshape.html +++ b/docs/build/html/python/_autosummary/mlx.core.reshape.html @@ -8,7 +8,7 @@ - mlx.core.reshape — MLX 0.16.1 documentation + mlx.core.reshape — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.right_shift.html b/docs/build/html/python/_autosummary/mlx.core.right_shift.html index 987bb3d56d..59f2107078 100644 --- a/docs/build/html/python/_autosummary/mlx.core.right_shift.html +++ b/docs/build/html/python/_autosummary/mlx.core.right_shift.html @@ -8,7 +8,7 @@ - mlx.core.right_shift — MLX 0.16.1 documentation + mlx.core.right_shift — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.round.html b/docs/build/html/python/_autosummary/mlx.core.round.html index 0c3a24901b..289e4e6975 100644 --- a/docs/build/html/python/_autosummary/mlx.core.round.html +++ b/docs/build/html/python/_autosummary/mlx.core.round.html @@ -8,7 +8,7 @@ - mlx.core.round — MLX 0.16.1 documentation + mlx.core.round — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.rsqrt.html b/docs/build/html/python/_autosummary/mlx.core.rsqrt.html index 6e016a10ea..5f96d38c64 100644 --- a/docs/build/html/python/_autosummary/mlx.core.rsqrt.html +++ b/docs/build/html/python/_autosummary/mlx.core.rsqrt.html @@ -8,7 +8,7 @@ - mlx.core.rsqrt — MLX 0.16.1 documentation + mlx.core.rsqrt — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.save.html b/docs/build/html/python/_autosummary/mlx.core.save.html index ce39c2d29c..ffe850ceb8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.save.html +++ b/docs/build/html/python/_autosummary/mlx.core.save.html @@ -8,7 +8,7 @@ - mlx.core.save — MLX 0.16.1 documentation + mlx.core.save — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.save_gguf.html b/docs/build/html/python/_autosummary/mlx.core.save_gguf.html index e6f18aa61e..72292add60 100644 --- a/docs/build/html/python/_autosummary/mlx.core.save_gguf.html +++ b/docs/build/html/python/_autosummary/mlx.core.save_gguf.html @@ -8,7 +8,7 @@ - mlx.core.save_gguf — MLX 0.16.1 documentation + mlx.core.save_gguf — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.save_safetensors.html b/docs/build/html/python/_autosummary/mlx.core.save_safetensors.html index 4e2e550b6a..c2bdc7a7ec 100644 --- a/docs/build/html/python/_autosummary/mlx.core.save_safetensors.html +++ b/docs/build/html/python/_autosummary/mlx.core.save_safetensors.html @@ -8,7 +8,7 @@ - mlx.core.save_safetensors — MLX 0.16.1 documentation + mlx.core.save_safetensors — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.savez.html b/docs/build/html/python/_autosummary/mlx.core.savez.html index 5d9973b578..99e9e223b6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.savez.html +++ b/docs/build/html/python/_autosummary/mlx.core.savez.html @@ -8,7 +8,7 @@ - mlx.core.savez — MLX 0.16.1 documentation + mlx.core.savez — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html b/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html index 16fccc69e9..50bbba3b2f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html +++ b/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html @@ -8,7 +8,7 @@ - mlx.core.savez_compressed — MLX 0.16.1 documentation + mlx.core.savez_compressed — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.set_default_device.html b/docs/build/html/python/_autosummary/mlx.core.set_default_device.html index 795ba4610a..4310feed3a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.set_default_device.html +++ b/docs/build/html/python/_autosummary/mlx.core.set_default_device.html @@ -8,7 +8,7 @@ - mlx.core.set_default_device — MLX 0.16.1 documentation + mlx.core.set_default_device — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html b/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html index 64c9be57d5..e3fdced1d6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html +++ b/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html @@ -8,7 +8,7 @@ - mlx.core.set_default_stream — MLX 0.16.1 documentation + mlx.core.set_default_stream — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.sigmoid.html b/docs/build/html/python/_autosummary/mlx.core.sigmoid.html index cb5587dd27..ed21fa2cc4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sigmoid.html +++ b/docs/build/html/python/_autosummary/mlx.core.sigmoid.html @@ -8,7 +8,7 @@ - mlx.core.sigmoid — MLX 0.16.1 documentation + mlx.core.sigmoid — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.sign.html b/docs/build/html/python/_autosummary/mlx.core.sign.html index d48df9a3c9..126a091f47 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sign.html +++ b/docs/build/html/python/_autosummary/mlx.core.sign.html @@ -8,7 +8,7 @@ - mlx.core.sign — MLX 0.16.1 documentation + mlx.core.sign — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.sin.html b/docs/build/html/python/_autosummary/mlx.core.sin.html index e179809cf0..fd867f128e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sin.html +++ b/docs/build/html/python/_autosummary/mlx.core.sin.html @@ -8,7 +8,7 @@ - mlx.core.sin — MLX 0.16.1 documentation + mlx.core.sin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.sinh.html b/docs/build/html/python/_autosummary/mlx.core.sinh.html index f83441c7e2..1f4e7a0cc7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sinh.html +++ b/docs/build/html/python/_autosummary/mlx.core.sinh.html @@ -8,7 +8,7 @@ - mlx.core.sinh — MLX 0.16.1 documentation + mlx.core.sinh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.softmax.html b/docs/build/html/python/_autosummary/mlx.core.softmax.html index a71532717f..50876f4114 100644 --- a/docs/build/html/python/_autosummary/mlx.core.softmax.html +++ b/docs/build/html/python/_autosummary/mlx.core.softmax.html @@ -8,7 +8,7 @@ - mlx.core.softmax — MLX 0.16.1 documentation + mlx.core.softmax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.sort.html b/docs/build/html/python/_autosummary/mlx.core.sort.html index b7cf82e1b2..897ca128f1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sort.html +++ b/docs/build/html/python/_autosummary/mlx.core.sort.html @@ -8,7 +8,7 @@ - mlx.core.sort — MLX 0.16.1 documentation + mlx.core.sort — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.split.html b/docs/build/html/python/_autosummary/mlx.core.split.html index a6ce389003..a2c58476ed 100644 --- a/docs/build/html/python/_autosummary/mlx.core.split.html +++ b/docs/build/html/python/_autosummary/mlx.core.split.html @@ -8,7 +8,7 @@ - mlx.core.split — MLX 0.16.1 documentation + mlx.core.split — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.sqrt.html b/docs/build/html/python/_autosummary/mlx.core.sqrt.html index 5741288a06..ef04ffb8a6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sqrt.html +++ b/docs/build/html/python/_autosummary/mlx.core.sqrt.html @@ -8,7 +8,7 @@ - mlx.core.sqrt — MLX 0.16.1 documentation + mlx.core.sqrt — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.square.html b/docs/build/html/python/_autosummary/mlx.core.square.html index 76f1a44c44..5fb6b31737 100644 --- a/docs/build/html/python/_autosummary/mlx.core.square.html +++ b/docs/build/html/python/_autosummary/mlx.core.square.html @@ -8,7 +8,7 @@ - mlx.core.square — MLX 0.16.1 documentation + mlx.core.square — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.squeeze.html b/docs/build/html/python/_autosummary/mlx.core.squeeze.html index 88fc1d71a2..679dcced44 100644 --- a/docs/build/html/python/_autosummary/mlx.core.squeeze.html +++ b/docs/build/html/python/_autosummary/mlx.core.squeeze.html @@ -8,7 +8,7 @@ - mlx.core.squeeze — MLX 0.16.1 documentation + mlx.core.squeeze — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.stack.html b/docs/build/html/python/_autosummary/mlx.core.stack.html index 96ac5fc6c7..a5c6bee4bb 100644 --- a/docs/build/html/python/_autosummary/mlx.core.stack.html +++ b/docs/build/html/python/_autosummary/mlx.core.stack.html @@ -8,7 +8,7 @@ - mlx.core.stack — MLX 0.16.1 documentation + mlx.core.stack — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.std.html b/docs/build/html/python/_autosummary/mlx.core.std.html index 0e1ee56263..748aa2d593 100644 --- a/docs/build/html/python/_autosummary/mlx.core.std.html +++ b/docs/build/html/python/_autosummary/mlx.core.std.html @@ -8,7 +8,7 @@ - mlx.core.std — MLX 0.16.1 documentation + mlx.core.std — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html b/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html index c6fc13ce23..1c3c183fbe 100644 --- a/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html +++ b/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html @@ -8,7 +8,7 @@ - mlx.core.stop_gradient — MLX 0.16.1 documentation + mlx.core.stop_gradient — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.subtract.html b/docs/build/html/python/_autosummary/mlx.core.subtract.html index 340c24ae1d..a5f84d74b4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.subtract.html +++ b/docs/build/html/python/_autosummary/mlx.core.subtract.html @@ -8,7 +8,7 @@ - mlx.core.subtract — MLX 0.16.1 documentation + mlx.core.subtract — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.sum.html b/docs/build/html/python/_autosummary/mlx.core.sum.html index 76cfba25af..f2c32ff90b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.sum.html +++ b/docs/build/html/python/_autosummary/mlx.core.sum.html @@ -8,7 +8,7 @@ - mlx.core.sum — MLX 0.16.1 documentation + mlx.core.sum — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.swapaxes.html b/docs/build/html/python/_autosummary/mlx.core.swapaxes.html index 1d27ba1563..af10016373 100644 --- a/docs/build/html/python/_autosummary/mlx.core.swapaxes.html +++ b/docs/build/html/python/_autosummary/mlx.core.swapaxes.html @@ -8,7 +8,7 @@ - mlx.core.swapaxes — MLX 0.16.1 documentation + mlx.core.swapaxes — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.synchronize.html b/docs/build/html/python/_autosummary/mlx.core.synchronize.html index 638e669066..5e284c445a 100644 --- a/docs/build/html/python/_autosummary/mlx.core.synchronize.html +++ b/docs/build/html/python/_autosummary/mlx.core.synchronize.html @@ -8,7 +8,7 @@ - mlx.core.synchronize — MLX 0.16.1 documentation + mlx.core.synchronize — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.take.html b/docs/build/html/python/_autosummary/mlx.core.take.html index f1a294edde..f22910a04b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.take.html +++ b/docs/build/html/python/_autosummary/mlx.core.take.html @@ -8,7 +8,7 @@ - mlx.core.take — MLX 0.16.1 documentation + mlx.core.take — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html b/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html index d83d7c52a0..f98b57b45d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html +++ b/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html @@ -8,7 +8,7 @@ - mlx.core.take_along_axis — MLX 0.16.1 documentation + mlx.core.take_along_axis — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.tan.html b/docs/build/html/python/_autosummary/mlx.core.tan.html index 35ad16308e..5be0ebf567 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tan.html +++ b/docs/build/html/python/_autosummary/mlx.core.tan.html @@ -8,7 +8,7 @@ - mlx.core.tan — MLX 0.16.1 documentation + mlx.core.tan — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.tanh.html b/docs/build/html/python/_autosummary/mlx.core.tanh.html index d301b1b05d..6c9e819b39 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tanh.html +++ b/docs/build/html/python/_autosummary/mlx.core.tanh.html @@ -8,7 +8,7 @@ - mlx.core.tanh — MLX 0.16.1 documentation + mlx.core.tanh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.tensordot.html b/docs/build/html/python/_autosummary/mlx.core.tensordot.html index b63a8dafbf..ba86d4bd96 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tensordot.html +++ b/docs/build/html/python/_autosummary/mlx.core.tensordot.html @@ -8,7 +8,7 @@ - mlx.core.tensordot — MLX 0.16.1 documentation + mlx.core.tensordot — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.tile.html b/docs/build/html/python/_autosummary/mlx.core.tile.html index f813b6efea..72c93d6100 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tile.html +++ b/docs/build/html/python/_autosummary/mlx.core.tile.html @@ -8,7 +8,7 @@ - mlx.core.tile — MLX 0.16.1 documentation + mlx.core.tile — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.topk.html b/docs/build/html/python/_autosummary/mlx.core.topk.html index 204da0eebb..6ecfb469ba 100644 --- a/docs/build/html/python/_autosummary/mlx.core.topk.html +++ b/docs/build/html/python/_autosummary/mlx.core.topk.html @@ -8,7 +8,7 @@ - mlx.core.topk — MLX 0.16.1 documentation + mlx.core.topk — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.trace.html b/docs/build/html/python/_autosummary/mlx.core.trace.html index 96a67b873c..6128a6219b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.trace.html +++ b/docs/build/html/python/_autosummary/mlx.core.trace.html @@ -8,7 +8,7 @@ - mlx.core.trace — MLX 0.16.1 documentation + mlx.core.trace — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.transpose.html b/docs/build/html/python/_autosummary/mlx.core.transpose.html index 0ad65fd084..362c681927 100644 --- a/docs/build/html/python/_autosummary/mlx.core.transpose.html +++ b/docs/build/html/python/_autosummary/mlx.core.transpose.html @@ -8,7 +8,7 @@ - mlx.core.transpose — MLX 0.16.1 documentation + mlx.core.transpose — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.tri.html b/docs/build/html/python/_autosummary/mlx.core.tri.html index 292a4eab65..fc81eb1081 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tri.html +++ b/docs/build/html/python/_autosummary/mlx.core.tri.html @@ -8,7 +8,7 @@ - mlx.core.tri — MLX 0.16.1 documentation + mlx.core.tri — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.tril.html b/docs/build/html/python/_autosummary/mlx.core.tril.html index 5fa057f2c8..f60a35c690 100644 --- a/docs/build/html/python/_autosummary/mlx.core.tril.html +++ b/docs/build/html/python/_autosummary/mlx.core.tril.html @@ -8,7 +8,7 @@ - mlx.core.tril — MLX 0.16.1 documentation + mlx.core.tril — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.triu.html b/docs/build/html/python/_autosummary/mlx.core.triu.html index 16f0ef3b0a..fce3c90d30 100644 --- a/docs/build/html/python/_autosummary/mlx.core.triu.html +++ b/docs/build/html/python/_autosummary/mlx.core.triu.html @@ -8,7 +8,7 @@ - mlx.core.triu — MLX 0.16.1 documentation + mlx.core.triu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html b/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html index 503887fcc3..3e533237b0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html +++ b/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html @@ -8,7 +8,7 @@ - mlx.core.value_and_grad — MLX 0.16.1 documentation + mlx.core.value_and_grad — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.var.html b/docs/build/html/python/_autosummary/mlx.core.var.html index 5797233a11..2e6b84929d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.var.html +++ b/docs/build/html/python/_autosummary/mlx.core.var.html @@ -8,7 +8,7 @@ - mlx.core.var — MLX 0.16.1 documentation + mlx.core.var — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.view.html b/docs/build/html/python/_autosummary/mlx.core.view.html index 0f1ce84226..c2e27dfccb 100644 --- a/docs/build/html/python/_autosummary/mlx.core.view.html +++ b/docs/build/html/python/_autosummary/mlx.core.view.html @@ -8,7 +8,7 @@ - mlx.core.view — MLX 0.16.1 documentation + mlx.core.view — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.vjp.html b/docs/build/html/python/_autosummary/mlx.core.vjp.html index 2b07105690..70cae97f53 100644 --- a/docs/build/html/python/_autosummary/mlx.core.vjp.html +++ b/docs/build/html/python/_autosummary/mlx.core.vjp.html @@ -8,7 +8,7 @@ - mlx.core.vjp — MLX 0.16.1 documentation + mlx.core.vjp — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.vmap.html b/docs/build/html/python/_autosummary/mlx.core.vmap.html index adc4890b4c..95f3423d42 100644 --- a/docs/build/html/python/_autosummary/mlx.core.vmap.html +++ b/docs/build/html/python/_autosummary/mlx.core.vmap.html @@ -8,7 +8,7 @@ - mlx.core.vmap — MLX 0.16.1 documentation + mlx.core.vmap — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.where.html b/docs/build/html/python/_autosummary/mlx.core.where.html index 5a835b047a..f9d50869ad 100644 --- a/docs/build/html/python/_autosummary/mlx.core.where.html +++ b/docs/build/html/python/_autosummary/mlx.core.where.html @@ -8,7 +8,7 @@ - mlx.core.where — MLX 0.16.1 documentation + mlx.core.where — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.zeros.html b/docs/build/html/python/_autosummary/mlx.core.zeros.html index d5a05d21bd..4b9cfcd560 100644 --- a/docs/build/html/python/_autosummary/mlx.core.zeros.html +++ b/docs/build/html/python/_autosummary/mlx.core.zeros.html @@ -8,7 +8,7 @@ - mlx.core.zeros — MLX 0.16.1 documentation + mlx.core.zeros — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.core.zeros_like.html b/docs/build/html/python/_autosummary/mlx.core.zeros_like.html index 585d02447c..a1ee50b9d4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.zeros_like.html +++ b/docs/build/html/python/_autosummary/mlx.core.zeros_like.html @@ -8,7 +8,7 @@ - mlx.core.zeros_like — MLX 0.16.1 documentation + mlx.core.zeros_like — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.nn.quantize.html b/docs/build/html/python/_autosummary/mlx.nn.quantize.html index b036f11c5f..de89492c73 100644 --- a/docs/build/html/python/_autosummary/mlx.nn.quantize.html +++ b/docs/build/html/python/_autosummary/mlx.nn.quantize.html @@ -8,7 +8,7 @@ - mlx.nn.quantize — MLX 0.16.1 documentation + mlx.nn.quantize — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html b/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html index 8103ba633f..212832743c 100644 --- a/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html +++ b/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html @@ -8,7 +8,7 @@ - mlx.nn.value_and_grad — MLX 0.16.1 documentation + mlx.nn.value_and_grad — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html b/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html index 8177c05af3..cb21b494c9 100644 --- a/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html +++ b/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html @@ -8,7 +8,7 @@ - mlx.optimizers.clip_grad_norm — MLX 0.16.1 documentation + mlx.optimizers.clip_grad_norm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html b/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html index 4061036d35..9d16d26dbf 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html @@ -8,7 +8,7 @@ - mlx.utils.tree_flatten — MLX 0.16.1 documentation + mlx.utils.tree_flatten — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_map.html b/docs/build/html/python/_autosummary/mlx.utils.tree_map.html index 648d38f571..d732a1afc0 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_map.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_map.html @@ -8,7 +8,7 @@ - mlx.utils.tree_map — MLX 0.16.1 documentation + mlx.utils.tree_map — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_map_with_path.html b/docs/build/html/python/_autosummary/mlx.utils.tree_map_with_path.html index 5efdfe3a3a..4925724249 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_map_with_path.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_map_with_path.html @@ -8,7 +8,7 @@ - mlx.utils.tree_map_with_path — MLX 0.16.1 documentation + mlx.utils.tree_map_with_path — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_reduce.html b/docs/build/html/python/_autosummary/mlx.utils.tree_reduce.html index ba4377552c..53a4b342a2 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_reduce.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_reduce.html @@ -8,7 +8,7 @@ - mlx.utils.tree_reduce — MLX 0.16.1 documentation + mlx.utils.tree_reduce — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html b/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html index 809cbbf7ca..d91090a1a8 100644 --- a/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html +++ b/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html @@ -8,7 +8,7 @@ - mlx.utils.tree_unflatten — MLX 0.16.1 documentation + mlx.utils.tree_unflatten — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/_autosummary/stream_class.html b/docs/build/html/python/_autosummary/stream_class.html index c5ea43dfcb..1a34ad056f 100644 --- a/docs/build/html/python/_autosummary/stream_class.html +++ b/docs/build/html/python/_autosummary/stream_class.html @@ -8,7 +8,7 @@ - mlx.core.Stream — MLX 0.16.1 documentation + mlx.core.Stream — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/array.html b/docs/build/html/python/array.html index ea41c62f61..88cfe3308f 100644 --- a/docs/build/html/python/array.html +++ b/docs/build/html/python/array.html @@ -8,7 +8,7 @@ - Array — MLX 0.16.1 documentation + Array — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/data_types.html b/docs/build/html/python/data_types.html index db79fc77f5..918f80dc09 100644 --- a/docs/build/html/python/data_types.html +++ b/docs/build/html/python/data_types.html @@ -8,7 +8,7 @@ - Data Types — MLX 0.16.1 documentation + Data Types — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra @@ -912,7 +914,7 @@ documentation for more information. Use
            • - + - + + + + - + + + + diff --git a/docs/build/html/python/metal.html b/docs/build/html/python/metal.html index 4eed55b9ea..80c11f59b6 100644 --- a/docs/build/html/python/metal.html +++ b/docs/build/html/python/metal.html @@ -8,7 +8,7 @@ - Metal — MLX 0.16.1 documentation + Metal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn.html b/docs/build/html/python/nn.html index 70c2caa76a..2a0747b8f4 100644 --- a/docs/build/html/python/nn.html +++ b/docs/build/html/python/nn.html @@ -8,7 +8,7 @@ - Neural Networks — MLX 0.16.1 documentation + Neural Networks — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html index 87b573c442..9d2e431831 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html @@ -8,7 +8,7 @@ - mlx.nn.ALiBi — MLX 0.16.1 documentation + mlx.nn.ALiBi — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html index 9fef328b34..dc8734f7f2 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html @@ -8,7 +8,7 @@ - mlx.nn.AvgPool1d — MLX 0.16.1 documentation + mlx.nn.AvgPool1d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html index 4701a2107b..63f1f2a20f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html @@ -8,7 +8,7 @@ - mlx.nn.AvgPool2d — MLX 0.16.1 documentation + mlx.nn.AvgPool2d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html index 3f446ae81e..8354e05e2b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html @@ -8,7 +8,7 @@ - mlx.nn.BatchNorm — MLX 0.16.1 documentation + mlx.nn.BatchNorm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html index c86948f869..f35ef488e4 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html @@ -8,7 +8,7 @@ - mlx.nn.Conv1d — MLX 0.16.1 documentation + mlx.nn.Conv1d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html index 4184a5a596..bc6479a8c7 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html @@ -8,7 +8,7 @@ - mlx.nn.Conv2d — MLX 0.16.1 documentation + mlx.nn.Conv2d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html index 2dcf9fb1c1..1c7d903eb6 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html @@ -8,7 +8,7 @@ - mlx.nn.Conv3d — MLX 0.16.1 documentation + mlx.nn.Conv3d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html index cd6263d128..63b0d4a56f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html @@ -8,7 +8,7 @@ - mlx.nn.Dropout — MLX 0.16.1 documentation + mlx.nn.Dropout — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html index 41f8533718..c606ffa389 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html @@ -8,7 +8,7 @@ - mlx.nn.Dropout2d — MLX 0.16.1 documentation + mlx.nn.Dropout2d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html index 54b7dc464c..e52bf87e1b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html @@ -8,7 +8,7 @@ - mlx.nn.Dropout3d — MLX 0.16.1 documentation + mlx.nn.Dropout3d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html index 0016ee76d8..4f8a491a33 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html @@ -8,7 +8,7 @@ - mlx.nn.Embedding — MLX 0.16.1 documentation + mlx.nn.Embedding — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html index 9ad90f14c0..9d88be2aa7 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html @@ -8,7 +8,7 @@ - mlx.nn.GELU — MLX 0.16.1 documentation + mlx.nn.GELU — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra @@ -855,8 +857,13 @@ document.write(`

            However, if approx is set to ‘precise’ or ‘fast’ it applies

            \[\begin{split}\textrm{GELUApprox}(x) &= 0.5 * x * \left(1 + \text{Tanh}\left((\sqrt{2 / \pi} * \left(x + 0.044715 * x^3\right)\right)\right) \\ -\textrm{GELUFast}(x) &= x * \sigma\left(1.773 * x\right)\end{split}\]
            +\textrm{GELUFast}(x) &= x * \sigma\left(1.702 * x\right)\end{split}\]

            respectively.

            +
            +

            Note

            +

            For compatibility with the PyTorch API, ‘tanh’ can be used as an alias +for ‘precise’.

            +

            See gelu(), gelu_approx() and gelu_fast_approx() for the functional equivalents and information regarding error bounds.

            diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html index 731f2a1dec..9dd2378c0a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html @@ -8,7 +8,7 @@ - mlx.nn.GLU — MLX 0.16.1 documentation + mlx.nn.GLU — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html index 752b79a76c..5f5d4c3147 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html @@ -8,7 +8,7 @@ - mlx.nn.GRU — MLX 0.16.1 documentation + mlx.nn.GRU — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html index 7482a45ede..2ccdf80c3d 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html @@ -8,7 +8,7 @@ - mlx.nn.GroupNorm — MLX 0.16.1 documentation + mlx.nn.GroupNorm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html b/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html index 98e5dacca6..f43581c82a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html @@ -8,7 +8,7 @@ - mlx.nn.HardShrink — MLX 0.16.1 documentation + mlx.nn.HardShrink — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html b/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html index 459c534a47..06c8058c54 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html @@ -8,7 +8,7 @@ - mlx.nn.HardTanh — MLX 0.16.1 documentation + mlx.nn.HardTanh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html index 5e68b69d15..dd2c86fd87 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html @@ -8,7 +8,7 @@ - mlx.nn.Hardswish — MLX 0.16.1 documentation + mlx.nn.Hardswish — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html index a32e8e258a..fbced76759 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html @@ -8,7 +8,7 @@ - mlx.nn.InstanceNorm — MLX 0.16.1 documentation + mlx.nn.InstanceNorm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html index a534dfeb9e..6989ad523a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html @@ -8,7 +8,7 @@ - mlx.nn.LSTM — MLX 0.16.1 documentation + mlx.nn.LSTM — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html index 812a3399dd..77f0091302 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html @@ -8,7 +8,7 @@ - mlx.nn.LayerNorm — MLX 0.16.1 documentation + mlx.nn.LayerNorm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html index 38cee30755..e8ef71959f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html @@ -8,7 +8,7 @@ - mlx.nn.LeakyReLU — MLX 0.16.1 documentation + mlx.nn.LeakyReLU — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html index 4dc306c6fd..2c7029572b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html @@ -8,7 +8,7 @@ - mlx.nn.Linear — MLX 0.16.1 documentation + mlx.nn.Linear — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html index a5380327ea..b8bea1de04 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html @@ -8,7 +8,7 @@ - mlx.nn.MaxPool1d — MLX 0.16.1 documentation + mlx.nn.MaxPool1d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html index 5d752df9e3..f68ec82342 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html @@ -8,7 +8,7 @@ - mlx.nn.MaxPool2d — MLX 0.16.1 documentation + mlx.nn.MaxPool2d — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html index dfde829337..77e0127b6d 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html @@ -8,7 +8,7 @@ - mlx.nn.Mish — MLX 0.16.1 documentation + mlx.nn.Mish — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html index 6ffb556a30..9be0d231ff 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html @@ -8,7 +8,7 @@ - mlx.nn.Module.apply — MLX 0.16.1 documentation + mlx.nn.Module.apply — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html index 8ece62f0a2..e03ba67bad 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.apply_to_modules — MLX 0.16.1 documentation + mlx.nn.Module.apply_to_modules — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html index 9f7fd604bf..b815d0e685 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html @@ -8,7 +8,7 @@ - mlx.nn.Module.children — MLX 0.16.1 documentation + mlx.nn.Module.children — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html index 363ebaf213..9bcc00d093 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html @@ -8,7 +8,7 @@ - mlx.nn.Module.eval — MLX 0.16.1 documentation + mlx.nn.Module.eval — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html index 6448d7c296..584cb07a0c 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html @@ -8,7 +8,7 @@ - mlx.nn.Module.filter_and_map — MLX 0.16.1 documentation + mlx.nn.Module.filter_and_map — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html index 6c3289c371..6c1823eff6 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html @@ -8,7 +8,7 @@ - mlx.nn.Module.freeze — MLX 0.16.1 documentation + mlx.nn.Module.freeze — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html index b0342998bf..c131501be6 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.leaf_modules — MLX 0.16.1 documentation + mlx.nn.Module.leaf_modules — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html index 889d7017cf..6834e03b3f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html @@ -8,7 +8,7 @@ - mlx.nn.Module.load_weights — MLX 0.16.1 documentation + mlx.nn.Module.load_weights — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html index 3e8c2b0e3d..f2c074fe2a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.modules — MLX 0.16.1 documentation + mlx.nn.Module.modules — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html index b6af2babf2..59d1375581 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.named_modules — MLX 0.16.1 documentation + mlx.nn.Module.named_modules — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html index 7c931d5c6a..5e4584406a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html @@ -8,7 +8,7 @@ - mlx.nn.Module.parameters — MLX 0.16.1 documentation + mlx.nn.Module.parameters — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html index 6ba8dc18df..4f178ff67d 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html @@ -8,7 +8,7 @@ - mlx.nn.Module.save_weights — MLX 0.16.1 documentation + mlx.nn.Module.save_weights — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html index 0da0dfe009..6b3375587c 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html @@ -8,7 +8,7 @@ - mlx.nn.Module.set_dtype — MLX 0.16.1 documentation + mlx.nn.Module.set_dtype — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html index 7dbc3dfa28..469a0f4b6f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html @@ -8,7 +8,7 @@ - mlx.nn.Module.state — MLX 0.16.1 documentation + mlx.nn.Module.state — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html index aceaac1cfe..728561c916 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html @@ -8,7 +8,7 @@ - mlx.nn.Module.train — MLX 0.16.1 documentation + mlx.nn.Module.train — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html index 714b1c0ad3..e9f9838e0a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html @@ -8,7 +8,7 @@ - mlx.nn.Module.trainable_parameters — MLX 0.16.1 documentation + mlx.nn.Module.trainable_parameters — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html index 587bf0e325..31d14e7c35 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html @@ -8,7 +8,7 @@ - mlx.nn.Module.training — MLX 0.16.1 documentation + mlx.nn.Module.training — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html index e7eab13a67..a81df789a8 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html @@ -8,7 +8,7 @@ - mlx.nn.Module.unfreeze — MLX 0.16.1 documentation + mlx.nn.Module.unfreeze — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html index 87faba7f5a..6921ecd45b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html @@ -8,7 +8,7 @@ - mlx.nn.Module.update — MLX 0.16.1 documentation + mlx.nn.Module.update — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html index 78952beeab..19f1696711 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html @@ -8,7 +8,7 @@ - mlx.nn.Module.update_modules — MLX 0.16.1 documentation + mlx.nn.Module.update_modules — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html index 18a9011897..933c711679 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html @@ -8,7 +8,7 @@ - mlx.nn.MultiHeadAttention — MLX 0.16.1 documentation + mlx.nn.MultiHeadAttention — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html index 6e7c2c59b2..ab4ce18606 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html @@ -8,7 +8,7 @@ - mlx.nn.PReLU — MLX 0.16.1 documentation + mlx.nn.PReLU — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html index e46aba950f..5fe22c045b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html @@ -8,7 +8,7 @@ - mlx.nn.QuantizedEmbedding — MLX 0.16.1 documentation + mlx.nn.QuantizedEmbedding — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html index d5f1956e4b..626322af91 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html @@ -8,7 +8,7 @@ - mlx.nn.QuantizedLinear — MLX 0.16.1 documentation + mlx.nn.QuantizedLinear — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html index c70e856401..32f14ef361 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html @@ -8,7 +8,7 @@ - mlx.nn.RMSNorm — MLX 0.16.1 documentation + mlx.nn.RMSNorm — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html index b732708ccc..7c0334e65a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html @@ -8,7 +8,7 @@ - mlx.nn.RNN — MLX 0.16.1 documentation + mlx.nn.RNN — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html index 80d40d186d..1f5535f7b8 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html @@ -8,7 +8,7 @@ - mlx.nn.ReLU — MLX 0.16.1 documentation + mlx.nn.ReLU — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html index 99fd5062dd..c6df6a020b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html @@ -8,7 +8,7 @@ - mlx.nn.ReLU6 — MLX 0.16.1 documentation + mlx.nn.ReLU6 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html index 6d1bd45907..112d3b7d28 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html @@ -8,7 +8,7 @@ - mlx.nn.RoPE — MLX 0.16.1 documentation + mlx.nn.RoPE — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html index f06fc7458f..6038dc32d1 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html @@ -8,7 +8,7 @@ - mlx.nn.SELU — MLX 0.16.1 documentation + mlx.nn.SELU — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html index 6f81764c36..cc1dd454cc 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html @@ -8,7 +8,7 @@ - mlx.nn.Sequential — MLX 0.16.1 documentation + mlx.nn.Sequential — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html index 6f3061c97d..f118768f4c 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html @@ -8,7 +8,7 @@ - mlx.nn.SiLU — MLX 0.16.1 documentation + mlx.nn.SiLU — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html index 78c610cf16..46602819e0 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html @@ -8,7 +8,7 @@ - mlx.nn.SinusoidalPositionalEncoding — MLX 0.16.1 documentation + mlx.nn.SinusoidalPositionalEncoding — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html index 43ee457e03..e38106a378 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html @@ -8,7 +8,7 @@ - mlx.nn.Softmax — MLX 0.16.1 documentation + mlx.nn.Softmax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html index db214eb14f..ec3e5ec7c8 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html @@ -8,7 +8,7 @@ - mlx.nn.Softmin — MLX 0.16.1 documentation + mlx.nn.Softmin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html index c8c81f9893..a91c62fa53 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html @@ -8,7 +8,7 @@ - mlx.nn.Softplus — MLX 0.16.1 documentation + mlx.nn.Softplus — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html index 071a7155e8..55a307816b 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html @@ -8,7 +8,7 @@ - mlx.nn.Softshrink — MLX 0.16.1 documentation + mlx.nn.Softshrink — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html index 2a2d05d0bb..8337c04898 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html @@ -8,7 +8,7 @@ - mlx.nn.Softsign — MLX 0.16.1 documentation + mlx.nn.Softsign — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html index 4d8ec244c0..5b848e9eb2 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html @@ -8,7 +8,7 @@ - mlx.nn.Step — MLX 0.16.1 documentation + mlx.nn.Step — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html index b12cfbac6b..28e2977ccc 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html @@ -8,7 +8,7 @@ - mlx.nn.Tanh — MLX 0.16.1 documentation + mlx.nn.Tanh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html index 14a7f6b394..a148b9d226 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html @@ -8,7 +8,7 @@ - mlx.nn.Transformer — MLX 0.16.1 documentation + mlx.nn.Transformer — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html index 7af8bdd261..839a101c5e 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html @@ -8,7 +8,7 @@ - mlx.nn.Upsample — MLX 0.16.1 documentation + mlx.nn.Upsample — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html index 63b0a9b603..caf8e3699f 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html @@ -8,7 +8,7 @@ - mlx.nn.init.constant — MLX 0.16.1 documentation + mlx.nn.init.constant — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html index 2d76d4bec6..3835e0ba27 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html @@ -8,7 +8,7 @@ - mlx.nn.init.glorot_normal — MLX 0.16.1 documentation + mlx.nn.init.glorot_normal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html index 825eecd5ce..30a17ce6d7 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html @@ -8,7 +8,7 @@ - mlx.nn.init.glorot_uniform — MLX 0.16.1 documentation + mlx.nn.init.glorot_uniform — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html index f22971cc0d..ef1ba54d89 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html @@ -8,7 +8,7 @@ - mlx.nn.init.he_normal — MLX 0.16.1 documentation + mlx.nn.init.he_normal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html index 8f04b61b73..a6a17de6ed 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html @@ -8,7 +8,7 @@ - mlx.nn.init.he_uniform — MLX 0.16.1 documentation + mlx.nn.init.he_uniform — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html index 258979d76c..d472ecded5 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html @@ -8,7 +8,7 @@ - mlx.nn.init.identity — MLX 0.16.1 documentation + mlx.nn.init.identity — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html index ed82d0e6d5..e1bfef9955 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html @@ -8,7 +8,7 @@ - mlx.nn.init.normal — MLX 0.16.1 documentation + mlx.nn.init.normal — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html index 1d6dba783e..bdb93caf9d 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html @@ -8,7 +8,7 @@ - mlx.nn.init.uniform — MLX 0.16.1 documentation + mlx.nn.init.uniform — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html index f8392f6f58..d9af0df9fa 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html @@ -8,7 +8,7 @@ - mlx.nn.elu — MLX 0.16.1 documentation + mlx.nn.elu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html index d98a9b4deb..4dd739c42a 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html @@ -8,7 +8,7 @@ - mlx.nn.gelu — MLX 0.16.1 documentation + mlx.nn.gelu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html index 345a0e4052..43d28510ed 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html @@ -8,7 +8,7 @@ - mlx.nn.gelu_approx — MLX 0.16.1 documentation + mlx.nn.gelu_approx — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html index 771f9daef3..1387d6e27c 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html @@ -8,7 +8,7 @@ - mlx.nn.gelu_fast_approx — MLX 0.16.1 documentation + mlx.nn.gelu_fast_approx — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html index 992d1cc101..ee1db5d116 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html @@ -8,7 +8,7 @@ - mlx.nn.glu — MLX 0.16.1 documentation + mlx.nn.glu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html index 5a6478b2b4..820c3db4e0 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html @@ -8,7 +8,7 @@ - mlx.nn.hard_shrink — MLX 0.16.1 documentation + mlx.nn.hard_shrink — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html index f1d32e935a..669467faf3 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html @@ -8,7 +8,7 @@ - mlx.nn.hard_tanh — MLX 0.16.1 documentation + mlx.nn.hard_tanh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html index 2082622cc2..fd1f8e7e59 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html @@ -8,7 +8,7 @@ - mlx.nn.hardswish — MLX 0.16.1 documentation + mlx.nn.hardswish — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html index f10ca6d119..067b2be636 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html @@ -8,7 +8,7 @@ - mlx.nn.leaky_relu — MLX 0.16.1 documentation + mlx.nn.leaky_relu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html index b10b2e2677..42930e70a0 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html @@ -8,7 +8,7 @@ - mlx.nn.log_sigmoid — MLX 0.16.1 documentation + mlx.nn.log_sigmoid — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html index e8fa10d084..764974e6e1 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html @@ -8,7 +8,7 @@ - mlx.nn.log_softmax — MLX 0.16.1 documentation + mlx.nn.log_softmax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html index c8844ece70..3e8531adfb 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html @@ -8,7 +8,7 @@ - mlx.nn.losses.binary_cross_entropy — MLX 0.16.1 documentation + mlx.nn.losses.binary_cross_entropy — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html index 0500a996dc..d83a6ad393 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.cosine_similarity_loss — MLX 0.16.1 documentation + mlx.nn.losses.cosine_similarity_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html index ee92ba8582..545a64f5b3 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html @@ -8,7 +8,7 @@ - mlx.nn.losses.cross_entropy — MLX 0.16.1 documentation + mlx.nn.losses.cross_entropy — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html index 21c2255e45..290a41c283 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.gaussian_nll_loss — MLX 0.16.1 documentation + mlx.nn.losses.gaussian_nll_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html index 3a1e83d798..a1bf1035d6 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.hinge_loss — MLX 0.16.1 documentation + mlx.nn.losses.hinge_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html index 5aea944b78..d595efb055 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.huber_loss — MLX 0.16.1 documentation + mlx.nn.losses.huber_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html index c377bf25c5..0a8104a8cf 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.kl_div_loss — MLX 0.16.1 documentation + mlx.nn.losses.kl_div_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html index f500ba8d0f..16e0fba567 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.l1_loss — MLX 0.16.1 documentation + mlx.nn.losses.l1_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html index 43061455fc..4ec702b322 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.log_cosh_loss — MLX 0.16.1 documentation + mlx.nn.losses.log_cosh_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html index 7c9e6ce48f..e18dd0bf80 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.margin_ranking_loss — MLX 0.16.1 documentation + mlx.nn.losses.margin_ranking_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html index 8a7af2f754..e4af5b17fd 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.mse_loss — MLX 0.16.1 documentation + mlx.nn.losses.mse_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html index 435976e5c8..b7bf192900 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.nll_loss — MLX 0.16.1 documentation + mlx.nn.losses.nll_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html index 76093ff234..5126880c4c 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.smooth_l1_loss — MLX 0.16.1 documentation + mlx.nn.losses.smooth_l1_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html index f5a8c26ce7..65656bfd79 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html @@ -8,7 +8,7 @@ - mlx.nn.losses.triplet_loss — MLX 0.16.1 documentation + mlx.nn.losses.triplet_loss — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html index e71fdd2944..17894f6a12 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html @@ -8,7 +8,7 @@ - mlx.nn.mish — MLX 0.16.1 documentation + mlx.nn.mish — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html index f1f6f0785d..f9eb91ef79 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html @@ -8,7 +8,7 @@ - mlx.nn.prelu — MLX 0.16.1 documentation + mlx.nn.prelu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html index 7918dc7222..f78b97d9e3 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html @@ -8,7 +8,7 @@ - mlx.nn.relu — MLX 0.16.1 documentation + mlx.nn.relu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html index 760b04d3a8..508c68c647 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html @@ -8,7 +8,7 @@ - mlx.nn.relu6 — MLX 0.16.1 documentation + mlx.nn.relu6 — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html index fd702a7be6..5c6c797ca5 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html @@ -8,7 +8,7 @@ - mlx.nn.selu — MLX 0.16.1 documentation + mlx.nn.selu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html index 845f214883..e70c9f8234 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html @@ -8,7 +8,7 @@ - mlx.nn.sigmoid — MLX 0.16.1 documentation + mlx.nn.sigmoid — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html index aa65e46a15..daa56f0104 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html @@ -8,7 +8,7 @@ - mlx.nn.silu — MLX 0.16.1 documentation + mlx.nn.silu — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html index fe0e6fa1d2..23fb287777 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html @@ -8,7 +8,7 @@ - mlx.nn.softmax — MLX 0.16.1 documentation + mlx.nn.softmax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html index e8f2fd38ec..d0db3d76c0 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html @@ -8,7 +8,7 @@ - mlx.nn.softmin — MLX 0.16.1 documentation + mlx.nn.softmin — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html index ac4c9990ae..0b53d6aa54 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html @@ -8,7 +8,7 @@ - mlx.nn.softplus — MLX 0.16.1 documentation + mlx.nn.softplus — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html index 99cfef2e3a..25eb71777b 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html @@ -8,7 +8,7 @@ - mlx.nn.softshrink — MLX 0.16.1 documentation + mlx.nn.softshrink — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html index a0ef6a7f42..4fdd92246f 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html @@ -8,7 +8,7 @@ - mlx.nn.step — MLX 0.16.1 documentation + mlx.nn.step — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html index d47ac4b047..94baa89040 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html @@ -8,7 +8,7 @@ - mlx.nn.tanh — MLX 0.16.1 documentation + mlx.nn.tanh — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/functions.html b/docs/build/html/python/nn/functions.html index 11573a5d62..7bd6279974 100644 --- a/docs/build/html/python/nn/functions.html +++ b/docs/build/html/python/nn/functions.html @@ -8,7 +8,7 @@ - Functions — MLX 0.16.1 documentation + Functions — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/init.html b/docs/build/html/python/nn/init.html index f65c6caeba..ece68041b1 100644 --- a/docs/build/html/python/nn/init.html +++ b/docs/build/html/python/nn/init.html @@ -8,7 +8,7 @@ - Initializers — MLX 0.16.1 documentation + Initializers — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/layers.html b/docs/build/html/python/nn/layers.html index 1c08a6569d..f584c69d31 100644 --- a/docs/build/html/python/nn/layers.html +++ b/docs/build/html/python/nn/layers.html @@ -8,7 +8,7 @@ - Layers — MLX 0.16.1 documentation + Layers — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/losses.html b/docs/build/html/python/nn/losses.html index e5072b9f7c..ad8fed1c98 100644 --- a/docs/build/html/python/nn/losses.html +++ b/docs/build/html/python/nn/losses.html @@ -8,7 +8,7 @@ - Loss Functions — MLX 0.16.1 documentation + Loss Functions — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/nn/module.html b/docs/build/html/python/nn/module.html index 87527034de..2b540e849f 100644 --- a/docs/build/html/python/nn/module.html +++ b/docs/build/html/python/nn/module.html @@ -8,7 +8,7 @@ - Module — MLX 0.16.1 documentation + Module — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/ops.html b/docs/build/html/python/ops.html index 546c3b8aaa..817316afb6 100644 --- a/docs/build/html/python/ops.html +++ b/docs/build/html/python/ops.html @@ -8,7 +8,7 @@ - Operations — MLX 0.16.1 documentation + Operations — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra @@ -1153,7 +1155,7 @@ document.write(`
            • - + diff --git a/docs/build/html/python/optimizers.html b/docs/build/html/python/optimizers.html index fbca6b6c92..904fe81c64 100644 --- a/docs/build/html/python/optimizers.html +++ b/docs/build/html/python/optimizers.html @@ -8,7 +8,7 @@ - Optimizers — MLX 0.16.1 documentation + Optimizers — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html index 376b5785c1..06ef95c6ce 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdaDelta.html @@ -8,7 +8,7 @@ - mlx.optimizers.AdaDelta — MLX 0.16.1 documentation + mlx.optimizers.AdaDelta — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html index 9c1569a586..d7fae7f90f 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html @@ -8,7 +8,7 @@ - mlx.optimizers.Adafactor — MLX 0.16.1 documentation + mlx.optimizers.Adafactor — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html index 9e55aa2240..5b02064f32 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html @@ -8,7 +8,7 @@ - mlx.optimizers.Adagrad — MLX 0.16.1 documentation + mlx.optimizers.Adagrad — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html index ef6e72ac8b..bbf11a3578 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html @@ -8,7 +8,7 @@ - mlx.optimizers.Adam — MLX 0.16.1 documentation + mlx.optimizers.Adam — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html index e305f9f31a..d96cd42abb 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html @@ -8,7 +8,7 @@ - mlx.optimizers.AdamW — MLX 0.16.1 documentation + mlx.optimizers.AdamW — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html index 3402bf78aa..5c37e0f8a6 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html @@ -8,7 +8,7 @@ - mlx.optimizers.Adamax — MLX 0.16.1 documentation + mlx.optimizers.Adamax — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html index 834bb80bd3..9a9526bcbf 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html @@ -8,7 +8,7 @@ - mlx.optimizers.Lion — MLX 0.16.1 documentation + mlx.optimizers.Lion — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html index 295e9788ab..245d416f06 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html @@ -8,7 +8,7 @@ - mlx.optimizers.Optimizer.apply_gradients — MLX 0.16.1 documentation + mlx.optimizers.Optimizer.apply_gradients — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html index 6bd5e76bb8..75b10af5b3 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html @@ -8,7 +8,7 @@ - mlx.optimizers.Optimizer.init — MLX 0.16.1 documentation + mlx.optimizers.Optimizer.init — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html index d59b3227a9..b3ec72da49 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html @@ -8,7 +8,7 @@ - mlx.optimizers.Optimizer.state — MLX 0.16.1 documentation + mlx.optimizers.Optimizer.state — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html index 01a13c1d32..e3e63eded9 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html @@ -8,7 +8,7 @@ - mlx.optimizers.Optimizer.update — MLX 0.16.1 documentation + mlx.optimizers.Optimizer.update — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html index 4acd664a35..0395767bdc 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html @@ -8,7 +8,7 @@ - mlx.optimizers.RMSprop — MLX 0.16.1 documentation + mlx.optimizers.RMSprop — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html index ed5854ec54..be27150d0d 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html @@ -8,7 +8,7 @@ - mlx.optimizers.SGD — MLX 0.16.1 documentation + mlx.optimizers.SGD — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html index 7f60ac15f2..7de71d75db 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html @@ -8,7 +8,7 @@ - mlx.optimizers.cosine_decay — MLX 0.16.1 documentation + mlx.optimizers.cosine_decay — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html index 1bebc60c37..523a58f984 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html @@ -8,7 +8,7 @@ - mlx.optimizers.exponential_decay — MLX 0.16.1 documentation + mlx.optimizers.exponential_decay — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html index c5ef73578d..119664df9d 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html @@ -8,7 +8,7 @@ - mlx.optimizers.join_schedules — MLX 0.16.1 documentation + mlx.optimizers.join_schedules — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -132,8 +132,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -458,8 +458,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html index 8b9ee6ebde..8accfb0c27 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html @@ -8,7 +8,7 @@ - mlx.optimizers.linear_schedule — MLX 0.16.1 documentation + mlx.optimizers.linear_schedule — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html index 5ab5faaeb4..39eb1578bc 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html @@ -8,7 +8,7 @@ - mlx.optimizers.step_decay — MLX 0.16.1 documentation + mlx.optimizers.step_decay — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/common_optimizers.html b/docs/build/html/python/optimizers/common_optimizers.html index 10418ec70b..4fde1bb16c 100644 --- a/docs/build/html/python/optimizers/common_optimizers.html +++ b/docs/build/html/python/optimizers/common_optimizers.html @@ -8,7 +8,7 @@ - Common Optimizers — MLX 0.16.1 documentation + Common Optimizers — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/optimizer.html b/docs/build/html/python/optimizers/optimizer.html index 56fea43f2b..3ba9e50e42 100644 --- a/docs/build/html/python/optimizers/optimizer.html +++ b/docs/build/html/python/optimizers/optimizer.html @@ -8,7 +8,7 @@ - Optimizer — MLX 0.16.1 documentation + Optimizer — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/optimizers/schedulers.html b/docs/build/html/python/optimizers/schedulers.html index a9cb0de38d..c1e2568dab 100644 --- a/docs/build/html/python/optimizers/schedulers.html +++ b/docs/build/html/python/optimizers/schedulers.html @@ -8,7 +8,7 @@ - Schedulers — MLX 0.16.1 documentation + Schedulers — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/random.html b/docs/build/html/python/random.html index 90e93be58e..ee34961a15 100644 --- a/docs/build/html/python/random.html +++ b/docs/build/html/python/random.html @@ -8,7 +8,7 @@ - Random — MLX 0.16.1 documentation + Random — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/transforms.html b/docs/build/html/python/transforms.html index 44516b1cb9..eb0746fff7 100644 --- a/docs/build/html/python/transforms.html +++ b/docs/build/html/python/transforms.html @@ -8,7 +8,7 @@ - Transforms — MLX 0.16.1 documentation + Transforms — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/tree_utils.html b/docs/build/html/python/tree_utils.html index 30f59c7754..09224128cd 100644 --- a/docs/build/html/python/tree_utils.html +++ b/docs/build/html/python/tree_utils.html @@ -8,7 +8,7 @@ - Tree Utils — MLX 0.16.1 documentation + Tree Utils — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/quantized_8h.html b/docs/build/html/quantized_8h.html index decdc7644b..9184a2c2c9 100644 --- a/docs/build/html/quantized_8h.html +++ b/docs/build/html/quantized_8h.html @@ -127,10 +127,10 @@ Functions
            •  - + - + @@ -166,6 +166,15 @@ Functions + + + + + + + + +

            __init__(*args, **kwargs)

            complexfloating

            floating

            inexact

            signedinteger

            unsignedinteger

            integer

            number

            generic

            Dtype

            An object to hold the type of a array.

            DtypeCategory

            DtypeCategory(value)

            Type to hold categories of dtypes.

            issubdtype(-> bool diff --git a/docs/build/html/python/devices_and_streams.html b/docs/build/html/python/devices_and_streams.html index 5539dcc179..1ef7107e8b 100644 --- a/docs/build/html/python/devices_and_streams.html +++ b/docs/build/html/python/devices_and_streams.html @@ -8,7 +8,7 @@ - Devices and Streams — MLX 0.16.1 documentation + Devices and Streams — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@

          • Linear Algebra diff --git a/docs/build/html/python/distributed.html b/docs/build/html/python/distributed.html index 0b3d4c707e..e133e40a28 100644 --- a/docs/build/html/python/distributed.html +++ b/docs/build/html/python/distributed.html @@ -8,7 +8,7 @@ - Distributed Communication — MLX 0.16.1 documentation + Distributed Communication — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/fast.html b/docs/build/html/python/fast.html index 799b71d3c5..2bedb6ad60 100644 --- a/docs/build/html/python/fast.html +++ b/docs/build/html/python/fast.html @@ -8,7 +8,7 @@ - Fast — MLX 0.16.1 documentation + Fast — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/fft.html b/docs/build/html/python/fft.html index 03a1988719..212b145e35 100644 --- a/docs/build/html/python/fft.html +++ b/docs/build/html/python/fft.html @@ -8,7 +8,7 @@ - FFT — MLX 0.16.1 documentation + FFT — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/python/linalg.html b/docs/build/html/python/linalg.html index a41552f2b6..96a9477068 100644 --- a/docs/build/html/python/linalg.html +++ b/docs/build/html/python/linalg.html @@ -8,7 +8,7 @@ - Linear Algebra — MLX 0.16.1 documentation + Linear Algebra — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra @@ -839,12 +841,18 @@ document.write(`

            • inv(a, *[, stream])

            Compute the inverse of a square matrix.

            norm(a, /[, ord, axis, keepdims, stream])

            tri_inv(a[, upper, stream])

            Compute the inverse of a triangular square matrix.

            norm(a, /[, ord, axis, keepdims, stream])

            Matrix or vector norm.

            cholesky(a[, upper, stream])

            cholesky(a[, upper, stream])

            Compute the Cholesky decomposition of a real symmetric positive semi-definite matrix.

            cholesky_inv(L[, upper, stream])

            Compute the inverse of a real symmetric positive semi-definite matrix using it's Cholesky decomposition L.

            qr(a, *[, stream])

            The QR factorization of the input matrix.

            partition(a, /, kth[, axis, stream])

            Returns a partitioned copy of the array such that the smaller kth elements are first.

            pad(a, pad_width[, constant_values, stream])

            pad(a, pad_width[, mode, constant_values, ...])

            Pad an array with a constant value

            power(a, b, /, *[, stream])

            template<typename T , const int group_size, const int bits>
            METAL_FUNC void qvm_impl (const device T *x, const device uint32_t *w, const device T *scales, const device T *biases, device T *y, const constant int &in_vec_size, const constant int &out_vec_size, uint3 tid, uint simd_gid, uint simd_lid)
             
            template<typename T , const int BM, const int BK, const int BN, const int group_size, const int bits, const bool aligned_N>
            template<typename T , const int group_size, const int bits, const bool aligned_N, const int BM = 32, const int BK = 32, const int BN = 32>
            METAL_FUNC void qmm_t_impl (const device T *x, const device uint32_t *w, const device T *scales, const device T *biases, device T *y, threadgroup T *Xs, threadgroup T *Ws, const constant int &M, const constant int &N, const constant int &K, uint3 tid, uint lid, uint simd_gid, uint simd_lid)
             
            template<typename T , const int BM, const int BK, const int BN, const int group_size, const int bits>
            template<typename T , const int group_size, const int bits, const int BM = 32, const int BK = 32, const int BN = 32>
            METAL_FUNC void qmm_n_impl (const device T *x, const device uint32_t *w, const device T *scales, const device T *biases, device T *y, threadgroup T *Xs, threadgroup T *Ws, const constant int &M, const constant int &N, const constant int &K, uint3 tid, uint lid, uint simd_gid, uint simd_lid)
             
            template<typename T >
            template<typename T , const int group_size, const int bits, const int BM = 32, const int BK = 32, const int BN = 32>
            void bs_qmm_n (const device T *x, const device uint32_t *w, const device T *scales, const device T *biases, const device uint32_t *lhs_indices, const device uint32_t *rhs_indices, device T *y, const constant int &M, const constant int &N, const constant int &K, const constant int &batch_ndims, const constant int *batch_shape, const constant size_t *lhs_strides, const constant size_t *rhs_strides, const constant int &x_batch_ndims, const constant int *x_shape, const constant size_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant size_t *w_strides, const constant size_t *s_strides, const constant size_t *b_strides, uint3 tid, uint lid, uint simd_gid, uint simd_lid)
             
            template<typename T , const int group_size, const int bits>
            void affine_quantize (const device T *w, device uint8_t *out, device T *scales, device T *biases, uint index)
             
            template<typename T , const int group_size, const int bits>
            void affine_quantize_scales_biases (const device T *w, const device T *scales, const device T *biases, device uint8_t *out, uint index)
             
            template<typename T , const int group_size, const int bits>
            void affine_dequantize (const device uint8_t *w, const device T *scales, const device T *biases, device T *out, uint index)
             
            @@ -304,6 +313,120 @@ template<typename T >

            Variables

            • +
            +
            + +

            ◆ affine_dequantize()

            + +
            +
            +
            +template<typename T , const int group_size, const int bits>
            + + + + + + + + + + + + + + + + + + + + + + + + + + +
            void affine_dequantize (const device uint8_t * w,
            const device T * scales,
            const device T * biases,
            device T * out,
            uint index )
            +
            + +
            +
            + +

            ◆ affine_quantize()

            + +
            +
            +
            +template<typename T , const int group_size, const int bits>
            + + + + + + + + + + + + + + + + + + + + + + + + + + +
            void affine_quantize (const device T * w,
            device uint8_t * out,
            device T * scales,
            device T * biases,
            uint index )
            +
            + +
            +
            + +

            ◆ affine_quantize_scales_biases()

            + +
            +
            +
            +template<typename T , const int group_size, const int bits>
            + + + + + + + + + + + + + + + + + + + + + + + + + + +
            void affine_quantize_scales_biases (const device T * w,
            const device T * scales,
            const device T * biases,
            device uint8_t * out,
            uint index )
            +
            +
            @@ -1275,7 +1398,7 @@ template<typename T , const int group_size, const int bits, const int BM = 32
            -template<typename T , const int BM, const int BK, const int BN, const int group_size, const int bits>
            +template<typename T , const int group_size, const int bits, const int BM = 32, const int BK = 32, const int BN = 32>
            @@ -1431,7 +1554,7 @@ template<typename T , const int group_size, const int bits, const bool aligne
            -template<typename T , const int BM, const int BK, const int BN, const int group_size, const int bits, const bool aligned_N>
            +template<typename T , const int group_size, const int bits, const bool aligned_N, const int BM = 32, const int BK = 32, const int BN = 32>
            METAL_FUNC void qmm_n_impl
            diff --git a/docs/build/html/quantized_8h_source.html b/docs/build/html/quantized_8h_source.html index 2b7cea96a4..c825c7ac3c 100644 --- a/docs/build/html/quantized_8h_source.html +++ b/docs/build/html/quantized_8h_source.html @@ -803,12 +803,12 @@ $(function() { codefold.init(0); });
            690
            691template <
            692 typename T,
            -
            693 const int BM,
            -
            694 const int BK,
            -
            695 const int BN,
            -
            696 const int group_size,
            -
            697 const int bits,
            -
            698 const bool aligned_N>
            +
            693 const int group_size,
            +
            694 const int bits,
            +
            695 const bool aligned_N,
            +
            696 const int BM = 32,
            +
            697 const int BK = 32,
            +
            698 const int BN = 32>
            699METAL_FUNC void qmm_t_impl(
            700 const device T* x,
            @@ -927,11 +927,11 @@ $(function() { codefold.init(0); });
            812
            813template <
            814 typename T,
            -
            815 const int BM,
            -
            816 const int BK,
            -
            817 const int BN,
            -
            818 const int group_size,
            -
            819 const int bits>
            +
            815 const int group_size,
            +
            816 const int bits,
            +
            817 const int BM = 32,
            +
            818 const int BK = 32,
            +
            819 const int BN = 32>
            820METAL_FUNC void qmm_n_impl(
            821 const device T* x,
            @@ -1225,7 +1225,7 @@ $(function() { codefold.init(0); });
            1099 threadgroup T Xs[BM * BK_padded];
            1100 threadgroup T Ws[BN * BK_padded];
            1101
            -
            1102 qmm_t_impl<T, BM, BK, BN, group_size, bits, aligned_N>(
            +
            1102 qmm_t_impl<T, group_size, bits, aligned_N, BM, BK, BN>(
            1103 x, w, scales, biases, y, Xs, Ws, M, N, K, tid, lid, simd_gid, simd_lid);
            1104}
            @@ -1259,7 +1259,7 @@ $(function() { codefold.init(0); });
            1131 threadgroup T Xs[BM * BK_padded];
            1132 threadgroup T Ws[BK * BN_padded];
            1133
            -
            1134 qmm_n_impl<T, BM, BK, BN, group_size, bits>(
            +
            1134 qmm_n_impl<T, group_size, bits, BM, BK, BN>(
            1135 x, w, scales, biases, y, Xs, Ws, M, N, K, tid, lid, simd_gid, simd_lid);
            1136}
            @@ -1518,7 +1518,7 @@ $(function() { codefold.init(0); });
            1382 s_strides,
            1383 b_strides,
            1384 tid);
            -
            1385 qmm_t_impl<T, BM, BK, BN, group_size, bits, aligned_N>(
            +
            1385 qmm_t_impl<T, group_size, bits, aligned_N, BM, BK, BN>(
            1386 x, w, scales, biases, y, Xs, Ws, M, N, K, tid, lid, simd_gid, simd_lid);
            1387}
            @@ -1588,15 +1588,169 @@ $(function() { codefold.init(0); });
            1450 s_strides,
            1451 b_strides,
            1452 tid);
            -
            1453 qmm_n_impl<T, BM, BK, BN, group_size, bits>(
            +
            1453 qmm_n_impl<T, group_size, bits, BM, BK, BN>(
            1454 x, w, scales, biases, y, Xs, Ws, M, N, K, tid, lid, simd_gid, simd_lid);
            1455}
            +
            1456
            +
            1457template <typename T, const int group_size, const int bits>
            +
            +
            1458[[kernel]] void affine_quantize(
            +
            1459 const device T* w [[buffer(0)]],
            +
            1460 device uint8_t* out [[buffer(1)]],
            +
            1461 device T* scales [[buffer(2)]],
            +
            1462 device T* biases [[buffer(3)]],
            +
            1463 uint index [[thread_position_in_grid]]) {
            +
            1464 constexpr T eps = T(1e-7);
            +
            1465 constexpr int simd_size = 32;
            +
            1466 constexpr int uint8_bits = 8;
            +
            1467 constexpr T n_bins = (1 << bits) - 1;
            +
            1468 constexpr int packs_per_int = uint8_bits / bits;
            +
            1469 constexpr int values_per_reduce = group_size / simd_size;
            +
            1470 constexpr int writes_per_reduce = packs_per_int / values_per_reduce;
            +
            1471 constexpr int writes_per_pack =
            +
            1472 writes_per_reduce > 1 ? 1 : values_per_reduce / packs_per_int;
            +
            1473
            +
            1474 static_assert(
            +
            1475 group_size % simd_size == 0,
            +
            1476 "Group size must be divisible by simd size.");
            +
            1477
            +
            1478 int in_index = index * values_per_reduce;
            +
            1479 int out_index = index * writes_per_pack;
            +
            1480
            +
            1481 T w_thread[values_per_reduce];
            +
            1482 T w_min = Limits<T>::max;
            +
            1483 T w_max = 0;
            +
            1484
            +
            1485#pragma clang loop unroll(full)
            +
            1486 for (int i = 0; i < values_per_reduce; i++) {
            +
            1487 T val = w[in_index + i];
            +
            1488 w_thread[i] = val;
            +
            1489 w_min = min(w_min, val);
            +
            1490 w_max = max(w_max, val);
            +
            1491 }
            +
            1492
            +
            1493 w_min = simd_min(w_min);
            +
            1494 w_max = simd_max(w_max);
            +
            1495
            +
            1496 T scale = max((w_max - w_min) / n_bins, eps);
            +
            1497 bool side = abs(w_min) > abs(w_max);
            +
            1498 scale = side ? scale : -scale;
            +
            1499 T edge = side ? w_min : w_max;
            +
            1500 T q0 = round(edge / scale);
            +
            1501 bool at_zero = q0 == 0.0f;
            +
            1502 scale = at_zero ? scale : edge / q0;
            +
            1503 T bias = at_zero ? T(0) : edge;
            +
            1504
            +
            1505 // Write out the scales and biases
            +
            1506 int gindex = in_index / group_size;
            +
            1507 if (in_index % group_size == 0) {
            +
            1508 scales[gindex] = scale;
            +
            1509 biases[gindex] = bias;
            +
            1510 }
            +
            1511
            +
            1512 uint8_t output = 0;
            +
            1513#pragma clang loop unroll(full)
            +
            1514 for (int i = 0; i < values_per_reduce; i++) {
            +
            1515 uint8_t val = min(round((w_thread[i] - bias) / scale), n_bins);
            +
            1516 if (bits == 8) {
            +
            1517 output = val;
            +
            1518 } else {
            +
            1519 output += val << (bits * (i % packs_per_int));
            +
            1520 }
            +
            1521
            +
            1522 if (packs_per_int < values_per_reduce &&
            +
            1523 i % packs_per_int == packs_per_int - 1) {
            +
            1524 out[out_index + i / packs_per_int] = output;
            +
            1525 output = 0;
            +
            1526 } else {
            +
            1527#pragma clang loop unroll(full)
            +
            1528 for (int j = 0; j < writes_per_reduce - 1; j++) {
            +
            1529 uint8_t sval = simd_shuffle_down(val, j + 1);
            +
            1530 output += sval << (bits * (values_per_reduce + j + i));
            +
            1531 }
            +
            1532 }
            +
            1533 }
            +
            1534 if (writes_per_reduce > 0 && out_index % writes_per_reduce == 0) {
            +
            1535 out[out_index / writes_per_reduce] = output;
            +
            1536 }
            +
            1537}
            +
            +
            1538
            +
            1539template <typename T, const int group_size, const int bits>
            +
            +
            1540[[kernel]] void affine_quantize_scales_biases(
            +
            1541 const device T* w [[buffer(0)]],
            +
            1542 const device T* scales [[buffer(1)]],
            +
            1543 const device T* biases [[buffer(2)]],
            +
            1544 device uint8_t* out [[buffer(3)]],
            +
            1545 uint index [[thread_position_in_grid]]) {
            +
            1546 constexpr int uint8_bits = 8;
            +
            1547 constexpr int packs_per_int = uint8_bits / bits;
            +
            1548 constexpr T n_bins = (1 << bits) - 1;
            +
            1549
            +
            1550 int in_index = index * packs_per_int;
            +
            1551 int gindex = in_index / group_size;
            +
            1552 T scale = scales[gindex];
            +
            1553 T bias = biases[gindex];
            +
            1554
            +
            1555 uint8_t output = 0;
            +
            1556#pragma clang loop unroll(full)
            +
            1557 for (int i = 0; i < packs_per_int; i++) {
            +
            1558 uint8_t val = min(round((w[in_index + i] - bias) / scale), n_bins);
            +
            1559 if (bits == 8) {
            +
            1560 output = val;
            +
            1561 } else {
            +
            1562 output += val << (bits * i);
            +
            1563 }
            +
            1564 }
            +
            1565 out[index] = output;
            +
            1566}
            +
            +
            1567
            +
            1568template <typename T, const int group_size, const int bits>
            +
            +
            1569[[kernel]] void affine_dequantize(
            +
            1570 const device uint8_t* w [[buffer(0)]],
            +
            1571 const device T* scales [[buffer(1)]],
            +
            1572 const device T* biases [[buffer(2)]],
            +
            1573 device T* out [[buffer(3)]],
            +
            1574 uint index [[thread_position_in_grid]]) {
            +
            1575 constexpr int uint8_bits = 8;
            +
            1576 constexpr int packs_per_int = uint8_bits / bits;
            +
            1577
            +
            1578 int oindex = index * packs_per_int;
            +
            1579 int gindex = oindex / group_size;
            +
            1580 T scale = scales[gindex];
            +
            1581 T bias = biases[gindex];
            +
            1582 uint val = w[index];
            +
            1583
            +
            1584#pragma clang loop unroll(full)
            +
            1585 for (int i = 0; i < packs_per_int; i++) {
            +
            1586 uint8_t d;
            +
            1587 if (bits == 2) {
            +
            1588 d = (val >> (bits * i)) & 0x03;
            +
            1589 } else if (bits == 4) {
            +
            1590 d = (val >> (bits * i)) & 0x0f;
            +
            1591 } else if (bits == 8) {
            +
            1592 d = val;
            +
            1593 }
            +
            1594 out[oindex + i] = scale * d + bias;
            +
            1595 }
            +
            1596}
            +
            +
            simd_size
            static constant constexpr const uint8_t simd_size
            Definition ops.h:8
            elem_to_loc_broadcast
            METAL_FUNC ulong2 elem_to_loc_broadcast(uint elem, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, int ndim)
            Definition utils.h:7
            elem_to_loc
            METAL_FUNC stride_t elem_to_loc(uint elem, device const int *shape, device const stride_t *strides, int ndim)
            Definition utils.h:77
            metal
            Definition bf16.h:265
            +
            metal::simd_max
            METAL_FUNC bfloat16_t simd_max(bfloat16_t data)
            Definition bf16_math.h:392
            +
            metal::round
            METAL_FUNC bfloat16_t round(bfloat16_t x)
            Definition bf16_math.h:234
            metal::min
            METAL_FUNC bfloat16_t min(bfloat16_t x, bfloat16_t y)
            Definition bf16_math.h:234
            metal::simd_sum
            METAL_FUNC bfloat16_t simd_sum(bfloat16_t data)
            Definition bf16_math.h:392
            +
            metal::max
            METAL_FUNC bfloat16_t max(bfloat16_t x, bfloat16_t y)
            Definition bf16_math.h:234
            +
            metal::abs
            METAL_FUNC bfloat16_t abs(bfloat16_t x)
            Definition bf16_math.h:234
            +
            metal::simd_min
            METAL_FUNC bfloat16_t simd_min(bfloat16_t data)
            Definition bf16_math.h:392
            +
            metal::simd_shuffle_down
            METAL_FUNC bfloat16_t simd_shuffle_down(bfloat16_t data, ushort delta)
            Definition bf16_math.h:391
            mlx::core::random::bits
            array bits(const std::vector< int > &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})
            Generate an array with type uint32 filled with random bits.
            MLX_MTL_CONST
            #define MLX_MTL_CONST
            Definition quantized.h:8
            bs_qmv
            void bs_qmv(const device uint32_t *w, const device T *scales, const device T *biases, const device T *x, const device uint32_t *lhs_indices, const device uint32_t *rhs_indices, device T *y, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &batch_ndims, const constant int *batch_shape, const constant size_t *lhs_strides, const constant size_t *rhs_strides, const constant int &x_batch_ndims, const constant int *x_shape, const constant size_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant size_t *w_strides, const constant size_t *s_strides, const constant size_t *b_strides, uint3 tid, uint simd_gid, uint simd_lid)
            Definition quantized.h:1200
            @@ -1605,6 +1759,8 @@ $(function() { codefold.init(0); });
            adjust_matrix_offsets
            METAL_FUNC void adjust_matrix_offsets(const device T *&x, const device uint32_t *&w, const device T *&scales, const device T *&biases, const device uint32_t *lhs_indices, const device uint32_t *rhs_indices, device T *&y, int output_stride, const constant int &batch_ndims, const constant int *batch_shape, const constant size_t *lhs_strides, const constant size_t *rhs_strides, const constant int &x_batch_ndims, const constant int *x_shape, const constant size_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant size_t *w_strides, const constant size_t *s_strides, const constant size_t *b_strides, uint3 tid)
            Definition quantized.h:946
            bs_qmm_t
            void bs_qmm_t(const device T *x, const device uint32_t *w, const device T *scales, const device T *biases, const device uint32_t *lhs_indices, const device uint32_t *rhs_indices, device T *y, const constant int &M, const constant int &N, const constant int &K, const constant int &batch_ndims, const constant int *batch_shape, const constant size_t *lhs_strides, const constant size_t *rhs_strides, const constant int &x_batch_ndims, const constant int *x_shape, const constant size_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant size_t *w_strides, const constant size_t *s_strides, const constant size_t *b_strides, uint3 tid, uint lid, uint simd_gid, uint simd_lid)
            Definition quantized.h:1329
            bs_qvm
            void bs_qvm(const device T *x, const device uint32_t *w, const device T *scales, const device T *biases, const device uint32_t *lhs_indices, const device uint32_t *rhs_indices, device T *y, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &batch_ndims, const constant int *batch_shape, const constant size_t *lhs_strides, const constant size_t *rhs_strides, const constant int &x_batch_ndims, const constant int *x_shape, const constant size_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant size_t *w_strides, const constant size_t *s_strides, const constant size_t *b_strides, uint3 tid, uint simd_gid, uint simd_lid)
            Definition quantized.h:1261
            +
            affine_quantize
            void affine_quantize(const device T *w, device uint8_t *out, device T *scales, device T *biases, uint index)
            Definition quantized.h:1458
            +
            affine_quantize_scales_biases
            void affine_quantize_scales_biases(const device T *w, const device T *scales, const device T *biases, device uint8_t *out, uint index)
            Definition quantized.h:1540
            qmm_n_impl
            METAL_FUNC void qmm_n_impl(const device T *x, const device uint32_t *w, const device T *scales, const device T *biases, device T *y, threadgroup T *Xs, threadgroup T *Ws, const constant int &M, const constant int &N, const constant int &K, uint3 tid, uint lid, uint simd_gid, uint simd_lid)
            Definition quantized.h:820
            SIMD_SIZE
            static constant constexpr const int SIMD_SIZE
            Definition quantized.h:10
            bs_qmv_fast
            void bs_qmv_fast(const device uint32_t *w, const device T *scales, const device T *biases, const device T *x, const device uint32_t *lhs_indices, const device uint32_t *rhs_indices, device T *y, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &batch_ndims, const constant int *batch_shape, const constant size_t *lhs_strides, const constant size_t *rhs_strides, const constant int &x_batch_ndims, const constant int *x_shape, const constant size_t *x_strides, const constant int &w_batch_ndims, const constant int *w_shape, const constant size_t *w_strides, const constant size_t *s_strides, const constant size_t *b_strides, uint3 tid, uint simd_gid, uint simd_lid)
            Definition quantized.h:1139
            @@ -1620,8 +1776,10 @@ $(function() { codefold.init(0); });
            qmv_fast
            void qmv_fast(const device uint32_t *w, const device T *scales, const device T *biases, const device T *x, device T *y, const constant int &in_vec_size, const constant int &out_vec_size, uint3 tid, uint simd_gid, uint simd_lid)
            Definition quantized.h:1000
            qmv
            void qmv(const device uint32_t *w, const device T *scales, const device T *biases, const device T *x, device T *y, const constant int &in_vec_size, const constant int &out_vec_size, uint3 tid, uint simd_gid, uint simd_lid)
            Definition quantized.h:1025
            qouter
            void qouter(const thread uint8_t *w, U x, U scale, U bias, thread U *result)
            Definition quantized.h:186
            +
            affine_dequantize
            void affine_dequantize(const device uint8_t *w, const device T *scales, const device T *biases, device T *out, uint index)
            Definition quantized.h:1569
            dequantize
            void dequantize(const device uint8_t *w, U scale, U bias, threadgroup U *w_local)
            Definition quantized.h:218
            qmm_t
            void qmm_t(const device T *x, const device uint32_t *w, const device T *scales, const device T *biases, device T *y, const constant int &M, const constant int &N, const constant int &K, uint3 tid, uint lid, uint simd_gid, uint simd_lid)
            Definition quantized.h:1082
            +
            Limits
            Definition utils.h:17
            QuantizedBlockLoader
            Definition quantized.h:261
            QuantizedBlockLoader::group_stride
            const int group_stride
            Definition quantized.h:281
            QuantizedBlockLoader::BCOLS_PACKED
            static constant constexpr const short BCOLS_PACKED
            Definition quantized.h:273
            diff --git a/docs/build/html/random_8h_source.html b/docs/build/html/random_8h_source.html index 5142138ee0..7afbda9a3f 100644 --- a/docs/build/html/random_8h_source.html +++ b/docs/build/html/random_8h_source.html @@ -379,11 +379,11 @@ $(function() { codefold.init(0); });
            mlx::core::random::laplace
            array laplace(const std::vector< int > &shape, Dtype dtype, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})
            Generate samples from the laplace distribution.
            mlx::core::random::uniform
            array uniform(const array &low, const array &high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})
            Generate uniform random numbers between low and high.
            mlx::core::to_stream
            Stream to_stream(StreamOrDevice s)
            -
            mlx::core::int32
            constexpr Dtype int32
            Definition dtype.h:69
            -
            mlx::core::float32
            constexpr Dtype float32
            Definition dtype.h:73
            +
            mlx::core::int32
            constexpr Dtype int32
            Definition dtype.h:67
            +
            mlx::core::float32
            constexpr Dtype float32
            Definition dtype.h:71
            mlx::core::StreamOrDevice
            std::variant< std::monostate, Stream, Device > StreamOrDevice
            Definition utils.h:14
            -
            mlx::core::Dtype
            Definition dtype.h:15
            +
            mlx::core::Dtype
            Definition dtype.h:13
            diff --git a/docs/build/html/search.html b/docs/build/html/search.html index cf517280b6..0aafbbf979 100644 --- a/docs/build/html/search.html +++ b/docs/build/html/search.html @@ -6,7 +6,7 @@ - Search - MLX 0.16.1 documentation + Search - MLX 0.16.2 documentation @@ -34,7 +34,7 @@ - + @@ -130,8 +130,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -456,8 +456,10 @@
          • Linear Algebra diff --git a/docs/build/html/search/all_0.js b/docs/build/html/search/all_0.js index f8b83e4a35..b003eda175 100644 --- a/docs/build/html/search/all_0.js +++ b/docs/build/html/search/all_0.js @@ -5,10 +5,6 @@ var searchData= ['_5fcontiguous_5fstrided_5freduce_2',['_contiguous_strided_reduce',['../reduce__col_8h.html#a6d72f6a88a37d8e031d0ac33f26ecbb4',1,'reduce_col.h']]], ['_5fmlx_5fbfloat16_3',['_MLX_BFloat16',['../struct___m_l_x___b_float16.html',1,'_MLX_BFloat16'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html',1,'mlx::core::_MLX_BFloat16'],['../struct___m_l_x___b_float16.html#ab1af7700f5d1e4ab567da6a34fa84668',1,'_MLX_BFloat16::_MLX_BFloat16() thread=default'],['../struct___m_l_x___b_float16.html#ae5c51644c3bd7cda6b796cb63c60c0b4',1,'_MLX_BFloat16::_MLX_BFloat16() threadgroup=default'],['../struct___m_l_x___b_float16.html#a21998a3c852d0e0f52681f8b453172bf',1,'_MLX_BFloat16::_MLX_BFloat16() device=default'],['../struct___m_l_x___b_float16.html#a64d8fc2e2463d7fa19cd3d5dd1ffdae8',1,'_MLX_BFloat16::_MLX_BFloat16() const ant=default'],['../struct___m_l_x___b_float16.html#a50d825f05a162d0ac133ad8b6f3c3112',1,'_MLX_BFloat16::_MLX_BFloat16(uint16_t bits, bits_to_bfloat_struct)'],['../struct___m_l_x___b_float16.html#a24c5736f234e09a0c82b00c7e44cc547',1,'_MLX_BFloat16::_MLX_BFloat16(T x) thread'],['../struct___m_l_x___b_float16.html#adeb880f31121c6dc40ce47765c6c7455',1,'_MLX_BFloat16::_MLX_BFloat16(T x) threadgroup'],['../struct___m_l_x___b_float16.html#ad2701d003e8fad168c89abc3907c6e53',1,'_MLX_BFloat16::_MLX_BFloat16(T x) device'],['../struct___m_l_x___b_float16.html#aec7fa716fd621ce1843338027bcb0118',1,'_MLX_BFloat16::_MLX_BFloat16(T x) const ant'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#a2c81f14fea4c01255a191f2146515917',1,'mlx::core::_MLX_BFloat16::_MLX_BFloat16()=default'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#ad43561d38ca00f9c37e8b130220233c0',1,'mlx::core::_MLX_BFloat16::_MLX_BFloat16(_MLX_BFloat16 const &)=default'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aedbead2d935a12e8d5a4ff6269ba9ab2',1,'mlx::core::_MLX_BFloat16::_MLX_BFloat16(const float &x)']]], ['_5fmlx_5ffloat16_4',['_MLX_Float16',['../structmlx_1_1core_1_1___m_l_x___float16.html',1,'mlx::core::_MLX_Float16'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a69a4ab5b456c4f3b786f43632e9a4fbc',1,'mlx::core::_MLX_Float16::_MLX_Float16()=default'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a0d7ed78b78c6d446f220f83b0cdebb86',1,'mlx::core::_MLX_Float16::_MLX_Float16(_MLX_Float16 const &)=default'],['../structmlx_1_1core_1_1___m_l_x___float16.html#afde284cbe678e0333ae277ffc8b131c0',1,'mlx::core::_MLX_Float16::_MLX_Float16(const float &x)']]], - ['_5fmps_5fprivate_5fcls_5',['_MPS_PRIVATE_CLS',['../mps_2gemm_8h.html#a2986cabc5a9ec53428151949c0369602',1,'gemm.h']]], - ['_5fmps_5fprivate_5fsel_6',['_MPS_PRIVATE_SEL',['../mps_2gemm_8h.html#a10e733259894b7fddd11f46b9eabfe5f',1,'gemm.h']]], - ['_5fmtl_5fprivate_5fdef_5fcls_7',['_MTL_PRIVATE_DEF_CLS',['../namespace_m_t_l_1_1_private_1_1_class.html#af6fc5dd227064897589f56d8912a8fd6',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSMatrixDescriptor)'],['../namespace_m_t_l_1_1_private_1_1_class.html#a23ba07d1dd78ce20ecba16d88faba5e2',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSMatrix)'],['../namespace_m_t_l_1_1_private_1_1_class.html#aa94ef8e094d91ad36af140823c15b59d',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSVectorDescriptor)'],['../namespace_m_t_l_1_1_private_1_1_class.html#aeff20cff997534929deb18ce0b7e947f',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSVector)'],['../namespace_m_t_l_1_1_private_1_1_class.html#a48f22abaf491778683a68e7db28839c9',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSKernel)'],['../namespace_m_t_l_1_1_private_1_1_class.html#a24a734c27edbaf99bbb3c3741fe98688',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSMatrixMultiplication)'],['../namespace_m_t_l_1_1_private_1_1_class.html#ac162f79565a61ed1f2316ca1a38c774d',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSMatrixVectorMultiplication)']]], - ['_5fmtl_5fprivate_5fdef_5fsel_8',['_MTL_PRIVATE_DEF_SEL',['../namespace_m_t_l_1_1_private_1_1_selector.html#acc50646a91989ebaca3cf945f42f2b62',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(matrixDescriptorWithRows_columns_rowBytes_dataType, "matrixDescriptorWithRows:columns:rowBytes:dataType:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a11464ac9c57b12981683bf25afc6ae77',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(matrixDescriptorWithRows_columns_matrices_rowBytes_matrixBytes_dataType, "matrixDescriptorWithRows:columns:matrices:rowBytes:matrixBytes:dataType:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a30c8b5bf66838e00c44d4cef997bee78',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(rows, "rows")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a5628c8d22de5d0339cdfd37cb8e103b6',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(initWithBuffer_descriptor, "initWithBuffer:descriptor:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#ae842530803a52e75c546db1f9424f933',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(initWithDevice_, "initWithDevice:transposeLeft:transposeRight:" "resultRows:resultColumns:interiorColumns:alpha:beta:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a8631d028899e57b928d1240c4cdda929',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(encodeToCommandBuffer_leftMatrix_rightMatrix_resultMatrix, "encodeToCommandBuffer:leftMatrix:rightMatrix:resultMatrix:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a071584fae1458eba0ed41e02e640d34b',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setLeftMatrixOrigin_, "setLeftMatrixOrigin:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a8534190f99aac6ab3a01c0f6965e8882',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setRightMatrixOrigin_, "setRightMatrixOrigin:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a8b1a7f7c97c62753b3cb7447d5aa4798',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setResultMatrixOrigin_, "setResultMatrixOrigin:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#af84065144e91d5fb295e23b42f3be8cc',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setBatchStart_, "setBatchStart:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#ac13dc3039959f5f5d970f02cbc89c16d',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setBatchSize_, "setBatchSize:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a46fcb509b385834113fe0b0f27ec768b',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(vectorDescriptorWithLength_dataType, "vectorDescriptorWithLength:dataType:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a4c7a7cb16b14dd691e1211913572c6a8',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(vectorDescriptorWithLength_vectors_vectorBytes_dataType, "vectorDescriptorWithLength:vectors:vectorBytes:dataType:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#ab84cb9467edcc1a5ce32f001cb46c656',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(initWithDevice_transpose_rows_columns_alpha_beta, "initWithDevice:transpose:rows:columns:alpha:beta:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#aec59058ff2c151d67647ad2680074090',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(encodeToCommandBuffer_inputMatrix_inputVector_resultVector, "encodeToCommandBuffer:inputMatrix:inputVector:resultVector:")']]], - ['_5fnomask_9',['_NoMask',['../struct___no_mask.html',1,'']]], - ['_5fnumeric_5flimits_5fimpl_3c_20bfloat16_5ft_20_3e_10',['_numeric_limits_impl< bfloat16_t >',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html',1,'metal']]] + ['_5fnomask_5',['_NoMask',['../struct___no_mask.html',1,'']]], + ['_5fnumeric_5flimits_5fimpl_3c_20bfloat16_5ft_20_3e_6',['_numeric_limits_impl< bfloat16_t >',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html',1,'metal']]] ]; diff --git a/docs/build/html/search/all_1.js b/docs/build/html/search/all_1.js index ae26198f2f..85e273653a 100644 --- a/docs/build/html/search/all_1.js +++ b/docs/build/html/search/all_1.js @@ -20,89 +20,92 @@ var searchData= ['adj_5fout_5fw_17',['adj_out_w',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html#ab971bf879079895189331fbeaf33c211',1,'mlx::steel::Conv2DGeneralJumpParams']]], ['adjust_5fmatrix_5foffsets_18',['adjust_matrix_offsets',['../quantized_8h.html#a3ab400746ad77be89c30d25638e01698',1,'quantized.h']]], ['advance_19',['advance',['../classpocketfft_1_1detail_1_1multi__iter.html#a5ddcc0666125b3cb6c0d62b30befdd2c',1,'pocketfft::detail::multi_iter::advance()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a73a9ecd3008d2bd35aaa00bf9fac074f',1,'pocketfft::detail::simple_iter::advance()'],['../classpocketfft_1_1detail_1_1rev__iter.html#ad1918c84ae963188afc7599629b29686',1,'pocketfft::detail::rev_iter::advance()']]], - ['align_5fk_20',['align_K',['../steel__gemm__fused_8h.html#a8bdd2cecf97aa5b033152b1d0f0d2416',1,'steel_gemm_fused.h']]], - ['align_5fm_21',['align_M',['../steel__gemm__fused_8h.html#a55af226dc74b0026b7d4b865142a6d21',1,'steel_gemm_fused.h']]], - ['align_5fn_22',['align_N',['../steel__gemm__fused_8h.html#aa3b267252df2dcbfdde8c5f174d27036',1,'steel_gemm_fused.h']]], - ['aligned_5falloc_23',['aligned_alloc',['../namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76',1,'pocketfft::detail']]], - ['aligned_5fallocator_24',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html',1,'pocketfft::detail::threading::aligned_allocator< T >'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator(const aligned_allocator< U > &)'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator()=default']]], - ['aligned_5fdealloc_25',['aligned_dealloc',['../namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8',1,'pocketfft::detail']]], - ['all_26',['all',['../group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68',1,'mlx::core::all(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4',1,'mlx::core::all(const array &a, StreamOrDevice s={})'],['../group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59',1,'mlx::core::all(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc',1,'mlx::core::all(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['all_5fgather_27',['all_gather',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04',1,'mlx::core::distributed::detail::all_gather()'],['../namespacemlx_1_1core_1_1distributed.html#abf45499dd1209ba7162908f14634d2ba',1,'mlx::core::distributed::all_gather()']]], - ['all_5freduce_28',['all_reduce',['../reduce__all_8h.html#a8d143d2fea044e4808b69cc3723e33c7',1,'reduce_all.h']]], - ['all_5freduce_5fdispatch_29',['all_reduce_dispatch',['../namespacemlx_1_1core.html#a3ab0fd997d9a35782106ff083a72e098',1,'mlx::core']]], - ['all_5freduce_5fno_5fatomics_30',['all_reduce_no_atomics',['../reduce__all_8h.html#a09d140566cb3e6418a821be1e42db282',1,'reduce_all.h']]], - ['all_5fsum_31',['all_sum',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960',1,'mlx::core::distributed::detail::all_sum()'],['../namespacemlx_1_1core_1_1distributed.html#ad1c6aa544781765060552cacfa845001',1,'mlx::core::distributed::all_sum()']]], - ['allclose_32',['allclose',['../group__ops.html#gaf0cd4257de7542daf9faf5e605e31020',1,'mlx::core']]], - ['allgather_33',['AllGather',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html',1,'mlx::core::distributed::AllGather'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ac9d0f1ba0d9602fa22e8f2b60f1231e0',1,'mlx::core::distributed::AllGather::AllGather()']]], - ['alloc_34',['alloc',['../class_m_p_s_1_1_matrix.html#a906d97b4b2365bec4acf278ef22205b4',1,'MPS::Matrix::alloc()'],['../class_m_p_s_1_1_matrix_multiplication.html#a9f1400b36672bd6f228a80982e5b2717',1,'MPS::MatrixMultiplication::alloc()'],['../class_m_p_s_1_1_vector.html#ac08919c16342247963b14766e65332a3',1,'MPS::Vector::alloc()'],['../class_m_p_s_1_1_matrix_vector_multiplication.html#aeca798759333aae7fdd594835e93d16a',1,'MPS::MatrixVectorMultiplication::alloc()']]], - ['alloc_5ftmp_35',['alloc_tmp',['../namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, size_t axsize, size_t elemsize)'],['../namespacepocketfft_1_1detail.html#a13832735696303b9559c4663631d5475',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, const shape_t &axes, size_t elemsize)']]], - ['allocate_36',['allocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826',1,'pocketfft::detail::threading::aligned_allocator']]], - ['allocator_37',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html',1,'mlx::core::allocator::Allocator'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a5803678a418fef687fc65fa9d5c37b65',1,'mlx::core::allocator::Allocator::Allocator()=default'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#aa05c081ce80dc036f9d3dd8c195259d2',1,'mlx::core::allocator::Allocator::Allocator(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a8e8ce346a16cf0c62847bed9289f9959',1,'mlx::core::allocator::Allocator::Allocator(Allocator &&other)=delete']]], - ['allocator_38',['allocator',['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#abf84c726a37df68345589b897b2e35f0',1,'mlx::core::allocator::CommonAllocator::allocator'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#afa1c5a725309caff163c492b5b84491e',1,'mlx::core::metal::MetalAllocator::allocator'],['../namespacemlx_1_1core_1_1allocator.html#aa23e2f20a336d0b159c097087194634e',1,'mlx::core::allocator::allocator()'],['../namespacemlx_1_1core_1_1metal.html#a74b3558bd518aecde6b14b0ba5e1a0d5',1,'mlx::core::metal::allocator()']]], - ['allocator_2eh_39',['allocator.h',['../allocator_8h.html',1,'(Global Namespace)'],['../backend_2metal_2allocator_8h.html',1,'(Global Namespace)']]], - ['allreduce_40',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html',1,'mlx::core::distributed::AllReduce'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a936b320f7b0a435f96e018c25f35e7ed',1,'mlx::core::distributed::AllReduce::AllReduce()']]], - ['alpha_41',['alpha',['../struct_m_l_x_fast_attention_params.html#a932266d04fa7d6e27d4a4a2c175f1477',1,'MLXFastAttentionParams::alpha'],['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#af8693d96512eff3e125d33d203920710',1,'mlx::steel::GEMMAddMMParams::alpha'],['../structmlx_1_1steel_1_1_transform_axpby.html#ab3223b49c6b3b7f89eba91aeaff9dcff',1,'mlx::steel::TransformAxpby::alpha']]], - ['and_42',['And',['../struct_and.html',1,'And< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924aba3b7fb927f6b6c8b198a9cdc3dd9e02',1,'mlx::core::distributed::AllReduce::And'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23dab14e7d426f45ae7f029f4e00210fbae4',1,'mlx::core::BitwiseBinary::And'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a5cc3412a1f243dcb11661bca42daea93',1,'mlx::core::Reduce::And']]], - ['any_43',['any',['../group__ops.html#ga8598dd718fb05cb28535e250372d4e6f',1,'mlx::core::any(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gad37df97f253a963bece124198dbaf9ba',1,'mlx::core::any(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef',1,'mlx::core::any(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce',1,'mlx::core::any(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['apply_44',['apply',['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply()'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const']]], - ['apply_5fepilogue_45',['apply_epilogue',['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)']]], - ['apply_5fepilogue_5fsafe_46',['apply_epilogue_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA']]], - ['apply_5finplace_5fop_47',['apply_inplace_op',['../structmlx_1_1steel_1_1_block_loader.html#adb4ca2cc193630a779de552fa8847ddf',1,'mlx::steel::BlockLoader']]], - ['arange_48',['Arange',['../classmlx_1_1core_1_1_arange.html',1,'mlx::core::Arange'],['../classmlx_1_1core_1_1_arange.html#a1a70c3b0b9c67d5a9446c141c5b7c574',1,'mlx::core::Arange::Arange()']]], - ['arange_49',['arange',['../namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b',1,'mlx::core::arange()'],['../namespacemlx_1_1core_1_1metal.html#a272c36f0faf2570cbb2f36030e9a3f26',1,'mlx::core::metal::arange()'],['../metal_2kernels_2arange_8h.html#a1e5126ee6ae0164c2343230c4d87c03e',1,'arange(): arange.h'],['../group__ops.html#ga7ca088b8090b9f84f2e08345cf3f835a',1,'mlx::core::arange(double start, double stop, double step, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga4c36b841dc5cba391dad029be5a0ad98',1,'mlx::core::arange(double start, double stop, double step, StreamOrDevice s={})'],['../group__ops.html#ga8d7cf9eb15e2daf1469058907e8abc85',1,'mlx::core::arange(double start, double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga74566a14e69ba6a25f5a35e7ade5c282',1,'mlx::core::arange(double start, double stop, StreamOrDevice s={})'],['../group__ops.html#ga345aa27af3dae3646b8b4b1068e89a3e',1,'mlx::core::arange(double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaae179075d0fe23f4bd53fdf8c41f4c70',1,'mlx::core::arange(double stop, StreamOrDevice s={})'],['../group__ops.html#ga6b945f513077c2978afc1a952c884860',1,'mlx::core::arange(int start, int stop, int step, StreamOrDevice s={})'],['../group__ops.html#ga1c39fcc6eaa1c1867735c7f849d708d6',1,'mlx::core::arange(int start, int stop, StreamOrDevice s={})'],['../group__ops.html#gafe6e4580452c873cac294f16129e633f',1,'mlx::core::arange(int stop, StreamOrDevice s={})']]], - ['arange_2eh_50',['arange.h',['../common_2arange_8h.html',1,'(Global Namespace)'],['../metal_2jit_2arange_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2arange_8h.html',1,'(Global Namespace)']]], - ['arange_5fkernels_51',['arange_kernels',['../metal_2jit_2arange_8h.html#a2f49fb7bdc0a90230077fe2023e6e5c0',1,'arange.h']]], - ['arccos_52',['ArcCos',['../struct_arc_cos.html',1,'ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html',1,'mlx::core::ArcCos'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html',1,'mlx::core::detail::ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html#a66f4ee841d17923d93241b71ea5103e9',1,'mlx::core::ArcCos::ArcCos()']]], - ['arccos_53',['arccos',['../group__ops.html#ga08bec7cb10c84466487b507fc5bf9776',1,'mlx::core']]], - ['arccosh_54',['ArcCosh',['../struct_arc_cosh.html',1,'ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html',1,'mlx::core::ArcCosh'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html',1,'mlx::core::detail::ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html#a34597054db467941a2a883c653ba4d71',1,'mlx::core::ArcCosh::ArcCosh()']]], - ['arccosh_55',['arccosh',['../group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5',1,'mlx::core']]], - ['arcsin_56',['ArcSin',['../struct_arc_sin.html',1,'ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html',1,'mlx::core::ArcSin'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html',1,'mlx::core::detail::ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html#a97cb8c3d4d9d6abc627dec49a404f013',1,'mlx::core::ArcSin::ArcSin()']]], - ['arcsin_57',['arcsin',['../group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619',1,'mlx::core']]], - ['arcsinh_58',['ArcSinh',['../struct_arc_sinh.html',1,'ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html',1,'mlx::core::ArcSinh'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html',1,'mlx::core::detail::ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html#a30076b222788deeaaf9ad92d3c535f20',1,'mlx::core::ArcSinh::ArcSinh()']]], - ['arcsinh_59',['arcsinh',['../group__ops.html#gac62e2cedc49ef2c90dd8584000317450',1,'mlx::core']]], - ['arctan_60',['ArcTan',['../struct_arc_tan.html',1,'ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html',1,'mlx::core::ArcTan'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html',1,'mlx::core::detail::ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html#a3511153bbd421e89fd9294cdb3f79b44',1,'mlx::core::ArcTan::ArcTan()']]], - ['arctan_61',['arctan',['../group__ops.html#gaa041f3f070e68f4946db07516b7d092e',1,'mlx::core']]], - ['arctan2_62',['ArcTan2',['../struct_arc_tan2.html',1,'ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html',1,'mlx::core::ArcTan2'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html',1,'mlx::core::detail::ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html#aa1a4ebab9924b6bcc80df5b52ed0121a',1,'mlx::core::ArcTan2::ArcTan2()']]], - ['arctan2_63',['arctan2',['../group__ops.html#ga6caba9c92b5989123501f909cc7da354',1,'mlx::core']]], - ['arctanh_64',['ArcTanh',['../struct_arc_tanh.html',1,'ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html',1,'mlx::core::ArcTanh'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html',1,'mlx::core::detail::ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html#a17857bd0e2a3ecf1f7bf8e1a3d354358',1,'mlx::core::ArcTanh::ArcTanh()']]], - ['arctanh_65',['arctanh',['../group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e',1,'mlx::core']]], - ['argmax_66',['ArgMax',['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3acc6659315ab0001abd37cbfcbe837e7e',1,'mlx::core::ArgReduce']]], - ['argmax_67',['argmax',['../group__ops.html#gae60b0b5339b9c50b9970260faf613e83',1,'mlx::core::argmax(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8',1,'mlx::core::argmax(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5',1,'mlx::core::argmax(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['argmin_68',['ArgMin',['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3a93a8a9221545ae9518d289d9ac4d09e9',1,'mlx::core::ArgReduce']]], - ['argmin_69',['argmin',['../group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71',1,'mlx::core::argmin(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga6bc577c5ab10cd9c848ba81321595070',1,'mlx::core::argmin(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81',1,'mlx::core::argmin(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['argpartition_70',['ArgPartition',['../classmlx_1_1core_1_1_arg_partition.html',1,'mlx::core::ArgPartition'],['../classmlx_1_1core_1_1_arg_partition.html#ab54b13dbf92351ba1ac06fd3e5a802df',1,'mlx::core::ArgPartition::ArgPartition()']]], - ['argpartition_71',['argpartition',['../group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd',1,'mlx::core::argpartition(const array &a, int kth, StreamOrDevice s={})'],['../group__ops.html#ga7b15c654c7463def57857a0e239989a3',1,'mlx::core::argpartition(const array &a, int kth, int axis, StreamOrDevice s={})']]], - ['argreduce_72',['ArgReduce',['../classmlx_1_1core_1_1_arg_reduce.html',1,'mlx::core::ArgReduce'],['../classmlx_1_1core_1_1_arg_reduce.html#aaccf8021dc24895656e25142eb65aa03',1,'mlx::core::ArgReduce::ArgReduce()']]], - ['argsort_73',['ArgSort',['../classmlx_1_1core_1_1_arg_sort.html',1,'mlx::core::ArgSort'],['../classmlx_1_1core_1_1_arg_sort.html#a38507a8445302a81cb44674c4a5fc0b0',1,'mlx::core::ArgSort::ArgSort()']]], - ['argsort_74',['argsort',['../group__ops.html#ga8df3b2703bf671457422894dd870cdc5',1,'mlx::core::argsort(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6',1,'mlx::core::argsort(const array &a, int axis, StreamOrDevice s={})']]], - ['argument_5fencoder_75',['argument_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#a6e33e2b1287324fb4a6575e0da5e5881',1,'mlx::core::metal::Device']]], - ['arr_76',['arr',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail::arr< T >'],['../classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42',1,'pocketfft::detail::arr::arr()'],['../classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd',1,'pocketfft::detail::arr::arr(size_t n)'],['../classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0',1,'pocketfft::detail::arr::arr(arr &&other)']]], - ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20t0_20_3e_20_3e_77',['arr< pocketfft::detail::cmplx< T0 > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20thigh_20_3e_20_3e_78',['arr< pocketfft::detail::cmplx< Thigh > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_3c_20t0_20_3e_79',['arr< T0 >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_5finfo_80',['arr_info',['../classpocketfft_1_1detail_1_1arr__info.html',1,'pocketfft::detail::arr_info'],['../classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e',1,'pocketfft::detail::arr_info::arr_info()']]], - ['array_81',['array',['../classmlx_1_1core_1_1array.html',1,'mlx::core::array'],['../classmlx_1_1core_1_1array.html#a75fac72da3ce214fa3737df92a64b232',1,'mlx::core::array::array(T val, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a6db4b8c28c767cc16ad2785ece496dca',1,'mlx::core::array::array(const std::complex< float > &val, Dtype dtype=complex64)'],['../classmlx_1_1core_1_1array.html#a3e506a53b9c7567448f7809dda680210',1,'mlx::core::array::array(It data, std::vector< int > shape, Dtype dtype=TypeToDtype< typename std::iterator_traits< It >::value_type >())'],['../classmlx_1_1core_1_1array.html#a87f170384f4fb93decf2b80ae7280f00',1,'mlx::core::array::array(std::initializer_list< T > data, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a46642301da11e3eb4312c37349fbc9d7',1,'mlx::core::array::array(std::initializer_list< float > data)'],['../classmlx_1_1core_1_1array.html#a5e1812029394bfb1a706c83611286f49',1,'mlx::core::array::array(std::initializer_list< int > data, Dtype dtype)'],['../classmlx_1_1core_1_1array.html#a44e57a41819321e0d796e08cb9a06e4b',1,'mlx::core::array::array(std::initializer_list< T > data, std::vector< int > shape, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a5b5f562ff14c150842cb61628e531663',1,'mlx::core::array::array(allocator::Buffer data, std::vector< int > shape, Dtype dtype, deleter_t deleter=allocator::free)'],['../classmlx_1_1core_1_1array.html#a297df274e2da5cb884257bbeffd6b187',1,'mlx::core::array::array(const array &other)=default'],['../classmlx_1_1core_1_1array.html#ab6cbccbba66cc54acda4390b19f0397c',1,'mlx::core::array::array(array &&other)=default'],['../classmlx_1_1core_1_1array.html#adaade8f4bb7f8ecc0ba07efb17cd2620',1,'mlx::core::array::array(std::vector< int > shape, Dtype dtype, std::shared_ptr< Primitive > primitive, std::vector< array > inputs)']]], - ['array_20operations_82',['Core array operations',['../group__ops.html',1,'']]], - ['array_2eh_83',['array.h',['../array_8h.html',1,'']]], - ['array_5fequal_84',['array_equal',['../group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645',1,'mlx::core::array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})'],['../group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14',1,'mlx::core::array_equal(const array &a, const array &b, StreamOrDevice s={})']]], - ['arrayiterator_85',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html',1,'mlx::core::array::ArrayIterator'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ad3afcb24c6db7642bbc06835f7f3e27a',1,'mlx::core::array::ArrayIterator::ArrayIterator()']]], - ['as_5foffset_86',['As_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a138ed1bbad2ca88d3a3c7d162cd36562',1,'mlx::steel::BlockMMA']]], - ['as_5fstrided_87',['as_strided',['../group__ops.html#ga8de80ecef30fc560003d40f61a38b99d',1,'mlx::core']]], - ['asimd_88',['Asimd',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92ac4b483cb58c361d7a657e0738c070',1,'mlx::steel::BlockMMA']]], - ['asin_89',['asin',['../namespacemetal.html#a16e843194df3fd136404bf80ba5ac95c',1,'metal::asin()'],['../namespacemetal_1_1fast.html#a769455a283da99654b6e42c3acf13eb1',1,'metal::fast::asin()'],['../namespacemetal_1_1precise.html#adc7b8b6e12e320cb32030f728dcbf438',1,'metal::precise::asin()']]], - ['asinh_90',['asinh',['../namespacemetal.html#abcc3251866930cfe880f89e7473d0e63',1,'metal::asinh()'],['../namespacemetal_1_1fast.html#a4367034b7b3e14310803bb2be975a556',1,'metal::fast::asinh()'],['../namespacemetal_1_1precise.html#aaad1cdde6687c8011fbc5fda1bb13424',1,'metal::precise::asinh()']]], - ['asstrided_91',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html',1,'mlx::core::AsStrided'],['../classmlx_1_1core_1_1_as_strided.html#a80c0547f72ed53374eafc57d57b5d4af',1,'mlx::core::AsStrided::AsStrided()']]], - ['astype_92',['AsType',['../classmlx_1_1core_1_1_as_type.html',1,'mlx::core::AsType'],['../classmlx_1_1core_1_1_as_type.html#a8c3241d402a8977bb4db037e225f5b47',1,'mlx::core::AsType::AsType()']]], - ['astype_93',['astype',['../group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62',1,'mlx::core']]], - ['async_5feval_94',['async_eval',['../namespacemlx_1_1core.html#a15dda19aa7fa1fc5fca35df5cf963297',1,'mlx::core']]], - ['atan_95',['atan',['../namespacemetal.html#a80a771553d9a0012b93620d19c48b00f',1,'metal::atan()'],['../namespacemetal_1_1fast.html#a769503b4b7f89071d0983258c5a3ac5a',1,'metal::fast::atan()'],['../namespacemetal_1_1precise.html#aaaf4b5f4786a912089bbf0ae7619a6be',1,'metal::precise::atan()']]], - ['atan2_96',['atan2',['../namespacemetal.html#a1d430793eaa38ccf0d07145e3fcd1e61',1,'metal::atan2()'],['../namespacemetal_1_1fast.html#a00e687ea46f5affe26e6aef8fd62b89a',1,'metal::fast::atan2()'],['../namespacemetal_1_1precise.html#a6f161b049cc6884f87b09b33c2d1cd7f',1,'metal::precise::atan2()']]], - ['atanh_97',['atanh',['../namespacemetal.html#a57116427997ba71dd3863bfb15de33bf',1,'metal::atanh()'],['../namespacemetal_1_1fast.html#af24608fc605db9a14427d37c36dc1c53',1,'metal::fast::atanh()'],['../namespacemetal_1_1precise.html#a902994837653b90c47f4285673e712c4',1,'metal::precise::atanh()']]], - ['atleast_5f1d_98',['atleast_1d',['../group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396',1,'mlx::core::atleast_1d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5',1,'mlx::core::atleast_1d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atleast_5f2d_99',['atleast_2d',['../group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087',1,'mlx::core::atleast_2d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga9950299a80c2562f13448758f856d1f5',1,'mlx::core::atleast_2d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atleast_5f3d_100',['atleast_3d',['../group__ops.html#ga4afd919601e67782ff964465919956a0',1,'mlx::core::atleast_3d(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaffdf742ad79440a60dda40062a8074fe',1,'mlx::core::atleast_3d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atomic_2eh_101',['atomic.h',['../atomic_8h.html',1,'']]], - ['atomic_5fupdate_102',['atomic_update',['../struct_none.html#a9d7414f58b3a526758007a4a1da72c74',1,'None::atomic_update()'],['../struct_and.html#a9ddf6d3f16dee000a5f0a30af4d43a7f',1,'And::atomic_update(device mlx_atomic< unsigned int > *out, bool val, int elem_idx, int offset=0)'],['../struct_and.html#af4a5c9f276d21580c0acaf1ff7064c57',1,'And::atomic_update(device mlx_atomic< bool > *out, bool val, uint offset=0)'],['../struct_or.html#a40840c0506bf4ed9a0771ba42c9ebb0f',1,'Or::atomic_update(device mlx_atomic< unsigned int > *out, bool val, uint elem_idx, uint offset=0)'],['../struct_or.html#aad1e6b5b015cd0910625da725afabdf3',1,'Or::atomic_update(device mlx_atomic< bool > *out, bool val, uint offset=0)'],['../struct_sum.html#a3188318e74823f00b9367317ec22ad6b',1,'Sum::atomic_update()'],['../struct_prod.html#a7ad8c7c18b1c54ff1d158683744a324d',1,'Prod::atomic_update()'],['../struct_min.html#a2377172960d21faa67b13c6b1eec5c69',1,'Min::atomic_update()'],['../struct_max.html#ac478a5cd1f567164d786240589d9260c',1,'Max::atomic_update()']]], - ['attach_5fevent_103',['attach_event',['../classmlx_1_1core_1_1array.html#a000c3cfe13cb378bf0523b62816190da',1,'mlx::core::array']]], - ['available_104',['available',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078a308bd3e5bf976888b120dd36d0c2d2ae',1,'mlx::core::array']]] + ['affine_5fdequantize_20',['affine_dequantize',['../quantized_8h.html#aecc3db76dcae742776e71ec8fca470c5',1,'affine_dequantize(): quantized.h'],['../namespacemlx_1_1core_1_1fast.html#a12c7ef41409d6fb378008e67b6fab328',1,'mlx::core::fast::affine_dequantize()']]], + ['affine_5fquantize_21',['affine_quantize',['../quantized_8h.html#a47610f886f988d84c3e789eb564a6c44',1,'affine_quantize(): quantized.h'],['../namespacemlx_1_1core_1_1fast.html#aa4b5f6886b2288cb6dfdd8598579f080',1,'mlx::core::fast::affine_quantize(const array &w, int group_size=64, int bits=4, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fast.html#a638c7e9b9ea8677f01786d8f9738baf8',1,'mlx::core::fast::affine_quantize(const array &w, const array &scales, const array &biases, int group_size=64, int bits=4, StreamOrDevice s={})']]], + ['affine_5fquantize_5fscales_5fbiases_22',['affine_quantize_scales_biases',['../quantized_8h.html#a57cd320aab5fa1abb65f79b0b781f575',1,'quantized.h']]], + ['affinequantize_23',['AffineQuantize',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html',1,'mlx::core::fast::AffineQuantize'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a84d5fa9e8c3de407fbcc5f38d2ed1473',1,'mlx::core::fast::AffineQuantize::AffineQuantize()']]], + ['align_5fk_24',['align_K',['../steel__gemm__fused_8h.html#a8bdd2cecf97aa5b033152b1d0f0d2416',1,'steel_gemm_fused.h']]], + ['align_5fm_25',['align_M',['../steel__gemm__fused_8h.html#a55af226dc74b0026b7d4b865142a6d21',1,'steel_gemm_fused.h']]], + ['align_5fn_26',['align_N',['../steel__gemm__fused_8h.html#aa3b267252df2dcbfdde8c5f174d27036',1,'steel_gemm_fused.h']]], + ['aligned_5falloc_27',['aligned_alloc',['../namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76',1,'pocketfft::detail']]], + ['aligned_5fallocator_28',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html',1,'pocketfft::detail::threading::aligned_allocator< T >'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator(const aligned_allocator< U > &)'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator()=default']]], + ['aligned_5fdealloc_29',['aligned_dealloc',['../namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8',1,'pocketfft::detail']]], + ['all_30',['all',['../group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68',1,'mlx::core::all(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4',1,'mlx::core::all(const array &a, StreamOrDevice s={})'],['../group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59',1,'mlx::core::all(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc',1,'mlx::core::all(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['all_5fgather_31',['all_gather',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04',1,'mlx::core::distributed::detail::all_gather()'],['../namespacemlx_1_1core_1_1distributed.html#abf45499dd1209ba7162908f14634d2ba',1,'mlx::core::distributed::all_gather()']]], + ['all_5freduce_32',['all_reduce',['../reduce__all_8h.html#a8d143d2fea044e4808b69cc3723e33c7',1,'reduce_all.h']]], + ['all_5freduce_5fdispatch_33',['all_reduce_dispatch',['../namespacemlx_1_1core.html#a3ab0fd997d9a35782106ff083a72e098',1,'mlx::core']]], + ['all_5freduce_5fno_5fatomics_34',['all_reduce_no_atomics',['../reduce__all_8h.html#a09d140566cb3e6418a821be1e42db282',1,'reduce_all.h']]], + ['all_5fsum_35',['all_sum',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960',1,'mlx::core::distributed::detail::all_sum()'],['../namespacemlx_1_1core_1_1distributed.html#ad1c6aa544781765060552cacfa845001',1,'mlx::core::distributed::all_sum()']]], + ['allclose_36',['allclose',['../group__ops.html#gaf0cd4257de7542daf9faf5e605e31020',1,'mlx::core']]], + ['allgather_37',['AllGather',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html',1,'mlx::core::distributed::AllGather'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ac9d0f1ba0d9602fa22e8f2b60f1231e0',1,'mlx::core::distributed::AllGather::AllGather()']]], + ['alloc_5ftmp_38',['alloc_tmp',['../namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, size_t axsize, size_t elemsize)'],['../namespacepocketfft_1_1detail.html#a13832735696303b9559c4663631d5475',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, const shape_t &axes, size_t elemsize)']]], + ['allocate_39',['allocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826',1,'pocketfft::detail::threading::aligned_allocator']]], + ['allocator_40',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html',1,'mlx::core::allocator::Allocator'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a5803678a418fef687fc65fa9d5c37b65',1,'mlx::core::allocator::Allocator::Allocator()=default'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#aa05c081ce80dc036f9d3dd8c195259d2',1,'mlx::core::allocator::Allocator::Allocator(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a8e8ce346a16cf0c62847bed9289f9959',1,'mlx::core::allocator::Allocator::Allocator(Allocator &&other)=delete']]], + ['allocator_41',['allocator',['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#abf84c726a37df68345589b897b2e35f0',1,'mlx::core::allocator::CommonAllocator::allocator'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#afa1c5a725309caff163c492b5b84491e',1,'mlx::core::metal::MetalAllocator::allocator'],['../namespacemlx_1_1core_1_1allocator.html#aa23e2f20a336d0b159c097087194634e',1,'mlx::core::allocator::allocator()'],['../namespacemlx_1_1core_1_1metal.html#a74b3558bd518aecde6b14b0ba5e1a0d5',1,'mlx::core::metal::allocator()']]], + ['allocator_2eh_42',['allocator.h',['../allocator_8h.html',1,'(Global Namespace)'],['../backend_2metal_2allocator_8h.html',1,'(Global Namespace)']]], + ['allreduce_43',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html',1,'mlx::core::distributed::AllReduce'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a936b320f7b0a435f96e018c25f35e7ed',1,'mlx::core::distributed::AllReduce::AllReduce()']]], + ['alpha_44',['alpha',['../struct_m_l_x_fast_attention_params.html#a932266d04fa7d6e27d4a4a2c175f1477',1,'MLXFastAttentionParams::alpha'],['../structmlx_1_1steel_1_1_g_e_m_m_add_m_m_params.html#af8693d96512eff3e125d33d203920710',1,'mlx::steel::GEMMAddMMParams::alpha'],['../structmlx_1_1steel_1_1_transform_axpby.html#ab3223b49c6b3b7f89eba91aeaff9dcff',1,'mlx::steel::TransformAxpby::alpha']]], + ['and_45',['And',['../struct_and.html',1,'And< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924aba3b7fb927f6b6c8b198a9cdc3dd9e02',1,'mlx::core::distributed::AllReduce::And'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23dab14e7d426f45ae7f029f4e00210fbae4',1,'mlx::core::BitwiseBinary::And'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a5cc3412a1f243dcb11661bca42daea93',1,'mlx::core::Reduce::And']]], + ['any_46',['any',['../group__ops.html#ga8598dd718fb05cb28535e250372d4e6f',1,'mlx::core::any(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gad37df97f253a963bece124198dbaf9ba',1,'mlx::core::any(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef',1,'mlx::core::any(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce',1,'mlx::core::any(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['apply_47',['apply',['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply(InT x) const'],['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply(InT x) const'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const']]], + ['apply_5fepilogue_48',['apply_epilogue',['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)']]], + ['apply_5fepilogue_5fsafe_49',['apply_epilogue_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA']]], + ['apply_5finplace_5fop_50',['apply_inplace_op',['../structmlx_1_1steel_1_1_block_loader.html#adb4ca2cc193630a779de552fa8847ddf',1,'mlx::steel::BlockLoader']]], + ['arange_51',['Arange',['../classmlx_1_1core_1_1_arange.html',1,'mlx::core::Arange'],['../classmlx_1_1core_1_1_arange.html#a1a70c3b0b9c67d5a9446c141c5b7c574',1,'mlx::core::Arange::Arange()']]], + ['arange_52',['arange',['../namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b',1,'mlx::core::arange()'],['../namespacemlx_1_1core_1_1metal.html#a272c36f0faf2570cbb2f36030e9a3f26',1,'mlx::core::metal::arange()'],['../metal_2kernels_2arange_8h.html#a1e5126ee6ae0164c2343230c4d87c03e',1,'arange(): arange.h'],['../group__ops.html#ga7ca088b8090b9f84f2e08345cf3f835a',1,'mlx::core::arange(double start, double stop, double step, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga4c36b841dc5cba391dad029be5a0ad98',1,'mlx::core::arange(double start, double stop, double step, StreamOrDevice s={})'],['../group__ops.html#ga8d7cf9eb15e2daf1469058907e8abc85',1,'mlx::core::arange(double start, double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga74566a14e69ba6a25f5a35e7ade5c282',1,'mlx::core::arange(double start, double stop, StreamOrDevice s={})'],['../group__ops.html#ga345aa27af3dae3646b8b4b1068e89a3e',1,'mlx::core::arange(double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaae179075d0fe23f4bd53fdf8c41f4c70',1,'mlx::core::arange(double stop, StreamOrDevice s={})'],['../group__ops.html#ga6b945f513077c2978afc1a952c884860',1,'mlx::core::arange(int start, int stop, int step, StreamOrDevice s={})'],['../group__ops.html#ga1c39fcc6eaa1c1867735c7f849d708d6',1,'mlx::core::arange(int start, int stop, StreamOrDevice s={})'],['../group__ops.html#gafe6e4580452c873cac294f16129e633f',1,'mlx::core::arange(int stop, StreamOrDevice s={})']]], + ['arange_2eh_53',['arange.h',['../common_2arange_8h.html',1,'(Global Namespace)'],['../metal_2jit_2arange_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2arange_8h.html',1,'(Global Namespace)']]], + ['arange_5fkernels_54',['arange_kernels',['../metal_2jit_2arange_8h.html#a2f49fb7bdc0a90230077fe2023e6e5c0',1,'arange.h']]], + ['arccos_55',['ArcCos',['../struct_arc_cos.html',1,'ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html',1,'mlx::core::ArcCos'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html',1,'mlx::core::detail::ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html#a66f4ee841d17923d93241b71ea5103e9',1,'mlx::core::ArcCos::ArcCos()']]], + ['arccos_56',['arccos',['../group__ops.html#ga08bec7cb10c84466487b507fc5bf9776',1,'mlx::core']]], + ['arccosh_57',['ArcCosh',['../struct_arc_cosh.html',1,'ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html',1,'mlx::core::ArcCosh'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html',1,'mlx::core::detail::ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html#a34597054db467941a2a883c653ba4d71',1,'mlx::core::ArcCosh::ArcCosh()']]], + ['arccosh_58',['arccosh',['../group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5',1,'mlx::core']]], + ['arcsin_59',['ArcSin',['../struct_arc_sin.html',1,'ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html',1,'mlx::core::ArcSin'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html',1,'mlx::core::detail::ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html#a97cb8c3d4d9d6abc627dec49a404f013',1,'mlx::core::ArcSin::ArcSin()']]], + ['arcsin_60',['arcsin',['../group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619',1,'mlx::core']]], + ['arcsinh_61',['ArcSinh',['../struct_arc_sinh.html',1,'ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html',1,'mlx::core::ArcSinh'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html',1,'mlx::core::detail::ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html#a30076b222788deeaaf9ad92d3c535f20',1,'mlx::core::ArcSinh::ArcSinh()']]], + ['arcsinh_62',['arcsinh',['../group__ops.html#gac62e2cedc49ef2c90dd8584000317450',1,'mlx::core']]], + ['arctan_63',['ArcTan',['../struct_arc_tan.html',1,'ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html',1,'mlx::core::ArcTan'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html',1,'mlx::core::detail::ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html#a3511153bbd421e89fd9294cdb3f79b44',1,'mlx::core::ArcTan::ArcTan()']]], + ['arctan_64',['arctan',['../group__ops.html#gaa041f3f070e68f4946db07516b7d092e',1,'mlx::core']]], + ['arctan2_65',['ArcTan2',['../struct_arc_tan2.html',1,'ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html',1,'mlx::core::ArcTan2'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html',1,'mlx::core::detail::ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html#aa1a4ebab9924b6bcc80df5b52ed0121a',1,'mlx::core::ArcTan2::ArcTan2()']]], + ['arctan2_66',['arctan2',['../group__ops.html#ga6caba9c92b5989123501f909cc7da354',1,'mlx::core']]], + ['arctanh_67',['ArcTanh',['../struct_arc_tanh.html',1,'ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html',1,'mlx::core::ArcTanh'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html',1,'mlx::core::detail::ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html#a17857bd0e2a3ecf1f7bf8e1a3d354358',1,'mlx::core::ArcTanh::ArcTanh()']]], + ['arctanh_68',['arctanh',['../group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e',1,'mlx::core']]], + ['argmax_69',['ArgMax',['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3acc6659315ab0001abd37cbfcbe837e7e',1,'mlx::core::ArgReduce']]], + ['argmax_70',['argmax',['../group__ops.html#gae60b0b5339b9c50b9970260faf613e83',1,'mlx::core::argmax(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8',1,'mlx::core::argmax(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5',1,'mlx::core::argmax(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['argmin_71',['ArgMin',['../classmlx_1_1core_1_1_arg_reduce.html#a920ed48caaba76683be0d1f1ed4a8bd3a93a8a9221545ae9518d289d9ac4d09e9',1,'mlx::core::ArgReduce']]], + ['argmin_72',['argmin',['../group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71',1,'mlx::core::argmin(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga6bc577c5ab10cd9c848ba81321595070',1,'mlx::core::argmin(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81',1,'mlx::core::argmin(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['argpartition_73',['ArgPartition',['../classmlx_1_1core_1_1_arg_partition.html',1,'mlx::core::ArgPartition'],['../classmlx_1_1core_1_1_arg_partition.html#ab54b13dbf92351ba1ac06fd3e5a802df',1,'mlx::core::ArgPartition::ArgPartition()']]], + ['argpartition_74',['argpartition',['../group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd',1,'mlx::core::argpartition(const array &a, int kth, StreamOrDevice s={})'],['../group__ops.html#ga7b15c654c7463def57857a0e239989a3',1,'mlx::core::argpartition(const array &a, int kth, int axis, StreamOrDevice s={})']]], + ['argreduce_75',['ArgReduce',['../classmlx_1_1core_1_1_arg_reduce.html',1,'mlx::core::ArgReduce'],['../classmlx_1_1core_1_1_arg_reduce.html#aaccf8021dc24895656e25142eb65aa03',1,'mlx::core::ArgReduce::ArgReduce()']]], + ['argsort_76',['ArgSort',['../classmlx_1_1core_1_1_arg_sort.html',1,'mlx::core::ArgSort'],['../classmlx_1_1core_1_1_arg_sort.html#a38507a8445302a81cb44674c4a5fc0b0',1,'mlx::core::ArgSort::ArgSort()']]], + ['argsort_77',['argsort',['../group__ops.html#ga8df3b2703bf671457422894dd870cdc5',1,'mlx::core::argsort(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6',1,'mlx::core::argsort(const array &a, int axis, StreamOrDevice s={})']]], + ['argument_5fencoder_78',['argument_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#a6e33e2b1287324fb4a6575e0da5e5881',1,'mlx::core::metal::Device']]], + ['arr_79',['arr',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail::arr< T >'],['../classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42',1,'pocketfft::detail::arr::arr()'],['../classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd',1,'pocketfft::detail::arr::arr(size_t n)'],['../classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0',1,'pocketfft::detail::arr::arr(arr &&other)']]], + ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20t0_20_3e_20_3e_80',['arr< pocketfft::detail::cmplx< T0 > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20thigh_20_3e_20_3e_81',['arr< pocketfft::detail::cmplx< Thigh > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_3c_20t0_20_3e_82',['arr< T0 >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_5finfo_83',['arr_info',['../classpocketfft_1_1detail_1_1arr__info.html',1,'pocketfft::detail::arr_info'],['../classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e',1,'pocketfft::detail::arr_info::arr_info()']]], + ['array_84',['array',['../classmlx_1_1core_1_1array.html',1,'mlx::core::array'],['../classmlx_1_1core_1_1array.html#a75fac72da3ce214fa3737df92a64b232',1,'mlx::core::array::array(T val, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a6db4b8c28c767cc16ad2785ece496dca',1,'mlx::core::array::array(const std::complex< float > &val, Dtype dtype=complex64)'],['../classmlx_1_1core_1_1array.html#a3e506a53b9c7567448f7809dda680210',1,'mlx::core::array::array(It data, std::vector< int > shape, Dtype dtype=TypeToDtype< typename std::iterator_traits< It >::value_type >())'],['../classmlx_1_1core_1_1array.html#a87f170384f4fb93decf2b80ae7280f00',1,'mlx::core::array::array(std::initializer_list< T > data, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a46642301da11e3eb4312c37349fbc9d7',1,'mlx::core::array::array(std::initializer_list< float > data)'],['../classmlx_1_1core_1_1array.html#a5e1812029394bfb1a706c83611286f49',1,'mlx::core::array::array(std::initializer_list< int > data, Dtype dtype)'],['../classmlx_1_1core_1_1array.html#a44e57a41819321e0d796e08cb9a06e4b',1,'mlx::core::array::array(std::initializer_list< T > data, std::vector< int > shape, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a5b5f562ff14c150842cb61628e531663',1,'mlx::core::array::array(allocator::Buffer data, std::vector< int > shape, Dtype dtype, deleter_t deleter=allocator::free)'],['../classmlx_1_1core_1_1array.html#a297df274e2da5cb884257bbeffd6b187',1,'mlx::core::array::array(const array &other)=default'],['../classmlx_1_1core_1_1array.html#ab6cbccbba66cc54acda4390b19f0397c',1,'mlx::core::array::array(array &&other)=default'],['../classmlx_1_1core_1_1array.html#adaade8f4bb7f8ecc0ba07efb17cd2620',1,'mlx::core::array::array(std::vector< int > shape, Dtype dtype, std::shared_ptr< Primitive > primitive, std::vector< array > inputs)']]], + ['array_20operations_85',['Core array operations',['../group__ops.html',1,'']]], + ['array_2eh_86',['array.h',['../array_8h.html',1,'']]], + ['array_5fequal_87',['array_equal',['../group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645',1,'mlx::core::array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})'],['../group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14',1,'mlx::core::array_equal(const array &a, const array &b, StreamOrDevice s={})']]], + ['arrayiterator_88',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html',1,'mlx::core::array::ArrayIterator'],['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ad3afcb24c6db7642bbc06835f7f3e27a',1,'mlx::core::array::ArrayIterator::ArrayIterator()']]], + ['as_5foffset_89',['As_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a138ed1bbad2ca88d3a3c7d162cd36562',1,'mlx::steel::BlockMMA']]], + ['as_5fstrided_90',['as_strided',['../group__ops.html#ga8de80ecef30fc560003d40f61a38b99d',1,'mlx::core']]], + ['asimd_91',['Asimd',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92ac4b483cb58c361d7a657e0738c070',1,'mlx::steel::BlockMMA']]], + ['asin_92',['asin',['../namespacemetal.html#a16e843194df3fd136404bf80ba5ac95c',1,'metal::asin()'],['../namespacemetal_1_1fast.html#a769455a283da99654b6e42c3acf13eb1',1,'metal::fast::asin()'],['../namespacemetal_1_1precise.html#adc7b8b6e12e320cb32030f728dcbf438',1,'metal::precise::asin()']]], + ['asinh_93',['asinh',['../namespacemetal.html#abcc3251866930cfe880f89e7473d0e63',1,'metal::asinh()'],['../namespacemetal_1_1fast.html#a4367034b7b3e14310803bb2be975a556',1,'metal::fast::asinh()'],['../namespacemetal_1_1precise.html#aaad1cdde6687c8011fbc5fda1bb13424',1,'metal::precise::asinh()']]], + ['asstrided_94',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html',1,'mlx::core::AsStrided'],['../classmlx_1_1core_1_1_as_strided.html#a80c0547f72ed53374eafc57d57b5d4af',1,'mlx::core::AsStrided::AsStrided()']]], + ['astype_95',['AsType',['../classmlx_1_1core_1_1_as_type.html',1,'mlx::core::AsType'],['../classmlx_1_1core_1_1_as_type.html#a8c3241d402a8977bb4db037e225f5b47',1,'mlx::core::AsType::AsType()']]], + ['astype_96',['astype',['../group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62',1,'mlx::core']]], + ['async_5feval_97',['async_eval',['../namespacemlx_1_1core.html#a15dda19aa7fa1fc5fca35df5cf963297',1,'mlx::core']]], + ['atan_98',['atan',['../namespacemetal.html#a80a771553d9a0012b93620d19c48b00f',1,'metal::atan()'],['../namespacemetal_1_1fast.html#a769503b4b7f89071d0983258c5a3ac5a',1,'metal::fast::atan()'],['../namespacemetal_1_1precise.html#aaaf4b5f4786a912089bbf0ae7619a6be',1,'metal::precise::atan()']]], + ['atan2_99',['atan2',['../namespacemetal.html#a1d430793eaa38ccf0d07145e3fcd1e61',1,'metal::atan2()'],['../namespacemetal_1_1fast.html#a00e687ea46f5affe26e6aef8fd62b89a',1,'metal::fast::atan2()'],['../namespacemetal_1_1precise.html#a6f161b049cc6884f87b09b33c2d1cd7f',1,'metal::precise::atan2()']]], + ['atanh_100',['atanh',['../namespacemetal.html#a57116427997ba71dd3863bfb15de33bf',1,'metal::atanh()'],['../namespacemetal_1_1fast.html#af24608fc605db9a14427d37c36dc1c53',1,'metal::fast::atanh()'],['../namespacemetal_1_1precise.html#a902994837653b90c47f4285673e712c4',1,'metal::precise::atanh()']]], + ['atleast_5f1d_101',['atleast_1d',['../group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396',1,'mlx::core::atleast_1d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5',1,'mlx::core::atleast_1d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atleast_5f2d_102',['atleast_2d',['../group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087',1,'mlx::core::atleast_2d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga9950299a80c2562f13448758f856d1f5',1,'mlx::core::atleast_2d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atleast_5f3d_103',['atleast_3d',['../group__ops.html#ga4afd919601e67782ff964465919956a0',1,'mlx::core::atleast_3d(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaffdf742ad79440a60dda40062a8074fe',1,'mlx::core::atleast_3d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atomic_2eh_104',['atomic.h',['../atomic_8h.html',1,'']]], + ['atomic_5fupdate_105',['atomic_update',['../struct_none.html#a9d7414f58b3a526758007a4a1da72c74',1,'None::atomic_update()'],['../struct_and.html#a9ddf6d3f16dee000a5f0a30af4d43a7f',1,'And::atomic_update(device mlx_atomic< unsigned int > *out, bool val, int elem_idx, int offset=0)'],['../struct_and.html#af4a5c9f276d21580c0acaf1ff7064c57',1,'And::atomic_update(device mlx_atomic< bool > *out, bool val, uint offset=0)'],['../struct_or.html#a40840c0506bf4ed9a0771ba42c9ebb0f',1,'Or::atomic_update(device mlx_atomic< unsigned int > *out, bool val, uint elem_idx, uint offset=0)'],['../struct_or.html#aad1e6b5b015cd0910625da725afabdf3',1,'Or::atomic_update(device mlx_atomic< bool > *out, bool val, uint offset=0)'],['../struct_sum.html#a3188318e74823f00b9367317ec22ad6b',1,'Sum::atomic_update()'],['../struct_prod.html#a7ad8c7c18b1c54ff1d158683744a324d',1,'Prod::atomic_update()'],['../struct_min.html#a2377172960d21faa67b13c6b1eec5c69',1,'Min::atomic_update()'],['../struct_max.html#ac478a5cd1f567164d786240589d9260c',1,'Max::atomic_update()']]], + ['attach_5fevent_106',['attach_event',['../classmlx_1_1core_1_1array.html#a000c3cfe13cb378bf0523b62816190da',1,'mlx::core::array']]], + ['available_107',['available',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078a308bd3e5bf976888b120dd36d0c2d2ae',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/all_10.js b/docs/build/html/search/all_10.js index 376a7bf3f3..af7db5786f 100644 --- a/docs/build/html/search/all_10.js +++ b/docs/build/html/search/all_10.js @@ -2,7 +2,7 @@ var searchData= [ ['pack_5ffactor_0',['pack_factor',['../struct_quantized_block_loader.html#a8eae73a0c04bf1e41fb96131f6aa500d',1,'QuantizedBlockLoader']]], ['pad_1',['Pad',['../classmlx_1_1core_1_1_pad.html',1,'mlx::core::Pad'],['../classmlx_1_1core_1_1_pad.html#ad03da2c40b1e1f2fdf2649d00fa4ab43',1,'mlx::core::Pad::Pad()']]], - ['pad_2',['pad',['../struct_m_l_x_conv_params.html#ae84a9afb3a95b57e0b763bb0ebda0753',1,'MLXConvParams::pad'],['../group__ops.html#ga4fd04d6472ae7cdb94a174edfe338ea3',1,'mlx::core::pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), StreamOrDevice s={})'],['../group__ops.html#gaca4e00d5e4ed9e9f29d56a17f19c2e31',1,'mlx::core::pad(const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), StreamOrDevice s={})'],['../group__ops.html#ga608164c4f5b4ac2fba374219285d90bc',1,'mlx::core::pad(const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), StreamOrDevice s={})'],['../group__ops.html#ga2e57e26c4d7f3d18802c164656cf915c',1,'mlx::core::pad(const array &a, int pad_width, const array &pad_value=array(0), StreamOrDevice s={})']]], + ['pad_2',['pad',['../struct_m_l_x_conv_params.html#ae84a9afb3a95b57e0b763bb0ebda0753',1,'MLXConvParams::pad'],['../group__ops.html#ga81c2f138ea81d78affd42e9a646cacbc',1,'mlx::core::pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#ga8495c9902b0a67f288918cb528d5abeb',1,'mlx::core::pad(const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#gaeaae2f9637b0b3e2b4c6c0517e285935',1,'mlx::core::pad(const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#ga5b8684361f57cbfd5e6eedf82a62ded0',1,'mlx::core::pad(const array &a, int pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})']]], ['pad_5fgpu_3',['pad_gpu',['../namespacemlx_1_1core.html#a6e2054d396ae487d810642dc19cdd0b0',1,'mlx::core']]], ['params_4',['params',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a09b4719415c5bddb0bb70c704b1d8d02',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a4f8c792ede675d14b70dd19fcf3c5aee',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::params'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a1fed11be2e8d9d594dcdf60e32b936b1',1,'mlx::steel::Conv2DWeightBlockLoader::params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a335c573456ede3dd34bda1eec9842fe2',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::params'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#af67adf4550d69231a259e79c1aae9acc',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a09fd92c74ef57c20b48bc780153365ba',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::params'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ae4759d18c0e5cc3530b3da8493008419',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::params']]], ['params_2eh_5',['params.h',['../conv_2params_8h.html',1,'(Global Namespace)'],['../gemm_2params_8h.html',1,'(Global Namespace)']]], @@ -46,21 +46,22 @@ var searchData= ['pre_5fout_43',['pre_out',['../struct_read_writer.html#a94da8aa85fa2916eaa3eaeb11499234a',1,'ReadWriter::pre_out(float2 elem) const'],['../struct_read_writer.html#a185553204b07a407ef02c41dd78e8239',1,'ReadWriter::pre_out(float2 elem, int length) const']]], ['prepare_5fslice_44',['prepare_slice',['../namespacemlx_1_1core.html#a07ae007c87cf2d723a0fa7b25a2b6a93',1,'mlx::core']]], ['prev_45',['prev',['../backend_2metal_2allocator_8h.html#aadb9e075b376adbd9ff6ba23663113bd',1,'allocator.h']]], - ['primitive_46',['Primitive',['../classmlx_1_1core_1_1_primitive.html',1,'mlx::core::Primitive'],['../classmlx_1_1core_1_1_primitive.html#afc69f22ee1f6e8a9ecc2c3a8f43b8fdb',1,'mlx::core::Primitive::Primitive(Stream stream)'],['../classmlx_1_1core_1_1_primitive.html#a3349f745fae50ca7627f79a731a19e32',1,'mlx::core::Primitive::Primitive(const Primitive &other)=delete'],['../classmlx_1_1core_1_1_primitive.html#a342da891b9882bdee9a0e0c1ac826eda',1,'mlx::core::Primitive::Primitive(Primitive &&other)=delete']]], + ['primitive_46',['Primitive',['../classmlx_1_1core_1_1_primitive.html',1,'mlx::core']]], ['primitive_47',['primitive',['../classmlx_1_1core_1_1array.html#a790548666511d8c6d9f92ee79d2ce14c',1,'mlx::core::array']]], - ['primitive_5fid_48',['primitive_id',['../classmlx_1_1core_1_1array.html#af5ad83605d4eea81561246873bee1d7c',1,'mlx::core::array']]], - ['primitive_5fptr_49',['primitive_ptr',['../classmlx_1_1core_1_1array.html#a5119cd616ec3c05d65878944b8889469',1,'mlx::core::array']]], - ['primitives_2eh_50',['primitives.h',['../distributed_2primitives_8h.html',1,'(Global Namespace)'],['../primitives_8h.html',1,'(Global Namespace)']]], - ['print_51',['print',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a6814f9008a683c6911d5b8991ef770ab',1,'mlx::core::distributed::AllReduce::print()'],['../classmlx_1_1core_1_1_primitive.html#ae1aff91354ce036596088a3e19474ecb',1,'mlx::core::Primitive::print()'],['../classmlx_1_1core_1_1_abs.html#a643d6db5116eed978e3208804a992107',1,'mlx::core::Abs::print()'],['../classmlx_1_1core_1_1_add.html#a8a96345aa63724f22b68bca7b861211d',1,'mlx::core::Add::print()'],['../classmlx_1_1core_1_1_add_m_m.html#a1262ac2c4c6e9ff6b6047bf7605e5cc9',1,'mlx::core::AddMM::print()'],['../classmlx_1_1core_1_1_arange.html#abd73d2b793da796dc7cf04c9f7d5c19e',1,'mlx::core::Arange::print()'],['../classmlx_1_1core_1_1_arc_cos.html#aa48d8bec4efbac569d809cf11648b739',1,'mlx::core::ArcCos::print()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6a9a2ab0cc360d7e2f9676db17f8e630',1,'mlx::core::ArcCosh::print()'],['../classmlx_1_1core_1_1_arc_sin.html#a895a35c9dd22fdb06e7b971bfd6fde87',1,'mlx::core::ArcSin::print()'],['../classmlx_1_1core_1_1_arc_sinh.html#aa8b2934a8a0b2eedec8257bbb5726430',1,'mlx::core::ArcSinh::print()'],['../classmlx_1_1core_1_1_arc_tan.html#ab0309e4feca36f221b3d672dc92cac05',1,'mlx::core::ArcTan::print()'],['../classmlx_1_1core_1_1_arc_tan2.html#abdfef9f572d06df1251c28222756a361',1,'mlx::core::ArcTan2::print()'],['../classmlx_1_1core_1_1_arc_tanh.html#aa9549311240d7ba225b84e1df9ad8523',1,'mlx::core::ArcTanh::print()'],['../classmlx_1_1core_1_1_arg_partition.html#aa8678d94fa1571ea71a7bf790cdb8d63',1,'mlx::core::ArgPartition::print()'],['../classmlx_1_1core_1_1_arg_reduce.html#a153a6d8dba7301c4fcd0e429154ead8f',1,'mlx::core::ArgReduce::print()'],['../classmlx_1_1core_1_1_arg_sort.html#a0b59ce43e0982d634a01631728b419bd',1,'mlx::core::ArgSort::print()'],['../classmlx_1_1core_1_1_as_type.html#aa617e29147c14bd5d1fa8ad0bf65af0c',1,'mlx::core::AsType::print()'],['../classmlx_1_1core_1_1_as_strided.html#af2e21b77ea9e6c70bca45224967745bf',1,'mlx::core::AsStrided::print()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a69b28e239da7fdb89f0a9f9467dd797d',1,'mlx::core::BitwiseBinary::print()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a37ecf6fa296d28efb7651a3c510fe159',1,'mlx::core::BlockMaskedMM::print()'],['../classmlx_1_1core_1_1_gather_m_m.html#ae7a6f4eecb15e95b21e6c87068ebd758',1,'mlx::core::GatherMM::print()'],['../classmlx_1_1core_1_1_broadcast.html#a6a610412861c6e472f930b6721b99a11',1,'mlx::core::Broadcast::print()'],['../classmlx_1_1core_1_1_ceil.html#a14a0048dd6496341cacaddada68276ee',1,'mlx::core::Ceil::print()'],['../classmlx_1_1core_1_1_compiled.html#a271521f92eef49c39799f38e26b64a9b',1,'mlx::core::Compiled::print()'],['../classmlx_1_1core_1_1_concatenate.html#a56f29b585a6d1d958954a68dcc893f33',1,'mlx::core::Concatenate::print()'],['../classmlx_1_1core_1_1_conjugate.html#a40281539bbd543ac8fd8e28650de17e4',1,'mlx::core::Conjugate::print()'],['../classmlx_1_1core_1_1_convolution.html#a844eab7c4cc99e775cfb561265ed14fd',1,'mlx::core::Convolution::print()'],['../classmlx_1_1core_1_1_copy.html#acfa1a02ab9cdab593e928faa515a8008',1,'mlx::core::Copy::print()'],['../classmlx_1_1core_1_1_cos.html#a81858457e4bea931a4bc6f6e38b0f696',1,'mlx::core::Cos::print()'],['../classmlx_1_1core_1_1_cosh.html#ac247faad68c1050cda9f72d7d6d040e2',1,'mlx::core::Cosh::print()'],['../classmlx_1_1core_1_1_custom_transforms.html#a2ddbacbc468271b11caee0ad97005298',1,'mlx::core::CustomTransforms::print()'],['../classmlx_1_1core_1_1_depends.html#aed575b0d927f4341f60442c70adeeb82',1,'mlx::core::Depends::print()'],['../classmlx_1_1core_1_1_divide.html#af3c15337ac15522cc34ed98b97895bb6',1,'mlx::core::Divide::print()'],['../classmlx_1_1core_1_1_div_mod.html#a7edbed50d07869d921e529157931b7a1',1,'mlx::core::DivMod::print()'],['../classmlx_1_1core_1_1_select.html#a678285f2c0b9dae85692399c3aa692a7',1,'mlx::core::Select::print()'],['../classmlx_1_1core_1_1_remainder.html#aeaecac5ea8e606d7ecd393d8019029e4',1,'mlx::core::Remainder::print()'],['../classmlx_1_1core_1_1_equal.html#a0787bf32f0b405a8b2ac809d2d990774',1,'mlx::core::Equal::print()'],['../classmlx_1_1core_1_1_erf.html#a186af7b783cf832c3b25eec3a09f5a0c',1,'mlx::core::Erf::print()'],['../classmlx_1_1core_1_1_erf_inv.html#a0acb31bd5780abf61877bd1a3e0fd4f9',1,'mlx::core::ErfInv::print()'],['../classmlx_1_1core_1_1_exp.html#ad87cc1b2ae595a613b03b0fdca63ae6a',1,'mlx::core::Exp::print()'],['../classmlx_1_1core_1_1_expm1.html#af1a99266fc50aa5948cdd298e2916ef1',1,'mlx::core::Expm1::print()'],['../classmlx_1_1core_1_1_f_f_t.html#a15a2a5f7647f5fb78611a251d3270edf',1,'mlx::core::FFT::print()'],['../classmlx_1_1core_1_1_floor.html#ac289e87c5fac15e2f491e2513be610f6',1,'mlx::core::Floor::print()'],['../classmlx_1_1core_1_1_full.html#a68e08303f4960ab373b84a3312edc013',1,'mlx::core::Full::print()'],['../classmlx_1_1core_1_1_gather.html#a9d57637a8a65008683c3847251bdcf91',1,'mlx::core::Gather::print()'],['../classmlx_1_1core_1_1_greater.html#aa2980e45cd2c79ebfb394012d3108a04',1,'mlx::core::Greater::print()'],['../classmlx_1_1core_1_1_greater_equal.html#ab98045c861d2d2ffb0398c2c1d671cef',1,'mlx::core::GreaterEqual::print()'],['../classmlx_1_1core_1_1_hadamard.html#a3df6e7e3b3b71bf50be5f1a05d0870b6',1,'mlx::core::Hadamard::print()'],['../classmlx_1_1core_1_1_less.html#ad67e6f66d7b75546fd98dbee6b631d78',1,'mlx::core::Less::print()'],['../classmlx_1_1core_1_1_less_equal.html#a409842d3862113c53cbbdf7467a06950',1,'mlx::core::LessEqual::print()'],['../classmlx_1_1core_1_1_load.html#a54e08a0ca41b7c9f1a76b00c889f0bfa',1,'mlx::core::Load::print()'],['../classmlx_1_1core_1_1_log.html#a7b946d98d4a228c6be9f606a3bd8a30d',1,'mlx::core::Log::print()'],['../classmlx_1_1core_1_1_log1p.html#a8a1569dde30440ce11ea466ccc69d2d4',1,'mlx::core::Log1p::print()'],['../classmlx_1_1core_1_1_logical_not.html#a001ff3eca46440f0d8a287e0b98a8a2c',1,'mlx::core::LogicalNot::print()'],['../classmlx_1_1core_1_1_logical_and.html#a9a5220eb56e1fd94fd879394ee5ad397',1,'mlx::core::LogicalAnd::print()'],['../classmlx_1_1core_1_1_logical_or.html#a6becc5fbfadde850de9857099dcd5003',1,'mlx::core::LogicalOr::print()'],['../classmlx_1_1core_1_1_log_add_exp.html#a702a2eff0bd1ae7b6fb829dd0b0b11b9',1,'mlx::core::LogAddExp::print()'],['../classmlx_1_1core_1_1_matmul.html#abb4a16a265a05d56a2f5d2e89d6f9dfd',1,'mlx::core::Matmul::print()'],['../classmlx_1_1core_1_1_maximum.html#a3b708a1d6b526719c62850294776f8ca',1,'mlx::core::Maximum::print()'],['../classmlx_1_1core_1_1_minimum.html#a137677bf32c626a768b732a7b8575512',1,'mlx::core::Minimum::print()'],['../classmlx_1_1core_1_1_multiply.html#aa4f1f7af68346ce80c2636df415c9909',1,'mlx::core::Multiply::print()'],['../classmlx_1_1core_1_1_negative.html#a0d5c30e267ff6468d64f1987f9f83f91',1,'mlx::core::Negative::print()'],['../classmlx_1_1core_1_1_not_equal.html#a12aa2f764880d29e627540610b63af09',1,'mlx::core::NotEqual::print()'],['../classmlx_1_1core_1_1_number_of_elements.html#aecde30826970938f3aa688979a668f52',1,'mlx::core::NumberOfElements::print()'],['../classmlx_1_1core_1_1_pad.html#af87754daaf51f6a6cf8bd4949ca1e70a',1,'mlx::core::Pad::print()'],['../classmlx_1_1core_1_1_partition.html#ab5c7aa4fed325475b33d4004649f0dc0',1,'mlx::core::Partition::print()'],['../classmlx_1_1core_1_1_power.html#a33e2d7ff078426fe66ea2370ceb5af60',1,'mlx::core::Power::print()'],['../classmlx_1_1core_1_1_quantized_matmul.html#aaef8c96d4d40b4fa08ced540d341a4db',1,'mlx::core::QuantizedMatmul::print()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a53c3fa7beb51ce2e1c2da28633406fe0',1,'mlx::core::GatherQMM::print()'],['../classmlx_1_1core_1_1_random_bits.html#a8a5593c34fd868d94b36a8ced1390271',1,'mlx::core::RandomBits::print()'],['../classmlx_1_1core_1_1_reshape.html#a0f2323d5d67ece0eb25ecff565b21862',1,'mlx::core::Reshape::print()'],['../classmlx_1_1core_1_1_reduce.html#a399be3a89553787a0a687706881f03cd',1,'mlx::core::Reduce::print()'],['../classmlx_1_1core_1_1_round.html#af0dfe8943109c936b35ab0082f566f72',1,'mlx::core::Round::print()'],['../classmlx_1_1core_1_1_scan.html#ad5b6308c79e9b985a49df35eadd15b22',1,'mlx::core::Scan::print()'],['../classmlx_1_1core_1_1_scatter.html#aa9d45cbfb27b814517f6016092b30efa',1,'mlx::core::Scatter::print()'],['../classmlx_1_1core_1_1_sigmoid.html#ad4cd19938e5159754aa7516f405580c2',1,'mlx::core::Sigmoid::print()'],['../classmlx_1_1core_1_1_sign.html#a2aa0720fe0a6d2408eb43c25d3d45b0a',1,'mlx::core::Sign::print()'],['../classmlx_1_1core_1_1_sin.html#a73b31005551015897f15c00e8b0222e4',1,'mlx::core::Sin::print()'],['../classmlx_1_1core_1_1_sinh.html#a5b4753d52e80799d4fea0b9172d25a77',1,'mlx::core::Sinh::print()'],['../classmlx_1_1core_1_1_slice.html#a50851148948d924b71817cfbd4401504',1,'mlx::core::Slice::print()'],['../classmlx_1_1core_1_1_slice_update.html#a751eefb9922c56479b4b0de2ad45439b',1,'mlx::core::SliceUpdate::print()'],['../classmlx_1_1core_1_1_softmax.html#aa783610ef6b82b92681e78fc99412d83',1,'mlx::core::Softmax::print()'],['../classmlx_1_1core_1_1_sort.html#ada81b9343f80958174eba708452927a2',1,'mlx::core::Sort::print()'],['../classmlx_1_1core_1_1_split.html#ad0c31fe5972643cc75fde10445fc47f2',1,'mlx::core::Split::print()'],['../classmlx_1_1core_1_1_square.html#a75feb558cd1d615e96309dd7d1e81384',1,'mlx::core::Square::print()'],['../classmlx_1_1core_1_1_sqrt.html#a8681c8de2f50049848d320c47f713c0f',1,'mlx::core::Sqrt::print()'],['../classmlx_1_1core_1_1_stop_gradient.html#acc7a7d51cbf014dae8ba3d20bedcad50',1,'mlx::core::StopGradient::print()'],['../classmlx_1_1core_1_1_subtract.html#a3834fd305435fb5a8e512566832e372b',1,'mlx::core::Subtract::print()'],['../classmlx_1_1core_1_1_tan.html#aeea7c284d595a2a928d5f28a55e9be7f',1,'mlx::core::Tan::print()'],['../classmlx_1_1core_1_1_tanh.html#a73f4976d641daf697cc1a231d773d78e',1,'mlx::core::Tanh::print()'],['../classmlx_1_1core_1_1_uniform.html#a01510998719b19df137451cc37850b8d',1,'mlx::core::Uniform::print()'],['../classmlx_1_1core_1_1_view.html#a513b034919a8a494add3155f910a360c',1,'mlx::core::View::print()'],['../classmlx_1_1core_1_1_transpose.html#ac6c87b850f4e5560aa13a5e1e9f9fe04',1,'mlx::core::Transpose::print()'],['../classmlx_1_1core_1_1_q_r_f.html#aba3526722b3a52b41fa8103b909f7f3b',1,'mlx::core::QRF::print()'],['../classmlx_1_1core_1_1_s_v_d.html#ab87a4e7ef857936bea66ba9e24662f53',1,'mlx::core::SVD::print()'],['../classmlx_1_1core_1_1_inverse.html#a543f18f1ce5c06c897141091e95a66e9',1,'mlx::core::Inverse::print()'],['../classmlx_1_1core_1_1_cholesky.html#a0a8b51ff7f5369d22bdc58910d4aaf84',1,'mlx::core::Cholesky::print()'],['../structmlx_1_1core_1_1_print_formatter.html#a79fad4cf5844db8c92b066539146281b',1,'mlx::core::PrintFormatter::print(std::ostream &os, bool val)'],['../structmlx_1_1core_1_1_print_formatter.html#a8da448a8adae671b26359341ea514316',1,'mlx::core::PrintFormatter::print(std::ostream &os, int16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9d750c134a6fbfa8251c5b1d01d73287',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#adbbb9cbff767f9db73c659a0c07ba633',1,'mlx::core::PrintFormatter::print(std::ostream &os, int32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a520adb07fafd911b22bc24b295e4f6cf',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ab0c702f1ae201e17cd328c9855cf522e',1,'mlx::core::PrintFormatter::print(std::ostream &os, int64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac59a5137ddd8b32aae057bb9826ee80d',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac4b7895d1168cfc1a3d1186d8a414d2f',1,'mlx::core::PrintFormatter::print(std::ostream &os, float16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ae21005f92bc641f2d657096f5d176a6d',1,'mlx::core::PrintFormatter::print(std::ostream &os, bfloat16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a57af5c32561b95d6ac2a3a1dc4f5d43e',1,'mlx::core::PrintFormatter::print(std::ostream &os, float val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9e1dc67c9afb0a09966336504790823d',1,'mlx::core::PrintFormatter::print(std::ostream &os, complex64_t val)']]], - ['print_5fcomplex_5fconstant_52',['print_complex_constant',['../namespacemlx_1_1core.html#a2b78f270942c6eb185e8045f1c5b4286',1,'mlx::core']]], - ['print_5fconstant_53',['print_constant',['../namespacemlx_1_1core.html#a7d11b000895d44d183260634f4192d92',1,'mlx::core']]], - ['print_5ffloat_5fconstant_54',['print_float_constant',['../namespacemlx_1_1core.html#a93a8ac59c644b801ec8881a58368caf2',1,'mlx::core']]], - ['print_5fgraph_55',['print_graph',['../namespacemlx_1_1core.html#a8ac23fb7f4d4c52e592d6296e63b80d4',1,'mlx::core::print_graph(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a17505ed8064dcaddc011cb3d52da2523',1,'mlx::core::print_graph(std::ostream &os, Arrays &&... outputs)']]], - ['print_5fint_5fconstant_56',['print_int_constant',['../namespacemlx_1_1core.html#a948ce3dfc4520d3aa98b33e42f617c64',1,'mlx::core']]], - ['printformatter_57',['PrintFormatter',['../structmlx_1_1core_1_1_print_formatter.html',1,'mlx::core']]], - ['prod_58',['Prod',['../struct_prod.html',1,'Prod< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924acdd1ec09a2fd99c81c561b5c63a4b482',1,'mlx::core::distributed::AllReduce::Prod'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9ac5b077bfec55fe2b141b197dfa00ecf7',1,'mlx::core::Reduce::Prod'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1a33edce755ed1a74632c302ad93a14789',1,'mlx::core::Scan::Prod'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca5e43e8ffd1f5ba49826e2e7ac3450466',1,'mlx::core::Scatter::Prod']]], - ['prod_59',['prod',['../structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89',1,'pocketfft::detail::util::prod()'],['../group__ops.html#ga4a09b7241d564d92548bc2773eb1d544',1,'mlx::core::prod(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga61832191f3c42ea549cf04953edc3602',1,'mlx::core::prod(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2b3935108f641e20a70dbf63f540d970',1,'mlx::core::prod(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0',1,'mlx::core::prod(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['promote_5ftypes_60',['promote_types',['../namespacemlx_1_1core.html#a505922e54acd43114308e3bdbda0e497',1,'mlx::core']]], - ['ptr_61',['ptr',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#a990643feac06961c5599aac098c17b94',1,'mlx::core::allocator::Buffer::ptr() const'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#acb15b2f057568828ea09635ed968b62a',1,'mlx::core::allocator::Buffer::ptr()']]], - ['push_62',['push',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#ae8ebd6bc1b4d18e75bd046005e7dde7b',1,'pocketfft::detail::threading::concurrent_queue']]] + ['primitive_48',['Primitive',['../classmlx_1_1core_1_1_primitive.html#afc69f22ee1f6e8a9ecc2c3a8f43b8fdb',1,'mlx::core::Primitive::Primitive(Stream stream)'],['../classmlx_1_1core_1_1_primitive.html#a3349f745fae50ca7627f79a731a19e32',1,'mlx::core::Primitive::Primitive(const Primitive &other)=delete'],['../classmlx_1_1core_1_1_primitive.html#a342da891b9882bdee9a0e0c1ac826eda',1,'mlx::core::Primitive::Primitive(Primitive &&other)=delete']]], + ['primitive_5fid_49',['primitive_id',['../classmlx_1_1core_1_1array.html#af5ad83605d4eea81561246873bee1d7c',1,'mlx::core::array']]], + ['primitive_5fptr_50',['primitive_ptr',['../classmlx_1_1core_1_1array.html#a5119cd616ec3c05d65878944b8889469',1,'mlx::core::array']]], + ['primitives_2eh_51',['primitives.h',['../distributed_2primitives_8h.html',1,'(Global Namespace)'],['../primitives_8h.html',1,'(Global Namespace)']]], + ['print_52',['print',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a6814f9008a683c6911d5b8991ef770ab',1,'mlx::core::distributed::AllReduce::print()'],['../classmlx_1_1core_1_1_primitive.html#ae1aff91354ce036596088a3e19474ecb',1,'mlx::core::Primitive::print()'],['../classmlx_1_1core_1_1_abs.html#a643d6db5116eed978e3208804a992107',1,'mlx::core::Abs::print()'],['../classmlx_1_1core_1_1_add.html#a8a96345aa63724f22b68bca7b861211d',1,'mlx::core::Add::print()'],['../classmlx_1_1core_1_1_add_m_m.html#a1262ac2c4c6e9ff6b6047bf7605e5cc9',1,'mlx::core::AddMM::print()'],['../classmlx_1_1core_1_1_arange.html#abd73d2b793da796dc7cf04c9f7d5c19e',1,'mlx::core::Arange::print()'],['../classmlx_1_1core_1_1_arc_cos.html#aa48d8bec4efbac569d809cf11648b739',1,'mlx::core::ArcCos::print()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6a9a2ab0cc360d7e2f9676db17f8e630',1,'mlx::core::ArcCosh::print()'],['../classmlx_1_1core_1_1_arc_sin.html#a895a35c9dd22fdb06e7b971bfd6fde87',1,'mlx::core::ArcSin::print()'],['../classmlx_1_1core_1_1_arc_sinh.html#aa8b2934a8a0b2eedec8257bbb5726430',1,'mlx::core::ArcSinh::print()'],['../classmlx_1_1core_1_1_arc_tan.html#ab0309e4feca36f221b3d672dc92cac05',1,'mlx::core::ArcTan::print()'],['../classmlx_1_1core_1_1_arc_tan2.html#abdfef9f572d06df1251c28222756a361',1,'mlx::core::ArcTan2::print()'],['../classmlx_1_1core_1_1_arc_tanh.html#aa9549311240d7ba225b84e1df9ad8523',1,'mlx::core::ArcTanh::print()'],['../classmlx_1_1core_1_1_arg_partition.html#aa8678d94fa1571ea71a7bf790cdb8d63',1,'mlx::core::ArgPartition::print()'],['../classmlx_1_1core_1_1_arg_reduce.html#a153a6d8dba7301c4fcd0e429154ead8f',1,'mlx::core::ArgReduce::print()'],['../classmlx_1_1core_1_1_arg_sort.html#a0b59ce43e0982d634a01631728b419bd',1,'mlx::core::ArgSort::print()'],['../classmlx_1_1core_1_1_as_type.html#aa617e29147c14bd5d1fa8ad0bf65af0c',1,'mlx::core::AsType::print()'],['../classmlx_1_1core_1_1_as_strided.html#af2e21b77ea9e6c70bca45224967745bf',1,'mlx::core::AsStrided::print()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a69b28e239da7fdb89f0a9f9467dd797d',1,'mlx::core::BitwiseBinary::print()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a37ecf6fa296d28efb7651a3c510fe159',1,'mlx::core::BlockMaskedMM::print()'],['../classmlx_1_1core_1_1_gather_m_m.html#ae7a6f4eecb15e95b21e6c87068ebd758',1,'mlx::core::GatherMM::print()'],['../classmlx_1_1core_1_1_broadcast.html#a6a610412861c6e472f930b6721b99a11',1,'mlx::core::Broadcast::print()'],['../classmlx_1_1core_1_1_ceil.html#a14a0048dd6496341cacaddada68276ee',1,'mlx::core::Ceil::print()'],['../classmlx_1_1core_1_1_compiled.html#a271521f92eef49c39799f38e26b64a9b',1,'mlx::core::Compiled::print()'],['../classmlx_1_1core_1_1_concatenate.html#a56f29b585a6d1d958954a68dcc893f33',1,'mlx::core::Concatenate::print()'],['../classmlx_1_1core_1_1_conjugate.html#a40281539bbd543ac8fd8e28650de17e4',1,'mlx::core::Conjugate::print()'],['../classmlx_1_1core_1_1_convolution.html#a844eab7c4cc99e775cfb561265ed14fd',1,'mlx::core::Convolution::print()'],['../classmlx_1_1core_1_1_copy.html#acfa1a02ab9cdab593e928faa515a8008',1,'mlx::core::Copy::print()'],['../classmlx_1_1core_1_1_cos.html#a81858457e4bea931a4bc6f6e38b0f696',1,'mlx::core::Cos::print()'],['../classmlx_1_1core_1_1_cosh.html#ac247faad68c1050cda9f72d7d6d040e2',1,'mlx::core::Cosh::print()'],['../classmlx_1_1core_1_1_custom_transforms.html#a2ddbacbc468271b11caee0ad97005298',1,'mlx::core::CustomTransforms::print()'],['../classmlx_1_1core_1_1_depends.html#aed575b0d927f4341f60442c70adeeb82',1,'mlx::core::Depends::print()'],['../classmlx_1_1core_1_1_divide.html#af3c15337ac15522cc34ed98b97895bb6',1,'mlx::core::Divide::print()'],['../classmlx_1_1core_1_1_div_mod.html#a7edbed50d07869d921e529157931b7a1',1,'mlx::core::DivMod::print()'],['../classmlx_1_1core_1_1_select.html#a678285f2c0b9dae85692399c3aa692a7',1,'mlx::core::Select::print()'],['../classmlx_1_1core_1_1_remainder.html#aeaecac5ea8e606d7ecd393d8019029e4',1,'mlx::core::Remainder::print()'],['../classmlx_1_1core_1_1_equal.html#a0787bf32f0b405a8b2ac809d2d990774',1,'mlx::core::Equal::print()'],['../classmlx_1_1core_1_1_erf.html#a186af7b783cf832c3b25eec3a09f5a0c',1,'mlx::core::Erf::print()'],['../classmlx_1_1core_1_1_erf_inv.html#a0acb31bd5780abf61877bd1a3e0fd4f9',1,'mlx::core::ErfInv::print()'],['../classmlx_1_1core_1_1_exp.html#ad87cc1b2ae595a613b03b0fdca63ae6a',1,'mlx::core::Exp::print()'],['../classmlx_1_1core_1_1_expm1.html#af1a99266fc50aa5948cdd298e2916ef1',1,'mlx::core::Expm1::print()'],['../classmlx_1_1core_1_1_f_f_t.html#a15a2a5f7647f5fb78611a251d3270edf',1,'mlx::core::FFT::print()'],['../classmlx_1_1core_1_1_floor.html#ac289e87c5fac15e2f491e2513be610f6',1,'mlx::core::Floor::print()'],['../classmlx_1_1core_1_1_full.html#a68e08303f4960ab373b84a3312edc013',1,'mlx::core::Full::print()'],['../classmlx_1_1core_1_1_gather.html#a9d57637a8a65008683c3847251bdcf91',1,'mlx::core::Gather::print()'],['../classmlx_1_1core_1_1_greater.html#aa2980e45cd2c79ebfb394012d3108a04',1,'mlx::core::Greater::print()'],['../classmlx_1_1core_1_1_greater_equal.html#ab98045c861d2d2ffb0398c2c1d671cef',1,'mlx::core::GreaterEqual::print()'],['../classmlx_1_1core_1_1_hadamard.html#a3df6e7e3b3b71bf50be5f1a05d0870b6',1,'mlx::core::Hadamard::print()'],['../classmlx_1_1core_1_1_less.html#ad67e6f66d7b75546fd98dbee6b631d78',1,'mlx::core::Less::print()'],['../classmlx_1_1core_1_1_less_equal.html#a409842d3862113c53cbbdf7467a06950',1,'mlx::core::LessEqual::print()'],['../classmlx_1_1core_1_1_load.html#a54e08a0ca41b7c9f1a76b00c889f0bfa',1,'mlx::core::Load::print()'],['../classmlx_1_1core_1_1_log.html#a7b946d98d4a228c6be9f606a3bd8a30d',1,'mlx::core::Log::print()'],['../classmlx_1_1core_1_1_log1p.html#a8a1569dde30440ce11ea466ccc69d2d4',1,'mlx::core::Log1p::print()'],['../classmlx_1_1core_1_1_logical_not.html#a001ff3eca46440f0d8a287e0b98a8a2c',1,'mlx::core::LogicalNot::print()'],['../classmlx_1_1core_1_1_logical_and.html#a9a5220eb56e1fd94fd879394ee5ad397',1,'mlx::core::LogicalAnd::print()'],['../classmlx_1_1core_1_1_logical_or.html#a6becc5fbfadde850de9857099dcd5003',1,'mlx::core::LogicalOr::print()'],['../classmlx_1_1core_1_1_log_add_exp.html#a702a2eff0bd1ae7b6fb829dd0b0b11b9',1,'mlx::core::LogAddExp::print()'],['../classmlx_1_1core_1_1_matmul.html#abb4a16a265a05d56a2f5d2e89d6f9dfd',1,'mlx::core::Matmul::print()'],['../classmlx_1_1core_1_1_maximum.html#a3b708a1d6b526719c62850294776f8ca',1,'mlx::core::Maximum::print()'],['../classmlx_1_1core_1_1_minimum.html#a137677bf32c626a768b732a7b8575512',1,'mlx::core::Minimum::print()'],['../classmlx_1_1core_1_1_multiply.html#aa4f1f7af68346ce80c2636df415c9909',1,'mlx::core::Multiply::print()'],['../classmlx_1_1core_1_1_negative.html#a0d5c30e267ff6468d64f1987f9f83f91',1,'mlx::core::Negative::print()'],['../classmlx_1_1core_1_1_not_equal.html#a12aa2f764880d29e627540610b63af09',1,'mlx::core::NotEqual::print()'],['../classmlx_1_1core_1_1_number_of_elements.html#aecde30826970938f3aa688979a668f52',1,'mlx::core::NumberOfElements::print()'],['../classmlx_1_1core_1_1_pad.html#af87754daaf51f6a6cf8bd4949ca1e70a',1,'mlx::core::Pad::print()'],['../classmlx_1_1core_1_1_partition.html#ab5c7aa4fed325475b33d4004649f0dc0',1,'mlx::core::Partition::print()'],['../classmlx_1_1core_1_1_power.html#a33e2d7ff078426fe66ea2370ceb5af60',1,'mlx::core::Power::print()'],['../classmlx_1_1core_1_1_quantized_matmul.html#aaef8c96d4d40b4fa08ced540d341a4db',1,'mlx::core::QuantizedMatmul::print()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a53c3fa7beb51ce2e1c2da28633406fe0',1,'mlx::core::GatherQMM::print()'],['../classmlx_1_1core_1_1_random_bits.html#a8a5593c34fd868d94b36a8ced1390271',1,'mlx::core::RandomBits::print()'],['../classmlx_1_1core_1_1_reshape.html#a0f2323d5d67ece0eb25ecff565b21862',1,'mlx::core::Reshape::print()'],['../classmlx_1_1core_1_1_reduce.html#a399be3a89553787a0a687706881f03cd',1,'mlx::core::Reduce::print()'],['../classmlx_1_1core_1_1_round.html#af0dfe8943109c936b35ab0082f566f72',1,'mlx::core::Round::print()'],['../classmlx_1_1core_1_1_scan.html#ad5b6308c79e9b985a49df35eadd15b22',1,'mlx::core::Scan::print()'],['../classmlx_1_1core_1_1_scatter.html#aa9d45cbfb27b814517f6016092b30efa',1,'mlx::core::Scatter::print()'],['../classmlx_1_1core_1_1_sigmoid.html#ad4cd19938e5159754aa7516f405580c2',1,'mlx::core::Sigmoid::print()'],['../classmlx_1_1core_1_1_sign.html#a2aa0720fe0a6d2408eb43c25d3d45b0a',1,'mlx::core::Sign::print()'],['../classmlx_1_1core_1_1_sin.html#a73b31005551015897f15c00e8b0222e4',1,'mlx::core::Sin::print()'],['../classmlx_1_1core_1_1_sinh.html#a5b4753d52e80799d4fea0b9172d25a77',1,'mlx::core::Sinh::print()'],['../classmlx_1_1core_1_1_slice.html#a50851148948d924b71817cfbd4401504',1,'mlx::core::Slice::print()'],['../classmlx_1_1core_1_1_slice_update.html#a751eefb9922c56479b4b0de2ad45439b',1,'mlx::core::SliceUpdate::print()'],['../classmlx_1_1core_1_1_softmax.html#aa783610ef6b82b92681e78fc99412d83',1,'mlx::core::Softmax::print()'],['../classmlx_1_1core_1_1_sort.html#ada81b9343f80958174eba708452927a2',1,'mlx::core::Sort::print()'],['../classmlx_1_1core_1_1_split.html#ad0c31fe5972643cc75fde10445fc47f2',1,'mlx::core::Split::print()'],['../classmlx_1_1core_1_1_square.html#a75feb558cd1d615e96309dd7d1e81384',1,'mlx::core::Square::print()'],['../classmlx_1_1core_1_1_sqrt.html#a8681c8de2f50049848d320c47f713c0f',1,'mlx::core::Sqrt::print()'],['../classmlx_1_1core_1_1_stop_gradient.html#acc7a7d51cbf014dae8ba3d20bedcad50',1,'mlx::core::StopGradient::print()'],['../classmlx_1_1core_1_1_subtract.html#a3834fd305435fb5a8e512566832e372b',1,'mlx::core::Subtract::print()'],['../classmlx_1_1core_1_1_tan.html#aeea7c284d595a2a928d5f28a55e9be7f',1,'mlx::core::Tan::print()'],['../classmlx_1_1core_1_1_tanh.html#a73f4976d641daf697cc1a231d773d78e',1,'mlx::core::Tanh::print()'],['../classmlx_1_1core_1_1_uniform.html#a01510998719b19df137451cc37850b8d',1,'mlx::core::Uniform::print()'],['../classmlx_1_1core_1_1_view.html#a513b034919a8a494add3155f910a360c',1,'mlx::core::View::print()'],['../classmlx_1_1core_1_1_transpose.html#ac6c87b850f4e5560aa13a5e1e9f9fe04',1,'mlx::core::Transpose::print()'],['../classmlx_1_1core_1_1_q_r_f.html#aba3526722b3a52b41fa8103b909f7f3b',1,'mlx::core::QRF::print()'],['../classmlx_1_1core_1_1_s_v_d.html#ab87a4e7ef857936bea66ba9e24662f53',1,'mlx::core::SVD::print()'],['../classmlx_1_1core_1_1_inverse.html#a543f18f1ce5c06c897141091e95a66e9',1,'mlx::core::Inverse::print()'],['../classmlx_1_1core_1_1_cholesky.html#a0a8b51ff7f5369d22bdc58910d4aaf84',1,'mlx::core::Cholesky::print()'],['../structmlx_1_1core_1_1_print_formatter.html#a79fad4cf5844db8c92b066539146281b',1,'mlx::core::PrintFormatter::print(std::ostream &os, bool val)'],['../structmlx_1_1core_1_1_print_formatter.html#a8da448a8adae671b26359341ea514316',1,'mlx::core::PrintFormatter::print(std::ostream &os, int16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9d750c134a6fbfa8251c5b1d01d73287',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#adbbb9cbff767f9db73c659a0c07ba633',1,'mlx::core::PrintFormatter::print(std::ostream &os, int32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a520adb07fafd911b22bc24b295e4f6cf',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ab0c702f1ae201e17cd328c9855cf522e',1,'mlx::core::PrintFormatter::print(std::ostream &os, int64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac59a5137ddd8b32aae057bb9826ee80d',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac4b7895d1168cfc1a3d1186d8a414d2f',1,'mlx::core::PrintFormatter::print(std::ostream &os, float16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ae21005f92bc641f2d657096f5d176a6d',1,'mlx::core::PrintFormatter::print(std::ostream &os, bfloat16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a57af5c32561b95d6ac2a3a1dc4f5d43e',1,'mlx::core::PrintFormatter::print(std::ostream &os, float val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9e1dc67c9afb0a09966336504790823d',1,'mlx::core::PrintFormatter::print(std::ostream &os, complex64_t val)']]], + ['print_5fcomplex_5fconstant_53',['print_complex_constant',['../namespacemlx_1_1core.html#a2b78f270942c6eb185e8045f1c5b4286',1,'mlx::core']]], + ['print_5fconstant_54',['print_constant',['../namespacemlx_1_1core.html#a7d11b000895d44d183260634f4192d92',1,'mlx::core']]], + ['print_5ffloat_5fconstant_55',['print_float_constant',['../namespacemlx_1_1core.html#a93a8ac59c644b801ec8881a58368caf2',1,'mlx::core']]], + ['print_5fgraph_56',['print_graph',['../namespacemlx_1_1core.html#a8ac23fb7f4d4c52e592d6296e63b80d4',1,'mlx::core::print_graph(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a17505ed8064dcaddc011cb3d52da2523',1,'mlx::core::print_graph(std::ostream &os, Arrays &&... outputs)']]], + ['print_5fint_5fconstant_57',['print_int_constant',['../namespacemlx_1_1core.html#a948ce3dfc4520d3aa98b33e42f617c64',1,'mlx::core']]], + ['printformatter_58',['PrintFormatter',['../structmlx_1_1core_1_1_print_formatter.html',1,'mlx::core']]], + ['prod_59',['Prod',['../struct_prod.html',1,'Prod< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924acdd1ec09a2fd99c81c561b5c63a4b482',1,'mlx::core::distributed::AllReduce::Prod'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9ac5b077bfec55fe2b141b197dfa00ecf7',1,'mlx::core::Reduce::Prod'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1a33edce755ed1a74632c302ad93a14789',1,'mlx::core::Scan::Prod'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca5e43e8ffd1f5ba49826e2e7ac3450466',1,'mlx::core::Scatter::Prod']]], + ['prod_60',['prod',['../structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89',1,'pocketfft::detail::util::prod()'],['../group__ops.html#ga4a09b7241d564d92548bc2773eb1d544',1,'mlx::core::prod(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga61832191f3c42ea549cf04953edc3602',1,'mlx::core::prod(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2b3935108f641e20a70dbf63f540d970',1,'mlx::core::prod(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0',1,'mlx::core::prod(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['promote_5ftypes_61',['promote_types',['../namespacemlx_1_1core.html#a505922e54acd43114308e3bdbda0e497',1,'mlx::core']]], + ['ptr_62',['ptr',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#a990643feac06961c5599aac098c17b94',1,'mlx::core::allocator::Buffer::ptr() const'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#acb15b2f057568828ea09635ed968b62a',1,'mlx::core::allocator::Buffer::ptr()']]], + ['push_63',['push',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#ae8ebd6bc1b4d18e75bd046005e7dde7b',1,'pocketfft::detail::threading::concurrent_queue']]] ]; diff --git a/docs/build/html/search/all_12.js b/docs/build/html/search/all_12.js index c0f4a1d816..e909d0fca5 100644 --- a/docs/build/html/search/all_12.js +++ b/docs/build/html/search/all_12.js @@ -80,7 +80,7 @@ var searchData= ['reductionoptype_77',['ReductionOpType',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65',1,'mlx::core']]], ['reductionplan_78',['ReductionPlan',['../structmlx_1_1core_1_1_reduction_plan.html',1,'mlx::core::ReductionPlan'],['../structmlx_1_1core_1_1_reduction_plan.html#a3df6acc08ef92647aab19d68b0961610',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_, std::vector< int > shape_, std::vector< size_t > strides_)'],['../structmlx_1_1core_1_1_reduction_plan.html#aec7496f3740a0b0d51aaa606f6fd68f4',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_)']]], ['reference_79',['reference',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a44e2e1f29191c20ec4390de4fa0bd59f',1,'mlx::core::array::ArrayIterator']]], - ['register_5flibrary_80',['register_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a45945f2efcd242d915ffa2171e92bf9d',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::string &lib_path)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a6993ce23a485b725b687628b77751ad2',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::function< std::string(const std::string &)> &lib_path_func=get_colocated_mtllib_path)']]], + ['register_5flibrary_80',['register_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a45945f2efcd242d915ffa2171e92bf9d',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::string &lib_path)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a99ff72689b7beb65ad4541391b0eeabf',1,'mlx::core::metal::Device::register_library(const std::string &lib_name)']]], ['remainder_81',['Remainder',['../structmlx_1_1core_1_1detail_1_1_remainder.html',1,'mlx::core::detail::Remainder'],['../classmlx_1_1core_1_1_remainder.html',1,'mlx::core::Remainder'],['../struct_remainder.html',1,'Remainder'],['../classmlx_1_1core_1_1_remainder.html#a4f3eada4a21898af4a77d1d27ce14641',1,'mlx::core::Remainder::Remainder()']]], ['remainder_82',['remainder',['../group__ops.html#ga99f5c904f724156a814d7817188351d2',1,'mlx::core']]], ['remaining_83',['remaining',['../classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd',1,'pocketfft::detail::multi_iter::remaining()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6',1,'pocketfft::detail::simple_iter::remaining()'],['../classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63',1,'pocketfft::detail::rev_iter::remaining()']]], @@ -121,8 +121,7 @@ var searchData= ['row_5freduce_5fgeneral_5fmed_118',['row_reduce_general_med',['../reduce__row_8h.html#ad77d444bcfa9e6afcc1ffc1dcb919f1c',1,'reduce_row.h']]], ['row_5freduce_5fgeneral_5fno_5fatomics_119',['row_reduce_general_no_atomics',['../reduce__row_8h.html#a034fdccc83e66871550c2d42399f0ab5',1,'reduce_row.h']]], ['row_5freduce_5fgeneral_5fsmall_120',['row_reduce_general_small',['../reduce__row_8h.html#aea2b1ba640fffb71c73550ec747b3148',1,'reduce_row.h']]], - ['rows_121',['rows',['../class_m_p_s_1_1_matrix_descriptor.html#a04ce0dd1d4632ff8c4328e4b62d71fc0',1,'MPS::MatrixDescriptor']]], - ['rsqrt_122',['Rsqrt',['../structmlx_1_1core_1_1detail_1_1_rsqrt.html',1,'mlx::core::detail::Rsqrt'],['../struct_rsqrt.html',1,'Rsqrt']]], - ['rsqrt_123',['rsqrt',['../namespacemetal.html#a1cf4b605c0aa7ff5bfe5e979a16f5157',1,'metal::rsqrt()'],['../namespacemetal_1_1fast.html#aa62097c750f1e4b69d09277f19976ab1',1,'metal::fast::rsqrt()'],['../namespacemetal_1_1precise.html#afb397b477745f12a44423934fa2b05ac',1,'metal::precise::rsqrt()'],['../group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1',1,'mlx::core::rsqrt()']]], - ['run_124',['run',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel']]] + ['rsqrt_121',['Rsqrt',['../structmlx_1_1core_1_1detail_1_1_rsqrt.html',1,'mlx::core::detail::Rsqrt'],['../struct_rsqrt.html',1,'Rsqrt']]], + ['rsqrt_122',['rsqrt',['../namespacemetal.html#a1cf4b605c0aa7ff5bfe5e979a16f5157',1,'metal::rsqrt()'],['../namespacemetal_1_1fast.html#aa62097c750f1e4b69d09277f19976ab1',1,'metal::fast::rsqrt()'],['../namespacemetal_1_1precise.html#afb397b477745f12a44423934fa2b05ac',1,'metal::precise::rsqrt()'],['../group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1',1,'mlx::core::rsqrt()']]], + ['run_123',['run',['../struct_g_e_m_v_kernel.html#ac4a7b5011a0ea938ab1949bb1767fc1a',1,'GEMVKernel::run()'],['../struct_g_e_m_v_t_kernel.html#a5d68656832de892f33db939005713927',1,'GEMVTKernel::run()'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run()']]] ]; diff --git a/docs/build/html/search/all_13.js b/docs/build/html/search/all_13.js index 625fd6f33c..b26cbaadc4 100644 --- a/docs/build/html/search/all_13.js +++ b/docs/build/html/search/all_13.js @@ -46,154 +46,150 @@ var searchData= ['set_5fstatus_43',['set_status',['../classmlx_1_1core_1_1array.html#a63598018999b49f1340b183cb303f05c',1,'mlx::core::array']]], ['set_5ftracer_44',['set_tracer',['../classmlx_1_1core_1_1array.html#af26e6be1a9e6239471a4c24310c0c7c8',1,'mlx::core::array']]], ['set_5fvalue_45',['set_value',['../classmlx_1_1core_1_1_event.html#a0d077b11f4b28f882b42440b7ac6d40d',1,'mlx::core::Event']]], - ['setbatchsize_46',['setBatchSize',['../class_m_p_s_1_1_matrix_multiplication.html#ac24cc7d71cdf3fd5bc4291ba2bfcd9e7',1,'MPS::MatrixMultiplication']]], - ['setbatchstart_47',['setBatchStart',['../class_m_p_s_1_1_matrix_multiplication.html#a82008429b4f14b5eba13142886b76cde',1,'MPS::MatrixMultiplication']]], - ['setleftmatrixorigin_48',['setLeftMatrixOrigin',['../class_m_p_s_1_1_matrix_multiplication.html#a8b905b75090af80d97cf09b05cdad528',1,'MPS::MatrixMultiplication']]], - ['setresultmatrixorigin_49',['setResultMatrixOrigin',['../class_m_p_s_1_1_matrix_multiplication.html#a86647678c55adffc51114d036445ba10',1,'MPS::MatrixMultiplication']]], - ['setrightmatrixorigin_50',['setRightMatrixOrigin',['../class_m_p_s_1_1_matrix_multiplication.html#afaa45acd37397737ef4d94e8803be7e2',1,'MPS::MatrixMultiplication']]], - ['shape_51',['shape',['../structmlx_1_1core_1_1_reduction_plan.html#a6cfa8771fa9caf6fdcc3d74c9fca83ae',1,'mlx::core::ReductionPlan::shape'],['../classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0',1,'pocketfft::detail::arr_info::shape() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8',1,'pocketfft::detail::arr_info::shape(size_t i) const'],['../classmlx_1_1core_1_1array.html#a4a2a2c8a4a5beafd723fc13f2055d55d',1,'mlx::core::array::shape() const'],['../classmlx_1_1core_1_1array.html#a51ed0c45666264dc172d06fba159eb8f',1,'mlx::core::array::shape(int dim) const']]], - ['shape_5ft_52',['shape_t',['../namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::detail']]], - ['shapes_53',['shapes',['../struct_indices.html#a5ab170f1a77636180889ddfffd4f7d2f',1,'Indices']]], - ['shapes_5fwithout_5freduction_5faxes_54',['shapes_without_reduction_axes',['../namespacemlx_1_1core.html#a44c3ea6db6553c3f6552b9ba64a69494',1,'mlx::core']]], - ['shared_5fbuffer_5fslice_55',['shared_buffer_slice',['../namespacemlx_1_1core.html#a427f2c255dfc6e1f83f97587b08e71bc',1,'mlx::core']]], - ['shp_56',['shp',['../classpocketfft_1_1detail_1_1arr__info.html#a2467e9e01de1ba4d7cd28c1af783da8d',1,'pocketfft::detail::arr_info']]], - ['shutdown_57',['shutdown',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411',1,'pocketfft::detail::threading::thread_pool']]], - ['siblings_58',['siblings',['../classmlx_1_1core_1_1array.html#acf80fde8f743f65ad5b4be69fcb7a74d',1,'mlx::core::array::siblings() const'],['../classmlx_1_1core_1_1array.html#a7263f23e70a580a9bc2129fbcde36e6c',1,'mlx::core::array::siblings()']]], - ['sigmoid_59',['Sigmoid',['../structmlx_1_1core_1_1detail_1_1_sigmoid.html',1,'mlx::core::detail::Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html',1,'mlx::core::Sigmoid'],['../struct_sigmoid.html',1,'Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html#a47eca99113ec19f0eb60b6a0472c592b',1,'mlx::core::Sigmoid::Sigmoid()']]], - ['sigmoid_60',['sigmoid',['../group__ops.html#ga708abf8f79609cd6831db7c38cafac0e',1,'mlx::core']]], - ['sign_61',['Sign',['../structmlx_1_1core_1_1detail_1_1_sign.html',1,'mlx::core::detail::Sign'],['../classmlx_1_1core_1_1_sign.html',1,'mlx::core::Sign'],['../struct_sign.html',1,'Sign'],['../classmlx_1_1core_1_1_sign.html#afe951e50907bc23a601ec5fa9eae5763',1,'mlx::core::Sign::Sign()']]], - ['sign_62',['sign',['../group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa',1,'mlx::core']]], - ['signal_63',['signal',['../classmlx_1_1core_1_1_event.html#a65a858445506a61be5889ae0e3651b89',1,'mlx::core::Event']]], - ['signaling_5fnan_64',['signaling_NaN',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ad1f76a43c7d51a3765174aa6e0dd9f80',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['signedinteger_65',['signedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2daed58b4631ff157bec9e35ed1182d2c10',1,'mlx::core::Dtype::signedinteger'],['../namespacemlx_1_1core.html#a24e1618af591d737d73729665e868001',1,'mlx::core::signedinteger']]], - ['simd_5fbroadcast_66',['simd_broadcast',['../namespacemetal.html#a498f1e85107eb5f01ba4435977f8efe0',1,'metal']]], - ['simd_5fexclusive_5fscan_67',['simd_exclusive_scan',['../struct_cum_sum.html#a64d1933942de4df2d44dd90e9e5e4460',1,'CumSum::simd_exclusive_scan()'],['../struct_cum_prod.html#a5b0577295853a48856b84fd4f935dff8',1,'CumProd::simd_exclusive_scan()'],['../struct_cum_prod_3_01bool_01_4.html#a1a86e9398bae24182b7be0a6577bf223',1,'CumProd< bool >::simd_exclusive_scan()'],['../struct_cum_max.html#ae11b67aa6c998e9a01615b2a79af4403',1,'CumMax::simd_exclusive_scan()'],['../struct_cum_min.html#a83e65017ff33018b585c043fb803773b',1,'CumMin::simd_exclusive_scan()']]], - ['simd_5fmax_68',['simd_max',['../namespacemetal.html#a048cad0aca52cb737ebf103e76bd1c49',1,'metal']]], - ['simd_5fmin_69',['simd_min',['../namespacemetal.html#ae9e2a23e00724ba2d7868bc4112b386b',1,'metal']]], - ['simd_5fprefix_5fexclusive_5fproduct_70',['simd_prefix_exclusive_product',['../namespacemetal.html#a5ca40242390b632f737e29636829b2e4',1,'metal']]], - ['simd_5fprefix_5fexclusive_5fsum_71',['simd_prefix_exclusive_sum',['../namespacemetal.html#abfbb70c7471f28bf7ff36a612ad014b2',1,'metal']]], - ['simd_5fprefix_5finclusive_5fproduct_72',['simd_prefix_inclusive_product',['../namespacemetal.html#a6ca6a7e1996228fa536e969e9e45c446',1,'metal']]], - ['simd_5fprefix_5finclusive_5fsum_73',['simd_prefix_inclusive_sum',['../namespacemetal.html#a567acb18199ac0107712eb8cb8aeb8e9',1,'metal']]], - ['simd_5fproduct_74',['simd_product',['../namespacemetal.html#ac6e883a04e2265a9790d7db76059e1b4',1,'metal']]], - ['simd_5freduce_75',['simd_reduce',['../struct_and.html#ab4df5517e050cfa7257c34c6b0bd980b',1,'And::simd_reduce()'],['../struct_or.html#aef6b5abbbf3fd22981b8d27562022605',1,'Or::simd_reduce()'],['../struct_sum.html#acf10e0604098391526da3b8271928b5c',1,'Sum::simd_reduce()'],['../struct_prod.html#ab0d7a38360b971b4d99c4ad2db310069',1,'Prod::simd_reduce()'],['../struct_min.html#afe8d7cefac624fd4267c1b252ac38528',1,'Min::simd_reduce()'],['../struct_max.html#a4f74440c6f1ce8247d5a87c2a8882080',1,'Max::simd_reduce()']]], - ['simd_5fscan_76',['simd_scan',['../struct_cum_sum.html#a83342af06f6a700067e3b451636852d6',1,'CumSum::simd_scan()'],['../struct_cum_prod.html#a860dbf6146de088d07175bd165b74004',1,'CumProd::simd_scan()'],['../struct_cum_prod_3_01bool_01_4.html#abeb5ec4237b330e7219f4e881cf10d7a',1,'CumProd< bool >::simd_scan()'],['../struct_cum_max.html#adc9ec8bb09b4433d4c2f03022c43d781',1,'CumMax::simd_scan()'],['../struct_cum_min.html#a0a1005d91b1c90e90e2c6dbd6c296649',1,'CumMin::simd_scan()']]], - ['simd_5fshuffle_77',['simd_shuffle',['../namespacemetal.html#a259ed115bc3c58f88eb35830916b26d4',1,'metal']]], - ['simd_5fshuffle_5fand_5ffill_5fdown_78',['simd_shuffle_and_fill_down',['../namespacemetal.html#ae29a06f0eac636ad7af21dea5b04938b',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a0ee6239fa29a5f9ee0201e0dc5ddc8e0',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], - ['simd_5fshuffle_5fand_5ffill_5fup_79',['simd_shuffle_and_fill_up',['../namespacemetal.html#a1ca14116bf50639b214d8414b5bbaaa6',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a5138d5cdc18139e135707916a243cd8e',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], - ['simd_5fshuffle_5fdown_80',['simd_shuffle_down',['../namespacemetal.html#af6e2dd7ae087aba6abac4f0350b7611c',1,'metal::simd_shuffle_down()'],['../backend_2metal_2kernels_2utils_8h.html#aba6279624b1d30c525efee856a222b5c',1,'simd_shuffle_down(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0c1e4d782fcc56e1ab5565cef12430dd',1,'simd_shuffle_down(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a48ae83a8caf5c74810df60b6c6cdb062',1,'simd_shuffle_down(bool data, uint16_t delta): utils.h']]], - ['simd_5fshuffle_5frotate_5fdown_81',['simd_shuffle_rotate_down',['../namespacemetal.html#a4bb203647a421032db47e73cd649841b',1,'metal']]], - ['simd_5fshuffle_5frotate_5fup_82',['simd_shuffle_rotate_up',['../namespacemetal.html#a729b22077d6c944491a6027c18ea80c9',1,'metal']]], - ['simd_5fshuffle_5fup_83',['simd_shuffle_up',['../namespacemetal.html#afe81c5fbde3f4890458b081909242c55',1,'metal']]], - ['simd_5fshuffle_5fxor_84',['simd_shuffle_xor',['../namespacemetal.html#a5017efc9605e069cfb507137cd1a1852',1,'metal']]], - ['simd_5fsize_85',['simd_size',['../backend_2metal_2kernels_2reduction_2ops_8h.html#a515b75d563a93d3c09ee677948dc83e3',1,'ops.h']]], - ['simd_5fsize_86',['SIMD_SIZE',['../quantized_8h.html#a62969a218d93680f5e35d0c61b160b99',1,'quantized.h']]], - ['simd_5fstride_5fa_87',['simd_stride_a',['../structmlx_1_1steel_1_1_block_m_m_a.html#ab52d2644053c5bb28ab9b24fab8b194a',1,'mlx::steel::BlockMMA']]], - ['simd_5fstride_5fb_88',['simd_stride_b',['../structmlx_1_1steel_1_1_block_m_m_a.html#af5c60b04602f72b705192911ed8749da',1,'mlx::steel::BlockMMA']]], - ['simd_5fsum_89',['simd_sum',['../namespacemetal.html#a85181e37a00cb4a4217f1bb25389bce5',1,'metal']]], - ['simd_5fxor_90',['simd_xor',['../namespacemetal.html#a1308decbf2d5c33d34d6be523ea1c30f',1,'metal']]], - ['simple_5fiter_91',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html',1,'pocketfft::detail::simple_iter'],['../classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398',1,'pocketfft::detail::simple_iter::simple_iter()']]], - ['simplevalueandgradfn_92',['SimpleValueAndGradFn',['../namespacemlx_1_1core.html#a2689b8f1181648cb1685204fea9f3066',1,'mlx::core']]], - ['sin_93',['Sin',['../structmlx_1_1core_1_1detail_1_1_sin.html',1,'mlx::core::detail::Sin'],['../classmlx_1_1core_1_1_sin.html',1,'mlx::core::Sin'],['../struct_sin.html',1,'Sin'],['../classmlx_1_1core_1_1_sin.html#a10d1ecc0ca96e79cdf55b57073d126ea',1,'mlx::core::Sin::Sin()']]], - ['sin_94',['sin',['../namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258',1,'pocketfft::detail::sin()'],['../namespacemetal.html#a619a159ca5f2ddfe3647d3a6bb6e804c',1,'metal::sin()'],['../namespacemetal_1_1fast.html#a3af771cfe7a135104f9d063147dba270',1,'metal::fast::sin()'],['../namespacemetal_1_1precise.html#a71acf77ffd29c56f56afae0195c98a1c',1,'metal::precise::sin()'],['../group__ops.html#gaebf0a73ad3732fba39df37826c235692',1,'mlx::core::sin()']]], - ['sincos_5f2pibyn_95',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html',1,'pocketfft::detail::sincos_2pibyn< T >'],['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40',1,'pocketfft::detail::sincos_2pibyn::sincos_2pibyn()']]], - ['sinh_96',['Sinh',['../structmlx_1_1core_1_1detail_1_1_sinh.html',1,'mlx::core::detail::Sinh'],['../classmlx_1_1core_1_1_sinh.html',1,'mlx::core::Sinh'],['../struct_sinh.html',1,'Sinh'],['../classmlx_1_1core_1_1_sinh.html#a4a4f6814d403c2ce5d6c574b0dca3c96',1,'mlx::core::Sinh::Sinh()']]], - ['sinh_97',['sinh',['../namespacemetal.html#a83ba4235ae350ab8880a9df09158620b',1,'metal::sinh()'],['../namespacemetal_1_1fast.html#a990d90b3440e38d1fb4ff5065c6c189b',1,'metal::fast::sinh()'],['../namespacemetal_1_1precise.html#abc8f4f59dd6e7204ab5d84f0af96331c',1,'metal::precise::sinh()'],['../group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7',1,'mlx::core::sinh()']]], - ['sinpi_98',['sinpi',['../namespacemetal.html#ae9655f7fa2ba6c0625ca25fbb278e269',1,'metal::sinpi()'],['../namespacemetal_1_1fast.html#ab07a32fe544aa304577d29e0251e87b2',1,'metal::fast::sinpi()'],['../namespacemetal_1_1precise.html#a78b17dab93519d9c82c2575dafec49c9',1,'metal::precise::sinpi()']]], - ['size_99',['size',['../structmlx_1_1core_1_1_dtype.html#adc24e03df52237e9ba03e2451f321b66',1,'mlx::core::Dtype::size'],['../classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952',1,'pocketfft::detail::arr::size()'],['../classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5',1,'pocketfft::detail::arr_info::size()'],['../classmlx_1_1core_1_1array.html#a598f87161926d9e0b516860f0ea2c8f6',1,'mlx::core::array::size()'],['../structmlx_1_1core_1_1distributed_1_1_group.html#abd96a09217e3d1bcc522888257d22cef',1,'mlx::core::distributed::Group::size()']]], - ['size_5fof_100',['size_of',['../namespacemlx_1_1core.html#add4794cc0ffe5d717fc146084a235d95',1,'mlx::core']]], - ['slice_101',['Slice',['../classmlx_1_1core_1_1_slice.html',1,'mlx::core::Slice'],['../classmlx_1_1core_1_1_slice.html#a8a38feb7bb6b72bdeebb83f053e2fd7f',1,'mlx::core::Slice::Slice()']]], - ['slice_102',['slice',['../group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21',1,'mlx::core::slice(const array &a, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})'],['../group__ops.html#ga90b243e25b5fbbd7a9732f7ac48f6403',1,'mlx::core::slice(const array &a, const std::vector< int > &start, const std::vector< int > &stop, StreamOrDevice s={})']]], - ['slice_5fgpu_103',['slice_gpu',['../namespacemlx_1_1core.html#a5ab133d67da9c16621f61690e27fccfd',1,'mlx::core']]], - ['slice_5fupdate_104',['slice_update',['../group__ops.html#ga3875660e4ce2c8add8bfcf8144078708',1,'mlx::core::slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})'],['../group__ops.html#ga03ffbbb4d989a463ef43f41ebf7eabef',1,'mlx::core::slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, StreamOrDevice s={})']]], - ['sliceupdate_105',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html',1,'mlx::core::SliceUpdate'],['../classmlx_1_1core_1_1_slice_update.html#aa30a7f22f557c56e1a2b5fcf44488990',1,'mlx::core::SliceUpdate::SliceUpdate()']]], - ['slicing_2eh_106',['slicing.h',['../common_2slicing_8h.html',1,'(Global Namespace)'],['../metal_2slicing_8h.html',1,'(Global Namespace)']]], - ['sm_107',['sm',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa85451edf6900fd6af164d4d50889ae3',1,'mlx::steel::BlockMMA']]], - ['sn_108',['sn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ade420e8b811d597345783c324c23a34a',1,'mlx::steel::BlockMMA']]], - ['softmax_109',['Softmax',['../classmlx_1_1core_1_1_softmax.html',1,'mlx::core::Softmax'],['../classmlx_1_1core_1_1_softmax.html#a4ec686aac4e06f0dfe2cbd6801af40eb',1,'mlx::core::Softmax::Softmax()']]], - ['softmax_110',['softmax',['../namespacemlx_1_1core_1_1metal.html#a4fe937c2c584fd646926057f31d54ca6',1,'mlx::core::metal::softmax()'],['../group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6',1,'mlx::core::softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga1ae3614d07d873892a530d14c3857d0b',1,'mlx::core::softmax(const array &a, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga06f570d73716a24303e6de3aaba4457b',1,'mlx::core::softmax(const array &a, int axis, bool precise=false, StreamOrDevice s={})']]], - ['softmax_2eh_111',['softmax.h',['../jit_2softmax_8h.html',1,'(Global Namespace)'],['../kernels_2softmax_8h.html',1,'(Global Namespace)']]], - ['softmax_5fexp_112',['softmax_exp',['../kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b',1,'softmax.h']]], - ['softmax_5fkernels_113',['softmax_kernels',['../jit_2softmax_8h.html#a1cbfb210a9a765c6620e9f1247ccef12',1,'softmax.h']]], - ['softmax_5flooped_114',['softmax_looped',['../kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276',1,'softmax.h']]], - ['softmax_5fn_5freads_115',['SOFTMAX_N_READS',['../defines_8h.html#a722995df24286b27b7da3d74b73f768d',1,'defines.h']]], - ['softmax_5fsingle_5frow_116',['softmax_single_row',['../kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b',1,'softmax.h']]], - ['sort_117',['Sort',['../classmlx_1_1core_1_1_sort.html',1,'mlx::core::Sort'],['../classmlx_1_1core_1_1_sort.html#a62943032dbd72e85ceb9b4b7211f4a44',1,'mlx::core::Sort::Sort()']]], - ['sort_118',['sort',['../struct_thread_sort.html#ad9ab3e6b47f7e9b91c0f3b773596986d',1,'ThreadSort::sort()'],['../struct_block_merge_sort.html#acc970f5eb963f7f2010f5ae5ea8b8bc0',1,'BlockMergeSort::sort()'],['../namespacemlx_1_1core_1_1metal.html#ab77c9a9ecaeeab8c66b712862777c24b',1,'mlx::core::metal::sort()'],['../group__ops.html#ga7fb616054665b3c2d61fa234f501f079',1,'mlx::core::sort(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaae1bc47aa737f705d0e5884270063fea',1,'mlx::core::sort(const array &a, int axis, StreamOrDevice s={})']]], - ['sort_2eh_119',['sort.h',['../sort_8h.html',1,'']]], - ['special_5fmul_120',['special_mul',['../structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874',1,'pocketfft::detail::cmplx::special_mul()'],['../namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76',1,'pocketfft::detail::special_mul()']]], - ['split_121',['Split',['../classmlx_1_1core_1_1_split.html',1,'mlx::core::Split'],['../classmlx_1_1core_1_1_split.html#a897c746ecfdff5119cc5ae3f20499385',1,'mlx::core::Split::Split()']]], - ['split_122',['split',['../structmlx_1_1core_1_1distributed_1_1_group.html#abbf40f8979488806bc5bca9ecc4130e9',1,'mlx::core::distributed::Group::split()'],['../group__ops.html#ga7534290bceab5fb3831a05d67bebce7d',1,'mlx::core::split(const array &a, int num_splits, int axis, StreamOrDevice s={})'],['../group__ops.html#ga56882d24e5fde59c266774624c892d41',1,'mlx::core::split(const array &a, int num_splits, StreamOrDevice s={})'],['../group__ops.html#ga2cfcb1a53924882e30476c9016c5de74',1,'mlx::core::split(const array &a, const std::vector< int > &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#gac324dfa3e26d3a14a35ab7962e36f0e1',1,'mlx::core::split(const array &a, const std::vector< int > &indices, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a42847b435d037a977592e355eed072af',1,'mlx::core::random::split(const array &key, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7ec057064c7326c41b536f08178861e5',1,'mlx::core::random::split(const array &key, int num, StreamOrDevice s={})']]], - ['split_5fk_5fpartition_5fsize_123',['split_k_partition_size',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a9f5a67b2343645b570e109c3837d4042',1,'mlx::steel::GEMMSpiltKParams']]], - ['split_5fk_5fpartition_5fstride_124',['split_k_partition_stride',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a12144ce89d404812cd862611d770b9fb',1,'mlx::steel::GEMMSpiltKParams']]], - ['split_5fk_5fpartitions_125',['split_k_partitions',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#ae06c27116905d4ff3b9b436e588a93fd',1,'mlx::steel::GEMMSpiltKParams']]], - ['sqrt_126',['Sqrt',['../structmlx_1_1core_1_1detail_1_1_sqrt.html',1,'mlx::core::detail::Sqrt'],['../classmlx_1_1core_1_1_sqrt.html',1,'mlx::core::Sqrt'],['../struct_sqrt.html',1,'Sqrt'],['../classmlx_1_1core_1_1_sqrt.html#a6682a7c31ca427c9d2c5ddb6a479bf29',1,'mlx::core::Sqrt::Sqrt()']]], - ['sqrt_127',['sqrt',['../namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3',1,'pocketfft::detail::sqrt()'],['../namespacemetal.html#ab3f4d4852ca0e591104fbd8e5b50d31b',1,'metal::sqrt()'],['../namespacemetal_1_1fast.html#a4218a85c7d8a74cb8055b4755205627e',1,'metal::fast::sqrt()'],['../namespacemetal_1_1precise.html#acb213467361cd2cab93a8d5ea1aa5bfd',1,'metal::precise::sqrt()'],['../group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1',1,'mlx::core::sqrt()']]], - ['square_128',['Square',['../structmlx_1_1core_1_1detail_1_1_square.html',1,'mlx::core::detail::Square'],['../classmlx_1_1core_1_1_square.html',1,'mlx::core::Square'],['../struct_square.html',1,'Square'],['../classmlx_1_1core_1_1_square.html#ab94e28d5c92e6febc1c74e525f730dc4',1,'mlx::core::Square::Square()']]], - ['square_129',['square',['../group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e',1,'mlx::core']]], - ['squeeze_130',['squeeze',['../group__ops.html#ga710daa7ec721bd4d3f326082cb195576',1,'mlx::core::squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3',1,'mlx::core::squeeze(const array &a, int axis, StreamOrDevice s={})'],['../group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad',1,'mlx::core::squeeze(const array &a, StreamOrDevice s={})']]], - ['src_131',['src',['../struct_quantized_block_loader.html#ad85c6b7e07c81307b3b91eb4dd7be30b',1,'QuantizedBlockLoader::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a656a46ee27486482b45ff90b3d626255',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a57552e9cfbafad71d47b2f3a8e027bdf',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7bfbcc4a1e3eef7aef5dd8e8c374a95f',1,'mlx::steel::Conv2DWeightBlockLoader::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#afe21e46e08523232830c25eb1b4ade16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b04a69952404a04029dacc424df6e8f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1ee2922961b5fcb1db577928c4d9d731',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a50f458dbb74d61be2ed24727d8d43614',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src'],['../structmlx_1_1steel_1_1_block_loader.html#a13004952d0bf2030b95acb621a3779dd',1,'mlx::steel::BlockLoader::src']]], - ['src_5fld_132',['src_ld',['../struct_quantized_block_loader.html#a8050977d473d1a24fae5c833e609839e',1,'QuantizedBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7464ec687323fa79050702952ed9084f',1,'mlx::steel::Conv2DWeightBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#aa6bedc0cbb447eaf70c03f2e26df2cb2',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6918c1df7712c4e408e2871467ea7987',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src_ld'],['../structmlx_1_1steel_1_1_block_loader.html#aadafc50f7f06af434149d7469df4714d',1,'mlx::steel::BlockLoader::src_ld']]], - ['stack_133',['stack',['../group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032',1,'mlx::core::stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga82216209dce901296fc737fe8efa5c94',1,'mlx::core::stack(const std::vector< array > &arrays, StreamOrDevice s={})']]], - ['start_5fcapture_134',['start_capture',['../namespacemlx_1_1core_1_1metal.html#aa47cb5651bf3b65c46ab216b7e504d77',1,'mlx::core::metal']]], - ['start_5fconcurrent_135',['start_concurrent',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::metal::CommandEncoder']]], - ['start_5frow_136',['start_row',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a220e033b689c8d6a6f319dae02b38334',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], - ['status_137',['Status',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078',1,'mlx::core::array']]], - ['status_138',['status',['../classmlx_1_1core_1_1array.html#a7102659be87e9ef62966696ab9b07dad',1,'mlx::core::array']]], - ['std_139',['std',['../group__ops.html#ga2a466024f8061febc0a64be557644cb0',1,'mlx::core::std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#gafdcb04d77c64405a3990078a77dd984c',1,'mlx::core::std(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8',1,'mlx::core::std(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga62721a206df671ef5797449eea97af9f',1,'mlx::core::std(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], - ['steel_5fconst_140',['STEEL_CONST',['../steel_2defines_8h.html#a90b91c866313ffa46eff6d9cc944ad2b',1,'defines.h']]], - ['steel_5fconv_141',['steel_conv',['../namespacemlx_1_1core_1_1metal.html#a92f1e559b1121d545746f81ff86eaca1',1,'mlx::core::metal']]], - ['steel_5fconv_2eh_142',['steel_conv.h',['../jit_2steel__conv_8h.html',1,'(Global Namespace)'],['../kernels_2steel_2conv_2kernels_2steel__conv_8h.html',1,'(Global Namespace)']]], - ['steel_5fconv_5fgeneral_143',['steel_conv_general',['../namespacemlx_1_1core_1_1metal.html#a02edb6a90bdf30f4c9f0d6c25b0267b5',1,'mlx::core::metal']]], - ['steel_5fconv_5fgeneral_2eh_144',['steel_conv_general.h',['../steel__conv__general_8h.html',1,'']]], - ['steel_5fconv_5fgeneral_5fkernels_145',['steel_conv_general_kernels',['../jit_2steel__conv_8h.html#ae4ca1720029316b08ea92b7662347d47',1,'steel_conv.h']]], - ['steel_5fconv_5fkernels_146',['steel_conv_kernels',['../jit_2steel__conv_8h.html#a386d79077465df56659416fd84adb513',1,'steel_conv.h']]], - ['steel_5fgemm_2eh_147',['steel_gemm.h',['../steel__gemm_8h.html',1,'']]], - ['steel_5fgemm_5ffused_148',['steel_gemm_fused',['../namespacemlx_1_1core_1_1metal.html#a17764366deed71c160fb26091400a803',1,'mlx::core::metal']]], - ['steel_5fgemm_5ffused_2eh_149',['steel_gemm_fused.h',['../steel__gemm__fused_8h.html',1,'']]], - ['steel_5fgemm_5ffused_5fkernels_150',['steel_gemm_fused_kernels',['../steel__gemm_8h.html#a4c6009fd5357b730805f2fd4ba6e093e',1,'steel_gemm.h']]], - ['steel_5fgemm_5fmasked_151',['steel_gemm_masked',['../namespacemlx_1_1core_1_1metal.html#a962272ca73d26c08f76f706a128fd71f',1,'mlx::core::metal']]], - ['steel_5fgemm_5fmasked_2eh_152',['steel_gemm_masked.h',['../steel__gemm__masked_8h.html',1,'']]], - ['steel_5fgemm_5fmasked_5fkernels_153',['steel_gemm_masked_kernels',['../steel__gemm_8h.html#a62a358fd3ec5365081920d07aceb581c',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_154',['steel_gemm_splitk',['../namespacemlx_1_1core_1_1metal.html#ad0dfd40ba7c09755711ceb731e57a5ac',1,'mlx::core::metal']]], - ['steel_5fgemm_5fsplitk_2eh_155',['steel_gemm_splitk.h',['../steel__gemm__splitk_8h.html',1,'']]], - ['steel_5fgemm_5fsplitk_5faccum_5faxbpy_5fkernels_156',['steel_gemm_splitk_accum_axbpy_kernels',['../steel__gemm_8h.html#a40a86e1381c241aba8511e51a981a4bf',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_5faccum_5fkernels_157',['steel_gemm_splitk_accum_kernels',['../steel__gemm_8h.html#a144a64b8d94f0371fb144e2cc308fcf9',1,'steel_gemm.h']]], - ['steel_5fgemm_5fsplitk_5fkernels_158',['steel_gemm_splitk_kernels',['../steel__gemm_8h.html#a92108ab01d826e38bca83d8569b947d9',1,'steel_gemm.h']]], - ['steel_5fmatmul_159',['steel_matmul',['../namespacemlx_1_1core.html#ab43a7633794498e1c6775cca829eb886',1,'mlx::core']]], - ['steel_5fmatmul_5fconv_5fgroups_160',['steel_matmul_conv_groups',['../namespacemlx_1_1core.html#afe8386cea0c7b846dc78649927fd0c75',1,'mlx::core']]], - ['steel_5fpragma_5funroll_161',['STEEL_PRAGMA_UNROLL',['../steel_2defines_8h.html#a5a5c3095b132a7589bc19cd5cb80e2c6',1,'defines.h']]], - ['stop_162',['stop',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a456ad1c0c9e731833a2f8411c4ed51aa',1,'mlx::core::scheduler::StreamThread']]], - ['stop_5fcapture_163',['stop_capture',['../namespacemlx_1_1core_1_1metal.html#ac90714424e36fb01e04550de69b8314f',1,'mlx::core::metal']]], - ['stop_5fgradient_164',['stop_gradient',['../group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f',1,'mlx::core']]], - ['stopgradient_165',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html',1,'mlx::core::StopGradient'],['../classmlx_1_1core_1_1_stop_gradient.html#ac70d1ab819d04e00f76bc25aeebaf84f',1,'mlx::core::StopGradient::StopGradient()']]], - ['store_5fresult_166',['store_result',['../structmlx_1_1steel_1_1_block_m_m_a.html#a85c57ad4092079f068683cb525fb08ee',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const']]], - ['store_5fresult_5fsafe_167',['store_result_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#af94bafeb285c240051e86418fccb89fe',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const']]], - ['str_168',['str',['../classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4',1,'pocketfft::detail::arr_info::str'],['../struct_m_l_x_conv_params.html#a862191e8ab1bc8a47aa1396b36d46058',1,'MLXConvParams::str']]], - ['stream_169',['Stream',['../structmlx_1_1core_1_1_stream.html',1,'mlx::core']]], - ['stream_170',['stream',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a8462e4acffcd385c6248bd7102e6bcb1',1,'mlx::core::scheduler::StreamThread::stream'],['../classmlx_1_1core_1_1_event.html#a193143bad31b68c699fa27f135b45614',1,'mlx::core::Event::stream()'],['../classmlx_1_1core_1_1_primitive.html#a46e6257397a662528f9f831842ac456a',1,'mlx::core::Primitive::stream()']]], - ['stream_171',['Stream',['../structmlx_1_1core_1_1_stream.html#a7f0815ff4886da74cbbff5f93d82dd3e',1,'mlx::core::Stream']]], - ['stream_2eh_172',['stream.h',['../stream_8h.html',1,'']]], - ['streamcontext_173',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html',1,'mlx::core::StreamContext'],['../structmlx_1_1core_1_1_stream_context.html#a89d803151e9d7dce29382aa83d5c6ef1',1,'mlx::core::StreamContext::StreamContext()']]], - ['streamordevice_174',['StreamOrDevice',['../namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58',1,'mlx::core']]], - ['streamthread_175',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html',1,'mlx::core::scheduler::StreamThread'],['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#ac528109a11abcb82e6e221c5efa4493c',1,'mlx::core::scheduler::StreamThread::StreamThread()']]], - ['stride_176',['stride',['../classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093',1,'pocketfft::detail::arr_info::stride() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b',1,'pocketfft::detail::arr_info::stride(size_t i) const']]], - ['stride_5fin_177',['stride_in',['../classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c',1,'pocketfft::detail::multi_iter']]], - ['stride_5fout_178',['stride_out',['../classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7',1,'pocketfft::detail::multi_iter']]], - ['stride_5ft_179',['stride_t',['../namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::detail']]], - ['strided_5fdevice_5fidx_180',['strided_device_idx',['../struct_read_writer.html#a4c0b12484aac4fd6759d67c190391989',1,'ReadWriter']]], - ['strided_5freduce_5fgeneral_5fdispatch_181',['strided_reduce_general_dispatch',['../namespacemlx_1_1core.html#aa0332c64ee9965f05026c30a0b778000',1,'mlx::core']]], - ['strided_5fscan_182',['strided_scan',['../kernels_2scan_8h.html#aef54ac73154fb9024eda760a9dfde607',1,'scan.h']]], - ['strided_5fshared_5fidx_183',['strided_shared_idx',['../struct_read_writer.html#ace40adb02cfb33d89c98353327c251fc',1,'ReadWriter']]], - ['strides_184',['strides',['../structmlx_1_1core_1_1_reduction_plan.html#a9bf7cae845ab633247c1811613ece8bd',1,'mlx::core::ReductionPlan::strides'],['../struct_indices.html#a7f73d7652f0f751e6a06c2663e329a4a',1,'Indices::strides'],['../classmlx_1_1core_1_1array.html#a186cf2648da92584d5c1c8b24e69629b',1,'mlx::core::array::strides() const'],['../classmlx_1_1core_1_1array.html#a919f850ca087d1c40aa68f854cb30be2',1,'mlx::core::array::strides(int dim) const']]], - ['submit_185',['submit',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4',1,'pocketfft::detail::threading::thread_pool']]], - ['subtract_186',['Subtract',['../structmlx_1_1core_1_1detail_1_1_subtract.html',1,'mlx::core::detail::Subtract'],['../classmlx_1_1core_1_1_subtract.html',1,'mlx::core::Subtract'],['../struct_subtract.html',1,'Subtract'],['../classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c',1,'mlx::core::Subtract::Subtract()']]], - ['subtract_187',['subtract',['../group__ops.html#ga196c240d3d0fcbb4713802c485e15133',1,'mlx::core']]], - ['sum_188',['Sum',['../struct_sum.html',1,'Sum< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a1fc7c1f09c80650ab0497e2d6781d65f',1,'mlx::core::distributed::AllReduce::Sum'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a8582875544f1d3d396a1a376473ef1dd',1,'mlx::core::Reduce::Sum'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ade23893033e4849f5596e7ce76a5fc36',1,'mlx::core::Scan::Sum'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca14abe2d8818efa71726be4e156813d6f',1,'mlx::core::Scatter::Sum']]], - ['sum_189',['sum',['../group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6',1,'mlx::core::sum(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3627754d7868487bdab1bd83f05d9c81',1,'mlx::core::sum(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4',1,'mlx::core::sum(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d',1,'mlx::core::sum(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['svd_190',['SVD',['../classmlx_1_1core_1_1_s_v_d.html',1,'mlx::core::SVD'],['../classmlx_1_1core_1_1_s_v_d.html#ae89ff583e34fa894cccb8e7a475ee6d1',1,'mlx::core::SVD::SVD()']]], - ['svd_191',['svd',['../namespacemlx_1_1core_1_1linalg.html#a64364b880e99914cf47bf756fa8dbaf0',1,'mlx::core::linalg']]], - ['swapaxes_192',['swapaxes',['../group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb',1,'mlx::core']]], - ['swizzle_193',['swizzle',['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle']]], - ['swizzle_5flog_194',['swizzle_log',['../struct_m_l_x_fast_attention_params.html#a68a338d522ffeb6761b7b168869361e2',1,'MLXFastAttentionParams::swizzle_log'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#ad0713159d4f710cd9a066596593d8840',1,'mlx::steel::ImplicitGemmConv2DParams::swizzle_log'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#af9ff2c06dd8994126634531440325be7',1,'mlx::steel::GEMMParams::swizzle_log']]], - ['synchronize_195',['synchronize',['../namespacemlx_1_1core.html#a14287949d82ffefad0306cef5eb5f9e4',1,'mlx::core::synchronize()'],['../namespacemlx_1_1core.html#a6648a71937b055e5ff513d98056c2fb5',1,'mlx::core::synchronize(Stream)']]] + ['set_5fvector_5fbytes_46',['set_vector_bytes',['../namespacemlx_1_1core.html#a62340bbaa8b216539688a60adcb568bf',1,'mlx::core::set_vector_bytes(CommandEncoder &enc, const std::vector< T > &vec, size_t nelems, int idx)'],['../namespacemlx_1_1core.html#ae309cb543dfb0239cfccc53a8ad0408e',1,'mlx::core::set_vector_bytes(CommandEncoder &enc, const std::vector< T > &vec, int idx)']]], + ['shape_47',['shape',['../structmlx_1_1core_1_1_reduction_plan.html#a6cfa8771fa9caf6fdcc3d74c9fca83ae',1,'mlx::core::ReductionPlan::shape'],['../classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0',1,'pocketfft::detail::arr_info::shape() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8',1,'pocketfft::detail::arr_info::shape(size_t i) const'],['../classmlx_1_1core_1_1array.html#a4a2a2c8a4a5beafd723fc13f2055d55d',1,'mlx::core::array::shape() const'],['../classmlx_1_1core_1_1array.html#a51ed0c45666264dc172d06fba159eb8f',1,'mlx::core::array::shape(int dim) const']]], + ['shape_5ft_48',['shape_t',['../namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1',1,'pocketfft::detail']]], + ['shapes_49',['shapes',['../struct_indices.html#a5ab170f1a77636180889ddfffd4f7d2f',1,'Indices']]], + ['shapes_5fwithout_5freduction_5faxes_50',['shapes_without_reduction_axes',['../namespacemlx_1_1core.html#a44c3ea6db6553c3f6552b9ba64a69494',1,'mlx::core']]], + ['shared_5fbuffer_5fslice_51',['shared_buffer_slice',['../namespacemlx_1_1core.html#a427f2c255dfc6e1f83f97587b08e71bc',1,'mlx::core']]], + ['shp_52',['shp',['../classpocketfft_1_1detail_1_1arr__info.html#a2467e9e01de1ba4d7cd28c1af783da8d',1,'pocketfft::detail::arr_info']]], + ['shutdown_53',['shutdown',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411',1,'pocketfft::detail::threading::thread_pool']]], + ['siblings_54',['siblings',['../classmlx_1_1core_1_1array.html#acf80fde8f743f65ad5b4be69fcb7a74d',1,'mlx::core::array::siblings() const'],['../classmlx_1_1core_1_1array.html#a7263f23e70a580a9bc2129fbcde36e6c',1,'mlx::core::array::siblings()']]], + ['sigmoid_55',['Sigmoid',['../structmlx_1_1core_1_1detail_1_1_sigmoid.html',1,'mlx::core::detail::Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html',1,'mlx::core::Sigmoid'],['../struct_sigmoid.html',1,'Sigmoid'],['../classmlx_1_1core_1_1_sigmoid.html#a47eca99113ec19f0eb60b6a0472c592b',1,'mlx::core::Sigmoid::Sigmoid()']]], + ['sigmoid_56',['sigmoid',['../group__ops.html#ga708abf8f79609cd6831db7c38cafac0e',1,'mlx::core']]], + ['sign_57',['Sign',['../structmlx_1_1core_1_1detail_1_1_sign.html',1,'mlx::core::detail::Sign'],['../classmlx_1_1core_1_1_sign.html',1,'mlx::core::Sign'],['../struct_sign.html',1,'Sign'],['../classmlx_1_1core_1_1_sign.html#afe951e50907bc23a601ec5fa9eae5763',1,'mlx::core::Sign::Sign()']]], + ['sign_58',['sign',['../group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa',1,'mlx::core']]], + ['signal_59',['signal',['../classmlx_1_1core_1_1_event.html#a65a858445506a61be5889ae0e3651b89',1,'mlx::core::Event']]], + ['signaling_5fnan_60',['signaling_NaN',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ad1f76a43c7d51a3765174aa6e0dd9f80',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['signedinteger_61',['signedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2daed58b4631ff157bec9e35ed1182d2c10',1,'mlx::core::Dtype::signedinteger'],['../namespacemlx_1_1core.html#a24e1618af591d737d73729665e868001',1,'mlx::core::signedinteger']]], + ['simd_5fbroadcast_62',['simd_broadcast',['../namespacemetal.html#a498f1e85107eb5f01ba4435977f8efe0',1,'metal']]], + ['simd_5fexclusive_5fscan_63',['simd_exclusive_scan',['../struct_cum_sum.html#a64d1933942de4df2d44dd90e9e5e4460',1,'CumSum::simd_exclusive_scan()'],['../struct_cum_prod.html#a5b0577295853a48856b84fd4f935dff8',1,'CumProd::simd_exclusive_scan()'],['../struct_cum_prod_3_01bool_01_4.html#a1a86e9398bae24182b7be0a6577bf223',1,'CumProd< bool >::simd_exclusive_scan()'],['../struct_cum_max.html#ae11b67aa6c998e9a01615b2a79af4403',1,'CumMax::simd_exclusive_scan()'],['../struct_cum_min.html#a83e65017ff33018b585c043fb803773b',1,'CumMin::simd_exclusive_scan()']]], + ['simd_5fmax_64',['simd_max',['../namespacemetal.html#a048cad0aca52cb737ebf103e76bd1c49',1,'metal']]], + ['simd_5fmin_65',['simd_min',['../namespacemetal.html#ae9e2a23e00724ba2d7868bc4112b386b',1,'metal']]], + ['simd_5fprefix_5fexclusive_5fproduct_66',['simd_prefix_exclusive_product',['../namespacemetal.html#a5ca40242390b632f737e29636829b2e4',1,'metal']]], + ['simd_5fprefix_5fexclusive_5fsum_67',['simd_prefix_exclusive_sum',['../namespacemetal.html#abfbb70c7471f28bf7ff36a612ad014b2',1,'metal']]], + ['simd_5fprefix_5finclusive_5fproduct_68',['simd_prefix_inclusive_product',['../namespacemetal.html#a6ca6a7e1996228fa536e969e9e45c446',1,'metal']]], + ['simd_5fprefix_5finclusive_5fsum_69',['simd_prefix_inclusive_sum',['../namespacemetal.html#a567acb18199ac0107712eb8cb8aeb8e9',1,'metal']]], + ['simd_5fproduct_70',['simd_product',['../namespacemetal.html#ac6e883a04e2265a9790d7db76059e1b4',1,'metal']]], + ['simd_5freduce_71',['simd_reduce',['../struct_and.html#ab4df5517e050cfa7257c34c6b0bd980b',1,'And::simd_reduce()'],['../struct_or.html#aef6b5abbbf3fd22981b8d27562022605',1,'Or::simd_reduce()'],['../struct_sum.html#acf10e0604098391526da3b8271928b5c',1,'Sum::simd_reduce()'],['../struct_prod.html#ab0d7a38360b971b4d99c4ad2db310069',1,'Prod::simd_reduce()'],['../struct_min.html#afe8d7cefac624fd4267c1b252ac38528',1,'Min::simd_reduce()'],['../struct_max.html#a4f74440c6f1ce8247d5a87c2a8882080',1,'Max::simd_reduce()']]], + ['simd_5fscan_72',['simd_scan',['../struct_cum_sum.html#a83342af06f6a700067e3b451636852d6',1,'CumSum::simd_scan()'],['../struct_cum_prod.html#a860dbf6146de088d07175bd165b74004',1,'CumProd::simd_scan()'],['../struct_cum_prod_3_01bool_01_4.html#abeb5ec4237b330e7219f4e881cf10d7a',1,'CumProd< bool >::simd_scan()'],['../struct_cum_max.html#adc9ec8bb09b4433d4c2f03022c43d781',1,'CumMax::simd_scan()'],['../struct_cum_min.html#a0a1005d91b1c90e90e2c6dbd6c296649',1,'CumMin::simd_scan()']]], + ['simd_5fshuffle_73',['simd_shuffle',['../namespacemetal.html#a259ed115bc3c58f88eb35830916b26d4',1,'metal']]], + ['simd_5fshuffle_5fand_5ffill_5fdown_74',['simd_shuffle_and_fill_down',['../namespacemetal.html#ae29a06f0eac636ad7af21dea5b04938b',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a0ee6239fa29a5f9ee0201e0dc5ddc8e0',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], + ['simd_5fshuffle_5fand_5ffill_5fup_75',['simd_shuffle_and_fill_up',['../namespacemetal.html#a1ca14116bf50639b214d8414b5bbaaa6',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a5138d5cdc18139e135707916a243cd8e',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], + ['simd_5fshuffle_5fdown_76',['simd_shuffle_down',['../namespacemetal.html#af6e2dd7ae087aba6abac4f0350b7611c',1,'metal::simd_shuffle_down()'],['../backend_2metal_2kernels_2utils_8h.html#aba6279624b1d30c525efee856a222b5c',1,'simd_shuffle_down(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0c1e4d782fcc56e1ab5565cef12430dd',1,'simd_shuffle_down(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a48ae83a8caf5c74810df60b6c6cdb062',1,'simd_shuffle_down(bool data, uint16_t delta): utils.h']]], + ['simd_5fshuffle_5frotate_5fdown_77',['simd_shuffle_rotate_down',['../namespacemetal.html#a4bb203647a421032db47e73cd649841b',1,'metal']]], + ['simd_5fshuffle_5frotate_5fup_78',['simd_shuffle_rotate_up',['../namespacemetal.html#a729b22077d6c944491a6027c18ea80c9',1,'metal']]], + ['simd_5fshuffle_5fup_79',['simd_shuffle_up',['../namespacemetal.html#afe81c5fbde3f4890458b081909242c55',1,'metal']]], + ['simd_5fshuffle_5fxor_80',['simd_shuffle_xor',['../namespacemetal.html#a5017efc9605e069cfb507137cd1a1852',1,'metal']]], + ['simd_5fsize_81',['simd_size',['../backend_2metal_2kernels_2reduction_2ops_8h.html#a515b75d563a93d3c09ee677948dc83e3',1,'ops.h']]], + ['simd_5fsize_82',['SIMD_SIZE',['../quantized_8h.html#a62969a218d93680f5e35d0c61b160b99',1,'quantized.h']]], + ['simd_5fstride_5fa_83',['simd_stride_a',['../structmlx_1_1steel_1_1_block_m_m_a.html#ab52d2644053c5bb28ab9b24fab8b194a',1,'mlx::steel::BlockMMA']]], + ['simd_5fstride_5fb_84',['simd_stride_b',['../structmlx_1_1steel_1_1_block_m_m_a.html#af5c60b04602f72b705192911ed8749da',1,'mlx::steel::BlockMMA']]], + ['simd_5fsum_85',['simd_sum',['../namespacemetal.html#a85181e37a00cb4a4217f1bb25389bce5',1,'metal']]], + ['simd_5fxor_86',['simd_xor',['../namespacemetal.html#a1308decbf2d5c33d34d6be523ea1c30f',1,'metal']]], + ['simple_5fiter_87',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html',1,'pocketfft::detail::simple_iter'],['../classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398',1,'pocketfft::detail::simple_iter::simple_iter()']]], + ['simplevalueandgradfn_88',['SimpleValueAndGradFn',['../namespacemlx_1_1core.html#a2689b8f1181648cb1685204fea9f3066',1,'mlx::core']]], + ['sin_89',['Sin',['../structmlx_1_1core_1_1detail_1_1_sin.html',1,'mlx::core::detail::Sin'],['../classmlx_1_1core_1_1_sin.html',1,'mlx::core::Sin'],['../struct_sin.html',1,'Sin'],['../classmlx_1_1core_1_1_sin.html#a10d1ecc0ca96e79cdf55b57073d126ea',1,'mlx::core::Sin::Sin()']]], + ['sin_90',['sin',['../namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258',1,'pocketfft::detail::sin()'],['../namespacemetal.html#a619a159ca5f2ddfe3647d3a6bb6e804c',1,'metal::sin()'],['../namespacemetal_1_1fast.html#a3af771cfe7a135104f9d063147dba270',1,'metal::fast::sin()'],['../namespacemetal_1_1precise.html#a71acf77ffd29c56f56afae0195c98a1c',1,'metal::precise::sin()'],['../group__ops.html#gaebf0a73ad3732fba39df37826c235692',1,'mlx::core::sin()']]], + ['sincos_5f2pibyn_91',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html',1,'pocketfft::detail::sincos_2pibyn< T >'],['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40',1,'pocketfft::detail::sincos_2pibyn::sincos_2pibyn()']]], + ['sinh_92',['Sinh',['../structmlx_1_1core_1_1detail_1_1_sinh.html',1,'mlx::core::detail::Sinh'],['../classmlx_1_1core_1_1_sinh.html',1,'mlx::core::Sinh'],['../struct_sinh.html',1,'Sinh'],['../classmlx_1_1core_1_1_sinh.html#a4a4f6814d403c2ce5d6c574b0dca3c96',1,'mlx::core::Sinh::Sinh()']]], + ['sinh_93',['sinh',['../namespacemetal.html#a83ba4235ae350ab8880a9df09158620b',1,'metal::sinh()'],['../namespacemetal_1_1fast.html#a990d90b3440e38d1fb4ff5065c6c189b',1,'metal::fast::sinh()'],['../namespacemetal_1_1precise.html#abc8f4f59dd6e7204ab5d84f0af96331c',1,'metal::precise::sinh()'],['../group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7',1,'mlx::core::sinh()']]], + ['sinpi_94',['sinpi',['../namespacemetal.html#ae9655f7fa2ba6c0625ca25fbb278e269',1,'metal::sinpi()'],['../namespacemetal_1_1fast.html#ab07a32fe544aa304577d29e0251e87b2',1,'metal::fast::sinpi()'],['../namespacemetal_1_1precise.html#a78b17dab93519d9c82c2575dafec49c9',1,'metal::precise::sinpi()']]], + ['size_95',['size',['../structmlx_1_1core_1_1_dtype.html#adc24e03df52237e9ba03e2451f321b66',1,'mlx::core::Dtype::size'],['../classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952',1,'pocketfft::detail::arr::size()'],['../classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5',1,'pocketfft::detail::arr_info::size()'],['../classmlx_1_1core_1_1array.html#a598f87161926d9e0b516860f0ea2c8f6',1,'mlx::core::array::size()'],['../structmlx_1_1core_1_1distributed_1_1_group.html#abd96a09217e3d1bcc522888257d22cef',1,'mlx::core::distributed::Group::size()']]], + ['size_5fof_96',['size_of',['../namespacemlx_1_1core.html#add4794cc0ffe5d717fc146084a235d95',1,'mlx::core']]], + ['slice_97',['Slice',['../classmlx_1_1core_1_1_slice.html',1,'mlx::core::Slice'],['../classmlx_1_1core_1_1_slice.html#a8a38feb7bb6b72bdeebb83f053e2fd7f',1,'mlx::core::Slice::Slice()']]], + ['slice_98',['slice',['../group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21',1,'mlx::core::slice(const array &a, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})'],['../group__ops.html#ga90b243e25b5fbbd7a9732f7ac48f6403',1,'mlx::core::slice(const array &a, const std::vector< int > &start, const std::vector< int > &stop, StreamOrDevice s={})']]], + ['slice_5fgpu_99',['slice_gpu',['../namespacemlx_1_1core.html#a5ab133d67da9c16621f61690e27fccfd',1,'mlx::core']]], + ['slice_5fupdate_100',['slice_update',['../group__ops.html#ga3875660e4ce2c8add8bfcf8144078708',1,'mlx::core::slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})'],['../group__ops.html#ga03ffbbb4d989a463ef43f41ebf7eabef',1,'mlx::core::slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, StreamOrDevice s={})']]], + ['sliceupdate_101',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html',1,'mlx::core::SliceUpdate'],['../classmlx_1_1core_1_1_slice_update.html#aa30a7f22f557c56e1a2b5fcf44488990',1,'mlx::core::SliceUpdate::SliceUpdate()']]], + ['slicing_2eh_102',['slicing.h',['../common_2slicing_8h.html',1,'(Global Namespace)'],['../metal_2slicing_8h.html',1,'(Global Namespace)']]], + ['sm_103',['sm',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa85451edf6900fd6af164d4d50889ae3',1,'mlx::steel::BlockMMA']]], + ['sn_104',['sn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ade420e8b811d597345783c324c23a34a',1,'mlx::steel::BlockMMA']]], + ['softmax_105',['Softmax',['../classmlx_1_1core_1_1_softmax.html',1,'mlx::core::Softmax'],['../classmlx_1_1core_1_1_softmax.html#a4ec686aac4e06f0dfe2cbd6801af40eb',1,'mlx::core::Softmax::Softmax()']]], + ['softmax_106',['softmax',['../namespacemlx_1_1core_1_1metal.html#a4fe937c2c584fd646926057f31d54ca6',1,'mlx::core::metal::softmax()'],['../group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6',1,'mlx::core::softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga1ae3614d07d873892a530d14c3857d0b',1,'mlx::core::softmax(const array &a, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga06f570d73716a24303e6de3aaba4457b',1,'mlx::core::softmax(const array &a, int axis, bool precise=false, StreamOrDevice s={})']]], + ['softmax_2eh_107',['softmax.h',['../jit_2softmax_8h.html',1,'(Global Namespace)'],['../kernels_2softmax_8h.html',1,'(Global Namespace)']]], + ['softmax_5fexp_108',['softmax_exp',['../kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b',1,'softmax.h']]], + ['softmax_5fkernels_109',['softmax_kernels',['../jit_2softmax_8h.html#a1cbfb210a9a765c6620e9f1247ccef12',1,'softmax.h']]], + ['softmax_5flooped_110',['softmax_looped',['../kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276',1,'softmax.h']]], + ['softmax_5fn_5freads_111',['SOFTMAX_N_READS',['../defines_8h.html#a722995df24286b27b7da3d74b73f768d',1,'defines.h']]], + ['softmax_5fsingle_5frow_112',['softmax_single_row',['../kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b',1,'softmax.h']]], + ['sort_113',['Sort',['../classmlx_1_1core_1_1_sort.html',1,'mlx::core::Sort'],['../classmlx_1_1core_1_1_sort.html#a62943032dbd72e85ceb9b4b7211f4a44',1,'mlx::core::Sort::Sort()']]], + ['sort_114',['sort',['../struct_thread_sort.html#ad9ab3e6b47f7e9b91c0f3b773596986d',1,'ThreadSort::sort()'],['../struct_block_merge_sort.html#acc970f5eb963f7f2010f5ae5ea8b8bc0',1,'BlockMergeSort::sort()'],['../namespacemlx_1_1core_1_1metal.html#ab77c9a9ecaeeab8c66b712862777c24b',1,'mlx::core::metal::sort()'],['../group__ops.html#ga7fb616054665b3c2d61fa234f501f079',1,'mlx::core::sort(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaae1bc47aa737f705d0e5884270063fea',1,'mlx::core::sort(const array &a, int axis, StreamOrDevice s={})']]], + ['sort_2eh_115',['sort.h',['../sort_8h.html',1,'']]], + ['special_5fmul_116',['special_mul',['../structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874',1,'pocketfft::detail::cmplx::special_mul()'],['../namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76',1,'pocketfft::detail::special_mul()']]], + ['split_117',['Split',['../classmlx_1_1core_1_1_split.html',1,'mlx::core::Split'],['../classmlx_1_1core_1_1_split.html#a897c746ecfdff5119cc5ae3f20499385',1,'mlx::core::Split::Split()']]], + ['split_118',['split',['../structmlx_1_1core_1_1distributed_1_1_group.html#abbf40f8979488806bc5bca9ecc4130e9',1,'mlx::core::distributed::Group::split()'],['../group__ops.html#ga7534290bceab5fb3831a05d67bebce7d',1,'mlx::core::split(const array &a, int num_splits, int axis, StreamOrDevice s={})'],['../group__ops.html#ga56882d24e5fde59c266774624c892d41',1,'mlx::core::split(const array &a, int num_splits, StreamOrDevice s={})'],['../group__ops.html#ga2cfcb1a53924882e30476c9016c5de74',1,'mlx::core::split(const array &a, const std::vector< int > &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#gac324dfa3e26d3a14a35ab7962e36f0e1',1,'mlx::core::split(const array &a, const std::vector< int > &indices, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a42847b435d037a977592e355eed072af',1,'mlx::core::random::split(const array &key, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7ec057064c7326c41b536f08178861e5',1,'mlx::core::random::split(const array &key, int num, StreamOrDevice s={})']]], + ['split_5fk_5fpartition_5fsize_119',['split_k_partition_size',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a9f5a67b2343645b570e109c3837d4042',1,'mlx::steel::GEMMSpiltKParams']]], + ['split_5fk_5fpartition_5fstride_120',['split_k_partition_stride',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a12144ce89d404812cd862611d770b9fb',1,'mlx::steel::GEMMSpiltKParams']]], + ['split_5fk_5fpartitions_121',['split_k_partitions',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#ae06c27116905d4ff3b9b436e588a93fd',1,'mlx::steel::GEMMSpiltKParams']]], + ['sqrt_122',['Sqrt',['../structmlx_1_1core_1_1detail_1_1_sqrt.html',1,'mlx::core::detail::Sqrt'],['../classmlx_1_1core_1_1_sqrt.html',1,'mlx::core::Sqrt'],['../struct_sqrt.html',1,'Sqrt'],['../classmlx_1_1core_1_1_sqrt.html#a6682a7c31ca427c9d2c5ddb6a479bf29',1,'mlx::core::Sqrt::Sqrt()']]], + ['sqrt_123',['sqrt',['../namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3',1,'pocketfft::detail::sqrt()'],['../namespacemetal.html#ab3f4d4852ca0e591104fbd8e5b50d31b',1,'metal::sqrt()'],['../namespacemetal_1_1fast.html#a4218a85c7d8a74cb8055b4755205627e',1,'metal::fast::sqrt()'],['../namespacemetal_1_1precise.html#acb213467361cd2cab93a8d5ea1aa5bfd',1,'metal::precise::sqrt()'],['../group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1',1,'mlx::core::sqrt()']]], + ['square_124',['Square',['../structmlx_1_1core_1_1detail_1_1_square.html',1,'mlx::core::detail::Square'],['../classmlx_1_1core_1_1_square.html',1,'mlx::core::Square'],['../struct_square.html',1,'Square'],['../classmlx_1_1core_1_1_square.html#ab94e28d5c92e6febc1c74e525f730dc4',1,'mlx::core::Square::Square()']]], + ['square_125',['square',['../group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e',1,'mlx::core']]], + ['squeeze_126',['squeeze',['../group__ops.html#ga710daa7ec721bd4d3f326082cb195576',1,'mlx::core::squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3',1,'mlx::core::squeeze(const array &a, int axis, StreamOrDevice s={})'],['../group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad',1,'mlx::core::squeeze(const array &a, StreamOrDevice s={})']]], + ['src_127',['src',['../struct_quantized_block_loader.html#ad85c6b7e07c81307b3b91eb4dd7be30b',1,'QuantizedBlockLoader::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a656a46ee27486482b45ff90b3d626255',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a57552e9cfbafad71d47b2f3a8e027bdf',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7bfbcc4a1e3eef7aef5dd8e8c374a95f',1,'mlx::steel::Conv2DWeightBlockLoader::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#afe21e46e08523232830c25eb1b4ade16',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a8b04a69952404a04029dacc424df6e8f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1ee2922961b5fcb1db577928c4d9d731',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::src'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a50f458dbb74d61be2ed24727d8d43614',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src'],['../structmlx_1_1steel_1_1_block_loader.html#a13004952d0bf2030b95acb621a3779dd',1,'mlx::steel::BlockLoader::src']]], + ['src_5fld_128',['src_ld',['../struct_quantized_block_loader.html#a8050977d473d1a24fae5c833e609839e',1,'QuantizedBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7464ec687323fa79050702952ed9084f',1,'mlx::steel::Conv2DWeightBlockLoader::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#aa6bedc0cbb447eaf70c03f2e26df2cb2',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::src_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a6918c1df7712c4e408e2871467ea7987',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::src_ld'],['../structmlx_1_1steel_1_1_block_loader.html#aadafc50f7f06af434149d7469df4714d',1,'mlx::steel::BlockLoader::src_ld']]], + ['stack_129',['stack',['../group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032',1,'mlx::core::stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga82216209dce901296fc737fe8efa5c94',1,'mlx::core::stack(const std::vector< array > &arrays, StreamOrDevice s={})']]], + ['start_5fcapture_130',['start_capture',['../namespacemlx_1_1core_1_1metal.html#aa47cb5651bf3b65c46ab216b7e504d77',1,'mlx::core::metal']]], + ['start_5fconcurrent_131',['start_concurrent',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::metal::CommandEncoder']]], + ['start_5frow_132',['start_row',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a220e033b689c8d6a6f319dae02b38334',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], + ['status_133',['Status',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078',1,'mlx::core::array']]], + ['status_134',['status',['../classmlx_1_1core_1_1array.html#a7102659be87e9ef62966696ab9b07dad',1,'mlx::core::array']]], + ['std_135',['std',['../group__ops.html#ga2a466024f8061febc0a64be557644cb0',1,'mlx::core::std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#gafdcb04d77c64405a3990078a77dd984c',1,'mlx::core::std(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8',1,'mlx::core::std(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga62721a206df671ef5797449eea97af9f',1,'mlx::core::std(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], + ['steel_5fconst_136',['STEEL_CONST',['../steel_2defines_8h.html#a90b91c866313ffa46eff6d9cc944ad2b',1,'defines.h']]], + ['steel_5fconv_137',['steel_conv',['../namespacemlx_1_1core_1_1metal.html#a92f1e559b1121d545746f81ff86eaca1',1,'mlx::core::metal']]], + ['steel_5fconv_2eh_138',['steel_conv.h',['../jit_2steel__conv_8h.html',1,'(Global Namespace)'],['../kernels_2steel_2conv_2kernels_2steel__conv_8h.html',1,'(Global Namespace)']]], + ['steel_5fconv_5fgeneral_139',['steel_conv_general',['../namespacemlx_1_1core_1_1metal.html#a02edb6a90bdf30f4c9f0d6c25b0267b5',1,'mlx::core::metal']]], + ['steel_5fconv_5fgeneral_2eh_140',['steel_conv_general.h',['../steel__conv__general_8h.html',1,'']]], + ['steel_5fconv_5fgeneral_5fkernels_141',['steel_conv_general_kernels',['../jit_2steel__conv_8h.html#ae4ca1720029316b08ea92b7662347d47',1,'steel_conv.h']]], + ['steel_5fconv_5fkernels_142',['steel_conv_kernels',['../jit_2steel__conv_8h.html#a386d79077465df56659416fd84adb513',1,'steel_conv.h']]], + ['steel_5fgemm_2eh_143',['steel_gemm.h',['../steel__gemm_8h.html',1,'']]], + ['steel_5fgemm_5ffused_144',['steel_gemm_fused',['../namespacemlx_1_1core_1_1metal.html#a17764366deed71c160fb26091400a803',1,'mlx::core::metal']]], + ['steel_5fgemm_5ffused_2eh_145',['steel_gemm_fused.h',['../steel__gemm__fused_8h.html',1,'']]], + ['steel_5fgemm_5ffused_5fkernels_146',['steel_gemm_fused_kernels',['../steel__gemm_8h.html#a4c6009fd5357b730805f2fd4ba6e093e',1,'steel_gemm.h']]], + ['steel_5fgemm_5fmasked_147',['steel_gemm_masked',['../namespacemlx_1_1core_1_1metal.html#a962272ca73d26c08f76f706a128fd71f',1,'mlx::core::metal']]], + ['steel_5fgemm_5fmasked_2eh_148',['steel_gemm_masked.h',['../steel__gemm__masked_8h.html',1,'']]], + ['steel_5fgemm_5fmasked_5fkernels_149',['steel_gemm_masked_kernels',['../steel__gemm_8h.html#a62a358fd3ec5365081920d07aceb581c',1,'steel_gemm.h']]], + ['steel_5fgemm_5fsplitk_150',['steel_gemm_splitk',['../namespacemlx_1_1core_1_1metal.html#ad0dfd40ba7c09755711ceb731e57a5ac',1,'mlx::core::metal']]], + ['steel_5fgemm_5fsplitk_2eh_151',['steel_gemm_splitk.h',['../steel__gemm__splitk_8h.html',1,'']]], + ['steel_5fgemm_5fsplitk_5faccum_5faxbpy_5fkernels_152',['steel_gemm_splitk_accum_axbpy_kernels',['../steel__gemm_8h.html#a40a86e1381c241aba8511e51a981a4bf',1,'steel_gemm.h']]], + ['steel_5fgemm_5fsplitk_5faccum_5fkernels_153',['steel_gemm_splitk_accum_kernels',['../steel__gemm_8h.html#a144a64b8d94f0371fb144e2cc308fcf9',1,'steel_gemm.h']]], + ['steel_5fgemm_5fsplitk_5fkernels_154',['steel_gemm_splitk_kernels',['../steel__gemm_8h.html#a92108ab01d826e38bca83d8569b947d9',1,'steel_gemm.h']]], + ['steel_5fmatmul_155',['steel_matmul',['../namespacemlx_1_1core.html#ab43a7633794498e1c6775cca829eb886',1,'mlx::core']]], + ['steel_5fmatmul_5fconv_5fgroups_156',['steel_matmul_conv_groups',['../namespacemlx_1_1core.html#afe8386cea0c7b846dc78649927fd0c75',1,'mlx::core']]], + ['steel_5fpragma_5funroll_157',['STEEL_PRAGMA_UNROLL',['../steel_2defines_8h.html#a5a5c3095b132a7589bc19cd5cb80e2c6',1,'defines.h']]], + ['stop_158',['stop',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a456ad1c0c9e731833a2f8411c4ed51aa',1,'mlx::core::scheduler::StreamThread']]], + ['stop_5fcapture_159',['stop_capture',['../namespacemlx_1_1core_1_1metal.html#ac90714424e36fb01e04550de69b8314f',1,'mlx::core::metal']]], + ['stop_5fgradient_160',['stop_gradient',['../group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f',1,'mlx::core']]], + ['stopgradient_161',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html',1,'mlx::core::StopGradient'],['../classmlx_1_1core_1_1_stop_gradient.html#ac70d1ab819d04e00f76bc25aeebaf84f',1,'mlx::core::StopGradient::StopGradient()']]], + ['store_5fresult_162',['store_result',['../structmlx_1_1steel_1_1_block_m_m_a.html#a85c57ad4092079f068683cb525fb08ee',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const']]], + ['store_5fresult_5fsafe_163',['store_result_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#af94bafeb285c240051e86418fccb89fe',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const']]], + ['str_164',['str',['../classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4',1,'pocketfft::detail::arr_info::str'],['../struct_m_l_x_conv_params.html#a862191e8ab1bc8a47aa1396b36d46058',1,'MLXConvParams::str']]], + ['stream_165',['Stream',['../structmlx_1_1core_1_1_stream.html',1,'mlx::core']]], + ['stream_166',['stream',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a8462e4acffcd385c6248bd7102e6bcb1',1,'mlx::core::scheduler::StreamThread::stream'],['../classmlx_1_1core_1_1_event.html#a193143bad31b68c699fa27f135b45614',1,'mlx::core::Event::stream()'],['../classmlx_1_1core_1_1_primitive.html#a46e6257397a662528f9f831842ac456a',1,'mlx::core::Primitive::stream()']]], + ['stream_167',['Stream',['../structmlx_1_1core_1_1_stream.html#a7f0815ff4886da74cbbff5f93d82dd3e',1,'mlx::core::Stream']]], + ['stream_2eh_168',['stream.h',['../stream_8h.html',1,'']]], + ['streamcontext_169',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html',1,'mlx::core::StreamContext'],['../structmlx_1_1core_1_1_stream_context.html#a89d803151e9d7dce29382aa83d5c6ef1',1,'mlx::core::StreamContext::StreamContext()']]], + ['streamordevice_170',['StreamOrDevice',['../namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58',1,'mlx::core']]], + ['streamthread_171',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html',1,'mlx::core::scheduler::StreamThread'],['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#ac528109a11abcb82e6e221c5efa4493c',1,'mlx::core::scheduler::StreamThread::StreamThread()']]], + ['stride_172',['stride',['../classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093',1,'pocketfft::detail::arr_info::stride() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b',1,'pocketfft::detail::arr_info::stride(size_t i) const']]], + ['stride_5fin_173',['stride_in',['../classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c',1,'pocketfft::detail::multi_iter']]], + ['stride_5fout_174',['stride_out',['../classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7',1,'pocketfft::detail::multi_iter']]], + ['stride_5ft_175',['stride_t',['../namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84',1,'pocketfft::detail']]], + ['strided_5fdevice_5fidx_176',['strided_device_idx',['../struct_read_writer.html#a4c0b12484aac4fd6759d67c190391989',1,'ReadWriter']]], + ['strided_5freduce_5fgeneral_5fdispatch_177',['strided_reduce_general_dispatch',['../namespacemlx_1_1core.html#aa0332c64ee9965f05026c30a0b778000',1,'mlx::core']]], + ['strided_5fscan_178',['strided_scan',['../kernels_2scan_8h.html#aef54ac73154fb9024eda760a9dfde607',1,'scan.h']]], + ['strided_5fshared_5fidx_179',['strided_shared_idx',['../struct_read_writer.html#ace40adb02cfb33d89c98353327c251fc',1,'ReadWriter']]], + ['strides_180',['strides',['../structmlx_1_1core_1_1_reduction_plan.html#a9bf7cae845ab633247c1811613ece8bd',1,'mlx::core::ReductionPlan::strides'],['../struct_indices.html#a7f73d7652f0f751e6a06c2663e329a4a',1,'Indices::strides'],['../classmlx_1_1core_1_1array.html#a186cf2648da92584d5c1c8b24e69629b',1,'mlx::core::array::strides() const'],['../classmlx_1_1core_1_1array.html#a919f850ca087d1c40aa68f854cb30be2',1,'mlx::core::array::strides(int dim) const']]], + ['submit_181',['submit',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4',1,'pocketfft::detail::threading::thread_pool']]], + ['subtract_182',['Subtract',['../structmlx_1_1core_1_1detail_1_1_subtract.html',1,'mlx::core::detail::Subtract'],['../classmlx_1_1core_1_1_subtract.html',1,'mlx::core::Subtract'],['../struct_subtract.html',1,'Subtract'],['../classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c',1,'mlx::core::Subtract::Subtract()']]], + ['subtract_183',['subtract',['../group__ops.html#ga196c240d3d0fcbb4713802c485e15133',1,'mlx::core']]], + ['sum_184',['Sum',['../struct_sum.html',1,'Sum< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a1fc7c1f09c80650ab0497e2d6781d65f',1,'mlx::core::distributed::AllReduce::Sum'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a8582875544f1d3d396a1a376473ef1dd',1,'mlx::core::Reduce::Sum'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ade23893033e4849f5596e7ce76a5fc36',1,'mlx::core::Scan::Sum'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca14abe2d8818efa71726be4e156813d6f',1,'mlx::core::Scatter::Sum']]], + ['sum_185',['sum',['../group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6',1,'mlx::core::sum(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3627754d7868487bdab1bd83f05d9c81',1,'mlx::core::sum(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4',1,'mlx::core::sum(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d',1,'mlx::core::sum(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['svd_186',['SVD',['../classmlx_1_1core_1_1_s_v_d.html',1,'mlx::core::SVD'],['../classmlx_1_1core_1_1_s_v_d.html#ae89ff583e34fa894cccb8e7a475ee6d1',1,'mlx::core::SVD::SVD()']]], + ['svd_187',['svd',['../namespacemlx_1_1core_1_1linalg.html#a64364b880e99914cf47bf756fa8dbaf0',1,'mlx::core::linalg']]], + ['swapaxes_188',['swapaxes',['../group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb',1,'mlx::core']]], + ['swizzle_189',['swizzle',['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle']]], + ['swizzle_5flog_190',['swizzle_log',['../struct_m_l_x_fast_attention_params.html#a68a338d522ffeb6761b7b168869361e2',1,'MLXFastAttentionParams::swizzle_log'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#ad0713159d4f710cd9a066596593d8840',1,'mlx::steel::ImplicitGemmConv2DParams::swizzle_log'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#af9ff2c06dd8994126634531440325be7',1,'mlx::steel::GEMMParams::swizzle_log']]], + ['synchronize_191',['synchronize',['../namespacemlx_1_1core.html#a14287949d82ffefad0306cef5eb5f9e4',1,'mlx::core::synchronize()'],['../namespacemlx_1_1core.html#a6648a71937b055e5ff513d98056c2fb5',1,'mlx::core::synchronize(Stream)']]] ]; diff --git a/docs/build/html/search/all_14.js b/docs/build/html/search/all_14.js index cde4c71e78..528e5a333b 100644 --- a/docs/build/html/search/all_14.js +++ b/docs/build/html/search/all_14.js @@ -27,56 +27,61 @@ var searchData= ['ternary_5fops_24',['ternary_ops',['../namespacemlx_1_1core_1_1metal.html#a11b593b07e9a33e5f78fe4695fb99ec9',1,'mlx::core::metal']]], ['ternary_5fops_2eh_25',['ternary_ops.h',['../ternary__ops_8h.html',1,'']]], ['ternary_5fv_26',['ternary_v',['../metal_2kernels_2ternary_8h.html#a83f93644d21ee774e06e8190d0725ccb',1,'ternary.h']]], - ['tgp_5fmem_5fsize_27',['tgp_mem_size',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a1ec583584e69dcbbb72106390a4fc5da',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fmem_5fsize_5fa_28',['tgp_mem_size_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ac00b149d76a903c2f91b0f477dc5037f',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fmem_5fsize_5fb_29',['tgp_mem_size_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a105af1069668028c6f1bc6d6dd162298',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fpadding_5fa_30',['tgp_padding_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad547704ccbff6c2076abeffa6628c5a0',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fpadding_5fb_31',['tgp_padding_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad1b03941e869017558423c08b08bc094',1,'mlx::steel::GEMMKernel']]], - ['tgp_5fsize_32',['tgp_size',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a9058ddb73e30e83fb9c548ba22817d64',1,'mlx::steel::GEMMKernel']]], - ['thread_33',['thread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a449de02bf2ac80d8fe2f208fa7eac359',1,'mlx::core::scheduler::StreamThread']]], - ['thread_5fcount_34',['thread_count',['../structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87',1,'pocketfft::detail::util']]], - ['thread_5ffn_35',['thread_fn',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a06a62c21c1174e4eb4d242e50aad7adf',1,'mlx::core::scheduler::StreamThread']]], - ['thread_5fid_36',['thread_id',['../namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b',1,'pocketfft::detail::threading']]], - ['thread_5fidx_37',['thread_idx',['../struct_quantized_block_loader.html#a50821537ea747bc03295a09bb0eef475',1,'QuantizedBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a70da26a715135d973f88371a70255be9',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ac18de37cde1459595bfe18b0d5ef146d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ab1cb2ade639787243e0325dcd3dc0a11',1,'mlx::steel::Conv2DWeightBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9642399b8066e29123524f36ebc7b482',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acacdac168004c87fee27c8554ac905a7',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a401f0c7cf1588552556603c7ffba2316',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08a517bc50caf41155b98be0690bfe44',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::thread_idx'],['../structmlx_1_1steel_1_1_block_loader.html#a064e2cc77e0b1cf0f8027929e031775b',1,'mlx::steel::BlockLoader::thread_idx']]], - ['thread_5fmap_38',['thread_map',['../namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e',1,'pocketfft::detail::threading']]], - ['thread_5fpool_39',['thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html',1,'pocketfft::detail::threading::thread_pool'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560',1,'pocketfft::detail::threading::thread_pool::thread_pool(size_t nthreads)'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0',1,'pocketfft::detail::threading::thread_pool::thread_pool()']]], - ['thread_5fsort_5ft_40',['thread_sort_t',['../struct_block_merge_sort.html#ad2474d16721f4ceb954125728a0e2ea2',1,'BlockMergeSort']]], - ['thread_5fswap_41',['thread_swap',['../sort_8h.html#a6e8c2da4975a8001fd5ddf211a3058b7',1,'sort.h']]], - ['threads_5fper_5ftg_42',['threads_per_tg',['../struct_read_writer.html#a64c58e358da22358df3075448ea23893',1,'ReadWriter']]], - ['threadsort_43',['ThreadSort',['../struct_thread_sort.html',1,'']]], - ['threefry_2eh_44',['threefry.h',['../threefry_8h.html',1,'']]], - ['threefry2x32_5fhash_45',['threefry2x32_hash',['../namespacemlx_1_1core_1_1random.html#ac7e92c89a2bac1b0bed922a3d4c3c66b',1,'mlx::core::random']]], - ['tile_46',['tile',['../group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361',1,'mlx::core']]], - ['tile_5fstride_47',['tile_stride',['../struct_quantized_block_loader.html#ac3f651c1a645291d1037a2cc8ded2320',1,'QuantizedBlockLoader::tile_stride'],['../structmlx_1_1steel_1_1_block_loader.html#ab87876699d55473620c7ea99f9da911d',1,'mlx::steel::BlockLoader::tile_stride']]], - ['tile_5fstride_5fa_48',['tile_stride_a',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8fddaa78913cdc8eea5e1cf7d2776330',1,'mlx::steel::BlockMMA']]], - ['tile_5fstride_5fb_49',['tile_stride_b',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae3f35453b3afbaac9df64ad5966b34a4',1,'mlx::steel::BlockMMA']]], - ['tiles_5fm_50',['tiles_m',['../struct_m_l_x_fast_attention_params.html#a0df159c839fc27b9426b8ac4336cc0ad',1,'MLXFastAttentionParams::tiles_m'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a4c5e33edf70be99cf93ac5723c12eb24',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#ad23a5a7f74cd5859741a36e4bc7823ca',1,'mlx::steel::GEMMParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a0970989624e17088d5326c2e198cb95b',1,'mlx::steel::GEMMSpiltKParams::tiles_m']]], - ['tiles_5fn_51',['tiles_n',['../struct_m_l_x_fast_attention_params.html#a608aa256216ac6d80af00209303d2029',1,'MLXFastAttentionParams::tiles_n'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a76f9f381e7187a993d65128b9b681b2d',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0e6b8b629232f1b43fbce9a395174bed',1,'mlx::steel::GEMMParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a5b46dfb9cee3606efa05d217349a20a6',1,'mlx::steel::GEMMSpiltKParams::tiles_n']]], - ['tm_52',['tm',['../structmlx_1_1steel_1_1_block_m_m_a.html#ab84d70540d09ce115794f343849f016f',1,'mlx::steel::BlockMMA']]], - ['tm_53',['TM',['../structmlx_1_1steel_1_1_block_m_m_a.html#aba5f749fdf32d8bd9d9e29f2a9ae4591',1,'mlx::steel::BlockMMA']]], - ['tm_5fstride_54',['TM_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a5b0029866f493363942133b55bff7307',1,'mlx::steel::BlockMMA']]], - ['tn_55',['TN',['../structmlx_1_1steel_1_1_block_m_m_a.html#a706ae779c1f8d2eb18f19c248567d424',1,'mlx::steel::BlockMMA']]], - ['tn_56',['tn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ad7a2033983cfbb474c50c2878057d8f1',1,'mlx::steel::BlockMMA']]], - ['tn_5fstride_57',['TN_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8b3690b383afd26563efb38f9c375e50',1,'mlx::steel::BlockMMA']]], - ['to_5fbnns_5fdtype_58',['to_bnns_dtype',['../namespacemlx_1_1core.html#a7a4193f37b1de9c33c31d1da09c77edb',1,'mlx::core']]], - ['to_5fstream_59',['to_stream',['../namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9',1,'mlx::core']]], - ['topk_60',['topk',['../group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a',1,'mlx::core::topk(const array &a, int k, StreamOrDevice s={})'],['../group__ops.html#ga35b8436c79ff953f6c809598b646f498',1,'mlx::core::topk(const array &a, int k, int axis, StreamOrDevice s={})']]], - ['trace_61',['trace',['../group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, StreamOrDevice s={})'],['../group__ops.html#gaf25c00108feaafaa6350a4434cb0062e',1,'mlx::core::trace(const array &a, StreamOrDevice s={})']]], - ['transformadd_62',['TransformAdd',['../structmlx_1_1steel_1_1_transform_add.html',1,'mlx::steel::TransformAdd< OutT, InT >'],['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd()']]], - ['transformaxpby_63',['TransformAxpby',['../structmlx_1_1steel_1_1_transform_axpby.html',1,'mlx::steel::TransformAxpby< OutT, InT >'],['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby()']]], - ['transformnone_64',['TransformNone',['../structmlx_1_1steel_1_1_transform_none.html',1,'mlx::steel']]], - ['transforms_2eh_65',['transforms.h',['../backend_2metal_2kernels_2steel_2gemm_2transforms_8h.html',1,'(Global Namespace)'],['../transforms_8h.html',1,'(Global Namespace)']]], - ['transforms_5fimpl_2eh_66',['transforms_impl.h',['../transforms__impl_8h.html',1,'']]], - ['transpose_67',['Transpose',['../classmlx_1_1core_1_1_transpose.html',1,'mlx::core::Transpose'],['../classmlx_1_1core_1_1_transpose.html#a1a9ba023584c61c7ac93d6dce536760a',1,'mlx::core::Transpose::Transpose()']]], - ['transpose_68',['transpose',['../group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b',1,'mlx::core::transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5',1,'mlx::core::transpose(const array &a, std::initializer_list< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a',1,'mlx::core::transpose(const array &a, StreamOrDevice s={})']]], - ['tri_69',['tri',['../group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78',1,'mlx::core::tri(int n, int m, int k, Dtype type, StreamOrDevice s={})'],['../group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5',1,'mlx::core::tri(int n, Dtype type, StreamOrDevice s={})']]], - ['tril_70',['tril',['../group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2',1,'mlx::core']]], - ['triu_71',['triu',['../group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412',1,'mlx::core']]], - ['trows_72',['TROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a2528ff5ed472e4ed35415ada42276b07',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3957fb263fe040fe70683fd1d7b06487',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a94f0ce5bb7d87bc1fb6a7c2ba2b892d4',1,'mlx::steel::Conv2DWeightBlockLoader::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a789683f9ac9d9309d07c05f3bdedd2fd',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a5cefb1285ed13ad3490198e9303453de',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a2aff22af70f685f858adea73f5575cf7',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acbc28f364381166faaeec2783dc88e10',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::TROWS']]], - ['trunc_73',['trunc',['../namespacemetal.html#a93cb75a11a362bfc8310ea19c554c887',1,'metal::trunc()'],['../namespacemetal_1_1fast.html#aa62e1075e86c626d97038f16e9433415',1,'metal::fast::trunc()'],['../namespacemetal_1_1precise.html#a334183e7a2dd49b983d072d1e8ee2b27',1,'metal::precise::trunc()']]], - ['truncated_5fnormal_74',['truncated_normal',['../namespacemlx_1_1core_1_1random.html#a00aa5746bac6d729d2ba9465153bb279',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a39663eda0fd7b274d01499a7b1c9035f',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['try_5fpop_75',['try_pop',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43',1,'pocketfft::detail::threading::concurrent_queue']]], - ['two_76',['two',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421a41877eab6fa3db7d7ed2cda9eba14251',1,'mlx::core::Log']]], - ['type_77',['type',['../structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e',1,'pocketfft::detail::add_vec::type'],['../structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html#a257b1c81fb9f559c48ee90497013494e',1,'pocketfft::detail::add_vec< cmplx< T > >::type'],['../structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea',1,'pocketfft::detail::ExecDcst::type'],['../structmlx_1_1core_1_1_reduction_plan.html#a24e407f13d4d02156380ecc1a6748a76',1,'mlx::core::ReductionPlan::type'],['../structmlx_1_1core_1_1_device.html#a763264ec90f7f23c5dced36c3f0db2e5',1,'mlx::core::Device::type']]], - ['typetodtype_78',['TypeToDtype',['../structmlx_1_1core_1_1_type_to_dtype.html',1,'mlx::core']]] + ['ternary_5fv2_27',['ternary_v2',['../metal_2kernels_2ternary_8h.html#a3e610f3b01966bdbf23fdfebe5d2c508',1,'ternary.h']]], + ['tgp_5fmem_5fsize_28',['tgp_mem_size',['../struct_g_e_m_v_kernel.html#a9ef4d0e62094d7033069f5dda5efb236',1,'GEMVKernel::tgp_mem_size'],['../struct_g_e_m_v_t_kernel.html#a48a09a21d7b822f380d040c752b785d7',1,'GEMVTKernel::tgp_mem_size'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a1ec583584e69dcbbb72106390a4fc5da',1,'mlx::steel::GEMMKernel::tgp_mem_size']]], + ['tgp_5fmem_5fsize_5fa_29',['tgp_mem_size_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ac00b149d76a903c2f91b0f477dc5037f',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fmem_5fsize_5fb_30',['tgp_mem_size_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a105af1069668028c6f1bc6d6dd162298',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fpadding_5fa_31',['tgp_padding_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad547704ccbff6c2076abeffa6628c5a0',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fpadding_5fb_32',['tgp_padding_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad1b03941e869017558423c08b08bc094',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fsize_33',['tgp_size',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a9058ddb73e30e83fb9c548ba22817d64',1,'mlx::steel::GEMMKernel']]], + ['thread_34',['thread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a449de02bf2ac80d8fe2f208fa7eac359',1,'mlx::core::scheduler::StreamThread']]], + ['thread_5fcount_35',['thread_count',['../structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87',1,'pocketfft::detail::util']]], + ['thread_5ffn_36',['thread_fn',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a06a62c21c1174e4eb4d242e50aad7adf',1,'mlx::core::scheduler::StreamThread']]], + ['thread_5fid_37',['thread_id',['../namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b',1,'pocketfft::detail::threading']]], + ['thread_5fidx_38',['thread_idx',['../struct_quantized_block_loader.html#a50821537ea747bc03295a09bb0eef475',1,'QuantizedBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a70da26a715135d973f88371a70255be9',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ac18de37cde1459595bfe18b0d5ef146d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ab1cb2ade639787243e0325dcd3dc0a11',1,'mlx::steel::Conv2DWeightBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9642399b8066e29123524f36ebc7b482',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acacdac168004c87fee27c8554ac905a7',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a401f0c7cf1588552556603c7ffba2316',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08a517bc50caf41155b98be0690bfe44',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::thread_idx'],['../structmlx_1_1steel_1_1_block_loader.html#a064e2cc77e0b1cf0f8027929e031775b',1,'mlx::steel::BlockLoader::thread_idx']]], + ['thread_5fmap_39',['thread_map',['../namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e',1,'pocketfft::detail::threading']]], + ['thread_5fpool_40',['thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html',1,'pocketfft::detail::threading::thread_pool'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560',1,'pocketfft::detail::threading::thread_pool::thread_pool(size_t nthreads)'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0',1,'pocketfft::detail::threading::thread_pool::thread_pool()']]], + ['thread_5fsort_5ft_41',['thread_sort_t',['../struct_block_merge_sort.html#ad2474d16721f4ceb954125728a0e2ea2',1,'BlockMergeSort']]], + ['thread_5fswap_42',['thread_swap',['../sort_8h.html#a6e8c2da4975a8001fd5ddf211a3058b7',1,'sort.h']]], + ['threads_5fper_5ftg_43',['threads_per_tg',['../struct_read_writer.html#a64c58e358da22358df3075448ea23893',1,'ReadWriter']]], + ['threadsm_44',['threadsM',['../struct_g_e_m_v_kernel.html#a1dd943fcbf5e7be435fc36bed589a641',1,'GEMVKernel::threadsM'],['../struct_g_e_m_v_t_kernel.html#a4a53e73a581aa8881b1f86ce653519e6',1,'GEMVTKernel::threadsM']]], + ['threadsn_45',['threadsN',['../struct_g_e_m_v_kernel.html#a47bfab7d21dd18760d3e0937ad36b19d',1,'GEMVKernel::threadsN'],['../struct_g_e_m_v_t_kernel.html#ade6f15a9744616de9dd71498ad7e758d',1,'GEMVTKernel::threadsN']]], + ['threadsort_46',['ThreadSort',['../struct_thread_sort.html',1,'']]], + ['threefry_2eh_47',['threefry.h',['../threefry_8h.html',1,'']]], + ['threefry2x32_5fhash_48',['threefry2x32_hash',['../namespacemlx_1_1core_1_1random.html#ac7e92c89a2bac1b0bed922a3d4c3c66b',1,'mlx::core::random']]], + ['tile_49',['tile',['../group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361',1,'mlx::core']]], + ['tile_5fstride_50',['tile_stride',['../struct_quantized_block_loader.html#ac3f651c1a645291d1037a2cc8ded2320',1,'QuantizedBlockLoader::tile_stride'],['../structmlx_1_1steel_1_1_block_loader.html#ab87876699d55473620c7ea99f9da911d',1,'mlx::steel::BlockLoader::tile_stride']]], + ['tile_5fstride_5fa_51',['tile_stride_a',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8fddaa78913cdc8eea5e1cf7d2776330',1,'mlx::steel::BlockMMA']]], + ['tile_5fstride_5fb_52',['tile_stride_b',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae3f35453b3afbaac9df64ad5966b34a4',1,'mlx::steel::BlockMMA']]], + ['tiles_5fm_53',['tiles_m',['../struct_m_l_x_fast_attention_params.html#a0df159c839fc27b9426b8ac4336cc0ad',1,'MLXFastAttentionParams::tiles_m'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a4c5e33edf70be99cf93ac5723c12eb24',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#ad23a5a7f74cd5859741a36e4bc7823ca',1,'mlx::steel::GEMMParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a0970989624e17088d5326c2e198cb95b',1,'mlx::steel::GEMMSpiltKParams::tiles_m']]], + ['tiles_5fn_54',['tiles_n',['../struct_m_l_x_fast_attention_params.html#a608aa256216ac6d80af00209303d2029',1,'MLXFastAttentionParams::tiles_n'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a76f9f381e7187a993d65128b9b681b2d',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0e6b8b629232f1b43fbce9a395174bed',1,'mlx::steel::GEMMParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a5b46dfb9cee3606efa05d217349a20a6',1,'mlx::steel::GEMMSpiltKParams::tiles_n']]], + ['tm_55',['TM',['../structmlx_1_1steel_1_1_block_m_m_a.html#aba5f749fdf32d8bd9d9e29f2a9ae4591',1,'mlx::steel::BlockMMA']]], + ['tm_56',['tm',['../structmlx_1_1steel_1_1_block_m_m_a.html#ab84d70540d09ce115794f343849f016f',1,'mlx::steel::BlockMMA']]], + ['tm_5fstride_57',['TM_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a5b0029866f493363942133b55bff7307',1,'mlx::steel::BlockMMA']]], + ['tn_58',['tn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ad7a2033983cfbb474c50c2878057d8f1',1,'mlx::steel::BlockMMA']]], + ['tn_59',['TN',['../structmlx_1_1steel_1_1_block_m_m_a.html#a706ae779c1f8d2eb18f19c248567d424',1,'mlx::steel::BlockMMA']]], + ['tn_5fstride_60',['TN_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8b3690b383afd26563efb38f9c375e50',1,'mlx::steel::BlockMMA']]], + ['to_5fbnns_5fdtype_61',['to_bnns_dtype',['../namespacemlx_1_1core.html#a7a4193f37b1de9c33c31d1da09c77edb',1,'mlx::core']]], + ['to_5fstream_62',['to_stream',['../namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9',1,'mlx::core']]], + ['topk_63',['topk',['../group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a',1,'mlx::core::topk(const array &a, int k, StreamOrDevice s={})'],['../group__ops.html#ga35b8436c79ff953f6c809598b646f498',1,'mlx::core::topk(const array &a, int k, int axis, StreamOrDevice s={})']]], + ['trace_64',['trace',['../group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, StreamOrDevice s={})'],['../group__ops.html#gaf25c00108feaafaa6350a4434cb0062e',1,'mlx::core::trace(const array &a, StreamOrDevice s={})']]], + ['transformadd_65',['TransformAdd',['../structmlx_1_1steel_1_1_transform_add.html',1,'mlx::steel::TransformAdd< OutT, InT >'],['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd::TransformAdd()']]], + ['transformaxpby_66',['TransformAxpby',['../structmlx_1_1steel_1_1_transform_axpby.html',1,'mlx::steel::TransformAxpby< OutT, InT >'],['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby::TransformAxpby()']]], + ['transformnone_67',['TransformNone',['../structmlx_1_1steel_1_1_transform_none.html',1,'mlx::steel']]], + ['transforms_2eh_68',['transforms.h',['../backend_2metal_2kernels_2steel_2gemm_2transforms_8h.html',1,'(Global Namespace)'],['../transforms_8h.html',1,'(Global Namespace)']]], + ['transforms_5fimpl_2eh_69',['transforms_impl.h',['../transforms__impl_8h.html',1,'']]], + ['transpose_70',['Transpose',['../classmlx_1_1core_1_1_transpose.html',1,'mlx::core::Transpose'],['../classmlx_1_1core_1_1_transpose.html#a1a9ba023584c61c7ac93d6dce536760a',1,'mlx::core::Transpose::Transpose()']]], + ['transpose_71',['transpose',['../group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b',1,'mlx::core::transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5',1,'mlx::core::transpose(const array &a, std::initializer_list< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a',1,'mlx::core::transpose(const array &a, StreamOrDevice s={})']]], + ['tri_72',['tri',['../group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78',1,'mlx::core::tri(int n, int m, int k, Dtype type, StreamOrDevice s={})'],['../group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5',1,'mlx::core::tri(int n, Dtype type, StreamOrDevice s={})']]], + ['tri_5finv_73',['tri_inv',['../namespacemlx_1_1core_1_1linalg.html#aba1994571326326717b5b5e38c2e0661',1,'mlx::core::linalg']]], + ['tril_74',['tril',['../group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2',1,'mlx::core']]], + ['triu_75',['triu',['../group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412',1,'mlx::core']]], + ['trows_76',['TROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a2528ff5ed472e4ed35415ada42276b07',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3957fb263fe040fe70683fd1d7b06487',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a94f0ce5bb7d87bc1fb6a7c2ba2b892d4',1,'mlx::steel::Conv2DWeightBlockLoader::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a789683f9ac9d9309d07c05f3bdedd2fd',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a5cefb1285ed13ad3490198e9303453de',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a2aff22af70f685f858adea73f5575cf7',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acbc28f364381166faaeec2783dc88e10',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::TROWS']]], + ['trunc_77',['trunc',['../namespacemetal.html#a93cb75a11a362bfc8310ea19c554c887',1,'metal::trunc()'],['../namespacemetal_1_1fast.html#aa62e1075e86c626d97038f16e9433415',1,'metal::fast::trunc()'],['../namespacemetal_1_1precise.html#a334183e7a2dd49b983d072d1e8ee2b27',1,'metal::precise::trunc()']]], + ['truncated_5fnormal_78',['truncated_normal',['../namespacemlx_1_1core_1_1random.html#a00aa5746bac6d729d2ba9465153bb279',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a39663eda0fd7b274d01499a7b1c9035f',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['try_5fpop_79',['try_pop',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43',1,'pocketfft::detail::threading::concurrent_queue']]], + ['two_80',['two',['../classmlx_1_1core_1_1_log.html#a044a23e8b1422984628e1cd5ab506421a41877eab6fa3db7d7ed2cda9eba14251',1,'mlx::core::Log']]], + ['type_81',['type',['../structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e',1,'pocketfft::detail::add_vec::type'],['../structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html#a257b1c81fb9f559c48ee90497013494e',1,'pocketfft::detail::add_vec< cmplx< T > >::type'],['../structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea',1,'pocketfft::detail::ExecDcst::type'],['../structmlx_1_1core_1_1_reduction_plan.html#a24e407f13d4d02156380ecc1a6748a76',1,'mlx::core::ReductionPlan::type'],['../structmlx_1_1core_1_1_device.html#a763264ec90f7f23c5dced36c3f0db2e5',1,'mlx::core::Device::type']]], + ['type_5fto_5fname_82',['type_to_name',['../namespacemlx_1_1core.html#af1fdfdaa5644394362e6baba30701bae',1,'mlx::core']]], + ['typetodtype_83',['TypeToDtype',['../structmlx_1_1core_1_1_type_to_dtype.html',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/all_15.js b/docs/build/html/search/all_15.js index 6c70d6aad8..800ec9e9fe 100644 --- a/docs/build/html/search/all_15.js +++ b/docs/build/html/search/all_15.js @@ -14,14 +14,15 @@ var searchData= ['unary_5fops_11',['unary_ops',['../namespacemlx_1_1core_1_1metal.html#a17b471fa52ea5f24ee63e081f46528f5',1,'mlx::core::metal']]], ['unary_5fops_2eh_12',['unary_ops.h',['../unary__ops_8h.html',1,'']]], ['unary_5fv_13',['unary_v',['../metal_2kernels_2unary_8h.html#ada52df4976078091cd7b47c152f326fb',1,'unary.h']]], - ['unaryprimitive_14',['UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html',1,'mlx::core::UnaryPrimitive'],['../classmlx_1_1core_1_1_unary_primitive.html#a189f6d4ed369f82a4b724a29eb056d4e',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(Stream stream)'],['../classmlx_1_1core_1_1_unary_primitive.html#a9935cffc4f246d3d883bc3d26c5163f2',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a780281fb04e2daf1be630c124bd605e3',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(UnaryPrimitive &&other)=delete']]], - ['uniform_15',['Uniform',['../classmlx_1_1core_1_1_uniform.html',1,'mlx::core::Uniform'],['../classmlx_1_1core_1_1_uniform.html#a626aa1091aa77b4a32c02290106b85e1',1,'mlx::core::Uniform::Uniform()']]], - ['uniform_16',['uniform',['../namespacemlx_1_1core_1_1random.html#adaa626cf75ab891978954bd1eb79a38b',1,'mlx::core::random::uniform(const array &low, const array &high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ac0dd00f7e37377d621f9f5bfb5a3f8e4',1,'mlx::core::random::uniform(T low, U high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2922e133d9f82dcf925bae0a784cc4a7',1,'mlx::core::random::uniform(const std::vector< int > &shape, Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a133d2855ff4d8daf41029cffdf43cdf9',1,'mlx::core::random::uniform(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['unscheduled_17',['unscheduled',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078ae8a9988458b0355001674020a45656fb',1,'mlx::core::array']]], - ['unsignedinteger_18',['unsignedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da9c035d4e66b2c72f583cde964cf3a0d3',1,'mlx::core::Dtype::unsignedinteger'],['../namespacemlx_1_1core.html#a42e9706a5521bb25eaf12ccad94bfc81',1,'mlx::core::unsignedinteger']]], - ['update_19',['update',['../struct_and.html#a514334a0fd865164d9c83fc07d499783',1,'And::update()'],['../struct_or.html#a942628b1b3c6be9f486c15081a0cba8e',1,'Or::update()']]], - ['use_5fout_5fsource_20',['use_out_source',['../steel__gemm__fused_8h.html#a3fe4e4382bda8a419557a5e6f77bc084',1,'steel_gemm_fused.h']]], - ['util_21',['util',['../structpocketfft_1_1detail_1_1util.html',1,'pocketfft::detail']]], - ['utils_22',['utils',['../namespacemlx_1_1core_1_1metal.html#a529dc6c2d4a37ba544b66b2c3cd792cc',1,'mlx::core::metal']]], - ['utils_2eh_23',['utils.h',['../backend_2accelerate_2utils_8h.html',1,'(Global Namespace)'],['../backend_2common_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2utils_8h.html',1,'(Global Namespace)'],['../utils_8h.html',1,'(Global Namespace)']]] + ['unary_5fv2_14',['unary_v2',['../metal_2kernels_2unary_8h.html#a95f7b9c0112cdc84ae10465fb74d29dd',1,'unary.h']]], + ['unaryprimitive_15',['UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html',1,'mlx::core::UnaryPrimitive'],['../classmlx_1_1core_1_1_unary_primitive.html#a189f6d4ed369f82a4b724a29eb056d4e',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(Stream stream)'],['../classmlx_1_1core_1_1_unary_primitive.html#a9935cffc4f246d3d883bc3d26c5163f2',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a780281fb04e2daf1be630c124bd605e3',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(UnaryPrimitive &&other)=delete']]], + ['uniform_16',['Uniform',['../classmlx_1_1core_1_1_uniform.html',1,'mlx::core::Uniform'],['../classmlx_1_1core_1_1_uniform.html#a626aa1091aa77b4a32c02290106b85e1',1,'mlx::core::Uniform::Uniform()']]], + ['uniform_17',['uniform',['../namespacemlx_1_1core_1_1random.html#adaa626cf75ab891978954bd1eb79a38b',1,'mlx::core::random::uniform(const array &low, const array &high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ac0dd00f7e37377d621f9f5bfb5a3f8e4',1,'mlx::core::random::uniform(T low, U high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2922e133d9f82dcf925bae0a784cc4a7',1,'mlx::core::random::uniform(const std::vector< int > &shape, Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a133d2855ff4d8daf41029cffdf43cdf9',1,'mlx::core::random::uniform(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['unscheduled_18',['unscheduled',['../classmlx_1_1core_1_1array.html#a199726612fa8a4bcd5c2d05eadad7078ae8a9988458b0355001674020a45656fb',1,'mlx::core::array']]], + ['unsignedinteger_19',['unsignedinteger',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da9c035d4e66b2c72f583cde964cf3a0d3',1,'mlx::core::Dtype::unsignedinteger'],['../namespacemlx_1_1core.html#a42e9706a5521bb25eaf12ccad94bfc81',1,'mlx::core::unsignedinteger']]], + ['update_20',['update',['../struct_and.html#a514334a0fd865164d9c83fc07d499783',1,'And::update()'],['../struct_or.html#a942628b1b3c6be9f486c15081a0cba8e',1,'Or::update()']]], + ['use_5fout_5fsource_21',['use_out_source',['../steel__gemm__fused_8h.html#a3fe4e4382bda8a419557a5e6f77bc084',1,'steel_gemm_fused.h']]], + ['util_22',['util',['../structpocketfft_1_1detail_1_1util.html',1,'pocketfft::detail']]], + ['utils_23',['utils',['../namespacemlx_1_1core_1_1metal.html#a529dc6c2d4a37ba544b66b2c3cd792cc',1,'mlx::core::metal']]], + ['utils_2eh_24',['utils.h',['../backend_2accelerate_2utils_8h.html',1,'(Global Namespace)'],['../backend_2common_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2steel_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2kernels_2utils_8h.html',1,'(Global Namespace)'],['../backend_2metal_2utils_8h.html',1,'(Global Namespace)'],['../utils_8h.html',1,'(Global Namespace)']]] ]; diff --git a/docs/build/html/search/all_16.js b/docs/build/html/search/all_16.js index 3c91ba0c20..9909cd5419 100644 --- a/docs/build/html/search/all_16.js +++ b/docs/build/html/search/all_16.js @@ -12,16 +12,14 @@ var searchData= ['valueandgradfn_9',['ValueAndGradFn',['../namespacemlx_1_1core.html#ab79d66ddf1ec38b2f2c01234892a2230',1,'mlx::core']]], ['var_10',['var',['../group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9',1,'mlx::core::var(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga7d7b38d118fa2613214078ef0f7d5a42',1,'mlx::core::var(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga78ddeb966cbe7a5b0aa17e1de43025f2',1,'mlx::core::var(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga4fbf3e3f98f2e4956faf87af320aa9d0',1,'mlx::core::var(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], ['vec_5fsize_11',['vec_size',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#adcc83bf6c02391cc2375e55c06a1c9a4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a71c313e1597a2bb99f7b07d434e119d2',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a10109dc9553207f5a365799e4969c6d2',1,'mlx::steel::Conv2DWeightBlockLoader::vec_size'],['../structmlx_1_1steel_1_1_channel_helper.html#a2b24f991a9380fdad6b51a038770b925',1,'mlx::steel::ChannelHelper::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a71449551bbfe56058440755dfd50fc75',1,'mlx::steel::ChannelHelper< 1 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acfb18991a77a9d1d4a79918ac5f387af',1,'mlx::steel::ChannelHelper< 2 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a5cb83774601c29564a6bbc010fc0bf7f',1,'mlx::steel::ChannelHelper< 3 >::vec_size'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#af28cdbe2a3c027d95832de07f60448ca',1,'mlx::steel::ChannelHelper< 4 >::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a6b0b18428516d1d6dcae3beb3faee81c',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a006153d274aa13d5fd4448b4607fff3a',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1587047caa339cf5b2c06adc4b332ab8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::vec_size'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08dba753ec7c8ea2892775746933b3e7',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::vec_size'],['../structmlx_1_1steel_1_1_block_loader.html#a58bdf9b9c81962733e22ecdeae28c092',1,'mlx::steel::BlockLoader::vec_size']]], - ['vector_12',['Vector',['../class_m_p_s_1_1_vector.html',1,'MPS::Vector'],['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a57dea6f5039281b7fee517fc43bf3110',1,'mlx::core::Vector']]], - ['vectordescriptor_13',['VectorDescriptor',['../class_m_p_s_1_1_vector_descriptor.html',1,'MPS']]], - ['vectordescriptor_14',['vectorDescriptor',['../class_m_p_s_1_1_vector_descriptor.html#a2bac55ff9bffcebba0755abcca09ec15',1,'MPS::VectorDescriptor::vectorDescriptor(NS::UInteger length, NS::UInteger dataType)'],['../class_m_p_s_1_1_vector_descriptor.html#ab6534f1c9536613d8e78ed72fe12b275',1,'MPS::VectorDescriptor::vectorDescriptor(NS::UInteger length, NS::UInteger vectors, NS::UInteger vectorBytes, NS::UInteger dataType)']]], - ['view_15',['View',['../classmlx_1_1core_1_1_view.html',1,'mlx::core::View'],['../classmlx_1_1core_1_1_view.html#ad7eed156c308e9a29a8b41f965ec941e',1,'mlx::core::View::View()']]], - ['view_16',['view',['../group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02',1,'mlx::core']]], - ['vjp_17',['vjp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abbf6d1d63dcda207ad7d9eeb4fc36225',1,'mlx::core::distributed::AllReduce::vjp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#aa5eff6fc128b71220899aab8ab9116fb',1,'mlx::core::distributed::AllGather::vjp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a74be4bcd0382f7f6400bf73fd5569c91',1,'mlx::core::fast::Custom::vjp()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#aacfbbbc15fcee0a5ce4f519ca3cca5eb',1,'mlx::core::fast::RMSNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#ae5e1b5df0705a6b1d141691a4396b0b6',1,'mlx::core::fast::LayerNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#ad999105414badd66c8fd9e069454a533',1,'mlx::core::fast::RoPE::vjp()'],['../classmlx_1_1core_1_1_primitive.html#a1dcb6807326eeab62474c6a0e3836d42',1,'mlx::core::Primitive::vjp()'],['../classmlx_1_1core_1_1_abs.html#aa2dd8ec0989e716b77394ac349b34592',1,'mlx::core::Abs::vjp()'],['../classmlx_1_1core_1_1_add.html#ac28e581862880e24ed2b99bb6a916607',1,'mlx::core::Add::vjp()'],['../classmlx_1_1core_1_1_add_m_m.html#ac1562a37cec6928e01281926ebeb47c6',1,'mlx::core::AddMM::vjp()'],['../classmlx_1_1core_1_1_arc_cos.html#a78e73e5e639d1249c7fe9614bf157c92',1,'mlx::core::ArcCos::vjp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a856c677f16e2b3f2edd2491e35db2d26',1,'mlx::core::ArcCosh::vjp()'],['../classmlx_1_1core_1_1_arc_sin.html#ab4057cd5ef1a8359f97493018e10d3a1',1,'mlx::core::ArcSin::vjp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a7988ee5b9e1e7e498dcab73d61ba147e',1,'mlx::core::ArcSinh::vjp()'],['../classmlx_1_1core_1_1_arc_tan.html#a5fefc3634b96a67ff8ae011a8ee180c2',1,'mlx::core::ArcTan::vjp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a99840c282e37b2b2a9c312e6e8ade1d2',1,'mlx::core::ArcTan2::vjp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a07da5797f7aaf3dfe43bf24e8562ac72',1,'mlx::core::ArcTanh::vjp()'],['../classmlx_1_1core_1_1_as_type.html#ac38a4f889311a3b5e5be9a67dcb93e18',1,'mlx::core::AsType::vjp()'],['../classmlx_1_1core_1_1_as_strided.html#a34783284c9b2f5b4a62c3c3ee5dd4062',1,'mlx::core::AsStrided::vjp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6131ed1c317ff8700a3e9b13fdaa9d61',1,'mlx::core::BitwiseBinary::vjp()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a1adf20087ee2f685bf39c2724b8e7120',1,'mlx::core::BlockMaskedMM::vjp()'],['../classmlx_1_1core_1_1_gather_m_m.html#a76c9f27c57354f6230b43944882e1bda',1,'mlx::core::GatherMM::vjp()'],['../classmlx_1_1core_1_1_broadcast.html#a0318847c9be40f00b23907ad56037d18',1,'mlx::core::Broadcast::vjp()'],['../classmlx_1_1core_1_1_ceil.html#ac2f5a2bd84b8f013e5ce688419a88acb',1,'mlx::core::Ceil::vjp()'],['../classmlx_1_1core_1_1_compiled.html#a32462e65c52f84b708188130cc508133',1,'mlx::core::Compiled::vjp()'],['../classmlx_1_1core_1_1_concatenate.html#a8155db9100ec3b8bd0bc94baeaeee3b0',1,'mlx::core::Concatenate::vjp()'],['../classmlx_1_1core_1_1_convolution.html#af8eb9c0c055ad20aa74b547016917690',1,'mlx::core::Convolution::vjp()'],['../classmlx_1_1core_1_1_copy.html#a6c4dee582001e9983e9517485ee37efd',1,'mlx::core::Copy::vjp()'],['../classmlx_1_1core_1_1_cos.html#a51d84113728e651ef9d4a1fe671c4d00',1,'mlx::core::Cos::vjp()'],['../classmlx_1_1core_1_1_cosh.html#a0791abd4305a333fb3b181a5357ce0f4',1,'mlx::core::Cosh::vjp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa1da36cef632df767cd9809d6cf06209',1,'mlx::core::CustomTransforms::vjp()'],['../classmlx_1_1core_1_1_depends.html#a02996fa45f01f7cb9f37074d5f8ccab0',1,'mlx::core::Depends::vjp()'],['../classmlx_1_1core_1_1_divide.html#ad3af7c70cad22c1a1a75b4a78ef793b6',1,'mlx::core::Divide::vjp()'],['../classmlx_1_1core_1_1_div_mod.html#a8c914a07f666a1d9377a27ed5d55e7c1',1,'mlx::core::DivMod::vjp()'],['../classmlx_1_1core_1_1_select.html#a9b522487b78fceeca7f827cd1c29a9a3',1,'mlx::core::Select::vjp()'],['../classmlx_1_1core_1_1_remainder.html#ab18f7bca1027ae71847a50da0933cec6',1,'mlx::core::Remainder::vjp()'],['../classmlx_1_1core_1_1_equal.html#af3c1bfcd1bf50922fc00e302bb193736',1,'mlx::core::Equal::vjp()'],['../classmlx_1_1core_1_1_erf.html#a1f529e95a42a2d69a8b18979d3ee2909',1,'mlx::core::Erf::vjp()'],['../classmlx_1_1core_1_1_erf_inv.html#a48afff12a58ddefae7ae0245c3580189',1,'mlx::core::ErfInv::vjp()'],['../classmlx_1_1core_1_1_exp.html#a94b9b7d137c3640d290b96c5e8b7e1a8',1,'mlx::core::Exp::vjp()'],['../classmlx_1_1core_1_1_expm1.html#af6ce416169190479c9792bb9cdbe2f43',1,'mlx::core::Expm1::vjp()'],['../classmlx_1_1core_1_1_f_f_t.html#aafc895614a6e368c0e6d64af20d01090',1,'mlx::core::FFT::vjp()'],['../classmlx_1_1core_1_1_floor.html#a589e2cf99b6fd1a5ba85534a2a31338e',1,'mlx::core::Floor::vjp()'],['../classmlx_1_1core_1_1_full.html#a49e76e7a8641f990701abc1b3bd49969',1,'mlx::core::Full::vjp()'],['../classmlx_1_1core_1_1_gather.html#aacf612a8f5f1cdbbfd19707d8d33c426',1,'mlx::core::Gather::vjp()'],['../classmlx_1_1core_1_1_greater.html#a341766a8a7e41d2a1160d35d4e781679',1,'mlx::core::Greater::vjp()'],['../classmlx_1_1core_1_1_greater_equal.html#a62f07a4ac54c708307c82aac0e5693ee',1,'mlx::core::GreaterEqual::vjp()'],['../classmlx_1_1core_1_1_hadamard.html#af4134775427b8998d66f489468b98656',1,'mlx::core::Hadamard::vjp()'],['../classmlx_1_1core_1_1_less.html#aaf205d389b5e602e0814b68f66de8f50',1,'mlx::core::Less::vjp()'],['../classmlx_1_1core_1_1_less_equal.html#aab2aab7590c299885e815c18eedd1028',1,'mlx::core::LessEqual::vjp()'],['../classmlx_1_1core_1_1_log.html#a40885dccfbf928c4d035881be1d49280',1,'mlx::core::Log::vjp()'],['../classmlx_1_1core_1_1_log1p.html#a3113c1d2b4c5e73d0b470f42dc48a880',1,'mlx::core::Log1p::vjp()'],['../classmlx_1_1core_1_1_logical_not.html#af2c3c241cf3910fbaba013c69d052a50',1,'mlx::core::LogicalNot::vjp()'],['../classmlx_1_1core_1_1_logical_and.html#ae42f8fc454577b0fd6410cae9d5f3b54',1,'mlx::core::LogicalAnd::vjp()'],['../classmlx_1_1core_1_1_logical_or.html#a51aed488f52d5031998689af9cb17847',1,'mlx::core::LogicalOr::vjp()'],['../classmlx_1_1core_1_1_log_add_exp.html#ae231af0ed24a93eb647ee58c2d2b20b4',1,'mlx::core::LogAddExp::vjp()'],['../classmlx_1_1core_1_1_matmul.html#a524136cca481598ea20894d85ca66bb0',1,'mlx::core::Matmul::vjp()'],['../classmlx_1_1core_1_1_maximum.html#a7de15d7b28784e24bbfc7e85ddcbcff3',1,'mlx::core::Maximum::vjp()'],['../classmlx_1_1core_1_1_minimum.html#a48a0cbe3a6c4f7473c00e343f63b5204',1,'mlx::core::Minimum::vjp()'],['../classmlx_1_1core_1_1_multiply.html#a74b7556ec03e2c3d3f971666d06f5db1',1,'mlx::core::Multiply::vjp()'],['../classmlx_1_1core_1_1_negative.html#a889585f056d33bda30c30311257af52a',1,'mlx::core::Negative::vjp()'],['../classmlx_1_1core_1_1_not_equal.html#a0361f29f4ae1235bdf3f3304527e2d4b',1,'mlx::core::NotEqual::vjp()'],['../classmlx_1_1core_1_1_pad.html#ad8a7e547644f2717a24322968e971038',1,'mlx::core::Pad::vjp()'],['../classmlx_1_1core_1_1_partition.html#a7110772b6cd2d430a2b825cf5c952ca9',1,'mlx::core::Partition::vjp()'],['../classmlx_1_1core_1_1_power.html#a1453bb8307d6ff33134f1e00263bf082',1,'mlx::core::Power::vjp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#acb975e272b4a88ab232ef7f7c3a2bf26',1,'mlx::core::QuantizedMatmul::vjp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#ae08a4b7d28902d46f39e66beeb0e23ab',1,'mlx::core::GatherQMM::vjp()'],['../classmlx_1_1core_1_1_reshape.html#ab17294ecc6b5d4e89626fb48c7516365',1,'mlx::core::Reshape::vjp()'],['../classmlx_1_1core_1_1_reduce.html#a684883d2a96315f548ca769510e28e4e',1,'mlx::core::Reduce::vjp()'],['../classmlx_1_1core_1_1_round.html#af8f085e08b7fa8840c52a20b12ca35ce',1,'mlx::core::Round::vjp()'],['../classmlx_1_1core_1_1_scan.html#aaf13f72620b4b5d6a20e1228930e848e',1,'mlx::core::Scan::vjp()'],['../classmlx_1_1core_1_1_scatter.html#a0b51287fba789bb139ed61d40a0c636a',1,'mlx::core::Scatter::vjp()'],['../classmlx_1_1core_1_1_sigmoid.html#aac2f56a4c8362e36a28e232758ca52cf',1,'mlx::core::Sigmoid::vjp()'],['../classmlx_1_1core_1_1_sign.html#aa60ac52edd739fbdf388a997acd01bce',1,'mlx::core::Sign::vjp()'],['../classmlx_1_1core_1_1_sin.html#aedefe550ab4b0687858981bc0bcfbfa0',1,'mlx::core::Sin::vjp()'],['../classmlx_1_1core_1_1_sinh.html#a6b39fdd429bbb4de389e7c904fd561f0',1,'mlx::core::Sinh::vjp()'],['../classmlx_1_1core_1_1_slice.html#a291746a527ff991b66249fb2b54b685f',1,'mlx::core::Slice::vjp()'],['../classmlx_1_1core_1_1_slice_update.html#aedcdc60a0477997a96306c02b66d3f77',1,'mlx::core::SliceUpdate::vjp()'],['../classmlx_1_1core_1_1_softmax.html#abb68c311c45ee422a7c966accde9041b',1,'mlx::core::Softmax::vjp()'],['../classmlx_1_1core_1_1_sort.html#a3a8900dce53ee4eb7a1b83806e629358',1,'mlx::core::Sort::vjp()'],['../classmlx_1_1core_1_1_split.html#a7e8730f9cffa9872fff6f8d577031674',1,'mlx::core::Split::vjp()'],['../classmlx_1_1core_1_1_square.html#abcd9516da7f02dc906368c23b0bca263',1,'mlx::core::Square::vjp()'],['../classmlx_1_1core_1_1_sqrt.html#a08a21bd2c3a016f042d95aca294e68f3',1,'mlx::core::Sqrt::vjp()'],['../classmlx_1_1core_1_1_subtract.html#a3a3322be7c3bcaa0397cf099091df16b',1,'mlx::core::Subtract::vjp()'],['../classmlx_1_1core_1_1_tan.html#a4639836cff03d73c769387d6943e92d7',1,'mlx::core::Tan::vjp()'],['../classmlx_1_1core_1_1_tanh.html#afe7b05e2b36b99c3a1b66f0cd3544e95',1,'mlx::core::Tanh::vjp()'],['../classmlx_1_1core_1_1_transpose.html#ac7805aa29b34afdf8852554f1e759f80',1,'mlx::core::Transpose::vjp()'],['../namespacemlx_1_1core.html#a1b33e2c2e3471420490cf0be2de6de18',1,'mlx::core::vjp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &cotangents)'],['../namespacemlx_1_1core.html#a2065a11249c3f4356ffd69b7a8c487ff',1,'mlx::core::vjp(const std::function< array(const array &)> &fun, const array &primal, const array &cotangent)']]], - ['vlen_18',['VLEN',['../structpocketfft_1_1detail_1_1_v_l_e_n.html',1,'pocketfft::detail']]], - ['vmap_19',['vmap',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a3f2dc71859847ca675ec4bfbe125035a',1,'mlx::core::distributed::AllReduce::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ad532d1d51f089dec3c84799b724ea031',1,'mlx::core::distributed::AllGather::vmap()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a7f4c3a4c48c6807faa36fb31e39dad8d',1,'mlx::core::fast::Custom::vmap()'],['../classmlx_1_1core_1_1_primitive.html#ac632b9619dd7a6a0f177bd36202e8103',1,'mlx::core::Primitive::vmap()'],['../classmlx_1_1core_1_1_abs.html#a4c9c98f1d71432fd3752ad9a6a8e7f2f',1,'mlx::core::Abs::vmap()'],['../classmlx_1_1core_1_1_add.html#a0e557d4d896153f84a25532562e4c646',1,'mlx::core::Add::vmap()'],['../classmlx_1_1core_1_1_add_m_m.html#a73ce80b3a37ec2523943028d50ebce81',1,'mlx::core::AddMM::vmap()'],['../classmlx_1_1core_1_1_arc_cos.html#a7548e23ace6827674aa6d284d44ccf83',1,'mlx::core::ArcCos::vmap()'],['../classmlx_1_1core_1_1_arc_cosh.html#af8ff78e910a9e485a203e1d3347bd461',1,'mlx::core::ArcCosh::vmap()'],['../classmlx_1_1core_1_1_arc_sin.html#a7cabb1e5a2bda44944378822c671ec82',1,'mlx::core::ArcSin::vmap()'],['../classmlx_1_1core_1_1_arc_sinh.html#a9e72b9751939387c333b5d4e19a37f6d',1,'mlx::core::ArcSinh::vmap()'],['../classmlx_1_1core_1_1_arc_tan.html#a1fb921554544a56498bc54f82e4a0556',1,'mlx::core::ArcTan::vmap()'],['../classmlx_1_1core_1_1_arc_tan2.html#ae02cb9fbf25e93dc1d7fbc9e3fb28634',1,'mlx::core::ArcTan2::vmap()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6ddcae68873559211cb91e7740dfc040',1,'mlx::core::ArcTanh::vmap()'],['../classmlx_1_1core_1_1_arg_partition.html#a441093795bcc31495ab5fbc9957b740a',1,'mlx::core::ArgPartition::vmap()'],['../classmlx_1_1core_1_1_arg_reduce.html#abfec42fa06ea15edaf393593751fb1ba',1,'mlx::core::ArgReduce::vmap()'],['../classmlx_1_1core_1_1_arg_sort.html#a3522bbbe4626a467394c1a8a9d7ac34e',1,'mlx::core::ArgSort::vmap()'],['../classmlx_1_1core_1_1_as_type.html#a7ebaf86fd6cad4a1ecfd7cde1ee0b0cc',1,'mlx::core::AsType::vmap()'],['../classmlx_1_1core_1_1_bitwise_binary.html#aa10be55f05bc1868bf4b375dc475f965',1,'mlx::core::BitwiseBinary::vmap()'],['../classmlx_1_1core_1_1_broadcast.html#aee4c71c2588ad01eb57e10f346cd666f',1,'mlx::core::Broadcast::vmap()'],['../classmlx_1_1core_1_1_ceil.html#ae86819990b43bdb0c2b3a25719b3a7a4',1,'mlx::core::Ceil::vmap()'],['../classmlx_1_1core_1_1_compiled.html#a732e7548f53977b4513bb7f30a04c30d',1,'mlx::core::Compiled::vmap()'],['../classmlx_1_1core_1_1_concatenate.html#a58c54dcf8e4b045d25edd3afc2caffc1',1,'mlx::core::Concatenate::vmap()'],['../classmlx_1_1core_1_1_conjugate.html#a2c7632c8ae0ca07777e23a0a79344e60',1,'mlx::core::Conjugate::vmap()'],['../classmlx_1_1core_1_1_copy.html#a669b10253c15b769d90058d1ad7d0e61',1,'mlx::core::Copy::vmap()'],['../classmlx_1_1core_1_1_cos.html#aec9460daf0131156734013d03b230cd6',1,'mlx::core::Cos::vmap()'],['../classmlx_1_1core_1_1_cosh.html#a1ab2386e7d96219b6e4a525f7dac0406',1,'mlx::core::Cosh::vmap()'],['../classmlx_1_1core_1_1_custom_transforms.html#a906a2ff30d9c5281fbf1fa927e4c021b',1,'mlx::core::CustomTransforms::vmap()'],['../classmlx_1_1core_1_1_divide.html#a83e7da52831165b3a026e97b63770242',1,'mlx::core::Divide::vmap()'],['../classmlx_1_1core_1_1_div_mod.html#ae709e0fdd83994bd1d156e0d0e6a7942',1,'mlx::core::DivMod::vmap()'],['../classmlx_1_1core_1_1_select.html#a84e80361c8cf02536b4b98098793550f',1,'mlx::core::Select::vmap()'],['../classmlx_1_1core_1_1_remainder.html#a79867e1099a2e3c2d3e87407b2ab6e3d',1,'mlx::core::Remainder::vmap()'],['../classmlx_1_1core_1_1_equal.html#aea9cc3c88924ac824d72c39c2e83b0ca',1,'mlx::core::Equal::vmap()'],['../classmlx_1_1core_1_1_erf.html#abe554f553356654a3e800ba368108aaa',1,'mlx::core::Erf::vmap()'],['../classmlx_1_1core_1_1_erf_inv.html#ad5d7634e8568af8cc4a54a558a48d0e9',1,'mlx::core::ErfInv::vmap()'],['../classmlx_1_1core_1_1_exp.html#a0fcd579fe148b4c3dbc72e514b81bb37',1,'mlx::core::Exp::vmap()'],['../classmlx_1_1core_1_1_expm1.html#aa4caa848b2ea97e71ee3dd33de039296',1,'mlx::core::Expm1::vmap()'],['../classmlx_1_1core_1_1_f_f_t.html#ac32d6cc9b67289124f855ea68a61ede1',1,'mlx::core::FFT::vmap()'],['../classmlx_1_1core_1_1_floor.html#aea4dc79a65774990e775ad49519a5d10',1,'mlx::core::Floor::vmap()'],['../classmlx_1_1core_1_1_full.html#afc57ab6bd9ebdbbf042af54a59785d95',1,'mlx::core::Full::vmap()'],['../classmlx_1_1core_1_1_gather.html#abab0c4c204e66489825ce80d2194a275',1,'mlx::core::Gather::vmap()'],['../classmlx_1_1core_1_1_greater.html#a6d8267411fc4951de781f9e8e6c53aa0',1,'mlx::core::Greater::vmap()'],['../classmlx_1_1core_1_1_greater_equal.html#ab0e1be93eb01b0ce7fa83e953f5e3e1d',1,'mlx::core::GreaterEqual::vmap()'],['../classmlx_1_1core_1_1_hadamard.html#a9f1a172e6246859e813002abe9b8f99c',1,'mlx::core::Hadamard::vmap()'],['../classmlx_1_1core_1_1_less.html#a5fee5956cf087d8405359121aa62ba7e',1,'mlx::core::Less::vmap()'],['../classmlx_1_1core_1_1_less_equal.html#a3d5df21db184f2b7620cda9da1684480',1,'mlx::core::LessEqual::vmap()'],['../classmlx_1_1core_1_1_log.html#a007ddbcf911093231f607a8b9ed5cd49',1,'mlx::core::Log::vmap()'],['../classmlx_1_1core_1_1_log1p.html#a7122576f95ce479926bbbbc690891f71',1,'mlx::core::Log1p::vmap()'],['../classmlx_1_1core_1_1_logical_not.html#a5308a271619ee74df561b0aaf525915d',1,'mlx::core::LogicalNot::vmap()'],['../classmlx_1_1core_1_1_logical_and.html#aacc5f6f53ffc327b7771485e3da2a4e5',1,'mlx::core::LogicalAnd::vmap()'],['../classmlx_1_1core_1_1_logical_or.html#a6e2e77e6aaf47872b2e96b151c32daf3',1,'mlx::core::LogicalOr::vmap()'],['../classmlx_1_1core_1_1_log_add_exp.html#a82190aa1421a9734b6e9480debffac78',1,'mlx::core::LogAddExp::vmap()'],['../classmlx_1_1core_1_1_matmul.html#a3a1c6e70bac300240760fe41a58340c2',1,'mlx::core::Matmul::vmap()'],['../classmlx_1_1core_1_1_maximum.html#ab664918e0d71cfec1318a9879e78c5d3',1,'mlx::core::Maximum::vmap()'],['../classmlx_1_1core_1_1_minimum.html#adab0f31acf68075a0be908d8eb882980',1,'mlx::core::Minimum::vmap()'],['../classmlx_1_1core_1_1_multiply.html#ae7e82c8fc8cbaf4e00c27eb54fac7dbf',1,'mlx::core::Multiply::vmap()'],['../classmlx_1_1core_1_1_negative.html#a1f8a6079e272f1a0599f88a1a8419cf0',1,'mlx::core::Negative::vmap()'],['../classmlx_1_1core_1_1_not_equal.html#ab8b57932f03c8eee664bf89adeaa43b5',1,'mlx::core::NotEqual::vmap()'],['../classmlx_1_1core_1_1_number_of_elements.html#a977d83eae845b8bd8c0b98b48cb1c6c2',1,'mlx::core::NumberOfElements::vmap()'],['../classmlx_1_1core_1_1_pad.html#a85658812a0f3275ba3eb74b7c75686cf',1,'mlx::core::Pad::vmap()'],['../classmlx_1_1core_1_1_partition.html#aa0cc55e4d4d2cb5d129d32832321df2c',1,'mlx::core::Partition::vmap()'],['../classmlx_1_1core_1_1_power.html#a5e22749592413a9adbdc877b03b87c8f',1,'mlx::core::Power::vmap()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a3434394140177b285f971c9ffe7e8763',1,'mlx::core::QuantizedMatmul::vmap()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a13ce5e138ebddb8780a034452f68892f',1,'mlx::core::GatherQMM::vmap()'],['../classmlx_1_1core_1_1_random_bits.html#a0dc12f053c6492f934bc18031412c415',1,'mlx::core::RandomBits::vmap()'],['../classmlx_1_1core_1_1_reshape.html#ae239dd3c6cab147e4af572dc58204f9d',1,'mlx::core::Reshape::vmap()'],['../classmlx_1_1core_1_1_reduce.html#abab1b5aa01ccad44f213f510c3596b38',1,'mlx::core::Reduce::vmap()'],['../classmlx_1_1core_1_1_round.html#a6fad8799a7982e1ccbe05be7cc38a7fd',1,'mlx::core::Round::vmap()'],['../classmlx_1_1core_1_1_scan.html#a297c7cc89c9bf9d186ebdebb634c7804',1,'mlx::core::Scan::vmap()'],['../classmlx_1_1core_1_1_sigmoid.html#a12712c23037e38192cbccd2d4b14cc85',1,'mlx::core::Sigmoid::vmap()'],['../classmlx_1_1core_1_1_sign.html#aa7296045907015b4e0ae8a93e5e6e295',1,'mlx::core::Sign::vmap()'],['../classmlx_1_1core_1_1_sin.html#a45533996f3d72d9dd97d4c61cd684fba',1,'mlx::core::Sin::vmap()'],['../classmlx_1_1core_1_1_sinh.html#ae171df22bc34c32e31b8135dc4caa788',1,'mlx::core::Sinh::vmap()'],['../classmlx_1_1core_1_1_slice.html#ae33583b0db22fcfeae34dfe1c0e3eaa2',1,'mlx::core::Slice::vmap()'],['../classmlx_1_1core_1_1_slice_update.html#adbf1c76de6ab2f986758530d351d6fa3',1,'mlx::core::SliceUpdate::vmap()'],['../classmlx_1_1core_1_1_softmax.html#ad22d3dcc71054d3dba529cf2dc981e19',1,'mlx::core::Softmax::vmap()'],['../classmlx_1_1core_1_1_sort.html#abfabb9e625cc0cb9335c7454ed27505c',1,'mlx::core::Sort::vmap()'],['../classmlx_1_1core_1_1_split.html#ab7c40e02a842e83bdb4698608472c7a6',1,'mlx::core::Split::vmap()'],['../classmlx_1_1core_1_1_square.html#a55bf43f878d4741c57a08d5fef472ea5',1,'mlx::core::Square::vmap()'],['../classmlx_1_1core_1_1_sqrt.html#a9d30e306ce08980c27d98c898577017e',1,'mlx::core::Sqrt::vmap()'],['../classmlx_1_1core_1_1_stop_gradient.html#aca680c8befef81da414c4375b11b16b0',1,'mlx::core::StopGradient::vmap()'],['../classmlx_1_1core_1_1_subtract.html#aa98f960e621a767c8a03624fd292f098',1,'mlx::core::Subtract::vmap()'],['../classmlx_1_1core_1_1_tan.html#ae2f67ca2adc83b10009cf28498bf58b7',1,'mlx::core::Tan::vmap()'],['../classmlx_1_1core_1_1_tanh.html#a32df3564c1ecb858c1ba9f855376762f',1,'mlx::core::Tanh::vmap()'],['../classmlx_1_1core_1_1_uniform.html#ad795037d5b1820e98f4268f166609926',1,'mlx::core::Uniform::vmap()'],['../classmlx_1_1core_1_1_view.html#a2230d3e5f434fb2b888de50b529ac121',1,'mlx::core::View::vmap()'],['../classmlx_1_1core_1_1_transpose.html#a5ef848b69def9a246665b67e6e3ffdfe',1,'mlx::core::Transpose::vmap()'],['../classmlx_1_1core_1_1_s_v_d.html#a0366c958f6cdac8d1d9e1a4eda53fae8',1,'mlx::core::SVD::vmap()'],['../classmlx_1_1core_1_1_inverse.html#a98419b9f0b8a6c9185fe012d523552c2',1,'mlx::core::Inverse::vmap()'],['../classmlx_1_1core_1_1_cholesky.html#ab5c3f6199ec3b399c91243a05d116aa5',1,'mlx::core::Cholesky::vmap()'],['../namespacemlx_1_1core.html#a933289d4688479e1c4d8ba04332c406b',1,'mlx::core::vmap'],['../namespacemlx_1_1core.html#a8481a3bb4c12c2b7dc6ba576c2be3d0d',1,'mlx::core::vmap(const std::function< array(const array &, const array &)> &fun, int in_axis_a=0, int in_axis_b=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a95a7757e8d18fced38acfc6a3e8d686a',1,'mlx::core::vmap(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &in_axes={}, const std::vector< int > &out_axes={})']]], - ['vmap_5freplace_20',['vmap_replace',['../namespacemlx_1_1core_1_1detail.html#a31a5582530faea230eb8acafc0f7e154',1,'mlx::core::detail']]], - ['vmap_5ftrace_21',['vmap_trace',['../namespacemlx_1_1core_1_1detail.html#a5ba794afe1a557e0505887cfb481c515',1,'mlx::core::detail']]], - ['vtype_22',['VTYPE',['../structpocketfft_1_1detail_1_1_v_t_y_p_e.html',1,'pocketfft::detail']]], - ['vtype_5ft_23',['vtype_t',['../namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62',1,'pocketfft::detail']]] + ['vector_12',['Vector',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a57dea6f5039281b7fee517fc43bf3110',1,'mlx::core']]], + ['view_13',['View',['../classmlx_1_1core_1_1_view.html',1,'mlx::core::View'],['../classmlx_1_1core_1_1_view.html#ad7eed156c308e9a29a8b41f965ec941e',1,'mlx::core::View::View()']]], + ['view_14',['view',['../group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02',1,'mlx::core']]], + ['vjp_15',['vjp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abbf6d1d63dcda207ad7d9eeb4fc36225',1,'mlx::core::distributed::AllReduce::vjp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#aa5eff6fc128b71220899aab8ab9116fb',1,'mlx::core::distributed::AllGather::vjp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a74be4bcd0382f7f6400bf73fd5569c91',1,'mlx::core::fast::Custom::vjp()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#aacfbbbc15fcee0a5ce4f519ca3cca5eb',1,'mlx::core::fast::RMSNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#ae5e1b5df0705a6b1d141691a4396b0b6',1,'mlx::core::fast::LayerNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#ad999105414badd66c8fd9e069454a533',1,'mlx::core::fast::RoPE::vjp()'],['../classmlx_1_1core_1_1_primitive.html#a1dcb6807326eeab62474c6a0e3836d42',1,'mlx::core::Primitive::vjp()'],['../classmlx_1_1core_1_1_abs.html#aa2dd8ec0989e716b77394ac349b34592',1,'mlx::core::Abs::vjp()'],['../classmlx_1_1core_1_1_add.html#ac28e581862880e24ed2b99bb6a916607',1,'mlx::core::Add::vjp()'],['../classmlx_1_1core_1_1_add_m_m.html#ac1562a37cec6928e01281926ebeb47c6',1,'mlx::core::AddMM::vjp()'],['../classmlx_1_1core_1_1_arc_cos.html#a78e73e5e639d1249c7fe9614bf157c92',1,'mlx::core::ArcCos::vjp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a856c677f16e2b3f2edd2491e35db2d26',1,'mlx::core::ArcCosh::vjp()'],['../classmlx_1_1core_1_1_arc_sin.html#ab4057cd5ef1a8359f97493018e10d3a1',1,'mlx::core::ArcSin::vjp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a7988ee5b9e1e7e498dcab73d61ba147e',1,'mlx::core::ArcSinh::vjp()'],['../classmlx_1_1core_1_1_arc_tan.html#a5fefc3634b96a67ff8ae011a8ee180c2',1,'mlx::core::ArcTan::vjp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a99840c282e37b2b2a9c312e6e8ade1d2',1,'mlx::core::ArcTan2::vjp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a07da5797f7aaf3dfe43bf24e8562ac72',1,'mlx::core::ArcTanh::vjp()'],['../classmlx_1_1core_1_1_as_type.html#ac38a4f889311a3b5e5be9a67dcb93e18',1,'mlx::core::AsType::vjp()'],['../classmlx_1_1core_1_1_as_strided.html#a34783284c9b2f5b4a62c3c3ee5dd4062',1,'mlx::core::AsStrided::vjp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6131ed1c317ff8700a3e9b13fdaa9d61',1,'mlx::core::BitwiseBinary::vjp()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a1adf20087ee2f685bf39c2724b8e7120',1,'mlx::core::BlockMaskedMM::vjp()'],['../classmlx_1_1core_1_1_gather_m_m.html#a76c9f27c57354f6230b43944882e1bda',1,'mlx::core::GatherMM::vjp()'],['../classmlx_1_1core_1_1_broadcast.html#a0318847c9be40f00b23907ad56037d18',1,'mlx::core::Broadcast::vjp()'],['../classmlx_1_1core_1_1_ceil.html#ac2f5a2bd84b8f013e5ce688419a88acb',1,'mlx::core::Ceil::vjp()'],['../classmlx_1_1core_1_1_compiled.html#a32462e65c52f84b708188130cc508133',1,'mlx::core::Compiled::vjp()'],['../classmlx_1_1core_1_1_concatenate.html#a8155db9100ec3b8bd0bc94baeaeee3b0',1,'mlx::core::Concatenate::vjp()'],['../classmlx_1_1core_1_1_convolution.html#af8eb9c0c055ad20aa74b547016917690',1,'mlx::core::Convolution::vjp()'],['../classmlx_1_1core_1_1_copy.html#a6c4dee582001e9983e9517485ee37efd',1,'mlx::core::Copy::vjp()'],['../classmlx_1_1core_1_1_cos.html#a51d84113728e651ef9d4a1fe671c4d00',1,'mlx::core::Cos::vjp()'],['../classmlx_1_1core_1_1_cosh.html#a0791abd4305a333fb3b181a5357ce0f4',1,'mlx::core::Cosh::vjp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa1da36cef632df767cd9809d6cf06209',1,'mlx::core::CustomTransforms::vjp()'],['../classmlx_1_1core_1_1_depends.html#a02996fa45f01f7cb9f37074d5f8ccab0',1,'mlx::core::Depends::vjp()'],['../classmlx_1_1core_1_1_divide.html#ad3af7c70cad22c1a1a75b4a78ef793b6',1,'mlx::core::Divide::vjp()'],['../classmlx_1_1core_1_1_div_mod.html#a8c914a07f666a1d9377a27ed5d55e7c1',1,'mlx::core::DivMod::vjp()'],['../classmlx_1_1core_1_1_select.html#a9b522487b78fceeca7f827cd1c29a9a3',1,'mlx::core::Select::vjp()'],['../classmlx_1_1core_1_1_remainder.html#ab18f7bca1027ae71847a50da0933cec6',1,'mlx::core::Remainder::vjp()'],['../classmlx_1_1core_1_1_equal.html#af3c1bfcd1bf50922fc00e302bb193736',1,'mlx::core::Equal::vjp()'],['../classmlx_1_1core_1_1_erf.html#a1f529e95a42a2d69a8b18979d3ee2909',1,'mlx::core::Erf::vjp()'],['../classmlx_1_1core_1_1_erf_inv.html#a48afff12a58ddefae7ae0245c3580189',1,'mlx::core::ErfInv::vjp()'],['../classmlx_1_1core_1_1_exp.html#a94b9b7d137c3640d290b96c5e8b7e1a8',1,'mlx::core::Exp::vjp()'],['../classmlx_1_1core_1_1_expm1.html#af6ce416169190479c9792bb9cdbe2f43',1,'mlx::core::Expm1::vjp()'],['../classmlx_1_1core_1_1_f_f_t.html#aafc895614a6e368c0e6d64af20d01090',1,'mlx::core::FFT::vjp()'],['../classmlx_1_1core_1_1_floor.html#a589e2cf99b6fd1a5ba85534a2a31338e',1,'mlx::core::Floor::vjp()'],['../classmlx_1_1core_1_1_full.html#a49e76e7a8641f990701abc1b3bd49969',1,'mlx::core::Full::vjp()'],['../classmlx_1_1core_1_1_gather.html#aacf612a8f5f1cdbbfd19707d8d33c426',1,'mlx::core::Gather::vjp()'],['../classmlx_1_1core_1_1_greater.html#a341766a8a7e41d2a1160d35d4e781679',1,'mlx::core::Greater::vjp()'],['../classmlx_1_1core_1_1_greater_equal.html#a62f07a4ac54c708307c82aac0e5693ee',1,'mlx::core::GreaterEqual::vjp()'],['../classmlx_1_1core_1_1_hadamard.html#af4134775427b8998d66f489468b98656',1,'mlx::core::Hadamard::vjp()'],['../classmlx_1_1core_1_1_less.html#aaf205d389b5e602e0814b68f66de8f50',1,'mlx::core::Less::vjp()'],['../classmlx_1_1core_1_1_less_equal.html#aab2aab7590c299885e815c18eedd1028',1,'mlx::core::LessEqual::vjp()'],['../classmlx_1_1core_1_1_log.html#a40885dccfbf928c4d035881be1d49280',1,'mlx::core::Log::vjp()'],['../classmlx_1_1core_1_1_log1p.html#a3113c1d2b4c5e73d0b470f42dc48a880',1,'mlx::core::Log1p::vjp()'],['../classmlx_1_1core_1_1_logical_not.html#af2c3c241cf3910fbaba013c69d052a50',1,'mlx::core::LogicalNot::vjp()'],['../classmlx_1_1core_1_1_logical_and.html#ae42f8fc454577b0fd6410cae9d5f3b54',1,'mlx::core::LogicalAnd::vjp()'],['../classmlx_1_1core_1_1_logical_or.html#a51aed488f52d5031998689af9cb17847',1,'mlx::core::LogicalOr::vjp()'],['../classmlx_1_1core_1_1_log_add_exp.html#ae231af0ed24a93eb647ee58c2d2b20b4',1,'mlx::core::LogAddExp::vjp()'],['../classmlx_1_1core_1_1_matmul.html#a524136cca481598ea20894d85ca66bb0',1,'mlx::core::Matmul::vjp()'],['../classmlx_1_1core_1_1_maximum.html#a7de15d7b28784e24bbfc7e85ddcbcff3',1,'mlx::core::Maximum::vjp()'],['../classmlx_1_1core_1_1_minimum.html#a48a0cbe3a6c4f7473c00e343f63b5204',1,'mlx::core::Minimum::vjp()'],['../classmlx_1_1core_1_1_multiply.html#a74b7556ec03e2c3d3f971666d06f5db1',1,'mlx::core::Multiply::vjp()'],['../classmlx_1_1core_1_1_negative.html#a889585f056d33bda30c30311257af52a',1,'mlx::core::Negative::vjp()'],['../classmlx_1_1core_1_1_not_equal.html#a0361f29f4ae1235bdf3f3304527e2d4b',1,'mlx::core::NotEqual::vjp()'],['../classmlx_1_1core_1_1_pad.html#ad8a7e547644f2717a24322968e971038',1,'mlx::core::Pad::vjp()'],['../classmlx_1_1core_1_1_partition.html#a7110772b6cd2d430a2b825cf5c952ca9',1,'mlx::core::Partition::vjp()'],['../classmlx_1_1core_1_1_power.html#a1453bb8307d6ff33134f1e00263bf082',1,'mlx::core::Power::vjp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#acb975e272b4a88ab232ef7f7c3a2bf26',1,'mlx::core::QuantizedMatmul::vjp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#ae08a4b7d28902d46f39e66beeb0e23ab',1,'mlx::core::GatherQMM::vjp()'],['../classmlx_1_1core_1_1_reshape.html#ab17294ecc6b5d4e89626fb48c7516365',1,'mlx::core::Reshape::vjp()'],['../classmlx_1_1core_1_1_reduce.html#a684883d2a96315f548ca769510e28e4e',1,'mlx::core::Reduce::vjp()'],['../classmlx_1_1core_1_1_round.html#af8f085e08b7fa8840c52a20b12ca35ce',1,'mlx::core::Round::vjp()'],['../classmlx_1_1core_1_1_scan.html#aaf13f72620b4b5d6a20e1228930e848e',1,'mlx::core::Scan::vjp()'],['../classmlx_1_1core_1_1_scatter.html#a0b51287fba789bb139ed61d40a0c636a',1,'mlx::core::Scatter::vjp()'],['../classmlx_1_1core_1_1_sigmoid.html#aac2f56a4c8362e36a28e232758ca52cf',1,'mlx::core::Sigmoid::vjp()'],['../classmlx_1_1core_1_1_sign.html#aa60ac52edd739fbdf388a997acd01bce',1,'mlx::core::Sign::vjp()'],['../classmlx_1_1core_1_1_sin.html#aedefe550ab4b0687858981bc0bcfbfa0',1,'mlx::core::Sin::vjp()'],['../classmlx_1_1core_1_1_sinh.html#a6b39fdd429bbb4de389e7c904fd561f0',1,'mlx::core::Sinh::vjp()'],['../classmlx_1_1core_1_1_slice.html#a291746a527ff991b66249fb2b54b685f',1,'mlx::core::Slice::vjp()'],['../classmlx_1_1core_1_1_slice_update.html#aedcdc60a0477997a96306c02b66d3f77',1,'mlx::core::SliceUpdate::vjp()'],['../classmlx_1_1core_1_1_softmax.html#abb68c311c45ee422a7c966accde9041b',1,'mlx::core::Softmax::vjp()'],['../classmlx_1_1core_1_1_sort.html#a3a8900dce53ee4eb7a1b83806e629358',1,'mlx::core::Sort::vjp()'],['../classmlx_1_1core_1_1_split.html#a7e8730f9cffa9872fff6f8d577031674',1,'mlx::core::Split::vjp()'],['../classmlx_1_1core_1_1_square.html#abcd9516da7f02dc906368c23b0bca263',1,'mlx::core::Square::vjp()'],['../classmlx_1_1core_1_1_sqrt.html#a08a21bd2c3a016f042d95aca294e68f3',1,'mlx::core::Sqrt::vjp()'],['../classmlx_1_1core_1_1_subtract.html#a3a3322be7c3bcaa0397cf099091df16b',1,'mlx::core::Subtract::vjp()'],['../classmlx_1_1core_1_1_tan.html#a4639836cff03d73c769387d6943e92d7',1,'mlx::core::Tan::vjp()'],['../classmlx_1_1core_1_1_tanh.html#afe7b05e2b36b99c3a1b66f0cd3544e95',1,'mlx::core::Tanh::vjp()'],['../classmlx_1_1core_1_1_transpose.html#ac7805aa29b34afdf8852554f1e759f80',1,'mlx::core::Transpose::vjp()'],['../namespacemlx_1_1core.html#a1b33e2c2e3471420490cf0be2de6de18',1,'mlx::core::vjp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &cotangents)'],['../namespacemlx_1_1core.html#a2065a11249c3f4356ffd69b7a8c487ff',1,'mlx::core::vjp(const std::function< array(const array &)> &fun, const array &primal, const array &cotangent)']]], + ['vlen_16',['VLEN',['../structpocketfft_1_1detail_1_1_v_l_e_n.html',1,'pocketfft::detail']]], + ['vmap_17',['vmap',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a3f2dc71859847ca675ec4bfbe125035a',1,'mlx::core::distributed::AllReduce::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ad532d1d51f089dec3c84799b724ea031',1,'mlx::core::distributed::AllGather::vmap()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a7f4c3a4c48c6807faa36fb31e39dad8d',1,'mlx::core::fast::Custom::vmap()'],['../classmlx_1_1core_1_1_primitive.html#ac632b9619dd7a6a0f177bd36202e8103',1,'mlx::core::Primitive::vmap()'],['../classmlx_1_1core_1_1_abs.html#a4c9c98f1d71432fd3752ad9a6a8e7f2f',1,'mlx::core::Abs::vmap()'],['../classmlx_1_1core_1_1_add.html#a0e557d4d896153f84a25532562e4c646',1,'mlx::core::Add::vmap()'],['../classmlx_1_1core_1_1_add_m_m.html#a73ce80b3a37ec2523943028d50ebce81',1,'mlx::core::AddMM::vmap()'],['../classmlx_1_1core_1_1_arc_cos.html#a7548e23ace6827674aa6d284d44ccf83',1,'mlx::core::ArcCos::vmap()'],['../classmlx_1_1core_1_1_arc_cosh.html#af8ff78e910a9e485a203e1d3347bd461',1,'mlx::core::ArcCosh::vmap()'],['../classmlx_1_1core_1_1_arc_sin.html#a7cabb1e5a2bda44944378822c671ec82',1,'mlx::core::ArcSin::vmap()'],['../classmlx_1_1core_1_1_arc_sinh.html#a9e72b9751939387c333b5d4e19a37f6d',1,'mlx::core::ArcSinh::vmap()'],['../classmlx_1_1core_1_1_arc_tan.html#a1fb921554544a56498bc54f82e4a0556',1,'mlx::core::ArcTan::vmap()'],['../classmlx_1_1core_1_1_arc_tan2.html#ae02cb9fbf25e93dc1d7fbc9e3fb28634',1,'mlx::core::ArcTan2::vmap()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6ddcae68873559211cb91e7740dfc040',1,'mlx::core::ArcTanh::vmap()'],['../classmlx_1_1core_1_1_arg_partition.html#a441093795bcc31495ab5fbc9957b740a',1,'mlx::core::ArgPartition::vmap()'],['../classmlx_1_1core_1_1_arg_reduce.html#abfec42fa06ea15edaf393593751fb1ba',1,'mlx::core::ArgReduce::vmap()'],['../classmlx_1_1core_1_1_arg_sort.html#a3522bbbe4626a467394c1a8a9d7ac34e',1,'mlx::core::ArgSort::vmap()'],['../classmlx_1_1core_1_1_as_type.html#a7ebaf86fd6cad4a1ecfd7cde1ee0b0cc',1,'mlx::core::AsType::vmap()'],['../classmlx_1_1core_1_1_bitwise_binary.html#aa10be55f05bc1868bf4b375dc475f965',1,'mlx::core::BitwiseBinary::vmap()'],['../classmlx_1_1core_1_1_broadcast.html#aee4c71c2588ad01eb57e10f346cd666f',1,'mlx::core::Broadcast::vmap()'],['../classmlx_1_1core_1_1_ceil.html#ae86819990b43bdb0c2b3a25719b3a7a4',1,'mlx::core::Ceil::vmap()'],['../classmlx_1_1core_1_1_compiled.html#a732e7548f53977b4513bb7f30a04c30d',1,'mlx::core::Compiled::vmap()'],['../classmlx_1_1core_1_1_concatenate.html#a58c54dcf8e4b045d25edd3afc2caffc1',1,'mlx::core::Concatenate::vmap()'],['../classmlx_1_1core_1_1_conjugate.html#a2c7632c8ae0ca07777e23a0a79344e60',1,'mlx::core::Conjugate::vmap()'],['../classmlx_1_1core_1_1_copy.html#a669b10253c15b769d90058d1ad7d0e61',1,'mlx::core::Copy::vmap()'],['../classmlx_1_1core_1_1_cos.html#aec9460daf0131156734013d03b230cd6',1,'mlx::core::Cos::vmap()'],['../classmlx_1_1core_1_1_cosh.html#a1ab2386e7d96219b6e4a525f7dac0406',1,'mlx::core::Cosh::vmap()'],['../classmlx_1_1core_1_1_custom_transforms.html#a906a2ff30d9c5281fbf1fa927e4c021b',1,'mlx::core::CustomTransforms::vmap()'],['../classmlx_1_1core_1_1_divide.html#a83e7da52831165b3a026e97b63770242',1,'mlx::core::Divide::vmap()'],['../classmlx_1_1core_1_1_div_mod.html#ae709e0fdd83994bd1d156e0d0e6a7942',1,'mlx::core::DivMod::vmap()'],['../classmlx_1_1core_1_1_select.html#a84e80361c8cf02536b4b98098793550f',1,'mlx::core::Select::vmap()'],['../classmlx_1_1core_1_1_remainder.html#a79867e1099a2e3c2d3e87407b2ab6e3d',1,'mlx::core::Remainder::vmap()'],['../classmlx_1_1core_1_1_equal.html#aea9cc3c88924ac824d72c39c2e83b0ca',1,'mlx::core::Equal::vmap()'],['../classmlx_1_1core_1_1_erf.html#abe554f553356654a3e800ba368108aaa',1,'mlx::core::Erf::vmap()'],['../classmlx_1_1core_1_1_erf_inv.html#ad5d7634e8568af8cc4a54a558a48d0e9',1,'mlx::core::ErfInv::vmap()'],['../classmlx_1_1core_1_1_exp.html#a0fcd579fe148b4c3dbc72e514b81bb37',1,'mlx::core::Exp::vmap()'],['../classmlx_1_1core_1_1_expm1.html#aa4caa848b2ea97e71ee3dd33de039296',1,'mlx::core::Expm1::vmap()'],['../classmlx_1_1core_1_1_f_f_t.html#ac32d6cc9b67289124f855ea68a61ede1',1,'mlx::core::FFT::vmap()'],['../classmlx_1_1core_1_1_floor.html#aea4dc79a65774990e775ad49519a5d10',1,'mlx::core::Floor::vmap()'],['../classmlx_1_1core_1_1_full.html#afc57ab6bd9ebdbbf042af54a59785d95',1,'mlx::core::Full::vmap()'],['../classmlx_1_1core_1_1_gather.html#abab0c4c204e66489825ce80d2194a275',1,'mlx::core::Gather::vmap()'],['../classmlx_1_1core_1_1_greater.html#a6d8267411fc4951de781f9e8e6c53aa0',1,'mlx::core::Greater::vmap()'],['../classmlx_1_1core_1_1_greater_equal.html#ab0e1be93eb01b0ce7fa83e953f5e3e1d',1,'mlx::core::GreaterEqual::vmap()'],['../classmlx_1_1core_1_1_hadamard.html#a9f1a172e6246859e813002abe9b8f99c',1,'mlx::core::Hadamard::vmap()'],['../classmlx_1_1core_1_1_less.html#a5fee5956cf087d8405359121aa62ba7e',1,'mlx::core::Less::vmap()'],['../classmlx_1_1core_1_1_less_equal.html#a3d5df21db184f2b7620cda9da1684480',1,'mlx::core::LessEqual::vmap()'],['../classmlx_1_1core_1_1_log.html#a007ddbcf911093231f607a8b9ed5cd49',1,'mlx::core::Log::vmap()'],['../classmlx_1_1core_1_1_log1p.html#a7122576f95ce479926bbbbc690891f71',1,'mlx::core::Log1p::vmap()'],['../classmlx_1_1core_1_1_logical_not.html#a5308a271619ee74df561b0aaf525915d',1,'mlx::core::LogicalNot::vmap()'],['../classmlx_1_1core_1_1_logical_and.html#aacc5f6f53ffc327b7771485e3da2a4e5',1,'mlx::core::LogicalAnd::vmap()'],['../classmlx_1_1core_1_1_logical_or.html#a6e2e77e6aaf47872b2e96b151c32daf3',1,'mlx::core::LogicalOr::vmap()'],['../classmlx_1_1core_1_1_log_add_exp.html#a82190aa1421a9734b6e9480debffac78',1,'mlx::core::LogAddExp::vmap()'],['../classmlx_1_1core_1_1_matmul.html#a3a1c6e70bac300240760fe41a58340c2',1,'mlx::core::Matmul::vmap()'],['../classmlx_1_1core_1_1_maximum.html#ab664918e0d71cfec1318a9879e78c5d3',1,'mlx::core::Maximum::vmap()'],['../classmlx_1_1core_1_1_minimum.html#adab0f31acf68075a0be908d8eb882980',1,'mlx::core::Minimum::vmap()'],['../classmlx_1_1core_1_1_multiply.html#ae7e82c8fc8cbaf4e00c27eb54fac7dbf',1,'mlx::core::Multiply::vmap()'],['../classmlx_1_1core_1_1_negative.html#a1f8a6079e272f1a0599f88a1a8419cf0',1,'mlx::core::Negative::vmap()'],['../classmlx_1_1core_1_1_not_equal.html#ab8b57932f03c8eee664bf89adeaa43b5',1,'mlx::core::NotEqual::vmap()'],['../classmlx_1_1core_1_1_number_of_elements.html#a977d83eae845b8bd8c0b98b48cb1c6c2',1,'mlx::core::NumberOfElements::vmap()'],['../classmlx_1_1core_1_1_pad.html#a85658812a0f3275ba3eb74b7c75686cf',1,'mlx::core::Pad::vmap()'],['../classmlx_1_1core_1_1_partition.html#aa0cc55e4d4d2cb5d129d32832321df2c',1,'mlx::core::Partition::vmap()'],['../classmlx_1_1core_1_1_power.html#a5e22749592413a9adbdc877b03b87c8f',1,'mlx::core::Power::vmap()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a3434394140177b285f971c9ffe7e8763',1,'mlx::core::QuantizedMatmul::vmap()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a13ce5e138ebddb8780a034452f68892f',1,'mlx::core::GatherQMM::vmap()'],['../classmlx_1_1core_1_1_random_bits.html#a0dc12f053c6492f934bc18031412c415',1,'mlx::core::RandomBits::vmap()'],['../classmlx_1_1core_1_1_reshape.html#ae239dd3c6cab147e4af572dc58204f9d',1,'mlx::core::Reshape::vmap()'],['../classmlx_1_1core_1_1_reduce.html#abab1b5aa01ccad44f213f510c3596b38',1,'mlx::core::Reduce::vmap()'],['../classmlx_1_1core_1_1_round.html#a6fad8799a7982e1ccbe05be7cc38a7fd',1,'mlx::core::Round::vmap()'],['../classmlx_1_1core_1_1_scan.html#a297c7cc89c9bf9d186ebdebb634c7804',1,'mlx::core::Scan::vmap()'],['../classmlx_1_1core_1_1_scatter.html#a696c38b373a7a7c71bc112bd1117e322',1,'mlx::core::Scatter::vmap()'],['../classmlx_1_1core_1_1_sigmoid.html#a12712c23037e38192cbccd2d4b14cc85',1,'mlx::core::Sigmoid::vmap()'],['../classmlx_1_1core_1_1_sign.html#aa7296045907015b4e0ae8a93e5e6e295',1,'mlx::core::Sign::vmap()'],['../classmlx_1_1core_1_1_sin.html#a45533996f3d72d9dd97d4c61cd684fba',1,'mlx::core::Sin::vmap()'],['../classmlx_1_1core_1_1_sinh.html#ae171df22bc34c32e31b8135dc4caa788',1,'mlx::core::Sinh::vmap()'],['../classmlx_1_1core_1_1_slice.html#ae33583b0db22fcfeae34dfe1c0e3eaa2',1,'mlx::core::Slice::vmap()'],['../classmlx_1_1core_1_1_slice_update.html#adbf1c76de6ab2f986758530d351d6fa3',1,'mlx::core::SliceUpdate::vmap()'],['../classmlx_1_1core_1_1_softmax.html#ad22d3dcc71054d3dba529cf2dc981e19',1,'mlx::core::Softmax::vmap()'],['../classmlx_1_1core_1_1_sort.html#abfabb9e625cc0cb9335c7454ed27505c',1,'mlx::core::Sort::vmap()'],['../classmlx_1_1core_1_1_split.html#ab7c40e02a842e83bdb4698608472c7a6',1,'mlx::core::Split::vmap()'],['../classmlx_1_1core_1_1_square.html#a55bf43f878d4741c57a08d5fef472ea5',1,'mlx::core::Square::vmap()'],['../classmlx_1_1core_1_1_sqrt.html#a9d30e306ce08980c27d98c898577017e',1,'mlx::core::Sqrt::vmap()'],['../classmlx_1_1core_1_1_stop_gradient.html#aca680c8befef81da414c4375b11b16b0',1,'mlx::core::StopGradient::vmap()'],['../classmlx_1_1core_1_1_subtract.html#aa98f960e621a767c8a03624fd292f098',1,'mlx::core::Subtract::vmap()'],['../classmlx_1_1core_1_1_tan.html#ae2f67ca2adc83b10009cf28498bf58b7',1,'mlx::core::Tan::vmap()'],['../classmlx_1_1core_1_1_tanh.html#a32df3564c1ecb858c1ba9f855376762f',1,'mlx::core::Tanh::vmap()'],['../classmlx_1_1core_1_1_uniform.html#ad795037d5b1820e98f4268f166609926',1,'mlx::core::Uniform::vmap()'],['../classmlx_1_1core_1_1_view.html#a2230d3e5f434fb2b888de50b529ac121',1,'mlx::core::View::vmap()'],['../classmlx_1_1core_1_1_transpose.html#a5ef848b69def9a246665b67e6e3ffdfe',1,'mlx::core::Transpose::vmap()'],['../classmlx_1_1core_1_1_s_v_d.html#a0366c958f6cdac8d1d9e1a4eda53fae8',1,'mlx::core::SVD::vmap()'],['../classmlx_1_1core_1_1_inverse.html#a98419b9f0b8a6c9185fe012d523552c2',1,'mlx::core::Inverse::vmap()'],['../classmlx_1_1core_1_1_cholesky.html#ab5c3f6199ec3b399c91243a05d116aa5',1,'mlx::core::Cholesky::vmap()'],['../namespacemlx_1_1core.html#a933289d4688479e1c4d8ba04332c406b',1,'mlx::core::vmap'],['../namespacemlx_1_1core.html#a8481a3bb4c12c2b7dc6ba576c2be3d0d',1,'mlx::core::vmap(const std::function< array(const array &, const array &)> &fun, int in_axis_a=0, int in_axis_b=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a95a7757e8d18fced38acfc6a3e8d686a',1,'mlx::core::vmap(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &in_axes={}, const std::vector< int > &out_axes={})']]], + ['vmap_5freplace_18',['vmap_replace',['../namespacemlx_1_1core_1_1detail.html#a31a5582530faea230eb8acafc0f7e154',1,'mlx::core::detail']]], + ['vmap_5ftrace_19',['vmap_trace',['../namespacemlx_1_1core_1_1detail.html#a5ba794afe1a557e0505887cfb481c515',1,'mlx::core::detail']]], + ['vtype_20',['VTYPE',['../structpocketfft_1_1detail_1_1_v_t_y_p_e.html',1,'pocketfft::detail']]], + ['vtype_5ft_21',['vtype_t',['../namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62',1,'pocketfft::detail']]] ]; diff --git a/docs/build/html/search/all_2.js b/docs/build/html/search/all_2.js index 6f93d9a25d..3b8ffbb8fa 100644 --- a/docs/build/html/search/all_2.js +++ b/docs/build/html/search/all_2.js @@ -44,56 +44,61 @@ var searchData= ['binary_5fg_5fnd1_41',['binary_g_nd1',['../metal_2kernels_2binary_8h.html#a6808bfb006cb5473da087a2758d0d867',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, constant const size_t &a_stride, constant const size_t &b_stride, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ad1fad37c168192b212a4294f4cf78133',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const size_t &a_stride, constant const size_t &b_stride, uint index): binary_two.h']]], ['binary_5fg_5fnd2_42',['binary_g_nd2',['../metal_2kernels_2binary_8h.html#a8cd5989852ec704c6fd132ae28f4fc14',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a03f7c15a1607576755abb65c542ae347',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim): binary_two.h']]], ['binary_5fg_5fnd3_43',['binary_g_nd3',['../metal_2kernels_2binary_8h.html#ac4979e60b993f7ffb602bcb91cd68bc9',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a43e5943460996c43060d1f3aa1309ba6',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim): binary_two.h']]], - ['binary_5fop_5fgpu_44',['binary_op_gpu',['../namespacemlx_1_1core.html#a63f3c166ea4916faa5d038668cfe16b1',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s)'],['../namespacemlx_1_1core.html#a6437ced86c227e0807352ae6b16da796',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)']]], - ['binary_5fop_5fgpu_5finplace_45',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#afee8898eaceab6dd5e90357f4a0613c4',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s)'],['../namespacemlx_1_1core.html#ad53410af1488f83d74035e3132fa16af',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)']]], + ['binary_5fop_5fgpu_44',['binary_op_gpu',['../namespacemlx_1_1core.html#ad884f4a36308b5b4f8a5d990d2e086df',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a094876ea5a2a2445ab64efc8222da202',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], + ['binary_5fop_5fgpu_5finplace_45',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#a8616c0b7b0fc118a75400bc86404c367',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a7e6af6624e322e7ad60a3873a66e18a3',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], ['binary_5fops_46',['binary_ops',['../namespacemlx_1_1core_1_1metal.html#a8db7f9cc781d4bfb08423a401665f322',1,'mlx::core::metal']]], ['binary_5fops_2eh_47',['binary_ops.h',['../binary__ops_8h.html',1,'']]], ['binary_5fss_48',['binary_ss',['../metal_2kernels_2binary_8h.html#a242b8b29a852c255467e50628c6dccf5',1,'binary_ss(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#af8a791ac7ca88d32cd8f4e9ac0f9ab4f',1,'binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], ['binary_5fsv_49',['binary_sv',['../metal_2kernels_2binary_8h.html#a4116c35f2e4632366d1611d5a95ba141',1,'binary_sv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab18c6ecf5065275c93701efd095c916c',1,'binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5ftwo_50',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]], - ['binary_5ftwo_2eh_51',['binary_two.h',['../common_2binary__two_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__two_8h.html',1,'(Global Namespace)']]], - ['binary_5fvs_52',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fvv_53',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['bits_54',['bits',['../namespacemlx_1_1core_1_1random.html#abb895baa477f5a06b5f88e69245f1825',1,'mlx::core::random::bits(const std::vector< int > &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a548ffed4ba3107b89885ff850ffce5f4',1,'mlx::core::random::bits(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['bits_5f_55',['bits_',['../struct___m_l_x___b_float16.html#a4113263b63e3757ea8334cc4f0f5c3c8',1,'_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aca48963f820065c3d8ecab24265ab3fc',1,'mlx::core::_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a5203fe52424fd32bce6eb7917dd9288b',1,'mlx::core::_MLX_Float16::bits_']]], - ['bits_5fto_5fbfloat_56',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]], - ['bits_5fto_5fbfloat_5fstruct_57',['bits_to_bfloat_struct',['../struct___m_l_x___b_float16_1_1bits__to__bfloat__struct.html',1,'_MLX_BFloat16']]], - ['bitwise_5fand_58',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]], - ['bitwise_5for_59',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]], - ['bitwise_5fxor_60',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]], - ['bitwiseand_61',['BitwiseAnd',['../struct_bitwise_and.html',1,'BitwiseAnd'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html',1,'mlx::core::detail::BitwiseAnd']]], - ['bitwisebinary_62',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html',1,'mlx::core::BitwiseBinary'],['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary::BitwiseBinary()']]], - ['bitwiseor_63',['BitwiseOr',['../struct_bitwise_or.html',1,'BitwiseOr'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html',1,'mlx::core::detail::BitwiseOr']]], - ['bitwisexor_64',['BitwiseXor',['../struct_bitwise_xor.html',1,'BitwiseXor'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html',1,'mlx::core::detail::BitwiseXor']]], - ['bj_65',['bj',['../struct_quantized_block_loader.html#ae2add92b2aaf3414e91f0470b9b0cc00',1,'QuantizedBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a7ae9e41f50c0c63c35b63086a1c22cc3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a6fd3dd7b74d91609fa9dd61c657a0e32',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a6f2fdcaf5a67567cca38ae3d8120ab37',1,'mlx::steel::Conv2DWeightBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a7cf448573d41fbc67f8dfc65b7aef2b2',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#adaa261fc2e8e694aedab4ebd60b52e5e',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#ace16704025bc6e6204c306a357f3a8b8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acec010e10d5733654963407af38d4f67',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_block_loader.html#a78c326e75ee35a484685771143047cd4',1,'mlx::steel::BlockLoader::bj']]], - ['block_5fmasked_5fgemm_66',['block_masked_gemm',['../steel__gemm__masked_8h.html#af805e998b2046ee30c2b4be813e3af97',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#a477932e2ae9d49366f7ede6db63f9cac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]], - ['block_5fmasked_5fmm_67',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]], - ['block_5fmerge_5fsort_5ft_68',['block_merge_sort_t',['../struct_kernel_merge_sort.html#adae7850e057fc30d5328c7b3dcc998fa',1,'KernelMergeSort::block_merge_sort_t'],['../struct_kernel_multi_block_merge_sort.html#af27e9af4b58640c0aa620bc4efc68dff',1,'KernelMultiBlockMergeSort::block_merge_sort_t']]], - ['block_5fsort_69',['block_sort',['../struct_kernel_merge_sort.html#a56b644ec66f7fb5c01b280f124304be9',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a322ed2eac315a561e0fd90af2fd577eb',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]], - ['block_5fsort_5fnc_70',['block_sort_nc',['../sort_8h.html#a29229399f51e5c440ffe5c9b99b27598',1,'sort.h']]], - ['blockloader_71',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader()']]], - ['blockmaskedmm_72',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html',1,'mlx::core::BlockMaskedMM'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM::BlockMaskedMM()']]], - ['blockmergesort_73',['BlockMergeSort',['../struct_block_merge_sort.html',1,'']]], - ['blockmma_74',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA()']]], - ['blockswizzle_75',['BlockSwizzle',['../structmlx_1_1steel_1_1_block_swizzle.html',1,'mlx::steel']]], - ['bluestein_5ffft_76',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]], - ['bool4_5for_5fuint_77',['bool4_or_uint',['../unionbool4__or__uint.html',1,'']]], - ['bool_5f_78',['bool_',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa467afb5838aa377d55cce81f84c5512b',1,'mlx::core::Dtype::bool_'],['../namespacemlx_1_1core.html#a113d2bac7e4aa6a4cb4a5c3242527b82',1,'mlx::core::bool_']]], - ['broadcast_79',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html',1,'mlx::core::Broadcast'],['../classmlx_1_1core_1_1_broadcast.html#accbab8433c93e281608a268d11afaefb',1,'mlx::core::Broadcast::Broadcast()']]], - ['broadcast_5farrays_80',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]], - ['broadcast_5fshapes_81',['broadcast_shapes',['../namespacemlx_1_1core.html#a075e07def338cd9d815182d0e6a656c0',1,'mlx::core']]], - ['broadcast_5fto_82',['broadcast_to',['../group__ops.html#gad256e86cc1a6e6b3832e392baa90318d',1,'mlx::core']]], - ['brows_83',['BROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ac070c6bd5be85b1ae805e18890db4fd4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a10591ea957605a9c662f93d59ff3410d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae9b86b05b23153ea1abaeead456c491c',1,'mlx::steel::Conv2DWeightBlockLoader::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a343984fb74ec579a4404278dbbc7e7b5',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acc8140aae84694f62e6324dbb6a614a4',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aba1e1c8012e4e50f0e9bcfb9486c1781',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a015a0c56de74a0c4d51953a7e94fbba8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BROWS']]], - ['bs_5foffset_84',['Bs_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92f6aeee432f53638447eac842f43eca',1,'mlx::steel::BlockMMA']]], - ['bs_5fqmm_5fn_85',['bs_qmm_n',['../quantized_8h.html#acfc43edcf0c742ddc090d22cb7229284',1,'quantized.h']]], - ['bs_5fqmm_5ft_86',['bs_qmm_t',['../quantized_8h.html#a3acc2ace766cb855f13da2d1834e7dc7',1,'quantized.h']]], - ['bs_5fqmv_87',['bs_qmv',['../quantized_8h.html#a03ba4a4a5fe9955319b0aa477d2d7d98',1,'quantized.h']]], - ['bs_5fqmv_5ffast_88',['bs_qmv_fast',['../quantized_8h.html#a6f92bb9f3d29d707bfc680bebe1c80f7',1,'quantized.h']]], - ['bs_5fqvm_89',['bs_qvm',['../quantized_8h.html#a3af1c89416632c9275b8218a33cb8a04',1,'quantized.h']]], - ['bsimd_90',['Bsimd',['../structmlx_1_1steel_1_1_block_m_m_a.html#a3de506004b428d3487bae05ad9e757e6',1,'mlx::steel::BlockMMA']]], - ['buf_91',['buf',['../struct_read_writer.html#a23bac3c96dd0265ddbee1f256be45ff5',1,'ReadWriter::buf'],['../backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697',1,'buf: allocator.h']]], - ['buffer_92',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html',1,'mlx::core::allocator::Buffer'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer::Buffer()']]], - ['buffer_93',['buffer',['../structmlx_1_1core_1_1array_1_1_data.html#a9a51e2d12ba505027cc0fca86bdd39ad',1,'mlx::core::array::Data::buffer'],['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]], - ['buffers_94',['buffers',['../struct_indices.html#ad705070a740579c07d109ae4f3d86e76',1,'Indices']]], - ['build_5flib_5fname_95',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]] + ['binary_5fsv2_50',['binary_sv2',['../metal_2kernels_2binary_8h.html#aa8c48b1b21d8f5a181f5443de2346589',1,'binary_sv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a08822ff98ea6f61a98b49a9e9a38b891',1,'binary_sv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5ftwo_51',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]], + ['binary_5ftwo_2eh_52',['binary_two.h',['../common_2binary__two_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2binary__two_8h.html',1,'(Global Namespace)']]], + ['binary_5fvs_53',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fvs2_54',['binary_vs2',['../metal_2kernels_2binary_8h.html#a48bd82eb10f9c623ce7d28daec4fa512',1,'binary_vs2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a273d2f31691f2c64623c2a97eab344be',1,'binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5fvv_55',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fvv2_56',['binary_vv2',['../metal_2kernels_2binary_8h.html#a19dbbf8fea68b64bdd25dc8d36865171',1,'binary_vv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12e80730e43dfaa4c79ce8d5f99edc50',1,'binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['bits_57',['bits',['../namespacemlx_1_1core_1_1random.html#abb895baa477f5a06b5f88e69245f1825',1,'mlx::core::random::bits(const std::vector< int > &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a548ffed4ba3107b89885ff850ffce5f4',1,'mlx::core::random::bits(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['bits_5f_58',['bits_',['../struct___m_l_x___b_float16.html#a4113263b63e3757ea8334cc4f0f5c3c8',1,'_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aca48963f820065c3d8ecab24265ab3fc',1,'mlx::core::_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a5203fe52424fd32bce6eb7917dd9288b',1,'mlx::core::_MLX_Float16::bits_']]], + ['bits_5fto_5fbfloat_59',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]], + ['bits_5fto_5fbfloat_5fstruct_60',['bits_to_bfloat_struct',['../struct___m_l_x___b_float16_1_1bits__to__bfloat__struct.html',1,'_MLX_BFloat16']]], + ['bitwise_5fand_61',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]], + ['bitwise_5for_62',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]], + ['bitwise_5fxor_63',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]], + ['bitwiseand_64',['BitwiseAnd',['../struct_bitwise_and.html',1,'BitwiseAnd'],['../structmlx_1_1core_1_1detail_1_1_bitwise_and.html',1,'mlx::core::detail::BitwiseAnd']]], + ['bitwisebinary_65',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html',1,'mlx::core::BitwiseBinary'],['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary::BitwiseBinary()']]], + ['bitwiseor_66',['BitwiseOr',['../struct_bitwise_or.html',1,'BitwiseOr'],['../structmlx_1_1core_1_1detail_1_1_bitwise_or.html',1,'mlx::core::detail::BitwiseOr']]], + ['bitwisexor_67',['BitwiseXor',['../struct_bitwise_xor.html',1,'BitwiseXor'],['../structmlx_1_1core_1_1detail_1_1_bitwise_xor.html',1,'mlx::core::detail::BitwiseXor']]], + ['bj_68',['bj',['../struct_quantized_block_loader.html#ae2add92b2aaf3414e91f0470b9b0cc00',1,'QuantizedBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a7ae9e41f50c0c63c35b63086a1c22cc3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a6fd3dd7b74d91609fa9dd61c657a0e32',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a6f2fdcaf5a67567cca38ae3d8120ab37',1,'mlx::steel::Conv2DWeightBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a7cf448573d41fbc67f8dfc65b7aef2b2',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#adaa261fc2e8e694aedab4ebd60b52e5e',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#ace16704025bc6e6204c306a357f3a8b8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acec010e10d5733654963407af38d4f67',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_block_loader.html#a78c326e75ee35a484685771143047cd4',1,'mlx::steel::BlockLoader::bj']]], + ['block_5fmasked_5fgemm_69',['block_masked_gemm',['../steel__gemm__masked_8h.html#af805e998b2046ee30c2b4be813e3af97',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#a477932e2ae9d49366f7ede6db63f9cac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]], + ['block_5fmasked_5fmm_70',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]], + ['block_5fmerge_5fsort_5ft_71',['block_merge_sort_t',['../struct_kernel_merge_sort.html#adae7850e057fc30d5328c7b3dcc998fa',1,'KernelMergeSort::block_merge_sort_t'],['../struct_kernel_multi_block_merge_sort.html#af27e9af4b58640c0aa620bc4efc68dff',1,'KernelMultiBlockMergeSort::block_merge_sort_t']]], + ['block_5fsort_72',['block_sort',['../struct_kernel_merge_sort.html#a56b644ec66f7fb5c01b280f124304be9',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a322ed2eac315a561e0fd90af2fd577eb',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]], + ['block_5fsort_5fnc_73',['block_sort_nc',['../sort_8h.html#a29229399f51e5c440ffe5c9b99b27598',1,'sort.h']]], + ['blockloader_74',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html',1,'mlx::steel::BlockLoader< T, BROWS, BCOLS, dst_ld, reduction_dim, tgp_size, alignment, n_reads, TCOLS, TROWS >'],['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader::BlockLoader()']]], + ['blockm_75',['blockM',['../struct_g_e_m_v_kernel.html#a7281520100658811076400060663903c',1,'GEMVKernel::blockM'],['../struct_g_e_m_v_t_kernel.html#a2ae8ce535d59cccf453381b4485a77f0',1,'GEMVTKernel::blockM']]], + ['blockmaskedmm_76',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html',1,'mlx::core::BlockMaskedMM'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM::BlockMaskedMM()']]], + ['blockmergesort_77',['BlockMergeSort',['../struct_block_merge_sort.html',1,'']]], + ['blockmma_78',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html',1,'mlx::steel::BlockMMA< T, U, BM, BN, BK, WM, WN, transpose_a, transpose_b, lda_tgp, ldb_tgp, AccumType, Epilogue >'],['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA::BlockMMA()']]], + ['blockn_79',['blockN',['../struct_g_e_m_v_kernel.html#a2fef17f9c9aa0bdf530ad3554fb0988b',1,'GEMVKernel::blockN'],['../struct_g_e_m_v_t_kernel.html#a60be87666006ba0bf88bc8e6902da42a',1,'GEMVTKernel::blockN']]], + ['blockswizzle_80',['BlockSwizzle',['../structmlx_1_1steel_1_1_block_swizzle.html',1,'mlx::steel']]], + ['bluestein_5ffft_81',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]], + ['bool4_5for_5fuint_82',['bool4_or_uint',['../unionbool4__or__uint.html',1,'']]], + ['bool_5f_83',['bool_',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa467afb5838aa377d55cce81f84c5512b',1,'mlx::core::Dtype::bool_'],['../namespacemlx_1_1core.html#a113d2bac7e4aa6a4cb4a5c3242527b82',1,'mlx::core::bool_']]], + ['broadcast_84',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html',1,'mlx::core::Broadcast'],['../classmlx_1_1core_1_1_broadcast.html#accbab8433c93e281608a268d11afaefb',1,'mlx::core::Broadcast::Broadcast()']]], + ['broadcast_5farrays_85',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]], + ['broadcast_5fshapes_86',['broadcast_shapes',['../namespacemlx_1_1core.html#a075e07def338cd9d815182d0e6a656c0',1,'mlx::core']]], + ['broadcast_5fto_87',['broadcast_to',['../group__ops.html#gad256e86cc1a6e6b3832e392baa90318d',1,'mlx::core']]], + ['brows_88',['BROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ac070c6bd5be85b1ae805e18890db4fd4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a10591ea957605a9c662f93d59ff3410d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae9b86b05b23153ea1abaeead456c491c',1,'mlx::steel::Conv2DWeightBlockLoader::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a343984fb74ec579a4404278dbbc7e7b5',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acc8140aae84694f62e6324dbb6a614a4',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aba1e1c8012e4e50f0e9bcfb9486c1781',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a015a0c56de74a0c4d51953a7e94fbba8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BROWS']]], + ['bs_5foffset_89',['Bs_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92f6aeee432f53638447eac842f43eca',1,'mlx::steel::BlockMMA']]], + ['bs_5fqmm_5fn_90',['bs_qmm_n',['../quantized_8h.html#acfc43edcf0c742ddc090d22cb7229284',1,'quantized.h']]], + ['bs_5fqmm_5ft_91',['bs_qmm_t',['../quantized_8h.html#a3acc2ace766cb855f13da2d1834e7dc7',1,'quantized.h']]], + ['bs_5fqmv_92',['bs_qmv',['../quantized_8h.html#a03ba4a4a5fe9955319b0aa477d2d7d98',1,'quantized.h']]], + ['bs_5fqmv_5ffast_93',['bs_qmv_fast',['../quantized_8h.html#a6f92bb9f3d29d707bfc680bebe1c80f7',1,'quantized.h']]], + ['bs_5fqvm_94',['bs_qvm',['../quantized_8h.html#a3af1c89416632c9275b8218a33cb8a04',1,'quantized.h']]], + ['bsimd_95',['Bsimd',['../structmlx_1_1steel_1_1_block_m_m_a.html#a3de506004b428d3487bae05ad9e757e6',1,'mlx::steel::BlockMMA']]], + ['buf_96',['buf',['../struct_read_writer.html#a23bac3c96dd0265ddbee1f256be45ff5',1,'ReadWriter::buf'],['../backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697',1,'buf: allocator.h']]], + ['buffer_97',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html',1,'mlx::core::allocator::Buffer'],['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer::Buffer()']]], + ['buffer_98',['buffer',['../structmlx_1_1core_1_1array_1_1_data.html#a9a51e2d12ba505027cc0fca86bdd39ad',1,'mlx::core::array::Data::buffer'],['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]], + ['buffers_99',['buffers',['../struct_indices.html#ad705070a740579c07d109ae4f3d86e76',1,'Indices']]], + ['build_5flib_5fname_100',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/all_3.js b/docs/build/html/search/all_3.js index 82fb12bc47..66d5a87f9b 100644 --- a/docs/build/html/search/all_3.js +++ b/docs/build/html/search/all_3.js @@ -1,7 +1,7 @@ var searchData= [ - ['c_0',['C',['../struct_m_l_x_conv_params.html#a0953063962ac3b5a027243289e72fbb2',1,'MLXConvParams']]], - ['c_1',['c',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a4a8a08f09d37b73795649038408b5f33',1,'mlx::core::Dtype']]], + ['c_0',['c',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715a4a8a08f09d37b73795649038408b5f33',1,'mlx::core::Dtype']]], + ['c_1',['C',['../struct_m_l_x_conv_params.html#a0953063962ac3b5a027243289e72fbb2',1,'MLXConvParams']]], ['c2c_2',['c2c',['../namespacepocketfft_1_1detail.html#ab585ac594ae1253d4659e7b9e1623c8a',1,'pocketfft::detail']]], ['c2r_3',['c2r',['../namespacepocketfft_1_1detail.html#ab26cbfed16f487b987f50bf63bfc1ab9',1,'pocketfft::detail::c2r(const shape_t &shape_out, const stride_t &stride_in, const stride_t &stride_out, size_t axis, bool forward, const std::complex< T > *data_in, T *data_out, T fct, size_t nthreads=1)'],['../namespacepocketfft_1_1detail.html#a788506fff59f8e13056247076cac51c1',1,'pocketfft::detail::c2r(const shape_t &shape_out, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, bool forward, const std::complex< T > *data_in, T *data_out, T fct, size_t nthreads=1)']]], ['can_5fconvert_5ffrom_5fbfloat_4',['can_convert_from_bfloat',['../backend_2metal_2kernels_2bf16_8h.html#a7e5992f7fcd8f2cdadcc1d7f6aefbb5a',1,'bf16.h']]], @@ -26,119 +26,122 @@ var searchData= ['checkpoint_23',['checkpoint',['../namespacemlx_1_1core.html#a26127b71b2ec65c51d7627e71847083d',1,'mlx::core']]], ['cholesky_24',['Cholesky',['../classmlx_1_1core_1_1_cholesky.html',1,'mlx::core::Cholesky'],['../classmlx_1_1core_1_1_cholesky.html#a6ae2e30b85f99f4f0d7f14c7949818ab',1,'mlx::core::Cholesky::Cholesky()']]], ['cholesky_25',['cholesky',['../namespacemlx_1_1core_1_1linalg.html#a46c8a4f806f0a97a4323e91189aa512b',1,'mlx::core::linalg']]], - ['clear_5fcache_26',['clear_cache',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a447c1eb38c00d2e8e521675297f4a9b1',1,'mlx::core::metal::MetalAllocator::clear_cache()'],['../namespacemlx_1_1core_1_1metal.html#a22b3384ebd17f2fca198f81b9f1b6dc3',1,'mlx::core::metal::clear_cache()']]], - ['clip_27',['clip',['../group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8',1,'mlx::core']]], - ['cmplx_28',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail::cmplx< T >'],['../structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c',1,'pocketfft::detail::cmplx::cmplx()'],['../structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a',1,'pocketfft::detail::cmplx::cmplx(T r_, T i_)']]], - ['cmplx_3c_20t0_20_3e_29',['cmplx< T0 >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], - ['cmplx_3c_20thigh_20_3e_30',['cmplx< Thigh >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], - ['cndarr_31',['cndarr',['../classpocketfft_1_1detail_1_1cndarr.html',1,'pocketfft::detail::cndarr< T >'],['../classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129',1,'pocketfft::detail::cndarr::cndarr()']]], - ['col_5fcontiguous_32',['col_contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#ae24709026598d635e6b5c24a15f8a802',1,'mlx::core::array::Flags']]], - ['col_5freduce_5fgeneral_33',['col_reduce_general',['../reduce__col_8h.html#a3b14333e195c0a07b70069bebf85d5c3',1,'reduce_col.h']]], - ['col_5freduce_5fgeneral_5fno_5fatomics_34',['col_reduce_general_no_atomics',['../reduce__col_8h.html#aef03a4131c710ba8f94a666f58719eb7',1,'reduce_col.h']]], - ['col_5freduce_5fsmall_35',['col_reduce_small',['../reduce__col_8h.html#a7da96ffb66e1fda27bad2852c2285b94',1,'reduce_col.h']]], - ['collapse_5fcontiguous_5fdims_36',['collapse_contiguous_dims',['../namespacemlx_1_1core.html#a9d151ba3e138be1954d2f51f85806b0c',1,'mlx::core::collapse_contiguous_dims(const std::vector< int > &shape, const std::vector< std::vector< stride_t > > strides)'],['../namespacemlx_1_1core.html#a8430e0baac3f6d8a2ab22428f9c0b7e2',1,'mlx::core::collapse_contiguous_dims(const std::vector< array > &xs)'],['../namespacemlx_1_1core.html#ac813412cce77fc1340dcfefc6e099276',1,'mlx::core::collapse_contiguous_dims(Arrays &&... xs)']]], - ['commandencoder_37',['CommandEncoder',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html',1,'mlx::core::metal::CommandEncoder'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::metal::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::metal::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete']]], - ['commit_5fcommand_5fbuffer_38',['commit_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a95248f1387824067fd4fed23ace5ac0c',1,'mlx::core::metal::Device']]], - ['commonallocator_39',['CommonAllocator',['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html',1,'mlx::core::allocator']]], - ['communication_5fstream_40',['communication_stream',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6',1,'mlx::core::distributed::detail']]], - ['compile_41',['compile',['../namespacemlx_1_1core.html#ab93149e46a6d8f3e1988123fab508dc2',1,'mlx::core::compile'],['../namespacemlx_1_1core_1_1detail.html#a4d14dee182dc9a8bea64efdaad310a1f',1,'mlx::core::detail::compile']]], - ['compile_2eh_42',['compile.h',['../compile_8h.html',1,'']]], - ['compile_5favailable_5ffor_5fdevice_43',['compile_available_for_device',['../namespacemlx_1_1core_1_1detail.html#aeeff2ba6ec3d9d4ed090de6d2681dbc2',1,'mlx::core::detail']]], - ['compile_5fclear_5fcache_44',['compile_clear_cache',['../namespacemlx_1_1core_1_1detail.html#a3fb927c209b946aefebb195993fbe4cf',1,'mlx::core::detail']]], - ['compile_5ferase_45',['compile_erase',['../namespacemlx_1_1core_1_1detail.html#a69eb76a14f845ca000f1ccb2edda0175',1,'mlx::core::detail']]], - ['compile_5fimpl_2eh_46',['compile_impl.h',['../compile__impl_8h.html',1,'']]], - ['compiled_47',['Compiled',['../classmlx_1_1core_1_1_compiled.html',1,'mlx::core::Compiled'],['../classmlx_1_1core_1_1_compiled.html#a2d8cefff835c419a48a077d306b8e051',1,'mlx::core::Compiled::Compiled()']]], - ['compiled_2eh_48',['compiled.h',['../compiled_8h.html',1,'']]], - ['compiled_5fallocate_5foutputs_49',['compiled_allocate_outputs',['../namespacemlx_1_1core.html#ab8c3c4fc05745f586de922c8266f4fce',1,'mlx::core']]], - ['compiled_5fcheck_5fcontiguity_50',['compiled_check_contiguity',['../namespacemlx_1_1core.html#a3b900ab319948c5a01a3ecd30a709027',1,'mlx::core']]], - ['compiled_5fpreamble_2eh_51',['compiled_preamble.h',['../compiled__preamble_8h.html',1,'']]], - ['compilemode_52',['CompileMode',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4',1,'mlx::core']]], - ['complex_2eh_53',['complex.h',['../backend_2metal_2kernels_2complex_8h.html',1,'(Global Namespace)'],['../types_2complex_8h.html',1,'(Global Namespace)']]], - ['complex128_5ft_54',['complex128_t',['../structmlx_1_1core_1_1complex128__t.html',1,'mlx::core::complex128_t'],['../structmlx_1_1core_1_1complex128__t.html#aa15d0b805f8790f7c7b76fc7b9d677e0',1,'mlx::core::complex128_t::complex128_t(double v, double u)'],['../structmlx_1_1core_1_1complex128__t.html#abf2842253b874f9f13f39ea68a89e5b6',1,'mlx::core::complex128_t::complex128_t(std::complex< double > v)'],['../structmlx_1_1core_1_1complex128__t.html#a526fba96d7e815360cb4226af085a1bf',1,'mlx::core::complex128_t::complex128_t(T x)']]], - ['complex64_55',['complex64',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa8c022579455bcd2c681f007e84f4e2cf',1,'mlx::core::Dtype::complex64'],['../namespacemlx_1_1core.html#af99db87e0078bfcdb383f5689bc874d4',1,'mlx::core::complex64']]], - ['complex64_5ft_56',['complex64_t',['../structcomplex64__t.html',1,'complex64_t'],['../structmlx_1_1core_1_1complex64__t.html',1,'mlx::core::complex64_t'],['../structcomplex64__t.html#adbd392a5e92d31997380ad0a38be4be8',1,'complex64_t::complex64_t(float real, float imag)'],['../structcomplex64__t.html#a33a2452eb33b5ed53655773539c357a5',1,'complex64_t::complex64_t(T x) thread'],['../structcomplex64__t.html#a89b65ace8588b7bf215355f705eb23d9',1,'complex64_t::complex64_t(T x) threadgroup'],['../structcomplex64__t.html#ac81b486f642fb3b26c5d659917bdbcd0',1,'complex64_t::complex64_t(T x) device'],['../structcomplex64__t.html#a0a27a41206400f1e62b60ceb56960c93',1,'complex64_t::complex64_t(T x) const ant'],['../structmlx_1_1core_1_1complex64__t.html#a697cc973ae27d63c8e00d830e780bd8c',1,'mlx::core::complex64_t::complex64_t(float v, float u)'],['../structmlx_1_1core_1_1complex64__t.html#ae065e39938f9c4374b4116f4c67d4d09',1,'mlx::core::complex64_t::complex64_t(std::complex< float > v)'],['../structmlx_1_1core_1_1complex64__t.html#a2232cbbe591a9d2bc228cb23fac38b50',1,'mlx::core::complex64_t::complex64_t(T x)']]], - ['complex_5fbinop_57',['complex_binop',['../types_2complex_8h.html#a9c7995d495359894e1b30c0f1678d6bd',1,'complex.h']]], - ['complex_5fbinop_5fhelper_58',['complex_binop_helper',['../types_2complex_8h.html#ac6890f9852de12339b09b65757ebc8c4',1,'complex.h']]], - ['complex_5fmul_59',['complex_mul',['../radix_8h.html#a5bfc53b531214c9ce277bebc18aa67d6',1,'radix.h']]], - ['complex_5fmul_5fconj_60',['complex_mul_conj',['../radix_8h.html#a0e2dfd3d1dda09f47ccc64eec35629f3',1,'radix.h']]], - ['complexfloating_61',['complexfloating',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dafb203630099d501ff7c255a574bc4812',1,'mlx::core::Dtype::complexfloating'],['../namespacemlx_1_1core.html#a70b8e88c9df750af984757105af33423',1,'mlx::core::complexfloating']]], - ['compute_5fstrided_5findices_62',['compute_strided_indices',['../struct_read_writer.html#a7c903fbb8b85a856ba5564d7df537cdf',1,'ReadWriter']]], - ['concatenate_63',['Concatenate',['../classmlx_1_1core_1_1_concatenate.html',1,'mlx::core::Concatenate'],['../classmlx_1_1core_1_1_concatenate.html#acff07853de2d31faeec7c4ca40ce0888',1,'mlx::core::Concatenate::Concatenate()']]], - ['concatenate_64',['concatenate',['../group__ops.html#gabdc36fa65697d0361c8d67495de77129',1,'mlx::core::concatenate(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#gaa95c34ca3a8877f2c50cb60e7fa312b8',1,'mlx::core::concatenate(const std::vector< array > &arrays, StreamOrDevice s={})']]], - ['concatenate_5fgpu_65',['concatenate_gpu',['../namespacemlx_1_1core.html#a050299d0d366ca5c9d09d1004dcc3e7d',1,'mlx::core']]], - ['concurrent_5fqueue_66',['concurrent_queue',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], - ['concurrent_5fqueue_3c_20std_3a_3afunction_3c_20void_28_29_3e_20_3e_67',['concurrent_queue< std::function< void()> >',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], - ['concurrentcontext_68',['ConcurrentContext',['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html',1,'mlx::core::metal::CommandEncoder::ConcurrentContext'],['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::metal::CommandEncoder::ConcurrentContext::ConcurrentContext()']]], - ['cond_69',['cond',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4ffd524d6a5bedd1a303b63bdde6701c',1,'mlx::core::scheduler::StreamThread']]], - ['conj_70',['conj',['../namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3',1,'pocketfft::detail']]], - ['conjugate_71',['Conjugate',['../struct_conjugate.html',1,'Conjugate'],['../classmlx_1_1core_1_1_conjugate.html',1,'mlx::core::Conjugate'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html',1,'mlx::core::detail::Conjugate'],['../classmlx_1_1core_1_1_conjugate.html#a627f9e6a8729fb3ffb3ca3228d007c87',1,'mlx::core::Conjugate::Conjugate()']]], - ['conjugate_72',['conjugate',['../group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23',1,'mlx::core']]], - ['contiguous_73',['contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#afd0ab11e7a486a2a8e50ee84b971ac8a',1,'mlx::core::array::Flags']]], - ['contiguous_5fscan_74',['contiguous_scan',['../kernels_2scan_8h.html#a1f9c197dc09641f3fbb2e886c85b0675',1,'scan.h']]], - ['contiguousallreduce_75',['ContiguousAllReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ae4e34c7154eb8dc47aa8503209730424',1,'mlx::core']]], - ['contiguousreduce_76',['ContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ad2547f25dffe8d8936dbec25601cfc84',1,'mlx::core']]], - ['contiguousstridedreduce_77',['ContiguousStridedReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ab48dac7508a2c790de1bdc33f29177ed',1,'mlx::core']]], - ['conv_78',['conv',['../namespacemlx_1_1core_1_1metal.html#ab1704e853394c725668c06752ebb5c24',1,'mlx::core::metal']]], - ['conv_2eh_79',['conv.h',['../conv_8h.html',1,'']]], - ['conv1d_80',['conv1d',['../group__ops.html#ga30d47e08093c03a3676f235f9f559411',1,'mlx::core']]], - ['conv2d_81',['conv2d',['../group__ops.html#ga73b02833229678786e7f302d458d5a83',1,'mlx::core']]], - ['conv2dgeneralbaseinfo_82',['Conv2DGeneralBaseInfo',['../structmlx_1_1steel_1_1_conv2_d_general_base_info.html',1,'mlx::steel']]], - ['conv2dgeneraljumpparams_83',['Conv2DGeneralJumpParams',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html',1,'mlx::steel']]], - ['conv2dinputblockloadergeneral_84',['Conv2DInputBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html',1,'mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1d83af561a483432bf8dcb42e734b23b',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::Conv2DInputBlockLoaderGeneral()']]], - ['conv2dinputblockloaderlargefilter_85',['Conv2DInputBlockLoaderLargeFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8755116a535539744e4947bc69f9c50f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::Conv2DInputBlockLoaderLargeFilter()']]], - ['conv2dinputblockloadersmallchannels_86',['Conv2DInputBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ab9fd3fdeab94470dde3326f1dd5c455a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::Conv2DInputBlockLoaderSmallChannels()']]], - ['conv2dinputblockloadersmallfilter_87',['Conv2DInputBlockLoaderSmallFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0a2cbf57c51cd928722e3f06aafcf933',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::Conv2DInputBlockLoaderSmallFilter()']]], - ['conv2dweightblockloader_88',['Conv2DWeightBlockLoader',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html',1,'mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a9a7dca3512b64cffb6eac305d795831c',1,'mlx::steel::Conv2DWeightBlockLoader::Conv2DWeightBlockLoader()']]], - ['conv2dweightblockloadergeneral_89',['Conv2DWeightBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ad0550fabbdc9297559381a5b488e9af1',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::Conv2DWeightBlockLoaderGeneral()']]], - ['conv2dweightblockloadersmallchannels_90',['Conv2DWeightBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae1806ea1c19713819dee83a38ab35fa6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::Conv2DWeightBlockLoaderSmallChannels()']]], - ['conv3d_91',['conv3d',['../group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca',1,'mlx::core']]], - ['conv_5fgeneral_92',['conv_general',['../group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e',1,'mlx::core::conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})'],['../group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d',1,'mlx::core::conv_general(const array &input, const array &weight, std::vector< int > stride={}, std::vector< int > padding={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})']]], - ['convolution_93',['Convolution',['../classmlx_1_1core_1_1_convolution.html',1,'mlx::core::Convolution'],['../classmlx_1_1core_1_1_convolution.html#a6f1de77b719bb13217b0d8c64cabb8ef',1,'mlx::core::Convolution::Convolution()']]], - ['copy_94',['Copy',['../classmlx_1_1core_1_1_copy.html',1,'mlx::core::Copy'],['../classmlx_1_1core_1_1_copy.html#a6243e044af119105ffaaed7d405cd584',1,'mlx::core::Copy::Copy()']]], - ['copy_95',['copy',['../namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2',1,'mlx::core::copy()'],['../namespacemlx_1_1core_1_1metal.html#aa215e631e2680f04a591b88d91571719',1,'mlx::core::metal::copy()'],['../group__ops.html#gae306e93af12f774bd80bad6c231b09d6',1,'mlx::core::copy()']]], - ['copy_2eh_96',['copy.h',['../common_2copy_8h.html',1,'(Global Namespace)'],['../metal_2copy_8h.html',1,'(Global Namespace)'],['../metal_2jit_2copy_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2copy_8h.html',1,'(Global Namespace)']]], - ['copy_5fg_97',['copy_g',['../metal_2kernels_2copy_8h.html#a1ea80f8edb842dd0ed8ffe553a2238b2',1,'copy.h']]], - ['copy_5fg_5fnd_98',['copy_g_nd',['../metal_2kernels_2copy_8h.html#a07336a7a024125b4f6d3e8556d04058f',1,'copy.h']]], - ['copy_5fg_5fnd1_99',['copy_g_nd1',['../metal_2kernels_2copy_8h.html#aba4530a7db6a61ca36f50e4f5e58fb77',1,'copy.h']]], - ['copy_5fg_5fnd2_100',['copy_g_nd2',['../metal_2kernels_2copy_8h.html#aee678c7c31119f3e609685589f37490c',1,'copy.h']]], - ['copy_5fg_5fnd3_101',['copy_g_nd3',['../metal_2kernels_2copy_8h.html#a821f8f3f3891159a295c66fc25aed1ff',1,'copy.h']]], - ['copy_5fgg_102',['copy_gg',['../metal_2kernels_2copy_8h.html#a9fd1d98f0dcf976089754cd47f55784d',1,'copy.h']]], - ['copy_5fgg_5fnd_103',['copy_gg_nd',['../metal_2kernels_2copy_8h.html#a3197d34e9c68e7059419ae0635667059',1,'copy.h']]], - ['copy_5fgg_5fnd1_104',['copy_gg_nd1',['../metal_2kernels_2copy_8h.html#a3278d9c999718bee3ccbe2922f501bf1',1,'copy.h']]], - ['copy_5fgg_5fnd2_105',['copy_gg_nd2',['../metal_2kernels_2copy_8h.html#a3e2d3cc7f34f56170409b6735f51a950',1,'copy.h']]], - ['copy_5fgg_5fnd3_106',['copy_gg_nd3',['../metal_2kernels_2copy_8h.html#a59f43b5bffed936d7559ceb06a10aabd',1,'copy.h']]], - ['copy_5fgpu_107',['copy_gpu',['../namespacemlx_1_1core.html#addaa46a13ac2deb1d9ce621338320e0e',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a6a6f4e46c8fc44fdc74c50ace02bcf38',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype)']]], - ['copy_5fgpu_5finplace_108',['copy_gpu_inplace',['../namespacemlx_1_1core.html#a69e30f5d30a6d72ac0ffe4886f24b7ba',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a8e1ccb0ed9387b0a789311d9f8964803',1,'mlx::core::copy_gpu_inplace(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#ae55b801b09ccf55cba96278163a9b1ef',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const std::vector< int64_t > &istride, int64_t ioffset, CopyType ctype, const Stream &s)']]], - ['copy_5fhartley_109',['copy_hartley',['../namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], - ['copy_5finplace_110',['copy_inplace',['../namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64',1,'mlx::core::copy_inplace(const array &src, array &dst, CopyType ctype)'],['../namespacemlx_1_1core.html#aad636e2d0b2f882cadd1b438f4daa9ed',1,'mlx::core::copy_inplace(const array &src, array &dst, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)']]], - ['copy_5finput_111',['copy_input',['../namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)'],['../namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, vtype_t< T > *dst)'],['../namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, T *dst)']]], - ['copy_5fkernels_112',['copy_kernels',['../metal_2jit_2copy_8h.html#a02dd446c4670b26ee069a70e951be785',1,'copy.h']]], - ['copy_5foutput_113',['copy_output',['../namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)'],['../namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], - ['copy_5fs_114',['copy_s',['../metal_2kernels_2copy_8h.html#aef09f9b9475345b1bba121d037d222ea',1,'copy.h']]], - ['copy_5fshared_5fbuffer_115',['copy_shared_buffer',['../classmlx_1_1core_1_1array.html#a28df7a333d90a311c49bc4bce7a1ad6d',1,'mlx::core::array::copy_shared_buffer(const array &other, const std::vector< size_t > &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a92974c656c35a972ad241f80584bbd29',1,'mlx::core::array::copy_shared_buffer(const array &other)']]], - ['copy_5fv_116',['copy_v',['../metal_2kernels_2copy_8h.html#ae26a13e0c8e6c15f7b10078e65970659',1,'copy.h']]], - ['copytype_117',['CopyType',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337',1,'mlx::core']]], - ['core_20array_20operations_118',['Core array operations',['../group__ops.html',1,'']]], - ['cos_119',['Cos',['../struct_cos.html',1,'Cos'],['../classmlx_1_1core_1_1_cos.html',1,'mlx::core::Cos'],['../structmlx_1_1core_1_1detail_1_1_cos.html',1,'mlx::core::detail::Cos'],['../classmlx_1_1core_1_1_cos.html#a2acb9fcf0901462189c476756fd99995',1,'mlx::core::Cos::Cos()']]], - ['cos_120',['cos',['../namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d',1,'pocketfft::detail::cos()'],['../namespacemetal.html#a2fa4778a6fe2fa43253ea724e5a608a3',1,'metal::cos()'],['../namespacemetal_1_1fast.html#a75b6bb32fa3870eda46a7bfc9f481f88',1,'metal::fast::cos()'],['../namespacemetal_1_1precise.html#ac4941f62e7d8ab9d7cabbd967aa9f220',1,'metal::precise::cos()'],['../group__ops.html#ga39dfdf72b556012aa35ff27a94116e74',1,'mlx::core::cos()']]], - ['cosh_121',['Cosh',['../struct_cosh.html',1,'Cosh'],['../classmlx_1_1core_1_1_cosh.html',1,'mlx::core::Cosh'],['../structmlx_1_1core_1_1detail_1_1_cosh.html',1,'mlx::core::detail::Cosh'],['../classmlx_1_1core_1_1_cosh.html#a44e8ac2e09a55ec32e9dc6641eedc8f1',1,'mlx::core::Cosh::Cosh()']]], - ['cosh_122',['cosh',['../namespacemetal.html#a8a68a88cc110830d057dbd71431b93c0',1,'metal::cosh()'],['../namespacemetal_1_1fast.html#a31544ad9de28012a4ddda86e3966a77e',1,'metal::fast::cosh()'],['../namespacemetal_1_1precise.html#a72d86d508300a9b58f4ccbbe70da4fbc',1,'metal::precise::cosh()'],['../group__ops.html#ga2181b71cda88007a3092be4795ff0715',1,'mlx::core::cosh()']]], - ['cosine_123',['cosine',['../structpocketfft_1_1detail_1_1_exec_dcst.html#a185023fc1e386cc8f233b79c49c1fd8a',1,'pocketfft::detail::ExecDcst']]], - ['cospi_124',['cospi',['../namespacemetal.html#a5c2f37939ad705ddea4409d3bedb8ce1',1,'metal::cospi()'],['../namespacemetal_1_1fast.html#a9906b41f75319b384ffb570cc94d67ce',1,'metal::fast::cospi()'],['../namespacemetal_1_1precise.html#a2392b78bd196efdbbac65901c4ab20e7',1,'metal::precise::cospi()']]], - ['cost_5fguess_125',['cost_guess',['../structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0',1,'pocketfft::detail::util']]], - ['count_5fdown_126',['count_down',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342',1,'pocketfft::detail::threading::latch']]], - ['cpu_127',['cpu',['../structmlx_1_1core_1_1_device.html#a69ee81924251dec96f1945c9d91506fd',1,'mlx::core::Device::cpu'],['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdbad9747e2da342bdb995f6389533ad1a3d',1,'mlx::core::Device::cpu']]], - ['cummax_128',['CumMax',['../struct_cum_max.html',1,'']]], - ['cummax_129',['cummax',['../group__ops.html#gaee37cac8476e8f8d666bcded5bc59143',1,'mlx::core']]], - ['cummin_130',['CumMin',['../struct_cum_min.html',1,'']]], - ['cummin_131',['cummin',['../group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8',1,'mlx::core']]], - ['cumprod_132',['CumProd',['../struct_cum_prod.html',1,'']]], - ['cumprod_133',['cumprod',['../group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f',1,'mlx::core']]], - ['cumprod_3c_20bool_20_3e_134',['CumProd< bool >',['../struct_cum_prod_3_01bool_01_4.html',1,'']]], - ['cumsum_135',['CumSum',['../struct_cum_sum.html',1,'']]], - ['cumsum_136',['cumsum',['../group__ops.html#gaddc825a5c173e195ab0fda83ad630420',1,'mlx::core']]], - ['custom_137',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html',1,'mlx::core::fast::Custom'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a4186fea23f7156c38960426821fca313',1,'mlx::core::fast::Custom::Custom()']]], - ['custom_5ffunction_138',['custom_function',['../namespacemlx_1_1core.html#a58c5b448f357b48e86599eb8eeea141d',1,'mlx::core']]], - ['custom_5fvjp_139',['custom_vjp',['../namespacemlx_1_1core.html#a3fa1f0ad360f3e16c146384276b1c467',1,'mlx::core']]], - ['customtransforms_140',['CustomTransforms',['../classmlx_1_1core_1_1_custom_transforms.html',1,'mlx::core::CustomTransforms'],['../classmlx_1_1core_1_1_custom_transforms.html#ab52abadb9c6f6db83d087c7b751be488',1,'mlx::core::CustomTransforms::CustomTransforms()']]] + ['cholesky_5finv_26',['cholesky_inv',['../namespacemlx_1_1core_1_1linalg.html#aef0fe4894c5cf98792d59859c6d20511',1,'mlx::core::linalg']]], + ['clear_5fcache_27',['clear_cache',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a447c1eb38c00d2e8e521675297f4a9b1',1,'mlx::core::metal::MetalAllocator::clear_cache()'],['../namespacemlx_1_1core_1_1metal.html#a22b3384ebd17f2fca198f81b9f1b6dc3',1,'mlx::core::metal::clear_cache()']]], + ['clip_28',['clip',['../group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8',1,'mlx::core']]], + ['cmplx_29',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail::cmplx< T >'],['../structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c',1,'pocketfft::detail::cmplx::cmplx()'],['../structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a',1,'pocketfft::detail::cmplx::cmplx(T r_, T i_)']]], + ['cmplx_3c_20t0_20_3e_30',['cmplx< T0 >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], + ['cmplx_3c_20thigh_20_3e_31',['cmplx< Thigh >',['../structpocketfft_1_1detail_1_1cmplx.html',1,'pocketfft::detail']]], + ['cndarr_32',['cndarr',['../classpocketfft_1_1detail_1_1cndarr.html',1,'pocketfft::detail::cndarr< T >'],['../classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129',1,'pocketfft::detail::cndarr::cndarr()']]], + ['col_5fcontiguous_33',['col_contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#ae24709026598d635e6b5c24a15f8a802',1,'mlx::core::array::Flags']]], + ['col_5freduce_5fgeneral_34',['col_reduce_general',['../reduce__col_8h.html#a3b14333e195c0a07b70069bebf85d5c3',1,'reduce_col.h']]], + ['col_5freduce_5fgeneral_5fno_5fatomics_35',['col_reduce_general_no_atomics',['../reduce__col_8h.html#aef03a4131c710ba8f94a666f58719eb7',1,'reduce_col.h']]], + ['col_5freduce_5fsmall_36',['col_reduce_small',['../reduce__col_8h.html#a7da96ffb66e1fda27bad2852c2285b94',1,'reduce_col.h']]], + ['collapse_5fcontiguous_5fdims_37',['collapse_contiguous_dims',['../namespacemlx_1_1core.html#a9d151ba3e138be1954d2f51f85806b0c',1,'mlx::core::collapse_contiguous_dims(const std::vector< int > &shape, const std::vector< std::vector< stride_t > > strides)'],['../namespacemlx_1_1core.html#a8430e0baac3f6d8a2ab22428f9c0b7e2',1,'mlx::core::collapse_contiguous_dims(const std::vector< array > &xs)'],['../namespacemlx_1_1core.html#ac813412cce77fc1340dcfefc6e099276',1,'mlx::core::collapse_contiguous_dims(Arrays &&... xs)']]], + ['commandencoder_38',['CommandEncoder',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html',1,'mlx::core::metal::CommandEncoder'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::metal::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::metal::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete']]], + ['commit_5fcommand_5fbuffer_39',['commit_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a95248f1387824067fd4fed23ace5ac0c',1,'mlx::core::metal::Device']]], + ['commonallocator_40',['CommonAllocator',['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html',1,'mlx::core::allocator']]], + ['communication_5fstream_41',['communication_stream',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6',1,'mlx::core::distributed::detail']]], + ['compile_42',['compile',['../namespacemlx_1_1core.html#ab93149e46a6d8f3e1988123fab508dc2',1,'mlx::core::compile'],['../namespacemlx_1_1core_1_1detail.html#a4d14dee182dc9a8bea64efdaad310a1f',1,'mlx::core::detail::compile']]], + ['compile_2eh_43',['compile.h',['../compile_8h.html',1,'']]], + ['compile_5favailable_5ffor_5fdevice_44',['compile_available_for_device',['../namespacemlx_1_1core_1_1detail.html#aeeff2ba6ec3d9d4ed090de6d2681dbc2',1,'mlx::core::detail']]], + ['compile_5fclear_5fcache_45',['compile_clear_cache',['../namespacemlx_1_1core_1_1detail.html#a3fb927c209b946aefebb195993fbe4cf',1,'mlx::core::detail']]], + ['compile_5ferase_46',['compile_erase',['../namespacemlx_1_1core_1_1detail.html#a69eb76a14f845ca000f1ccb2edda0175',1,'mlx::core::detail']]], + ['compile_5fimpl_2eh_47',['compile_impl.h',['../compile__impl_8h.html',1,'']]], + ['compiled_48',['Compiled',['../classmlx_1_1core_1_1_compiled.html',1,'mlx::core::Compiled'],['../classmlx_1_1core_1_1_compiled.html#a2d8cefff835c419a48a077d306b8e051',1,'mlx::core::Compiled::Compiled()']]], + ['compiled_2eh_49',['compiled.h',['../compiled_8h.html',1,'']]], + ['compiled_5fallocate_5foutputs_50',['compiled_allocate_outputs',['../namespacemlx_1_1core.html#ab8c3c4fc05745f586de922c8266f4fce',1,'mlx::core']]], + ['compiled_5fcheck_5fcontiguity_51',['compiled_check_contiguity',['../namespacemlx_1_1core.html#a3b900ab319948c5a01a3ecd30a709027',1,'mlx::core']]], + ['compiled_5fpreamble_2eh_52',['compiled_preamble.h',['../compiled__preamble_8h.html',1,'']]], + ['compilemode_53',['CompileMode',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4',1,'mlx::core']]], + ['complex_2eh_54',['complex.h',['../backend_2metal_2kernels_2complex_8h.html',1,'(Global Namespace)'],['../types_2complex_8h.html',1,'(Global Namespace)']]], + ['complex128_5ft_55',['complex128_t',['../structmlx_1_1core_1_1complex128__t.html',1,'mlx::core::complex128_t'],['../structmlx_1_1core_1_1complex128__t.html#aa15d0b805f8790f7c7b76fc7b9d677e0',1,'mlx::core::complex128_t::complex128_t(double v, double u)'],['../structmlx_1_1core_1_1complex128__t.html#abf2842253b874f9f13f39ea68a89e5b6',1,'mlx::core::complex128_t::complex128_t(std::complex< double > v)'],['../structmlx_1_1core_1_1complex128__t.html#a526fba96d7e815360cb4226af085a1bf',1,'mlx::core::complex128_t::complex128_t(T x)']]], + ['complex64_56',['complex64',['../structmlx_1_1core_1_1_dtype.html#ade845ef5dcebead13a37fe696436e1daa8c022579455bcd2c681f007e84f4e2cf',1,'mlx::core::Dtype::complex64'],['../namespacemlx_1_1core.html#af99db87e0078bfcdb383f5689bc874d4',1,'mlx::core::complex64']]], + ['complex64_5ft_57',['complex64_t',['../structcomplex64__t.html',1,'complex64_t'],['../structmlx_1_1core_1_1complex64__t.html',1,'mlx::core::complex64_t'],['../structcomplex64__t.html#adbd392a5e92d31997380ad0a38be4be8',1,'complex64_t::complex64_t(float real, float imag)'],['../structcomplex64__t.html#a33a2452eb33b5ed53655773539c357a5',1,'complex64_t::complex64_t(T x) thread'],['../structcomplex64__t.html#a89b65ace8588b7bf215355f705eb23d9',1,'complex64_t::complex64_t(T x) threadgroup'],['../structcomplex64__t.html#ac81b486f642fb3b26c5d659917bdbcd0',1,'complex64_t::complex64_t(T x) device'],['../structcomplex64__t.html#a0a27a41206400f1e62b60ceb56960c93',1,'complex64_t::complex64_t(T x) const ant'],['../structmlx_1_1core_1_1complex64__t.html#a697cc973ae27d63c8e00d830e780bd8c',1,'mlx::core::complex64_t::complex64_t(float v, float u)'],['../structmlx_1_1core_1_1complex64__t.html#ae065e39938f9c4374b4116f4c67d4d09',1,'mlx::core::complex64_t::complex64_t(std::complex< float > v)'],['../structmlx_1_1core_1_1complex64__t.html#a2232cbbe591a9d2bc228cb23fac38b50',1,'mlx::core::complex64_t::complex64_t(T x)']]], + ['complex_5fbinop_58',['complex_binop',['../types_2complex_8h.html#a9c7995d495359894e1b30c0f1678d6bd',1,'complex.h']]], + ['complex_5fbinop_5fhelper_59',['complex_binop_helper',['../types_2complex_8h.html#ac6890f9852de12339b09b65757ebc8c4',1,'complex.h']]], + ['complex_5fmul_60',['complex_mul',['../radix_8h.html#a5bfc53b531214c9ce277bebc18aa67d6',1,'radix.h']]], + ['complex_5fmul_5fconj_61',['complex_mul_conj',['../radix_8h.html#a0e2dfd3d1dda09f47ccc64eec35629f3',1,'radix.h']]], + ['complexfloating_62',['complexfloating',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dafb203630099d501ff7c255a574bc4812',1,'mlx::core::Dtype::complexfloating'],['../namespacemlx_1_1core.html#a70b8e88c9df750af984757105af33423',1,'mlx::core::complexfloating']]], + ['compute_5fstrided_5findices_63',['compute_strided_indices',['../struct_read_writer.html#a7c903fbb8b85a856ba5564d7df537cdf',1,'ReadWriter']]], + ['concatenate_64',['Concatenate',['../classmlx_1_1core_1_1_concatenate.html',1,'mlx::core::Concatenate'],['../classmlx_1_1core_1_1_concatenate.html#acff07853de2d31faeec7c4ca40ce0888',1,'mlx::core::Concatenate::Concatenate()']]], + ['concatenate_65',['concatenate',['../group__ops.html#gabdc36fa65697d0361c8d67495de77129',1,'mlx::core::concatenate(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#gaa95c34ca3a8877f2c50cb60e7fa312b8',1,'mlx::core::concatenate(const std::vector< array > &arrays, StreamOrDevice s={})']]], + ['concatenate_5fgpu_66',['concatenate_gpu',['../namespacemlx_1_1core.html#a050299d0d366ca5c9d09d1004dcc3e7d',1,'mlx::core']]], + ['concurrent_5fqueue_67',['concurrent_queue',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], + ['concurrent_5fqueue_3c_20std_3a_3afunction_3c_20void_28_29_3e_20_3e_68',['concurrent_queue< std::function< void()> >',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html',1,'pocketfft::detail::threading']]], + ['concurrentcontext_69',['ConcurrentContext',['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html',1,'mlx::core::metal::CommandEncoder::ConcurrentContext'],['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::metal::CommandEncoder::ConcurrentContext::ConcurrentContext()']]], + ['cond_70',['cond',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4ffd524d6a5bedd1a303b63bdde6701c',1,'mlx::core::scheduler::StreamThread']]], + ['conj_71',['conj',['../namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3',1,'pocketfft::detail']]], + ['conjugate_72',['Conjugate',['../struct_conjugate.html',1,'Conjugate'],['../classmlx_1_1core_1_1_conjugate.html',1,'mlx::core::Conjugate'],['../structmlx_1_1core_1_1detail_1_1_conjugate.html',1,'mlx::core::detail::Conjugate'],['../classmlx_1_1core_1_1_conjugate.html#a627f9e6a8729fb3ffb3ca3228d007c87',1,'mlx::core::Conjugate::Conjugate()']]], + ['conjugate_73',['conjugate',['../group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23',1,'mlx::core']]], + ['contiguous_74',['contiguous',['../structmlx_1_1core_1_1array_1_1_flags.html#afd0ab11e7a486a2a8e50ee84b971ac8a',1,'mlx::core::array::Flags']]], + ['contiguous_5fscan_75',['contiguous_scan',['../kernels_2scan_8h.html#a1f9c197dc09641f3fbb2e886c85b0675',1,'scan.h']]], + ['contiguousallreduce_76',['ContiguousAllReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ae4e34c7154eb8dc47aa8503209730424',1,'mlx::core']]], + ['contiguousreduce_77',['ContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ad2547f25dffe8d8936dbec25601cfc84',1,'mlx::core']]], + ['contiguousstridedreduce_78',['ContiguousStridedReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ab48dac7508a2c790de1bdc33f29177ed',1,'mlx::core']]], + ['conv_79',['conv',['../namespacemlx_1_1core_1_1metal.html#ab1704e853394c725668c06752ebb5c24',1,'mlx::core::metal']]], + ['conv_2eh_80',['conv.h',['../conv_8h.html',1,'']]], + ['conv1d_81',['conv1d',['../group__ops.html#ga30d47e08093c03a3676f235f9f559411',1,'mlx::core']]], + ['conv2d_82',['conv2d',['../group__ops.html#ga73b02833229678786e7f302d458d5a83',1,'mlx::core']]], + ['conv2dgeneralbaseinfo_83',['Conv2DGeneralBaseInfo',['../structmlx_1_1steel_1_1_conv2_d_general_base_info.html',1,'mlx::steel']]], + ['conv2dgeneraljumpparams_84',['Conv2DGeneralJumpParams',['../structmlx_1_1steel_1_1_conv2_d_general_jump_params.html',1,'mlx::steel']]], + ['conv2dinputblockloadergeneral_85',['Conv2DInputBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html',1,'mlx::steel::Conv2DInputBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1d83af561a483432bf8dcb42e734b23b',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::Conv2DInputBlockLoaderGeneral()']]], + ['conv2dinputblockloaderlargefilter_86',['Conv2DInputBlockLoaderLargeFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8755116a535539744e4947bc69f9c50f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::Conv2DInputBlockLoaderLargeFilter()']]], + ['conv2dinputblockloadersmallchannels_87',['Conv2DInputBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ab9fd3fdeab94470dde3326f1dd5c455a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::Conv2DInputBlockLoaderSmallChannels()']]], + ['conv2dinputblockloadersmallfilter_88',['Conv2DInputBlockLoaderSmallFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0a2cbf57c51cd928722e3f06aafcf933',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::Conv2DInputBlockLoaderSmallFilter()']]], + ['conv2dweightblockloader_89',['Conv2DWeightBlockLoader',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html',1,'mlx::steel::Conv2DWeightBlockLoader< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a9a7dca3512b64cffb6eac305d795831c',1,'mlx::steel::Conv2DWeightBlockLoader::Conv2DWeightBlockLoader()']]], + ['conv2dweightblockloadergeneral_90',['Conv2DWeightBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral< T, BM, BN, BK, tgp_size, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ad0550fabbdc9297559381a5b488e9af1',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::Conv2DWeightBlockLoaderGeneral()']]], + ['conv2dweightblockloadersmallchannels_91',['Conv2DWeightBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels< T, BM, BN, BK, tgp_size, n_channels, tgp_padding >'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae1806ea1c19713819dee83a38ab35fa6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::Conv2DWeightBlockLoaderSmallChannels()']]], + ['conv3d_92',['conv3d',['../group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca',1,'mlx::core']]], + ['conv_5fgeneral_93',['conv_general',['../group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e',1,'mlx::core::conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})'],['../group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d',1,'mlx::core::conv_general(const array &input, const array &weight, std::vector< int > stride={}, std::vector< int > padding={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})']]], + ['convolution_94',['Convolution',['../classmlx_1_1core_1_1_convolution.html',1,'mlx::core::Convolution'],['../classmlx_1_1core_1_1_convolution.html#a6f1de77b719bb13217b0d8c64cabb8ef',1,'mlx::core::Convolution::Convolution()']]], + ['copy_95',['Copy',['../classmlx_1_1core_1_1_copy.html',1,'mlx::core::Copy'],['../classmlx_1_1core_1_1_copy.html#a6243e044af119105ffaaed7d405cd584',1,'mlx::core::Copy::Copy()']]], + ['copy_96',['copy',['../namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2',1,'mlx::core::copy()'],['../namespacemlx_1_1core_1_1metal.html#aa215e631e2680f04a591b88d91571719',1,'mlx::core::metal::copy()'],['../group__ops.html#gae306e93af12f774bd80bad6c231b09d6',1,'mlx::core::copy()']]], + ['copy_2eh_97',['copy.h',['../common_2copy_8h.html',1,'(Global Namespace)'],['../metal_2copy_8h.html',1,'(Global Namespace)'],['../metal_2jit_2copy_8h.html',1,'(Global Namespace)'],['../metal_2kernels_2copy_8h.html',1,'(Global Namespace)']]], + ['copy_5fg_98',['copy_g',['../metal_2kernels_2copy_8h.html#a1ea80f8edb842dd0ed8ffe553a2238b2',1,'copy.h']]], + ['copy_5fg_5fnd_99',['copy_g_nd',['../metal_2kernels_2copy_8h.html#a07336a7a024125b4f6d3e8556d04058f',1,'copy.h']]], + ['copy_5fg_5fnd1_100',['copy_g_nd1',['../metal_2kernels_2copy_8h.html#aba4530a7db6a61ca36f50e4f5e58fb77',1,'copy.h']]], + ['copy_5fg_5fnd2_101',['copy_g_nd2',['../metal_2kernels_2copy_8h.html#aee678c7c31119f3e609685589f37490c',1,'copy.h']]], + ['copy_5fg_5fnd3_102',['copy_g_nd3',['../metal_2kernels_2copy_8h.html#a821f8f3f3891159a295c66fc25aed1ff',1,'copy.h']]], + ['copy_5fgg_103',['copy_gg',['../metal_2kernels_2copy_8h.html#a9fd1d98f0dcf976089754cd47f55784d',1,'copy.h']]], + ['copy_5fgg_5fnd_104',['copy_gg_nd',['../metal_2kernels_2copy_8h.html#a3197d34e9c68e7059419ae0635667059',1,'copy.h']]], + ['copy_5fgg_5fnd1_105',['copy_gg_nd1',['../metal_2kernels_2copy_8h.html#a3278d9c999718bee3ccbe2922f501bf1',1,'copy.h']]], + ['copy_5fgg_5fnd2_106',['copy_gg_nd2',['../metal_2kernels_2copy_8h.html#a3e2d3cc7f34f56170409b6735f51a950',1,'copy.h']]], + ['copy_5fgg_5fnd3_107',['copy_gg_nd3',['../metal_2kernels_2copy_8h.html#a59f43b5bffed936d7559ceb06a10aabd',1,'copy.h']]], + ['copy_5fgpu_108',['copy_gpu',['../namespacemlx_1_1core.html#addaa46a13ac2deb1d9ce621338320e0e',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a6a6f4e46c8fc44fdc74c50ace02bcf38',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype)']]], + ['copy_5fgpu_5finplace_109',['copy_gpu_inplace',['../namespacemlx_1_1core.html#a69e30f5d30a6d72ac0ffe4886f24b7ba',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a8e1ccb0ed9387b0a789311d9f8964803',1,'mlx::core::copy_gpu_inplace(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#ae55b801b09ccf55cba96278163a9b1ef',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const std::vector< int64_t > &istride, int64_t ioffset, CopyType ctype, const Stream &s)']]], + ['copy_5fhartley_110',['copy_hartley',['../namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], + ['copy_5finplace_111',['copy_inplace',['../namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64',1,'mlx::core::copy_inplace(const array &src, array &dst, CopyType ctype)'],['../namespacemlx_1_1core.html#aad636e2d0b2f882cadd1b438f4daa9ed',1,'mlx::core::copy_inplace(const array &src, array &dst, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)']]], + ['copy_5finput_112',['copy_input',['../namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)'],['../namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, vtype_t< T > *dst)'],['../namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, T *dst)']]], + ['copy_5fkernels_113',['copy_kernels',['../metal_2jit_2copy_8h.html#a02dd446c4670b26ee069a70e951be785',1,'copy.h']]], + ['copy_5foutput_114',['copy_output',['../namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)'],['../namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], + ['copy_5fs_115',['copy_s',['../metal_2kernels_2copy_8h.html#aef09f9b9475345b1bba121d037d222ea',1,'copy.h']]], + ['copy_5fs2_116',['copy_s2',['../metal_2kernels_2copy_8h.html#a8023e9335cc5334847a8d315042be3a3',1,'copy.h']]], + ['copy_5fshared_5fbuffer_117',['copy_shared_buffer',['../classmlx_1_1core_1_1array.html#a28df7a333d90a311c49bc4bce7a1ad6d',1,'mlx::core::array::copy_shared_buffer(const array &other, const std::vector< size_t > &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a92974c656c35a972ad241f80584bbd29',1,'mlx::core::array::copy_shared_buffer(const array &other)']]], + ['copy_5fv_118',['copy_v',['../metal_2kernels_2copy_8h.html#ae26a13e0c8e6c15f7b10078e65970659',1,'copy.h']]], + ['copy_5fv2_119',['copy_v2',['../metal_2kernels_2copy_8h.html#aee14a5326f53d9b30b0b38e27d180ef3',1,'copy.h']]], + ['copytype_120',['CopyType',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337',1,'mlx::core']]], + ['core_20array_20operations_121',['Core array operations',['../group__ops.html',1,'']]], + ['cos_122',['Cos',['../struct_cos.html',1,'Cos'],['../classmlx_1_1core_1_1_cos.html',1,'mlx::core::Cos'],['../structmlx_1_1core_1_1detail_1_1_cos.html',1,'mlx::core::detail::Cos'],['../classmlx_1_1core_1_1_cos.html#a2acb9fcf0901462189c476756fd99995',1,'mlx::core::Cos::Cos()']]], + ['cos_123',['cos',['../namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d',1,'pocketfft::detail::cos()'],['../namespacemetal.html#a2fa4778a6fe2fa43253ea724e5a608a3',1,'metal::cos()'],['../namespacemetal_1_1fast.html#a75b6bb32fa3870eda46a7bfc9f481f88',1,'metal::fast::cos()'],['../namespacemetal_1_1precise.html#ac4941f62e7d8ab9d7cabbd967aa9f220',1,'metal::precise::cos()'],['../group__ops.html#ga39dfdf72b556012aa35ff27a94116e74',1,'mlx::core::cos()']]], + ['cosh_124',['Cosh',['../struct_cosh.html',1,'Cosh'],['../classmlx_1_1core_1_1_cosh.html',1,'mlx::core::Cosh'],['../structmlx_1_1core_1_1detail_1_1_cosh.html',1,'mlx::core::detail::Cosh'],['../classmlx_1_1core_1_1_cosh.html#a44e8ac2e09a55ec32e9dc6641eedc8f1',1,'mlx::core::Cosh::Cosh()']]], + ['cosh_125',['cosh',['../namespacemetal.html#a8a68a88cc110830d057dbd71431b93c0',1,'metal::cosh()'],['../namespacemetal_1_1fast.html#a31544ad9de28012a4ddda86e3966a77e',1,'metal::fast::cosh()'],['../namespacemetal_1_1precise.html#a72d86d508300a9b58f4ccbbe70da4fbc',1,'metal::precise::cosh()'],['../group__ops.html#ga2181b71cda88007a3092be4795ff0715',1,'mlx::core::cosh()']]], + ['cosine_126',['cosine',['../structpocketfft_1_1detail_1_1_exec_dcst.html#a185023fc1e386cc8f233b79c49c1fd8a',1,'pocketfft::detail::ExecDcst']]], + ['cospi_127',['cospi',['../namespacemetal.html#a5c2f37939ad705ddea4409d3bedb8ce1',1,'metal::cospi()'],['../namespacemetal_1_1fast.html#a9906b41f75319b384ffb570cc94d67ce',1,'metal::fast::cospi()'],['../namespacemetal_1_1precise.html#a2392b78bd196efdbbac65901c4ab20e7',1,'metal::precise::cospi()']]], + ['cost_5fguess_128',['cost_guess',['../structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0',1,'pocketfft::detail::util']]], + ['count_5fdown_129',['count_down',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342',1,'pocketfft::detail::threading::latch']]], + ['cpu_130',['cpu',['../structmlx_1_1core_1_1_device.html#a69ee81924251dec96f1945c9d91506fd',1,'mlx::core::Device::cpu'],['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdbad9747e2da342bdb995f6389533ad1a3d',1,'mlx::core::Device::cpu']]], + ['cummax_131',['CumMax',['../struct_cum_max.html',1,'']]], + ['cummax_132',['cummax',['../group__ops.html#gaee37cac8476e8f8d666bcded5bc59143',1,'mlx::core']]], + ['cummin_133',['CumMin',['../struct_cum_min.html',1,'']]], + ['cummin_134',['cummin',['../group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8',1,'mlx::core']]], + ['cumprod_135',['CumProd',['../struct_cum_prod.html',1,'']]], + ['cumprod_136',['cumprod',['../group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f',1,'mlx::core']]], + ['cumprod_3c_20bool_20_3e_137',['CumProd< bool >',['../struct_cum_prod_3_01bool_01_4.html',1,'']]], + ['cumsum_138',['CumSum',['../struct_cum_sum.html',1,'']]], + ['cumsum_139',['cumsum',['../group__ops.html#gaddc825a5c173e195ab0fda83ad630420',1,'mlx::core']]], + ['custom_140',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html',1,'mlx::core::fast::Custom'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a4186fea23f7156c38960426821fca313',1,'mlx::core::fast::Custom::Custom()']]], + ['custom_5ffunction_141',['custom_function',['../namespacemlx_1_1core.html#a58c5b448f357b48e86599eb8eeea141d',1,'mlx::core']]], + ['custom_5fvjp_142',['custom_vjp',['../namespacemlx_1_1core.html#a3fa1f0ad360f3e16c146384276b1c467',1,'mlx::core']]], + ['customtransforms_143',['CustomTransforms',['../classmlx_1_1core_1_1_custom_transforms.html',1,'mlx::core::CustomTransforms'],['../classmlx_1_1core_1_1_custom_transforms.html#ab52abadb9c6f6db83d087c7b751be488',1,'mlx::core::CustomTransforms::CustomTransforms()']]] ]; diff --git a/docs/build/html/search/all_4.js b/docs/build/html/search/all_4.js index a8b5f68dcf..201433edba 100644 --- a/docs/build/html/search/all_4.js +++ b/docs/build/html/search/all_4.js @@ -6,62 +6,55 @@ var searchData= ['data_3',['Data',['../structmlx_1_1core_1_1array_1_1_data.html#a77e2ea35fac1d54e4062468a432e1482',1,'mlx::core::array::Data::Data(allocator::Buffer buffer, deleter_t d=allocator::free)'],['../structmlx_1_1core_1_1array_1_1_data.html#a50f242040b123052e48e18c244ff70fc',1,'mlx::core::array::Data::Data(const Data &d)=delete']]], ['data_5fshared_5fptr_4',['data_shared_ptr',['../classmlx_1_1core_1_1array.html#ab84c792117e29cdf90ef3433303f6141',1,'mlx::core::array']]], ['data_5fsize_5',['data_size',['../classmlx_1_1core_1_1array.html#afaf2a370fa35d96af1b27a4b814e3bfd',1,'mlx::core::array']]], - ['datatype_6',['DataType',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288',1,'MPS::DataType'],['../namespace_m_p_s.html#a96effe6e5c9d287c6b86491509c78be2',1,'MPS::DataType']]], - ['datatypealternateencodingbit_7',['DataTypeAlternateEncodingBit',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a9681e9100f89bbc0484c4195ff1cef03',1,'MPS']]], - ['datatypebfloat16_8',['DataTypeBFloat16',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a39c004722f824f460d83cca680b242ca',1,'MPS']]], - ['datatypefloat16_9',['DataTypeFloat16',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a4fb70692e99c4ff2dc6028275e5a3868',1,'MPS']]], - ['datatypefloat32_10',['DataTypeFloat32',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a2156c04e979b8d8bd39d4ae17d53ded8',1,'MPS']]], - ['datatypefloatbit_11',['DataTypeFloatBit',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a330c9014eb076d89dae3d69f3b49c04a',1,'MPS']]], - ['dct_12',['dct',['../namespacepocketfft_1_1detail.html#a60615f5b685314c658346c309d5ef2ba',1,'pocketfft::detail']]], - ['deallocate_13',['deallocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a2a99b8e296d26b255e9937ba5f30e76f',1,'pocketfft::detail::threading::aligned_allocator']]], - ['decompose_5fhadamard_14',['decompose_hadamard',['../namespacemlx_1_1core.html#a3a8fe7ba84714dbb5fdc81e93a07abc8',1,'mlx::core']]], - ['default_5f_15',['default_',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#ab5993daeed822c6b970caddab7e3fd90',1,'mlx::core::random::KeySequence']]], - ['default_5fdevice_16',['default_device',['../namespacemlx_1_1core.html#a0196171cfe6ee2953113abce597dc815',1,'mlx::core']]], - ['default_5fstream_17',['default_stream',['../namespacemlx_1_1core.html#ac198b7e282957c724c84a435e8f1215e',1,'mlx::core']]], - ['defaultcontiguousreduce_18',['DefaultContiguousReduce',['../structmlx_1_1core_1_1_default_contiguous_reduce.html',1,'mlx::core::DefaultContiguousReduce< T, U, Op >'],['../structmlx_1_1core_1_1_default_contiguous_reduce.html#aeb4fb7fa1a4c8e7d1da1f450ce95c57f',1,'mlx::core::DefaultContiguousReduce::DefaultContiguousReduce()']]], - ['defaultstridedreduce_19',['DefaultStridedReduce',['../structmlx_1_1core_1_1_default_strided_reduce.html',1,'mlx::core::DefaultStridedReduce< T, U, Op >'],['../structmlx_1_1core_1_1_default_strided_reduce.html#a477e5dd0dd33071e48769d11d19a13fb',1,'mlx::core::DefaultStridedReduce::DefaultStridedReduce()']]], - ['define_5fdefault_5fis_5fequivalent_20',['DEFINE_DEFAULT_IS_EQUIVALENT',['../primitives_8h.html#a0fb9d19207dc4869aca35abfbdf4d70a',1,'primitives.h']]], - ['define_5fgrads_21',['DEFINE_GRADS',['../primitives_8h.html#a77abdcb55bc2eb0f9a45edc5ee639bf6',1,'primitives.h']]], - ['define_5finput_5foutput_5fshape_22',['DEFINE_INPUT_OUTPUT_SHAPE',['../primitives_8h.html#a649a06267b75e007224ea4ddefedb999',1,'primitives.h']]], - ['define_5fprint_23',['DEFINE_PRINT',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a8af1e90d4aa56f31ec40ad152ebd2421',1,'mlx::core::distributed::AllGather::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae6eea81b5e3789c2f6f376cc07f0a47c',1,'mlx::core::fast::RMSNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a9895733eab845e11484d86cf6ecedced',1,'mlx::core::fast::RMSNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a467fcf02b3ddf1d8b6d476b244ae3568',1,'mlx::core::fast::LayerNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a5ab3eb5402c7e8060916056eb2b7887f',1,'mlx::core::fast::LayerNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a2b06fe64fa8feca65140632087065e16',1,'mlx::core::fast::RoPE::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6cc2092fa5b8e7585921b8e0f3ec3db7',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_PRINT()'],['../primitives_8h.html#a1d3a37af519e16f6a703b1e9ebd0f592',1,'DEFINE_PRINT: primitives.h']]], - ['define_5fvmap_24',['DEFINE_VMAP',['../primitives_8h.html#adc0fbd79fe0d1114dc85da4ed99798bd',1,'primitives.h']]], - ['defines_2eh_25',['defines.h',['../defines_8h.html',1,'(Global Namespace)'],['../steel_2defines_8h.html',1,'(Global Namespace)']]], - ['degrees_26',['degrees',['../group__ops.html#ga3a70569b50e1083c5ded199d73fb960c',1,'mlx::core']]], - ['deleter_5ft_27',['deleter_t',['../namespacemlx_1_1core.html#a1e6cec03ebd80fd2d6b12b288367bfa8',1,'mlx::core']]], - ['denorm_5fmin_28',['denorm_min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a6a9dbcba4dd79cad50876dda506b9eed',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['depends_29',['Depends',['../classmlx_1_1core_1_1_depends.html',1,'mlx::core::Depends'],['../classmlx_1_1core_1_1_depends.html#a4ccb792c99f5d8d133d3fac29f7d3f62',1,'mlx::core::Depends::Depends()']]], - ['depends_30',['depends',['../group__ops.html#gac4a51a68fbe1725436b026d2fbb95759',1,'mlx::core']]], - ['dequantize_31',['dequantize',['../quantized_8h.html#aecff265b63566d0d5689cfc4e5b037d2',1,'dequantize(): quantized.h'],['../group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077',1,'mlx::core::dequantize()']]], - ['detach_32',['detach',['../classmlx_1_1core_1_1array.html#a84948c29df8c957904919c8602692bd2',1,'mlx::core::array']]], - ['device_33',['Device',['../structmlx_1_1core_1_1_device.html',1,'mlx::core::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html',1,'mlx::core::metal::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html#ae0db74570eb4b19d8cf19774db91bfd6',1,'mlx::core::metal::Device::Device()'],['../classmlx_1_1core_1_1metal_1_1_device.html#abf59a4addb5473f9e814e3651ba85f06',1,'mlx::core::metal::Device::Device(const Device &)=delete'],['../structmlx_1_1core_1_1_device.html#a481ccfb94d689994396bd353e966b489',1,'mlx::core::Device::Device()']]], - ['device_34',['device',['../structmlx_1_1core_1_1_stream.html#a406b1b0162287a4162fab1f70e2ff3bb',1,'mlx::core::Stream::device'],['../class_m_p_s_1_1_kernel.html#a46793d1c80a4df313c067103fcae6024',1,'MPS::Kernel::device()'],['../classmlx_1_1core_1_1_primitive.html#a8ae61e3289c4134232a69295268f8261',1,'mlx::core::Primitive::device()'],['../namespacemlx_1_1core_1_1metal.html#a910797b74824e6ee576fbb533dee8b57',1,'mlx::core::metal::device()']]], - ['device_2eh_35',['device.h',['../backend_2metal_2device_8h.html',1,'(Global Namespace)'],['../device_8h.html',1,'(Global Namespace)']]], - ['device_5finfo_36',['device_info',['../namespacemlx_1_1core_1_1metal.html#a6ad19c44efabb7423f973407926ead61',1,'mlx::core::metal']]], - ['devicetype_37',['DeviceType',['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdb',1,'mlx::core::Device']]], - ['diag_38',['diag',['../group__ops.html#ga11af511875640e1fa88e0ca87e199344',1,'mlx::core']]], - ['diagonal_39',['diagonal',['../group__ops.html#ga9236b085a88ead3128ed8079d009cac6',1,'mlx::core']]], - ['difference_5ftype_40',['difference_type',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#adcee44c77980fc2370a2c31e203aead5',1,'mlx::core::array::ArrayIterator']]], - ['digits_41',['digits',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#af6a681edff230c8d734a1feefb8d1879',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['digits10_42',['digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a0f48dd0c8a2d2dfa825067fb212b2e6b',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['disable_5fcompile_43',['disable_compile',['../namespacemlx_1_1core.html#a5f5fea955057bb3842b271b037909e66',1,'mlx::core']]], - ['disabled_44',['disabled',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a075ae3d2fc31640504f814f60e5ef713',1,'mlx::core']]], - ['dispatchthreadgroups_45',['dispatchThreadgroups',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a74bcd8e35f80f5a62db48c4a2bb0173e',1,'mlx::core::metal::CommandEncoder']]], - ['dispatchthreads_46',['dispatchThreads',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a1e41477f2f489e38499f7830a91c9810',1,'mlx::core::metal::CommandEncoder']]], - ['distprimitive_47',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html',1,'mlx::core::distributed::DistPrimitive'],['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a9e4b0b92776aca0c1331450ae4955cef',1,'mlx::core::distributed::DistPrimitive::DistPrimitive()']]], - ['distributed_2eh_48',['distributed.h',['../distributed_8h.html',1,'']]], - ['divide_49',['Divide',['../struct_divide.html',1,'Divide'],['../structmlx_1_1core_1_1detail_1_1_divide.html',1,'mlx::core::detail::Divide'],['../classmlx_1_1core_1_1_divide.html',1,'mlx::core::Divide'],['../classmlx_1_1core_1_1_divide.html#a62fc71e8998be65ff18285dbbd21eedb',1,'mlx::core::Divide::Divide()']]], - ['divide_50',['divide',['../namespacemetal.html#a2aea493fc1a874970b77ed0031e965df',1,'metal::divide()'],['../namespacemetal_1_1fast.html#ae70bc2185e4649369cf7b15f5e1d48be',1,'metal::fast::divide()'],['../namespacemetal_1_1precise.html#aec0982cdb96a08b61f51129150d82e9d',1,'metal::precise::divide()'],['../group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f',1,'mlx::core::divide()']]], - ['divmod_51',['DivMod',['../struct_div_mod.html',1,'DivMod'],['../classmlx_1_1core_1_1_div_mod.html',1,'mlx::core::DivMod'],['../classmlx_1_1core_1_1_div_mod.html#a859e3b6149cdceab1c7ccfd2246fb826',1,'mlx::core::DivMod::DivMod()']]], - ['divmod_52',['divmod',['../group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894',1,'mlx::core']]], - ['do_5faxpby_53',['do_axpby',['../steel__gemm__fused_8h.html#a703f06c849c89c37af7b1d27b0804a29',1,'steel_gemm_fused.h']]], - ['do_5fgather_54',['do_gather',['../steel__gemm__fused_8h.html#a60efac3ac3b7cd64d096bbae38a3ac69',1,'steel_gemm_fused.h']]], - ['do_5fread_55',['do_read',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a13eb86acf6abe288c19645935a47d2ad',1,'mlx::steel::Conv2DWeightBlockLoader::do_read'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a640155880483e1042ec5f647b9adaac6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::do_read']]], - ['dst_56',['dst',['../struct_quantized_block_loader.html#a9857214690fe6abad0e19d1045152f83',1,'QuantizedBlockLoader::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ae048eb79f8b8d98f0fe8805c30fbb09f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8598bf23a2bce6af13c876cbfa76449f',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aea6494838175225d02cbc7768a646ec7',1,'mlx::steel::Conv2DWeightBlockLoader::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a59a4fffc1dc2f3fadfb3fdd1b886da70',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a24e20e4c1dd1ebf9534bfa2b3e050ed3',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aa84c4ad43a5defb83ba1a5f49a7adb2a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8474daf268013e138a84fc1c4bff7352',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::dst'],['../structmlx_1_1steel_1_1_block_loader.html#af34c184a19846e4b40ba54b2946589ec',1,'mlx::steel::BlockLoader::dst'],['../namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::detail::dst()']]], - ['dst_5fld_57',['dst_ld',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a91192d512e7a18c2d16a139065000959',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a9e59da7e4436e61b2d3c3f982355910b',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a0ff5a6d503e0bbac4634030a75ab818d',1,'mlx::steel::Conv2DWeightBlockLoader::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ae71570942c7b0ad8e67c62662b336c4a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ac18eeebea26cc6da434ead6eb4397350',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a07c85eab8cbf7b02c60df29cf32031ef',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aae121ca6016fc6c7255027b3641f3a09',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::dst_ld']]], - ['dtype_58',['Dtype',['../structmlx_1_1core_1_1_dtype.html',1,'mlx::core']]], - ['dtype_59',['dtype',['../classmlx_1_1core_1_1array.html#ae29e7d6fbfbea1e5e321a8d1ea3cfacd',1,'mlx::core::array']]], - ['dtype_60',['Dtype',['../structmlx_1_1core_1_1_dtype.html#aec17f0a4a51729e5ac40b62f0aa765d1',1,'mlx::core::Dtype']]], - ['dtype_2eh_61',['dtype.h',['../dtype_8h.html',1,'']]], - ['dtype_5ffrom_5farray_5fprotocol_62',['dtype_from_array_protocol',['../namespacemlx_1_1core.html#a14e0dfe96b41a04be91553799eb07594',1,'mlx::core']]], - ['dtype_5fto_5farray_5fprotocol_63',['dtype_to_array_protocol',['../namespacemlx_1_1core.html#a80509820694d2f74f0d262d13a90774b',1,'mlx::core']]] + ['dct_6',['dct',['../namespacepocketfft_1_1detail.html#a60615f5b685314c658346c309d5ef2ba',1,'pocketfft::detail']]], + ['deallocate_7',['deallocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a2a99b8e296d26b255e9937ba5f30e76f',1,'pocketfft::detail::threading::aligned_allocator']]], + ['debug_5fset_5fprimitive_5fbuffer_5flabel_8',['debug_set_primitive_buffer_label',['../namespacemlx_1_1core.html#a489e45b3a5cd8b46e8ea56b9132eb230',1,'mlx::core']]], + ['debug_5fset_5fstream_5fqueue_5flabel_9',['debug_set_stream_queue_label',['../namespacemlx_1_1core.html#a79817d2432e782e596c9c49a08b93be2',1,'mlx::core']]], + ['decompose_5fhadamard_10',['decompose_hadamard',['../namespacemlx_1_1core.html#a3a8fe7ba84714dbb5fdc81e93a07abc8',1,'mlx::core']]], + ['default_5f_11',['default_',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#ab5993daeed822c6b970caddab7e3fd90',1,'mlx::core::random::KeySequence']]], + ['default_5fdevice_12',['default_device',['../namespacemlx_1_1core.html#a0196171cfe6ee2953113abce597dc815',1,'mlx::core']]], + ['default_5fstream_13',['default_stream',['../namespacemlx_1_1core.html#ac198b7e282957c724c84a435e8f1215e',1,'mlx::core']]], + ['defaultcontiguousreduce_14',['DefaultContiguousReduce',['../structmlx_1_1core_1_1_default_contiguous_reduce.html',1,'mlx::core::DefaultContiguousReduce< T, U, Op >'],['../structmlx_1_1core_1_1_default_contiguous_reduce.html#aeb4fb7fa1a4c8e7d1da1f450ce95c57f',1,'mlx::core::DefaultContiguousReduce::DefaultContiguousReduce()']]], + ['defaultstridedreduce_15',['DefaultStridedReduce',['../structmlx_1_1core_1_1_default_strided_reduce.html',1,'mlx::core::DefaultStridedReduce< T, U, Op >'],['../structmlx_1_1core_1_1_default_strided_reduce.html#a477e5dd0dd33071e48769d11d19a13fb',1,'mlx::core::DefaultStridedReduce::DefaultStridedReduce()']]], + ['define_5fdefault_5fis_5fequivalent_16',['DEFINE_DEFAULT_IS_EQUIVALENT',['../primitives_8h.html#a0fb9d19207dc4869aca35abfbdf4d70a',1,'primitives.h']]], + ['define_5fgrads_17',['DEFINE_GRADS',['../primitives_8h.html#a77abdcb55bc2eb0f9a45edc5ee639bf6',1,'primitives.h']]], + ['define_5finput_5foutput_5fshape_18',['DEFINE_INPUT_OUTPUT_SHAPE',['../primitives_8h.html#a649a06267b75e007224ea4ddefedb999',1,'primitives.h']]], + ['define_5fprint_19',['DEFINE_PRINT',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a8af1e90d4aa56f31ec40ad152ebd2421',1,'mlx::core::distributed::AllGather::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae6eea81b5e3789c2f6f376cc07f0a47c',1,'mlx::core::fast::RMSNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a9895733eab845e11484d86cf6ecedced',1,'mlx::core::fast::RMSNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a467fcf02b3ddf1d8b6d476b244ae3568',1,'mlx::core::fast::LayerNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a5ab3eb5402c7e8060916056eb2b7887f',1,'mlx::core::fast::LayerNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a2b06fe64fa8feca65140632087065e16',1,'mlx::core::fast::RoPE::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6cc2092fa5b8e7585921b8e0f3ec3db7',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a4b8f1b1f633002c8ca6fa8f0ef4dd587',1,'mlx::core::fast::AffineQuantize::DEFINE_PRINT()'],['../primitives_8h.html#a1d3a37af519e16f6a703b1e9ebd0f592',1,'DEFINE_PRINT: primitives.h']]], + ['define_5fvmap_20',['DEFINE_VMAP',['../primitives_8h.html#adc0fbd79fe0d1114dc85da4ed99798bd',1,'primitives.h']]], + ['defines_2eh_21',['defines.h',['../defines_8h.html',1,'(Global Namespace)'],['../steel_2defines_8h.html',1,'(Global Namespace)']]], + ['degrees_22',['degrees',['../group__ops.html#ga3a70569b50e1083c5ded199d73fb960c',1,'mlx::core']]], + ['deleter_5ft_23',['deleter_t',['../namespacemlx_1_1core.html#a1e6cec03ebd80fd2d6b12b288367bfa8',1,'mlx::core']]], + ['denorm_5fmin_24',['denorm_min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a6a9dbcba4dd79cad50876dda506b9eed',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['depends_25',['Depends',['../classmlx_1_1core_1_1_depends.html',1,'mlx::core::Depends'],['../classmlx_1_1core_1_1_depends.html#a4ccb792c99f5d8d133d3fac29f7d3f62',1,'mlx::core::Depends::Depends()']]], + ['depends_26',['depends',['../group__ops.html#gac4a51a68fbe1725436b026d2fbb95759',1,'mlx::core']]], + ['dequantize_27',['dequantize',['../quantized_8h.html#aecff265b63566d0d5689cfc4e5b037d2',1,'dequantize(): quantized.h'],['../group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077',1,'mlx::core::dequantize()']]], + ['detach_28',['detach',['../classmlx_1_1core_1_1array.html#a84948c29df8c957904919c8602692bd2',1,'mlx::core::array']]], + ['device_29',['Device',['../structmlx_1_1core_1_1_device.html',1,'mlx::core::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html',1,'mlx::core::metal::Device'],['../classmlx_1_1core_1_1metal_1_1_device.html#ae0db74570eb4b19d8cf19774db91bfd6',1,'mlx::core::metal::Device::Device()'],['../classmlx_1_1core_1_1metal_1_1_device.html#abf59a4addb5473f9e814e3651ba85f06',1,'mlx::core::metal::Device::Device(const Device &)=delete'],['../structmlx_1_1core_1_1_device.html#a481ccfb94d689994396bd353e966b489',1,'mlx::core::Device::Device()']]], + ['device_30',['device',['../structmlx_1_1core_1_1_stream.html#a406b1b0162287a4162fab1f70e2ff3bb',1,'mlx::core::Stream::device'],['../classmlx_1_1core_1_1_primitive.html#a8ae61e3289c4134232a69295268f8261',1,'mlx::core::Primitive::device()'],['../namespacemlx_1_1core_1_1metal.html#a910797b74824e6ee576fbb533dee8b57',1,'mlx::core::metal::device()']]], + ['device_2eh_31',['device.h',['../backend_2metal_2device_8h.html',1,'(Global Namespace)'],['../device_8h.html',1,'(Global Namespace)']]], + ['device_5finfo_32',['device_info',['../namespacemlx_1_1core_1_1metal.html#a6ad19c44efabb7423f973407926ead61',1,'mlx::core::metal']]], + ['devicetype_33',['DeviceType',['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdb',1,'mlx::core::Device']]], + ['diag_34',['diag',['../group__ops.html#ga11af511875640e1fa88e0ca87e199344',1,'mlx::core']]], + ['diagonal_35',['diagonal',['../group__ops.html#ga9236b085a88ead3128ed8079d009cac6',1,'mlx::core']]], + ['difference_5ftype_36',['difference_type',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#adcee44c77980fc2370a2c31e203aead5',1,'mlx::core::array::ArrayIterator']]], + ['digits_37',['digits',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#af6a681edff230c8d734a1feefb8d1879',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['digits10_38',['digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a0f48dd0c8a2d2dfa825067fb212b2e6b',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['disable_5fcompile_39',['disable_compile',['../namespacemlx_1_1core.html#a5f5fea955057bb3842b271b037909e66',1,'mlx::core']]], + ['disabled_40',['disabled',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a075ae3d2fc31640504f814f60e5ef713',1,'mlx::core']]], + ['dispatchthreadgroups_41',['dispatchThreadgroups',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a74bcd8e35f80f5a62db48c4a2bb0173e',1,'mlx::core::metal::CommandEncoder']]], + ['dispatchthreads_42',['dispatchThreads',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a1e41477f2f489e38499f7830a91c9810',1,'mlx::core::metal::CommandEncoder']]], + ['distprimitive_43',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html',1,'mlx::core::distributed::DistPrimitive'],['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a9e4b0b92776aca0c1331450ae4955cef',1,'mlx::core::distributed::DistPrimitive::DistPrimitive()']]], + ['distributed_2eh_44',['distributed.h',['../distributed_8h.html',1,'']]], + ['divide_45',['Divide',['../struct_divide.html',1,'Divide'],['../structmlx_1_1core_1_1detail_1_1_divide.html',1,'mlx::core::detail::Divide'],['../classmlx_1_1core_1_1_divide.html',1,'mlx::core::Divide'],['../classmlx_1_1core_1_1_divide.html#a62fc71e8998be65ff18285dbbd21eedb',1,'mlx::core::Divide::Divide()']]], + ['divide_46',['divide',['../namespacemetal.html#a2aea493fc1a874970b77ed0031e965df',1,'metal::divide()'],['../namespacemetal_1_1fast.html#ae70bc2185e4649369cf7b15f5e1d48be',1,'metal::fast::divide()'],['../namespacemetal_1_1precise.html#aec0982cdb96a08b61f51129150d82e9d',1,'metal::precise::divide()'],['../group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f',1,'mlx::core::divide()']]], + ['divmod_47',['DivMod',['../struct_div_mod.html',1,'DivMod'],['../classmlx_1_1core_1_1_div_mod.html',1,'mlx::core::DivMod'],['../classmlx_1_1core_1_1_div_mod.html#a859e3b6149cdceab1c7ccfd2246fb826',1,'mlx::core::DivMod::DivMod()']]], + ['divmod_48',['divmod',['../group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894',1,'mlx::core']]], + ['do_5faxpby_49',['do_axpby',['../steel__gemm__fused_8h.html#a703f06c849c89c37af7b1d27b0804a29',1,'steel_gemm_fused.h']]], + ['do_5fgather_50',['do_gather',['../steel__gemm__fused_8h.html#a60efac3ac3b7cd64d096bbae38a3ac69',1,'steel_gemm_fused.h']]], + ['do_5fread_51',['do_read',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a13eb86acf6abe288c19645935a47d2ad',1,'mlx::steel::Conv2DWeightBlockLoader::do_read'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a640155880483e1042ec5f647b9adaac6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::do_read']]], + ['dst_52',['dst',['../struct_quantized_block_loader.html#a9857214690fe6abad0e19d1045152f83',1,'QuantizedBlockLoader::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ae048eb79f8b8d98f0fe8805c30fbb09f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8598bf23a2bce6af13c876cbfa76449f',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aea6494838175225d02cbc7768a646ec7',1,'mlx::steel::Conv2DWeightBlockLoader::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a59a4fffc1dc2f3fadfb3fdd1b886da70',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a24e20e4c1dd1ebf9534bfa2b3e050ed3',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::dst'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aa84c4ad43a5defb83ba1a5f49a7adb2a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::dst'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8474daf268013e138a84fc1c4bff7352',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::dst'],['../structmlx_1_1steel_1_1_block_loader.html#af34c184a19846e4b40ba54b2946589ec',1,'mlx::steel::BlockLoader::dst'],['../namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::detail::dst()']]], + ['dst_5fld_53',['dst_ld',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a91192d512e7a18c2d16a139065000959',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a9e59da7e4436e61b2d3c3f982355910b',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a0ff5a6d503e0bbac4634030a75ab818d',1,'mlx::steel::Conv2DWeightBlockLoader::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ae71570942c7b0ad8e67c62662b336c4a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ac18eeebea26cc6da434ead6eb4397350',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a07c85eab8cbf7b02c60df29cf32031ef',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::dst_ld'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aae121ca6016fc6c7255027b3641f3a09',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::dst_ld']]], + ['dtype_54',['Dtype',['../structmlx_1_1core_1_1_dtype.html',1,'mlx::core::Dtype'],['../structmlx_1_1core_1_1_dtype.html#aec17f0a4a51729e5ac40b62f0aa765d1',1,'mlx::core::Dtype::Dtype()']]], + ['dtype_55',['dtype',['../classmlx_1_1core_1_1array.html#ae29e7d6fbfbea1e5e321a8d1ea3cfacd',1,'mlx::core::array']]], + ['dtype_2eh_56',['dtype.h',['../dtype_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/all_5.js b/docs/build/html/search/all_5.js index 7b178b1678..ca654ee420 100644 --- a/docs/build/html/search/all_5.js +++ b/docs/build/html/search/all_5.js @@ -19,41 +19,40 @@ var searchData= ['enable_5fcompile_16',['enable_compile',['../namespacemlx_1_1core.html#a1983a2466bff3bae4d23cf34bd0946c9',1,'mlx::core']]], ['enable_5ffor_5farrays_5ft_17',['enable_for_arrays_t',['../namespacemlx_1_1core.html#af89751d79339f3e4d9318ea97d64d114',1,'mlx::core']]], ['enabled_18',['enabled',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4aa10311459433adf322f2590a4987c423',1,'mlx::core']]], - ['encodetocommandbuffer_19',['encodeToCommandBuffer',['../class_m_p_s_1_1_matrix_multiplication.html#ad30480456e66b79fa47086a3411c4921',1,'MPS::MatrixMultiplication::encodeToCommandBuffer()'],['../class_m_p_s_1_1_matrix_vector_multiplication.html#adefd2759c843035c73da2d9aad8ffb8a',1,'MPS::MatrixVectorMultiplication::encodeToCommandBuffer()']]], - ['end_20',['end',['../classmlx_1_1core_1_1array.html#a5daf64552fb450825c9b382f3a5fa2d4',1,'mlx::core::array']]], - ['end_5fencoding_21',['end_encoding',['../classmlx_1_1core_1_1metal_1_1_device.html#a60689f97347811b27e8c5ca23e0372bf',1,'mlx::core::metal::Device']]], - ['enqueue_22',['enqueue',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4918720319cf224a1b4208568964c286',1,'mlx::core::scheduler::StreamThread::enqueue()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a03809c783bd1866362dc7cb9118abbcc',1,'mlx::core::scheduler::Scheduler::enqueue()'],['../namespacemlx_1_1core_1_1scheduler.html#aa2d4eacf5d5cbc778a51aafd4fd8e4d7',1,'mlx::core::scheduler::enqueue()']]], - ['epsilon_23',['epsilon',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a96c4197e3076f0aa9065370b8ece49ca',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['equal_24',['Equal',['../struct_equal.html',1,'Equal'],['../structmlx_1_1core_1_1detail_1_1_equal.html',1,'mlx::core::detail::Equal'],['../classmlx_1_1core_1_1_equal.html',1,'mlx::core::Equal'],['../classmlx_1_1core_1_1_equal.html#a4af81cf2dd071db5bbf8ce1df95fdf36',1,'mlx::core::Equal::Equal()']]], - ['equal_25',['equal',['../group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde',1,'mlx::core']]], - ['erf_26',['Erf',['../struct_erf.html',1,'Erf'],['../structmlx_1_1core_1_1detail_1_1_erf.html',1,'mlx::core::detail::Erf'],['../classmlx_1_1core_1_1_erf.html',1,'mlx::core::Erf'],['../classmlx_1_1core_1_1_erf.html#a702f76f848928d8d7d3d0881ac6e4c82',1,'mlx::core::Erf::Erf()']]], - ['erf_27',['erf',['../erf_8h.html#a6ce199ee56105c67adbf8c48c019a8b2',1,'erf(): erf.h'],['../group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0',1,'mlx::core::erf()']]], - ['erf_2eh_28',['erf.h',['../erf_8h.html',1,'']]], - ['erfinv_29',['ErfInv',['../struct_erf_inv.html',1,'ErfInv'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html',1,'mlx::core::detail::ErfInv'],['../classmlx_1_1core_1_1_erf_inv.html',1,'mlx::core::ErfInv'],['../classmlx_1_1core_1_1_erf_inv.html#a5d0279247b67da4592311559f04e1478',1,'mlx::core::ErfInv::ErfInv()']]], - ['erfinv_30',['erfinv',['../erf_8h.html#a1846e0d683c7aff826bb32addcc3b885',1,'erfinv(): erf.h'],['../group__ops.html#ga76fb9062c64264e34d2e07013390557c',1,'mlx::core::erfinv()']]], - ['eval_31',['eval',['../classmlx_1_1core_1_1array.html#a2820c45188071a22175e9fa42e10a49a',1,'mlx::core::array::eval()'],['../namespacemlx_1_1core.html#a7d6e097d8effed52f4713672e471f299',1,'mlx::core::eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#adb14f689c9f75f7901edb196c2bfb971',1,'mlx::core::eval(Arrays &&... outputs)']]], - ['eval_5fcpu_32',['eval_cpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#acdc1965ad64ee9ee6328fe150a97902e',1,'mlx::core::distributed::AllReduce::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ab721fe0072fffbddbc3c4334dd033ba5',1,'mlx::core::distributed::AllGather::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a7da6e0cfd630958d9633b2e2bd97a54f',1,'mlx::core::fast::RMSNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#adfc1d52bc266466ab29ee45fd8fab439',1,'mlx::core::fast::RMSNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a5d7a4c1c9ee84e327d1c371733108c05',1,'mlx::core::fast::LayerNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a0d8c4c6e7462befc38f7e08244fa1c2b',1,'mlx::core::fast::LayerNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a05a7d595c6b9dadf7ddfd6e3fd402f0e',1,'mlx::core::fast::RoPE::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ae20851e002f7fcb6d4f97817596f6328',1,'mlx::core::fast::ScaledDotProductAttention::eval_cpu()'],['../classmlx_1_1core_1_1_primitive.html#a1596dc50b910538eae14878e98f07575',1,'mlx::core::Primitive::eval_cpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a7e8f6f5d6ae0a33f6abc0f5a46e0b132',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#aa0ed6e32c36200a3ff9bc592c9b300db',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0d3e697496ef8e842d21195cb3c14e60',1,'mlx::core::Abs::eval_cpu()'],['../classmlx_1_1core_1_1_add.html#a5bacfc51dfa2a5a931bad2dd7bdc7a5f',1,'mlx::core::Add::eval_cpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a15694e3bf2ed5c193237b2b9ca00867c',1,'mlx::core::AddMM::eval_cpu()'],['../classmlx_1_1core_1_1_arange.html#aba44432491cbd599bf72712f5f4267a1',1,'mlx::core::Arange::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a58dcba9e706cb12bab062bb7fa5fa006',1,'mlx::core::ArcCos::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#a0f6d989bcbbc38f15ef17a136879a9c9',1,'mlx::core::ArcCosh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sin.html#ab3542492c14021329788de8f2a9be1e4',1,'mlx::core::ArcSin::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a52574b24d8d16839c58673f51f8ac066',1,'mlx::core::ArcSinh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a1211bc31241227528f04435239ddb9a3',1,'mlx::core::ArcTan::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a13094e6b702769928ca0da468f5ce45c',1,'mlx::core::ArcTan2::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a5af9224e1f1ffec412b0baa0af7e1ecd',1,'mlx::core::ArcTanh::eval_cpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a896f75c5325798ac3f9093f6a4581828',1,'mlx::core::ArgPartition::eval_cpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#ad8d48725623ede1ff654fa13eccf2287',1,'mlx::core::ArgReduce::eval_cpu()'],['../classmlx_1_1core_1_1_arg_sort.html#a022079683774bfeb531b3a002cff16fa',1,'mlx::core::ArgSort::eval_cpu()'],['../classmlx_1_1core_1_1_as_type.html#aa89dbf4d73b00c6a44cffd04d5bb228d',1,'mlx::core::AsType::eval_cpu()'],['../classmlx_1_1core_1_1_as_strided.html#acdd4705e4503ff0b124215c4676b4193',1,'mlx::core::AsStrided::eval_cpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a2194bf585213bda1b2966aa02d2fe283',1,'mlx::core::BitwiseBinary::eval_cpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aa85da478cdc6d4a97be06e5d4abee1f2',1,'mlx::core::BlockMaskedMM::eval_cpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#a62352074a480df0e1f879b0bae425730',1,'mlx::core::GatherMM::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast.html#a53d48d9778e2d4c24a124cd767900780',1,'mlx::core::Broadcast::eval_cpu()'],['../classmlx_1_1core_1_1_ceil.html#a9791801fff3f8b79944e15ac2a45a035',1,'mlx::core::Ceil::eval_cpu()'],['../classmlx_1_1core_1_1_compiled.html#ac45b1d0fedd85feefbff7ce7e168b151',1,'mlx::core::Compiled::eval_cpu()'],['../classmlx_1_1core_1_1_concatenate.html#a609e76bede7fc5581ec84ddcb727a258',1,'mlx::core::Concatenate::eval_cpu()'],['../classmlx_1_1core_1_1_conjugate.html#ae39643e2178f442ffba05139f8609d61',1,'mlx::core::Conjugate::eval_cpu()'],['../classmlx_1_1core_1_1_convolution.html#ac74256068da01730629109fa4fa8432b',1,'mlx::core::Convolution::eval_cpu()'],['../classmlx_1_1core_1_1_copy.html#af4a0ebec423e84ffe8083a5e9ed0d70c',1,'mlx::core::Copy::eval_cpu()'],['../classmlx_1_1core_1_1_cos.html#a061fc446268fe56237ae6b20ccf78152',1,'mlx::core::Cos::eval_cpu()'],['../classmlx_1_1core_1_1_cosh.html#ae8702df7e8f0e20cbeccb2a548961d3d',1,'mlx::core::Cosh::eval_cpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#adba1c40c77a2138df6b5f75483f62184',1,'mlx::core::CustomTransforms::eval_cpu()'],['../classmlx_1_1core_1_1_depends.html#a0c7ea6db97337591fa53c6e6bde41e5e',1,'mlx::core::Depends::eval_cpu()'],['../classmlx_1_1core_1_1_divide.html#a823443c2a8e8b81bbcaeee6ddbcdbf49',1,'mlx::core::Divide::eval_cpu()'],['../classmlx_1_1core_1_1_div_mod.html#ae350b7b93ad128e3133ee14f247193b3',1,'mlx::core::DivMod::eval_cpu()'],['../classmlx_1_1core_1_1_select.html#aa51aa36e0adbd69e0d23d7c7adf88de2',1,'mlx::core::Select::eval_cpu()'],['../classmlx_1_1core_1_1_remainder.html#ac6c6c86a0bf02e6e529eb87f6e617ccc',1,'mlx::core::Remainder::eval_cpu()'],['../classmlx_1_1core_1_1_equal.html#aabb8aa61fa581defddcdca1274b1b454',1,'mlx::core::Equal::eval_cpu()'],['../classmlx_1_1core_1_1_erf.html#a84ea16e43d5b7f83bbc2d5ece78a3fb6',1,'mlx::core::Erf::eval_cpu()'],['../classmlx_1_1core_1_1_erf_inv.html#af579627402af3249565134884701d39e',1,'mlx::core::ErfInv::eval_cpu()'],['../classmlx_1_1core_1_1_exp.html#a47934c5a5023bc7ae7ae89bff45ebb2c',1,'mlx::core::Exp::eval_cpu()'],['../classmlx_1_1core_1_1_expm1.html#ab9c8b7aa50fe4592d55f8957baac647a',1,'mlx::core::Expm1::eval_cpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a6bc262a0c2b5d4fe655e3e2e0ff28635',1,'mlx::core::FFT::eval_cpu()'],['../classmlx_1_1core_1_1_floor.html#a1a7dc5f571b7b73e7ef3cbdc1dd1fcf7',1,'mlx::core::Floor::eval_cpu()'],['../classmlx_1_1core_1_1_full.html#a3dccd3756599d7fd018b2af0093b082c',1,'mlx::core::Full::eval_cpu()'],['../classmlx_1_1core_1_1_gather.html#a9ed5587f0d04b59a2b9186c0aac21290',1,'mlx::core::Gather::eval_cpu()'],['../classmlx_1_1core_1_1_greater.html#abe1c03f311d0e0b610f3392a6566f2ae',1,'mlx::core::Greater::eval_cpu()'],['../classmlx_1_1core_1_1_greater_equal.html#a15469125b9bea89b64bfeac01590c075',1,'mlx::core::GreaterEqual::eval_cpu()'],['../classmlx_1_1core_1_1_hadamard.html#ab27d6a9df42b3aab41ace3073a4c880d',1,'mlx::core::Hadamard::eval_cpu()'],['../classmlx_1_1core_1_1_less.html#a32624124ffece066f496b3299056bcef',1,'mlx::core::Less::eval_cpu()'],['../classmlx_1_1core_1_1_less_equal.html#a55d1352b0e97841a92503bc57c19ed16',1,'mlx::core::LessEqual::eval_cpu()'],['../classmlx_1_1core_1_1_load.html#ada026ac30566f3109d8182e35d307c0a',1,'mlx::core::Load::eval_cpu()'],['../classmlx_1_1core_1_1_log.html#aadc7bb4cb24f3ecbbb9ed54a699ab74f',1,'mlx::core::Log::eval_cpu()'],['../classmlx_1_1core_1_1_log1p.html#a8192e5438de99c4cda056987935cba23',1,'mlx::core::Log1p::eval_cpu()'],['../classmlx_1_1core_1_1_logical_not.html#acf3f7b3b20ca69533536e0e0a05725b3',1,'mlx::core::LogicalNot::eval_cpu()'],['../classmlx_1_1core_1_1_logical_and.html#adbe1c1785af1a8b827289d22b0d170b3',1,'mlx::core::LogicalAnd::eval_cpu()'],['../classmlx_1_1core_1_1_logical_or.html#a13cd4cbf26589287e85aeaaca42d7f62',1,'mlx::core::LogicalOr::eval_cpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#abef17fb590b1a8d356f2a580e45d41f0',1,'mlx::core::LogAddExp::eval_cpu()'],['../classmlx_1_1core_1_1_matmul.html#a357a7f57a2a220a91977f810a69413fc',1,'mlx::core::Matmul::eval_cpu()'],['../classmlx_1_1core_1_1_maximum.html#a62b38fbe5f96db58c2b60165ac4eadcf',1,'mlx::core::Maximum::eval_cpu()'],['../classmlx_1_1core_1_1_minimum.html#a6b93f493ee87089943a8085fe59dfc6e',1,'mlx::core::Minimum::eval_cpu()'],['../classmlx_1_1core_1_1_multiply.html#a624fce06c047cdc4dfdbdcaaddb25f34',1,'mlx::core::Multiply::eval_cpu()'],['../classmlx_1_1core_1_1_negative.html#af43553dc418c8ebe75fa9cdcba103c3b',1,'mlx::core::Negative::eval_cpu()'],['../classmlx_1_1core_1_1_not_equal.html#a8f95f8b5873850b875b1641df8196047',1,'mlx::core::NotEqual::eval_cpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#acc328321cf5300874ee884367cbede3f',1,'mlx::core::NumberOfElements::eval_cpu()'],['../classmlx_1_1core_1_1_pad.html#aaf82dd163cd536fbf97304f8b29080cb',1,'mlx::core::Pad::eval_cpu()'],['../classmlx_1_1core_1_1_partition.html#a784596ab567f9f3cb4fe1a69466523d8',1,'mlx::core::Partition::eval_cpu()'],['../classmlx_1_1core_1_1_power.html#a6783da16fb6ff393aaa57737f1973206',1,'mlx::core::Power::eval_cpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ab3dfa73b74d8f4f2e9ab4f0eb016b0e3',1,'mlx::core::QuantizedMatmul::eval_cpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a89aae98bfbdd6563df44ef7d70f0bf8c',1,'mlx::core::GatherQMM::eval_cpu()'],['../classmlx_1_1core_1_1_random_bits.html#a5752d051cd16cf5f8d4754c0a656f0d2',1,'mlx::core::RandomBits::eval_cpu()'],['../classmlx_1_1core_1_1_reshape.html#a658de2c5f710991b48e14b2bd19b229f',1,'mlx::core::Reshape::eval_cpu()'],['../classmlx_1_1core_1_1_reduce.html#aeb8a58b560c0a09ae3a695df7829acfa',1,'mlx::core::Reduce::eval_cpu()'],['../classmlx_1_1core_1_1_round.html#ad066b0944b437f64ab546025efa00007',1,'mlx::core::Round::eval_cpu()'],['../classmlx_1_1core_1_1_scan.html#a15676d9fd066e935782a923fba3e940b',1,'mlx::core::Scan::eval_cpu()'],['../classmlx_1_1core_1_1_scatter.html#a7623f590f8b77167b5ebb4f14bc9dc97',1,'mlx::core::Scatter::eval_cpu()'],['../classmlx_1_1core_1_1_sigmoid.html#aa930ce05734cca529ebcb8d0ca8e1255',1,'mlx::core::Sigmoid::eval_cpu()'],['../classmlx_1_1core_1_1_sign.html#a7498ec993b66879be30c5d9762c45a97',1,'mlx::core::Sign::eval_cpu()'],['../classmlx_1_1core_1_1_sin.html#ab34f9cebc2aed55a0b6ab4c991f02eb5',1,'mlx::core::Sin::eval_cpu()'],['../classmlx_1_1core_1_1_sinh.html#ab6d5f6f40d177f6435f6a51c71b939dd',1,'mlx::core::Sinh::eval_cpu()'],['../classmlx_1_1core_1_1_slice.html#a4b13503f5b2f5c6a90d394b020f9b3f2',1,'mlx::core::Slice::eval_cpu()'],['../classmlx_1_1core_1_1_slice_update.html#ad82ca0e3ab88a0e086431050deea831b',1,'mlx::core::SliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_softmax.html#ac9ebc2eab1683b682e689ed8f4622b79',1,'mlx::core::Softmax::eval_cpu()'],['../classmlx_1_1core_1_1_sort.html#a459769a0241b2620e55bedaba19827cd',1,'mlx::core::Sort::eval_cpu()'],['../classmlx_1_1core_1_1_split.html#aff2889cb9074f0fda53edf8fa40b1fd4',1,'mlx::core::Split::eval_cpu()'],['../classmlx_1_1core_1_1_square.html#a1f4d327a705950616da63b83c2829e59',1,'mlx::core::Square::eval_cpu()'],['../classmlx_1_1core_1_1_sqrt.html#a5a64ecc4eef1e30a2963435dca7cefd5',1,'mlx::core::Sqrt::eval_cpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a56207714d374b08f60e4d9cdbc7340b2',1,'mlx::core::StopGradient::eval_cpu()'],['../classmlx_1_1core_1_1_subtract.html#a47574258b6c95f8ad260c114d6d36a12',1,'mlx::core::Subtract::eval_cpu()'],['../classmlx_1_1core_1_1_tan.html#a9c9a731158fa60eef30067fe0da9f3e9',1,'mlx::core::Tan::eval_cpu()'],['../classmlx_1_1core_1_1_tanh.html#af7ed4345f622da069e5b0284067923f5',1,'mlx::core::Tanh::eval_cpu()'],['../classmlx_1_1core_1_1_uniform.html#a037a2c96b79b70a64f2b637c9f1a432f',1,'mlx::core::Uniform::eval_cpu()'],['../classmlx_1_1core_1_1_view.html#a0ad6deb11914a242f10e8039fcb02497',1,'mlx::core::View::eval_cpu()'],['../classmlx_1_1core_1_1_transpose.html#a1fbcfcca43f9ec06c63a3c14708c30f8',1,'mlx::core::Transpose::eval_cpu()'],['../classmlx_1_1core_1_1_q_r_f.html#a48493887395d65a27f04de1804d277d2',1,'mlx::core::QRF::eval_cpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a637f5c39fa8b10722c04a066f6c1ada6',1,'mlx::core::SVD::eval_cpu()'],['../classmlx_1_1core_1_1_inverse.html#aeb1d8dc9bc4052a616023f65b3c7bb81',1,'mlx::core::Inverse::eval_cpu()'],['../classmlx_1_1core_1_1_cholesky.html#a4bdec36c1cc99aadf9a4a39d4c57bea5',1,'mlx::core::Cholesky::eval_cpu()']]], - ['eval_5fgpu_33',['eval_gpu',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a611ac8dfd9b4f49617ee8976ff3e15e9',1,'mlx::core::distributed::DistPrimitive::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae7955e8d43c097eecae264e804b4d8ca',1,'mlx::core::fast::RMSNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a48efb8fa84c4ba6cc9fb560ebbe01560',1,'mlx::core::fast::RMSNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a77abda7f47bffa2c037a5d60cccc1528',1,'mlx::core::fast::LayerNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a954a003a4a27c8c4c60a5a14142a9cc3',1,'mlx::core::fast::LayerNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a913b6b00fc518b25ac3947e4e15790f2',1,'mlx::core::fast::RoPE::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a505f38ba93a3499895f5312e0112e73d',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ad51666e69f670e286293aff96eb435a9',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, array &out)'],['../classmlx_1_1core_1_1_primitive.html#ad217376dcf5eff691d731566faec2ba2',1,'mlx::core::Primitive::eval_gpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a6b7f80abaf038d53ec6ffbb0dfac6adb',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#a971fe9ad47f6569118879ce1d0f41447',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0a976e636dd8505b473fbdddf949f514',1,'mlx::core::Abs::eval_gpu()'],['../classmlx_1_1core_1_1_add.html#aa0aacbc1e26b95a2f040f62aa4f69c3d',1,'mlx::core::Add::eval_gpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a5f933be14baebc32a0be0f9a69148aa9',1,'mlx::core::AddMM::eval_gpu()'],['../classmlx_1_1core_1_1_arange.html#a7a2e9787c6c3a78b4a6df91206974031',1,'mlx::core::Arange::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a46f72d4af89b0a0f5f203783fb44589c',1,'mlx::core::ArcCos::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#aa6a2587485a0e015ac2d5211d7d045fc',1,'mlx::core::ArcCosh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sin.html#a7fa4ae7a85bc8bed97ea258ae30762f3',1,'mlx::core::ArcSin::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79f648a86de4c10386a1ce3b5e38e8ac',1,'mlx::core::ArcSinh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a77866feb27028865d844070447c9a254',1,'mlx::core::ArcTan::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a76d3f0c29e0ff4642b8d39dac90d3f50',1,'mlx::core::ArcTan2::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a10566b9d3b2c7d090895b46d9040bc1d',1,'mlx::core::ArcTanh::eval_gpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a9a60995eaf85f63c877e86b23cbc15fc',1,'mlx::core::ArgPartition::eval_gpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#aafa982ce2abc0cd9e81e43aa2c823d29',1,'mlx::core::ArgReduce::eval_gpu()'],['../classmlx_1_1core_1_1_arg_sort.html#abc2d730850ec4ee8d7968b7417911709',1,'mlx::core::ArgSort::eval_gpu()'],['../classmlx_1_1core_1_1_as_type.html#a5b111b9d74c60d27b4a7ebaa49f96e0b',1,'mlx::core::AsType::eval_gpu()'],['../classmlx_1_1core_1_1_as_strided.html#ab6771a208323994927ca162ba7bb10ed',1,'mlx::core::AsStrided::eval_gpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#ac831a29fc46701b00bbe63ee33832afd',1,'mlx::core::BitwiseBinary::eval_gpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ab372b6df4de00a33795a052a23bb1df9',1,'mlx::core::BlockMaskedMM::eval_gpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#ad754c35f460a055cc383ad93a5f72da1',1,'mlx::core::GatherMM::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast.html#ab9bd9dbcedcefc9b29c84911b5ce69fe',1,'mlx::core::Broadcast::eval_gpu()'],['../classmlx_1_1core_1_1_ceil.html#abe178e0058e44b6618be414215e96887',1,'mlx::core::Ceil::eval_gpu()'],['../classmlx_1_1core_1_1_compiled.html#aa3d5ff0f2b3554ad48fbbf2a0f3336d5',1,'mlx::core::Compiled::eval_gpu()'],['../classmlx_1_1core_1_1_concatenate.html#a309a1c50e97f9925866433ee2841c474',1,'mlx::core::Concatenate::eval_gpu()'],['../classmlx_1_1core_1_1_conjugate.html#aff0a802166e3724db88ab5d3feb2d3de',1,'mlx::core::Conjugate::eval_gpu()'],['../classmlx_1_1core_1_1_convolution.html#a30b64109eeb1778f002b99447dff9dd2',1,'mlx::core::Convolution::eval_gpu()'],['../classmlx_1_1core_1_1_copy.html#a1eda7b2ea771a168f67421f0d384b3a1',1,'mlx::core::Copy::eval_gpu()'],['../classmlx_1_1core_1_1_cos.html#a5ef41aafad595f6cdd8c535e36e12060',1,'mlx::core::Cos::eval_gpu()'],['../classmlx_1_1core_1_1_cosh.html#a23f71b43792934c3ec0ebe9b74f32559',1,'mlx::core::Cosh::eval_gpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#a7b3538681acbb20af3ed37b0877f6667',1,'mlx::core::CustomTransforms::eval_gpu()'],['../classmlx_1_1core_1_1_depends.html#ae5057f65e69490ad0add8eeda2b75e28',1,'mlx::core::Depends::eval_gpu()'],['../classmlx_1_1core_1_1_divide.html#abffda0ce37221ddc28dc9eea794f6bc7',1,'mlx::core::Divide::eval_gpu()'],['../classmlx_1_1core_1_1_div_mod.html#a003117c9ecf3c06a27248f72a76348dc',1,'mlx::core::DivMod::eval_gpu()'],['../classmlx_1_1core_1_1_select.html#a2a82b6cba4c386b2b87f225a4b08ea9b',1,'mlx::core::Select::eval_gpu()'],['../classmlx_1_1core_1_1_remainder.html#a7919ea9b84e42522d51bf0d5a396e161',1,'mlx::core::Remainder::eval_gpu()'],['../classmlx_1_1core_1_1_equal.html#ac3757001fec42ceb5ece2954df42161c',1,'mlx::core::Equal::eval_gpu()'],['../classmlx_1_1core_1_1_erf.html#ad8551be664d767dccc3c0d8cc1eca008',1,'mlx::core::Erf::eval_gpu()'],['../classmlx_1_1core_1_1_erf_inv.html#a4a2413d0634db1f3dae1806ddfa632db',1,'mlx::core::ErfInv::eval_gpu()'],['../classmlx_1_1core_1_1_exp.html#a7d63695a97a14760fd33b5d4e6590822',1,'mlx::core::Exp::eval_gpu()'],['../classmlx_1_1core_1_1_expm1.html#a82930071f4b77d883b300f77966aff5f',1,'mlx::core::Expm1::eval_gpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a1c21b26d1e9ad7c4da78ae845721b2dd',1,'mlx::core::FFT::eval_gpu()'],['../classmlx_1_1core_1_1_floor.html#aaa29c83538099eb8f951c95a41f2eb65',1,'mlx::core::Floor::eval_gpu()'],['../classmlx_1_1core_1_1_full.html#aa54f99bb4cba12a551392dea56003872',1,'mlx::core::Full::eval_gpu()'],['../classmlx_1_1core_1_1_gather.html#aec48ee529cb2449915a7b27a3c4361e8',1,'mlx::core::Gather::eval_gpu()'],['../classmlx_1_1core_1_1_greater.html#ae8957cccf4c924d941f57a1bb751c878',1,'mlx::core::Greater::eval_gpu()'],['../classmlx_1_1core_1_1_greater_equal.html#ac246263b4548126c3d4ab7e392575d24',1,'mlx::core::GreaterEqual::eval_gpu()'],['../classmlx_1_1core_1_1_hadamard.html#a2470feb690f5463138490763c38b5733',1,'mlx::core::Hadamard::eval_gpu()'],['../classmlx_1_1core_1_1_less.html#a353335ce06ddbe8498d86d129c835917',1,'mlx::core::Less::eval_gpu()'],['../classmlx_1_1core_1_1_less_equal.html#acf035a82b11e6f63742143ea540fedac',1,'mlx::core::LessEqual::eval_gpu()'],['../classmlx_1_1core_1_1_load.html#a06933e887ea94a4d01d81195c5e07a3d',1,'mlx::core::Load::eval_gpu()'],['../classmlx_1_1core_1_1_log.html#aaaa49e9455f3a197bc319646b5ca6390',1,'mlx::core::Log::eval_gpu()'],['../classmlx_1_1core_1_1_log1p.html#a1b97decae7338d46874e736c95fa7431',1,'mlx::core::Log1p::eval_gpu()'],['../classmlx_1_1core_1_1_logical_not.html#a1d0d2bc93f935eca6c85ef7bf67f2d6a',1,'mlx::core::LogicalNot::eval_gpu()'],['../classmlx_1_1core_1_1_logical_and.html#a132b2eedaa3978de5a5350da3c2ca40f',1,'mlx::core::LogicalAnd::eval_gpu()'],['../classmlx_1_1core_1_1_logical_or.html#a3be1da328f0f8620de2e4fc1d22a077a',1,'mlx::core::LogicalOr::eval_gpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#acace355b62ec00df649f9f99e8f2eb7a',1,'mlx::core::LogAddExp::eval_gpu()'],['../classmlx_1_1core_1_1_matmul.html#a8707a4e9b75c769e8f1dbca15c6a1ae7',1,'mlx::core::Matmul::eval_gpu()'],['../classmlx_1_1core_1_1_maximum.html#ade0f721b10a6b3a12bdadd34c48f72a7',1,'mlx::core::Maximum::eval_gpu()'],['../classmlx_1_1core_1_1_minimum.html#aadc68afa0afbe2103f19d161f5e0a2ba',1,'mlx::core::Minimum::eval_gpu()'],['../classmlx_1_1core_1_1_multiply.html#a634fcb4e981d8d3f4d94252caf25bee0',1,'mlx::core::Multiply::eval_gpu()'],['../classmlx_1_1core_1_1_negative.html#a97f1b316eace0c6d9e576d766940c75b',1,'mlx::core::Negative::eval_gpu()'],['../classmlx_1_1core_1_1_not_equal.html#a61179747e34e203150e9c660dfddb5f2',1,'mlx::core::NotEqual::eval_gpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#a2c98c42915fb2bfe12f5c99ea553eff5',1,'mlx::core::NumberOfElements::eval_gpu()'],['../classmlx_1_1core_1_1_pad.html#aefd4d3a5bd8b6b35b266c9e558ada153',1,'mlx::core::Pad::eval_gpu()'],['../classmlx_1_1core_1_1_partition.html#a8eca1be21ae9ccfda46e6f3e85f506ef',1,'mlx::core::Partition::eval_gpu()'],['../classmlx_1_1core_1_1_power.html#a80577d4c0853c24027777c90a1ec7e11',1,'mlx::core::Power::eval_gpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a2812ad007d695ed1aaf9cf706fb9c4b3',1,'mlx::core::QuantizedMatmul::eval_gpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a86eb048afc95646b2e96ec5493e3d887',1,'mlx::core::GatherQMM::eval_gpu()'],['../classmlx_1_1core_1_1_random_bits.html#a578756866665358577418e4cdd94aa3a',1,'mlx::core::RandomBits::eval_gpu()'],['../classmlx_1_1core_1_1_reshape.html#aa1e85f28471875750c47351520b56059',1,'mlx::core::Reshape::eval_gpu()'],['../classmlx_1_1core_1_1_reduce.html#ae9caaf42edadfe73ea208d98f526890f',1,'mlx::core::Reduce::eval_gpu()'],['../classmlx_1_1core_1_1_round.html#af7fe5ff8f3db166c203b4be4b07f13ec',1,'mlx::core::Round::eval_gpu()'],['../classmlx_1_1core_1_1_scan.html#aef22c6fc2b2cb2a907cd8965c7413dde',1,'mlx::core::Scan::eval_gpu()'],['../classmlx_1_1core_1_1_scatter.html#ab304345db3d8cfeea15e27461ae2e678',1,'mlx::core::Scatter::eval_gpu()'],['../classmlx_1_1core_1_1_sigmoid.html#a7a6bd0222d51d7f25f2719a91ccdfeca',1,'mlx::core::Sigmoid::eval_gpu()'],['../classmlx_1_1core_1_1_sign.html#afa2b48b99a194106006b44af69ffda8b',1,'mlx::core::Sign::eval_gpu()'],['../classmlx_1_1core_1_1_sin.html#a6b59f1156cf8bdad8d45acd1d825cb5e',1,'mlx::core::Sin::eval_gpu()'],['../classmlx_1_1core_1_1_sinh.html#a5a1af2399f166d5b228b5e83a1837c75',1,'mlx::core::Sinh::eval_gpu()'],['../classmlx_1_1core_1_1_slice.html#aa53c21ff06a7c659e889af6b97d10a4a',1,'mlx::core::Slice::eval_gpu()'],['../classmlx_1_1core_1_1_slice_update.html#aac1a1d122e5697be057d63552141032b',1,'mlx::core::SliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_softmax.html#a35dac69ddcc7e2ec0e1a76fe93db85af',1,'mlx::core::Softmax::eval_gpu()'],['../classmlx_1_1core_1_1_sort.html#a4141c48f0e8670c728663f3722675382',1,'mlx::core::Sort::eval_gpu()'],['../classmlx_1_1core_1_1_split.html#a78ddda89c4daee73c74cfbc1e44656df',1,'mlx::core::Split::eval_gpu()'],['../classmlx_1_1core_1_1_square.html#a0ea2a78a5bb52daa4103263bf2f98045',1,'mlx::core::Square::eval_gpu()'],['../classmlx_1_1core_1_1_sqrt.html#a6d205e679a593d1ba20206c5c47ba501',1,'mlx::core::Sqrt::eval_gpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a907b96f0a1ce608e211d87ccf2b9ca89',1,'mlx::core::StopGradient::eval_gpu()'],['../classmlx_1_1core_1_1_subtract.html#a69021b23daf061764d97fabbc0f4f06c',1,'mlx::core::Subtract::eval_gpu()'],['../classmlx_1_1core_1_1_tan.html#aca7dbb4836507005a2032ac957a04d3f',1,'mlx::core::Tan::eval_gpu()'],['../classmlx_1_1core_1_1_tanh.html#a48df896599ae93dbce84a5c0f50cf761',1,'mlx::core::Tanh::eval_gpu()'],['../classmlx_1_1core_1_1_uniform.html#a5f88cbf2495f24f87cefd99aaaebe4d0',1,'mlx::core::Uniform::eval_gpu()'],['../classmlx_1_1core_1_1_view.html#add6e12ff1e476fe1db7718b14f21b075',1,'mlx::core::View::eval_gpu()'],['../classmlx_1_1core_1_1_transpose.html#a38d25739c08aa594a6775015a1d7d92e',1,'mlx::core::Transpose::eval_gpu()'],['../classmlx_1_1core_1_1_q_r_f.html#ae5fa3482192f4713605cd07e7fc1c6c9',1,'mlx::core::QRF::eval_gpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a7067b2207f826a25549d571856b94e83',1,'mlx::core::SVD::eval_gpu()'],['../classmlx_1_1core_1_1_inverse.html#a086fbbc947ad232e01686ad063a78ed2',1,'mlx::core::Inverse::eval_gpu()'],['../classmlx_1_1core_1_1_cholesky.html#a8c918594bf129888044ef37fcae56795',1,'mlx::core::Cholesky::eval_gpu()']]], - ['event_34',['Event',['../classmlx_1_1core_1_1_event.html',1,'mlx::core::Event'],['../classmlx_1_1core_1_1_event.html#a833506419b2110ad1abd89b2dd238b4d',1,'mlx::core::Event::Event()=default'],['../classmlx_1_1core_1_1_event.html#a13e4835f2ffb2cc22e29148a448ea184',1,'mlx::core::Event::Event(const Stream &steam)']]], - ['event_35',['event',['../classmlx_1_1core_1_1array.html#a0a8e4d6e67e739a712876bb36f88f9bf',1,'mlx::core::array']]], - ['event_2eh_36',['event.h',['../event_8h.html',1,'']]], - ['excess_37',['excess',['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#ada22a8bd8a89078cfa28874055c8e753',1,'mlx::steel::ChannelHelper< 1 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acc490f3999230aa592c61bbed7eb7cfe',1,'mlx::steel::ChannelHelper< 2 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#aae404674763f3dc73c5ab29169f8b80f',1,'mlx::steel::ChannelHelper< 3 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#aecdd8331fec703d739a6f07b9b901ac8',1,'mlx::steel::ChannelHelper< 4 >::excess']]], - ['exec_38',['exec',['../classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2',1,'pocketfft::detail::cfftp::exec()'],['../classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac',1,'pocketfft::detail::rfftp::exec()'],['../classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c',1,'pocketfft::detail::fftblue::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d',1,'pocketfft::detail::pocketfft_c::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33',1,'pocketfft::detail::pocketfft_r::exec()'],['../classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058',1,'pocketfft::detail::T_dct1::exec()'],['../classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26',1,'pocketfft::detail::T_dst1::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac',1,'pocketfft::detail::T_dcst23::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708',1,'pocketfft::detail::T_dcst4::exec()']]], - ['exec_5fr_39',['exec_r',['../classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00',1,'pocketfft::detail::fftblue']]], - ['execc2c_40',['ExecC2C',['../structpocketfft_1_1detail_1_1_exec_c2_c.html',1,'pocketfft::detail']]], - ['execdcst_41',['ExecDcst',['../structpocketfft_1_1detail_1_1_exec_dcst.html',1,'pocketfft::detail']]], - ['exechartley_42',['ExecHartley',['../structpocketfft_1_1detail_1_1_exec_hartley.html',1,'pocketfft::detail']]], - ['execr2r_43',['ExecR2R',['../structpocketfft_1_1detail_1_1_exec_r2_r.html',1,'pocketfft::detail']]], - ['exp_44',['Exp',['../struct_exp.html',1,'Exp'],['../structmlx_1_1core_1_1detail_1_1_exp.html',1,'mlx::core::detail::Exp'],['../classmlx_1_1core_1_1_exp.html',1,'mlx::core::Exp'],['../classmlx_1_1core_1_1_exp.html#a1d0a618cbb91ab29ef53b57ff6ed6e06',1,'mlx::core::Exp::Exp()']]], - ['exp_45',['exp',['../namespacemetal.html#ac2a0b3618d922ac014baac8189d44650',1,'metal::exp()'],['../namespacemetal_1_1fast.html#ad3dbd387b63373c29e3449609f763ede',1,'metal::fast::exp()'],['../namespacemetal_1_1precise.html#a8d8d2d5700ce432b33cf47cf22528e8f',1,'metal::precise::exp()'],['../group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125',1,'mlx::core::exp()']]], - ['exp10_46',['exp10',['../namespacemetal.html#a4c63707d13c89364496a48906631c204',1,'metal::exp10()'],['../namespacemetal_1_1fast.html#a453122f982485cbb4e471b3ac282ee5e',1,'metal::fast::exp10()'],['../namespacemetal_1_1precise.html#af9addb343c967da3a83e9e123a8521fd',1,'metal::precise::exp10()']]], - ['exp2_47',['exp2',['../namespacemetal.html#a228201c20777848804a4d0589c1d33e7',1,'metal::exp2()'],['../namespacemetal_1_1fast.html#ac092b65a46720adaf22f6266671d2d71',1,'metal::fast::exp2()'],['../namespacemetal_1_1precise.html#a92a880bd2197efc0da0f8f0f7ec1e4c9',1,'metal::precise::exp2()']]], - ['expand_5fdims_48',['expand_dims',['../group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133',1,'mlx::core::expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a',1,'mlx::core::expand_dims(const array &a, int axis, StreamOrDevice s={})']]], - ['expm1_49',['Expm1',['../struct_expm1.html',1,'Expm1'],['../structmlx_1_1core_1_1detail_1_1_expm1.html',1,'mlx::core::detail::Expm1'],['../classmlx_1_1core_1_1_expm1.html',1,'mlx::core::Expm1'],['../classmlx_1_1core_1_1_expm1.html#a47c2a1b2a4ef6bb07ba77c55ddddaec2',1,'mlx::core::Expm1::Expm1()']]], - ['expm1_50',['expm1',['../group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0',1,'mlx::core']]], - ['expm1f_51',['expm1f',['../expm1f_8h.html#a87f66d30e185950f42ce3641783cdc40',1,'expm1f.h']]], - ['expm1f_2eh_52',['expm1f.h',['../expm1f_8h.html',1,'']]], - ['expm1f_5fscaled_5funchecked_53',['expm1f_scaled_unchecked',['../expm1f_8h.html#adf20e03405fba634ca8d01acac24592e',1,'expm1f.h']]], - ['export_5fto_5fdot_54',['export_to_dot',['../namespacemlx_1_1core.html#a57395bdf43d9c5c134e610c169222cca',1,'mlx::core::export_to_dot(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a839f94dbad44f0d37333006fc876b42e',1,'mlx::core::export_to_dot(std::ostream &os, Arrays &&... outputs)']]], - ['eye_55',['eye',['../group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3',1,'mlx::core::eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd',1,'mlx::core::eye(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga61657db78ef35d41112d362c869c25d2',1,'mlx::core::eye(int n, int m, StreamOrDevice s={})'],['../group__ops.html#ga908a15b42834be498a46856c99dfc779',1,'mlx::core::eye(int n, int m, int k, StreamOrDevice s={})'],['../group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f',1,'mlx::core::eye(int n, StreamOrDevice s={})']]] + ['end_19',['end',['../classmlx_1_1core_1_1array.html#a5daf64552fb450825c9b382f3a5fa2d4',1,'mlx::core::array']]], + ['end_5fencoding_20',['end_encoding',['../classmlx_1_1core_1_1metal_1_1_device.html#a60689f97347811b27e8c5ca23e0372bf',1,'mlx::core::metal::Device']]], + ['enqueue_21',['enqueue',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4918720319cf224a1b4208568964c286',1,'mlx::core::scheduler::StreamThread::enqueue()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a03809c783bd1866362dc7cb9118abbcc',1,'mlx::core::scheduler::Scheduler::enqueue()'],['../namespacemlx_1_1core_1_1scheduler.html#aa2d4eacf5d5cbc778a51aafd4fd8e4d7',1,'mlx::core::scheduler::enqueue()']]], + ['epsilon_22',['epsilon',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a96c4197e3076f0aa9065370b8ece49ca',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['equal_23',['Equal',['../struct_equal.html',1,'Equal'],['../structmlx_1_1core_1_1detail_1_1_equal.html',1,'mlx::core::detail::Equal'],['../classmlx_1_1core_1_1_equal.html',1,'mlx::core::Equal'],['../classmlx_1_1core_1_1_equal.html#a4af81cf2dd071db5bbf8ce1df95fdf36',1,'mlx::core::Equal::Equal()']]], + ['equal_24',['equal',['../group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde',1,'mlx::core']]], + ['erf_25',['Erf',['../struct_erf.html',1,'Erf'],['../structmlx_1_1core_1_1detail_1_1_erf.html',1,'mlx::core::detail::Erf'],['../classmlx_1_1core_1_1_erf.html',1,'mlx::core::Erf'],['../classmlx_1_1core_1_1_erf.html#a702f76f848928d8d7d3d0881ac6e4c82',1,'mlx::core::Erf::Erf()']]], + ['erf_26',['erf',['../erf_8h.html#a6ce199ee56105c67adbf8c48c019a8b2',1,'erf(): erf.h'],['../group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0',1,'mlx::core::erf()']]], + ['erf_2eh_27',['erf.h',['../erf_8h.html',1,'']]], + ['erfinv_28',['ErfInv',['../struct_erf_inv.html',1,'ErfInv'],['../structmlx_1_1core_1_1detail_1_1_erf_inv.html',1,'mlx::core::detail::ErfInv'],['../classmlx_1_1core_1_1_erf_inv.html',1,'mlx::core::ErfInv'],['../classmlx_1_1core_1_1_erf_inv.html#a5d0279247b67da4592311559f04e1478',1,'mlx::core::ErfInv::ErfInv()']]], + ['erfinv_29',['erfinv',['../erf_8h.html#a1846e0d683c7aff826bb32addcc3b885',1,'erfinv(): erf.h'],['../group__ops.html#ga76fb9062c64264e34d2e07013390557c',1,'mlx::core::erfinv()']]], + ['eval_30',['eval',['../classmlx_1_1core_1_1array.html#a2820c45188071a22175e9fa42e10a49a',1,'mlx::core::array::eval()'],['../namespacemlx_1_1core.html#a7d6e097d8effed52f4713672e471f299',1,'mlx::core::eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#adb14f689c9f75f7901edb196c2bfb971',1,'mlx::core::eval(Arrays &&... outputs)']]], + ['eval_5fcpu_31',['eval_cpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#acdc1965ad64ee9ee6328fe150a97902e',1,'mlx::core::distributed::AllReduce::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ab721fe0072fffbddbc3c4334dd033ba5',1,'mlx::core::distributed::AllGather::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a7da6e0cfd630958d9633b2e2bd97a54f',1,'mlx::core::fast::RMSNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#adfc1d52bc266466ab29ee45fd8fab439',1,'mlx::core::fast::RMSNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a5d7a4c1c9ee84e327d1c371733108c05',1,'mlx::core::fast::LayerNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a0d8c4c6e7462befc38f7e08244fa1c2b',1,'mlx::core::fast::LayerNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a05a7d595c6b9dadf7ddfd6e3fd402f0e',1,'mlx::core::fast::RoPE::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ae20851e002f7fcb6d4f97817596f6328',1,'mlx::core::fast::ScaledDotProductAttention::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a3b5d628628d245b38911118d4a0ff9fd',1,'mlx::core::fast::AffineQuantize::eval_cpu()'],['../classmlx_1_1core_1_1_primitive.html#a1596dc50b910538eae14878e98f07575',1,'mlx::core::Primitive::eval_cpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a7e8f6f5d6ae0a33f6abc0f5a46e0b132',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#aa0ed6e32c36200a3ff9bc592c9b300db',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0d3e697496ef8e842d21195cb3c14e60',1,'mlx::core::Abs::eval_cpu()'],['../classmlx_1_1core_1_1_add.html#a5bacfc51dfa2a5a931bad2dd7bdc7a5f',1,'mlx::core::Add::eval_cpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a15694e3bf2ed5c193237b2b9ca00867c',1,'mlx::core::AddMM::eval_cpu()'],['../classmlx_1_1core_1_1_arange.html#aba44432491cbd599bf72712f5f4267a1',1,'mlx::core::Arange::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a58dcba9e706cb12bab062bb7fa5fa006',1,'mlx::core::ArcCos::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#a0f6d989bcbbc38f15ef17a136879a9c9',1,'mlx::core::ArcCosh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sin.html#ab3542492c14021329788de8f2a9be1e4',1,'mlx::core::ArcSin::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a52574b24d8d16839c58673f51f8ac066',1,'mlx::core::ArcSinh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a1211bc31241227528f04435239ddb9a3',1,'mlx::core::ArcTan::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a13094e6b702769928ca0da468f5ce45c',1,'mlx::core::ArcTan2::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a5af9224e1f1ffec412b0baa0af7e1ecd',1,'mlx::core::ArcTanh::eval_cpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a896f75c5325798ac3f9093f6a4581828',1,'mlx::core::ArgPartition::eval_cpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#ad8d48725623ede1ff654fa13eccf2287',1,'mlx::core::ArgReduce::eval_cpu()'],['../classmlx_1_1core_1_1_arg_sort.html#a022079683774bfeb531b3a002cff16fa',1,'mlx::core::ArgSort::eval_cpu()'],['../classmlx_1_1core_1_1_as_type.html#aa89dbf4d73b00c6a44cffd04d5bb228d',1,'mlx::core::AsType::eval_cpu()'],['../classmlx_1_1core_1_1_as_strided.html#acdd4705e4503ff0b124215c4676b4193',1,'mlx::core::AsStrided::eval_cpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a2194bf585213bda1b2966aa02d2fe283',1,'mlx::core::BitwiseBinary::eval_cpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aa85da478cdc6d4a97be06e5d4abee1f2',1,'mlx::core::BlockMaskedMM::eval_cpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#a62352074a480df0e1f879b0bae425730',1,'mlx::core::GatherMM::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast.html#a53d48d9778e2d4c24a124cd767900780',1,'mlx::core::Broadcast::eval_cpu()'],['../classmlx_1_1core_1_1_ceil.html#a9791801fff3f8b79944e15ac2a45a035',1,'mlx::core::Ceil::eval_cpu()'],['../classmlx_1_1core_1_1_compiled.html#ac45b1d0fedd85feefbff7ce7e168b151',1,'mlx::core::Compiled::eval_cpu()'],['../classmlx_1_1core_1_1_concatenate.html#a609e76bede7fc5581ec84ddcb727a258',1,'mlx::core::Concatenate::eval_cpu()'],['../classmlx_1_1core_1_1_conjugate.html#ae39643e2178f442ffba05139f8609d61',1,'mlx::core::Conjugate::eval_cpu()'],['../classmlx_1_1core_1_1_convolution.html#ac74256068da01730629109fa4fa8432b',1,'mlx::core::Convolution::eval_cpu()'],['../classmlx_1_1core_1_1_copy.html#af4a0ebec423e84ffe8083a5e9ed0d70c',1,'mlx::core::Copy::eval_cpu()'],['../classmlx_1_1core_1_1_cos.html#a061fc446268fe56237ae6b20ccf78152',1,'mlx::core::Cos::eval_cpu()'],['../classmlx_1_1core_1_1_cosh.html#ae8702df7e8f0e20cbeccb2a548961d3d',1,'mlx::core::Cosh::eval_cpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#adba1c40c77a2138df6b5f75483f62184',1,'mlx::core::CustomTransforms::eval_cpu()'],['../classmlx_1_1core_1_1_depends.html#a0c7ea6db97337591fa53c6e6bde41e5e',1,'mlx::core::Depends::eval_cpu()'],['../classmlx_1_1core_1_1_divide.html#a823443c2a8e8b81bbcaeee6ddbcdbf49',1,'mlx::core::Divide::eval_cpu()'],['../classmlx_1_1core_1_1_div_mod.html#ae350b7b93ad128e3133ee14f247193b3',1,'mlx::core::DivMod::eval_cpu()'],['../classmlx_1_1core_1_1_select.html#aa51aa36e0adbd69e0d23d7c7adf88de2',1,'mlx::core::Select::eval_cpu()'],['../classmlx_1_1core_1_1_remainder.html#ac6c6c86a0bf02e6e529eb87f6e617ccc',1,'mlx::core::Remainder::eval_cpu()'],['../classmlx_1_1core_1_1_equal.html#aabb8aa61fa581defddcdca1274b1b454',1,'mlx::core::Equal::eval_cpu()'],['../classmlx_1_1core_1_1_erf.html#a84ea16e43d5b7f83bbc2d5ece78a3fb6',1,'mlx::core::Erf::eval_cpu()'],['../classmlx_1_1core_1_1_erf_inv.html#af579627402af3249565134884701d39e',1,'mlx::core::ErfInv::eval_cpu()'],['../classmlx_1_1core_1_1_exp.html#a47934c5a5023bc7ae7ae89bff45ebb2c',1,'mlx::core::Exp::eval_cpu()'],['../classmlx_1_1core_1_1_expm1.html#ab9c8b7aa50fe4592d55f8957baac647a',1,'mlx::core::Expm1::eval_cpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a6bc262a0c2b5d4fe655e3e2e0ff28635',1,'mlx::core::FFT::eval_cpu()'],['../classmlx_1_1core_1_1_floor.html#a1a7dc5f571b7b73e7ef3cbdc1dd1fcf7',1,'mlx::core::Floor::eval_cpu()'],['../classmlx_1_1core_1_1_full.html#a3dccd3756599d7fd018b2af0093b082c',1,'mlx::core::Full::eval_cpu()'],['../classmlx_1_1core_1_1_gather.html#a9ed5587f0d04b59a2b9186c0aac21290',1,'mlx::core::Gather::eval_cpu()'],['../classmlx_1_1core_1_1_greater.html#abe1c03f311d0e0b610f3392a6566f2ae',1,'mlx::core::Greater::eval_cpu()'],['../classmlx_1_1core_1_1_greater_equal.html#a15469125b9bea89b64bfeac01590c075',1,'mlx::core::GreaterEqual::eval_cpu()'],['../classmlx_1_1core_1_1_hadamard.html#ab27d6a9df42b3aab41ace3073a4c880d',1,'mlx::core::Hadamard::eval_cpu()'],['../classmlx_1_1core_1_1_less.html#a32624124ffece066f496b3299056bcef',1,'mlx::core::Less::eval_cpu()'],['../classmlx_1_1core_1_1_less_equal.html#a55d1352b0e97841a92503bc57c19ed16',1,'mlx::core::LessEqual::eval_cpu()'],['../classmlx_1_1core_1_1_load.html#ada026ac30566f3109d8182e35d307c0a',1,'mlx::core::Load::eval_cpu()'],['../classmlx_1_1core_1_1_log.html#aadc7bb4cb24f3ecbbb9ed54a699ab74f',1,'mlx::core::Log::eval_cpu()'],['../classmlx_1_1core_1_1_log1p.html#a8192e5438de99c4cda056987935cba23',1,'mlx::core::Log1p::eval_cpu()'],['../classmlx_1_1core_1_1_logical_not.html#acf3f7b3b20ca69533536e0e0a05725b3',1,'mlx::core::LogicalNot::eval_cpu()'],['../classmlx_1_1core_1_1_logical_and.html#adbe1c1785af1a8b827289d22b0d170b3',1,'mlx::core::LogicalAnd::eval_cpu()'],['../classmlx_1_1core_1_1_logical_or.html#a13cd4cbf26589287e85aeaaca42d7f62',1,'mlx::core::LogicalOr::eval_cpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#abef17fb590b1a8d356f2a580e45d41f0',1,'mlx::core::LogAddExp::eval_cpu()'],['../classmlx_1_1core_1_1_matmul.html#a357a7f57a2a220a91977f810a69413fc',1,'mlx::core::Matmul::eval_cpu()'],['../classmlx_1_1core_1_1_maximum.html#a62b38fbe5f96db58c2b60165ac4eadcf',1,'mlx::core::Maximum::eval_cpu()'],['../classmlx_1_1core_1_1_minimum.html#a6b93f493ee87089943a8085fe59dfc6e',1,'mlx::core::Minimum::eval_cpu()'],['../classmlx_1_1core_1_1_multiply.html#a624fce06c047cdc4dfdbdcaaddb25f34',1,'mlx::core::Multiply::eval_cpu()'],['../classmlx_1_1core_1_1_negative.html#af43553dc418c8ebe75fa9cdcba103c3b',1,'mlx::core::Negative::eval_cpu()'],['../classmlx_1_1core_1_1_not_equal.html#a8f95f8b5873850b875b1641df8196047',1,'mlx::core::NotEqual::eval_cpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#acc328321cf5300874ee884367cbede3f',1,'mlx::core::NumberOfElements::eval_cpu()'],['../classmlx_1_1core_1_1_pad.html#aaf82dd163cd536fbf97304f8b29080cb',1,'mlx::core::Pad::eval_cpu()'],['../classmlx_1_1core_1_1_partition.html#a784596ab567f9f3cb4fe1a69466523d8',1,'mlx::core::Partition::eval_cpu()'],['../classmlx_1_1core_1_1_power.html#a6783da16fb6ff393aaa57737f1973206',1,'mlx::core::Power::eval_cpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ab3dfa73b74d8f4f2e9ab4f0eb016b0e3',1,'mlx::core::QuantizedMatmul::eval_cpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a89aae98bfbdd6563df44ef7d70f0bf8c',1,'mlx::core::GatherQMM::eval_cpu()'],['../classmlx_1_1core_1_1_random_bits.html#a5752d051cd16cf5f8d4754c0a656f0d2',1,'mlx::core::RandomBits::eval_cpu()'],['../classmlx_1_1core_1_1_reshape.html#a658de2c5f710991b48e14b2bd19b229f',1,'mlx::core::Reshape::eval_cpu()'],['../classmlx_1_1core_1_1_reduce.html#aeb8a58b560c0a09ae3a695df7829acfa',1,'mlx::core::Reduce::eval_cpu()'],['../classmlx_1_1core_1_1_round.html#ad066b0944b437f64ab546025efa00007',1,'mlx::core::Round::eval_cpu()'],['../classmlx_1_1core_1_1_scan.html#a15676d9fd066e935782a923fba3e940b',1,'mlx::core::Scan::eval_cpu()'],['../classmlx_1_1core_1_1_scatter.html#a7623f590f8b77167b5ebb4f14bc9dc97',1,'mlx::core::Scatter::eval_cpu()'],['../classmlx_1_1core_1_1_sigmoid.html#aa930ce05734cca529ebcb8d0ca8e1255',1,'mlx::core::Sigmoid::eval_cpu()'],['../classmlx_1_1core_1_1_sign.html#a7498ec993b66879be30c5d9762c45a97',1,'mlx::core::Sign::eval_cpu()'],['../classmlx_1_1core_1_1_sin.html#ab34f9cebc2aed55a0b6ab4c991f02eb5',1,'mlx::core::Sin::eval_cpu()'],['../classmlx_1_1core_1_1_sinh.html#ab6d5f6f40d177f6435f6a51c71b939dd',1,'mlx::core::Sinh::eval_cpu()'],['../classmlx_1_1core_1_1_slice.html#a4b13503f5b2f5c6a90d394b020f9b3f2',1,'mlx::core::Slice::eval_cpu()'],['../classmlx_1_1core_1_1_slice_update.html#ad82ca0e3ab88a0e086431050deea831b',1,'mlx::core::SliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_softmax.html#ac9ebc2eab1683b682e689ed8f4622b79',1,'mlx::core::Softmax::eval_cpu()'],['../classmlx_1_1core_1_1_sort.html#a459769a0241b2620e55bedaba19827cd',1,'mlx::core::Sort::eval_cpu()'],['../classmlx_1_1core_1_1_split.html#aff2889cb9074f0fda53edf8fa40b1fd4',1,'mlx::core::Split::eval_cpu()'],['../classmlx_1_1core_1_1_square.html#a1f4d327a705950616da63b83c2829e59',1,'mlx::core::Square::eval_cpu()'],['../classmlx_1_1core_1_1_sqrt.html#a5a64ecc4eef1e30a2963435dca7cefd5',1,'mlx::core::Sqrt::eval_cpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a56207714d374b08f60e4d9cdbc7340b2',1,'mlx::core::StopGradient::eval_cpu()'],['../classmlx_1_1core_1_1_subtract.html#a47574258b6c95f8ad260c114d6d36a12',1,'mlx::core::Subtract::eval_cpu()'],['../classmlx_1_1core_1_1_tan.html#a9c9a731158fa60eef30067fe0da9f3e9',1,'mlx::core::Tan::eval_cpu()'],['../classmlx_1_1core_1_1_tanh.html#af7ed4345f622da069e5b0284067923f5',1,'mlx::core::Tanh::eval_cpu()'],['../classmlx_1_1core_1_1_uniform.html#a037a2c96b79b70a64f2b637c9f1a432f',1,'mlx::core::Uniform::eval_cpu()'],['../classmlx_1_1core_1_1_view.html#a0ad6deb11914a242f10e8039fcb02497',1,'mlx::core::View::eval_cpu()'],['../classmlx_1_1core_1_1_transpose.html#a1fbcfcca43f9ec06c63a3c14708c30f8',1,'mlx::core::Transpose::eval_cpu()'],['../classmlx_1_1core_1_1_q_r_f.html#a48493887395d65a27f04de1804d277d2',1,'mlx::core::QRF::eval_cpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a637f5c39fa8b10722c04a066f6c1ada6',1,'mlx::core::SVD::eval_cpu()'],['../classmlx_1_1core_1_1_inverse.html#aeb1d8dc9bc4052a616023f65b3c7bb81',1,'mlx::core::Inverse::eval_cpu()'],['../classmlx_1_1core_1_1_cholesky.html#a4bdec36c1cc99aadf9a4a39d4c57bea5',1,'mlx::core::Cholesky::eval_cpu()']]], + ['eval_5fgpu_32',['eval_gpu',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a611ac8dfd9b4f49617ee8976ff3e15e9',1,'mlx::core::distributed::DistPrimitive::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae7955e8d43c097eecae264e804b4d8ca',1,'mlx::core::fast::RMSNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a48efb8fa84c4ba6cc9fb560ebbe01560',1,'mlx::core::fast::RMSNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a77abda7f47bffa2c037a5d60cccc1528',1,'mlx::core::fast::LayerNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a954a003a4a27c8c4c60a5a14142a9cc3',1,'mlx::core::fast::LayerNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a913b6b00fc518b25ac3947e4e15790f2',1,'mlx::core::fast::RoPE::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a505f38ba93a3499895f5312e0112e73d',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ad51666e69f670e286293aff96eb435a9',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, array &out)'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a63812b2abaf26ad7e7fa4c9e82db1628',1,'mlx::core::fast::AffineQuantize::eval_gpu()'],['../classmlx_1_1core_1_1_primitive.html#ad217376dcf5eff691d731566faec2ba2',1,'mlx::core::Primitive::eval_gpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a6b7f80abaf038d53ec6ffbb0dfac6adb',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#a971fe9ad47f6569118879ce1d0f41447',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0a976e636dd8505b473fbdddf949f514',1,'mlx::core::Abs::eval_gpu()'],['../classmlx_1_1core_1_1_add.html#aa0aacbc1e26b95a2f040f62aa4f69c3d',1,'mlx::core::Add::eval_gpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a5f933be14baebc32a0be0f9a69148aa9',1,'mlx::core::AddMM::eval_gpu()'],['../classmlx_1_1core_1_1_arange.html#a7a2e9787c6c3a78b4a6df91206974031',1,'mlx::core::Arange::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a46f72d4af89b0a0f5f203783fb44589c',1,'mlx::core::ArcCos::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#aa6a2587485a0e015ac2d5211d7d045fc',1,'mlx::core::ArcCosh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sin.html#a7fa4ae7a85bc8bed97ea258ae30762f3',1,'mlx::core::ArcSin::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79f648a86de4c10386a1ce3b5e38e8ac',1,'mlx::core::ArcSinh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a77866feb27028865d844070447c9a254',1,'mlx::core::ArcTan::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a76d3f0c29e0ff4642b8d39dac90d3f50',1,'mlx::core::ArcTan2::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a10566b9d3b2c7d090895b46d9040bc1d',1,'mlx::core::ArcTanh::eval_gpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a9a60995eaf85f63c877e86b23cbc15fc',1,'mlx::core::ArgPartition::eval_gpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#aafa982ce2abc0cd9e81e43aa2c823d29',1,'mlx::core::ArgReduce::eval_gpu()'],['../classmlx_1_1core_1_1_arg_sort.html#abc2d730850ec4ee8d7968b7417911709',1,'mlx::core::ArgSort::eval_gpu()'],['../classmlx_1_1core_1_1_as_type.html#a5b111b9d74c60d27b4a7ebaa49f96e0b',1,'mlx::core::AsType::eval_gpu()'],['../classmlx_1_1core_1_1_as_strided.html#ab6771a208323994927ca162ba7bb10ed',1,'mlx::core::AsStrided::eval_gpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#ac831a29fc46701b00bbe63ee33832afd',1,'mlx::core::BitwiseBinary::eval_gpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ab372b6df4de00a33795a052a23bb1df9',1,'mlx::core::BlockMaskedMM::eval_gpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#ad754c35f460a055cc383ad93a5f72da1',1,'mlx::core::GatherMM::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast.html#ab9bd9dbcedcefc9b29c84911b5ce69fe',1,'mlx::core::Broadcast::eval_gpu()'],['../classmlx_1_1core_1_1_ceil.html#abe178e0058e44b6618be414215e96887',1,'mlx::core::Ceil::eval_gpu()'],['../classmlx_1_1core_1_1_compiled.html#aa3d5ff0f2b3554ad48fbbf2a0f3336d5',1,'mlx::core::Compiled::eval_gpu()'],['../classmlx_1_1core_1_1_concatenate.html#a309a1c50e97f9925866433ee2841c474',1,'mlx::core::Concatenate::eval_gpu()'],['../classmlx_1_1core_1_1_conjugate.html#aff0a802166e3724db88ab5d3feb2d3de',1,'mlx::core::Conjugate::eval_gpu()'],['../classmlx_1_1core_1_1_convolution.html#a30b64109eeb1778f002b99447dff9dd2',1,'mlx::core::Convolution::eval_gpu()'],['../classmlx_1_1core_1_1_copy.html#a1eda7b2ea771a168f67421f0d384b3a1',1,'mlx::core::Copy::eval_gpu()'],['../classmlx_1_1core_1_1_cos.html#a5ef41aafad595f6cdd8c535e36e12060',1,'mlx::core::Cos::eval_gpu()'],['../classmlx_1_1core_1_1_cosh.html#a23f71b43792934c3ec0ebe9b74f32559',1,'mlx::core::Cosh::eval_gpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#a7b3538681acbb20af3ed37b0877f6667',1,'mlx::core::CustomTransforms::eval_gpu()'],['../classmlx_1_1core_1_1_depends.html#ae5057f65e69490ad0add8eeda2b75e28',1,'mlx::core::Depends::eval_gpu()'],['../classmlx_1_1core_1_1_divide.html#abffda0ce37221ddc28dc9eea794f6bc7',1,'mlx::core::Divide::eval_gpu()'],['../classmlx_1_1core_1_1_div_mod.html#a003117c9ecf3c06a27248f72a76348dc',1,'mlx::core::DivMod::eval_gpu()'],['../classmlx_1_1core_1_1_select.html#a2a82b6cba4c386b2b87f225a4b08ea9b',1,'mlx::core::Select::eval_gpu()'],['../classmlx_1_1core_1_1_remainder.html#a7919ea9b84e42522d51bf0d5a396e161',1,'mlx::core::Remainder::eval_gpu()'],['../classmlx_1_1core_1_1_equal.html#ac3757001fec42ceb5ece2954df42161c',1,'mlx::core::Equal::eval_gpu()'],['../classmlx_1_1core_1_1_erf.html#ad8551be664d767dccc3c0d8cc1eca008',1,'mlx::core::Erf::eval_gpu()'],['../classmlx_1_1core_1_1_erf_inv.html#a4a2413d0634db1f3dae1806ddfa632db',1,'mlx::core::ErfInv::eval_gpu()'],['../classmlx_1_1core_1_1_exp.html#a7d63695a97a14760fd33b5d4e6590822',1,'mlx::core::Exp::eval_gpu()'],['../classmlx_1_1core_1_1_expm1.html#a82930071f4b77d883b300f77966aff5f',1,'mlx::core::Expm1::eval_gpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a1c21b26d1e9ad7c4da78ae845721b2dd',1,'mlx::core::FFT::eval_gpu()'],['../classmlx_1_1core_1_1_floor.html#aaa29c83538099eb8f951c95a41f2eb65',1,'mlx::core::Floor::eval_gpu()'],['../classmlx_1_1core_1_1_full.html#aa54f99bb4cba12a551392dea56003872',1,'mlx::core::Full::eval_gpu()'],['../classmlx_1_1core_1_1_gather.html#aec48ee529cb2449915a7b27a3c4361e8',1,'mlx::core::Gather::eval_gpu()'],['../classmlx_1_1core_1_1_greater.html#ae8957cccf4c924d941f57a1bb751c878',1,'mlx::core::Greater::eval_gpu()'],['../classmlx_1_1core_1_1_greater_equal.html#ac246263b4548126c3d4ab7e392575d24',1,'mlx::core::GreaterEqual::eval_gpu()'],['../classmlx_1_1core_1_1_hadamard.html#a2470feb690f5463138490763c38b5733',1,'mlx::core::Hadamard::eval_gpu()'],['../classmlx_1_1core_1_1_less.html#a353335ce06ddbe8498d86d129c835917',1,'mlx::core::Less::eval_gpu()'],['../classmlx_1_1core_1_1_less_equal.html#acf035a82b11e6f63742143ea540fedac',1,'mlx::core::LessEqual::eval_gpu()'],['../classmlx_1_1core_1_1_load.html#a06933e887ea94a4d01d81195c5e07a3d',1,'mlx::core::Load::eval_gpu()'],['../classmlx_1_1core_1_1_log.html#aaaa49e9455f3a197bc319646b5ca6390',1,'mlx::core::Log::eval_gpu()'],['../classmlx_1_1core_1_1_log1p.html#a1b97decae7338d46874e736c95fa7431',1,'mlx::core::Log1p::eval_gpu()'],['../classmlx_1_1core_1_1_logical_not.html#a1d0d2bc93f935eca6c85ef7bf67f2d6a',1,'mlx::core::LogicalNot::eval_gpu()'],['../classmlx_1_1core_1_1_logical_and.html#a132b2eedaa3978de5a5350da3c2ca40f',1,'mlx::core::LogicalAnd::eval_gpu()'],['../classmlx_1_1core_1_1_logical_or.html#a3be1da328f0f8620de2e4fc1d22a077a',1,'mlx::core::LogicalOr::eval_gpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#acace355b62ec00df649f9f99e8f2eb7a',1,'mlx::core::LogAddExp::eval_gpu()'],['../classmlx_1_1core_1_1_matmul.html#a8707a4e9b75c769e8f1dbca15c6a1ae7',1,'mlx::core::Matmul::eval_gpu()'],['../classmlx_1_1core_1_1_maximum.html#ade0f721b10a6b3a12bdadd34c48f72a7',1,'mlx::core::Maximum::eval_gpu()'],['../classmlx_1_1core_1_1_minimum.html#aadc68afa0afbe2103f19d161f5e0a2ba',1,'mlx::core::Minimum::eval_gpu()'],['../classmlx_1_1core_1_1_multiply.html#a634fcb4e981d8d3f4d94252caf25bee0',1,'mlx::core::Multiply::eval_gpu()'],['../classmlx_1_1core_1_1_negative.html#a97f1b316eace0c6d9e576d766940c75b',1,'mlx::core::Negative::eval_gpu()'],['../classmlx_1_1core_1_1_not_equal.html#a61179747e34e203150e9c660dfddb5f2',1,'mlx::core::NotEqual::eval_gpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#a2c98c42915fb2bfe12f5c99ea553eff5',1,'mlx::core::NumberOfElements::eval_gpu()'],['../classmlx_1_1core_1_1_pad.html#aefd4d3a5bd8b6b35b266c9e558ada153',1,'mlx::core::Pad::eval_gpu()'],['../classmlx_1_1core_1_1_partition.html#a8eca1be21ae9ccfda46e6f3e85f506ef',1,'mlx::core::Partition::eval_gpu()'],['../classmlx_1_1core_1_1_power.html#a80577d4c0853c24027777c90a1ec7e11',1,'mlx::core::Power::eval_gpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a2812ad007d695ed1aaf9cf706fb9c4b3',1,'mlx::core::QuantizedMatmul::eval_gpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a86eb048afc95646b2e96ec5493e3d887',1,'mlx::core::GatherQMM::eval_gpu()'],['../classmlx_1_1core_1_1_random_bits.html#a578756866665358577418e4cdd94aa3a',1,'mlx::core::RandomBits::eval_gpu()'],['../classmlx_1_1core_1_1_reshape.html#aa1e85f28471875750c47351520b56059',1,'mlx::core::Reshape::eval_gpu()'],['../classmlx_1_1core_1_1_reduce.html#ae9caaf42edadfe73ea208d98f526890f',1,'mlx::core::Reduce::eval_gpu()'],['../classmlx_1_1core_1_1_round.html#af7fe5ff8f3db166c203b4be4b07f13ec',1,'mlx::core::Round::eval_gpu()'],['../classmlx_1_1core_1_1_scan.html#aef22c6fc2b2cb2a907cd8965c7413dde',1,'mlx::core::Scan::eval_gpu()'],['../classmlx_1_1core_1_1_scatter.html#ab304345db3d8cfeea15e27461ae2e678',1,'mlx::core::Scatter::eval_gpu()'],['../classmlx_1_1core_1_1_sigmoid.html#a7a6bd0222d51d7f25f2719a91ccdfeca',1,'mlx::core::Sigmoid::eval_gpu()'],['../classmlx_1_1core_1_1_sign.html#afa2b48b99a194106006b44af69ffda8b',1,'mlx::core::Sign::eval_gpu()'],['../classmlx_1_1core_1_1_sin.html#a6b59f1156cf8bdad8d45acd1d825cb5e',1,'mlx::core::Sin::eval_gpu()'],['../classmlx_1_1core_1_1_sinh.html#a5a1af2399f166d5b228b5e83a1837c75',1,'mlx::core::Sinh::eval_gpu()'],['../classmlx_1_1core_1_1_slice.html#aa53c21ff06a7c659e889af6b97d10a4a',1,'mlx::core::Slice::eval_gpu()'],['../classmlx_1_1core_1_1_slice_update.html#aac1a1d122e5697be057d63552141032b',1,'mlx::core::SliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_softmax.html#a35dac69ddcc7e2ec0e1a76fe93db85af',1,'mlx::core::Softmax::eval_gpu()'],['../classmlx_1_1core_1_1_sort.html#a4141c48f0e8670c728663f3722675382',1,'mlx::core::Sort::eval_gpu()'],['../classmlx_1_1core_1_1_split.html#a78ddda89c4daee73c74cfbc1e44656df',1,'mlx::core::Split::eval_gpu()'],['../classmlx_1_1core_1_1_square.html#a0ea2a78a5bb52daa4103263bf2f98045',1,'mlx::core::Square::eval_gpu()'],['../classmlx_1_1core_1_1_sqrt.html#a6d205e679a593d1ba20206c5c47ba501',1,'mlx::core::Sqrt::eval_gpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a907b96f0a1ce608e211d87ccf2b9ca89',1,'mlx::core::StopGradient::eval_gpu()'],['../classmlx_1_1core_1_1_subtract.html#a69021b23daf061764d97fabbc0f4f06c',1,'mlx::core::Subtract::eval_gpu()'],['../classmlx_1_1core_1_1_tan.html#aca7dbb4836507005a2032ac957a04d3f',1,'mlx::core::Tan::eval_gpu()'],['../classmlx_1_1core_1_1_tanh.html#a48df896599ae93dbce84a5c0f50cf761',1,'mlx::core::Tanh::eval_gpu()'],['../classmlx_1_1core_1_1_uniform.html#a5f88cbf2495f24f87cefd99aaaebe4d0',1,'mlx::core::Uniform::eval_gpu()'],['../classmlx_1_1core_1_1_view.html#add6e12ff1e476fe1db7718b14f21b075',1,'mlx::core::View::eval_gpu()'],['../classmlx_1_1core_1_1_transpose.html#a38d25739c08aa594a6775015a1d7d92e',1,'mlx::core::Transpose::eval_gpu()'],['../classmlx_1_1core_1_1_q_r_f.html#ae5fa3482192f4713605cd07e7fc1c6c9',1,'mlx::core::QRF::eval_gpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a7067b2207f826a25549d571856b94e83',1,'mlx::core::SVD::eval_gpu()'],['../classmlx_1_1core_1_1_inverse.html#a086fbbc947ad232e01686ad063a78ed2',1,'mlx::core::Inverse::eval_gpu()'],['../classmlx_1_1core_1_1_cholesky.html#a8c918594bf129888044ef37fcae56795',1,'mlx::core::Cholesky::eval_gpu()']]], + ['event_33',['Event',['../classmlx_1_1core_1_1_event.html',1,'mlx::core::Event'],['../classmlx_1_1core_1_1_event.html#a833506419b2110ad1abd89b2dd238b4d',1,'mlx::core::Event::Event()=default'],['../classmlx_1_1core_1_1_event.html#a13e4835f2ffb2cc22e29148a448ea184',1,'mlx::core::Event::Event(const Stream &steam)']]], + ['event_34',['event',['../classmlx_1_1core_1_1array.html#a0a8e4d6e67e739a712876bb36f88f9bf',1,'mlx::core::array']]], + ['event_2eh_35',['event.h',['../event_8h.html',1,'']]], + ['excess_36',['excess',['../structmlx_1_1steel_1_1_channel_helper.html#afc34bf92168c1865a9611b319dbcd000',1,'mlx::steel::ChannelHelper::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#ada22a8bd8a89078cfa28874055c8e753',1,'mlx::steel::ChannelHelper< 1 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#acc490f3999230aa592c61bbed7eb7cfe',1,'mlx::steel::ChannelHelper< 2 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#aae404674763f3dc73c5ab29169f8b80f',1,'mlx::steel::ChannelHelper< 3 >::excess'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#aecdd8331fec703d739a6f07b9b901ac8',1,'mlx::steel::ChannelHelper< 4 >::excess']]], + ['exec_37',['exec',['../classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2',1,'pocketfft::detail::cfftp::exec()'],['../classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac',1,'pocketfft::detail::rfftp::exec()'],['../classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c',1,'pocketfft::detail::fftblue::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d',1,'pocketfft::detail::pocketfft_c::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33',1,'pocketfft::detail::pocketfft_r::exec()'],['../classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058',1,'pocketfft::detail::T_dct1::exec()'],['../classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26',1,'pocketfft::detail::T_dst1::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac',1,'pocketfft::detail::T_dcst23::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708',1,'pocketfft::detail::T_dcst4::exec()']]], + ['exec_5fr_38',['exec_r',['../classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00',1,'pocketfft::detail::fftblue']]], + ['execc2c_39',['ExecC2C',['../structpocketfft_1_1detail_1_1_exec_c2_c.html',1,'pocketfft::detail']]], + ['execdcst_40',['ExecDcst',['../structpocketfft_1_1detail_1_1_exec_dcst.html',1,'pocketfft::detail']]], + ['exechartley_41',['ExecHartley',['../structpocketfft_1_1detail_1_1_exec_hartley.html',1,'pocketfft::detail']]], + ['execr2r_42',['ExecR2R',['../structpocketfft_1_1detail_1_1_exec_r2_r.html',1,'pocketfft::detail']]], + ['exp_43',['Exp',['../struct_exp.html',1,'Exp'],['../structmlx_1_1core_1_1detail_1_1_exp.html',1,'mlx::core::detail::Exp'],['../classmlx_1_1core_1_1_exp.html',1,'mlx::core::Exp'],['../classmlx_1_1core_1_1_exp.html#a1d0a618cbb91ab29ef53b57ff6ed6e06',1,'mlx::core::Exp::Exp()']]], + ['exp_44',['exp',['../namespacemetal.html#ac2a0b3618d922ac014baac8189d44650',1,'metal::exp()'],['../namespacemetal_1_1fast.html#ad3dbd387b63373c29e3449609f763ede',1,'metal::fast::exp()'],['../namespacemetal_1_1precise.html#a8d8d2d5700ce432b33cf47cf22528e8f',1,'metal::precise::exp()'],['../group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125',1,'mlx::core::exp()']]], + ['exp10_45',['exp10',['../namespacemetal.html#a4c63707d13c89364496a48906631c204',1,'metal::exp10()'],['../namespacemetal_1_1fast.html#a453122f982485cbb4e471b3ac282ee5e',1,'metal::fast::exp10()'],['../namespacemetal_1_1precise.html#af9addb343c967da3a83e9e123a8521fd',1,'metal::precise::exp10()']]], + ['exp2_46',['exp2',['../namespacemetal.html#a228201c20777848804a4d0589c1d33e7',1,'metal::exp2()'],['../namespacemetal_1_1fast.html#ac092b65a46720adaf22f6266671d2d71',1,'metal::fast::exp2()'],['../namespacemetal_1_1precise.html#a92a880bd2197efc0da0f8f0f7ec1e4c9',1,'metal::precise::exp2()']]], + ['expand_5fdims_47',['expand_dims',['../group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133',1,'mlx::core::expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a',1,'mlx::core::expand_dims(const array &a, int axis, StreamOrDevice s={})']]], + ['expm1_48',['Expm1',['../struct_expm1.html',1,'Expm1'],['../structmlx_1_1core_1_1detail_1_1_expm1.html',1,'mlx::core::detail::Expm1'],['../classmlx_1_1core_1_1_expm1.html',1,'mlx::core::Expm1'],['../classmlx_1_1core_1_1_expm1.html#a47c2a1b2a4ef6bb07ba77c55ddddaec2',1,'mlx::core::Expm1::Expm1()']]], + ['expm1_49',['expm1',['../group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0',1,'mlx::core']]], + ['expm1f_50',['expm1f',['../expm1f_8h.html#a87f66d30e185950f42ce3641783cdc40',1,'expm1f.h']]], + ['expm1f_2eh_51',['expm1f.h',['../expm1f_8h.html',1,'']]], + ['expm1f_5fscaled_5funchecked_52',['expm1f_scaled_unchecked',['../expm1f_8h.html#adf20e03405fba634ca8d01acac24592e',1,'expm1f.h']]], + ['export_5fto_5fdot_53',['export_to_dot',['../namespacemlx_1_1core.html#a57395bdf43d9c5c134e610c169222cca',1,'mlx::core::export_to_dot(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a839f94dbad44f0d37333006fc876b42e',1,'mlx::core::export_to_dot(std::ostream &os, Arrays &&... outputs)']]], + ['eye_54',['eye',['../group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3',1,'mlx::core::eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd',1,'mlx::core::eye(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga61657db78ef35d41112d362c869c25d2',1,'mlx::core::eye(int n, int m, StreamOrDevice s={})'],['../group__ops.html#ga908a15b42834be498a46856c99dfc779',1,'mlx::core::eye(int n, int m, int k, StreamOrDevice s={})'],['../group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f',1,'mlx::core::eye(int n, StreamOrDevice s={})']]] ]; diff --git a/docs/build/html/search/all_7.js b/docs/build/html/search/all_7.js index d9d69806cf..84ce58d140 100644 --- a/docs/build/html/search/all_7.js +++ b/docs/build/html/search/all_7.js @@ -11,7 +11,7 @@ var searchData= ['gathermm_8',['GatherMM',['../classmlx_1_1core_1_1_gather_m_m.html',1,'mlx::core::GatherMM'],['../classmlx_1_1core_1_1_gather_m_m.html#afd9bbc08138181b80e2fb86536ff3f2a',1,'mlx::core::GatherMM::GatherMM()']]], ['gatherqmm_9',['GatherQMM',['../classmlx_1_1core_1_1_gather_q_m_m.html',1,'mlx::core::GatherQMM'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a60ed2ade7f10dd9c9314913a810f9360',1,'mlx::core::GatherQMM::GatherQMM()']]], ['gemm_10',['gemm',['../namespacemlx_1_1core_1_1metal.html#ac46fd23516a61fc56d997910e4144281',1,'mlx::core::metal::gemm()'],['../steel__gemm__fused_8h.html#aa40dd40b9a0bbf20c8911032ed0c3e6d',1,'gemm(const device T *A, const device T *B, const device T *C, device T *D, const constant GEMMParams *params, const constant GEMMAddMMParams *addmm_params, const constant int *batch_shape, const constant size_t *batch_strides, const constant uint32_t *lhs_indices, const constant uint32_t *rhs_indices, const constant uint32_t *C_indices, const constant int *operand_shape, const constant size_t *operand_strides, const constant packed_int3 &operand_batch_ndim, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_fused.h']]], - ['gemm_2eh_11',['gemm.h',['../kernels_2steel_2gemm_2gemm_8h.html',1,'(Global Namespace)'],['../mps_2gemm_8h.html',1,'(Global Namespace)']]], + ['gemm_2eh_11',['gemm.h',['../gemm_8h.html',1,'']]], ['gemm_5fk_5fiterations_12',['gemm_k_iterations',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a8b50863e4e2d3481c154be6c3629bf51',1,'mlx::steel::ImplicitGemmConv2DParams']]], ['gemm_5fk_5fiterations_5faligned_13',['gemm_k_iterations_aligned',['../struct_m_l_x_fast_attention_params.html#adbc0a13076da5f704498e57239cb2bf2',1,'MLXFastAttentionParams::gemm_k_iterations_aligned'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0d7f419ba265805b418e93ce1ca2e0f9',1,'mlx::steel::GEMMParams::gemm_k_iterations_aligned'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#aa37e05a03ac8b34ec7dc31ca42f68998',1,'mlx::steel::GEMMSpiltKParams::gemm_k_iterations_aligned']]], ['gemm_5floop_14',['gemm_loop',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a756d7bbcc96e2919cd65eec4bc135780',1,'mlx::steel::GEMMKernel']]], @@ -25,77 +25,85 @@ var searchData= ['gemmkernel_22',['GEMMKernel',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html',1,'mlx::steel']]], ['gemmparams_23',['GEMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_params.html',1,'mlx::steel']]], ['gemmspiltkparams_24',['GEMMSpiltKParams',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html',1,'mlx::steel']]], - ['general_25',['General',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a0db377921f4ce762c62526131097968f',1,'mlx::core']]], - ['general_5fc2r_26',['general_c2r',['../namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc',1,'pocketfft::detail']]], - ['general_5fnd_27',['general_nd',['../namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0',1,'pocketfft::detail']]], - ['general_5fr2c_28',['general_r2c',['../namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed',1,'pocketfft::detail']]], - ['generalcontiguousreduce_29',['GeneralContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a540cf31fe6858115a02e789938297cdb',1,'mlx::core']]], - ['generalgeneral_30',['GeneralGeneral',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a6fe62e8ce1fae1e70cb9eeaa67d29dab',1,'mlx::core']]], - ['generalreduce_31',['GeneralReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a3622f95ed0ec99657f9ad8ef39ec2184',1,'mlx::core']]], - ['generalstridedreduce_32',['GeneralStridedReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ad999b1a8ae1d7436efb5ffdfafb1dd3d',1,'mlx::core']]], - ['generic_33',['generic',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da3d517f8924ac7fd03699a29d97dc52d9',1,'mlx::core::Dtype::generic'],['../namespacemlx_1_1core.html#a34d69c4d46aa9b2a4a79dba7aba093d2',1,'mlx::core::generic']]], - ['get_5factive_5fmemory_34',['get_active_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a7a3ad4e33d57a47474c98e2f88e775d7',1,'mlx::core::metal::MetalAllocator::get_active_memory()'],['../namespacemlx_1_1core_1_1metal.html#a7b75c2639016ac4d350fa6c9da386667',1,'mlx::core::metal::get_active_memory()']]], - ['get_5farange_5fkernel_35',['get_arange_kernel',['../namespacemlx_1_1core.html#a76f614e9956a6ca05a9be4db5a483446',1,'mlx::core']]], - ['get_5fbinary_5fkernel_36',['get_binary_kernel',['../namespacemlx_1_1core.html#a4decd4a07d91487e6903f6e3c8b7513a',1,'mlx::core']]], - ['get_5fbinary_5ftwo_5fkernel_37',['get_binary_two_kernel',['../namespacemlx_1_1core.html#a4e809746f48e5dcf7fa63215d3f5e33e',1,'mlx::core']]], - ['get_5fcache_5fmemory_38',['get_cache_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ad3cabbe638917ca4114eb74dcabe381f',1,'mlx::core::metal::MetalAllocator::get_cache_memory()'],['../namespacemlx_1_1core_1_1metal.html#a43307654f62ed7c58e014be7fb03909c',1,'mlx::core::metal::get_cache_memory()']]], - ['get_5fcolocated_5fmtllib_5fpath_39',['get_colocated_mtllib_path',['../namespacemlx_1_1core_1_1metal.html#a5fd6ba2040e53a254b9d71ae7ebd315f',1,'mlx::core::metal']]], - ['get_5fcommand_5fbuffer_40',['get_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a5fe3970fbe92ccc55fce4241ffbe5210',1,'mlx::core::metal::Device']]], - ['get_5fcommand_5fbuffer_5fops_41',['get_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a064e1cb6a16de7a0619f6447622350f8',1,'mlx::core::metal::Device']]], - ['get_5fcommand_5fencoder_42',['get_command_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#affa682ef612def4890f5152f81ffb7e6',1,'mlx::core::metal::Device']]], - ['get_5fcopy_5fkernel_43',['get_copy_kernel',['../namespacemlx_1_1core.html#a05a220cff45f12439fde775983c6df78',1,'mlx::core']]], - ['get_5fdefault_5fstream_44',['get_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a2366c7b888e433608e203752edc92282',1,'mlx::core::scheduler::Scheduler']]], - ['get_5ffft_5fkernel_45',['get_fft_kernel',['../namespacemlx_1_1core.html#a1d4cffc3c78067b3d9a62d64f3fb686f',1,'mlx::core']]], - ['get_5ffunction_46',['get_function',['../classmlx_1_1core_1_1metal_1_1_device.html#abf4ecf035e8be2ff00d2a7e95475d7ac',1,'mlx::core::metal::Device::get_function(const std::string &base_name, MTL::Library *mtl_lib, const std::string &specialized_name="", const MTLFCList &func_consts={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#ac083535ead07161cb27ad31f8f64fb84',1,'mlx::core::metal::Device::get_function(const std::string &base_name, const std::string &lib_name="mlx", const std::string &specialized_name="", const MTLFCList &func_consts={})']]], - ['get_5fkernel_47',['get_kernel',['../classmlx_1_1core_1_1metal_1_1_device.html#a6810c4dcbcfbf93fc51d42aa5ff0fc3a',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#afa0cac9d800c21a8a7f6cb224256abaf',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, const std::string &lib_name="mlx", const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})']]], - ['get_5fkernel_5fpreamble_48',['get_kernel_preamble',['../compiled__preamble_8h.html#a1dfa17a0369fb90ff615c7461f5013f3',1,'compiled_preamble.h']]], - ['get_5flibrary_49',['get_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a86297d66de43ee77d0697f69c6bd0b87',1,'mlx::core::metal::Device::get_library(const std::string &name)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a961d3cdb4bfb14c1a0b0aeb05609f56e',1,'mlx::core::metal::Device::get_library(const std::string &name, const std::string &source_string, bool cache=true)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a8d8d9e28ffeea10d687b91ea480fb4bd',1,'mlx::core::metal::Device::get_library(const std::string &name, const MTL::StitchedLibraryDescriptor *desc, bool cache=true)']]], - ['get_5fmb_5fsort_5fkernel_50',['get_mb_sort_kernel',['../namespacemlx_1_1core.html#afb57825bb763050cc9a9d194aa41ac36',1,'mlx::core']]], - ['get_5fname_51',['get_name',['../structmlx_1_1core_1_1_node_namer.html#a1690dd38de288c0aee2bb53156eb770e',1,'mlx::core::NodeNamer']]], - ['get_5fpeak_5fmemory_52',['get_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ac7972a3fe58e69489de775a0f152da17',1,'mlx::core::metal::MetalAllocator::get_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#a4b67d680cefa95f0ed5801f0e14e48ce',1,'mlx::core::metal::get_peak_memory()']]], - ['get_5fplan_53',['get_plan',['../namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b',1,'pocketfft::detail']]], - ['get_5fpool_54',['get_pool',['../namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a',1,'pocketfft::detail::threading']]], - ['get_5fquantized_5fkernel_55',['get_quantized_kernel',['../namespacemlx_1_1core.html#aa3faeae5378bfaafe3ce3432a051e43e',1,'mlx::core']]], - ['get_5freduce_5finit_5fkernel_56',['get_reduce_init_kernel',['../namespacemlx_1_1core.html#a51c4bb09230348bd0252e22bfdc9bc89',1,'mlx::core']]], - ['get_5freduce_5fkernel_57',['get_reduce_kernel',['../namespacemlx_1_1core.html#a5d6843dcbdbd4d12851fc2edbafbc9e6',1,'mlx::core']]], - ['get_5freduction_5fplan_58',['get_reduction_plan',['../namespacemlx_1_1core.html#a1555dc378c5254e79199421761f26f2b',1,'mlx::core']]], - ['get_5fscan_5fkernel_59',['get_scan_kernel',['../namespacemlx_1_1core.html#aeefaff208444d3fa61ecc0946fe1de5f',1,'mlx::core']]], - ['get_5fshape_60',['get_shape',['../namespacemlx_1_1core.html#aab0d8a256957984acc1e3615c65c898e',1,'mlx::core']]], - ['get_5fsoftmax_5fkernel_61',['get_softmax_kernel',['../namespacemlx_1_1core.html#a35a412f688d79eb47e42d20a7c8650ee',1,'mlx::core']]], - ['get_5fsort_5fkernel_62',['get_sort_kernel',['../namespacemlx_1_1core.html#a84ebe6275218070f0ea320f126f64e22',1,'mlx::core']]], - ['get_5fsteel_5fconv_5fgeneral_5fkernel_63',['get_steel_conv_general_kernel',['../namespacemlx_1_1core.html#abce2b67044ee06a7bbe7a91ec7c8c48d',1,'mlx::core']]], - ['get_5fsteel_5fconv_5fkernel_64',['get_steel_conv_kernel',['../namespacemlx_1_1core.html#adce79d220672f5f3c65cc31d145ca9c4',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5ffused_5fkernel_65',['get_steel_gemm_fused_kernel',['../namespacemlx_1_1core.html#a84fa8e0aee321a9d614433a0b933103b',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fmasked_5fkernel_66',['get_steel_gemm_masked_kernel',['../namespacemlx_1_1core.html#ab5f60614e965144b451930fdf935e08d',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fsplitk_5faccum_5fkernel_67',['get_steel_gemm_splitk_accum_kernel',['../namespacemlx_1_1core.html#a195b86cad5bb99aa1bcd23952305af6b',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fsplitk_5fkernel_68',['get_steel_gemm_splitk_kernel',['../namespacemlx_1_1core.html#af48c6f2f72b61dbd6766e4f5fea85df5',1,'mlx::core']]], - ['get_5ftemplate_5fdefinition_69',['get_template_definition',['../namespacemlx_1_1core.html#aae0d19f0acdef2accd2428fb84c8a032',1,'mlx::core']]], - ['get_5fternary_5fkernel_70',['get_ternary_kernel',['../namespacemlx_1_1core.html#a54eb3b65375022428aab5f810e40624b',1,'mlx::core']]], - ['get_5ftwiddle_71',['get_twiddle',['../radix_8h.html#ac5cf950316b9445296ee9ecfc56a56bd',1,'radix.h']]], - ['get_5ftype_5fstring_72',['get_type_string',['../namespacemlx_1_1core.html#af776fd91dd60594dcfebbafd17f19068',1,'mlx::core']]], - ['get_5funary_5fkernel_73',['get_unary_kernel',['../namespacemlx_1_1core.html#a15175e8e2b1e26726c63393e4d68b628',1,'mlx::core']]], - ['gguf_2eh_74',['gguf.h',['../gguf_8h.html',1,'']]], - ['gguf_5fload_5fquantized_75',['gguf_load_quantized',['../namespacemlx_1_1core.html#a65dd68163bdaef3631e3724327782498',1,'mlx::core']]], - ['ggufload_76',['GGUFLoad',['../namespacemlx_1_1core.html#aa5b0f7f13a941e1f41c411194e9033c7',1,'mlx::core']]], - ['ggufmetadata_77',['GGUFMetaData',['../namespacemlx_1_1core.html#a8c2c1b9a37aadfb48f4c3a7e806e32e3',1,'mlx::core']]], - ['global_5fformatter_78',['global_formatter',['../namespacemlx_1_1core.html#af5a408a78cc934717dd711ddfda58ea6',1,'mlx::core']]], - ['good_79',['good',['../classmlx_1_1core_1_1io_1_1_reader.html#a005d0b52c1f34866f7412b7f41dabec3',1,'mlx::core::io::Reader::good()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a0b050c2c27487007e250e2e19560ffe4',1,'mlx::core::io::Writer::good()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a4ecb046dfa7dab346d096adab3a1c318',1,'mlx::core::io::FileReader::good()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9ec4934b26fb358d699ddce1482b2d54',1,'mlx::core::io::FileWriter::good()']]], - ['good_5fsize_5fcmplx_80',['good_size_cmplx',['../structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624',1,'pocketfft::detail::util']]], - ['good_5fsize_5freal_81',['good_size_real',['../structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa',1,'pocketfft::detail::util']]], - ['gpu_82',['gpu',['../structmlx_1_1core_1_1_device.html#a45ed081b56ae5d4ddd39c83a5d8a1616',1,'mlx::core::Device::gpu'],['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdba0aa0be2a866411d9ff03515227454947',1,'mlx::core::Device::gpu']]], - ['grad_83',['grad',['../namespacemlx_1_1core.html#a99994806f81e83f98746c9e5ba4c3673',1,'mlx::core::grad'],['../namespacemlx_1_1core.html#af482f6c64acd77c57ef5bb4b7be9726c',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a64bc619876b0f8cc81a2637ca81c99f7',1,'mlx::core::grad(const std::function< array(const array &)> &fun)']]], - ['graph_5futils_2eh_84',['graph_utils.h',['../graph__utils_8h.html',1,'']]], - ['greater_85',['Greater',['../struct_greater.html',1,'Greater'],['../structmlx_1_1core_1_1detail_1_1_greater.html',1,'mlx::core::detail::Greater'],['../classmlx_1_1core_1_1_greater.html',1,'mlx::core::Greater'],['../classmlx_1_1core_1_1_greater.html#a1d5992a66c020cd97a70e8e3d8cd1a1b',1,'mlx::core::Greater::Greater()']]], - ['greater_86',['greater',['../group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d',1,'mlx::core']]], - ['greater_5fequal_87',['greater_equal',['../group__ops.html#ga7153071bcfff6faad21332163fb9a430',1,'mlx::core']]], - ['greaterequal_88',['GreaterEqual',['../struct_greater_equal.html',1,'GreaterEqual'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html',1,'mlx::core::detail::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html',1,'mlx::core::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html#a19a3c49d5a9b40e17da0e56ef6908527',1,'mlx::core::GreaterEqual::GreaterEqual()']]], - ['grid_89',['grid',['../struct_read_writer.html#ac7a957f99873d3797081f5d620f3b2c8',1,'ReadWriter']]], - ['group_90',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html',1,'mlx::core::distributed']]], - ['group_91',['group',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8831cb61ac633431b78b5fb99c0ea9ff',1,'mlx::core::distributed::DistPrimitive']]], - ['group_92',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html#a6f84accc8d6734989b2757bf6cdd0152',1,'mlx::core::distributed::Group']]], - ['group_5fstep_5fcnt_93',['group_step_cnt',['../struct_quantized_block_loader.html#a234feacde36a4afc0d740332a3769fb6',1,'QuantizedBlockLoader']]], - ['group_5fsteps_94',['group_steps',['../struct_quantized_block_loader.html#a31e14175f3d4902d9fe5ab5a219f61ba',1,'QuantizedBlockLoader']]], - ['group_5fstride_95',['group_stride',['../struct_quantized_block_loader.html#a0ace7e3762ecfa5a4106e7dee7e1b6ab',1,'QuantizedBlockLoader']]], - ['groups_96',['groups',['../struct_m_l_x_conv_params.html#af7a5590ac0974c7841c7f8b9fda0cbed',1,'MLXConvParams']]], - ['gumbel_97',['gumbel',['../namespacemlx_1_1core_1_1random.html#aa849b765cd794306997bcbb9936d3d84',1,'mlx::core::random']]] + ['gemv_5fmasked_25',['gemv_masked',['../namespacemlx_1_1core_1_1metal.html#abc055b75e6a059618f279c35f8de36e7',1,'mlx::core::metal::gemv_masked()'],['../kernels_2gemv__masked_8h.html#ab3070d14cdecb1dd7dc220a551da6b7b',1,'gemv_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant size_t *vector_batch_stride, const constant size_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant size_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid): gemv_masked.h']]], + ['gemv_5fmasked_2eh_26',['gemv_masked.h',['../jit_2gemv__masked_8h.html',1,'(Global Namespace)'],['../kernels_2gemv__masked_8h.html',1,'(Global Namespace)']]], + ['gemv_5fmasked_5fkernel_27',['gemv_masked_kernel',['../jit_2gemv__masked_8h.html#a933f06c211f86c37673dee329ed6901f',1,'gemv_masked.h']]], + ['gemv_5ft_5fmasked_28',['gemv_t_masked',['../kernels_2gemv__masked_8h.html#a0c8d353fc453e448b2d0ed9a19431b63',1,'gemv_masked.h']]], + ['gemvkernel_29',['GEMVKernel',['../struct_g_e_m_v_kernel.html',1,'']]], + ['gemvtkernel_30',['GEMVTKernel',['../struct_g_e_m_v_t_kernel.html',1,'']]], + ['general_31',['General',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a0db377921f4ce762c62526131097968f',1,'mlx::core']]], + ['general_5fc2r_32',['general_c2r',['../namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc',1,'pocketfft::detail']]], + ['general_5fnd_33',['general_nd',['../namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0',1,'pocketfft::detail']]], + ['general_5fr2c_34',['general_r2c',['../namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed',1,'pocketfft::detail']]], + ['generalcontiguousreduce_35',['GeneralContiguousReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a540cf31fe6858115a02e789938297cdb',1,'mlx::core']]], + ['generalgeneral_36',['GeneralGeneral',['../namespacemlx_1_1core.html#abd84ff6c5245e4e170b2ef5247594337a6fe62e8ce1fae1e70cb9eeaa67d29dab',1,'mlx::core']]], + ['generalreduce_37',['GeneralReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65a3622f95ed0ec99657f9ad8ef39ec2184',1,'mlx::core']]], + ['generalstridedreduce_38',['GeneralStridedReduce',['../namespacemlx_1_1core.html#a12412984a1cabfe1189942c898f8fe65ad999b1a8ae1d7436efb5ffdfafb1dd3d',1,'mlx::core']]], + ['generic_39',['generic',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2da3d517f8924ac7fd03699a29d97dc52d9',1,'mlx::core::Dtype::generic'],['../namespacemlx_1_1core.html#a34d69c4d46aa9b2a4a79dba7aba093d2',1,'mlx::core::generic']]], + ['get_5f2d_5fgrid_5fdims_40',['get_2d_grid_dims',['../namespacemlx_1_1core.html#a8dc169474a51a1f4f761d5752819bd7c',1,'mlx::core']]], + ['get_5factive_5fmemory_41',['get_active_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a7a3ad4e33d57a47474c98e2f88e775d7',1,'mlx::core::metal::MetalAllocator::get_active_memory()'],['../namespacemlx_1_1core_1_1metal.html#a7b75c2639016ac4d350fa6c9da386667',1,'mlx::core::metal::get_active_memory()']]], + ['get_5farange_5fkernel_42',['get_arange_kernel',['../namespacemlx_1_1core.html#a76f614e9956a6ca05a9be4db5a483446',1,'mlx::core']]], + ['get_5fbinary_5fkernel_43',['get_binary_kernel',['../namespacemlx_1_1core.html#a4decd4a07d91487e6903f6e3c8b7513a',1,'mlx::core']]], + ['get_5fbinary_5ftwo_5fkernel_44',['get_binary_two_kernel',['../namespacemlx_1_1core.html#a4e809746f48e5dcf7fa63215d3f5e33e',1,'mlx::core']]], + ['get_5fblock_5fdims_45',['get_block_dims',['../namespacemlx_1_1core.html#a7b3f872f6046bdd4d429ee1f1c31cd40',1,'mlx::core']]], + ['get_5fcache_5fmemory_46',['get_cache_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ad3cabbe638917ca4114eb74dcabe381f',1,'mlx::core::metal::MetalAllocator::get_cache_memory()'],['../namespacemlx_1_1core_1_1metal.html#a43307654f62ed7c58e014be7fb03909c',1,'mlx::core::metal::get_cache_memory()']]], + ['get_5fcommand_5fbuffer_47',['get_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a5fe3970fbe92ccc55fce4241ffbe5210',1,'mlx::core::metal::Device']]], + ['get_5fcommand_5fbuffer_5fops_48',['get_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a064e1cb6a16de7a0619f6447622350f8',1,'mlx::core::metal::Device']]], + ['get_5fcommand_5fencoder_49',['get_command_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#affa682ef612def4890f5152f81ffb7e6',1,'mlx::core::metal::Device']]], + ['get_5fcopy_5fkernel_50',['get_copy_kernel',['../namespacemlx_1_1core.html#a05a220cff45f12439fde775983c6df78',1,'mlx::core']]], + ['get_5fdefault_5fstream_51',['get_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a2366c7b888e433608e203752edc92282',1,'mlx::core::scheduler::Scheduler']]], + ['get_5ffft_5fkernel_52',['get_fft_kernel',['../namespacemlx_1_1core.html#a1d4cffc3c78067b3d9a62d64f3fb686f',1,'mlx::core']]], + ['get_5ffunction_53',['get_function',['../classmlx_1_1core_1_1metal_1_1_device.html#abf4ecf035e8be2ff00d2a7e95475d7ac',1,'mlx::core::metal::Device::get_function(const std::string &base_name, MTL::Library *mtl_lib, const std::string &specialized_name="", const MTLFCList &func_consts={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#ac083535ead07161cb27ad31f8f64fb84',1,'mlx::core::metal::Device::get_function(const std::string &base_name, const std::string &lib_name="mlx", const std::string &specialized_name="", const MTLFCList &func_consts={})']]], + ['get_5fgemv_5fmasked_5fkernel_54',['get_gemv_masked_kernel',['../namespacemlx_1_1core.html#a90c24e0d0b99b68fad9deefcf4d3e818',1,'mlx::core']]], + ['get_5fkernel_55',['get_kernel',['../classmlx_1_1core_1_1metal_1_1_device.html#a6810c4dcbcfbf93fc51d42aa5ff0fc3a',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#afa0cac9d800c21a8a7f6cb224256abaf',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, const std::string &lib_name="mlx", const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})']]], + ['get_5fkernel_5fpreamble_56',['get_kernel_preamble',['../compiled__preamble_8h.html#a1dfa17a0369fb90ff615c7461f5013f3',1,'compiled_preamble.h']]], + ['get_5flibrary_57',['get_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a86297d66de43ee77d0697f69c6bd0b87',1,'mlx::core::metal::Device::get_library(const std::string &name)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a961d3cdb4bfb14c1a0b0aeb05609f56e',1,'mlx::core::metal::Device::get_library(const std::string &name, const std::string &source_string, bool cache=true)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a8d8d9e28ffeea10d687b91ea480fb4bd',1,'mlx::core::metal::Device::get_library(const std::string &name, const MTL::StitchedLibraryDescriptor *desc, bool cache=true)']]], + ['get_5fmb_5fsort_5fkernel_58',['get_mb_sort_kernel',['../namespacemlx_1_1core.html#afb57825bb763050cc9a9d194aa41ac36',1,'mlx::core']]], + ['get_5fname_59',['get_name',['../structmlx_1_1core_1_1_node_namer.html#a1690dd38de288c0aee2bb53156eb770e',1,'mlx::core::NodeNamer']]], + ['get_5fpeak_5fmemory_60',['get_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ac7972a3fe58e69489de775a0f152da17',1,'mlx::core::metal::MetalAllocator::get_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#a4b67d680cefa95f0ed5801f0e14e48ce',1,'mlx::core::metal::get_peak_memory()']]], + ['get_5fplan_61',['get_plan',['../namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b',1,'pocketfft::detail']]], + ['get_5fpool_62',['get_pool',['../namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a',1,'pocketfft::detail::threading']]], + ['get_5fprimitive_5fstring_63',['get_primitive_string',['../namespacemlx_1_1core.html#ad4be35b310a252edd80d9cf04f094a60',1,'mlx::core']]], + ['get_5fquantized_5fkernel_64',['get_quantized_kernel',['../namespacemlx_1_1core.html#aa3faeae5378bfaafe3ce3432a051e43e',1,'mlx::core']]], + ['get_5freduce_5finit_5fkernel_65',['get_reduce_init_kernel',['../namespacemlx_1_1core.html#a51c4bb09230348bd0252e22bfdc9bc89',1,'mlx::core']]], + ['get_5freduce_5fkernel_66',['get_reduce_kernel',['../namespacemlx_1_1core.html#a5d6843dcbdbd4d12851fc2edbafbc9e6',1,'mlx::core']]], + ['get_5freduction_5fplan_67',['get_reduction_plan',['../namespacemlx_1_1core.html#a1555dc378c5254e79199421761f26f2b',1,'mlx::core']]], + ['get_5fscan_5fkernel_68',['get_scan_kernel',['../namespacemlx_1_1core.html#aeefaff208444d3fa61ecc0946fe1de5f',1,'mlx::core']]], + ['get_5fshape_69',['get_shape',['../namespacemlx_1_1core.html#aab0d8a256957984acc1e3615c65c898e',1,'mlx::core']]], + ['get_5fsoftmax_5fkernel_70',['get_softmax_kernel',['../namespacemlx_1_1core.html#a35a412f688d79eb47e42d20a7c8650ee',1,'mlx::core']]], + ['get_5fsort_5fkernel_71',['get_sort_kernel',['../namespacemlx_1_1core.html#a84ebe6275218070f0ea320f126f64e22',1,'mlx::core']]], + ['get_5fsteel_5fconv_5fgeneral_5fkernel_72',['get_steel_conv_general_kernel',['../namespacemlx_1_1core.html#abce2b67044ee06a7bbe7a91ec7c8c48d',1,'mlx::core']]], + ['get_5fsteel_5fconv_5fkernel_73',['get_steel_conv_kernel',['../namespacemlx_1_1core.html#adce79d220672f5f3c65cc31d145ca9c4',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5ffused_5fkernel_74',['get_steel_gemm_fused_kernel',['../namespacemlx_1_1core.html#a84fa8e0aee321a9d614433a0b933103b',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fmasked_5fkernel_75',['get_steel_gemm_masked_kernel',['../namespacemlx_1_1core.html#ab5f60614e965144b451930fdf935e08d',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fsplitk_5faccum_5fkernel_76',['get_steel_gemm_splitk_accum_kernel',['../namespacemlx_1_1core.html#a195b86cad5bb99aa1bcd23952305af6b',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fsplitk_5fkernel_77',['get_steel_gemm_splitk_kernel',['../namespacemlx_1_1core.html#af48c6f2f72b61dbd6766e4f5fea85df5',1,'mlx::core']]], + ['get_5ftemplate_5fdefinition_78',['get_template_definition',['../namespacemlx_1_1core.html#aae0d19f0acdef2accd2428fb84c8a032',1,'mlx::core']]], + ['get_5fternary_5fkernel_79',['get_ternary_kernel',['../namespacemlx_1_1core.html#a54eb3b65375022428aab5f810e40624b',1,'mlx::core']]], + ['get_5ftwiddle_80',['get_twiddle',['../radix_8h.html#ac5cf950316b9445296ee9ecfc56a56bd',1,'radix.h']]], + ['get_5ftype_5fstring_81',['get_type_string',['../namespacemlx_1_1core.html#af776fd91dd60594dcfebbafd17f19068',1,'mlx::core']]], + ['get_5funary_5fkernel_82',['get_unary_kernel',['../namespacemlx_1_1core.html#a15175e8e2b1e26726c63393e4d68b628',1,'mlx::core']]], + ['gguf_2eh_83',['gguf.h',['../gguf_8h.html',1,'']]], + ['gguf_5fload_5fquantized_84',['gguf_load_quantized',['../namespacemlx_1_1core.html#a65dd68163bdaef3631e3724327782498',1,'mlx::core']]], + ['ggufload_85',['GGUFLoad',['../namespacemlx_1_1core.html#aa5b0f7f13a941e1f41c411194e9033c7',1,'mlx::core']]], + ['ggufmetadata_86',['GGUFMetaData',['../namespacemlx_1_1core.html#a8c2c1b9a37aadfb48f4c3a7e806e32e3',1,'mlx::core']]], + ['global_5fformatter_87',['global_formatter',['../namespacemlx_1_1core.html#af5a408a78cc934717dd711ddfda58ea6',1,'mlx::core']]], + ['good_88',['good',['../classmlx_1_1core_1_1io_1_1_reader.html#a005d0b52c1f34866f7412b7f41dabec3',1,'mlx::core::io::Reader::good()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a0b050c2c27487007e250e2e19560ffe4',1,'mlx::core::io::Writer::good()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a4ecb046dfa7dab346d096adab3a1c318',1,'mlx::core::io::FileReader::good()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9ec4934b26fb358d699ddce1482b2d54',1,'mlx::core::io::FileWriter::good()']]], + ['good_5fsize_5fcmplx_89',['good_size_cmplx',['../structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624',1,'pocketfft::detail::util']]], + ['good_5fsize_5freal_90',['good_size_real',['../structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa',1,'pocketfft::detail::util']]], + ['gpu_91',['gpu',['../structmlx_1_1core_1_1_device.html#a45ed081b56ae5d4ddd39c83a5d8a1616',1,'mlx::core::Device::gpu'],['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdba0aa0be2a866411d9ff03515227454947',1,'mlx::core::Device::gpu']]], + ['grad_92',['grad',['../namespacemlx_1_1core.html#a99994806f81e83f98746c9e5ba4c3673',1,'mlx::core::grad'],['../namespacemlx_1_1core.html#af482f6c64acd77c57ef5bb4b7be9726c',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a64bc619876b0f8cc81a2637ca81c99f7',1,'mlx::core::grad(const std::function< array(const array &)> &fun)']]], + ['graph_5futils_2eh_93',['graph_utils.h',['../graph__utils_8h.html',1,'']]], + ['greater_94',['Greater',['../struct_greater.html',1,'Greater'],['../structmlx_1_1core_1_1detail_1_1_greater.html',1,'mlx::core::detail::Greater'],['../classmlx_1_1core_1_1_greater.html',1,'mlx::core::Greater'],['../classmlx_1_1core_1_1_greater.html#a1d5992a66c020cd97a70e8e3d8cd1a1b',1,'mlx::core::Greater::Greater()']]], + ['greater_95',['greater',['../group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d',1,'mlx::core']]], + ['greater_5fequal_96',['greater_equal',['../group__ops.html#ga7153071bcfff6faad21332163fb9a430',1,'mlx::core']]], + ['greaterequal_97',['GreaterEqual',['../struct_greater_equal.html',1,'GreaterEqual'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html',1,'mlx::core::detail::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html',1,'mlx::core::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html#a19a3c49d5a9b40e17da0e56ef6908527',1,'mlx::core::GreaterEqual::GreaterEqual()']]], + ['grid_98',['grid',['../struct_read_writer.html#ac7a957f99873d3797081f5d620f3b2c8',1,'ReadWriter']]], + ['group_99',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html',1,'mlx::core::distributed::Group'],['../structmlx_1_1core_1_1distributed_1_1_group.html#a6f84accc8d6734989b2757bf6cdd0152',1,'mlx::core::distributed::Group::Group()']]], + ['group_100',['group',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8831cb61ac633431b78b5fb99c0ea9ff',1,'mlx::core::distributed::DistPrimitive']]], + ['group_5fstep_5fcnt_101',['group_step_cnt',['../struct_quantized_block_loader.html#a234feacde36a4afc0d740332a3769fb6',1,'QuantizedBlockLoader']]], + ['group_5fsteps_102',['group_steps',['../struct_quantized_block_loader.html#a31e14175f3d4902d9fe5ab5a219f61ba',1,'QuantizedBlockLoader']]], + ['group_5fstride_103',['group_stride',['../struct_quantized_block_loader.html#a0ace7e3762ecfa5a4106e7dee7e1b6ab',1,'QuantizedBlockLoader']]], + ['groups_104',['groups',['../struct_m_l_x_conv_params.html#af7a5590ac0974c7841c7f8b9fda0cbed',1,'MLXConvParams']]], + ['gumbel_105',['gumbel',['../namespacemlx_1_1core_1_1random.html#aa849b765cd794306997bcbb9936d3d84',1,'mlx::core::random']]] ]; diff --git a/docs/build/html/search/all_8.js b/docs/build/html/search/all_8.js index c84f934537..385c5b72f7 100644 --- a/docs/build/html/search/all_8.js +++ b/docs/build/html/search/all_8.js @@ -19,5 +19,9 @@ var searchData= ['half_5finplace_5fop_16',['half_inplace_op',['../fp16_8h.html#a6348c00d31a50b2df1b47d18af49c4b8',1,'fp16.h']]], ['half_5ftypes_2eh_17',['half_types.h',['../half__types_8h.html',1,'']]], ['has_5fbatch_18',['has_batch',['../steel__gemm__fused_8h.html#adffcdc900c19ff97f1523e43f1a5a6cc',1,'steel_gemm_fused.h']]], - ['has_5fprimitive_19',['has_primitive',['../classmlx_1_1core_1_1array.html#aa5aceab15241e7826cbaf8b8a41440c1',1,'mlx::core::array']]] + ['has_5fmul_5foperand_5fmask_19',['has_mul_operand_mask',['../struct_g_e_m_v_kernel.html#ad47223ee49b3cb7bf3746a2cec45f883',1,'GEMVKernel::has_mul_operand_mask'],['../struct_g_e_m_v_t_kernel.html#a8db6f01f96a36b216acd801c34a96ef5',1,'GEMVTKernel::has_mul_operand_mask']]], + ['has_5fmul_5foutput_5fmask_20',['has_mul_output_mask',['../struct_g_e_m_v_kernel.html#a0edbf2dd6a6563e7afa6dab6b670615c',1,'GEMVKernel::has_mul_output_mask'],['../struct_g_e_m_v_t_kernel.html#a8eb06f6569e4042e24fee220b11fa10d',1,'GEMVTKernel::has_mul_output_mask']]], + ['has_5foperand_5fmask_21',['has_operand_mask',['../struct_g_e_m_v_kernel.html#ab00784dff1512a7b0919fcb4cfa5d50e',1,'GEMVKernel::has_operand_mask'],['../struct_g_e_m_v_t_kernel.html#a6729d6e63e76a1e9c7c8e78d9aac4869',1,'GEMVTKernel::has_operand_mask']]], + ['has_5foutput_5fmask_22',['has_output_mask',['../struct_g_e_m_v_kernel.html#ab8b64c94f4c8f6f09c0777415589b487',1,'GEMVKernel::has_output_mask'],['../struct_g_e_m_v_t_kernel.html#aaefdf8f023da255bbb70a0c3e3408626',1,'GEMVTKernel::has_output_mask']]], + ['has_5fprimitive_23',['has_primitive',['../classmlx_1_1core_1_1array.html#aa5aceab15241e7826cbaf8b8a41440c1',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/all_9.js b/docs/build/html/search/all_9.js index 376863fcff..4ebf856c78 100644 --- a/docs/build/html/search/all_9.js +++ b/docs/build/html/search/all_9.js @@ -22,7 +22,7 @@ var searchData= ['indices_19',['Indices',['../struct_indices.html',1,'']]], ['inexact_20',['inexact',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dae03b116564cd944b048fde87dbd4d5c9',1,'mlx::core::Dtype::inexact'],['../namespacemlx_1_1core.html#a54c6fae21b7f2fea8e6f80011ef38534',1,'mlx::core::inexact']]], ['infinity_21',['infinity',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61eb741e7af49046beb863abf023b206',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['init_22',['init',['../struct_and.html#ab55eac58830ae01c45f18377eff021ea',1,'And::init'],['../struct_or.html#a6734c64e8048a209047fa18dd8bd49e8',1,'Or::init'],['../struct_sum.html#a433bf0fe3ce751c18e5269bbc19daa65',1,'Sum::init'],['../struct_prod.html#ac99f3882b6c3c0c17d70a82f2f2f2c56',1,'Prod::init'],['../struct_min.html#a322a8241f9e9f60ffa040e68c02af97c',1,'Min::init'],['../struct_max.html#aa985471d570c1009e5cbb174efcb9131',1,'Max::init'],['../struct_cum_sum.html#a987f7a15eded60081b74abc103d3054b',1,'CumSum::init'],['../struct_cum_prod.html#acf2ae2daa84d477b413deea679cab9e5',1,'CumProd::init'],['../struct_cum_prod_3_01bool_01_4.html#ae7a8b0ba9e6898356b87b18766e76d2c',1,'CumProd< bool >::init'],['../struct_cum_max.html#a16480052a2eeb4340e546838aab59cc4',1,'CumMax::init'],['../struct_cum_min.html#a8b67f739c620d0cc194b533190990ab9',1,'CumMin::init'],['../struct_less_than.html#abf97a6b0163048e4ba96460939dbd3a3',1,'LessThan::init'],['../class_m_p_s_1_1_matrix.html#a2398221610a7c610c85cf7fb3f18281b',1,'MPS::Matrix::init(MTL::Buffer *buffer, MatrixDescriptor *descriptor)'],['../class_m_p_s_1_1_matrix.html#a40771ae44d119abd56bd32b7cc38feda',1,'MPS::Matrix::init(const MTL::Buffer *buffer, MatrixDescriptor *descriptor)'],['../class_m_p_s_1_1_matrix_multiplication.html#a35b1a908807eb8ececd0241531dea823',1,'MPS::MatrixMultiplication::init()'],['../class_m_p_s_1_1_vector.html#a2ea0c0575489e4689d19640d1da1b179',1,'MPS::Vector::init(MTL::Buffer *buffer, VectorDescriptor *descriptor)'],['../class_m_p_s_1_1_vector.html#aa95af1607e9978c2757f8a5503a95244',1,'MPS::Vector::init(const MTL::Buffer *buffer, VectorDescriptor *descriptor)'],['../class_m_p_s_1_1_matrix_vector_multiplication.html#a62c065490befe32c6a59b888b1347c80',1,'MPS::MatrixVectorMultiplication::init()'],['../namespacemlx_1_1core_1_1distributed.html#a33633c058c7ec82cca4f237243c6810d',1,'mlx::core::distributed::init()']]], + ['init_22',['init',['../struct_and.html#ab55eac58830ae01c45f18377eff021ea',1,'And::init'],['../struct_or.html#a6734c64e8048a209047fa18dd8bd49e8',1,'Or::init'],['../struct_sum.html#a433bf0fe3ce751c18e5269bbc19daa65',1,'Sum::init'],['../struct_prod.html#ac99f3882b6c3c0c17d70a82f2f2f2c56',1,'Prod::init'],['../struct_min.html#a322a8241f9e9f60ffa040e68c02af97c',1,'Min::init'],['../struct_max.html#aa985471d570c1009e5cbb174efcb9131',1,'Max::init'],['../struct_cum_sum.html#a987f7a15eded60081b74abc103d3054b',1,'CumSum::init'],['../struct_cum_prod.html#acf2ae2daa84d477b413deea679cab9e5',1,'CumProd::init'],['../struct_cum_prod_3_01bool_01_4.html#ae7a8b0ba9e6898356b87b18766e76d2c',1,'CumProd< bool >::init'],['../struct_cum_max.html#a16480052a2eeb4340e546838aab59cc4',1,'CumMax::init'],['../struct_cum_min.html#a8b67f739c620d0cc194b533190990ab9',1,'CumMin::init'],['../struct_less_than.html#abf97a6b0163048e4ba96460939dbd3a3',1,'LessThan::init'],['../namespacemlx_1_1core_1_1distributed.html#a33633c058c7ec82cca4f237243c6810d',1,'mlx::core::distributed::init()']]], ['init_5freduce_23',['init_reduce',['../reduce__init_8h.html#a0088604ac2eaa6940689ff12c4ba5fc2',1,'reduce_init.h']]], ['inner_24',['inner',['../group__ops.html#ga654fec16a9746b390916697a2ab2546e',1,'mlx::core']]], ['inp_5fjump_5fc_25',['inp_jump_c',['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a78d30e843d65d1829623afb0b607f0a5',1,'mlx::steel::ImplicitGemmConv2DParams']]], @@ -45,7 +45,7 @@ var searchData= ['inv_42',['inv',['../struct_read_writer.html#a773fa8524515bdc2ff8b0e2060741813',1,'ReadWriter::inv'],['../namespacemlx_1_1core_1_1linalg.html#ad966a0b6bff176c9f933534ed62389a2',1,'mlx::core::linalg::inv()']]], ['inv_5f_43',['inv_',['../backend_2metal_2kernels_2fft_8h.html#a23a773ae455da5d0b9b5508ee2e7694b',1,'fft.h']]], ['inv_5falpha_44',['INV_ALPHA',['../struct_m_l_x_scaled_dot_product_attention_params.html#a7461e0e17cdc7d3fed80bb00d58d8644',1,'MLXScaledDotProductAttentionParams']]], - ['inverse_45',['Inverse',['../classmlx_1_1core_1_1_inverse.html',1,'mlx::core::Inverse'],['../classmlx_1_1core_1_1_inverse.html#a4fcdda66a176b520fe4655ab91862aad',1,'mlx::core::Inverse::Inverse()']]], + ['inverse_45',['Inverse',['../classmlx_1_1core_1_1_inverse.html',1,'mlx::core::Inverse'],['../classmlx_1_1core_1_1_inverse.html#a71467681e523abb725724490bfeb76ad',1,'mlx::core::Inverse::Inverse()']]], ['io_2eh_46',['io.h',['../io_8h.html',1,'']]], ['iofs_47',['iofs',['../classpocketfft_1_1detail_1_1multi__iter.html#ad33360d4a8ab8e6d72efadc6f9cb5bfa',1,'pocketfft::detail::multi_iter::iofs(size_t i) const'],['../classpocketfft_1_1detail_1_1multi__iter.html#a97462d97bdca6419d8d2f37c2031fe83',1,'pocketfft::detail::multi_iter::iofs(size_t j, size_t i) const']]], ['irfft_48',['irfft',['../namespacemlx_1_1core_1_1fft.html#aafa721d0492e9f74913a6e86b4896ad8',1,'mlx::core::fft::irfft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#afbd0035a3cf91f428838de1fcf01a3a3',1,'mlx::core::fft::irfft(const array &a, int axis=-1, StreamOrDevice s={})']]], @@ -55,25 +55,24 @@ var searchData= ['is_5farray_5fv_52',['is_array_v',['../namespacemlx_1_1core.html#a01b0d64a75dfa2e95d6c7b5c53d708af',1,'mlx::core']]], ['is_5farrays_5fv_53',['is_arrays_v',['../namespacemlx_1_1core.html#a94c1057929b390e5613304afa16dfbda',1,'mlx::core']]], ['is_5favailable_54',['is_available',['../classmlx_1_1core_1_1array.html#aebed1f37c19197be76105161102a8a40',1,'mlx::core::array::is_available()'],['../namespacemlx_1_1core_1_1metal.html#a0cdf2c08c7bc0927a86070adc206987f',1,'mlx::core::metal::is_available()'],['../namespacemlx_1_1core_1_1distributed.html#a95655473cd0032c06e5fe3fca85aeef3',1,'mlx::core::distributed::is_available()']]], - ['is_5fbig_5fendian_55',['is_big_endian',['../namespacemlx_1_1core.html#a625ed440df3fa57318017c1f2c589efe',1,'mlx::core']]], - ['is_5fdonatable_56',['is_donatable',['../classmlx_1_1core_1_1array.html#a4677a404b5d191af20b52649225de087',1,'mlx::core::array']]], - ['is_5fequivalent_57',['is_equivalent',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#af08b1294f3f93505a96fdfa85b1edd62',1,'mlx::core::fast::ScaledDotProductAttention::is_equivalent()'],['../classmlx_1_1core_1_1_primitive.html#a6140a502af4c2bbbc776ab26e9afebcd',1,'mlx::core::Primitive::is_equivalent()'],['../classmlx_1_1core_1_1_abs.html#ab6f0ec56bc7c048382297e12dabadc67',1,'mlx::core::Abs::is_equivalent()'],['../classmlx_1_1core_1_1_add.html#aba0a35410c3aac53d0f7a0c283d9ee3f',1,'mlx::core::Add::is_equivalent()'],['../classmlx_1_1core_1_1_add_m_m.html#a6e37c6882dba995a63fb6d8dfb01754f',1,'mlx::core::AddMM::is_equivalent()'],['../classmlx_1_1core_1_1_arange.html#a7b6a45cf9c4b109d4e0373f3fe576c35',1,'mlx::core::Arange::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cos.html#a39557461e3235801886675a9b7d25bf5',1,'mlx::core::ArcCos::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6928e827b9ac2e86e7d5b02b78150eee',1,'mlx::core::ArcCosh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sin.html#a13b5e39eeccaf32d94b8eb85b3b753ab',1,'mlx::core::ArcSin::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sinh.html#a63c7a765c7906242dc3371deec094f0f',1,'mlx::core::ArcSinh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan.html#a0e5b5fc7218143ecd0a8666d9137c34c',1,'mlx::core::ArcTan::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan2.html#aeaee58cd803d3ebf0b76574a409682cc',1,'mlx::core::ArcTan2::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tanh.html#ac8ecdd640043dab0461d49d7650679a2',1,'mlx::core::ArcTanh::is_equivalent()'],['../classmlx_1_1core_1_1_arg_partition.html#ad87509ce70b51fb75dfb9c3a05a5b31a',1,'mlx::core::ArgPartition::is_equivalent()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03b81a670dcb1e39bf7279e4d4583b97',1,'mlx::core::ArgReduce::is_equivalent()'],['../classmlx_1_1core_1_1_arg_sort.html#a048cd09c557d29d1111726f97010a845',1,'mlx::core::ArgSort::is_equivalent()'],['../classmlx_1_1core_1_1_as_type.html#a8e6c8b2428ab15c4fb43f2e3a8fb38af',1,'mlx::core::AsType::is_equivalent()'],['../classmlx_1_1core_1_1_as_strided.html#a1738c6aa0a3a3eb68530f0d5b436e094',1,'mlx::core::AsStrided::is_equivalent()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8cd6b916b4838a6c329cf4df8530c3b8',1,'mlx::core::BitwiseBinary::is_equivalent()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aef1c303955f9b8f445296372cf181160',1,'mlx::core::BlockMaskedMM::is_equivalent()'],['../classmlx_1_1core_1_1_gather_m_m.html#a163f17f6ce2c002f22e81b302777342b',1,'mlx::core::GatherMM::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast.html#a0e27692b0090ec451954649a36042616',1,'mlx::core::Broadcast::is_equivalent()'],['../classmlx_1_1core_1_1_ceil.html#aacd90acb56eb0649c1cef807aa21df52',1,'mlx::core::Ceil::is_equivalent()'],['../classmlx_1_1core_1_1_compiled.html#a63e5016458887813b4a59dee5a0a3f10',1,'mlx::core::Compiled::is_equivalent()'],['../classmlx_1_1core_1_1_concatenate.html#aaf8a72a0c30114460caf519580cc35d2',1,'mlx::core::Concatenate::is_equivalent()'],['../classmlx_1_1core_1_1_conjugate.html#af42f00a790c6bc5572bd8fe9e5b36c5e',1,'mlx::core::Conjugate::is_equivalent()'],['../classmlx_1_1core_1_1_convolution.html#afb87708a5e3aab2e9e663daa9d8863de',1,'mlx::core::Convolution::is_equivalent()'],['../classmlx_1_1core_1_1_copy.html#afcfa39465015f638e294aa954ea0f3da',1,'mlx::core::Copy::is_equivalent()'],['../classmlx_1_1core_1_1_cos.html#ab611ca38c987915659f7ffcce0370417',1,'mlx::core::Cos::is_equivalent()'],['../classmlx_1_1core_1_1_cosh.html#ae0bacccaf501f5349db0c13cca776ff9',1,'mlx::core::Cosh::is_equivalent()'],['../classmlx_1_1core_1_1_divide.html#a3dda091f05c4164c29bb8129e9712650',1,'mlx::core::Divide::is_equivalent()'],['../classmlx_1_1core_1_1_div_mod.html#af5fcf8ec8515d46844cbeeab6dafb38a',1,'mlx::core::DivMod::is_equivalent()'],['../classmlx_1_1core_1_1_select.html#afc3c333fac7f902c98839921ef2874c8',1,'mlx::core::Select::is_equivalent()'],['../classmlx_1_1core_1_1_remainder.html#a802039faaa2ed7b763ec3d7debcce814',1,'mlx::core::Remainder::is_equivalent()'],['../classmlx_1_1core_1_1_equal.html#a58c1c5003e43f47dc0788c1851deaa02',1,'mlx::core::Equal::is_equivalent()'],['../classmlx_1_1core_1_1_erf.html#abe99dfbc2954c3a7d5dec56ab165ee82',1,'mlx::core::Erf::is_equivalent()'],['../classmlx_1_1core_1_1_erf_inv.html#aaac9e3b454ba564f9c6e804ab6562832',1,'mlx::core::ErfInv::is_equivalent()'],['../classmlx_1_1core_1_1_exp.html#ac6e44bffe7a643ab4ca51e74c7328357',1,'mlx::core::Exp::is_equivalent()'],['../classmlx_1_1core_1_1_f_f_t.html#a0ede3bc8b6d77d560c0a750b68fddc06',1,'mlx::core::FFT::is_equivalent()'],['../classmlx_1_1core_1_1_floor.html#a24b64feb026c4fcd02fc481cffdb1c94',1,'mlx::core::Floor::is_equivalent()'],['../classmlx_1_1core_1_1_full.html#afafcbcae1e28597fe8f7fde289105792',1,'mlx::core::Full::is_equivalent()'],['../classmlx_1_1core_1_1_gather.html#a23ff1406dbf0c770e75ad47440b467aa',1,'mlx::core::Gather::is_equivalent()'],['../classmlx_1_1core_1_1_greater.html#a6877a6888614a618dc64296763ccabb1',1,'mlx::core::Greater::is_equivalent()'],['../classmlx_1_1core_1_1_greater_equal.html#a3daef8596b963026b602019bc56fc5fc',1,'mlx::core::GreaterEqual::is_equivalent()'],['../classmlx_1_1core_1_1_hadamard.html#a8a528d8d69a7343bdfd704a3e74230b8',1,'mlx::core::Hadamard::is_equivalent()'],['../classmlx_1_1core_1_1_less.html#a7d6ed6353a0dcefebd008026dbd3cd63',1,'mlx::core::Less::is_equivalent()'],['../classmlx_1_1core_1_1_less_equal.html#a76ee1438cf4bd109eae4e0b3472b26af',1,'mlx::core::LessEqual::is_equivalent()'],['../classmlx_1_1core_1_1_log.html#a2fc58ea4ca744db493b947d1136d05f8',1,'mlx::core::Log::is_equivalent()'],['../classmlx_1_1core_1_1_logical_not.html#aba53675da351cd9b71a73d475b4bbe99',1,'mlx::core::LogicalNot::is_equivalent()'],['../classmlx_1_1core_1_1_logical_and.html#a9572c35f72e0db2f7f86bbf42438a6be',1,'mlx::core::LogicalAnd::is_equivalent()'],['../classmlx_1_1core_1_1_logical_or.html#a9c8b10a5cf5c69fdc2362390197e4e71',1,'mlx::core::LogicalOr::is_equivalent()'],['../classmlx_1_1core_1_1_log_add_exp.html#a3cf9a202c05aff39919d713d6e2b32e4',1,'mlx::core::LogAddExp::is_equivalent()'],['../classmlx_1_1core_1_1_matmul.html#aab372b59eae0840fc4f75ef5719a2630',1,'mlx::core::Matmul::is_equivalent()'],['../classmlx_1_1core_1_1_maximum.html#a21fe93fbd7799682f481260aee8bdb46',1,'mlx::core::Maximum::is_equivalent()'],['../classmlx_1_1core_1_1_minimum.html#a56c54ee3293cc2cd84462b9ec7ac36b4',1,'mlx::core::Minimum::is_equivalent()'],['../classmlx_1_1core_1_1_multiply.html#ae288159fa2d6d35087a85aca8eafa9b2',1,'mlx::core::Multiply::is_equivalent()'],['../classmlx_1_1core_1_1_negative.html#ac2a4d8159c548639d6289980c8975823',1,'mlx::core::Negative::is_equivalent()'],['../classmlx_1_1core_1_1_not_equal.html#ac12fd6b3e2f2e7e4e622b59badf2c73d',1,'mlx::core::NotEqual::is_equivalent()'],['../classmlx_1_1core_1_1_number_of_elements.html#ad6a32565ccc64499e368e15bba0b438f',1,'mlx::core::NumberOfElements::is_equivalent()'],['../classmlx_1_1core_1_1_pad.html#aad7c3bfecafe435d6a8e807de4c7ea9b',1,'mlx::core::Pad::is_equivalent()'],['../classmlx_1_1core_1_1_partition.html#aabdf6ef4f2159b2bfe93e0e87d4772f8',1,'mlx::core::Partition::is_equivalent()'],['../classmlx_1_1core_1_1_power.html#a76b4ec9d1ff07f06189e414480453d68',1,'mlx::core::Power::is_equivalent()'],['../classmlx_1_1core_1_1_quantized_matmul.html#af28b36e3f40ea41785387800326cc8e1',1,'mlx::core::QuantizedMatmul::is_equivalent()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a6a7da6bcf657fcdb157c45bf35fdec11',1,'mlx::core::GatherQMM::is_equivalent()'],['../classmlx_1_1core_1_1_random_bits.html#a72ec915debf5823e7c0463045b2894e6',1,'mlx::core::RandomBits::is_equivalent()'],['../classmlx_1_1core_1_1_reshape.html#abd07c53af476777a04307e0423784cf3',1,'mlx::core::Reshape::is_equivalent()'],['../classmlx_1_1core_1_1_reduce.html#abe8f3327d617d0dd7438f066497ae08e',1,'mlx::core::Reduce::is_equivalent()'],['../classmlx_1_1core_1_1_round.html#aeb3d8607bbba7345a3142d4cbd4e6927',1,'mlx::core::Round::is_equivalent()'],['../classmlx_1_1core_1_1_scan.html#a54445a4d677ca4fe2a58d08eb5223ac6',1,'mlx::core::Scan::is_equivalent()'],['../classmlx_1_1core_1_1_scatter.html#a0208172562abdc90472e6eb5f84c987f',1,'mlx::core::Scatter::is_equivalent()'],['../classmlx_1_1core_1_1_sigmoid.html#a04814ba1b0edf8299d5ca1bcb8749d8e',1,'mlx::core::Sigmoid::is_equivalent()'],['../classmlx_1_1core_1_1_sign.html#a8c0934acbcc4b146e5aacd35a8c445bb',1,'mlx::core::Sign::is_equivalent()'],['../classmlx_1_1core_1_1_sin.html#af00b0e5516f884996ce7a97e6c1e3e6a',1,'mlx::core::Sin::is_equivalent()'],['../classmlx_1_1core_1_1_sinh.html#adcb1878996fd4902cd550042dd6ad70d',1,'mlx::core::Sinh::is_equivalent()'],['../classmlx_1_1core_1_1_slice.html#a43202c3b8966ae1db9ab82072e4918b0',1,'mlx::core::Slice::is_equivalent()'],['../classmlx_1_1core_1_1_slice_update.html#a60f588acced42391e6e5615ae8d16119',1,'mlx::core::SliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_softmax.html#a9215ed7bd36bc11276c58dfb9808d728',1,'mlx::core::Softmax::is_equivalent()'],['../classmlx_1_1core_1_1_sort.html#ae48f07cf641d54234fc4fb6529a33511',1,'mlx::core::Sort::is_equivalent()'],['../classmlx_1_1core_1_1_split.html#af25a0cc259573b9dce60d285eee18345',1,'mlx::core::Split::is_equivalent()'],['../classmlx_1_1core_1_1_square.html#a6abc881d44071019aa15481e5ea75ab2',1,'mlx::core::Square::is_equivalent()'],['../classmlx_1_1core_1_1_sqrt.html#ab871c2b8ab4a27a3f782a005d0e87c46',1,'mlx::core::Sqrt::is_equivalent()'],['../classmlx_1_1core_1_1_stop_gradient.html#a327539298b21d800d26482b94fce41b3',1,'mlx::core::StopGradient::is_equivalent()'],['../classmlx_1_1core_1_1_subtract.html#af1c05e1e3f703ba916d54f8ccbbd102b',1,'mlx::core::Subtract::is_equivalent()'],['../classmlx_1_1core_1_1_tan.html#afdf46288e7f60ea7f878688347dff7e4',1,'mlx::core::Tan::is_equivalent()'],['../classmlx_1_1core_1_1_tanh.html#a0692a1de2373b86eb394252ed4fecfda',1,'mlx::core::Tanh::is_equivalent()'],['../classmlx_1_1core_1_1_uniform.html#abb6048807a7c5b2e35a77e06a17f801b',1,'mlx::core::Uniform::is_equivalent()'],['../classmlx_1_1core_1_1_view.html#a7cb8403a96a47cb258caac4e3b850f64',1,'mlx::core::View::is_equivalent()'],['../classmlx_1_1core_1_1_transpose.html#a799ec3c3fa9f1b9e6177c755252a3eab',1,'mlx::core::Transpose::is_equivalent()']]], - ['is_5fmetal_5fatomic_58',['is_metal_atomic',['../atomic_8h.html#a91a8bdcae647947a83c6689d7f252d24',1,'atomic.h']]], - ['is_5fopen_59',['is_open',['../classmlx_1_1core_1_1io_1_1_reader.html#a780f504058bd9c80cb3d105046a9f985',1,'mlx::core::io::Reader::is_open()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a85aa36bdb0dbfb8c5b6cfd955b03417a',1,'mlx::core::io::Writer::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a97ccdbd79886d9b95b32e5cd4741d150',1,'mlx::core::io::FileReader::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#ad5d2ee671a81700cb1658c41309d6676',1,'mlx::core::io::FileWriter::is_open()']]], - ['is_5fpower_5fof_5f2_60',['is_power_of_2',['../namespacemlx_1_1core.html#adacbc4526e8964b267a8ec3eb1bc1a32',1,'mlx::core']]], - ['is_5fpower_5fof_5f2_5f_61',['is_power_of_2_',['../backend_2metal_2kernels_2fft_8h.html#abeb0f8d030ee27c89882b87eed13be2d',1,'fft.h']]], - ['is_5fready_62',['is_ready',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3',1,'pocketfft::detail::threading::latch']]], - ['is_5fsame_5fshape_63',['is_same_shape',['../namespacemlx_1_1core.html#ad4b664de4a4abd305827b30879b9da33',1,'mlx::core']]], - ['is_5fscalar_64',['is_scalar',['../namespacemlx_1_1core.html#a985c60929757190e0b4ec51f57c767d0',1,'mlx::core']]], - ['is_5fstatic_5fcast_65',['is_static_cast',['../namespacemlx_1_1core.html#afd9e740e567f9d7c28e00113caf46d5f',1,'mlx::core']]], - ['is_5ftracer_66',['is_tracer',['../classmlx_1_1core_1_1array.html#af9acb115019b995354d366c4ac6b968c',1,'mlx::core::array']]], - ['isclose_67',['isclose',['../group__ops.html#ga51eac95c04400921c54716de14b52491',1,'mlx::core']]], - ['isinf_68',['isinf',['../group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844',1,'mlx::core']]], - ['isnan_69',['isnan',['../namespacemetal.html#a83320ba983d90dd1fa5847b6940dc0bb',1,'metal::isnan()'],['../group__ops.html#ga175592792471b0ffb45196dca4711ba6',1,'mlx::core::isnan(const array &a, StreamOrDevice s={})']]], - ['isneginf_70',['isneginf',['../group__ops.html#ga1940523da381ed7be50656a3bc465ff3',1,'mlx::core']]], - ['isposinf_71',['isposinf',['../group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993',1,'mlx::core']]], - ['issubdtype_72',['issubdtype',['../namespacemlx_1_1core.html#aca9e69b06f4212eba44bf0ce6711d5f7',1,'mlx::core::issubdtype(const Dtype &a, const Dtype &b)'],['../namespacemlx_1_1core.html#aca39f224c1d17bde35dfcb9088430704',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype &b)'],['../namespacemlx_1_1core.html#ae9ee4a7c205df061c1caa7e62b7504e8',1,'mlx::core::issubdtype(const Dtype &a, const Dtype::Category &b)'],['../namespacemlx_1_1core.html#ab5b1a5a3d545a5de00c3117f76d71a1d',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype::Category &b)']]], - ['item_73',['item',['../classmlx_1_1core_1_1array.html#a90c5afddc2fa3028c0f8099bd64c8a99',1,'mlx::core::array::item()'],['../classmlx_1_1core_1_1array.html#a8650a99a6b7549bc823b03ad92590ff7',1,'mlx::core::array::item() const']]], - ['itemsize_74',['itemsize',['../classmlx_1_1core_1_1array.html#af329d9432c92de87cbaa2de8454eefc0',1,'mlx::core::array']]], - ['iterator_5fcategory_75',['iterator_category',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a2cbf481e39164245668b3be6cbcc614d',1,'mlx::core::array::ArrayIterator']]] + ['is_5fdonatable_55',['is_donatable',['../classmlx_1_1core_1_1array.html#a4677a404b5d191af20b52649225de087',1,'mlx::core::array']]], + ['is_5fequivalent_56',['is_equivalent',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#af08b1294f3f93505a96fdfa85b1edd62',1,'mlx::core::fast::ScaledDotProductAttention::is_equivalent()'],['../classmlx_1_1core_1_1_primitive.html#a6140a502af4c2bbbc776ab26e9afebcd',1,'mlx::core::Primitive::is_equivalent()'],['../classmlx_1_1core_1_1_abs.html#ab6f0ec56bc7c048382297e12dabadc67',1,'mlx::core::Abs::is_equivalent()'],['../classmlx_1_1core_1_1_add.html#aba0a35410c3aac53d0f7a0c283d9ee3f',1,'mlx::core::Add::is_equivalent()'],['../classmlx_1_1core_1_1_add_m_m.html#a6e37c6882dba995a63fb6d8dfb01754f',1,'mlx::core::AddMM::is_equivalent()'],['../classmlx_1_1core_1_1_arange.html#a7b6a45cf9c4b109d4e0373f3fe576c35',1,'mlx::core::Arange::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cos.html#a39557461e3235801886675a9b7d25bf5',1,'mlx::core::ArcCos::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6928e827b9ac2e86e7d5b02b78150eee',1,'mlx::core::ArcCosh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sin.html#a13b5e39eeccaf32d94b8eb85b3b753ab',1,'mlx::core::ArcSin::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sinh.html#a63c7a765c7906242dc3371deec094f0f',1,'mlx::core::ArcSinh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan.html#a0e5b5fc7218143ecd0a8666d9137c34c',1,'mlx::core::ArcTan::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan2.html#aeaee58cd803d3ebf0b76574a409682cc',1,'mlx::core::ArcTan2::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tanh.html#ac8ecdd640043dab0461d49d7650679a2',1,'mlx::core::ArcTanh::is_equivalent()'],['../classmlx_1_1core_1_1_arg_partition.html#ad87509ce70b51fb75dfb9c3a05a5b31a',1,'mlx::core::ArgPartition::is_equivalent()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03b81a670dcb1e39bf7279e4d4583b97',1,'mlx::core::ArgReduce::is_equivalent()'],['../classmlx_1_1core_1_1_arg_sort.html#a048cd09c557d29d1111726f97010a845',1,'mlx::core::ArgSort::is_equivalent()'],['../classmlx_1_1core_1_1_as_type.html#a8e6c8b2428ab15c4fb43f2e3a8fb38af',1,'mlx::core::AsType::is_equivalent()'],['../classmlx_1_1core_1_1_as_strided.html#a1738c6aa0a3a3eb68530f0d5b436e094',1,'mlx::core::AsStrided::is_equivalent()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8cd6b916b4838a6c329cf4df8530c3b8',1,'mlx::core::BitwiseBinary::is_equivalent()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aef1c303955f9b8f445296372cf181160',1,'mlx::core::BlockMaskedMM::is_equivalent()'],['../classmlx_1_1core_1_1_gather_m_m.html#a163f17f6ce2c002f22e81b302777342b',1,'mlx::core::GatherMM::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast.html#a0e27692b0090ec451954649a36042616',1,'mlx::core::Broadcast::is_equivalent()'],['../classmlx_1_1core_1_1_ceil.html#aacd90acb56eb0649c1cef807aa21df52',1,'mlx::core::Ceil::is_equivalent()'],['../classmlx_1_1core_1_1_compiled.html#a63e5016458887813b4a59dee5a0a3f10',1,'mlx::core::Compiled::is_equivalent()'],['../classmlx_1_1core_1_1_concatenate.html#aaf8a72a0c30114460caf519580cc35d2',1,'mlx::core::Concatenate::is_equivalent()'],['../classmlx_1_1core_1_1_conjugate.html#af42f00a790c6bc5572bd8fe9e5b36c5e',1,'mlx::core::Conjugate::is_equivalent()'],['../classmlx_1_1core_1_1_convolution.html#afb87708a5e3aab2e9e663daa9d8863de',1,'mlx::core::Convolution::is_equivalent()'],['../classmlx_1_1core_1_1_copy.html#afcfa39465015f638e294aa954ea0f3da',1,'mlx::core::Copy::is_equivalent()'],['../classmlx_1_1core_1_1_cos.html#ab611ca38c987915659f7ffcce0370417',1,'mlx::core::Cos::is_equivalent()'],['../classmlx_1_1core_1_1_cosh.html#ae0bacccaf501f5349db0c13cca776ff9',1,'mlx::core::Cosh::is_equivalent()'],['../classmlx_1_1core_1_1_divide.html#a3dda091f05c4164c29bb8129e9712650',1,'mlx::core::Divide::is_equivalent()'],['../classmlx_1_1core_1_1_div_mod.html#af5fcf8ec8515d46844cbeeab6dafb38a',1,'mlx::core::DivMod::is_equivalent()'],['../classmlx_1_1core_1_1_select.html#afc3c333fac7f902c98839921ef2874c8',1,'mlx::core::Select::is_equivalent()'],['../classmlx_1_1core_1_1_remainder.html#a802039faaa2ed7b763ec3d7debcce814',1,'mlx::core::Remainder::is_equivalent()'],['../classmlx_1_1core_1_1_equal.html#a58c1c5003e43f47dc0788c1851deaa02',1,'mlx::core::Equal::is_equivalent()'],['../classmlx_1_1core_1_1_erf.html#abe99dfbc2954c3a7d5dec56ab165ee82',1,'mlx::core::Erf::is_equivalent()'],['../classmlx_1_1core_1_1_erf_inv.html#aaac9e3b454ba564f9c6e804ab6562832',1,'mlx::core::ErfInv::is_equivalent()'],['../classmlx_1_1core_1_1_exp.html#ac6e44bffe7a643ab4ca51e74c7328357',1,'mlx::core::Exp::is_equivalent()'],['../classmlx_1_1core_1_1_f_f_t.html#a0ede3bc8b6d77d560c0a750b68fddc06',1,'mlx::core::FFT::is_equivalent()'],['../classmlx_1_1core_1_1_floor.html#a24b64feb026c4fcd02fc481cffdb1c94',1,'mlx::core::Floor::is_equivalent()'],['../classmlx_1_1core_1_1_full.html#afafcbcae1e28597fe8f7fde289105792',1,'mlx::core::Full::is_equivalent()'],['../classmlx_1_1core_1_1_gather.html#a23ff1406dbf0c770e75ad47440b467aa',1,'mlx::core::Gather::is_equivalent()'],['../classmlx_1_1core_1_1_greater.html#a6877a6888614a618dc64296763ccabb1',1,'mlx::core::Greater::is_equivalent()'],['../classmlx_1_1core_1_1_greater_equal.html#a3daef8596b963026b602019bc56fc5fc',1,'mlx::core::GreaterEqual::is_equivalent()'],['../classmlx_1_1core_1_1_hadamard.html#a8a528d8d69a7343bdfd704a3e74230b8',1,'mlx::core::Hadamard::is_equivalent()'],['../classmlx_1_1core_1_1_less.html#a7d6ed6353a0dcefebd008026dbd3cd63',1,'mlx::core::Less::is_equivalent()'],['../classmlx_1_1core_1_1_less_equal.html#a76ee1438cf4bd109eae4e0b3472b26af',1,'mlx::core::LessEqual::is_equivalent()'],['../classmlx_1_1core_1_1_log.html#a2fc58ea4ca744db493b947d1136d05f8',1,'mlx::core::Log::is_equivalent()'],['../classmlx_1_1core_1_1_logical_not.html#aba53675da351cd9b71a73d475b4bbe99',1,'mlx::core::LogicalNot::is_equivalent()'],['../classmlx_1_1core_1_1_logical_and.html#a9572c35f72e0db2f7f86bbf42438a6be',1,'mlx::core::LogicalAnd::is_equivalent()'],['../classmlx_1_1core_1_1_logical_or.html#a9c8b10a5cf5c69fdc2362390197e4e71',1,'mlx::core::LogicalOr::is_equivalent()'],['../classmlx_1_1core_1_1_log_add_exp.html#a3cf9a202c05aff39919d713d6e2b32e4',1,'mlx::core::LogAddExp::is_equivalent()'],['../classmlx_1_1core_1_1_matmul.html#aab372b59eae0840fc4f75ef5719a2630',1,'mlx::core::Matmul::is_equivalent()'],['../classmlx_1_1core_1_1_maximum.html#a21fe93fbd7799682f481260aee8bdb46',1,'mlx::core::Maximum::is_equivalent()'],['../classmlx_1_1core_1_1_minimum.html#a56c54ee3293cc2cd84462b9ec7ac36b4',1,'mlx::core::Minimum::is_equivalent()'],['../classmlx_1_1core_1_1_multiply.html#ae288159fa2d6d35087a85aca8eafa9b2',1,'mlx::core::Multiply::is_equivalent()'],['../classmlx_1_1core_1_1_negative.html#ac2a4d8159c548639d6289980c8975823',1,'mlx::core::Negative::is_equivalent()'],['../classmlx_1_1core_1_1_not_equal.html#ac12fd6b3e2f2e7e4e622b59badf2c73d',1,'mlx::core::NotEqual::is_equivalent()'],['../classmlx_1_1core_1_1_number_of_elements.html#ad6a32565ccc64499e368e15bba0b438f',1,'mlx::core::NumberOfElements::is_equivalent()'],['../classmlx_1_1core_1_1_pad.html#aad7c3bfecafe435d6a8e807de4c7ea9b',1,'mlx::core::Pad::is_equivalent()'],['../classmlx_1_1core_1_1_partition.html#aabdf6ef4f2159b2bfe93e0e87d4772f8',1,'mlx::core::Partition::is_equivalent()'],['../classmlx_1_1core_1_1_power.html#a76b4ec9d1ff07f06189e414480453d68',1,'mlx::core::Power::is_equivalent()'],['../classmlx_1_1core_1_1_quantized_matmul.html#af28b36e3f40ea41785387800326cc8e1',1,'mlx::core::QuantizedMatmul::is_equivalent()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a6a7da6bcf657fcdb157c45bf35fdec11',1,'mlx::core::GatherQMM::is_equivalent()'],['../classmlx_1_1core_1_1_random_bits.html#a72ec915debf5823e7c0463045b2894e6',1,'mlx::core::RandomBits::is_equivalent()'],['../classmlx_1_1core_1_1_reshape.html#abd07c53af476777a04307e0423784cf3',1,'mlx::core::Reshape::is_equivalent()'],['../classmlx_1_1core_1_1_reduce.html#abe8f3327d617d0dd7438f066497ae08e',1,'mlx::core::Reduce::is_equivalent()'],['../classmlx_1_1core_1_1_round.html#aeb3d8607bbba7345a3142d4cbd4e6927',1,'mlx::core::Round::is_equivalent()'],['../classmlx_1_1core_1_1_scan.html#a54445a4d677ca4fe2a58d08eb5223ac6',1,'mlx::core::Scan::is_equivalent()'],['../classmlx_1_1core_1_1_scatter.html#a0208172562abdc90472e6eb5f84c987f',1,'mlx::core::Scatter::is_equivalent()'],['../classmlx_1_1core_1_1_sigmoid.html#a04814ba1b0edf8299d5ca1bcb8749d8e',1,'mlx::core::Sigmoid::is_equivalent()'],['../classmlx_1_1core_1_1_sign.html#a8c0934acbcc4b146e5aacd35a8c445bb',1,'mlx::core::Sign::is_equivalent()'],['../classmlx_1_1core_1_1_sin.html#af00b0e5516f884996ce7a97e6c1e3e6a',1,'mlx::core::Sin::is_equivalent()'],['../classmlx_1_1core_1_1_sinh.html#adcb1878996fd4902cd550042dd6ad70d',1,'mlx::core::Sinh::is_equivalent()'],['../classmlx_1_1core_1_1_slice.html#a43202c3b8966ae1db9ab82072e4918b0',1,'mlx::core::Slice::is_equivalent()'],['../classmlx_1_1core_1_1_slice_update.html#a60f588acced42391e6e5615ae8d16119',1,'mlx::core::SliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_softmax.html#a9215ed7bd36bc11276c58dfb9808d728',1,'mlx::core::Softmax::is_equivalent()'],['../classmlx_1_1core_1_1_sort.html#ae48f07cf641d54234fc4fb6529a33511',1,'mlx::core::Sort::is_equivalent()'],['../classmlx_1_1core_1_1_split.html#af25a0cc259573b9dce60d285eee18345',1,'mlx::core::Split::is_equivalent()'],['../classmlx_1_1core_1_1_square.html#a6abc881d44071019aa15481e5ea75ab2',1,'mlx::core::Square::is_equivalent()'],['../classmlx_1_1core_1_1_sqrt.html#ab871c2b8ab4a27a3f782a005d0e87c46',1,'mlx::core::Sqrt::is_equivalent()'],['../classmlx_1_1core_1_1_stop_gradient.html#a327539298b21d800d26482b94fce41b3',1,'mlx::core::StopGradient::is_equivalent()'],['../classmlx_1_1core_1_1_subtract.html#af1c05e1e3f703ba916d54f8ccbbd102b',1,'mlx::core::Subtract::is_equivalent()'],['../classmlx_1_1core_1_1_tan.html#afdf46288e7f60ea7f878688347dff7e4',1,'mlx::core::Tan::is_equivalent()'],['../classmlx_1_1core_1_1_tanh.html#a0692a1de2373b86eb394252ed4fecfda',1,'mlx::core::Tanh::is_equivalent()'],['../classmlx_1_1core_1_1_uniform.html#abb6048807a7c5b2e35a77e06a17f801b',1,'mlx::core::Uniform::is_equivalent()'],['../classmlx_1_1core_1_1_view.html#a7cb8403a96a47cb258caac4e3b850f64',1,'mlx::core::View::is_equivalent()'],['../classmlx_1_1core_1_1_transpose.html#a799ec3c3fa9f1b9e6177c755252a3eab',1,'mlx::core::Transpose::is_equivalent()']]], + ['is_5fmetal_5fatomic_57',['is_metal_atomic',['../atomic_8h.html#a91a8bdcae647947a83c6689d7f252d24',1,'atomic.h']]], + ['is_5fopen_58',['is_open',['../classmlx_1_1core_1_1io_1_1_reader.html#a780f504058bd9c80cb3d105046a9f985',1,'mlx::core::io::Reader::is_open()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a85aa36bdb0dbfb8c5b6cfd955b03417a',1,'mlx::core::io::Writer::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a97ccdbd79886d9b95b32e5cd4741d150',1,'mlx::core::io::FileReader::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#ad5d2ee671a81700cb1658c41309d6676',1,'mlx::core::io::FileWriter::is_open()']]], + ['is_5fpower_5fof_5f2_59',['is_power_of_2',['../namespacemlx_1_1core.html#adacbc4526e8964b267a8ec3eb1bc1a32',1,'mlx::core']]], + ['is_5fpower_5fof_5f2_5f_60',['is_power_of_2_',['../backend_2metal_2kernels_2fft_8h.html#abeb0f8d030ee27c89882b87eed13be2d',1,'fft.h']]], + ['is_5fready_61',['is_ready',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3',1,'pocketfft::detail::threading::latch']]], + ['is_5fsame_5fshape_62',['is_same_shape',['../namespacemlx_1_1core.html#ad4b664de4a4abd305827b30879b9da33',1,'mlx::core']]], + ['is_5fscalar_63',['is_scalar',['../namespacemlx_1_1core.html#a985c60929757190e0b4ec51f57c767d0',1,'mlx::core']]], + ['is_5fstatic_5fcast_64',['is_static_cast',['../namespacemlx_1_1core.html#afd9e740e567f9d7c28e00113caf46d5f',1,'mlx::core']]], + ['is_5ftracer_65',['is_tracer',['../classmlx_1_1core_1_1array.html#af9acb115019b995354d366c4ac6b968c',1,'mlx::core::array']]], + ['isclose_66',['isclose',['../group__ops.html#ga51eac95c04400921c54716de14b52491',1,'mlx::core']]], + ['isinf_67',['isinf',['../group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844',1,'mlx::core']]], + ['isnan_68',['isnan',['../namespacemetal.html#a83320ba983d90dd1fa5847b6940dc0bb',1,'metal::isnan()'],['../group__ops.html#ga175592792471b0ffb45196dca4711ba6',1,'mlx::core::isnan(const array &a, StreamOrDevice s={})']]], + ['isneginf_69',['isneginf',['../group__ops.html#ga1940523da381ed7be50656a3bc465ff3',1,'mlx::core']]], + ['isposinf_70',['isposinf',['../group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993',1,'mlx::core']]], + ['issubdtype_71',['issubdtype',['../namespacemlx_1_1core.html#aca9e69b06f4212eba44bf0ce6711d5f7',1,'mlx::core::issubdtype(const Dtype &a, const Dtype &b)'],['../namespacemlx_1_1core.html#aca39f224c1d17bde35dfcb9088430704',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype &b)'],['../namespacemlx_1_1core.html#ae9ee4a7c205df061c1caa7e62b7504e8',1,'mlx::core::issubdtype(const Dtype &a, const Dtype::Category &b)'],['../namespacemlx_1_1core.html#ab5b1a5a3d545a5de00c3117f76d71a1d',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype::Category &b)']]], + ['item_72',['item',['../classmlx_1_1core_1_1array.html#a90c5afddc2fa3028c0f8099bd64c8a99',1,'mlx::core::array::item()'],['../classmlx_1_1core_1_1array.html#a8650a99a6b7549bc823b03ad92590ff7',1,'mlx::core::array::item() const']]], + ['itemsize_73',['itemsize',['../classmlx_1_1core_1_1array.html#af329d9432c92de87cbaa2de8454eefc0',1,'mlx::core::array']]], + ['iterator_5fcategory_74',['iterator_category',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#a2cbf481e39164245668b3be6cbcc614d',1,'mlx::core::array::ArrayIterator']]] ]; diff --git a/docs/build/html/search/all_b.js b/docs/build/html/search/all_b.js index db56319dc1..2d991d4530 100644 --- a/docs/build/html/search/all_b.js +++ b/docs/build/html/search/all_b.js @@ -2,13 +2,12 @@ var searchData= [ ['k_0',['K',['../struct_m_l_x_fast_attention_params.html#ada454f5ad22ec36a22d0ff596751af23',1,'MLXFastAttentionParams::K'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#ae1b0386e4cd1a7018f4b654c4e9493ba',1,'mlx::steel::ImplicitGemmConv2DParams::K'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#aa0851af4da8df820bdad9589ff517cff',1,'mlx::steel::GEMMParams::K'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a3733f9031e82e761ec44e72ed5c6d0e7',1,'mlx::steel::GEMMSpiltKParams::K']]], ['kdil_1',['kdil',['../struct_m_l_x_conv_params.html#a7611db8f1621c7e09fc685ed44073b14',1,'MLXConvParams']]], - ['kernel_2',['Kernel',['../class_m_p_s_1_1_kernel.html',1,'MPS']]], - ['kernelmergesort_3',['KernelMergeSort',['../struct_kernel_merge_sort.html',1,'']]], - ['kernelmultiblockmergesort_4',['KernelMultiBlockMergeSort',['../struct_kernel_multi_block_merge_sort.html',1,'']]], - ['kernels_2eh_5',['kernels.h',['../kernels_8h.html',1,'']]], - ['key_6',['key',['../namespacemlx_1_1core_1_1random.html#acf04b6f42de11383e86dcc7f98c67bd8',1,'mlx::core::random']]], - ['keysequence_7',['KeySequence',['../classmlx_1_1core_1_1random_1_1_key_sequence.html',1,'mlx::core::random::KeySequence'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a196eb6ce5ba1eb37cc8c67d6d1332bfe',1,'mlx::core::random::KeySequence::KeySequence()']]], - ['kind_8',['Kind',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715',1,'mlx::core::Dtype']]], - ['kindof_9',['kindof',['../namespacemlx_1_1core.html#ad527b86818823db040195785efd7d724',1,'mlx::core']]], - ['kv_5ftiles_10',['KV_TILES',['../struct_m_l_x_scaled_dot_product_attention_params.html#a58ef2765fd681e6b35b2ba72030610e0',1,'MLXScaledDotProductAttentionParams']]] + ['kernelmergesort_2',['KernelMergeSort',['../struct_kernel_merge_sort.html',1,'']]], + ['kernelmultiblockmergesort_3',['KernelMultiBlockMergeSort',['../struct_kernel_multi_block_merge_sort.html',1,'']]], + ['kernels_2eh_4',['kernels.h',['../kernels_8h.html',1,'']]], + ['key_5',['key',['../namespacemlx_1_1core_1_1random.html#acf04b6f42de11383e86dcc7f98c67bd8',1,'mlx::core::random']]], + ['keysequence_6',['KeySequence',['../classmlx_1_1core_1_1random_1_1_key_sequence.html',1,'mlx::core::random::KeySequence'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a196eb6ce5ba1eb37cc8c67d6d1332bfe',1,'mlx::core::random::KeySequence::KeySequence()']]], + ['kind_7',['Kind',['../structmlx_1_1core_1_1_dtype.html#adb1ea8b45a0c53e04a0e73b168702715',1,'mlx::core::Dtype']]], + ['kindof_8',['kindof',['../namespacemlx_1_1core.html#ad527b86818823db040195785efd7d724',1,'mlx::core']]], + ['kv_5ftiles_9',['KV_TILES',['../struct_m_l_x_scaled_dot_product_attention_params.html#a58ef2765fd681e6b35b2ba72030610e0',1,'MLXScaledDotProductAttentionParams']]] ]; diff --git a/docs/build/html/search/all_c.js b/docs/build/html/search/all_c.js index a468531887..7e4c48d56b 100644 --- a/docs/build/html/search/all_c.js +++ b/docs/build/html/search/all_c.js @@ -1,6 +1,6 @@ var searchData= [ - ['label_0',['label',['../class_m_p_s_1_1_kernel.html#aaf646bce46266bc11c419464ec72cdfe',1,'MPS::Kernel::label()'],['../classmlx_1_1core_1_1io_1_1_reader.html#a8c244daf52fd5ebb9a2c7e5f4ae793cd',1,'mlx::core::io::Reader::label()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a828125a9adcb7e90c8bcaba0fe47f854',1,'mlx::core::io::Writer::label()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a7415da4cb14c328ed07901ff1c266221',1,'mlx::core::io::FileReader::label()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a5093dce80ff0c51ea036a87e3e5fb456',1,'mlx::core::io::FileWriter::label()']]], + ['label_0',['label',['../classmlx_1_1core_1_1io_1_1_reader.html#a8c244daf52fd5ebb9a2c7e5f4ae793cd',1,'mlx::core::io::Reader::label()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a828125a9adcb7e90c8bcaba0fe47f854',1,'mlx::core::io::Writer::label()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a7415da4cb14c328ed07901ff1c266221',1,'mlx::core::io::FileReader::label()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a5093dce80ff0c51ea036a87e3e5fb456',1,'mlx::core::io::FileWriter::label()']]], ['lapack_5fhelper_2eh_1',['lapack_helper.h',['../lapack__helper_8h.html',1,'']]], ['laplace_2',['laplace',['../namespacemlx_1_1core_1_1random.html#ad52c7cb7709542c5a464df12bfc6b0b7',1,'mlx::core::random::laplace(const std::vector< int > &shape, Dtype dtype, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a0d0c393dd53d6fdd02e46844b3cf45d1',1,'mlx::core::random::laplace(const std::vector< int > &shape, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a0f73eb7d337529708937eef0080c0998',1,'mlx::core::random::laplace(const std::vector< int > &shape, const Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a96b6f40efbee7045a35218296ba12a37',1,'mlx::core::random::laplace(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], ['largest_5fprime_5ffactor_3',['largest_prime_factor',['../structpocketfft_1_1detail_1_1util.html#a0e9b92280f25063d1ebef942dd0d233c',1,'pocketfft::detail::util']]], @@ -51,10 +51,10 @@ var searchData= ['load_2eh_48',['load.h',['../load_8h.html',1,'']]], ['load_5fgguf_49',['load_gguf',['../namespacemlx_1_1core.html#a2aa12b351ce559deb14cda0a5292c2ce',1,'mlx::core']]], ['load_5fpadded_50',['load_padded',['../struct_read_writer.html#add5bd3f647793a5a19d63197a19df73c',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#af3ce6bbb1a8dfb3bab1ae18d3eb45bc0',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#ab116f4569bb9dc6eaef0d8d08472e239',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const']]], - ['load_5fsafe_51',['load_safe',['../struct_quantized_block_loader.html#a699dc9aa284b8fbf870310bbb224465b',1,'QuantizedBlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../kernels_2scan_8h.html#ae8eb101e538b85f8a4bcf451489ae0ac',1,'load_safe(): scan.h']]], + ['load_5fsafe_51',['load_safe',['../struct_g_e_m_v_kernel.html#a04bb72da9a93d6d1eba468fa311bbba7',1,'GEMVKernel::load_safe()'],['../struct_quantized_block_loader.html#a699dc9aa284b8fbf870310bbb224465b',1,'QuantizedBlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../kernels_2scan_8h.html#ae8eb101e538b85f8a4bcf451489ae0ac',1,'load_safe(): scan.h']]], ['load_5fsafetensors_52',['load_safetensors',['../namespacemlx_1_1core.html#a96cc40e1af8c4626c813ce4859f70a5c',1,'mlx::core::load_safetensors(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#af7eea1682a38d363c56a066321e6d526',1,'mlx::core::load_safetensors(const std::string &file, StreamOrDevice s={})']]], ['load_5fstrided_53',['load_strided',['../struct_read_writer.html#a998ef484bade81f726b9edfc6b878197',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a3d9c8cbc582cad6b5218339d0f721559',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a795a71a8e1f154a5af415ebe1b3f0713',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a0935b946b8bf2e769427fcbf2da2f7be',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a7d45368c74a8b7c632659504b3273a13',1,'ReadWriter::load_strided(int stride, int overall_n)']]], - ['load_5funsafe_54',['load_unsafe',['../struct_quantized_block_loader.html#a86009527cb4b53e4c21fd6b1f78cfefc',1,'QuantizedBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a961836be363409744e48e595d5e0c2ec',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8034abc10483487fc94313e3674d1111',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a69e2f7c9814d1cc1c5c267be8618dc55',1,'mlx::steel::Conv2DWeightBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#aa11d1a142bc868df462f48a7102147f3',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a0e262b003ac0e7ee6272585eac921704',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3859ca11b5991ef6ee9b99afdc3ea30a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8f078982186421f5b484c0b53af9c655',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader.html#a6c9e27f11f48b34580ed2c7e9cad9a27',1,'mlx::steel::BlockLoader::load_unsafe()'],['../kernels_2scan_8h.html#a9c415d07921f3961bad0a00a34f4a9a3',1,'load_unsafe(U values[N_READS], const device T *input): scan.h']]], + ['load_5funsafe_54',['load_unsafe',['../struct_g_e_m_v_kernel.html#a6013e9c5b2f72fa1311dd038172df0ce',1,'GEMVKernel::load_unsafe()'],['../struct_quantized_block_loader.html#a86009527cb4b53e4c21fd6b1f78cfefc',1,'QuantizedBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a961836be363409744e48e595d5e0c2ec',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8034abc10483487fc94313e3674d1111',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a69e2f7c9814d1cc1c5c267be8618dc55',1,'mlx::steel::Conv2DWeightBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#aa11d1a142bc868df462f48a7102147f3',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a0e262b003ac0e7ee6272585eac921704',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3859ca11b5991ef6ee9b99afdc3ea30a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8f078982186421f5b484c0b53af9c655',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader.html#a6c9e27f11f48b34580ed2c7e9cad9a27',1,'mlx::steel::BlockLoader::load_unsafe()'],['../kernels_2scan_8h.html#a9c415d07921f3961bad0a00a34f4a9a3',1,'load_unsafe(U values[N_READS], const device T *input): scan.h']]], ['load_5fvector_55',['load_vector',['../quantized_8h.html#a8dbace41de9e1e21dd59d016db11b3e9',1,'quantized.h']]], ['load_5fvector_5fsafe_56',['load_vector_safe',['../quantized_8h.html#aa69e143d646fad332c1a53e8c9b337b7',1,'quantized.h']]], ['loader_2eh_57',['loader.h',['../conv_2loader_8h.html',1,'(Global Namespace)'],['../gemm_2loader_8h.html',1,'(Global Namespace)']]], diff --git a/docs/build/html/search/all_d.js b/docs/build/html/search/all_d.js index dbbfa2f238..428815967c 100644 --- a/docs/build/html/search/all_d.js +++ b/docs/build/html/search/all_d.js @@ -3,111 +3,100 @@ var searchData= ['m_0',['M',['../struct_m_l_x_fast_attention_params.html#a5cd3ede5f41d5fdf8177cab3f059f4d8',1,'MLXFastAttentionParams::M'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a2117fc93662d5177c8f3e7c2dbb9e2db',1,'mlx::steel::ImplicitGemmConv2DParams::M'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a85b20a4c4558cc78d76fcbd045a9c694',1,'mlx::steel::GEMMParams::M'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a8bab0cf8a20d2abefe294a7505917e7e',1,'mlx::steel::GEMMSpiltKParams::M']]], ['make_5farrays_1',['make_arrays',['../classmlx_1_1core_1_1array.html#a1173db4e23f5a8230911cb8fba45d5e6',1,'mlx::core::array']]], ['make_5fcontiguous_5fstrides_2',['make_contiguous_strides',['../namespacemlx_1_1core.html#ac9fb1286a1a00395e901dbff80560895',1,'mlx::core']]], - ['make_5fsynchronize_5ftask_3',['make_synchronize_task',['../namespacemlx_1_1core_1_1metal.html#ab31abdda3052162d59f6590a89e38337',1,'mlx::core::metal']]], - ['make_5ftask_4',['make_task',['../namespacemlx_1_1core_1_1metal.html#a4552b7ccdfa7f3cc9895c09799d8048e',1,'mlx::core::metal']]], - ['malloc_5',['malloc',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a9a17d2c7a97772bf4a15e6c74af34ca4',1,'mlx::core::allocator::Allocator::malloc()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#a4f3d5de6b8c0eba22e9403b28a5ef3f0',1,'mlx::core::allocator::CommonAllocator::malloc()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a6c0feb9b1ff9977f76c69745393944bc',1,'mlx::core::metal::MetalAllocator::malloc()'],['../namespacemlx_1_1core_1_1allocator.html#a560d10a166e3c294f3757166f9bd6801',1,'mlx::core::allocator::malloc(size_t size)']]], - ['malloc_5for_5fwait_6',['malloc_or_wait',['../namespacemlx_1_1core_1_1allocator.html#a86ac0a11ff78f21e717f641716c34abc',1,'mlx::core::allocator']]], - ['mask_5fh_7',['mask_h',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0b892c1a7edb9ed20c076d8945855c19',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], - ['mask_5ft_8',['mask_t',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a270ab3da7c98a12525a59952742cc97d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], - ['mask_5fw_9',['mask_w',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a19ddba7259c3c2c02ed90f3f635557be',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], - ['matmul_10',['Matmul',['../classmlx_1_1core_1_1_matmul.html',1,'mlx::core::Matmul'],['../classmlx_1_1core_1_1_matmul.html#adef92f30ab35e540ccb316ea6b94e6f7',1,'mlx::core::Matmul::Matmul()']]], - ['matmul_11',['matmul',['../group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a',1,'mlx::core']]], - ['matmul_2eh_12',['matmul.h',['../matmul_8h.html',1,'']]], - ['matrix_13',['Matrix',['../class_m_p_s_1_1_matrix.html',1,'MPS']]], - ['matrixdescriptor_14',['MatrixDescriptor',['../class_m_p_s_1_1_matrix_descriptor.html',1,'MPS']]], - ['matrixdescriptor_15',['matrixDescriptor',['../class_m_p_s_1_1_matrix_descriptor.html#a7a99042b3c8afa037719ca316500a37b',1,'MPS::MatrixDescriptor::matrixDescriptor(NS::UInteger rows, NS::UInteger columns, NS::UInteger rowBytes, NS::UInteger dataType)'],['../class_m_p_s_1_1_matrix_descriptor.html#a67e5df7d48fc175efd5124b3b43b6366',1,'MPS::MatrixDescriptor::matrixDescriptor(NS::UInteger rows, NS::UInteger columns, NS::UInteger matrices, NS::UInteger rowBytes, NS::UInteger matrixBytes, NS::UInteger dataType)']]], - ['matrixmultiplication_16',['MatrixMultiplication',['../class_m_p_s_1_1_matrix_multiplication.html',1,'MPS']]], - ['matrixvectormultiplication_17',['MatrixVectorMultiplication',['../class_m_p_s_1_1_matrix_vector_multiplication.html',1,'MPS']]], - ['max_18',['Max',['../struct_max.html',1,'']]], - ['max_19',['max',['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits::max'],['../struct_limits_3_01uint8__t_01_4.html#a1570fb640e2e41f96776db5ca08d500c',1,'Limits< uint8_t >::max'],['../struct_limits_3_01uint16__t_01_4.html#a228b33556ba4cb7e6137ab6258628488',1,'Limits< uint16_t >::max'],['../struct_limits_3_01uint32__t_01_4.html#a91fa8f7214ec936976a8324c7431c651',1,'Limits< uint32_t >::max'],['../struct_limits_3_01uint64__t_01_4.html#aa8c2257881a4e1fa8596fa07dba5e107',1,'Limits< uint64_t >::max'],['../struct_limits_3_01int8__t_01_4.html#a96fed01fa9249226be69760652643289',1,'Limits< int8_t >::max'],['../struct_limits_3_01int16__t_01_4.html#a12d64c398ca7609b7c906f3cf1a6f678',1,'Limits< int16_t >::max'],['../struct_limits_3_01int32__t_01_4.html#af756344b31e84222dd73d3445dcd5640',1,'Limits< int32_t >::max'],['../struct_limits_3_01int64__t_01_4.html#ac9c420604c0f3d237ddfb2b8a2439224',1,'Limits< int64_t >::max'],['../struct_limits_3_01half_01_4.html#a4f9515dbf2a622074f121bea39a7b175',1,'Limits< half >::max'],['../struct_limits_3_01float_01_4.html#aba172b22b388190aa3969ef16885d8a6',1,'Limits< float >::max'],['../struct_limits_3_01bfloat16__t_01_4.html#a0ead3618da6718629ea9fa4670b5005f',1,'Limits< bfloat16_t >::max'],['../struct_limits_3_01bool_01_4.html#acbd2132145888d51220558a101ffcff4',1,'Limits< bool >::max'],['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a92320d40a58218e40cc414986ac95c50',1,'metal::_numeric_limits_impl< bfloat16_t >::max()']]], - ['max_20',['Max',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924ac00cf69bbba24f7ab08d3ad618705988',1,'mlx::core::distributed::AllReduce::Max'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a3d11c500ea4f7f639e20dd0755d39260',1,'mlx::core::Reduce::Max'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ad54b2905015a390708f79bae6cdac56d',1,'mlx::core::Scan::Max'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca1c2da7b96d743296fe660f5fc4072f16',1,'mlx::core::Scatter::Max']]], - ['max_21',['max',['../namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b',1,'metal::max()'],['../namespacemetal_1_1fast.html#a747e2e58092a27fb8b4dd3d16934fb52',1,'metal::fast::max()'],['../namespacemetal_1_1precise.html#a6a954a4e4e3753303d1dc734855a185f',1,'metal::precise::max()'],['../group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7',1,'mlx::core::max(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc',1,'mlx::core::max(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f',1,'mlx::core::max(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga7b638050e03a93f2896c981bc2850a47',1,'mlx::core::max(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['max3_22',['max3',['../namespacemetal.html#a00f9c0ad66d969794614f56912eed9c9',1,'metal::max3()'],['../namespacemetal_1_1fast.html#a6fc2cf18ffa8149561864c86dba0f803',1,'metal::fast::max3()'],['../namespacemetal_1_1precise.html#ac490e8614ebd2c9343af1ae6c0d4e82c',1,'metal::precise::max3()']]], - ['max_5fdigits10_23',['max_digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a8d3905e6f158379a0c52682266e8d0e2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['max_5fexponent_24',['max_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61bb136f819fa392c50bdf3c38f3aad2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['max_5fexponent10_25',['max_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a76bfb2deb0e0afc011f77bf5a6d0ed94',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['max_5foutput_5fsize_26',['MAX_OUTPUT_SIZE',['../backend_2metal_2kernels_2fft_8h.html#a28d683cf067736d76f867f30c066317e',1,'fft.h']]], - ['max_5fradix_27',['MAX_RADIX',['../backend_2metal_2kernels_2fft_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: fft.h'],['../readwrite_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: readwrite.h']]], - ['max_5freduce_5fspecialized_5fdims_28',['MAX_REDUCE_SPECIALIZED_DIMS',['../defines_8h.html#a15629f1b81a2b6f1cca26d07a2734623',1,'defines.h']]], - ['max_5fthreads_29',['max_threads',['../namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70',1,'pocketfft::detail::threading']]], - ['maximum_30',['Maximum',['../struct_maximum.html',1,'Maximum'],['../structmlx_1_1core_1_1detail_1_1_maximum.html',1,'mlx::core::detail::Maximum'],['../classmlx_1_1core_1_1_maximum.html',1,'mlx::core::Maximum'],['../classmlx_1_1core_1_1_maximum.html#a28389307e385efe1b2955b86b115e816',1,'mlx::core::Maximum::Maximum()']]], - ['maximum_31',['maximum',['../group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d',1,'mlx::core']]], - ['mb_5fblock_5fmerge_32',['mb_block_merge',['../sort_8h.html#ab381cd57f344bc7304ab580bfdc78807',1,'sort.h']]], - ['mb_5fblock_5fpartition_33',['mb_block_partition',['../sort_8h.html#a32cbe4163b8b0f5cb2c97b256119a4b2',1,'sort.h']]], - ['mb_5fblock_5fsort_34',['mb_block_sort',['../sort_8h.html#a2a0533103661dd378d6bfe949930650a',1,'sort.h']]], - ['mean_35',['mean',['../group__ops.html#gade46e768fd46b8b640eb16f26abeecef',1,'mlx::core::mean(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga52b59fdd8e8430538e564f5bbcfa31e6',1,'mlx::core::mean(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga066161f3d3e395a1d76c638cb680d444',1,'mlx::core::mean(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga45fba73eab0e3b6e128ed3ce2f43a5da',1,'mlx::core::mean(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['median3_36',['median3',['../namespacemetal.html#aa3ff49457ce3c93fc1c0897fd1525157',1,'metal::median3()'],['../namespacemetal_1_1fast.html#a742b55f1e4369921ee7f60d70185bfbc',1,'metal::fast::median3()'],['../namespacemetal_1_1precise.html#a14555ff99c4388493fec48e070144ae2',1,'metal::precise::median3()']]], - ['merge_5fpartition_37',['merge_partition',['../struct_block_merge_sort.html#ab2300cbecb23f3433bad888924c831ca',1,'BlockMergeSort::merge_partition()'],['../struct_kernel_multi_block_merge_sort.html#ab15895b4233aba0e279cc44a07a201fe',1,'KernelMultiBlockMergeSort::merge_partition()']]], - ['merge_5fstep_38',['merge_step',['../struct_block_merge_sort.html#ab65f190edf1851b37c39ad49ce99a43c',1,'BlockMergeSort']]], - ['meshgrid_39',['meshgrid',['../group__ops.html#ga577c911618575314de63d1060656a26e',1,'mlx::core']]], - ['metal_40',['metal',['../namespacemetal.html',1,'']]], - ['metal_2eh_41',['metal.h',['../metal_8h.html',1,'']]], - ['metal_3a_3afast_42',['fast',['../namespacemetal_1_1fast.html',1,'metal']]], - ['metal_3a_3aprecise_43',['precise',['../namespacemetal_1_1precise.html',1,'metal']]], - ['metal_5fimpl_2eh_44',['metal_impl.h',['../metal__impl_8h.html',1,'']]], - ['metalallocator_45',['MetalAllocator',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html',1,'mlx::core::metal']]], - ['min_46',['Min',['../struct_min.html',1,'']]], - ['min_47',['min',['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits::min'],['../struct_limits_3_01uint8__t_01_4.html#a408bd5a337e7292f06e63da81193629a',1,'Limits< uint8_t >::min'],['../struct_limits_3_01uint16__t_01_4.html#ae173984c3be8b6750f27daed581805fe',1,'Limits< uint16_t >::min'],['../struct_limits_3_01uint32__t_01_4.html#ab0c3975e02053b234c7b606ababa66e1',1,'Limits< uint32_t >::min'],['../struct_limits_3_01uint64__t_01_4.html#a80627f39e951398283942cefa48f4dd0',1,'Limits< uint64_t >::min'],['../struct_limits_3_01int8__t_01_4.html#a7a809307d2bba80382f0645d277eaa4b',1,'Limits< int8_t >::min'],['../struct_limits_3_01int16__t_01_4.html#adca7139647801e223c35b0abc7da5240',1,'Limits< int16_t >::min'],['../struct_limits_3_01int32__t_01_4.html#af336a1b22a8ed6a83a4cfb5bf8869771',1,'Limits< int32_t >::min'],['../struct_limits_3_01int64__t_01_4.html#a1c90fb96af515badaccaa835b08f7428',1,'Limits< int64_t >::min'],['../struct_limits_3_01half_01_4.html#aca7b036c257878bf1b80912fb5d4516d',1,'Limits< half >::min'],['../struct_limits_3_01float_01_4.html#a3225e334d372ee86128c89a440d8648f',1,'Limits< float >::min'],['../struct_limits_3_01bfloat16__t_01_4.html#a2fd1811b9f615b2b897904bc27d1cb49',1,'Limits< bfloat16_t >::min'],['../struct_limits_3_01bool_01_4.html#a139f787b57536d455490b8ef801d37cc',1,'Limits< bool >::min'],['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#adaed80031f5ca0ff69d30ec4c5d0c98f',1,'metal::_numeric_limits_impl< bfloat16_t >::min()']]], - ['min_48',['Min',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a4f685dcd48e6614d6bb2ccda4f2686ef',1,'mlx::core::distributed::AllReduce::Min'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a0d3d1f5c94725bdc42fa692e2c074418',1,'mlx::core::Reduce::Min'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1a7d2ee8f14f2e70a9d47170fecc6da898',1,'mlx::core::Scan::Min'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613cad914e4c3475ce9858f2de4bf35dcfdbf',1,'mlx::core::Scatter::Min']]], - ['min_49',['min',['../namespacemetal.html#a6653b28c9473087141eddce39878d4d3',1,'metal::min()'],['../namespacemetal_1_1fast.html#a3e958e56a4712687c381a0b64d123e61',1,'metal::fast::min()'],['../namespacemetal_1_1precise.html#afed0da2f7df3505b5dffa2389c3cb36e',1,'metal::precise::min()'],['../group__ops.html#gab27599802617a4c8f9964ab5f4ffee12',1,'mlx::core::min(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8',1,'mlx::core::min(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5',1,'mlx::core::min(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga36fa315eef677f4143868f552cd26d03',1,'mlx::core::min(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['min3_50',['min3',['../namespacemetal.html#a005510c8c0f964ce2b8aad3ba76a7a3f',1,'metal::min3()'],['../namespacemetal_1_1fast.html#a606a4c1b34ce05ea89ca5af81724036f',1,'metal::fast::min3()'],['../namespacemetal_1_1precise.html#a4d37ce31c3549ca4772a4ee29798e231',1,'metal::precise::min3()']]], - ['min_5fexponent_51',['min_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a13829f8c7a7c0efdc8946eff5d3c9470',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['min_5fexponent10_52',['min_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#aeaed172780720e06b8731cef3177e277',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['minimum_53',['Minimum',['../struct_minimum.html',1,'Minimum'],['../structmlx_1_1core_1_1detail_1_1_minimum.html',1,'mlx::core::detail::Minimum'],['../classmlx_1_1core_1_1_minimum.html',1,'mlx::core::Minimum'],['../classmlx_1_1core_1_1_minimum.html#ab0f2ce17108df44b82cff68886b0f6f5',1,'mlx::core::Minimum::Minimum()']]], - ['minimum_54',['minimum',['../group__ops.html#ga49ba00c090f81f331c91b0c97040bce0',1,'mlx::core']]], - ['mlx_55',['mlx',['../namespacemlx.html',1,'']]], - ['mlx_2eh_56',['mlx.h',['../mlx_8h.html',1,'']]], - ['mlx_3a_3acore_57',['core',['../namespacemlx_1_1core.html',1,'mlx']]], - ['mlx_3a_3acore_3a_3aallocator_58',['allocator',['../namespacemlx_1_1core_1_1allocator.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3adetail_59',['detail',['../namespacemlx_1_1core_1_1detail.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3adistributed_60',['distributed',['../namespacemlx_1_1core_1_1distributed.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3adistributed_3a_3adetail_61',['detail',['../namespacemlx_1_1core_1_1distributed_1_1detail.html',1,'mlx::core::distributed']]], - ['mlx_3a_3acore_3a_3afast_62',['fast',['../namespacemlx_1_1core_1_1fast.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3afft_63',['fft',['../namespacemlx_1_1core_1_1fft.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3aio_64',['io',['../namespacemlx_1_1core_1_1io.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3alinalg_65',['linalg',['../namespacemlx_1_1core_1_1linalg.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3ametal_66',['metal',['../namespacemlx_1_1core_1_1metal.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3arandom_67',['random',['../namespacemlx_1_1core_1_1random.html',1,'mlx::core']]], - ['mlx_3a_3acore_3a_3ascheduler_68',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html',1,'mlx::core']]], - ['mlx_3a_3asteel_69',['steel',['../namespacemlx_1_1steel.html',1,'mlx']]], - ['mlx_5fatomic_70',['mlx_atomic',['../structmlx__atomic.html',1,'']]], - ['mlx_5fatomic_3c_20t_2c_20enable_5fif_5ft_3c_20is_5fmetal_5fatomic_3c_20t_20_3e_20_3e_20_3e_71',['mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >',['../structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html',1,'']]], - ['mlx_5fatomic_5fcompare_5fexchange_5fweak_5fexplicit_72',['mlx_atomic_compare_exchange_weak_explicit',['../atomic_8h.html#af63dac2f2df485a85b490e58302491b3',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread T *expected, T val, uint offset): atomic.h'],['../atomic_8h.html#a0c328b82e1b4c8cd0e9c864ecac30136',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread uint *expected, uint val, uint offset): atomic.h']]], - ['mlx_5fatomic_5ffetch_5fadd_5fexplicit_73',['mlx_atomic_fetch_add_explicit',['../atomic_8h.html#a37cbd062017e3d056782dcfa7f5f6add',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fand_5fexplicit_74',['mlx_atomic_fetch_and_explicit',['../atomic_8h.html#a65bd5cae890db9b927d853a52bdc06e8',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_75',['mlx_atomic_fetch_max_explicit',['../atomic_8h.html#aafeb87661c0b216d994677807a78f6d7',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_3c_20float_20_3e_76',['mlx_atomic_fetch_max_explicit< float >',['../atomic_8h.html#a55a4a8fbb8951a18375bdb635cf81af0',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_77',['mlx_atomic_fetch_min_explicit',['../atomic_8h.html#ab962bd4d6ee8040e930e03f54c5ac16a',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_3c_20float_20_3e_78',['mlx_atomic_fetch_min_explicit< float >',['../atomic_8h.html#aa81c17d78ba4c9eb989f288000816c73',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5fmul_5fexplicit_79',['mlx_atomic_fetch_mul_explicit',['../atomic_8h.html#adea4b9574507634f411de665f4db7c3c',1,'atomic.h']]], - ['mlx_5fatomic_5ffetch_5for_5fexplicit_80',['mlx_atomic_fetch_or_explicit',['../atomic_8h.html#a995c2d054cc4faa7b86e9a8719ff3e6f',1,'atomic.h']]], - ['mlx_5fatomic_5fload_5fexplicit_81',['mlx_atomic_load_explicit',['../atomic_8h.html#a802615a5c326e8dd4ab13fefa0231d4e',1,'atomic.h']]], - ['mlx_5fatomic_5fstore_5fexplicit_82',['mlx_atomic_store_explicit',['../atomic_8h.html#a8108e66ef9a5fa8c62e6a7c6504c43aa',1,'atomic.h']]], - ['mlx_5flapack_5ffunc_83',['MLX_LAPACK_FUNC',['../lapack__helper_8h.html#ae22db9704827bf013a0a61f21a47464b',1,'lapack_helper.h']]], - ['mlx_5fmtl_5fconst_84',['MLX_MTL_CONST',['../quantized_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: quantized.h'],['../sort_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: sort.h']]], - ['mlx_5fmtl_5floop_5funroll_85',['MLX_MTL_LOOP_UNROLL',['../sort_8h.html#ad34b622323cebef136669fedd7229515',1,'sort.h']]], - ['mlx_5fmtl_5fpragma_5funroll_86',['MLX_MTL_PRAGMA_UNROLL',['../backend_2metal_2kernels_2utils_8h.html#a069b682d7d21827461544817d722bfd3',1,'utils.h']]], - ['mlxconvparams_87',['MLXConvParams',['../struct_m_l_x_conv_params.html',1,'']]], - ['mlxconvparams_3c_202_20_3e_88',['MLXConvParams< 2 >',['../struct_m_l_x_conv_params.html',1,'']]], - ['mlxfastattentionparams_89',['MLXFastAttentionParams',['../struct_m_l_x_fast_attention_params.html',1,'']]], - ['mlxscaleddotproductattentionparams_90',['MLXScaledDotProductAttentionParams',['../struct_m_l_x_scaled_dot_product_attention_params.html',1,'']]], - ['mma_91',['mma',['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA']]], - ['mma_2eh_92',['mma.h',['../mma_8h.html',1,'']]], - ['mma_5ft_93',['mma_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel']]], - ['move_5fshared_5fbuffer_94',['move_shared_buffer',['../classmlx_1_1core_1_1array.html#acce00db63e0f3d80f797b02397ade836',1,'mlx::core::array::move_shared_buffer(array other, const std::vector< size_t > &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a38d7ad605f8282e5e49d0c09e0555c78',1,'mlx::core::array::move_shared_buffer(array other)']]], - ['moveaxis_95',['moveaxis',['../group__ops.html#ga24067d10a842db2c9d509ea48135a2c3',1,'mlx::core']]], - ['mpinplace_96',['MPINPLACE',['../namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2',1,'pocketfft::detail']]], - ['mps_97',['MPS',['../namespace_m_p_s.html',1,'']]], - ['mtl_98',['MTL',['../namespace_m_t_l.html',1,'']]], - ['mtl_3a_3aprivate_99',['Private',['../namespace_m_t_l_1_1_private.html',1,'MTL']]], - ['mtl_3a_3aprivate_3a_3aclass_100',['Class',['../namespace_m_t_l_1_1_private_1_1_class.html',1,'MTL::Private']]], - ['mtl_3a_3aprivate_3a_3aselector_101',['Selector',['../namespace_m_t_l_1_1_private_1_1_selector.html',1,'MTL::Private']]], - ['mtl_5fconst_102',['MTL_CONST',['../defines_8h.html#a767ed9f2604de22b259cee02c4ce1d22',1,'defines.h']]], - ['mtl_5fdevice_103',['mtl_device',['../classmlx_1_1core_1_1metal_1_1_device.html#a31dba377f2be44a746db10d1b9367653',1,'mlx::core::metal::Device']]], - ['mtlfclist_104',['MTLFCList',['../namespacemlx_1_1core_1_1metal.html#a616e09a1ef321d527770721cef264c54',1,'mlx::core::metal']]], - ['mtx_105',['mtx',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a70410c9e612f871663929f1e8441a976',1,'mlx::core::scheduler::StreamThread']]], - ['multi_5fiter_106',['multi_iter',['../classpocketfft_1_1detail_1_1multi__iter.html',1,'pocketfft::detail::multi_iter< N >'],['../classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9',1,'pocketfft::detail::multi_iter::multi_iter()']]], - ['multiply_107',['Multiply',['../structmlx_1_1core_1_1detail_1_1_multiply.html',1,'mlx::core::detail::Multiply'],['../classmlx_1_1core_1_1_multiply.html',1,'mlx::core::Multiply'],['../struct_multiply.html',1,'Multiply'],['../classmlx_1_1core_1_1_multiply.html#aca5c50f900321f3eb4d6fbcbc225c00c',1,'mlx::core::Multiply::Multiply()']]], - ['multiply_108',['multiply',['../group__ops.html#gaf57392e641640b5d06e4c99518391c38',1,'mlx::core']]], - ['multivariate_5fnormal_109',['multivariate_normal',['../namespacemlx_1_1core_1_1random.html#a8c37da3c1c0c561cad7499d6d9db81fb',1,'mlx::core::random']]] + ['make_5fstring_3',['make_string',['../namespacemlx_1_1core.html#aed148d95e7b5221f1312473deded0d27',1,'mlx::core']]], + ['make_5fsynchronize_5ftask_4',['make_synchronize_task',['../namespacemlx_1_1core_1_1metal.html#ab31abdda3052162d59f6590a89e38337',1,'mlx::core::metal']]], + ['make_5ftask_5',['make_task',['../namespacemlx_1_1core_1_1metal.html#a4552b7ccdfa7f3cc9895c09799d8048e',1,'mlx::core::metal']]], + ['malloc_6',['malloc',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a9a17d2c7a97772bf4a15e6c74af34ca4',1,'mlx::core::allocator::Allocator::malloc()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#a4f3d5de6b8c0eba22e9403b28a5ef3f0',1,'mlx::core::allocator::CommonAllocator::malloc()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a6c0feb9b1ff9977f76c69745393944bc',1,'mlx::core::metal::MetalAllocator::malloc()'],['../namespacemlx_1_1core_1_1allocator.html#a560d10a166e3c294f3757166f9bd6801',1,'mlx::core::allocator::malloc(size_t size)']]], + ['malloc_5for_5fwait_7',['malloc_or_wait',['../namespacemlx_1_1core_1_1allocator.html#a86ac0a11ff78f21e717f641716c34abc',1,'mlx::core::allocator']]], + ['mask_5fh_8',['mask_h',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0b892c1a7edb9ed20c076d8945855c19',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], + ['mask_5ft_9',['mask_t',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a270ab3da7c98a12525a59952742cc97d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], + ['mask_5fw_10',['mask_w',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a19ddba7259c3c2c02ed90f3f635557be',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], + ['matmul_11',['Matmul',['../classmlx_1_1core_1_1_matmul.html',1,'mlx::core::Matmul'],['../classmlx_1_1core_1_1_matmul.html#adef92f30ab35e540ccb316ea6b94e6f7',1,'mlx::core::Matmul::Matmul()']]], + ['matmul_12',['matmul',['../group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a',1,'mlx::core']]], + ['matmul_2eh_13',['matmul.h',['../matmul_8h.html',1,'']]], + ['max_14',['Max',['../struct_max.html',1,'Max< U >'],['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924ac00cf69bbba24f7ab08d3ad618705988',1,'mlx::core::distributed::AllReduce::Max'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a3d11c500ea4f7f639e20dd0755d39260',1,'mlx::core::Reduce::Max'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1ad54b2905015a390708f79bae6cdac56d',1,'mlx::core::Scan::Max'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca1c2da7b96d743296fe660f5fc4072f16',1,'mlx::core::Scatter::Max']]], + ['max_15',['max',['../struct_limits.html#a2f0673b6f9da89ce1d64f9f3d74f50a8',1,'Limits::max'],['../struct_limits_3_01uint8__t_01_4.html#a1570fb640e2e41f96776db5ca08d500c',1,'Limits< uint8_t >::max'],['../struct_limits_3_01uint16__t_01_4.html#a228b33556ba4cb7e6137ab6258628488',1,'Limits< uint16_t >::max'],['../struct_limits_3_01uint32__t_01_4.html#a91fa8f7214ec936976a8324c7431c651',1,'Limits< uint32_t >::max'],['../struct_limits_3_01uint64__t_01_4.html#aa8c2257881a4e1fa8596fa07dba5e107',1,'Limits< uint64_t >::max'],['../struct_limits_3_01int8__t_01_4.html#a96fed01fa9249226be69760652643289',1,'Limits< int8_t >::max'],['../struct_limits_3_01int16__t_01_4.html#a12d64c398ca7609b7c906f3cf1a6f678',1,'Limits< int16_t >::max'],['../struct_limits_3_01int32__t_01_4.html#af756344b31e84222dd73d3445dcd5640',1,'Limits< int32_t >::max'],['../struct_limits_3_01int64__t_01_4.html#ac9c420604c0f3d237ddfb2b8a2439224',1,'Limits< int64_t >::max'],['../struct_limits_3_01half_01_4.html#a4f9515dbf2a622074f121bea39a7b175',1,'Limits< half >::max'],['../struct_limits_3_01float_01_4.html#aba172b22b388190aa3969ef16885d8a6',1,'Limits< float >::max'],['../struct_limits_3_01bfloat16__t_01_4.html#a0ead3618da6718629ea9fa4670b5005f',1,'Limits< bfloat16_t >::max'],['../struct_limits_3_01bool_01_4.html#acbd2132145888d51220558a101ffcff4',1,'Limits< bool >::max'],['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a92320d40a58218e40cc414986ac95c50',1,'metal::_numeric_limits_impl< bfloat16_t >::max()'],['../namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b',1,'metal::max()'],['../namespacemetal_1_1fast.html#a747e2e58092a27fb8b4dd3d16934fb52',1,'metal::fast::max()'],['../namespacemetal_1_1precise.html#a6a954a4e4e3753303d1dc734855a185f',1,'metal::precise::max()'],['../group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7',1,'mlx::core::max(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc',1,'mlx::core::max(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f',1,'mlx::core::max(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga7b638050e03a93f2896c981bc2850a47',1,'mlx::core::max(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['max3_16',['max3',['../namespacemetal.html#a00f9c0ad66d969794614f56912eed9c9',1,'metal::max3()'],['../namespacemetal_1_1fast.html#a6fc2cf18ffa8149561864c86dba0f803',1,'metal::fast::max3()'],['../namespacemetal_1_1precise.html#ac490e8614ebd2c9343af1ae6c0d4e82c',1,'metal::precise::max3()']]], + ['max_5fdigits10_17',['max_digits10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a8d3905e6f158379a0c52682266e8d0e2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['max_5fexponent_18',['max_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61bb136f819fa392c50bdf3c38f3aad2',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['max_5fexponent10_19',['max_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a76bfb2deb0e0afc011f77bf5a6d0ed94',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['max_5foutput_5fsize_20',['MAX_OUTPUT_SIZE',['../backend_2metal_2kernels_2fft_8h.html#a28d683cf067736d76f867f30c066317e',1,'fft.h']]], + ['max_5fradix_21',['MAX_RADIX',['../backend_2metal_2kernels_2fft_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: fft.h'],['../readwrite_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: readwrite.h']]], + ['max_5freduce_5fspecialized_5fdims_22',['MAX_REDUCE_SPECIALIZED_DIMS',['../defines_8h.html#a15629f1b81a2b6f1cca26d07a2734623',1,'defines.h']]], + ['max_5fthreads_23',['max_threads',['../namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70',1,'pocketfft::detail::threading']]], + ['maximum_24',['Maximum',['../struct_maximum.html',1,'Maximum'],['../structmlx_1_1core_1_1detail_1_1_maximum.html',1,'mlx::core::detail::Maximum'],['../classmlx_1_1core_1_1_maximum.html',1,'mlx::core::Maximum'],['../classmlx_1_1core_1_1_maximum.html#a28389307e385efe1b2955b86b115e816',1,'mlx::core::Maximum::Maximum()']]], + ['maximum_25',['maximum',['../group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d',1,'mlx::core']]], + ['mb_5fblock_5fmerge_26',['mb_block_merge',['../sort_8h.html#ab381cd57f344bc7304ab580bfdc78807',1,'sort.h']]], + ['mb_5fblock_5fpartition_27',['mb_block_partition',['../sort_8h.html#a32cbe4163b8b0f5cb2c97b256119a4b2',1,'sort.h']]], + ['mb_5fblock_5fsort_28',['mb_block_sort',['../sort_8h.html#a2a0533103661dd378d6bfe949930650a',1,'sort.h']]], + ['mean_29',['mean',['../group__ops.html#gade46e768fd46b8b640eb16f26abeecef',1,'mlx::core::mean(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga52b59fdd8e8430538e564f5bbcfa31e6',1,'mlx::core::mean(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga066161f3d3e395a1d76c638cb680d444',1,'mlx::core::mean(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga45fba73eab0e3b6e128ed3ce2f43a5da',1,'mlx::core::mean(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['median3_30',['median3',['../namespacemetal.html#aa3ff49457ce3c93fc1c0897fd1525157',1,'metal::median3()'],['../namespacemetal_1_1fast.html#a742b55f1e4369921ee7f60d70185bfbc',1,'metal::fast::median3()'],['../namespacemetal_1_1precise.html#a14555ff99c4388493fec48e070144ae2',1,'metal::precise::median3()']]], + ['merge_5fpartition_31',['merge_partition',['../struct_block_merge_sort.html#ab2300cbecb23f3433bad888924c831ca',1,'BlockMergeSort::merge_partition()'],['../struct_kernel_multi_block_merge_sort.html#ab15895b4233aba0e279cc44a07a201fe',1,'KernelMultiBlockMergeSort::merge_partition()']]], + ['merge_5fstep_32',['merge_step',['../struct_block_merge_sort.html#ab65f190edf1851b37c39ad49ce99a43c',1,'BlockMergeSort']]], + ['meshgrid_33',['meshgrid',['../group__ops.html#ga577c911618575314de63d1060656a26e',1,'mlx::core']]], + ['metal_34',['metal',['../namespacemetal.html',1,'']]], + ['metal_2eh_35',['metal.h',['../metal_8h.html',1,'']]], + ['metal_3a_3afast_36',['fast',['../namespacemetal_1_1fast.html',1,'metal']]], + ['metal_3a_3aprecise_37',['precise',['../namespacemetal_1_1precise.html',1,'metal']]], + ['metal_5fimpl_2eh_38',['metal_impl.h',['../metal__impl_8h.html',1,'']]], + ['metalallocator_39',['MetalAllocator',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html',1,'mlx::core::metal']]], + ['min_40',['Min',['../struct_min.html',1,'']]], + ['min_41',['min',['../struct_limits.html#a6e81584ba65a4dc6ff9366b458e3a20e',1,'Limits::min'],['../struct_limits_3_01uint8__t_01_4.html#a408bd5a337e7292f06e63da81193629a',1,'Limits< uint8_t >::min'],['../struct_limits_3_01uint16__t_01_4.html#ae173984c3be8b6750f27daed581805fe',1,'Limits< uint16_t >::min'],['../struct_limits_3_01uint32__t_01_4.html#ab0c3975e02053b234c7b606ababa66e1',1,'Limits< uint32_t >::min'],['../struct_limits_3_01uint64__t_01_4.html#a80627f39e951398283942cefa48f4dd0',1,'Limits< uint64_t >::min'],['../struct_limits_3_01int8__t_01_4.html#a7a809307d2bba80382f0645d277eaa4b',1,'Limits< int8_t >::min'],['../struct_limits_3_01int16__t_01_4.html#adca7139647801e223c35b0abc7da5240',1,'Limits< int16_t >::min'],['../struct_limits_3_01int32__t_01_4.html#af336a1b22a8ed6a83a4cfb5bf8869771',1,'Limits< int32_t >::min'],['../struct_limits_3_01int64__t_01_4.html#a1c90fb96af515badaccaa835b08f7428',1,'Limits< int64_t >::min'],['../struct_limits_3_01half_01_4.html#aca7b036c257878bf1b80912fb5d4516d',1,'Limits< half >::min'],['../struct_limits_3_01float_01_4.html#a3225e334d372ee86128c89a440d8648f',1,'Limits< float >::min'],['../struct_limits_3_01bfloat16__t_01_4.html#a2fd1811b9f615b2b897904bc27d1cb49',1,'Limits< bfloat16_t >::min'],['../struct_limits_3_01bool_01_4.html#a139f787b57536d455490b8ef801d37cc',1,'Limits< bool >::min'],['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#adaed80031f5ca0ff69d30ec4c5d0c98f',1,'metal::_numeric_limits_impl< bfloat16_t >::min()']]], + ['min_42',['Min',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abb4560980e5d01aed14175ce8f6fc924a4f685dcd48e6614d6bb2ccda4f2686ef',1,'mlx::core::distributed::AllReduce::Min'],['../classmlx_1_1core_1_1_reduce.html#a0848518b16ae6d4043d6be247bdf31c9a0d3d1f5c94725bdc42fa692e2c074418',1,'mlx::core::Reduce::Min'],['../classmlx_1_1core_1_1_scan.html#a47bf2ec54ead4b8f00f9f188518630f1a7d2ee8f14f2e70a9d47170fecc6da898',1,'mlx::core::Scan::Min'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613cad914e4c3475ce9858f2de4bf35dcfdbf',1,'mlx::core::Scatter::Min']]], + ['min_43',['min',['../namespacemetal.html#a6653b28c9473087141eddce39878d4d3',1,'metal::min()'],['../namespacemetal_1_1fast.html#a3e958e56a4712687c381a0b64d123e61',1,'metal::fast::min()'],['../namespacemetal_1_1precise.html#afed0da2f7df3505b5dffa2389c3cb36e',1,'metal::precise::min()'],['../group__ops.html#gab27599802617a4c8f9964ab5f4ffee12',1,'mlx::core::min(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8',1,'mlx::core::min(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5',1,'mlx::core::min(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga36fa315eef677f4143868f552cd26d03',1,'mlx::core::min(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['min3_44',['min3',['../namespacemetal.html#a005510c8c0f964ce2b8aad3ba76a7a3f',1,'metal::min3()'],['../namespacemetal_1_1fast.html#a606a4c1b34ce05ea89ca5af81724036f',1,'metal::fast::min3()'],['../namespacemetal_1_1precise.html#a4d37ce31c3549ca4772a4ee29798e231',1,'metal::precise::min3()']]], + ['min_5fexponent_45',['min_exponent',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a13829f8c7a7c0efdc8946eff5d3c9470',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['min_5fexponent10_46',['min_exponent10',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#aeaed172780720e06b8731cef3177e277',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['minimum_47',['Minimum',['../struct_minimum.html',1,'Minimum'],['../structmlx_1_1core_1_1detail_1_1_minimum.html',1,'mlx::core::detail::Minimum'],['../classmlx_1_1core_1_1_minimum.html',1,'mlx::core::Minimum'],['../classmlx_1_1core_1_1_minimum.html#ab0f2ce17108df44b82cff68886b0f6f5',1,'mlx::core::Minimum::Minimum()']]], + ['minimum_48',['minimum',['../group__ops.html#ga49ba00c090f81f331c91b0c97040bce0',1,'mlx::core']]], + ['mlx_49',['mlx',['../namespacemlx.html',1,'']]], + ['mlx_2eh_50',['mlx.h',['../mlx_8h.html',1,'']]], + ['mlx_3a_3acore_51',['core',['../namespacemlx_1_1core.html',1,'mlx']]], + ['mlx_3a_3acore_3a_3aallocator_52',['allocator',['../namespacemlx_1_1core_1_1allocator.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3adetail_53',['detail',['../namespacemlx_1_1core_1_1detail.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3adistributed_54',['distributed',['../namespacemlx_1_1core_1_1distributed.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3adistributed_3a_3adetail_55',['detail',['../namespacemlx_1_1core_1_1distributed_1_1detail.html',1,'mlx::core::distributed']]], + ['mlx_3a_3acore_3a_3afast_56',['fast',['../namespacemlx_1_1core_1_1fast.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3afft_57',['fft',['../namespacemlx_1_1core_1_1fft.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3aio_58',['io',['../namespacemlx_1_1core_1_1io.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3alinalg_59',['linalg',['../namespacemlx_1_1core_1_1linalg.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3ametal_60',['metal',['../namespacemlx_1_1core_1_1metal.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3arandom_61',['random',['../namespacemlx_1_1core_1_1random.html',1,'mlx::core']]], + ['mlx_3a_3acore_3a_3ascheduler_62',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html',1,'mlx::core']]], + ['mlx_3a_3asteel_63',['steel',['../namespacemlx_1_1steel.html',1,'mlx']]], + ['mlx_5fatomic_64',['mlx_atomic',['../structmlx__atomic.html',1,'']]], + ['mlx_5fatomic_3c_20t_2c_20enable_5fif_5ft_3c_20is_5fmetal_5fatomic_3c_20t_20_3e_20_3e_20_3e_65',['mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >',['../structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html',1,'']]], + ['mlx_5fatomic_5fcompare_5fexchange_5fweak_5fexplicit_66',['mlx_atomic_compare_exchange_weak_explicit',['../atomic_8h.html#af63dac2f2df485a85b490e58302491b3',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread T *expected, T val, uint offset): atomic.h'],['../atomic_8h.html#a0c328b82e1b4c8cd0e9c864ecac30136',1,'mlx_atomic_compare_exchange_weak_explicit(device mlx_atomic< T > *object, thread uint *expected, uint val, uint offset): atomic.h']]], + ['mlx_5fatomic_5ffetch_5fadd_5fexplicit_67',['mlx_atomic_fetch_add_explicit',['../atomic_8h.html#a37cbd062017e3d056782dcfa7f5f6add',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fand_5fexplicit_68',['mlx_atomic_fetch_and_explicit',['../atomic_8h.html#a65bd5cae890db9b927d853a52bdc06e8',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_69',['mlx_atomic_fetch_max_explicit',['../atomic_8h.html#aafeb87661c0b216d994677807a78f6d7',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmax_5fexplicit_3c_20float_20_3e_70',['mlx_atomic_fetch_max_explicit< float >',['../atomic_8h.html#a55a4a8fbb8951a18375bdb635cf81af0',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_71',['mlx_atomic_fetch_min_explicit',['../atomic_8h.html#ab962bd4d6ee8040e930e03f54c5ac16a',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmin_5fexplicit_3c_20float_20_3e_72',['mlx_atomic_fetch_min_explicit< float >',['../atomic_8h.html#aa81c17d78ba4c9eb989f288000816c73',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5fmul_5fexplicit_73',['mlx_atomic_fetch_mul_explicit',['../atomic_8h.html#adea4b9574507634f411de665f4db7c3c',1,'atomic.h']]], + ['mlx_5fatomic_5ffetch_5for_5fexplicit_74',['mlx_atomic_fetch_or_explicit',['../atomic_8h.html#a995c2d054cc4faa7b86e9a8719ff3e6f',1,'atomic.h']]], + ['mlx_5fatomic_5fload_5fexplicit_75',['mlx_atomic_load_explicit',['../atomic_8h.html#a802615a5c326e8dd4ab13fefa0231d4e',1,'atomic.h']]], + ['mlx_5fatomic_5fstore_5fexplicit_76',['mlx_atomic_store_explicit',['../atomic_8h.html#a8108e66ef9a5fa8c62e6a7c6504c43aa',1,'atomic.h']]], + ['mlx_5flapack_5ffunc_77',['MLX_LAPACK_FUNC',['../lapack__helper_8h.html#ae22db9704827bf013a0a61f21a47464b',1,'lapack_helper.h']]], + ['mlx_5fmtl_5fconst_78',['MLX_MTL_CONST',['../kernels_2gemv__masked_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: gemv_masked.h'],['../quantized_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: quantized.h'],['../sort_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: sort.h']]], + ['mlx_5fmtl_5floop_5funroll_79',['MLX_MTL_LOOP_UNROLL',['../sort_8h.html#ad34b622323cebef136669fedd7229515',1,'sort.h']]], + ['mlx_5fmtl_5fpragma_5funroll_80',['MLX_MTL_PRAGMA_UNROLL',['../kernels_2gemv__masked_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: gemv_masked.h'],['../backend_2metal_2kernels_2utils_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: utils.h']]], + ['mlxconvparams_81',['MLXConvParams',['../struct_m_l_x_conv_params.html',1,'']]], + ['mlxconvparams_3c_202_20_3e_82',['MLXConvParams< 2 >',['../struct_m_l_x_conv_params.html',1,'']]], + ['mlxfastattentionparams_83',['MLXFastAttentionParams',['../struct_m_l_x_fast_attention_params.html',1,'']]], + ['mlxscaleddotproductattentionparams_84',['MLXScaledDotProductAttentionParams',['../struct_m_l_x_scaled_dot_product_attention_params.html',1,'']]], + ['mma_85',['mma',['../structmlx_1_1steel_1_1_block_m_m_a.html#a6a2c2a6d5e767d52c41b42a9d36086b0',1,'mlx::steel::BlockMMA']]], + ['mma_2eh_86',['mma.h',['../mma_8h.html',1,'']]], + ['mma_5ft_87',['mma_t',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#add8c6a31011a4895667c2a94a5af3782',1,'mlx::steel::GEMMKernel']]], + ['move_5fshared_5fbuffer_88',['move_shared_buffer',['../classmlx_1_1core_1_1array.html#acce00db63e0f3d80f797b02397ade836',1,'mlx::core::array::move_shared_buffer(array other, const std::vector< size_t > &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a38d7ad605f8282e5e49d0c09e0555c78',1,'mlx::core::array::move_shared_buffer(array other)']]], + ['moveaxis_89',['moveaxis',['../group__ops.html#ga24067d10a842db2c9d509ea48135a2c3',1,'mlx::core']]], + ['mpinplace_90',['MPINPLACE',['../namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2',1,'pocketfft::detail']]], + ['mtl_5fconst_91',['MTL_CONST',['../defines_8h.html#a767ed9f2604de22b259cee02c4ce1d22',1,'defines.h']]], + ['mtl_5fdevice_92',['mtl_device',['../classmlx_1_1core_1_1metal_1_1_device.html#a31dba377f2be44a746db10d1b9367653',1,'mlx::core::metal::Device']]], + ['mtlfclist_93',['MTLFCList',['../namespacemlx_1_1core_1_1metal.html#a616e09a1ef321d527770721cef264c54',1,'mlx::core::metal']]], + ['mtx_94',['mtx',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a70410c9e612f871663929f1e8441a976',1,'mlx::core::scheduler::StreamThread']]], + ['multi_5fiter_95',['multi_iter',['../classpocketfft_1_1detail_1_1multi__iter.html',1,'pocketfft::detail::multi_iter< N >'],['../classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9',1,'pocketfft::detail::multi_iter::multi_iter()']]], + ['multiply_96',['Multiply',['../structmlx_1_1core_1_1detail_1_1_multiply.html',1,'mlx::core::detail::Multiply'],['../classmlx_1_1core_1_1_multiply.html',1,'mlx::core::Multiply'],['../struct_multiply.html',1,'Multiply'],['../classmlx_1_1core_1_1_multiply.html#aca5c50f900321f3eb4d6fbcbc225c00c',1,'mlx::core::Multiply::Multiply()']]], + ['multiply_97',['multiply',['../group__ops.html#gaf57392e641640b5d06e4c99518391c38',1,'mlx::core']]], + ['multivariate_5fnormal_98',['multivariate_normal',['../namespacemlx_1_1core_1_1random.html#a8c37da3c1c0c561cad7499d6d9db81fb',1,'mlx::core::random']]] ]; diff --git a/docs/build/html/search/all_e.js b/docs/build/html/search/all_e.js index f08f638931..383b2a56a4 100644 --- a/docs/build/html/search/all_e.js +++ b/docs/build/html/search/all_e.js @@ -1,7 +1,7 @@ var searchData= [ - ['n_0',['N',['../struct_m_l_x_fast_attention_params.html#ab42c792a80388002e34992cbd837a167',1,'MLXFastAttentionParams::N'],['../struct_m_l_x_conv_params.html#ae6b7054dc3cffa8e6aedeb29fa7da932',1,'MLXConvParams::N'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a213f5ea4018120d8b61ab82754aaba83',1,'mlx::steel::ImplicitGemmConv2DParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a174626ab98515d89923b2841a664b9a1',1,'mlx::steel::GEMMParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a1103e79fb8962812b9a3c9d5c902ff86',1,'mlx::steel::GEMMSpiltKParams::N']]], - ['n_1',['n',['../struct_read_writer.html#a655346c9ebfc33a69da3f1c1d4238dfb',1,'ReadWriter']]], + ['n_0',['n',['../struct_read_writer.html#a655346c9ebfc33a69da3f1c1d4238dfb',1,'ReadWriter']]], + ['n_1',['N',['../struct_m_l_x_fast_attention_params.html#ab42c792a80388002e34992cbd837a167',1,'MLXFastAttentionParams::N'],['../struct_m_l_x_conv_params.html#ae6b7054dc3cffa8e6aedeb29fa7da932',1,'MLXConvParams::N'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a213f5ea4018120d8b61ab82754aaba83',1,'mlx::steel::ImplicitGemmConv2DParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a174626ab98515d89923b2841a664b9a1',1,'mlx::steel::GEMMParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a1103e79fb8962812b9a3c9d5c902ff86',1,'mlx::steel::GEMMSpiltKParams::N']]], ['n_5factive_5ftasks_2',['n_active_tasks',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a3c9fa21442974acba3409d49bb033131',1,'mlx::core::scheduler::Scheduler::n_active_tasks()'],['../namespacemlx_1_1core_1_1scheduler.html#a9bf641981df5fc16b0fb0dbacc0c3afd',1,'mlx::core::scheduler::n_active_tasks()']]], ['n_5fchannels_3',['n_channels',['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a06c2fb9c93660e8f6916228cd77f9494',1,'mlx::steel::ChannelHelper< 1 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#ac66ff37bc2cf78d96667192a6cca73b5',1,'mlx::steel::ChannelHelper< 2 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a071c015713b7bab09930661165517eff',1,'mlx::steel::ChannelHelper< 3 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#a167b00a84adf93b60e3d7a943d5eb977',1,'mlx::steel::ChannelHelper< 4 >::n_channels']]], ['n_5fkv_5fheads_4',['N_KV_HEADS',['../struct_m_l_x_scaled_dot_product_attention_params.html#a68a292b9986c20560aca88394f82e9f7',1,'MLXScaledDotProductAttentionParams']]], @@ -16,28 +16,29 @@ var searchData= ['nd_5floop_13',['nd_loop',['../namespacemlx_1_1core.html#a9a9254ce9975ec247a2718bc02d6f201',1,'mlx::core']]], ['ndarr_14',['ndarr',['../classpocketfft_1_1detail_1_1ndarr.html',1,'pocketfft::detail::ndarr< T >'],['../classpocketfft_1_1detail_1_1ndarr.html#a8f0037a172d96cb1ad915a5069175fa2',1,'pocketfft::detail::ndarr::ndarr()']]], ['ndim_15',['ndim',['../struct_indices.html#a7dec359e91d0eb2b64e5461b54308313',1,'Indices::ndim'],['../classpocketfft_1_1detail_1_1arr__info.html#ac608c8af2a59a28a0012e308be7ee414',1,'pocketfft::detail::arr_info::ndim()'],['../classmlx_1_1core_1_1array.html#a53006e77d13d9d88b525ef577748939f',1,'mlx::core::array::ndim()']]], - ['negative_16',['Negative',['../structmlx_1_1core_1_1detail_1_1_negative.html',1,'mlx::core::detail::Negative'],['../classmlx_1_1core_1_1_negative.html',1,'mlx::core::Negative'],['../struct_negative.html',1,'Negative'],['../classmlx_1_1core_1_1_negative.html#aa3b73395d9fa5b7215dca488bc0d3c70',1,'mlx::core::Negative::Negative()']]], - ['negative_17',['negative',['../group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6',1,'mlx::core']]], - ['new_5fqueue_18',['new_queue',['../classmlx_1_1core_1_1metal_1_1_device.html#a8135ae2a8c1e6f3861e84d4e60c28b67',1,'mlx::core::metal::Device']]], - ['new_5fscoped_5fmemory_5fpool_19',['new_scoped_memory_pool',['../namespacemlx_1_1core_1_1metal.html#a6c52659362f6da31629216b65c297407',1,'mlx::core::metal']]], - ['new_5fstream_20',['new_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a157c8da85fa1bddb8eacf8515a3cc879',1,'mlx::core::scheduler::Scheduler::new_stream()'],['../namespacemlx_1_1core_1_1metal.html#a8b4188f9a090a1da42d62b8a369bf106',1,'mlx::core::metal::new_stream()'],['../namespacemlx_1_1core.html#a6f7c63a9be10337b3b96d527e1db3c2f',1,'mlx::core::new_stream()']]], - ['next_21',['next',['../struct_quantized_block_loader.html#a674138ef7c43cc45586ea9f8fd6f6bd9',1,'QuantizedBlockLoader::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a3b71f379ff9baf39830c92f4f1ecde52',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a78d2b0098311a278be8394edbd5fc731',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aae56c19bb562219770fec38e5666c6ce',1,'mlx::steel::Conv2DWeightBlockLoader::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af9ce1a767266664bea131a5437002c80',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a30b10bebde7f08b89d03bdd9ea0f48da',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3e5ee68ed0ee43f7e979dd4222f76a8c',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a11743cb1c108f42ccdc6e59204a5b3e8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a4193c5eac3ef093a740d5305b25d3e18',1,'mlx::core::random::KeySequence::next()'],['../backend_2metal_2allocator_8h.html#ae704ab07eac590091daa5fc4aec7bddb',1,'next: allocator.h']]], - ['next_5fpower_5fof_5f2_22',['next_power_of_2',['../namespacemlx_1_1core.html#a685c0530e338aabc622325685846ce93',1,'mlx::core']]], - ['nextafter_23',['nextafter',['../namespacemetal.html#a9547fd7b09164931986f6db4813bd72d',1,'metal::nextafter()'],['../namespacemetal_1_1fast.html#a4583e8be04fc0bd475b97b0934604f23',1,'metal::fast::nextafter()'],['../namespacemetal_1_1precise.html#ad012ceeb55b77f1533749b351331e026',1,'metal::precise::nextafter()']]], - ['no_5ffuse_24',['no_fuse',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4ada8df7fd43da6073fec4fe5666b03dbb',1,'mlx::core']]], - ['no_5fsimplify_25',['no_simplify',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a8e5611dfddbae6e68624c59aa3e4e3e2',1,'mlx::core']]], - ['nodenamer_26',['NodeNamer',['../structmlx_1_1core_1_1_node_namer.html',1,'mlx::core']]], - ['nomask_5ft_27',['nomask_t',['../steel__gemm__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'steel_gemm_masked.h']]], - ['none_28',['None',['../struct_none.html',1,'None'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca87a6a1927de175b71d7d0b5c11b8665c',1,'mlx::core::Scatter::None']]], - ['norm_29',['norm',['../namespacemlx_1_1core_1_1linalg.html#aba765b8e95e9a1d33d31f727a185919d',1,'mlx::core::linalg::norm(const array &a, const double ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#acaa85b4146821c268abecec2422c02d2',1,'mlx::core::linalg::norm(const array &a, const double ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#af1ebe0c6dcba9a1c49b5e397dddf3264',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#ae8da67e4c6e073f93889f1051203cd9e',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a229018071d5602e38d6248230f334a10',1,'mlx::core::linalg::norm(const array &a, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a44250cff34238f01471fd61e76036f03',1,'mlx::core::linalg::norm(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['normal_30',['normal',['../namespacemlx_1_1core_1_1random.html#aa3977269b77f66c6397c7c72f873d17a',1,'mlx::core::random::normal(const std::vector< int > &shape, Dtype dtype, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa3516ace3ebc5e80ca73fb8f50df436f',1,'mlx::core::random::normal(const std::vector< int > &shape, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a75ecf88e36a4013ecda18a91cb60befb',1,'mlx::core::random::normal(const std::vector< int > &shape, const Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7723a27a0093a1f51ca31d26c1d061c8',1,'mlx::core::random::normal(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['normalize_5faxis_31',['normalize_axis',['../namespacemlx_1_1core.html#a0181b5d72bf3d34448dabc70f7ff858d',1,'mlx::core']]], - ['not_5fequal_32',['not_equal',['../group__ops.html#ga28f22c5d2c399eee53be7b3facc11103',1,'mlx::core']]], - ['notequal_33',['NotEqual',['../structmlx_1_1core_1_1detail_1_1_not_equal.html',1,'mlx::core::detail::NotEqual'],['../classmlx_1_1core_1_1_not_equal.html',1,'mlx::core::NotEqual'],['../struct_not_equal.html',1,'NotEqual'],['../classmlx_1_1core_1_1_not_equal.html#ac568397bd17b5d9f25ad1a0ebadedbb9',1,'mlx::core::NotEqual::NotEqual()']]], - ['notify_5fnew_5ftask_34',['notify_new_task',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ae8aa34a9be8bc73508dd500000421173',1,'mlx::core::scheduler::Scheduler::notify_new_task()'],['../namespacemlx_1_1core_1_1scheduler.html#a6b7289e33cef665178fe614aac75c1b2',1,'mlx::core::scheduler::notify_new_task()']]], - ['notify_5ftask_5fcompletion_35',['notify_task_completion',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#abbb2b1c2f8bae2b9c7cc51db65f18a3b',1,'mlx::core::scheduler::Scheduler::notify_task_completion()'],['../namespacemlx_1_1core_1_1scheduler.html#a1d06ffdbab36790b78deb6e34adc737f',1,'mlx::core::scheduler::notify_task_completion()']]], - ['num_5fthreads_36',['num_threads',['../namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833',1,'pocketfft::detail::threading']]], - ['number_37',['number',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dab1bc248a7ff2b2e95569f56de68615df',1,'mlx::core::Dtype::number'],['../namespacemlx_1_1core.html#a069c0aab6b36aef34419534ec4a4310d',1,'mlx::core::number']]], - ['number_5fof_5felements_38',['number_of_elements',['../group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636',1,'mlx::core']]], - ['numberofelements_39',['NumberOfElements',['../classmlx_1_1core_1_1_number_of_elements.html',1,'mlx::core::NumberOfElements'],['../classmlx_1_1core_1_1_number_of_elements.html#ac64d7c40ae29d687f8b7d2fa33e13b06',1,'mlx::core::NumberOfElements::NumberOfElements()']]] + ['needs_5ftgp_5freduction_16',['needs_tgp_reduction',['../struct_g_e_m_v_kernel.html#ae8113fddf6fb637acfd12efd978b704c',1,'GEMVKernel::needs_tgp_reduction'],['../struct_g_e_m_v_t_kernel.html#a67be7ec69c3791f02e97ccdb00ae0e03',1,'GEMVTKernel::needs_tgp_reduction']]], + ['negative_17',['Negative',['../structmlx_1_1core_1_1detail_1_1_negative.html',1,'mlx::core::detail::Negative'],['../classmlx_1_1core_1_1_negative.html',1,'mlx::core::Negative'],['../struct_negative.html',1,'Negative'],['../classmlx_1_1core_1_1_negative.html#aa3b73395d9fa5b7215dca488bc0d3c70',1,'mlx::core::Negative::Negative()']]], + ['negative_18',['negative',['../group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6',1,'mlx::core']]], + ['new_5fqueue_19',['new_queue',['../classmlx_1_1core_1_1metal_1_1_device.html#a8135ae2a8c1e6f3861e84d4e60c28b67',1,'mlx::core::metal::Device']]], + ['new_5fscoped_5fmemory_5fpool_20',['new_scoped_memory_pool',['../namespacemlx_1_1core_1_1metal.html#a6c52659362f6da31629216b65c297407',1,'mlx::core::metal']]], + ['new_5fstream_21',['new_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a157c8da85fa1bddb8eacf8515a3cc879',1,'mlx::core::scheduler::Scheduler::new_stream()'],['../namespacemlx_1_1core_1_1metal.html#a8b4188f9a090a1da42d62b8a369bf106',1,'mlx::core::metal::new_stream()'],['../namespacemlx_1_1core.html#a6f7c63a9be10337b3b96d527e1db3c2f',1,'mlx::core::new_stream()']]], + ['next_22',['next',['../struct_quantized_block_loader.html#a674138ef7c43cc45586ea9f8fd6f6bd9',1,'QuantizedBlockLoader::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a3b71f379ff9baf39830c92f4f1ecde52',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a78d2b0098311a278be8394edbd5fc731',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#aae56c19bb562219770fec38e5666c6ce',1,'mlx::steel::Conv2DWeightBlockLoader::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af9ce1a767266664bea131a5437002c80',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a30b10bebde7f08b89d03bdd9ea0f48da',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::next()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3e5ee68ed0ee43f7e979dd4222f76a8c',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a11743cb1c108f42ccdc6e59204a5b3e8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::next()'],['../structmlx_1_1steel_1_1_block_loader.html#a6af21428f0e7c17b48ddedf4dd20a1e8',1,'mlx::steel::BlockLoader::next()'],['../classmlx_1_1core_1_1random_1_1_key_sequence.html#a4193c5eac3ef093a740d5305b25d3e18',1,'mlx::core::random::KeySequence::next()'],['../backend_2metal_2allocator_8h.html#ae704ab07eac590091daa5fc4aec7bddb',1,'next: allocator.h']]], + ['next_5fpower_5fof_5f2_23',['next_power_of_2',['../namespacemlx_1_1core.html#a685c0530e338aabc622325685846ce93',1,'mlx::core']]], + ['nextafter_24',['nextafter',['../namespacemetal.html#a9547fd7b09164931986f6db4813bd72d',1,'metal::nextafter()'],['../namespacemetal_1_1fast.html#a4583e8be04fc0bd475b97b0934604f23',1,'metal::fast::nextafter()'],['../namespacemetal_1_1precise.html#ad012ceeb55b77f1533749b351331e026',1,'metal::precise::nextafter()']]], + ['no_5ffuse_25',['no_fuse',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4ada8df7fd43da6073fec4fe5666b03dbb',1,'mlx::core']]], + ['no_5fsimplify_26',['no_simplify',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a8e5611dfddbae6e68624c59aa3e4e3e2',1,'mlx::core']]], + ['nodenamer_27',['NodeNamer',['../structmlx_1_1core_1_1_node_namer.html',1,'mlx::core']]], + ['nomask_5ft_28',['nomask_t',['../kernels_2gemv__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'nomask_t: gemv_masked.h'],['../steel__gemm__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'nomask_t: steel_gemm_masked.h']]], + ['none_29',['None',['../struct_none.html',1,'None'],['../classmlx_1_1core_1_1_scatter.html#a614d19af11dc30644b2b4941033b613ca87a6a1927de175b71d7d0b5c11b8665c',1,'mlx::core::Scatter::None']]], + ['norm_30',['norm',['../namespacemlx_1_1core_1_1linalg.html#aba765b8e95e9a1d33d31f727a185919d',1,'mlx::core::linalg::norm(const array &a, const double ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#acaa85b4146821c268abecec2422c02d2',1,'mlx::core::linalg::norm(const array &a, const double ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#af1ebe0c6dcba9a1c49b5e397dddf3264',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#ae8da67e4c6e073f93889f1051203cd9e',1,'mlx::core::linalg::norm(const array &a, const std::string &ord, int axis, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a229018071d5602e38d6248230f334a10',1,'mlx::core::linalg::norm(const array &a, const std::optional< std::vector< int > > &axis=std::nullopt, bool keepdims=false, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1linalg.html#a44250cff34238f01471fd61e76036f03',1,'mlx::core::linalg::norm(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['normal_31',['normal',['../namespacemlx_1_1core_1_1random.html#aa3977269b77f66c6397c7c72f873d17a',1,'mlx::core::random::normal(const std::vector< int > &shape, Dtype dtype, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#aa3516ace3ebc5e80ca73fb8f50df436f',1,'mlx::core::random::normal(const std::vector< int > &shape, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a75ecf88e36a4013ecda18a91cb60befb',1,'mlx::core::random::normal(const std::vector< int > &shape, const Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7723a27a0093a1f51ca31d26c1d061c8',1,'mlx::core::random::normal(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['normalize_5faxis_32',['normalize_axis',['../namespacemlx_1_1core.html#a0181b5d72bf3d34448dabc70f7ff858d',1,'mlx::core']]], + ['not_5fequal_33',['not_equal',['../group__ops.html#ga28f22c5d2c399eee53be7b3facc11103',1,'mlx::core']]], + ['notequal_34',['NotEqual',['../structmlx_1_1core_1_1detail_1_1_not_equal.html',1,'mlx::core::detail::NotEqual'],['../classmlx_1_1core_1_1_not_equal.html',1,'mlx::core::NotEqual'],['../struct_not_equal.html',1,'NotEqual'],['../classmlx_1_1core_1_1_not_equal.html#ac568397bd17b5d9f25ad1a0ebadedbb9',1,'mlx::core::NotEqual::NotEqual()']]], + ['notify_5fnew_5ftask_35',['notify_new_task',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#ae8aa34a9be8bc73508dd500000421173',1,'mlx::core::scheduler::Scheduler::notify_new_task()'],['../namespacemlx_1_1core_1_1scheduler.html#a6b7289e33cef665178fe614aac75c1b2',1,'mlx::core::scheduler::notify_new_task()']]], + ['notify_5ftask_5fcompletion_36',['notify_task_completion',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#abbb2b1c2f8bae2b9c7cc51db65f18a3b',1,'mlx::core::scheduler::Scheduler::notify_task_completion()'],['../namespacemlx_1_1core_1_1scheduler.html#a1d06ffdbab36790b78deb6e34adc737f',1,'mlx::core::scheduler::notify_task_completion()']]], + ['num_5fthreads_37',['num_threads',['../namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833',1,'pocketfft::detail::threading']]], + ['number_38',['number',['../structmlx_1_1core_1_1_dtype.html#ac091c39cbd6686ef69aa1e5a2425aa2dab1bc248a7ff2b2e95569f56de68615df',1,'mlx::core::Dtype::number'],['../namespacemlx_1_1core.html#a069c0aab6b36aef34419534ec4a4310d',1,'mlx::core::number']]], + ['number_5fof_5felements_39',['number_of_elements',['../group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636',1,'mlx::core']]], + ['numberofelements_40',['NumberOfElements',['../classmlx_1_1core_1_1_number_of_elements.html',1,'mlx::core::NumberOfElements'],['../classmlx_1_1core_1_1_number_of_elements.html#ac64d7c40ae29d687f8b7d2fa33e13b06',1,'mlx::core::NumberOfElements::NumberOfElements()']]] ]; diff --git a/docs/build/html/search/all_f.js b/docs/build/html/search/all_f.js index b944f8409c..54da314737 100644 --- a/docs/build/html/search/all_f.js +++ b/docs/build/html/search/all_f.js @@ -9,7 +9,7 @@ var searchData= ['op_6',['Op',['../classmlx_1_1core_1_1_bitwise_binary.html#a6f8b5d455d0c1770428a6bef1608f23d',1,'mlx::core::BitwiseBinary']]], ['op_7',['op',['../structmlx_1_1core_1_1_default_strided_reduce.html#ac871f55a7ddd205574974cb4492a240b',1,'mlx::core::DefaultStridedReduce::op'],['../structmlx_1_1core_1_1_default_contiguous_reduce.html#a1928f07db988715cc177999e386f4830',1,'mlx::core::DefaultContiguousReduce::op'],['../common_2binary_8h.html#a70228731d29946574b238d21fb4b360c',1,'op: binary.h']]], ['operations_8',['Core array operations',['../group__ops.html',1,'']]], - ['operator_20bool_9',['operator bool',['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant']]], + ['operator_20bool_9',['operator bool',['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant'],['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant']]], ['operator_20dtype_10',['operator Dtype',['../structmlx_1_1core_1_1_type_to_dtype.html#aefdd0fd6a5bbf0197a3996ccd4adea13',1,'mlx::core::TypeToDtype']]], ['operator_20float_11',['operator float',['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aaae72e5340ce91325f1925be36ba46cb',1,'mlx::core::_MLX_BFloat16::operator float()'],['../structmlx_1_1core_1_1complex128__t.html#a3e2faf180c0b785646a0e4296f709a5e',1,'mlx::core::complex128_t::operator float()'],['../structmlx_1_1core_1_1complex64__t.html#a90d224dd37308345086bb9cc882ef6fc',1,'mlx::core::complex64_t::operator float()'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a363de5054f3673bddc90293fc3c9bb99',1,'mlx::core::_MLX_Float16::operator float()']]], ['operator_20t_12',['operator T',['../struct___m_l_x___b_float16.html#aa7dfefdf0d15e102d2b8258c9ab01836',1,'_MLX_BFloat16::operator T() const thread'],['../struct___m_l_x___b_float16.html#a2546a8afa77e14ed5b3c5da79a281260',1,'_MLX_BFloat16::operator T() const threadgroup'],['../struct___m_l_x___b_float16.html#a1d523f87740fcb852db6ab57896c245a',1,'_MLX_BFloat16::operator T() const device'],['../struct___m_l_x___b_float16.html#a95acd29283024d7093a0bc58c9468a0a',1,'_MLX_BFloat16::operator T() const constant'],['../structcomplex64__t.html#a70e9b16031eeaff3baa601f400023fcd',1,'complex64_t::operator T() const thread'],['../structcomplex64__t.html#a4f3beea7ab6001189b782a74d1746b67',1,'complex64_t::operator T() const threadgroup'],['../structcomplex64__t.html#a9f4f7eca89ffe6c8d126a4145df6d9f2',1,'complex64_t::operator T() const device'],['../structcomplex64__t.html#ac33e2e5263fec76a4fb4418c6e1d8d14',1,'complex64_t::operator T() const constant']]], diff --git a/docs/build/html/search/classes_1.js b/docs/build/html/search/classes_1.js index 8ef031a700..f2ad7365e2 100644 --- a/docs/build/html/search/classes_1.js +++ b/docs/build/html/search/classes_1.js @@ -6,29 +6,30 @@ var searchData= ['add_5fvec_3',['add_vec',['../structpocketfft_1_1detail_1_1add__vec.html',1,'pocketfft::detail']]], ['add_5fvec_3c_20cmplx_3c_20t_20_3e_20_3e_4',['add_vec< cmplx< T > >',['../structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html',1,'pocketfft::detail']]], ['addmm_5',['AddMM',['../classmlx_1_1core_1_1_add_m_m.html',1,'mlx::core']]], - ['aligned_5fallocator_6',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html',1,'pocketfft::detail::threading']]], - ['allgather_7',['AllGather',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html',1,'mlx::core::distributed']]], - ['allocator_8',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html',1,'mlx::core::allocator']]], - ['allreduce_9',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html',1,'mlx::core::distributed']]], - ['and_10',['And',['../struct_and.html',1,'']]], - ['arange_11',['Arange',['../classmlx_1_1core_1_1_arange.html',1,'mlx::core']]], - ['arccos_12',['ArcCos',['../struct_arc_cos.html',1,'ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html',1,'mlx::core::ArcCos'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html',1,'mlx::core::detail::ArcCos']]], - ['arccosh_13',['ArcCosh',['../struct_arc_cosh.html',1,'ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html',1,'mlx::core::ArcCosh'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html',1,'mlx::core::detail::ArcCosh']]], - ['arcsin_14',['ArcSin',['../struct_arc_sin.html',1,'ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html',1,'mlx::core::ArcSin'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html',1,'mlx::core::detail::ArcSin']]], - ['arcsinh_15',['ArcSinh',['../struct_arc_sinh.html',1,'ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html',1,'mlx::core::ArcSinh'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html',1,'mlx::core::detail::ArcSinh']]], - ['arctan_16',['ArcTan',['../struct_arc_tan.html',1,'ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html',1,'mlx::core::ArcTan'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html',1,'mlx::core::detail::ArcTan']]], - ['arctan2_17',['ArcTan2',['../struct_arc_tan2.html',1,'ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html',1,'mlx::core::ArcTan2'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html',1,'mlx::core::detail::ArcTan2']]], - ['arctanh_18',['ArcTanh',['../struct_arc_tanh.html',1,'ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html',1,'mlx::core::ArcTanh'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html',1,'mlx::core::detail::ArcTanh']]], - ['argpartition_19',['ArgPartition',['../classmlx_1_1core_1_1_arg_partition.html',1,'mlx::core']]], - ['argreduce_20',['ArgReduce',['../classmlx_1_1core_1_1_arg_reduce.html',1,'mlx::core']]], - ['argsort_21',['ArgSort',['../classmlx_1_1core_1_1_arg_sort.html',1,'mlx::core']]], - ['arr_22',['arr',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20t0_20_3e_20_3e_23',['arr< pocketfft::detail::cmplx< T0 > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20thigh_20_3e_20_3e_24',['arr< pocketfft::detail::cmplx< Thigh > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_3c_20t0_20_3e_25',['arr< T0 >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], - ['arr_5finfo_26',['arr_info',['../classpocketfft_1_1detail_1_1arr__info.html',1,'pocketfft::detail']]], - ['array_27',['array',['../classmlx_1_1core_1_1array.html',1,'mlx::core']]], - ['arrayiterator_28',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html',1,'mlx::core::array']]], - ['asstrided_29',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html',1,'mlx::core']]], - ['astype_30',['AsType',['../classmlx_1_1core_1_1_as_type.html',1,'mlx::core']]] + ['affinequantize_6',['AffineQuantize',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html',1,'mlx::core::fast']]], + ['aligned_5fallocator_7',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html',1,'pocketfft::detail::threading']]], + ['allgather_8',['AllGather',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html',1,'mlx::core::distributed']]], + ['allocator_9',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html',1,'mlx::core::allocator']]], + ['allreduce_10',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html',1,'mlx::core::distributed']]], + ['and_11',['And',['../struct_and.html',1,'']]], + ['arange_12',['Arange',['../classmlx_1_1core_1_1_arange.html',1,'mlx::core']]], + ['arccos_13',['ArcCos',['../struct_arc_cos.html',1,'ArcCos'],['../classmlx_1_1core_1_1_arc_cos.html',1,'mlx::core::ArcCos'],['../structmlx_1_1core_1_1detail_1_1_arc_cos.html',1,'mlx::core::detail::ArcCos']]], + ['arccosh_14',['ArcCosh',['../struct_arc_cosh.html',1,'ArcCosh'],['../classmlx_1_1core_1_1_arc_cosh.html',1,'mlx::core::ArcCosh'],['../structmlx_1_1core_1_1detail_1_1_arc_cosh.html',1,'mlx::core::detail::ArcCosh']]], + ['arcsin_15',['ArcSin',['../struct_arc_sin.html',1,'ArcSin'],['../classmlx_1_1core_1_1_arc_sin.html',1,'mlx::core::ArcSin'],['../structmlx_1_1core_1_1detail_1_1_arc_sin.html',1,'mlx::core::detail::ArcSin']]], + ['arcsinh_16',['ArcSinh',['../struct_arc_sinh.html',1,'ArcSinh'],['../classmlx_1_1core_1_1_arc_sinh.html',1,'mlx::core::ArcSinh'],['../structmlx_1_1core_1_1detail_1_1_arc_sinh.html',1,'mlx::core::detail::ArcSinh']]], + ['arctan_17',['ArcTan',['../struct_arc_tan.html',1,'ArcTan'],['../classmlx_1_1core_1_1_arc_tan.html',1,'mlx::core::ArcTan'],['../structmlx_1_1core_1_1detail_1_1_arc_tan.html',1,'mlx::core::detail::ArcTan']]], + ['arctan2_18',['ArcTan2',['../struct_arc_tan2.html',1,'ArcTan2'],['../classmlx_1_1core_1_1_arc_tan2.html',1,'mlx::core::ArcTan2'],['../structmlx_1_1core_1_1detail_1_1_arc_tan2.html',1,'mlx::core::detail::ArcTan2']]], + ['arctanh_19',['ArcTanh',['../struct_arc_tanh.html',1,'ArcTanh'],['../classmlx_1_1core_1_1_arc_tanh.html',1,'mlx::core::ArcTanh'],['../structmlx_1_1core_1_1detail_1_1_arc_tanh.html',1,'mlx::core::detail::ArcTanh']]], + ['argpartition_20',['ArgPartition',['../classmlx_1_1core_1_1_arg_partition.html',1,'mlx::core']]], + ['argreduce_21',['ArgReduce',['../classmlx_1_1core_1_1_arg_reduce.html',1,'mlx::core']]], + ['argsort_22',['ArgSort',['../classmlx_1_1core_1_1_arg_sort.html',1,'mlx::core']]], + ['arr_23',['arr',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20t0_20_3e_20_3e_24',['arr< pocketfft::detail::cmplx< T0 > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_3c_20pocketfft_3a_3adetail_3a_3acmplx_3c_20thigh_20_3e_20_3e_25',['arr< pocketfft::detail::cmplx< Thigh > >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_3c_20t0_20_3e_26',['arr< T0 >',['../classpocketfft_1_1detail_1_1arr.html',1,'pocketfft::detail']]], + ['arr_5finfo_27',['arr_info',['../classpocketfft_1_1detail_1_1arr__info.html',1,'pocketfft::detail']]], + ['array_28',['array',['../classmlx_1_1core_1_1array.html',1,'mlx::core']]], + ['arrayiterator_29',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html',1,'mlx::core::array']]], + ['asstrided_30',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html',1,'mlx::core']]], + ['astype_31',['AsType',['../classmlx_1_1core_1_1_as_type.html',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/classes_15.js b/docs/build/html/search/classes_15.js index f7e2cd36bc..e6aaa08045 100644 --- a/docs/build/html/search/classes_15.js +++ b/docs/build/html/search/classes_15.js @@ -1,8 +1,6 @@ var searchData= [ - ['vector_0',['Vector',['../class_m_p_s_1_1_vector.html',1,'MPS']]], - ['vectordescriptor_1',['VectorDescriptor',['../class_m_p_s_1_1_vector_descriptor.html',1,'MPS']]], - ['view_2',['View',['../classmlx_1_1core_1_1_view.html',1,'mlx::core']]], - ['vlen_3',['VLEN',['../structpocketfft_1_1detail_1_1_v_l_e_n.html',1,'pocketfft::detail']]], - ['vtype_4',['VTYPE',['../structpocketfft_1_1detail_1_1_v_t_y_p_e.html',1,'pocketfft::detail']]] + ['view_0',['View',['../classmlx_1_1core_1_1_view.html',1,'mlx::core']]], + ['vlen_1',['VLEN',['../structpocketfft_1_1detail_1_1_v_l_e_n.html',1,'pocketfft::detail']]], + ['vtype_2',['VTYPE',['../structpocketfft_1_1detail_1_1_v_t_y_p_e.html',1,'pocketfft::detail']]] ]; diff --git a/docs/build/html/search/classes_7.js b/docs/build/html/search/classes_7.js index f28255f112..c9171a2275 100644 --- a/docs/build/html/search/classes_7.js +++ b/docs/build/html/search/classes_7.js @@ -7,7 +7,9 @@ var searchData= ['gemmkernel_4',['GEMMKernel',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html',1,'mlx::steel']]], ['gemmparams_5',['GEMMParams',['../structmlx_1_1steel_1_1_g_e_m_m_params.html',1,'mlx::steel']]], ['gemmspiltkparams_6',['GEMMSpiltKParams',['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html',1,'mlx::steel']]], - ['greater_7',['Greater',['../struct_greater.html',1,'Greater'],['../structmlx_1_1core_1_1detail_1_1_greater.html',1,'mlx::core::detail::Greater'],['../classmlx_1_1core_1_1_greater.html',1,'mlx::core::Greater']]], - ['greaterequal_8',['GreaterEqual',['../struct_greater_equal.html',1,'GreaterEqual'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html',1,'mlx::core::detail::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html',1,'mlx::core::GreaterEqual']]], - ['group_9',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html',1,'mlx::core::distributed']]] + ['gemvkernel_7',['GEMVKernel',['../struct_g_e_m_v_kernel.html',1,'']]], + ['gemvtkernel_8',['GEMVTKernel',['../struct_g_e_m_v_t_kernel.html',1,'']]], + ['greater_9',['Greater',['../struct_greater.html',1,'Greater'],['../structmlx_1_1core_1_1detail_1_1_greater.html',1,'mlx::core::detail::Greater'],['../classmlx_1_1core_1_1_greater.html',1,'mlx::core::Greater']]], + ['greaterequal_10',['GreaterEqual',['../struct_greater_equal.html',1,'GreaterEqual'],['../structmlx_1_1core_1_1detail_1_1_greater_equal.html',1,'mlx::core::detail::GreaterEqual'],['../classmlx_1_1core_1_1_greater_equal.html',1,'mlx::core::GreaterEqual']]], + ['group_11',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html',1,'mlx::core::distributed']]] ]; diff --git a/docs/build/html/search/classes_a.js b/docs/build/html/search/classes_a.js index 578cecfeaa..68da9be646 100644 --- a/docs/build/html/search/classes_a.js +++ b/docs/build/html/search/classes_a.js @@ -1,7 +1,6 @@ var searchData= [ - ['kernel_0',['Kernel',['../class_m_p_s_1_1_kernel.html',1,'MPS']]], - ['kernelmergesort_1',['KernelMergeSort',['../struct_kernel_merge_sort.html',1,'']]], - ['kernelmultiblockmergesort_2',['KernelMultiBlockMergeSort',['../struct_kernel_multi_block_merge_sort.html',1,'']]], - ['keysequence_3',['KeySequence',['../classmlx_1_1core_1_1random_1_1_key_sequence.html',1,'mlx::core::random']]] + ['kernelmergesort_0',['KernelMergeSort',['../struct_kernel_merge_sort.html',1,'']]], + ['kernelmultiblockmergesort_1',['KernelMultiBlockMergeSort',['../struct_kernel_multi_block_merge_sort.html',1,'']]], + ['keysequence_2',['KeySequence',['../classmlx_1_1core_1_1random_1_1_key_sequence.html',1,'mlx::core::random']]] ]; diff --git a/docs/build/html/search/classes_c.js b/docs/build/html/search/classes_c.js index 9c5ed3a2e8..55bf64bac4 100644 --- a/docs/build/html/search/classes_c.js +++ b/docs/build/html/search/classes_c.js @@ -1,21 +1,17 @@ var searchData= [ ['matmul_0',['Matmul',['../classmlx_1_1core_1_1_matmul.html',1,'mlx::core']]], - ['matrix_1',['Matrix',['../class_m_p_s_1_1_matrix.html',1,'MPS']]], - ['matrixdescriptor_2',['MatrixDescriptor',['../class_m_p_s_1_1_matrix_descriptor.html',1,'MPS']]], - ['matrixmultiplication_3',['MatrixMultiplication',['../class_m_p_s_1_1_matrix_multiplication.html',1,'MPS']]], - ['matrixvectormultiplication_4',['MatrixVectorMultiplication',['../class_m_p_s_1_1_matrix_vector_multiplication.html',1,'MPS']]], - ['max_5',['Max',['../struct_max.html',1,'']]], - ['maximum_6',['Maximum',['../struct_maximum.html',1,'Maximum'],['../structmlx_1_1core_1_1detail_1_1_maximum.html',1,'mlx::core::detail::Maximum'],['../classmlx_1_1core_1_1_maximum.html',1,'mlx::core::Maximum']]], - ['metalallocator_7',['MetalAllocator',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html',1,'mlx::core::metal']]], - ['min_8',['Min',['../struct_min.html',1,'']]], - ['minimum_9',['Minimum',['../struct_minimum.html',1,'Minimum'],['../structmlx_1_1core_1_1detail_1_1_minimum.html',1,'mlx::core::detail::Minimum'],['../classmlx_1_1core_1_1_minimum.html',1,'mlx::core::Minimum']]], - ['mlx_5fatomic_10',['mlx_atomic',['../structmlx__atomic.html',1,'']]], - ['mlx_5fatomic_3c_20t_2c_20enable_5fif_5ft_3c_20is_5fmetal_5fatomic_3c_20t_20_3e_20_3e_20_3e_11',['mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >',['../structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html',1,'']]], - ['mlxconvparams_12',['MLXConvParams',['../struct_m_l_x_conv_params.html',1,'']]], - ['mlxconvparams_3c_202_20_3e_13',['MLXConvParams< 2 >',['../struct_m_l_x_conv_params.html',1,'']]], - ['mlxfastattentionparams_14',['MLXFastAttentionParams',['../struct_m_l_x_fast_attention_params.html',1,'']]], - ['mlxscaleddotproductattentionparams_15',['MLXScaledDotProductAttentionParams',['../struct_m_l_x_scaled_dot_product_attention_params.html',1,'']]], - ['multi_5fiter_16',['multi_iter',['../classpocketfft_1_1detail_1_1multi__iter.html',1,'pocketfft::detail']]], - ['multiply_17',['Multiply',['../structmlx_1_1core_1_1detail_1_1_multiply.html',1,'mlx::core::detail::Multiply'],['../classmlx_1_1core_1_1_multiply.html',1,'mlx::core::Multiply'],['../struct_multiply.html',1,'Multiply']]] + ['max_1',['Max',['../struct_max.html',1,'']]], + ['maximum_2',['Maximum',['../struct_maximum.html',1,'Maximum'],['../structmlx_1_1core_1_1detail_1_1_maximum.html',1,'mlx::core::detail::Maximum'],['../classmlx_1_1core_1_1_maximum.html',1,'mlx::core::Maximum']]], + ['metalallocator_3',['MetalAllocator',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html',1,'mlx::core::metal']]], + ['min_4',['Min',['../struct_min.html',1,'']]], + ['minimum_5',['Minimum',['../struct_minimum.html',1,'Minimum'],['../structmlx_1_1core_1_1detail_1_1_minimum.html',1,'mlx::core::detail::Minimum'],['../classmlx_1_1core_1_1_minimum.html',1,'mlx::core::Minimum']]], + ['mlx_5fatomic_6',['mlx_atomic',['../structmlx__atomic.html',1,'']]], + ['mlx_5fatomic_3c_20t_2c_20enable_5fif_5ft_3c_20is_5fmetal_5fatomic_3c_20t_20_3e_20_3e_20_3e_7',['mlx_atomic< T, enable_if_t< is_metal_atomic< T > > >',['../structmlx__atomic_3_01_t_00_01enable__if__t_3_01is__metal__atomic_3_01_t_01_4_01_4_01_4.html',1,'']]], + ['mlxconvparams_8',['MLXConvParams',['../struct_m_l_x_conv_params.html',1,'']]], + ['mlxconvparams_3c_202_20_3e_9',['MLXConvParams< 2 >',['../struct_m_l_x_conv_params.html',1,'']]], + ['mlxfastattentionparams_10',['MLXFastAttentionParams',['../struct_m_l_x_fast_attention_params.html',1,'']]], + ['mlxscaleddotproductattentionparams_11',['MLXScaledDotProductAttentionParams',['../struct_m_l_x_scaled_dot_product_attention_params.html',1,'']]], + ['multi_5fiter_12',['multi_iter',['../classpocketfft_1_1detail_1_1multi__iter.html',1,'pocketfft::detail']]], + ['multiply_13',['Multiply',['../structmlx_1_1core_1_1detail_1_1_multiply.html',1,'mlx::core::detail::Multiply'],['../classmlx_1_1core_1_1_multiply.html',1,'mlx::core::Multiply'],['../struct_multiply.html',1,'Multiply']]] ]; diff --git a/docs/build/html/search/defines_0.js b/docs/build/html/search/defines_0.js index 0753f823cf..9812778117 100644 --- a/docs/build/html/search/defines_0.js +++ b/docs/build/html/search/defines_0.js @@ -1,7 +1,5 @@ var searchData= [ ['_5f_5fmlx_5fbfloat_5fnan_5f_5f_0',['__MLX_BFLOAT_NAN__',['../types_2bf16_8h.html#a78cafe726142a0166894e0c0a70b19f6',1,'bf16.h']]], - ['_5f_5fmlx_5fhalf_5fnan_5f_5f_1',['__MLX_HALF_NAN__',['../fp16_8h.html#a10abf57a099efdbb9db0c78e9c120e50',1,'fp16.h']]], - ['_5fmps_5fprivate_5fcls_2',['_MPS_PRIVATE_CLS',['../mps_2gemm_8h.html#a2986cabc5a9ec53428151949c0369602',1,'gemm.h']]], - ['_5fmps_5fprivate_5fsel_3',['_MPS_PRIVATE_SEL',['../mps_2gemm_8h.html#a10e733259894b7fddd11f46b9eabfe5f',1,'gemm.h']]] + ['_5f_5fmlx_5fhalf_5fnan_5f_5f_1',['__MLX_HALF_NAN__',['../fp16_8h.html#a10abf57a099efdbb9db0c78e9c120e50',1,'fp16.h']]] ]; diff --git a/docs/build/html/search/defines_8.js b/docs/build/html/search/defines_8.js index 175b93dec1..79205dfbe6 100644 --- a/docs/build/html/search/defines_8.js +++ b/docs/build/html/search/defines_8.js @@ -3,8 +3,8 @@ var searchData= ['max_5foutput_5fsize_0',['MAX_OUTPUT_SIZE',['../backend_2metal_2kernels_2fft_8h.html#a28d683cf067736d76f867f30c066317e',1,'fft.h']]], ['max_5fradix_1',['MAX_RADIX',['../backend_2metal_2kernels_2fft_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: fft.h'],['../readwrite_8h.html#a7b6e56afa21f022c5e754b000955735a',1,'MAX_RADIX: readwrite.h']]], ['mlx_5flapack_5ffunc_2',['MLX_LAPACK_FUNC',['../lapack__helper_8h.html#ae22db9704827bf013a0a61f21a47464b',1,'lapack_helper.h']]], - ['mlx_5fmtl_5fconst_3',['MLX_MTL_CONST',['../quantized_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: quantized.h'],['../sort_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: sort.h']]], + ['mlx_5fmtl_5fconst_3',['MLX_MTL_CONST',['../kernels_2gemv__masked_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: gemv_masked.h'],['../quantized_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: quantized.h'],['../sort_8h.html#a0386011c52d03e60885a31e6fbd903dd',1,'MLX_MTL_CONST: sort.h']]], ['mlx_5fmtl_5floop_5funroll_4',['MLX_MTL_LOOP_UNROLL',['../sort_8h.html#ad34b622323cebef136669fedd7229515',1,'sort.h']]], - ['mlx_5fmtl_5fpragma_5funroll_5',['MLX_MTL_PRAGMA_UNROLL',['../backend_2metal_2kernels_2utils_8h.html#a069b682d7d21827461544817d722bfd3',1,'utils.h']]], + ['mlx_5fmtl_5fpragma_5funroll_5',['MLX_MTL_PRAGMA_UNROLL',['../kernels_2gemv__masked_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: gemv_masked.h'],['../backend_2metal_2kernels_2utils_8h.html#a069b682d7d21827461544817d722bfd3',1,'MLX_MTL_PRAGMA_UNROLL: utils.h']]], ['mtl_5fconst_6',['MTL_CONST',['../defines_8h.html#a767ed9f2604de22b259cee02c4ce1d22',1,'defines.h']]] ]; diff --git a/docs/build/html/search/enums_2.js b/docs/build/html/search/enums_2.js index 6510e99b1b..84a072b313 100644 --- a/docs/build/html/search/enums_2.js +++ b/docs/build/html/search/enums_2.js @@ -1,5 +1,4 @@ var searchData= [ - ['datatype_0',['DataType',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288',1,'MPS']]], - ['devicetype_1',['DeviceType',['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdb',1,'mlx::core::Device']]] + ['devicetype_0',['DeviceType',['../structmlx_1_1core_1_1_device.html#ac45b3de9b3458d8f31005136cde20fdb',1,'mlx::core::Device']]] ]; diff --git a/docs/build/html/search/enumvalues_3.js b/docs/build/html/search/enumvalues_3.js index 437fc22c07..e83d149540 100644 --- a/docs/build/html/search/enumvalues_3.js +++ b/docs/build/html/search/enumvalues_3.js @@ -1,9 +1,4 @@ var searchData= [ - ['datatypealternateencodingbit_0',['DataTypeAlternateEncodingBit',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a9681e9100f89bbc0484c4195ff1cef03',1,'MPS']]], - ['datatypebfloat16_1',['DataTypeBFloat16',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a39c004722f824f460d83cca680b242ca',1,'MPS']]], - ['datatypefloat16_2',['DataTypeFloat16',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a4fb70692e99c4ff2dc6028275e5a3868',1,'MPS']]], - ['datatypefloat32_3',['DataTypeFloat32',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a2156c04e979b8d8bd39d4ae17d53ded8',1,'MPS']]], - ['datatypefloatbit_4',['DataTypeFloatBit',['../namespace_m_p_s.html#a89deea53e8654b369b5cde55bdcd1288a330c9014eb076d89dae3d69f3b49c04a',1,'MPS']]], - ['disabled_5',['disabled',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a075ae3d2fc31640504f814f60e5ef713',1,'mlx::core']]] + ['disabled_0',['disabled',['../namespacemlx_1_1core.html#adb15ff2b1ca5207fd4f6e631e2c3bcb4a075ae3d2fc31640504f814f60e5ef713',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/files_6.js b/docs/build/html/search/files_6.js index 735f6f9c43..d246684e7e 100644 --- a/docs/build/html/search/files_6.js +++ b/docs/build/html/search/files_6.js @@ -1,7 +1,8 @@ var searchData= [ ['gather_2eh_0',['gather.h',['../gather_8h.html',1,'']]], - ['gemm_2eh_1',['gemm.h',['../kernels_2steel_2gemm_2gemm_8h.html',1,'(Global Namespace)'],['../mps_2gemm_8h.html',1,'(Global Namespace)']]], - ['gguf_2eh_2',['gguf.h',['../gguf_8h.html',1,'']]], - ['graph_5futils_2eh_3',['graph_utils.h',['../graph__utils_8h.html',1,'']]] + ['gemm_2eh_1',['gemm.h',['../gemm_8h.html',1,'']]], + ['gemv_5fmasked_2eh_2',['gemv_masked.h',['../jit_2gemv__masked_8h.html',1,'(Global Namespace)'],['../kernels_2gemv__masked_8h.html',1,'(Global Namespace)']]], + ['gguf_2eh_3',['gguf.h',['../gguf_8h.html',1,'']]], + ['graph_5futils_2eh_4',['graph_utils.h',['../graph__utils_8h.html',1,'']]] ]; diff --git a/docs/build/html/search/functions_0.js b/docs/build/html/search/functions_0.js index f839c40355..7d740542cc 100644 --- a/docs/build/html/search/functions_0.js +++ b/docs/build/html/search/functions_0.js @@ -2,7 +2,5 @@ var searchData= [ ['_5fcontiguous_5fstrided_5freduce_0',['_contiguous_strided_reduce',['../reduce__col_8h.html#a6d72f6a88a37d8e031d0ac33f26ecbb4',1,'reduce_col.h']]], ['_5fmlx_5fbfloat16_1',['_MLX_BFloat16',['../struct___m_l_x___b_float16.html#ab1af7700f5d1e4ab567da6a34fa84668',1,'_MLX_BFloat16::_MLX_BFloat16() thread=default'],['../struct___m_l_x___b_float16.html#ae5c51644c3bd7cda6b796cb63c60c0b4',1,'_MLX_BFloat16::_MLX_BFloat16() threadgroup=default'],['../struct___m_l_x___b_float16.html#a21998a3c852d0e0f52681f8b453172bf',1,'_MLX_BFloat16::_MLX_BFloat16() device=default'],['../struct___m_l_x___b_float16.html#a64d8fc2e2463d7fa19cd3d5dd1ffdae8',1,'_MLX_BFloat16::_MLX_BFloat16() const ant=default'],['../struct___m_l_x___b_float16.html#a50d825f05a162d0ac133ad8b6f3c3112',1,'_MLX_BFloat16::_MLX_BFloat16(uint16_t bits, bits_to_bfloat_struct)'],['../struct___m_l_x___b_float16.html#a24c5736f234e09a0c82b00c7e44cc547',1,'_MLX_BFloat16::_MLX_BFloat16(T x) thread'],['../struct___m_l_x___b_float16.html#adeb880f31121c6dc40ce47765c6c7455',1,'_MLX_BFloat16::_MLX_BFloat16(T x) threadgroup'],['../struct___m_l_x___b_float16.html#ad2701d003e8fad168c89abc3907c6e53',1,'_MLX_BFloat16::_MLX_BFloat16(T x) device'],['../struct___m_l_x___b_float16.html#aec7fa716fd621ce1843338027bcb0118',1,'_MLX_BFloat16::_MLX_BFloat16(T x) const ant'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#a2c81f14fea4c01255a191f2146515917',1,'mlx::core::_MLX_BFloat16::_MLX_BFloat16()=default'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#ad43561d38ca00f9c37e8b130220233c0',1,'mlx::core::_MLX_BFloat16::_MLX_BFloat16(_MLX_BFloat16 const &)=default'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aedbead2d935a12e8d5a4ff6269ba9ab2',1,'mlx::core::_MLX_BFloat16::_MLX_BFloat16(const float &x)']]], - ['_5fmlx_5ffloat16_2',['_MLX_Float16',['../structmlx_1_1core_1_1___m_l_x___float16.html#a69a4ab5b456c4f3b786f43632e9a4fbc',1,'mlx::core::_MLX_Float16::_MLX_Float16()=default'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a0d7ed78b78c6d446f220f83b0cdebb86',1,'mlx::core::_MLX_Float16::_MLX_Float16(_MLX_Float16 const &)=default'],['../structmlx_1_1core_1_1___m_l_x___float16.html#afde284cbe678e0333ae277ffc8b131c0',1,'mlx::core::_MLX_Float16::_MLX_Float16(const float &x)']]], - ['_5fmtl_5fprivate_5fdef_5fcls_3',['_MTL_PRIVATE_DEF_CLS',['../namespace_m_t_l_1_1_private_1_1_class.html#af6fc5dd227064897589f56d8912a8fd6',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSMatrixDescriptor)'],['../namespace_m_t_l_1_1_private_1_1_class.html#a23ba07d1dd78ce20ecba16d88faba5e2',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSMatrix)'],['../namespace_m_t_l_1_1_private_1_1_class.html#aa94ef8e094d91ad36af140823c15b59d',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSVectorDescriptor)'],['../namespace_m_t_l_1_1_private_1_1_class.html#aeff20cff997534929deb18ce0b7e947f',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSVector)'],['../namespace_m_t_l_1_1_private_1_1_class.html#a48f22abaf491778683a68e7db28839c9',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSKernel)'],['../namespace_m_t_l_1_1_private_1_1_class.html#a24a734c27edbaf99bbb3c3741fe98688',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSMatrixMultiplication)'],['../namespace_m_t_l_1_1_private_1_1_class.html#ac162f79565a61ed1f2316ca1a38c774d',1,'MTL::Private::Class::_MTL_PRIVATE_DEF_CLS(MPSMatrixVectorMultiplication)']]], - ['_5fmtl_5fprivate_5fdef_5fsel_4',['_MTL_PRIVATE_DEF_SEL',['../namespace_m_t_l_1_1_private_1_1_selector.html#acc50646a91989ebaca3cf945f42f2b62',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(matrixDescriptorWithRows_columns_rowBytes_dataType, "matrixDescriptorWithRows:columns:rowBytes:dataType:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a11464ac9c57b12981683bf25afc6ae77',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(matrixDescriptorWithRows_columns_matrices_rowBytes_matrixBytes_dataType, "matrixDescriptorWithRows:columns:matrices:rowBytes:matrixBytes:dataType:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a30c8b5bf66838e00c44d4cef997bee78',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(rows, "rows")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a5628c8d22de5d0339cdfd37cb8e103b6',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(initWithBuffer_descriptor, "initWithBuffer:descriptor:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#ae842530803a52e75c546db1f9424f933',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(initWithDevice_, "initWithDevice:transposeLeft:transposeRight:" "resultRows:resultColumns:interiorColumns:alpha:beta:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a8631d028899e57b928d1240c4cdda929',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(encodeToCommandBuffer_leftMatrix_rightMatrix_resultMatrix, "encodeToCommandBuffer:leftMatrix:rightMatrix:resultMatrix:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a071584fae1458eba0ed41e02e640d34b',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setLeftMatrixOrigin_, "setLeftMatrixOrigin:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a8534190f99aac6ab3a01c0f6965e8882',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setRightMatrixOrigin_, "setRightMatrixOrigin:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a8b1a7f7c97c62753b3cb7447d5aa4798',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setResultMatrixOrigin_, "setResultMatrixOrigin:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#af84065144e91d5fb295e23b42f3be8cc',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setBatchStart_, "setBatchStart:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#ac13dc3039959f5f5d970f02cbc89c16d',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(setBatchSize_, "setBatchSize:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a46fcb509b385834113fe0b0f27ec768b',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(vectorDescriptorWithLength_dataType, "vectorDescriptorWithLength:dataType:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#a4c7a7cb16b14dd691e1211913572c6a8',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(vectorDescriptorWithLength_vectors_vectorBytes_dataType, "vectorDescriptorWithLength:vectors:vectorBytes:dataType:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#ab84cb9467edcc1a5ce32f001cb46c656',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(initWithDevice_transpose_rows_columns_alpha_beta, "initWithDevice:transpose:rows:columns:alpha:beta:")'],['../namespace_m_t_l_1_1_private_1_1_selector.html#aec59058ff2c151d67647ad2680074090',1,'MTL::Private::Selector::_MTL_PRIVATE_DEF_SEL(encodeToCommandBuffer_inputMatrix_inputVector_resultVector, "encodeToCommandBuffer:inputMatrix:inputVector:resultVector:")']]] + ['_5fmlx_5ffloat16_2',['_MLX_Float16',['../structmlx_1_1core_1_1___m_l_x___float16.html#a69a4ab5b456c4f3b786f43632e9a4fbc',1,'mlx::core::_MLX_Float16::_MLX_Float16()=default'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a0d7ed78b78c6d446f220f83b0cdebb86',1,'mlx::core::_MLX_Float16::_MLX_Float16(_MLX_Float16 const &)=default'],['../structmlx_1_1core_1_1___m_l_x___float16.html#afde284cbe678e0333ae277ffc8b131c0',1,'mlx::core::_MLX_Float16::_MLX_Float16(const float &x)']]] ]; diff --git a/docs/build/html/search/functions_1.js b/docs/build/html/search/functions_1.js index 5ca1d2c4be..38c204d0eb 100644 --- a/docs/build/html/search/functions_1.js +++ b/docs/build/html/search/functions_1.js @@ -10,70 +10,73 @@ var searchData= ['addmm_7',['addmm',['../group__ops.html#ga82a53e083205a965387b3c3e2463244a',1,'mlx::core']]], ['adjust_5fmatrix_5foffsets_8',['adjust_matrix_offsets',['../quantized_8h.html#a3ab400746ad77be89c30d25638e01698',1,'quantized.h']]], ['advance_9',['advance',['../classpocketfft_1_1detail_1_1multi__iter.html#a5ddcc0666125b3cb6c0d62b30befdd2c',1,'pocketfft::detail::multi_iter::advance()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a73a9ecd3008d2bd35aaa00bf9fac074f',1,'pocketfft::detail::simple_iter::advance()'],['../classpocketfft_1_1detail_1_1rev__iter.html#ad1918c84ae963188afc7599629b29686',1,'pocketfft::detail::rev_iter::advance()']]], - ['aligned_5falloc_10',['aligned_alloc',['../namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76',1,'pocketfft::detail']]], - ['aligned_5fallocator_11',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator(const aligned_allocator< U > &)'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator()=default']]], - ['aligned_5fdealloc_12',['aligned_dealloc',['../namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8',1,'pocketfft::detail']]], - ['all_13',['all',['../group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68',1,'mlx::core::all(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4',1,'mlx::core::all(const array &a, StreamOrDevice s={})'],['../group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59',1,'mlx::core::all(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc',1,'mlx::core::all(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['all_5fgather_14',['all_gather',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04',1,'mlx::core::distributed::detail::all_gather()'],['../namespacemlx_1_1core_1_1distributed.html#abf45499dd1209ba7162908f14634d2ba',1,'mlx::core::distributed::all_gather()']]], - ['all_5freduce_15',['all_reduce',['../reduce__all_8h.html#a8d143d2fea044e4808b69cc3723e33c7',1,'reduce_all.h']]], - ['all_5freduce_5fdispatch_16',['all_reduce_dispatch',['../namespacemlx_1_1core.html#a3ab0fd997d9a35782106ff083a72e098',1,'mlx::core']]], - ['all_5freduce_5fno_5fatomics_17',['all_reduce_no_atomics',['../reduce__all_8h.html#a09d140566cb3e6418a821be1e42db282',1,'reduce_all.h']]], - ['all_5fsum_18',['all_sum',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960',1,'mlx::core::distributed::detail::all_sum()'],['../namespacemlx_1_1core_1_1distributed.html#ad1c6aa544781765060552cacfa845001',1,'mlx::core::distributed::all_sum()']]], - ['allclose_19',['allclose',['../group__ops.html#gaf0cd4257de7542daf9faf5e605e31020',1,'mlx::core']]], - ['allgather_20',['AllGather',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ac9d0f1ba0d9602fa22e8f2b60f1231e0',1,'mlx::core::distributed::AllGather']]], - ['alloc_21',['alloc',['../class_m_p_s_1_1_matrix.html#a906d97b4b2365bec4acf278ef22205b4',1,'MPS::Matrix::alloc()'],['../class_m_p_s_1_1_matrix_multiplication.html#a9f1400b36672bd6f228a80982e5b2717',1,'MPS::MatrixMultiplication::alloc()'],['../class_m_p_s_1_1_vector.html#ac08919c16342247963b14766e65332a3',1,'MPS::Vector::alloc()'],['../class_m_p_s_1_1_matrix_vector_multiplication.html#aeca798759333aae7fdd594835e93d16a',1,'MPS::MatrixVectorMultiplication::alloc()']]], - ['alloc_5ftmp_22',['alloc_tmp',['../namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, size_t axsize, size_t elemsize)'],['../namespacepocketfft_1_1detail.html#a13832735696303b9559c4663631d5475',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, const shape_t &axes, size_t elemsize)']]], - ['allocate_23',['allocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826',1,'pocketfft::detail::threading::aligned_allocator']]], - ['allocator_24',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a5803678a418fef687fc65fa9d5c37b65',1,'mlx::core::allocator::Allocator::Allocator()=default'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#aa05c081ce80dc036f9d3dd8c195259d2',1,'mlx::core::allocator::Allocator::Allocator(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a8e8ce346a16cf0c62847bed9289f9959',1,'mlx::core::allocator::Allocator::Allocator(Allocator &&other)=delete']]], - ['allocator_25',['allocator',['../namespacemlx_1_1core_1_1allocator.html#aa23e2f20a336d0b159c097087194634e',1,'mlx::core::allocator::allocator()'],['../namespacemlx_1_1core_1_1metal.html#a74b3558bd518aecde6b14b0ba5e1a0d5',1,'mlx::core::metal::allocator()']]], - ['allreduce_26',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a936b320f7b0a435f96e018c25f35e7ed',1,'mlx::core::distributed::AllReduce']]], - ['any_27',['any',['../group__ops.html#ga8598dd718fb05cb28535e250372d4e6f',1,'mlx::core::any(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gad37df97f253a963bece124198dbaf9ba',1,'mlx::core::any(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef',1,'mlx::core::any(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce',1,'mlx::core::any(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['apply_28',['apply',['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply()'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const']]], - ['apply_5fepilogue_29',['apply_epilogue',['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)']]], - ['apply_5fepilogue_5fsafe_30',['apply_epilogue_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA']]], - ['apply_5finplace_5fop_31',['apply_inplace_op',['../structmlx_1_1steel_1_1_block_loader.html#adb4ca2cc193630a779de552fa8847ddf',1,'mlx::steel::BlockLoader']]], - ['arange_32',['Arange',['../classmlx_1_1core_1_1_arange.html#a1a70c3b0b9c67d5a9446c141c5b7c574',1,'mlx::core::Arange']]], - ['arange_33',['arange',['../namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b',1,'mlx::core::arange()'],['../namespacemlx_1_1core_1_1metal.html#a272c36f0faf2570cbb2f36030e9a3f26',1,'mlx::core::metal::arange()'],['../metal_2kernels_2arange_8h.html#a1e5126ee6ae0164c2343230c4d87c03e',1,'arange(): arange.h'],['../group__ops.html#ga7ca088b8090b9f84f2e08345cf3f835a',1,'mlx::core::arange(double start, double stop, double step, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga4c36b841dc5cba391dad029be5a0ad98',1,'mlx::core::arange(double start, double stop, double step, StreamOrDevice s={})'],['../group__ops.html#ga8d7cf9eb15e2daf1469058907e8abc85',1,'mlx::core::arange(double start, double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga74566a14e69ba6a25f5a35e7ade5c282',1,'mlx::core::arange(double start, double stop, StreamOrDevice s={})'],['../group__ops.html#ga345aa27af3dae3646b8b4b1068e89a3e',1,'mlx::core::arange(double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaae179075d0fe23f4bd53fdf8c41f4c70',1,'mlx::core::arange(double stop, StreamOrDevice s={})'],['../group__ops.html#ga6b945f513077c2978afc1a952c884860',1,'mlx::core::arange(int start, int stop, int step, StreamOrDevice s={})'],['../group__ops.html#ga1c39fcc6eaa1c1867735c7f849d708d6',1,'mlx::core::arange(int start, int stop, StreamOrDevice s={})'],['../group__ops.html#gafe6e4580452c873cac294f16129e633f',1,'mlx::core::arange(int stop, StreamOrDevice s={})']]], - ['arccos_34',['ArcCos',['../classmlx_1_1core_1_1_arc_cos.html#a66f4ee841d17923d93241b71ea5103e9',1,'mlx::core::ArcCos']]], - ['arccos_35',['arccos',['../group__ops.html#ga08bec7cb10c84466487b507fc5bf9776',1,'mlx::core']]], - ['arccosh_36',['ArcCosh',['../classmlx_1_1core_1_1_arc_cosh.html#a34597054db467941a2a883c653ba4d71',1,'mlx::core::ArcCosh']]], - ['arccosh_37',['arccosh',['../group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5',1,'mlx::core']]], - ['arcsin_38',['ArcSin',['../classmlx_1_1core_1_1_arc_sin.html#a97cb8c3d4d9d6abc627dec49a404f013',1,'mlx::core::ArcSin']]], - ['arcsin_39',['arcsin',['../group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619',1,'mlx::core']]], - ['arcsinh_40',['ArcSinh',['../classmlx_1_1core_1_1_arc_sinh.html#a30076b222788deeaaf9ad92d3c535f20',1,'mlx::core::ArcSinh']]], - ['arcsinh_41',['arcsinh',['../group__ops.html#gac62e2cedc49ef2c90dd8584000317450',1,'mlx::core']]], - ['arctan_42',['ArcTan',['../classmlx_1_1core_1_1_arc_tan.html#a3511153bbd421e89fd9294cdb3f79b44',1,'mlx::core::ArcTan']]], - ['arctan_43',['arctan',['../group__ops.html#gaa041f3f070e68f4946db07516b7d092e',1,'mlx::core']]], - ['arctan2_44',['ArcTan2',['../classmlx_1_1core_1_1_arc_tan2.html#aa1a4ebab9924b6bcc80df5b52ed0121a',1,'mlx::core::ArcTan2']]], - ['arctan2_45',['arctan2',['../group__ops.html#ga6caba9c92b5989123501f909cc7da354',1,'mlx::core']]], - ['arctanh_46',['ArcTanh',['../classmlx_1_1core_1_1_arc_tanh.html#a17857bd0e2a3ecf1f7bf8e1a3d354358',1,'mlx::core::ArcTanh']]], - ['arctanh_47',['arctanh',['../group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e',1,'mlx::core']]], - ['argmax_48',['argmax',['../group__ops.html#gae60b0b5339b9c50b9970260faf613e83',1,'mlx::core::argmax(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8',1,'mlx::core::argmax(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5',1,'mlx::core::argmax(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['argmin_49',['argmin',['../group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71',1,'mlx::core::argmin(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga6bc577c5ab10cd9c848ba81321595070',1,'mlx::core::argmin(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81',1,'mlx::core::argmin(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['argpartition_50',['ArgPartition',['../classmlx_1_1core_1_1_arg_partition.html#ab54b13dbf92351ba1ac06fd3e5a802df',1,'mlx::core::ArgPartition']]], - ['argpartition_51',['argpartition',['../group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd',1,'mlx::core::argpartition(const array &a, int kth, StreamOrDevice s={})'],['../group__ops.html#ga7b15c654c7463def57857a0e239989a3',1,'mlx::core::argpartition(const array &a, int kth, int axis, StreamOrDevice s={})']]], - ['argreduce_52',['ArgReduce',['../classmlx_1_1core_1_1_arg_reduce.html#aaccf8021dc24895656e25142eb65aa03',1,'mlx::core::ArgReduce']]], - ['argsort_53',['ArgSort',['../classmlx_1_1core_1_1_arg_sort.html#a38507a8445302a81cb44674c4a5fc0b0',1,'mlx::core::ArgSort']]], - ['argsort_54',['argsort',['../group__ops.html#ga8df3b2703bf671457422894dd870cdc5',1,'mlx::core::argsort(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6',1,'mlx::core::argsort(const array &a, int axis, StreamOrDevice s={})']]], - ['argument_5fencoder_55',['argument_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#a6e33e2b1287324fb4a6575e0da5e5881',1,'mlx::core::metal::Device']]], - ['arr_56',['arr',['../classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42',1,'pocketfft::detail::arr::arr()'],['../classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd',1,'pocketfft::detail::arr::arr(size_t n)'],['../classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0',1,'pocketfft::detail::arr::arr(arr &&other)']]], - ['arr_5finfo_57',['arr_info',['../classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e',1,'pocketfft::detail::arr_info']]], - ['array_58',['array',['../classmlx_1_1core_1_1array.html#a75fac72da3ce214fa3737df92a64b232',1,'mlx::core::array::array(T val, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a6db4b8c28c767cc16ad2785ece496dca',1,'mlx::core::array::array(const std::complex< float > &val, Dtype dtype=complex64)'],['../classmlx_1_1core_1_1array.html#a3e506a53b9c7567448f7809dda680210',1,'mlx::core::array::array(It data, std::vector< int > shape, Dtype dtype=TypeToDtype< typename std::iterator_traits< It >::value_type >())'],['../classmlx_1_1core_1_1array.html#a87f170384f4fb93decf2b80ae7280f00',1,'mlx::core::array::array(std::initializer_list< T > data, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a46642301da11e3eb4312c37349fbc9d7',1,'mlx::core::array::array(std::initializer_list< float > data)'],['../classmlx_1_1core_1_1array.html#a5e1812029394bfb1a706c83611286f49',1,'mlx::core::array::array(std::initializer_list< int > data, Dtype dtype)'],['../classmlx_1_1core_1_1array.html#a44e57a41819321e0d796e08cb9a06e4b',1,'mlx::core::array::array(std::initializer_list< T > data, std::vector< int > shape, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a5b5f562ff14c150842cb61628e531663',1,'mlx::core::array::array(allocator::Buffer data, std::vector< int > shape, Dtype dtype, deleter_t deleter=allocator::free)'],['../classmlx_1_1core_1_1array.html#a297df274e2da5cb884257bbeffd6b187',1,'mlx::core::array::array(const array &other)=default'],['../classmlx_1_1core_1_1array.html#ab6cbccbba66cc54acda4390b19f0397c',1,'mlx::core::array::array(array &&other)=default'],['../classmlx_1_1core_1_1array.html#adaade8f4bb7f8ecc0ba07efb17cd2620',1,'mlx::core::array::array(std::vector< int > shape, Dtype dtype, std::shared_ptr< Primitive > primitive, std::vector< array > inputs)']]], - ['array_5fequal_59',['array_equal',['../group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645',1,'mlx::core::array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})'],['../group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14',1,'mlx::core::array_equal(const array &a, const array &b, StreamOrDevice s={})']]], - ['arrayiterator_60',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ad3afcb24c6db7642bbc06835f7f3e27a',1,'mlx::core::array::ArrayIterator']]], - ['as_5fstrided_61',['as_strided',['../group__ops.html#ga8de80ecef30fc560003d40f61a38b99d',1,'mlx::core']]], - ['asin_62',['asin',['../namespacemetal.html#a16e843194df3fd136404bf80ba5ac95c',1,'metal::asin()'],['../namespacemetal_1_1fast.html#a769455a283da99654b6e42c3acf13eb1',1,'metal::fast::asin()'],['../namespacemetal_1_1precise.html#adc7b8b6e12e320cb32030f728dcbf438',1,'metal::precise::asin()']]], - ['asinh_63',['asinh',['../namespacemetal.html#abcc3251866930cfe880f89e7473d0e63',1,'metal::asinh()'],['../namespacemetal_1_1fast.html#a4367034b7b3e14310803bb2be975a556',1,'metal::fast::asinh()'],['../namespacemetal_1_1precise.html#aaad1cdde6687c8011fbc5fda1bb13424',1,'metal::precise::asinh()']]], - ['asstrided_64',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html#a80c0547f72ed53374eafc57d57b5d4af',1,'mlx::core::AsStrided']]], - ['astype_65',['AsType',['../classmlx_1_1core_1_1_as_type.html#a8c3241d402a8977bb4db037e225f5b47',1,'mlx::core::AsType']]], - ['astype_66',['astype',['../group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62',1,'mlx::core']]], - ['async_5feval_67',['async_eval',['../namespacemlx_1_1core.html#a15dda19aa7fa1fc5fca35df5cf963297',1,'mlx::core']]], - ['atan_68',['atan',['../namespacemetal.html#a80a771553d9a0012b93620d19c48b00f',1,'metal::atan()'],['../namespacemetal_1_1fast.html#a769503b4b7f89071d0983258c5a3ac5a',1,'metal::fast::atan()'],['../namespacemetal_1_1precise.html#aaaf4b5f4786a912089bbf0ae7619a6be',1,'metal::precise::atan()']]], - ['atan2_69',['atan2',['../namespacemetal.html#a1d430793eaa38ccf0d07145e3fcd1e61',1,'metal::atan2()'],['../namespacemetal_1_1fast.html#a00e687ea46f5affe26e6aef8fd62b89a',1,'metal::fast::atan2()'],['../namespacemetal_1_1precise.html#a6f161b049cc6884f87b09b33c2d1cd7f',1,'metal::precise::atan2()']]], - ['atanh_70',['atanh',['../namespacemetal.html#a57116427997ba71dd3863bfb15de33bf',1,'metal::atanh()'],['../namespacemetal_1_1fast.html#af24608fc605db9a14427d37c36dc1c53',1,'metal::fast::atanh()'],['../namespacemetal_1_1precise.html#a902994837653b90c47f4285673e712c4',1,'metal::precise::atanh()']]], - ['atleast_5f1d_71',['atleast_1d',['../group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396',1,'mlx::core::atleast_1d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5',1,'mlx::core::atleast_1d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atleast_5f2d_72',['atleast_2d',['../group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087',1,'mlx::core::atleast_2d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga9950299a80c2562f13448758f856d1f5',1,'mlx::core::atleast_2d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atleast_5f3d_73',['atleast_3d',['../group__ops.html#ga4afd919601e67782ff964465919956a0',1,'mlx::core::atleast_3d(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaffdf742ad79440a60dda40062a8074fe',1,'mlx::core::atleast_3d(const std::vector< array > &a, StreamOrDevice s={})']]], - ['atomic_5fupdate_74',['atomic_update',['../struct_none.html#a9d7414f58b3a526758007a4a1da72c74',1,'None::atomic_update()'],['../struct_and.html#a9ddf6d3f16dee000a5f0a30af4d43a7f',1,'And::atomic_update(device mlx_atomic< unsigned int > *out, bool val, int elem_idx, int offset=0)'],['../struct_and.html#af4a5c9f276d21580c0acaf1ff7064c57',1,'And::atomic_update(device mlx_atomic< bool > *out, bool val, uint offset=0)'],['../struct_or.html#a40840c0506bf4ed9a0771ba42c9ebb0f',1,'Or::atomic_update(device mlx_atomic< unsigned int > *out, bool val, uint elem_idx, uint offset=0)'],['../struct_or.html#aad1e6b5b015cd0910625da725afabdf3',1,'Or::atomic_update(device mlx_atomic< bool > *out, bool val, uint offset=0)'],['../struct_sum.html#a3188318e74823f00b9367317ec22ad6b',1,'Sum::atomic_update()'],['../struct_prod.html#a7ad8c7c18b1c54ff1d158683744a324d',1,'Prod::atomic_update()'],['../struct_min.html#a2377172960d21faa67b13c6b1eec5c69',1,'Min::atomic_update()'],['../struct_max.html#ac478a5cd1f567164d786240589d9260c',1,'Max::atomic_update()']]], - ['attach_5fevent_75',['attach_event',['../classmlx_1_1core_1_1array.html#a000c3cfe13cb378bf0523b62816190da',1,'mlx::core::array']]] + ['affine_5fdequantize_10',['affine_dequantize',['../quantized_8h.html#aecc3db76dcae742776e71ec8fca470c5',1,'affine_dequantize(): quantized.h'],['../namespacemlx_1_1core_1_1fast.html#a12c7ef41409d6fb378008e67b6fab328',1,'mlx::core::fast::affine_dequantize()']]], + ['affine_5fquantize_11',['affine_quantize',['../quantized_8h.html#a47610f886f988d84c3e789eb564a6c44',1,'affine_quantize(): quantized.h'],['../namespacemlx_1_1core_1_1fast.html#aa4b5f6886b2288cb6dfdd8598579f080',1,'mlx::core::fast::affine_quantize(const array &w, int group_size=64, int bits=4, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fast.html#a638c7e9b9ea8677f01786d8f9738baf8',1,'mlx::core::fast::affine_quantize(const array &w, const array &scales, const array &biases, int group_size=64, int bits=4, StreamOrDevice s={})']]], + ['affine_5fquantize_5fscales_5fbiases_12',['affine_quantize_scales_biases',['../quantized_8h.html#a57cd320aab5fa1abb65f79b0b781f575',1,'quantized.h']]], + ['affinequantize_13',['AffineQuantize',['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a84d5fa9e8c3de407fbcc5f38d2ed1473',1,'mlx::core::fast::AffineQuantize']]], + ['aligned_5falloc_14',['aligned_alloc',['../namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76',1,'pocketfft::detail']]], + ['aligned_5fallocator_15',['aligned_allocator',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator(const aligned_allocator< U > &)'],['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94',1,'pocketfft::detail::threading::aligned_allocator::aligned_allocator()=default']]], + ['aligned_5fdealloc_16',['aligned_dealloc',['../namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8',1,'pocketfft::detail']]], + ['all_17',['all',['../group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68',1,'mlx::core::all(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4',1,'mlx::core::all(const array &a, StreamOrDevice s={})'],['../group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59',1,'mlx::core::all(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc',1,'mlx::core::all(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['all_5fgather_18',['all_gather',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04',1,'mlx::core::distributed::detail::all_gather()'],['../namespacemlx_1_1core_1_1distributed.html#abf45499dd1209ba7162908f14634d2ba',1,'mlx::core::distributed::all_gather()']]], + ['all_5freduce_19',['all_reduce',['../reduce__all_8h.html#a8d143d2fea044e4808b69cc3723e33c7',1,'reduce_all.h']]], + ['all_5freduce_5fdispatch_20',['all_reduce_dispatch',['../namespacemlx_1_1core.html#a3ab0fd997d9a35782106ff083a72e098',1,'mlx::core']]], + ['all_5freduce_5fno_5fatomics_21',['all_reduce_no_atomics',['../reduce__all_8h.html#a09d140566cb3e6418a821be1e42db282',1,'reduce_all.h']]], + ['all_5fsum_22',['all_sum',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960',1,'mlx::core::distributed::detail::all_sum()'],['../namespacemlx_1_1core_1_1distributed.html#ad1c6aa544781765060552cacfa845001',1,'mlx::core::distributed::all_sum()']]], + ['allclose_23',['allclose',['../group__ops.html#gaf0cd4257de7542daf9faf5e605e31020',1,'mlx::core']]], + ['allgather_24',['AllGather',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ac9d0f1ba0d9602fa22e8f2b60f1231e0',1,'mlx::core::distributed::AllGather']]], + ['alloc_5ftmp_25',['alloc_tmp',['../namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, size_t axsize, size_t elemsize)'],['../namespacepocketfft_1_1detail.html#a13832735696303b9559c4663631d5475',1,'pocketfft::detail::alloc_tmp(const shape_t &shape, const shape_t &axes, size_t elemsize)']]], + ['allocate_26',['allocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826',1,'pocketfft::detail::threading::aligned_allocator']]], + ['allocator_27',['Allocator',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a5803678a418fef687fc65fa9d5c37b65',1,'mlx::core::allocator::Allocator::Allocator()=default'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#aa05c081ce80dc036f9d3dd8c195259d2',1,'mlx::core::allocator::Allocator::Allocator(const Allocator &other)=delete'],['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a8e8ce346a16cf0c62847bed9289f9959',1,'mlx::core::allocator::Allocator::Allocator(Allocator &&other)=delete']]], + ['allocator_28',['allocator',['../namespacemlx_1_1core_1_1allocator.html#aa23e2f20a336d0b159c097087194634e',1,'mlx::core::allocator::allocator()'],['../namespacemlx_1_1core_1_1metal.html#a74b3558bd518aecde6b14b0ba5e1a0d5',1,'mlx::core::metal::allocator()']]], + ['allreduce_29',['AllReduce',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a936b320f7b0a435f96e018c25f35e7ed',1,'mlx::core::distributed::AllReduce']]], + ['any_30',['any',['../group__ops.html#ga8598dd718fb05cb28535e250372d4e6f',1,'mlx::core::any(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gad37df97f253a963bece124198dbaf9ba',1,'mlx::core::any(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef',1,'mlx::core::any(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce',1,'mlx::core::any(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['apply_31',['apply',['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply(InT x) const'],['../struct_scale_op.html#a69f82bc925843a4e1c14dfe8ad2f3218',1,'ScaleOp::apply(InT x) const'],['../structmlx_1_1steel_1_1_transform_none.html#a84daa89be5b3348b5715bf8c5a01da75',1,'mlx::steel::TransformNone::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_none.html#ae4c397038f386b13eaa386638a0fce90',1,'mlx::steel::TransformNone::apply(InT x, OutT)'],['../structmlx_1_1steel_1_1_transform_add.html#afbb688d84443fd622b4dd2768cfe0acf',1,'mlx::steel::TransformAdd::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_add.html#a4923b0059d88099b2739f2cf0273ea19',1,'mlx::steel::TransformAdd::apply(InT x, OutT c)'],['../structmlx_1_1steel_1_1_transform_axpby.html#a14ad48b0189d6bdde06c66f1b567ae87',1,'mlx::steel::TransformAxpby::apply(InT x)'],['../structmlx_1_1steel_1_1_transform_axpby.html#aaf3a45e25d7abf7a34b48cc612e631ba',1,'mlx::steel::TransformAxpby::apply(InT x, OutT c) const']]], + ['apply_5fepilogue_32',['apply_epilogue',['../structmlx_1_1steel_1_1_block_m_m_a.html#af653c0808ba4fa9a25286f1febb7baff',1,'mlx::steel::BlockMMA::apply_epilogue(thread const UnaryEpilogue &epilogue_op)'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a823c56cbd2086f10272df7284a5247ae',1,'mlx::steel::BlockMMA::apply_epilogue(const device U *C, const int ldc, const int fdc, thread const BinaryEpilogue &epilogue_op)']]], + ['apply_5fepilogue_5fsafe_33',['apply_epilogue_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#a9e48f2d51099ec00171506724faab54a',1,'mlx::steel::BlockMMA']]], + ['apply_5finplace_5fop_34',['apply_inplace_op',['../structmlx_1_1steel_1_1_block_loader.html#adb4ca2cc193630a779de552fa8847ddf',1,'mlx::steel::BlockLoader']]], + ['arange_35',['Arange',['../classmlx_1_1core_1_1_arange.html#a1a70c3b0b9c67d5a9446c141c5b7c574',1,'mlx::core::Arange']]], + ['arange_36',['arange',['../namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b',1,'mlx::core::arange()'],['../namespacemlx_1_1core_1_1metal.html#a272c36f0faf2570cbb2f36030e9a3f26',1,'mlx::core::metal::arange()'],['../metal_2kernels_2arange_8h.html#a1e5126ee6ae0164c2343230c4d87c03e',1,'arange(): arange.h'],['../group__ops.html#ga7ca088b8090b9f84f2e08345cf3f835a',1,'mlx::core::arange(double start, double stop, double step, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga4c36b841dc5cba391dad029be5a0ad98',1,'mlx::core::arange(double start, double stop, double step, StreamOrDevice s={})'],['../group__ops.html#ga8d7cf9eb15e2daf1469058907e8abc85',1,'mlx::core::arange(double start, double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga74566a14e69ba6a25f5a35e7ade5c282',1,'mlx::core::arange(double start, double stop, StreamOrDevice s={})'],['../group__ops.html#ga345aa27af3dae3646b8b4b1068e89a3e',1,'mlx::core::arange(double stop, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#gaae179075d0fe23f4bd53fdf8c41f4c70',1,'mlx::core::arange(double stop, StreamOrDevice s={})'],['../group__ops.html#ga6b945f513077c2978afc1a952c884860',1,'mlx::core::arange(int start, int stop, int step, StreamOrDevice s={})'],['../group__ops.html#ga1c39fcc6eaa1c1867735c7f849d708d6',1,'mlx::core::arange(int start, int stop, StreamOrDevice s={})'],['../group__ops.html#gafe6e4580452c873cac294f16129e633f',1,'mlx::core::arange(int stop, StreamOrDevice s={})']]], + ['arccos_37',['ArcCos',['../classmlx_1_1core_1_1_arc_cos.html#a66f4ee841d17923d93241b71ea5103e9',1,'mlx::core::ArcCos']]], + ['arccos_38',['arccos',['../group__ops.html#ga08bec7cb10c84466487b507fc5bf9776',1,'mlx::core']]], + ['arccosh_39',['ArcCosh',['../classmlx_1_1core_1_1_arc_cosh.html#a34597054db467941a2a883c653ba4d71',1,'mlx::core::ArcCosh']]], + ['arccosh_40',['arccosh',['../group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5',1,'mlx::core']]], + ['arcsin_41',['ArcSin',['../classmlx_1_1core_1_1_arc_sin.html#a97cb8c3d4d9d6abc627dec49a404f013',1,'mlx::core::ArcSin']]], + ['arcsin_42',['arcsin',['../group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619',1,'mlx::core']]], + ['arcsinh_43',['ArcSinh',['../classmlx_1_1core_1_1_arc_sinh.html#a30076b222788deeaaf9ad92d3c535f20',1,'mlx::core::ArcSinh']]], + ['arcsinh_44',['arcsinh',['../group__ops.html#gac62e2cedc49ef2c90dd8584000317450',1,'mlx::core']]], + ['arctan_45',['ArcTan',['../classmlx_1_1core_1_1_arc_tan.html#a3511153bbd421e89fd9294cdb3f79b44',1,'mlx::core::ArcTan']]], + ['arctan_46',['arctan',['../group__ops.html#gaa041f3f070e68f4946db07516b7d092e',1,'mlx::core']]], + ['arctan2_47',['ArcTan2',['../classmlx_1_1core_1_1_arc_tan2.html#aa1a4ebab9924b6bcc80df5b52ed0121a',1,'mlx::core::ArcTan2']]], + ['arctan2_48',['arctan2',['../group__ops.html#ga6caba9c92b5989123501f909cc7da354',1,'mlx::core']]], + ['arctanh_49',['ArcTanh',['../classmlx_1_1core_1_1_arc_tanh.html#a17857bd0e2a3ecf1f7bf8e1a3d354358',1,'mlx::core::ArcTanh']]], + ['arctanh_50',['arctanh',['../group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e',1,'mlx::core']]], + ['argmax_51',['argmax',['../group__ops.html#gae60b0b5339b9c50b9970260faf613e83',1,'mlx::core::argmax(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8',1,'mlx::core::argmax(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5',1,'mlx::core::argmax(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['argmin_52',['argmin',['../group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71',1,'mlx::core::argmin(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga6bc577c5ab10cd9c848ba81321595070',1,'mlx::core::argmin(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81',1,'mlx::core::argmin(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['argpartition_53',['ArgPartition',['../classmlx_1_1core_1_1_arg_partition.html#ab54b13dbf92351ba1ac06fd3e5a802df',1,'mlx::core::ArgPartition']]], + ['argpartition_54',['argpartition',['../group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd',1,'mlx::core::argpartition(const array &a, int kth, StreamOrDevice s={})'],['../group__ops.html#ga7b15c654c7463def57857a0e239989a3',1,'mlx::core::argpartition(const array &a, int kth, int axis, StreamOrDevice s={})']]], + ['argreduce_55',['ArgReduce',['../classmlx_1_1core_1_1_arg_reduce.html#aaccf8021dc24895656e25142eb65aa03',1,'mlx::core::ArgReduce']]], + ['argsort_56',['ArgSort',['../classmlx_1_1core_1_1_arg_sort.html#a38507a8445302a81cb44674c4a5fc0b0',1,'mlx::core::ArgSort']]], + ['argsort_57',['argsort',['../group__ops.html#ga8df3b2703bf671457422894dd870cdc5',1,'mlx::core::argsort(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6',1,'mlx::core::argsort(const array &a, int axis, StreamOrDevice s={})']]], + ['argument_5fencoder_58',['argument_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#a6e33e2b1287324fb4a6575e0da5e5881',1,'mlx::core::metal::Device']]], + ['arr_59',['arr',['../classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42',1,'pocketfft::detail::arr::arr()'],['../classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd',1,'pocketfft::detail::arr::arr(size_t n)'],['../classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0',1,'pocketfft::detail::arr::arr(arr &&other)']]], + ['arr_5finfo_60',['arr_info',['../classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e',1,'pocketfft::detail::arr_info']]], + ['array_61',['array',['../classmlx_1_1core_1_1array.html#a75fac72da3ce214fa3737df92a64b232',1,'mlx::core::array::array(T val, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a6db4b8c28c767cc16ad2785ece496dca',1,'mlx::core::array::array(const std::complex< float > &val, Dtype dtype=complex64)'],['../classmlx_1_1core_1_1array.html#a3e506a53b9c7567448f7809dda680210',1,'mlx::core::array::array(It data, std::vector< int > shape, Dtype dtype=TypeToDtype< typename std::iterator_traits< It >::value_type >())'],['../classmlx_1_1core_1_1array.html#a87f170384f4fb93decf2b80ae7280f00',1,'mlx::core::array::array(std::initializer_list< T > data, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a46642301da11e3eb4312c37349fbc9d7',1,'mlx::core::array::array(std::initializer_list< float > data)'],['../classmlx_1_1core_1_1array.html#a5e1812029394bfb1a706c83611286f49',1,'mlx::core::array::array(std::initializer_list< int > data, Dtype dtype)'],['../classmlx_1_1core_1_1array.html#a44e57a41819321e0d796e08cb9a06e4b',1,'mlx::core::array::array(std::initializer_list< T > data, std::vector< int > shape, Dtype dtype=TypeToDtype< T >())'],['../classmlx_1_1core_1_1array.html#a5b5f562ff14c150842cb61628e531663',1,'mlx::core::array::array(allocator::Buffer data, std::vector< int > shape, Dtype dtype, deleter_t deleter=allocator::free)'],['../classmlx_1_1core_1_1array.html#a297df274e2da5cb884257bbeffd6b187',1,'mlx::core::array::array(const array &other)=default'],['../classmlx_1_1core_1_1array.html#ab6cbccbba66cc54acda4390b19f0397c',1,'mlx::core::array::array(array &&other)=default'],['../classmlx_1_1core_1_1array.html#adaade8f4bb7f8ecc0ba07efb17cd2620',1,'mlx::core::array::array(std::vector< int > shape, Dtype dtype, std::shared_ptr< Primitive > primitive, std::vector< array > inputs)']]], + ['array_5fequal_62',['array_equal',['../group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645',1,'mlx::core::array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})'],['../group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14',1,'mlx::core::array_equal(const array &a, const array &b, StreamOrDevice s={})']]], + ['arrayiterator_63',['ArrayIterator',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#ad3afcb24c6db7642bbc06835f7f3e27a',1,'mlx::core::array::ArrayIterator']]], + ['as_5fstrided_64',['as_strided',['../group__ops.html#ga8de80ecef30fc560003d40f61a38b99d',1,'mlx::core']]], + ['asin_65',['asin',['../namespacemetal.html#a16e843194df3fd136404bf80ba5ac95c',1,'metal::asin()'],['../namespacemetal_1_1fast.html#a769455a283da99654b6e42c3acf13eb1',1,'metal::fast::asin()'],['../namespacemetal_1_1precise.html#adc7b8b6e12e320cb32030f728dcbf438',1,'metal::precise::asin()']]], + ['asinh_66',['asinh',['../namespacemetal.html#abcc3251866930cfe880f89e7473d0e63',1,'metal::asinh()'],['../namespacemetal_1_1fast.html#a4367034b7b3e14310803bb2be975a556',1,'metal::fast::asinh()'],['../namespacemetal_1_1precise.html#aaad1cdde6687c8011fbc5fda1bb13424',1,'metal::precise::asinh()']]], + ['asstrided_67',['AsStrided',['../classmlx_1_1core_1_1_as_strided.html#a80c0547f72ed53374eafc57d57b5d4af',1,'mlx::core::AsStrided']]], + ['astype_68',['AsType',['../classmlx_1_1core_1_1_as_type.html#a8c3241d402a8977bb4db037e225f5b47',1,'mlx::core::AsType']]], + ['astype_69',['astype',['../group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62',1,'mlx::core']]], + ['async_5feval_70',['async_eval',['../namespacemlx_1_1core.html#a15dda19aa7fa1fc5fca35df5cf963297',1,'mlx::core']]], + ['atan_71',['atan',['../namespacemetal.html#a80a771553d9a0012b93620d19c48b00f',1,'metal::atan()'],['../namespacemetal_1_1fast.html#a769503b4b7f89071d0983258c5a3ac5a',1,'metal::fast::atan()'],['../namespacemetal_1_1precise.html#aaaf4b5f4786a912089bbf0ae7619a6be',1,'metal::precise::atan()']]], + ['atan2_72',['atan2',['../namespacemetal.html#a1d430793eaa38ccf0d07145e3fcd1e61',1,'metal::atan2()'],['../namespacemetal_1_1fast.html#a00e687ea46f5affe26e6aef8fd62b89a',1,'metal::fast::atan2()'],['../namespacemetal_1_1precise.html#a6f161b049cc6884f87b09b33c2d1cd7f',1,'metal::precise::atan2()']]], + ['atanh_73',['atanh',['../namespacemetal.html#a57116427997ba71dd3863bfb15de33bf',1,'metal::atanh()'],['../namespacemetal_1_1fast.html#af24608fc605db9a14427d37c36dc1c53',1,'metal::fast::atanh()'],['../namespacemetal_1_1precise.html#a902994837653b90c47f4285673e712c4',1,'metal::precise::atanh()']]], + ['atleast_5f1d_74',['atleast_1d',['../group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396',1,'mlx::core::atleast_1d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5',1,'mlx::core::atleast_1d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atleast_5f2d_75',['atleast_2d',['../group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087',1,'mlx::core::atleast_2d(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga9950299a80c2562f13448758f856d1f5',1,'mlx::core::atleast_2d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atleast_5f3d_76',['atleast_3d',['../group__ops.html#ga4afd919601e67782ff964465919956a0',1,'mlx::core::atleast_3d(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaffdf742ad79440a60dda40062a8074fe',1,'mlx::core::atleast_3d(const std::vector< array > &a, StreamOrDevice s={})']]], + ['atomic_5fupdate_77',['atomic_update',['../struct_none.html#a9d7414f58b3a526758007a4a1da72c74',1,'None::atomic_update()'],['../struct_and.html#a9ddf6d3f16dee000a5f0a30af4d43a7f',1,'And::atomic_update(device mlx_atomic< unsigned int > *out, bool val, int elem_idx, int offset=0)'],['../struct_and.html#af4a5c9f276d21580c0acaf1ff7064c57',1,'And::atomic_update(device mlx_atomic< bool > *out, bool val, uint offset=0)'],['../struct_or.html#a40840c0506bf4ed9a0771ba42c9ebb0f',1,'Or::atomic_update(device mlx_atomic< unsigned int > *out, bool val, uint elem_idx, uint offset=0)'],['../struct_or.html#aad1e6b5b015cd0910625da725afabdf3',1,'Or::atomic_update(device mlx_atomic< bool > *out, bool val, uint offset=0)'],['../struct_sum.html#a3188318e74823f00b9367317ec22ad6b',1,'Sum::atomic_update()'],['../struct_prod.html#a7ad8c7c18b1c54ff1d158683744a324d',1,'Prod::atomic_update()'],['../struct_min.html#a2377172960d21faa67b13c6b1eec5c69',1,'Min::atomic_update()'],['../struct_max.html#ac478a5cd1f567164d786240589d9260c',1,'Max::atomic_update()']]], + ['attach_5fevent_78',['attach_event',['../classmlx_1_1core_1_1array.html#a000c3cfe13cb378bf0523b62816190da',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/functions_10.js b/docs/build/html/search/functions_10.js index 545a22c8fb..e750b9fcc6 100644 --- a/docs/build/html/search/functions_10.js +++ b/docs/build/html/search/functions_10.js @@ -1,7 +1,7 @@ var searchData= [ ['pad_0',['Pad',['../classmlx_1_1core_1_1_pad.html#ad03da2c40b1e1f2fdf2649d00fa4ab43',1,'mlx::core::Pad']]], - ['pad_1',['pad',['../group__ops.html#ga4fd04d6472ae7cdb94a174edfe338ea3',1,'mlx::core::pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), StreamOrDevice s={})'],['../group__ops.html#gaca4e00d5e4ed9e9f29d56a17f19c2e31',1,'mlx::core::pad(const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), StreamOrDevice s={})'],['../group__ops.html#ga608164c4f5b4ac2fba374219285d90bc',1,'mlx::core::pad(const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), StreamOrDevice s={})'],['../group__ops.html#ga2e57e26c4d7f3d18802c164656cf915c',1,'mlx::core::pad(const array &a, int pad_width, const array &pad_value=array(0), StreamOrDevice s={})']]], + ['pad_1',['pad',['../group__ops.html#ga81c2f138ea81d78affd42e9a646cacbc',1,'mlx::core::pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#ga8495c9902b0a67f288918cb528d5abeb',1,'mlx::core::pad(const array &a, const std::vector< std::pair< int, int > > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#gaeaae2f9637b0b3e2b4c6c0517e285935',1,'mlx::core::pad(const array &a, const std::pair< int, int > &pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})'],['../group__ops.html#ga5b8684361f57cbfd5e6eedf82a62ded0',1,'mlx::core::pad(const array &a, int pad_width, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})']]], ['pad_5fgpu_2',['pad_gpu',['../namespacemlx_1_1core.html#a6e2054d396ae487d810642dc19cdd0b0',1,'mlx::core']]], ['partition_3',['Partition',['../classmlx_1_1core_1_1_partition.html#a7b82ca3895b6654308fac566b277ac0d',1,'mlx::core::Partition']]], ['partition_4',['partition',['../group__ops.html#gac1b30830a972fb9a2601379ad2b32405',1,'mlx::core::partition(const array &a, int kth, StreamOrDevice s={})'],['../group__ops.html#ga4fbea3a5f66cf81e3c119d1661119321',1,'mlx::core::partition(const array &a, int kth, int axis, StreamOrDevice s={})']]], @@ -19,8 +19,8 @@ var searchData= ['powr_16',['powr',['../namespacemetal.html#ae529e431f178bafedc18a889323c0bc2',1,'metal::powr()'],['../namespacemetal_1_1fast.html#a4293cbc94175b4dcc724fe4747eb5d5a',1,'metal::fast::powr()'],['../namespacemetal_1_1precise.html#ac9dbab0bd99b2b94e364aba5353bdcd7',1,'metal::precise::powr()']]], ['pre_5fout_17',['pre_out',['../struct_read_writer.html#a94da8aa85fa2916eaa3eaeb11499234a',1,'ReadWriter::pre_out(float2 elem) const'],['../struct_read_writer.html#a185553204b07a407ef02c41dd78e8239',1,'ReadWriter::pre_out(float2 elem, int length) const']]], ['prepare_5fslice_18',['prepare_slice',['../namespacemlx_1_1core.html#a07ae007c87cf2d723a0fa7b25a2b6a93',1,'mlx::core']]], - ['primitive_19',['Primitive',['../classmlx_1_1core_1_1_primitive.html#afc69f22ee1f6e8a9ecc2c3a8f43b8fdb',1,'mlx::core::Primitive::Primitive(Stream stream)'],['../classmlx_1_1core_1_1_primitive.html#a3349f745fae50ca7627f79a731a19e32',1,'mlx::core::Primitive::Primitive(const Primitive &other)=delete'],['../classmlx_1_1core_1_1_primitive.html#a342da891b9882bdee9a0e0c1ac826eda',1,'mlx::core::Primitive::Primitive(Primitive &&other)=delete']]], - ['primitive_20',['primitive',['../classmlx_1_1core_1_1array.html#a790548666511d8c6d9f92ee79d2ce14c',1,'mlx::core::array']]], + ['primitive_19',['primitive',['../classmlx_1_1core_1_1array.html#a790548666511d8c6d9f92ee79d2ce14c',1,'mlx::core::array']]], + ['primitive_20',['Primitive',['../classmlx_1_1core_1_1_primitive.html#afc69f22ee1f6e8a9ecc2c3a8f43b8fdb',1,'mlx::core::Primitive::Primitive(Stream stream)'],['../classmlx_1_1core_1_1_primitive.html#a3349f745fae50ca7627f79a731a19e32',1,'mlx::core::Primitive::Primitive(const Primitive &other)=delete'],['../classmlx_1_1core_1_1_primitive.html#a342da891b9882bdee9a0e0c1ac826eda',1,'mlx::core::Primitive::Primitive(Primitive &&other)=delete']]], ['primitive_5fid_21',['primitive_id',['../classmlx_1_1core_1_1array.html#af5ad83605d4eea81561246873bee1d7c',1,'mlx::core::array']]], ['primitive_5fptr_22',['primitive_ptr',['../classmlx_1_1core_1_1array.html#a5119cd616ec3c05d65878944b8889469',1,'mlx::core::array']]], ['print_23',['print',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a6814f9008a683c6911d5b8991ef770ab',1,'mlx::core::distributed::AllReduce::print()'],['../classmlx_1_1core_1_1_primitive.html#ae1aff91354ce036596088a3e19474ecb',1,'mlx::core::Primitive::print()'],['../classmlx_1_1core_1_1_abs.html#a643d6db5116eed978e3208804a992107',1,'mlx::core::Abs::print()'],['../classmlx_1_1core_1_1_add.html#a8a96345aa63724f22b68bca7b861211d',1,'mlx::core::Add::print()'],['../classmlx_1_1core_1_1_add_m_m.html#a1262ac2c4c6e9ff6b6047bf7605e5cc9',1,'mlx::core::AddMM::print()'],['../classmlx_1_1core_1_1_arange.html#abd73d2b793da796dc7cf04c9f7d5c19e',1,'mlx::core::Arange::print()'],['../classmlx_1_1core_1_1_arc_cos.html#aa48d8bec4efbac569d809cf11648b739',1,'mlx::core::ArcCos::print()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6a9a2ab0cc360d7e2f9676db17f8e630',1,'mlx::core::ArcCosh::print()'],['../classmlx_1_1core_1_1_arc_sin.html#a895a35c9dd22fdb06e7b971bfd6fde87',1,'mlx::core::ArcSin::print()'],['../classmlx_1_1core_1_1_arc_sinh.html#aa8b2934a8a0b2eedec8257bbb5726430',1,'mlx::core::ArcSinh::print()'],['../classmlx_1_1core_1_1_arc_tan.html#ab0309e4feca36f221b3d672dc92cac05',1,'mlx::core::ArcTan::print()'],['../classmlx_1_1core_1_1_arc_tan2.html#abdfef9f572d06df1251c28222756a361',1,'mlx::core::ArcTan2::print()'],['../classmlx_1_1core_1_1_arc_tanh.html#aa9549311240d7ba225b84e1df9ad8523',1,'mlx::core::ArcTanh::print()'],['../classmlx_1_1core_1_1_arg_partition.html#aa8678d94fa1571ea71a7bf790cdb8d63',1,'mlx::core::ArgPartition::print()'],['../classmlx_1_1core_1_1_arg_reduce.html#a153a6d8dba7301c4fcd0e429154ead8f',1,'mlx::core::ArgReduce::print()'],['../classmlx_1_1core_1_1_arg_sort.html#a0b59ce43e0982d634a01631728b419bd',1,'mlx::core::ArgSort::print()'],['../classmlx_1_1core_1_1_as_type.html#aa617e29147c14bd5d1fa8ad0bf65af0c',1,'mlx::core::AsType::print()'],['../classmlx_1_1core_1_1_as_strided.html#af2e21b77ea9e6c70bca45224967745bf',1,'mlx::core::AsStrided::print()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a69b28e239da7fdb89f0a9f9467dd797d',1,'mlx::core::BitwiseBinary::print()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a37ecf6fa296d28efb7651a3c510fe159',1,'mlx::core::BlockMaskedMM::print()'],['../classmlx_1_1core_1_1_gather_m_m.html#ae7a6f4eecb15e95b21e6c87068ebd758',1,'mlx::core::GatherMM::print()'],['../classmlx_1_1core_1_1_broadcast.html#a6a610412861c6e472f930b6721b99a11',1,'mlx::core::Broadcast::print()'],['../classmlx_1_1core_1_1_ceil.html#a14a0048dd6496341cacaddada68276ee',1,'mlx::core::Ceil::print()'],['../classmlx_1_1core_1_1_compiled.html#a271521f92eef49c39799f38e26b64a9b',1,'mlx::core::Compiled::print()'],['../classmlx_1_1core_1_1_concatenate.html#a56f29b585a6d1d958954a68dcc893f33',1,'mlx::core::Concatenate::print()'],['../classmlx_1_1core_1_1_conjugate.html#a40281539bbd543ac8fd8e28650de17e4',1,'mlx::core::Conjugate::print()'],['../classmlx_1_1core_1_1_convolution.html#a844eab7c4cc99e775cfb561265ed14fd',1,'mlx::core::Convolution::print()'],['../classmlx_1_1core_1_1_copy.html#acfa1a02ab9cdab593e928faa515a8008',1,'mlx::core::Copy::print()'],['../classmlx_1_1core_1_1_cos.html#a81858457e4bea931a4bc6f6e38b0f696',1,'mlx::core::Cos::print()'],['../classmlx_1_1core_1_1_cosh.html#ac247faad68c1050cda9f72d7d6d040e2',1,'mlx::core::Cosh::print()'],['../classmlx_1_1core_1_1_custom_transforms.html#a2ddbacbc468271b11caee0ad97005298',1,'mlx::core::CustomTransforms::print()'],['../classmlx_1_1core_1_1_depends.html#aed575b0d927f4341f60442c70adeeb82',1,'mlx::core::Depends::print()'],['../classmlx_1_1core_1_1_divide.html#af3c15337ac15522cc34ed98b97895bb6',1,'mlx::core::Divide::print()'],['../classmlx_1_1core_1_1_div_mod.html#a7edbed50d07869d921e529157931b7a1',1,'mlx::core::DivMod::print()'],['../classmlx_1_1core_1_1_select.html#a678285f2c0b9dae85692399c3aa692a7',1,'mlx::core::Select::print()'],['../classmlx_1_1core_1_1_remainder.html#aeaecac5ea8e606d7ecd393d8019029e4',1,'mlx::core::Remainder::print()'],['../classmlx_1_1core_1_1_equal.html#a0787bf32f0b405a8b2ac809d2d990774',1,'mlx::core::Equal::print()'],['../classmlx_1_1core_1_1_erf.html#a186af7b783cf832c3b25eec3a09f5a0c',1,'mlx::core::Erf::print()'],['../classmlx_1_1core_1_1_erf_inv.html#a0acb31bd5780abf61877bd1a3e0fd4f9',1,'mlx::core::ErfInv::print()'],['../classmlx_1_1core_1_1_exp.html#ad87cc1b2ae595a613b03b0fdca63ae6a',1,'mlx::core::Exp::print()'],['../classmlx_1_1core_1_1_expm1.html#af1a99266fc50aa5948cdd298e2916ef1',1,'mlx::core::Expm1::print()'],['../classmlx_1_1core_1_1_f_f_t.html#a15a2a5f7647f5fb78611a251d3270edf',1,'mlx::core::FFT::print()'],['../classmlx_1_1core_1_1_floor.html#ac289e87c5fac15e2f491e2513be610f6',1,'mlx::core::Floor::print()'],['../classmlx_1_1core_1_1_full.html#a68e08303f4960ab373b84a3312edc013',1,'mlx::core::Full::print()'],['../classmlx_1_1core_1_1_gather.html#a9d57637a8a65008683c3847251bdcf91',1,'mlx::core::Gather::print()'],['../classmlx_1_1core_1_1_greater.html#aa2980e45cd2c79ebfb394012d3108a04',1,'mlx::core::Greater::print()'],['../classmlx_1_1core_1_1_greater_equal.html#ab98045c861d2d2ffb0398c2c1d671cef',1,'mlx::core::GreaterEqual::print()'],['../classmlx_1_1core_1_1_hadamard.html#a3df6e7e3b3b71bf50be5f1a05d0870b6',1,'mlx::core::Hadamard::print()'],['../classmlx_1_1core_1_1_less.html#ad67e6f66d7b75546fd98dbee6b631d78',1,'mlx::core::Less::print()'],['../classmlx_1_1core_1_1_less_equal.html#a409842d3862113c53cbbdf7467a06950',1,'mlx::core::LessEqual::print()'],['../classmlx_1_1core_1_1_load.html#a54e08a0ca41b7c9f1a76b00c889f0bfa',1,'mlx::core::Load::print()'],['../classmlx_1_1core_1_1_log.html#a7b946d98d4a228c6be9f606a3bd8a30d',1,'mlx::core::Log::print()'],['../classmlx_1_1core_1_1_log1p.html#a8a1569dde30440ce11ea466ccc69d2d4',1,'mlx::core::Log1p::print()'],['../classmlx_1_1core_1_1_logical_not.html#a001ff3eca46440f0d8a287e0b98a8a2c',1,'mlx::core::LogicalNot::print()'],['../classmlx_1_1core_1_1_logical_and.html#a9a5220eb56e1fd94fd879394ee5ad397',1,'mlx::core::LogicalAnd::print()'],['../classmlx_1_1core_1_1_logical_or.html#a6becc5fbfadde850de9857099dcd5003',1,'mlx::core::LogicalOr::print()'],['../classmlx_1_1core_1_1_log_add_exp.html#a702a2eff0bd1ae7b6fb829dd0b0b11b9',1,'mlx::core::LogAddExp::print()'],['../classmlx_1_1core_1_1_matmul.html#abb4a16a265a05d56a2f5d2e89d6f9dfd',1,'mlx::core::Matmul::print()'],['../classmlx_1_1core_1_1_maximum.html#a3b708a1d6b526719c62850294776f8ca',1,'mlx::core::Maximum::print()'],['../classmlx_1_1core_1_1_minimum.html#a137677bf32c626a768b732a7b8575512',1,'mlx::core::Minimum::print()'],['../classmlx_1_1core_1_1_multiply.html#aa4f1f7af68346ce80c2636df415c9909',1,'mlx::core::Multiply::print()'],['../classmlx_1_1core_1_1_negative.html#a0d5c30e267ff6468d64f1987f9f83f91',1,'mlx::core::Negative::print()'],['../classmlx_1_1core_1_1_not_equal.html#a12aa2f764880d29e627540610b63af09',1,'mlx::core::NotEqual::print()'],['../classmlx_1_1core_1_1_number_of_elements.html#aecde30826970938f3aa688979a668f52',1,'mlx::core::NumberOfElements::print()'],['../classmlx_1_1core_1_1_pad.html#af87754daaf51f6a6cf8bd4949ca1e70a',1,'mlx::core::Pad::print()'],['../classmlx_1_1core_1_1_partition.html#ab5c7aa4fed325475b33d4004649f0dc0',1,'mlx::core::Partition::print()'],['../classmlx_1_1core_1_1_power.html#a33e2d7ff078426fe66ea2370ceb5af60',1,'mlx::core::Power::print()'],['../classmlx_1_1core_1_1_quantized_matmul.html#aaef8c96d4d40b4fa08ced540d341a4db',1,'mlx::core::QuantizedMatmul::print()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a53c3fa7beb51ce2e1c2da28633406fe0',1,'mlx::core::GatherQMM::print()'],['../classmlx_1_1core_1_1_random_bits.html#a8a5593c34fd868d94b36a8ced1390271',1,'mlx::core::RandomBits::print()'],['../classmlx_1_1core_1_1_reshape.html#a0f2323d5d67ece0eb25ecff565b21862',1,'mlx::core::Reshape::print()'],['../classmlx_1_1core_1_1_reduce.html#a399be3a89553787a0a687706881f03cd',1,'mlx::core::Reduce::print()'],['../classmlx_1_1core_1_1_round.html#af0dfe8943109c936b35ab0082f566f72',1,'mlx::core::Round::print()'],['../classmlx_1_1core_1_1_scan.html#ad5b6308c79e9b985a49df35eadd15b22',1,'mlx::core::Scan::print()'],['../classmlx_1_1core_1_1_scatter.html#aa9d45cbfb27b814517f6016092b30efa',1,'mlx::core::Scatter::print()'],['../classmlx_1_1core_1_1_sigmoid.html#ad4cd19938e5159754aa7516f405580c2',1,'mlx::core::Sigmoid::print()'],['../classmlx_1_1core_1_1_sign.html#a2aa0720fe0a6d2408eb43c25d3d45b0a',1,'mlx::core::Sign::print()'],['../classmlx_1_1core_1_1_sin.html#a73b31005551015897f15c00e8b0222e4',1,'mlx::core::Sin::print()'],['../classmlx_1_1core_1_1_sinh.html#a5b4753d52e80799d4fea0b9172d25a77',1,'mlx::core::Sinh::print()'],['../classmlx_1_1core_1_1_slice.html#a50851148948d924b71817cfbd4401504',1,'mlx::core::Slice::print()'],['../classmlx_1_1core_1_1_slice_update.html#a751eefb9922c56479b4b0de2ad45439b',1,'mlx::core::SliceUpdate::print()'],['../classmlx_1_1core_1_1_softmax.html#aa783610ef6b82b92681e78fc99412d83',1,'mlx::core::Softmax::print()'],['../classmlx_1_1core_1_1_sort.html#ada81b9343f80958174eba708452927a2',1,'mlx::core::Sort::print()'],['../classmlx_1_1core_1_1_split.html#ad0c31fe5972643cc75fde10445fc47f2',1,'mlx::core::Split::print()'],['../classmlx_1_1core_1_1_square.html#a75feb558cd1d615e96309dd7d1e81384',1,'mlx::core::Square::print()'],['../classmlx_1_1core_1_1_sqrt.html#a8681c8de2f50049848d320c47f713c0f',1,'mlx::core::Sqrt::print()'],['../classmlx_1_1core_1_1_stop_gradient.html#acc7a7d51cbf014dae8ba3d20bedcad50',1,'mlx::core::StopGradient::print()'],['../classmlx_1_1core_1_1_subtract.html#a3834fd305435fb5a8e512566832e372b',1,'mlx::core::Subtract::print()'],['../classmlx_1_1core_1_1_tan.html#aeea7c284d595a2a928d5f28a55e9be7f',1,'mlx::core::Tan::print()'],['../classmlx_1_1core_1_1_tanh.html#a73f4976d641daf697cc1a231d773d78e',1,'mlx::core::Tanh::print()'],['../classmlx_1_1core_1_1_uniform.html#a01510998719b19df137451cc37850b8d',1,'mlx::core::Uniform::print()'],['../classmlx_1_1core_1_1_view.html#a513b034919a8a494add3155f910a360c',1,'mlx::core::View::print()'],['../classmlx_1_1core_1_1_transpose.html#ac6c87b850f4e5560aa13a5e1e9f9fe04',1,'mlx::core::Transpose::print()'],['../classmlx_1_1core_1_1_q_r_f.html#aba3526722b3a52b41fa8103b909f7f3b',1,'mlx::core::QRF::print()'],['../classmlx_1_1core_1_1_s_v_d.html#ab87a4e7ef857936bea66ba9e24662f53',1,'mlx::core::SVD::print()'],['../classmlx_1_1core_1_1_inverse.html#a543f18f1ce5c06c897141091e95a66e9',1,'mlx::core::Inverse::print()'],['../classmlx_1_1core_1_1_cholesky.html#a0a8b51ff7f5369d22bdc58910d4aaf84',1,'mlx::core::Cholesky::print()'],['../structmlx_1_1core_1_1_print_formatter.html#a79fad4cf5844db8c92b066539146281b',1,'mlx::core::PrintFormatter::print(std::ostream &os, bool val)'],['../structmlx_1_1core_1_1_print_formatter.html#a8da448a8adae671b26359341ea514316',1,'mlx::core::PrintFormatter::print(std::ostream &os, int16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9d750c134a6fbfa8251c5b1d01d73287',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#adbbb9cbff767f9db73c659a0c07ba633',1,'mlx::core::PrintFormatter::print(std::ostream &os, int32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a520adb07fafd911b22bc24b295e4f6cf',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint32_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ab0c702f1ae201e17cd328c9855cf522e',1,'mlx::core::PrintFormatter::print(std::ostream &os, int64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac59a5137ddd8b32aae057bb9826ee80d',1,'mlx::core::PrintFormatter::print(std::ostream &os, uint64_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ac4b7895d1168cfc1a3d1186d8a414d2f',1,'mlx::core::PrintFormatter::print(std::ostream &os, float16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#ae21005f92bc641f2d657096f5d176a6d',1,'mlx::core::PrintFormatter::print(std::ostream &os, bfloat16_t val)'],['../structmlx_1_1core_1_1_print_formatter.html#a57af5c32561b95d6ac2a3a1dc4f5d43e',1,'mlx::core::PrintFormatter::print(std::ostream &os, float val)'],['../structmlx_1_1core_1_1_print_formatter.html#a9e1dc67c9afb0a09966336504790823d',1,'mlx::core::PrintFormatter::print(std::ostream &os, complex64_t val)']]], diff --git a/docs/build/html/search/functions_12.js b/docs/build/html/search/functions_12.js index bfd25c00d5..043c64fab9 100644 --- a/docs/build/html/search/functions_12.js +++ b/docs/build/html/search/functions_12.js @@ -34,7 +34,7 @@ var searchData= ['reduce_5futils_31',['reduce_utils',['../namespacemlx_1_1core_1_1metal.html#a2ec39572806310cf528aea06530e8af8',1,'mlx::core::metal']]], ['reduction_5fop_32',['reduction_op',['../namespacemlx_1_1core.html#aa08ffc1e8f2c58afb2d463496f827ef0',1,'mlx::core::reduction_op(const array &x, array &out, const std::vector< int > &axes, U init, OpS ops, OpC opc, Op op)'],['../namespacemlx_1_1core.html#a064d61b6ddc9e5d1e261a7e33de71083',1,'mlx::core::reduction_op(const array &x, array &out, const std::vector< int > &axes, U init, Op op)']]], ['reductionplan_33',['ReductionPlan',['../structmlx_1_1core_1_1_reduction_plan.html#a3df6acc08ef92647aab19d68b0961610',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_, std::vector< int > shape_, std::vector< size_t > strides_)'],['../structmlx_1_1core_1_1_reduction_plan.html#aec7496f3740a0b0d51aaa606f6fd68f4',1,'mlx::core::ReductionPlan::ReductionPlan(ReductionOpType type_)']]], - ['register_5flibrary_34',['register_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a45945f2efcd242d915ffa2171e92bf9d',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::string &lib_path)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a6993ce23a485b725b687628b77751ad2',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::function< std::string(const std::string &)> &lib_path_func=get_colocated_mtllib_path)']]], + ['register_5flibrary_34',['register_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a45945f2efcd242d915ffa2171e92bf9d',1,'mlx::core::metal::Device::register_library(const std::string &lib_name, const std::string &lib_path)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a99ff72689b7beb65ad4541391b0eeabf',1,'mlx::core::metal::Device::register_library(const std::string &lib_name)']]], ['remainder_35',['Remainder',['../classmlx_1_1core_1_1_remainder.html#a4f3eada4a21898af4a77d1d27ce14641',1,'mlx::core::Remainder']]], ['remainder_36',['remainder',['../group__ops.html#ga99f5c904f724156a814d7817188351d2',1,'mlx::core']]], ['remaining_37',['remaining',['../classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd',1,'pocketfft::detail::multi_iter::remaining()'],['../classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6',1,'pocketfft::detail::simple_iter::remaining()'],['../classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63',1,'pocketfft::detail::rev_iter::remaining()']]], @@ -70,7 +70,6 @@ var searchData= ['row_5freduce_5fgeneral_5fmed_67',['row_reduce_general_med',['../reduce__row_8h.html#ad77d444bcfa9e6afcc1ffc1dcb919f1c',1,'reduce_row.h']]], ['row_5freduce_5fgeneral_5fno_5fatomics_68',['row_reduce_general_no_atomics',['../reduce__row_8h.html#a034fdccc83e66871550c2d42399f0ab5',1,'reduce_row.h']]], ['row_5freduce_5fgeneral_5fsmall_69',['row_reduce_general_small',['../reduce__row_8h.html#aea2b1ba640fffb71c73550ec747b3148',1,'reduce_row.h']]], - ['rows_70',['rows',['../class_m_p_s_1_1_matrix_descriptor.html#a04ce0dd1d4632ff8c4328e4b62d71fc0',1,'MPS::MatrixDescriptor']]], - ['rsqrt_71',['rsqrt',['../namespacemetal.html#a1cf4b605c0aa7ff5bfe5e979a16f5157',1,'metal::rsqrt()'],['../namespacemetal_1_1fast.html#aa62097c750f1e4b69d09277f19976ab1',1,'metal::fast::rsqrt()'],['../namespacemetal_1_1precise.html#afb397b477745f12a44423934fa2b05ac',1,'metal::precise::rsqrt()'],['../group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1',1,'mlx::core::rsqrt()']]], - ['run_72',['run',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel']]] + ['rsqrt_70',['rsqrt',['../namespacemetal.html#a1cf4b605c0aa7ff5bfe5e979a16f5157',1,'metal::rsqrt()'],['../namespacemetal_1_1fast.html#aa62097c750f1e4b69d09277f19976ab1',1,'metal::fast::rsqrt()'],['../namespacemetal_1_1precise.html#afb397b477745f12a44423934fa2b05ac',1,'metal::precise::rsqrt()'],['../group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1',1,'mlx::core::rsqrt()']]], + ['run_71',['run',['../struct_g_e_m_v_kernel.html#ac4a7b5011a0ea938ab1949bb1767fc1a',1,'GEMVKernel::run()'],['../struct_g_e_m_v_t_kernel.html#a5d68656832de892f33db939005713927',1,'GEMVTKernel::run()'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a00e55d4a161758350ed7310817d2d2a5',1,'mlx::steel::GEMMKernel::run()']]] ]; diff --git a/docs/build/html/search/functions_13.js b/docs/build/html/search/functions_13.js index 4c860d619d..76b8f929e0 100644 --- a/docs/build/html/search/functions_13.js +++ b/docs/build/html/search/functions_13.js @@ -34,106 +34,102 @@ var searchData= ['set_5fstatus_31',['set_status',['../classmlx_1_1core_1_1array.html#a63598018999b49f1340b183cb303f05c',1,'mlx::core::array']]], ['set_5ftracer_32',['set_tracer',['../classmlx_1_1core_1_1array.html#af26e6be1a9e6239471a4c24310c0c7c8',1,'mlx::core::array']]], ['set_5fvalue_33',['set_value',['../classmlx_1_1core_1_1_event.html#a0d077b11f4b28f882b42440b7ac6d40d',1,'mlx::core::Event']]], - ['setbatchsize_34',['setBatchSize',['../class_m_p_s_1_1_matrix_multiplication.html#ac24cc7d71cdf3fd5bc4291ba2bfcd9e7',1,'MPS::MatrixMultiplication']]], - ['setbatchstart_35',['setBatchStart',['../class_m_p_s_1_1_matrix_multiplication.html#a82008429b4f14b5eba13142886b76cde',1,'MPS::MatrixMultiplication']]], - ['setleftmatrixorigin_36',['setLeftMatrixOrigin',['../class_m_p_s_1_1_matrix_multiplication.html#a8b905b75090af80d97cf09b05cdad528',1,'MPS::MatrixMultiplication']]], - ['setresultmatrixorigin_37',['setResultMatrixOrigin',['../class_m_p_s_1_1_matrix_multiplication.html#a86647678c55adffc51114d036445ba10',1,'MPS::MatrixMultiplication']]], - ['setrightmatrixorigin_38',['setRightMatrixOrigin',['../class_m_p_s_1_1_matrix_multiplication.html#afaa45acd37397737ef4d94e8803be7e2',1,'MPS::MatrixMultiplication']]], - ['shape_39',['shape',['../classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0',1,'pocketfft::detail::arr_info::shape() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8',1,'pocketfft::detail::arr_info::shape(size_t i) const'],['../classmlx_1_1core_1_1array.html#a4a2a2c8a4a5beafd723fc13f2055d55d',1,'mlx::core::array::shape() const'],['../classmlx_1_1core_1_1array.html#a51ed0c45666264dc172d06fba159eb8f',1,'mlx::core::array::shape(int dim) const']]], - ['shapes_5fwithout_5freduction_5faxes_40',['shapes_without_reduction_axes',['../namespacemlx_1_1core.html#a44c3ea6db6553c3f6552b9ba64a69494',1,'mlx::core']]], - ['shared_5fbuffer_5fslice_41',['shared_buffer_slice',['../namespacemlx_1_1core.html#a427f2c255dfc6e1f83f97587b08e71bc',1,'mlx::core']]], - ['shutdown_42',['shutdown',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411',1,'pocketfft::detail::threading::thread_pool']]], - ['siblings_43',['siblings',['../classmlx_1_1core_1_1array.html#acf80fde8f743f65ad5b4be69fcb7a74d',1,'mlx::core::array::siblings() const'],['../classmlx_1_1core_1_1array.html#a7263f23e70a580a9bc2129fbcde36e6c',1,'mlx::core::array::siblings()']]], - ['sigmoid_44',['Sigmoid',['../classmlx_1_1core_1_1_sigmoid.html#a47eca99113ec19f0eb60b6a0472c592b',1,'mlx::core::Sigmoid']]], - ['sigmoid_45',['sigmoid',['../group__ops.html#ga708abf8f79609cd6831db7c38cafac0e',1,'mlx::core']]], - ['sign_46',['Sign',['../classmlx_1_1core_1_1_sign.html#afe951e50907bc23a601ec5fa9eae5763',1,'mlx::core::Sign']]], - ['sign_47',['sign',['../group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa',1,'mlx::core']]], - ['signal_48',['signal',['../classmlx_1_1core_1_1_event.html#a65a858445506a61be5889ae0e3651b89',1,'mlx::core::Event']]], - ['signaling_5fnan_49',['signaling_NaN',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ad1f76a43c7d51a3765174aa6e0dd9f80',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['simd_5fbroadcast_50',['simd_broadcast',['../namespacemetal.html#a498f1e85107eb5f01ba4435977f8efe0',1,'metal']]], - ['simd_5fexclusive_5fscan_51',['simd_exclusive_scan',['../struct_cum_sum.html#a64d1933942de4df2d44dd90e9e5e4460',1,'CumSum::simd_exclusive_scan()'],['../struct_cum_prod.html#a5b0577295853a48856b84fd4f935dff8',1,'CumProd::simd_exclusive_scan()'],['../struct_cum_prod_3_01bool_01_4.html#a1a86e9398bae24182b7be0a6577bf223',1,'CumProd< bool >::simd_exclusive_scan()'],['../struct_cum_max.html#ae11b67aa6c998e9a01615b2a79af4403',1,'CumMax::simd_exclusive_scan()'],['../struct_cum_min.html#a83e65017ff33018b585c043fb803773b',1,'CumMin::simd_exclusive_scan()']]], - ['simd_5fmax_52',['simd_max',['../namespacemetal.html#a048cad0aca52cb737ebf103e76bd1c49',1,'metal']]], - ['simd_5fmin_53',['simd_min',['../namespacemetal.html#ae9e2a23e00724ba2d7868bc4112b386b',1,'metal']]], - ['simd_5fprefix_5fexclusive_5fproduct_54',['simd_prefix_exclusive_product',['../namespacemetal.html#a5ca40242390b632f737e29636829b2e4',1,'metal']]], - ['simd_5fprefix_5fexclusive_5fsum_55',['simd_prefix_exclusive_sum',['../namespacemetal.html#abfbb70c7471f28bf7ff36a612ad014b2',1,'metal']]], - ['simd_5fprefix_5finclusive_5fproduct_56',['simd_prefix_inclusive_product',['../namespacemetal.html#a6ca6a7e1996228fa536e969e9e45c446',1,'metal']]], - ['simd_5fprefix_5finclusive_5fsum_57',['simd_prefix_inclusive_sum',['../namespacemetal.html#a567acb18199ac0107712eb8cb8aeb8e9',1,'metal']]], - ['simd_5fproduct_58',['simd_product',['../namespacemetal.html#ac6e883a04e2265a9790d7db76059e1b4',1,'metal']]], - ['simd_5freduce_59',['simd_reduce',['../struct_and.html#ab4df5517e050cfa7257c34c6b0bd980b',1,'And::simd_reduce()'],['../struct_or.html#aef6b5abbbf3fd22981b8d27562022605',1,'Or::simd_reduce()'],['../struct_sum.html#acf10e0604098391526da3b8271928b5c',1,'Sum::simd_reduce()'],['../struct_prod.html#ab0d7a38360b971b4d99c4ad2db310069',1,'Prod::simd_reduce()'],['../struct_min.html#afe8d7cefac624fd4267c1b252ac38528',1,'Min::simd_reduce()'],['../struct_max.html#a4f74440c6f1ce8247d5a87c2a8882080',1,'Max::simd_reduce()']]], - ['simd_5fscan_60',['simd_scan',['../struct_cum_sum.html#a83342af06f6a700067e3b451636852d6',1,'CumSum::simd_scan()'],['../struct_cum_prod.html#a860dbf6146de088d07175bd165b74004',1,'CumProd::simd_scan()'],['../struct_cum_prod_3_01bool_01_4.html#abeb5ec4237b330e7219f4e881cf10d7a',1,'CumProd< bool >::simd_scan()'],['../struct_cum_max.html#adc9ec8bb09b4433d4c2f03022c43d781',1,'CumMax::simd_scan()'],['../struct_cum_min.html#a0a1005d91b1c90e90e2c6dbd6c296649',1,'CumMin::simd_scan()']]], - ['simd_5fshuffle_61',['simd_shuffle',['../namespacemetal.html#a259ed115bc3c58f88eb35830916b26d4',1,'metal']]], - ['simd_5fshuffle_5fand_5ffill_5fdown_62',['simd_shuffle_and_fill_down',['../namespacemetal.html#ae29a06f0eac636ad7af21dea5b04938b',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a0ee6239fa29a5f9ee0201e0dc5ddc8e0',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], - ['simd_5fshuffle_5fand_5ffill_5fup_63',['simd_shuffle_and_fill_up',['../namespacemetal.html#a1ca14116bf50639b214d8414b5bbaaa6',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a5138d5cdc18139e135707916a243cd8e',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], - ['simd_5fshuffle_5fdown_64',['simd_shuffle_down',['../namespacemetal.html#af6e2dd7ae087aba6abac4f0350b7611c',1,'metal::simd_shuffle_down()'],['../backend_2metal_2kernels_2utils_8h.html#aba6279624b1d30c525efee856a222b5c',1,'simd_shuffle_down(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0c1e4d782fcc56e1ab5565cef12430dd',1,'simd_shuffle_down(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a48ae83a8caf5c74810df60b6c6cdb062',1,'simd_shuffle_down(bool data, uint16_t delta): utils.h']]], - ['simd_5fshuffle_5frotate_5fdown_65',['simd_shuffle_rotate_down',['../namespacemetal.html#a4bb203647a421032db47e73cd649841b',1,'metal']]], - ['simd_5fshuffle_5frotate_5fup_66',['simd_shuffle_rotate_up',['../namespacemetal.html#a729b22077d6c944491a6027c18ea80c9',1,'metal']]], - ['simd_5fshuffle_5fup_67',['simd_shuffle_up',['../namespacemetal.html#afe81c5fbde3f4890458b081909242c55',1,'metal']]], - ['simd_5fshuffle_5fxor_68',['simd_shuffle_xor',['../namespacemetal.html#a5017efc9605e069cfb507137cd1a1852',1,'metal']]], - ['simd_5fsum_69',['simd_sum',['../namespacemetal.html#a85181e37a00cb4a4217f1bb25389bce5',1,'metal']]], - ['simd_5fxor_70',['simd_xor',['../namespacemetal.html#a1308decbf2d5c33d34d6be523ea1c30f',1,'metal']]], - ['simple_5fiter_71',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398',1,'pocketfft::detail::simple_iter']]], - ['sin_72',['Sin',['../classmlx_1_1core_1_1_sin.html#a10d1ecc0ca96e79cdf55b57073d126ea',1,'mlx::core::Sin']]], - ['sin_73',['sin',['../namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258',1,'pocketfft::detail::sin()'],['../namespacemetal.html#a619a159ca5f2ddfe3647d3a6bb6e804c',1,'metal::sin()'],['../namespacemetal_1_1fast.html#a3af771cfe7a135104f9d063147dba270',1,'metal::fast::sin()'],['../namespacemetal_1_1precise.html#a71acf77ffd29c56f56afae0195c98a1c',1,'metal::precise::sin()'],['../group__ops.html#gaebf0a73ad3732fba39df37826c235692',1,'mlx::core::sin()']]], - ['sincos_5f2pibyn_74',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40',1,'pocketfft::detail::sincos_2pibyn']]], - ['sinh_75',['Sinh',['../classmlx_1_1core_1_1_sinh.html#a4a4f6814d403c2ce5d6c574b0dca3c96',1,'mlx::core::Sinh']]], - ['sinh_76',['sinh',['../namespacemetal.html#a83ba4235ae350ab8880a9df09158620b',1,'metal::sinh()'],['../namespacemetal_1_1fast.html#a990d90b3440e38d1fb4ff5065c6c189b',1,'metal::fast::sinh()'],['../namespacemetal_1_1precise.html#abc8f4f59dd6e7204ab5d84f0af96331c',1,'metal::precise::sinh()'],['../group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7',1,'mlx::core::sinh()']]], - ['sinpi_77',['sinpi',['../namespacemetal.html#ae9655f7fa2ba6c0625ca25fbb278e269',1,'metal::sinpi()'],['../namespacemetal_1_1fast.html#ab07a32fe544aa304577d29e0251e87b2',1,'metal::fast::sinpi()'],['../namespacemetal_1_1precise.html#a78b17dab93519d9c82c2575dafec49c9',1,'metal::precise::sinpi()']]], - ['size_78',['size',['../classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952',1,'pocketfft::detail::arr::size()'],['../classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5',1,'pocketfft::detail::arr_info::size()'],['../classmlx_1_1core_1_1array.html#a598f87161926d9e0b516860f0ea2c8f6',1,'mlx::core::array::size()'],['../structmlx_1_1core_1_1distributed_1_1_group.html#abd96a09217e3d1bcc522888257d22cef',1,'mlx::core::distributed::Group::size()']]], - ['size_5fof_79',['size_of',['../namespacemlx_1_1core.html#add4794cc0ffe5d717fc146084a235d95',1,'mlx::core']]], - ['slice_80',['Slice',['../classmlx_1_1core_1_1_slice.html#a8a38feb7bb6b72bdeebb83f053e2fd7f',1,'mlx::core::Slice']]], - ['slice_81',['slice',['../group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21',1,'mlx::core::slice(const array &a, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})'],['../group__ops.html#ga90b243e25b5fbbd7a9732f7ac48f6403',1,'mlx::core::slice(const array &a, const std::vector< int > &start, const std::vector< int > &stop, StreamOrDevice s={})']]], - ['slice_5fgpu_82',['slice_gpu',['../namespacemlx_1_1core.html#a5ab133d67da9c16621f61690e27fccfd',1,'mlx::core']]], - ['slice_5fupdate_83',['slice_update',['../group__ops.html#ga3875660e4ce2c8add8bfcf8144078708',1,'mlx::core::slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})'],['../group__ops.html#ga03ffbbb4d989a463ef43f41ebf7eabef',1,'mlx::core::slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, StreamOrDevice s={})']]], - ['sliceupdate_84',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html#aa30a7f22f557c56e1a2b5fcf44488990',1,'mlx::core::SliceUpdate']]], - ['softmax_85',['Softmax',['../classmlx_1_1core_1_1_softmax.html#a4ec686aac4e06f0dfe2cbd6801af40eb',1,'mlx::core::Softmax']]], - ['softmax_86',['softmax',['../namespacemlx_1_1core_1_1metal.html#a4fe937c2c584fd646926057f31d54ca6',1,'mlx::core::metal::softmax()'],['../group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6',1,'mlx::core::softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga1ae3614d07d873892a530d14c3857d0b',1,'mlx::core::softmax(const array &a, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga06f570d73716a24303e6de3aaba4457b',1,'mlx::core::softmax(const array &a, int axis, bool precise=false, StreamOrDevice s={})']]], - ['softmax_5fexp_87',['softmax_exp',['../kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b',1,'softmax.h']]], - ['softmax_5flooped_88',['softmax_looped',['../kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276',1,'softmax.h']]], - ['softmax_5fsingle_5frow_89',['softmax_single_row',['../kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b',1,'softmax.h']]], - ['sort_90',['Sort',['../classmlx_1_1core_1_1_sort.html#a62943032dbd72e85ceb9b4b7211f4a44',1,'mlx::core::Sort']]], - ['sort_91',['sort',['../struct_thread_sort.html#ad9ab3e6b47f7e9b91c0f3b773596986d',1,'ThreadSort::sort()'],['../struct_block_merge_sort.html#acc970f5eb963f7f2010f5ae5ea8b8bc0',1,'BlockMergeSort::sort()'],['../namespacemlx_1_1core_1_1metal.html#ab77c9a9ecaeeab8c66b712862777c24b',1,'mlx::core::metal::sort()'],['../group__ops.html#ga7fb616054665b3c2d61fa234f501f079',1,'mlx::core::sort(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaae1bc47aa737f705d0e5884270063fea',1,'mlx::core::sort(const array &a, int axis, StreamOrDevice s={})']]], - ['special_5fmul_92',['special_mul',['../structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874',1,'pocketfft::detail::cmplx::special_mul()'],['../namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76',1,'pocketfft::detail::special_mul()']]], - ['split_93',['Split',['../classmlx_1_1core_1_1_split.html#a897c746ecfdff5119cc5ae3f20499385',1,'mlx::core::Split']]], - ['split_94',['split',['../structmlx_1_1core_1_1distributed_1_1_group.html#abbf40f8979488806bc5bca9ecc4130e9',1,'mlx::core::distributed::Group::split()'],['../group__ops.html#ga7534290bceab5fb3831a05d67bebce7d',1,'mlx::core::split(const array &a, int num_splits, int axis, StreamOrDevice s={})'],['../group__ops.html#ga56882d24e5fde59c266774624c892d41',1,'mlx::core::split(const array &a, int num_splits, StreamOrDevice s={})'],['../group__ops.html#ga2cfcb1a53924882e30476c9016c5de74',1,'mlx::core::split(const array &a, const std::vector< int > &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#gac324dfa3e26d3a14a35ab7962e36f0e1',1,'mlx::core::split(const array &a, const std::vector< int > &indices, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a42847b435d037a977592e355eed072af',1,'mlx::core::random::split(const array &key, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7ec057064c7326c41b536f08178861e5',1,'mlx::core::random::split(const array &key, int num, StreamOrDevice s={})']]], - ['sqrt_95',['Sqrt',['../classmlx_1_1core_1_1_sqrt.html#a6682a7c31ca427c9d2c5ddb6a479bf29',1,'mlx::core::Sqrt']]], - ['sqrt_96',['sqrt',['../namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3',1,'pocketfft::detail::sqrt()'],['../namespacemetal.html#ab3f4d4852ca0e591104fbd8e5b50d31b',1,'metal::sqrt()'],['../namespacemetal_1_1fast.html#a4218a85c7d8a74cb8055b4755205627e',1,'metal::fast::sqrt()'],['../namespacemetal_1_1precise.html#acb213467361cd2cab93a8d5ea1aa5bfd',1,'metal::precise::sqrt()'],['../group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1',1,'mlx::core::sqrt()']]], - ['square_97',['Square',['../classmlx_1_1core_1_1_square.html#ab94e28d5c92e6febc1c74e525f730dc4',1,'mlx::core::Square']]], - ['square_98',['square',['../group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e',1,'mlx::core']]], - ['squeeze_99',['squeeze',['../group__ops.html#ga710daa7ec721bd4d3f326082cb195576',1,'mlx::core::squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3',1,'mlx::core::squeeze(const array &a, int axis, StreamOrDevice s={})'],['../group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad',1,'mlx::core::squeeze(const array &a, StreamOrDevice s={})']]], - ['stack_100',['stack',['../group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032',1,'mlx::core::stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga82216209dce901296fc737fe8efa5c94',1,'mlx::core::stack(const std::vector< array > &arrays, StreamOrDevice s={})']]], - ['start_5fcapture_101',['start_capture',['../namespacemlx_1_1core_1_1metal.html#aa47cb5651bf3b65c46ab216b7e504d77',1,'mlx::core::metal']]], - ['start_5fconcurrent_102',['start_concurrent',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::metal::CommandEncoder']]], - ['status_103',['status',['../classmlx_1_1core_1_1array.html#a7102659be87e9ef62966696ab9b07dad',1,'mlx::core::array']]], - ['std_104',['std',['../group__ops.html#ga2a466024f8061febc0a64be557644cb0',1,'mlx::core::std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#gafdcb04d77c64405a3990078a77dd984c',1,'mlx::core::std(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8',1,'mlx::core::std(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga62721a206df671ef5797449eea97af9f',1,'mlx::core::std(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], - ['steel_5fconv_105',['steel_conv',['../namespacemlx_1_1core_1_1metal.html#a92f1e559b1121d545746f81ff86eaca1',1,'mlx::core::metal']]], - ['steel_5fconv_5fgeneral_106',['steel_conv_general',['../namespacemlx_1_1core_1_1metal.html#a02edb6a90bdf30f4c9f0d6c25b0267b5',1,'mlx::core::metal']]], - ['steel_5fgemm_5ffused_107',['steel_gemm_fused',['../namespacemlx_1_1core_1_1metal.html#a17764366deed71c160fb26091400a803',1,'mlx::core::metal']]], - ['steel_5fgemm_5fmasked_108',['steel_gemm_masked',['../namespacemlx_1_1core_1_1metal.html#a962272ca73d26c08f76f706a128fd71f',1,'mlx::core::metal']]], - ['steel_5fgemm_5fsplitk_109',['steel_gemm_splitk',['../namespacemlx_1_1core_1_1metal.html#ad0dfd40ba7c09755711ceb731e57a5ac',1,'mlx::core::metal']]], - ['steel_5fmatmul_110',['steel_matmul',['../namespacemlx_1_1core.html#ab43a7633794498e1c6775cca829eb886',1,'mlx::core']]], - ['steel_5fmatmul_5fconv_5fgroups_111',['steel_matmul_conv_groups',['../namespacemlx_1_1core.html#afe8386cea0c7b846dc78649927fd0c75',1,'mlx::core']]], - ['stop_5fcapture_112',['stop_capture',['../namespacemlx_1_1core_1_1metal.html#ac90714424e36fb01e04550de69b8314f',1,'mlx::core::metal']]], - ['stop_5fgradient_113',['stop_gradient',['../group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f',1,'mlx::core']]], - ['stopgradient_114',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html#ac70d1ab819d04e00f76bc25aeebaf84f',1,'mlx::core::StopGradient']]], - ['store_5fresult_115',['store_result',['../structmlx_1_1steel_1_1_block_m_m_a.html#a85c57ad4092079f068683cb525fb08ee',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const']]], - ['store_5fresult_5fsafe_116',['store_result_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#af94bafeb285c240051e86418fccb89fe',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const']]], - ['stream_117',['stream',['../classmlx_1_1core_1_1_event.html#a193143bad31b68c699fa27f135b45614',1,'mlx::core::Event::stream()'],['../classmlx_1_1core_1_1_primitive.html#a46e6257397a662528f9f831842ac456a',1,'mlx::core::Primitive::stream()']]], - ['stream_118',['Stream',['../structmlx_1_1core_1_1_stream.html#a7f0815ff4886da74cbbff5f93d82dd3e',1,'mlx::core::Stream']]], - ['streamcontext_119',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html#a89d803151e9d7dce29382aa83d5c6ef1',1,'mlx::core::StreamContext']]], - ['streamthread_120',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#ac528109a11abcb82e6e221c5efa4493c',1,'mlx::core::scheduler::StreamThread']]], - ['stride_121',['stride',['../classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093',1,'pocketfft::detail::arr_info::stride() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b',1,'pocketfft::detail::arr_info::stride(size_t i) const']]], - ['stride_5fin_122',['stride_in',['../classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c',1,'pocketfft::detail::multi_iter']]], - ['stride_5fout_123',['stride_out',['../classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7',1,'pocketfft::detail::multi_iter']]], - ['strided_5freduce_5fgeneral_5fdispatch_124',['strided_reduce_general_dispatch',['../namespacemlx_1_1core.html#aa0332c64ee9965f05026c30a0b778000',1,'mlx::core']]], - ['strided_5fscan_125',['strided_scan',['../kernels_2scan_8h.html#aef54ac73154fb9024eda760a9dfde607',1,'scan.h']]], - ['strides_126',['strides',['../classmlx_1_1core_1_1array.html#a186cf2648da92584d5c1c8b24e69629b',1,'mlx::core::array::strides() const'],['../classmlx_1_1core_1_1array.html#a919f850ca087d1c40aa68f854cb30be2',1,'mlx::core::array::strides(int dim) const']]], - ['submit_127',['submit',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4',1,'pocketfft::detail::threading::thread_pool']]], - ['subtract_128',['Subtract',['../classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c',1,'mlx::core::Subtract']]], - ['subtract_129',['subtract',['../group__ops.html#ga196c240d3d0fcbb4713802c485e15133',1,'mlx::core']]], - ['sum_130',['sum',['../group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6',1,'mlx::core::sum(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3627754d7868487bdab1bd83f05d9c81',1,'mlx::core::sum(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4',1,'mlx::core::sum(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d',1,'mlx::core::sum(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], - ['svd_131',['SVD',['../classmlx_1_1core_1_1_s_v_d.html#ae89ff583e34fa894cccb8e7a475ee6d1',1,'mlx::core::SVD']]], - ['svd_132',['svd',['../namespacemlx_1_1core_1_1linalg.html#a64364b880e99914cf47bf756fa8dbaf0',1,'mlx::core::linalg']]], - ['swapaxes_133',['swapaxes',['../group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb',1,'mlx::core']]], - ['swizzle_134',['swizzle',['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle']]], - ['synchronize_135',['synchronize',['../namespacemlx_1_1core.html#a14287949d82ffefad0306cef5eb5f9e4',1,'mlx::core::synchronize()'],['../namespacemlx_1_1core.html#a6648a71937b055e5ff513d98056c2fb5',1,'mlx::core::synchronize(Stream)']]] + ['set_5fvector_5fbytes_34',['set_vector_bytes',['../namespacemlx_1_1core.html#a62340bbaa8b216539688a60adcb568bf',1,'mlx::core::set_vector_bytes(CommandEncoder &enc, const std::vector< T > &vec, size_t nelems, int idx)'],['../namespacemlx_1_1core.html#ae309cb543dfb0239cfccc53a8ad0408e',1,'mlx::core::set_vector_bytes(CommandEncoder &enc, const std::vector< T > &vec, int idx)']]], + ['shape_35',['shape',['../classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0',1,'pocketfft::detail::arr_info::shape() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8',1,'pocketfft::detail::arr_info::shape(size_t i) const'],['../classmlx_1_1core_1_1array.html#a4a2a2c8a4a5beafd723fc13f2055d55d',1,'mlx::core::array::shape() const'],['../classmlx_1_1core_1_1array.html#a51ed0c45666264dc172d06fba159eb8f',1,'mlx::core::array::shape(int dim) const']]], + ['shapes_5fwithout_5freduction_5faxes_36',['shapes_without_reduction_axes',['../namespacemlx_1_1core.html#a44c3ea6db6553c3f6552b9ba64a69494',1,'mlx::core']]], + ['shared_5fbuffer_5fslice_37',['shared_buffer_slice',['../namespacemlx_1_1core.html#a427f2c255dfc6e1f83f97587b08e71bc',1,'mlx::core']]], + ['shutdown_38',['shutdown',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411',1,'pocketfft::detail::threading::thread_pool']]], + ['siblings_39',['siblings',['../classmlx_1_1core_1_1array.html#acf80fde8f743f65ad5b4be69fcb7a74d',1,'mlx::core::array::siblings() const'],['../classmlx_1_1core_1_1array.html#a7263f23e70a580a9bc2129fbcde36e6c',1,'mlx::core::array::siblings()']]], + ['sigmoid_40',['Sigmoid',['../classmlx_1_1core_1_1_sigmoid.html#a47eca99113ec19f0eb60b6a0472c592b',1,'mlx::core::Sigmoid']]], + ['sigmoid_41',['sigmoid',['../group__ops.html#ga708abf8f79609cd6831db7c38cafac0e',1,'mlx::core']]], + ['sign_42',['Sign',['../classmlx_1_1core_1_1_sign.html#afe951e50907bc23a601ec5fa9eae5763',1,'mlx::core::Sign']]], + ['sign_43',['sign',['../group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa',1,'mlx::core']]], + ['signal_44',['signal',['../classmlx_1_1core_1_1_event.html#a65a858445506a61be5889ae0e3651b89',1,'mlx::core::Event']]], + ['signaling_5fnan_45',['signaling_NaN',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#ad1f76a43c7d51a3765174aa6e0dd9f80',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['simd_5fbroadcast_46',['simd_broadcast',['../namespacemetal.html#a498f1e85107eb5f01ba4435977f8efe0',1,'metal']]], + ['simd_5fexclusive_5fscan_47',['simd_exclusive_scan',['../struct_cum_sum.html#a64d1933942de4df2d44dd90e9e5e4460',1,'CumSum::simd_exclusive_scan()'],['../struct_cum_prod.html#a5b0577295853a48856b84fd4f935dff8',1,'CumProd::simd_exclusive_scan()'],['../struct_cum_prod_3_01bool_01_4.html#a1a86e9398bae24182b7be0a6577bf223',1,'CumProd< bool >::simd_exclusive_scan()'],['../struct_cum_max.html#ae11b67aa6c998e9a01615b2a79af4403',1,'CumMax::simd_exclusive_scan()'],['../struct_cum_min.html#a83e65017ff33018b585c043fb803773b',1,'CumMin::simd_exclusive_scan()']]], + ['simd_5fmax_48',['simd_max',['../namespacemetal.html#a048cad0aca52cb737ebf103e76bd1c49',1,'metal']]], + ['simd_5fmin_49',['simd_min',['../namespacemetal.html#ae9e2a23e00724ba2d7868bc4112b386b',1,'metal']]], + ['simd_5fprefix_5fexclusive_5fproduct_50',['simd_prefix_exclusive_product',['../namespacemetal.html#a5ca40242390b632f737e29636829b2e4',1,'metal']]], + ['simd_5fprefix_5fexclusive_5fsum_51',['simd_prefix_exclusive_sum',['../namespacemetal.html#abfbb70c7471f28bf7ff36a612ad014b2',1,'metal']]], + ['simd_5fprefix_5finclusive_5fproduct_52',['simd_prefix_inclusive_product',['../namespacemetal.html#a6ca6a7e1996228fa536e969e9e45c446',1,'metal']]], + ['simd_5fprefix_5finclusive_5fsum_53',['simd_prefix_inclusive_sum',['../namespacemetal.html#a567acb18199ac0107712eb8cb8aeb8e9',1,'metal']]], + ['simd_5fproduct_54',['simd_product',['../namespacemetal.html#ac6e883a04e2265a9790d7db76059e1b4',1,'metal']]], + ['simd_5freduce_55',['simd_reduce',['../struct_and.html#ab4df5517e050cfa7257c34c6b0bd980b',1,'And::simd_reduce()'],['../struct_or.html#aef6b5abbbf3fd22981b8d27562022605',1,'Or::simd_reduce()'],['../struct_sum.html#acf10e0604098391526da3b8271928b5c',1,'Sum::simd_reduce()'],['../struct_prod.html#ab0d7a38360b971b4d99c4ad2db310069',1,'Prod::simd_reduce()'],['../struct_min.html#afe8d7cefac624fd4267c1b252ac38528',1,'Min::simd_reduce()'],['../struct_max.html#a4f74440c6f1ce8247d5a87c2a8882080',1,'Max::simd_reduce()']]], + ['simd_5fscan_56',['simd_scan',['../struct_cum_sum.html#a83342af06f6a700067e3b451636852d6',1,'CumSum::simd_scan()'],['../struct_cum_prod.html#a860dbf6146de088d07175bd165b74004',1,'CumProd::simd_scan()'],['../struct_cum_prod_3_01bool_01_4.html#abeb5ec4237b330e7219f4e881cf10d7a',1,'CumProd< bool >::simd_scan()'],['../struct_cum_max.html#adc9ec8bb09b4433d4c2f03022c43d781',1,'CumMax::simd_scan()'],['../struct_cum_min.html#a0a1005d91b1c90e90e2c6dbd6c296649',1,'CumMin::simd_scan()']]], + ['simd_5fshuffle_57',['simd_shuffle',['../namespacemetal.html#a259ed115bc3c58f88eb35830916b26d4',1,'metal']]], + ['simd_5fshuffle_5fand_5ffill_5fdown_58',['simd_shuffle_and_fill_down',['../namespacemetal.html#ae29a06f0eac636ad7af21dea5b04938b',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a0ee6239fa29a5f9ee0201e0dc5ddc8e0',1,'metal::simd_shuffle_and_fill_down(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], + ['simd_5fshuffle_5fand_5ffill_5fup_59',['simd_shuffle_and_fill_up',['../namespacemetal.html#a1ca14116bf50639b214d8414b5bbaaa6',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta, ushort modulo)'],['../namespacemetal.html#a5138d5cdc18139e135707916a243cd8e',1,'metal::simd_shuffle_and_fill_up(bfloat16_t data, bfloat16_t filling_data, ushort delta)']]], + ['simd_5fshuffle_5fdown_60',['simd_shuffle_down',['../namespacemetal.html#af6e2dd7ae087aba6abac4f0350b7611c',1,'metal::simd_shuffle_down()'],['../backend_2metal_2kernels_2utils_8h.html#aba6279624b1d30c525efee856a222b5c',1,'simd_shuffle_down(uint64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0c1e4d782fcc56e1ab5565cef12430dd',1,'simd_shuffle_down(int64_t data, uint16_t delta): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a48ae83a8caf5c74810df60b6c6cdb062',1,'simd_shuffle_down(bool data, uint16_t delta): utils.h']]], + ['simd_5fshuffle_5frotate_5fdown_61',['simd_shuffle_rotate_down',['../namespacemetal.html#a4bb203647a421032db47e73cd649841b',1,'metal']]], + ['simd_5fshuffle_5frotate_5fup_62',['simd_shuffle_rotate_up',['../namespacemetal.html#a729b22077d6c944491a6027c18ea80c9',1,'metal']]], + ['simd_5fshuffle_5fup_63',['simd_shuffle_up',['../namespacemetal.html#afe81c5fbde3f4890458b081909242c55',1,'metal']]], + ['simd_5fshuffle_5fxor_64',['simd_shuffle_xor',['../namespacemetal.html#a5017efc9605e069cfb507137cd1a1852',1,'metal']]], + ['simd_5fsum_65',['simd_sum',['../namespacemetal.html#a85181e37a00cb4a4217f1bb25389bce5',1,'metal']]], + ['simd_5fxor_66',['simd_xor',['../namespacemetal.html#a1308decbf2d5c33d34d6be523ea1c30f',1,'metal']]], + ['simple_5fiter_67',['simple_iter',['../classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398',1,'pocketfft::detail::simple_iter']]], + ['sin_68',['Sin',['../classmlx_1_1core_1_1_sin.html#a10d1ecc0ca96e79cdf55b57073d126ea',1,'mlx::core::Sin']]], + ['sin_69',['sin',['../namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258',1,'pocketfft::detail::sin()'],['../namespacemetal.html#a619a159ca5f2ddfe3647d3a6bb6e804c',1,'metal::sin()'],['../namespacemetal_1_1fast.html#a3af771cfe7a135104f9d063147dba270',1,'metal::fast::sin()'],['../namespacemetal_1_1precise.html#a71acf77ffd29c56f56afae0195c98a1c',1,'metal::precise::sin()'],['../group__ops.html#gaebf0a73ad3732fba39df37826c235692',1,'mlx::core::sin()']]], + ['sincos_5f2pibyn_70',['sincos_2pibyn',['../classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40',1,'pocketfft::detail::sincos_2pibyn']]], + ['sinh_71',['Sinh',['../classmlx_1_1core_1_1_sinh.html#a4a4f6814d403c2ce5d6c574b0dca3c96',1,'mlx::core::Sinh']]], + ['sinh_72',['sinh',['../namespacemetal.html#a83ba4235ae350ab8880a9df09158620b',1,'metal::sinh()'],['../namespacemetal_1_1fast.html#a990d90b3440e38d1fb4ff5065c6c189b',1,'metal::fast::sinh()'],['../namespacemetal_1_1precise.html#abc8f4f59dd6e7204ab5d84f0af96331c',1,'metal::precise::sinh()'],['../group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7',1,'mlx::core::sinh()']]], + ['sinpi_73',['sinpi',['../namespacemetal.html#ae9655f7fa2ba6c0625ca25fbb278e269',1,'metal::sinpi()'],['../namespacemetal_1_1fast.html#ab07a32fe544aa304577d29e0251e87b2',1,'metal::fast::sinpi()'],['../namespacemetal_1_1precise.html#a78b17dab93519d9c82c2575dafec49c9',1,'metal::precise::sinpi()']]], + ['size_74',['size',['../classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952',1,'pocketfft::detail::arr::size()'],['../classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5',1,'pocketfft::detail::arr_info::size()'],['../classmlx_1_1core_1_1array.html#a598f87161926d9e0b516860f0ea2c8f6',1,'mlx::core::array::size()'],['../structmlx_1_1core_1_1distributed_1_1_group.html#abd96a09217e3d1bcc522888257d22cef',1,'mlx::core::distributed::Group::size()']]], + ['size_5fof_75',['size_of',['../namespacemlx_1_1core.html#add4794cc0ffe5d717fc146084a235d95',1,'mlx::core']]], + ['slice_76',['Slice',['../classmlx_1_1core_1_1_slice.html#a8a38feb7bb6b72bdeebb83f053e2fd7f',1,'mlx::core::Slice']]], + ['slice_77',['slice',['../group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21',1,'mlx::core::slice(const array &a, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})'],['../group__ops.html#ga90b243e25b5fbbd7a9732f7ac48f6403',1,'mlx::core::slice(const array &a, const std::vector< int > &start, const std::vector< int > &stop, StreamOrDevice s={})']]], + ['slice_5fgpu_78',['slice_gpu',['../namespacemlx_1_1core.html#a5ab133d67da9c16621f61690e27fccfd',1,'mlx::core']]], + ['slice_5fupdate_79',['slice_update',['../group__ops.html#ga3875660e4ce2c8add8bfcf8144078708',1,'mlx::core::slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})'],['../group__ops.html#ga03ffbbb4d989a463ef43f41ebf7eabef',1,'mlx::core::slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, StreamOrDevice s={})']]], + ['sliceupdate_80',['SliceUpdate',['../classmlx_1_1core_1_1_slice_update.html#aa30a7f22f557c56e1a2b5fcf44488990',1,'mlx::core::SliceUpdate']]], + ['softmax_81',['Softmax',['../classmlx_1_1core_1_1_softmax.html#a4ec686aac4e06f0dfe2cbd6801af40eb',1,'mlx::core::Softmax']]], + ['softmax_82',['softmax',['../namespacemlx_1_1core_1_1metal.html#a4fe937c2c584fd646926057f31d54ca6',1,'mlx::core::metal::softmax()'],['../group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6',1,'mlx::core::softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga1ae3614d07d873892a530d14c3857d0b',1,'mlx::core::softmax(const array &a, bool precise=false, StreamOrDevice s={})'],['../group__ops.html#ga06f570d73716a24303e6de3aaba4457b',1,'mlx::core::softmax(const array &a, int axis, bool precise=false, StreamOrDevice s={})']]], + ['softmax_5fexp_83',['softmax_exp',['../kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b',1,'softmax.h']]], + ['softmax_5flooped_84',['softmax_looped',['../kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276',1,'softmax.h']]], + ['softmax_5fsingle_5frow_85',['softmax_single_row',['../kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b',1,'softmax.h']]], + ['sort_86',['Sort',['../classmlx_1_1core_1_1_sort.html#a62943032dbd72e85ceb9b4b7211f4a44',1,'mlx::core::Sort']]], + ['sort_87',['sort',['../struct_thread_sort.html#ad9ab3e6b47f7e9b91c0f3b773596986d',1,'ThreadSort::sort()'],['../struct_block_merge_sort.html#acc970f5eb963f7f2010f5ae5ea8b8bc0',1,'BlockMergeSort::sort()'],['../namespacemlx_1_1core_1_1metal.html#ab77c9a9ecaeeab8c66b712862777c24b',1,'mlx::core::metal::sort()'],['../group__ops.html#ga7fb616054665b3c2d61fa234f501f079',1,'mlx::core::sort(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaae1bc47aa737f705d0e5884270063fea',1,'mlx::core::sort(const array &a, int axis, StreamOrDevice s={})']]], + ['special_5fmul_88',['special_mul',['../structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874',1,'pocketfft::detail::cmplx::special_mul()'],['../namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76',1,'pocketfft::detail::special_mul()']]], + ['split_89',['Split',['../classmlx_1_1core_1_1_split.html#a897c746ecfdff5119cc5ae3f20499385',1,'mlx::core::Split']]], + ['split_90',['split',['../structmlx_1_1core_1_1distributed_1_1_group.html#abbf40f8979488806bc5bca9ecc4130e9',1,'mlx::core::distributed::Group::split()'],['../group__ops.html#ga7534290bceab5fb3831a05d67bebce7d',1,'mlx::core::split(const array &a, int num_splits, int axis, StreamOrDevice s={})'],['../group__ops.html#ga56882d24e5fde59c266774624c892d41',1,'mlx::core::split(const array &a, int num_splits, StreamOrDevice s={})'],['../group__ops.html#ga2cfcb1a53924882e30476c9016c5de74',1,'mlx::core::split(const array &a, const std::vector< int > &indices, int axis, StreamOrDevice s={})'],['../group__ops.html#gac324dfa3e26d3a14a35ab7962e36f0e1',1,'mlx::core::split(const array &a, const std::vector< int > &indices, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a42847b435d037a977592e355eed072af',1,'mlx::core::random::split(const array &key, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a7ec057064c7326c41b536f08178861e5',1,'mlx::core::random::split(const array &key, int num, StreamOrDevice s={})']]], + ['sqrt_91',['Sqrt',['../classmlx_1_1core_1_1_sqrt.html#a6682a7c31ca427c9d2c5ddb6a479bf29',1,'mlx::core::Sqrt']]], + ['sqrt_92',['sqrt',['../namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3',1,'pocketfft::detail::sqrt()'],['../namespacemetal.html#ab3f4d4852ca0e591104fbd8e5b50d31b',1,'metal::sqrt()'],['../namespacemetal_1_1fast.html#a4218a85c7d8a74cb8055b4755205627e',1,'metal::fast::sqrt()'],['../namespacemetal_1_1precise.html#acb213467361cd2cab93a8d5ea1aa5bfd',1,'metal::precise::sqrt()'],['../group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1',1,'mlx::core::sqrt()']]], + ['square_93',['Square',['../classmlx_1_1core_1_1_square.html#ab94e28d5c92e6febc1c74e525f730dc4',1,'mlx::core::Square']]], + ['square_94',['square',['../group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e',1,'mlx::core']]], + ['squeeze_95',['squeeze',['../group__ops.html#ga710daa7ec721bd4d3f326082cb195576',1,'mlx::core::squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3',1,'mlx::core::squeeze(const array &a, int axis, StreamOrDevice s={})'],['../group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad',1,'mlx::core::squeeze(const array &a, StreamOrDevice s={})']]], + ['stack_96',['stack',['../group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032',1,'mlx::core::stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#ga82216209dce901296fc737fe8efa5c94',1,'mlx::core::stack(const std::vector< array > &arrays, StreamOrDevice s={})']]], + ['start_5fcapture_97',['start_capture',['../namespacemlx_1_1core_1_1metal.html#aa47cb5651bf3b65c46ab216b7e504d77',1,'mlx::core::metal']]], + ['start_5fconcurrent_98',['start_concurrent',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a48b548a0b15f9d1279c938a1c6167034',1,'mlx::core::metal::CommandEncoder']]], + ['status_99',['status',['../classmlx_1_1core_1_1array.html#a7102659be87e9ef62966696ab9b07dad',1,'mlx::core::array']]], + ['std_100',['std',['../group__ops.html#ga2a466024f8061febc0a64be557644cb0',1,'mlx::core::std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#gafdcb04d77c64405a3990078a77dd984c',1,'mlx::core::std(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8',1,'mlx::core::std(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga62721a206df671ef5797449eea97af9f',1,'mlx::core::std(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], + ['steel_5fconv_101',['steel_conv',['../namespacemlx_1_1core_1_1metal.html#a92f1e559b1121d545746f81ff86eaca1',1,'mlx::core::metal']]], + ['steel_5fconv_5fgeneral_102',['steel_conv_general',['../namespacemlx_1_1core_1_1metal.html#a02edb6a90bdf30f4c9f0d6c25b0267b5',1,'mlx::core::metal']]], + ['steel_5fgemm_5ffused_103',['steel_gemm_fused',['../namespacemlx_1_1core_1_1metal.html#a17764366deed71c160fb26091400a803',1,'mlx::core::metal']]], + ['steel_5fgemm_5fmasked_104',['steel_gemm_masked',['../namespacemlx_1_1core_1_1metal.html#a962272ca73d26c08f76f706a128fd71f',1,'mlx::core::metal']]], + ['steel_5fgemm_5fsplitk_105',['steel_gemm_splitk',['../namespacemlx_1_1core_1_1metal.html#ad0dfd40ba7c09755711ceb731e57a5ac',1,'mlx::core::metal']]], + ['steel_5fmatmul_106',['steel_matmul',['../namespacemlx_1_1core.html#ab43a7633794498e1c6775cca829eb886',1,'mlx::core']]], + ['steel_5fmatmul_5fconv_5fgroups_107',['steel_matmul_conv_groups',['../namespacemlx_1_1core.html#afe8386cea0c7b846dc78649927fd0c75',1,'mlx::core']]], + ['stop_5fcapture_108',['stop_capture',['../namespacemlx_1_1core_1_1metal.html#ac90714424e36fb01e04550de69b8314f',1,'mlx::core::metal']]], + ['stop_5fgradient_109',['stop_gradient',['../group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f',1,'mlx::core']]], + ['stopgradient_110',['StopGradient',['../classmlx_1_1core_1_1_stop_gradient.html#ac70d1ab819d04e00f76bc25aeebaf84f',1,'mlx::core::StopGradient']]], + ['store_5fresult_111',['store_result',['../structmlx_1_1steel_1_1_block_m_m_a.html#a85c57ad4092079f068683cb525fb08ee',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7cf757e9785e23997b1417e024559ed3',1,'mlx::steel::BlockMMA::store_result(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, thread const Epilogue &epilogue_op) const']]], + ['store_5fresult_5fsafe_112',['store_result_safe',['../structmlx_1_1steel_1_1_block_m_m_a.html#af94bafeb285c240051e86418fccb89fe',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, short2 dst_tile_dims) const'],['../structmlx_1_1steel_1_1_block_m_m_a.html#a7b324c992750ed3aaa4c485f15b2f391',1,'mlx::steel::BlockMMA::store_result_safe(device U *D, const int ldd, const device U *C, const int ldc, const int fdc, short2 dst_tile_dims, thread const Epilogue &epilogue_op) const']]], + ['stream_113',['stream',['../classmlx_1_1core_1_1_event.html#a193143bad31b68c699fa27f135b45614',1,'mlx::core::Event::stream()'],['../classmlx_1_1core_1_1_primitive.html#a46e6257397a662528f9f831842ac456a',1,'mlx::core::Primitive::stream()']]], + ['stream_114',['Stream',['../structmlx_1_1core_1_1_stream.html#a7f0815ff4886da74cbbff5f93d82dd3e',1,'mlx::core::Stream']]], + ['streamcontext_115',['StreamContext',['../structmlx_1_1core_1_1_stream_context.html#a89d803151e9d7dce29382aa83d5c6ef1',1,'mlx::core::StreamContext']]], + ['streamthread_116',['StreamThread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#ac528109a11abcb82e6e221c5efa4493c',1,'mlx::core::scheduler::StreamThread']]], + ['stride_117',['stride',['../classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093',1,'pocketfft::detail::arr_info::stride() const'],['../classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b',1,'pocketfft::detail::arr_info::stride(size_t i) const']]], + ['stride_5fin_118',['stride_in',['../classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c',1,'pocketfft::detail::multi_iter']]], + ['stride_5fout_119',['stride_out',['../classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7',1,'pocketfft::detail::multi_iter']]], + ['strided_5freduce_5fgeneral_5fdispatch_120',['strided_reduce_general_dispatch',['../namespacemlx_1_1core.html#aa0332c64ee9965f05026c30a0b778000',1,'mlx::core']]], + ['strided_5fscan_121',['strided_scan',['../kernels_2scan_8h.html#aef54ac73154fb9024eda760a9dfde607',1,'scan.h']]], + ['strides_122',['strides',['../classmlx_1_1core_1_1array.html#a186cf2648da92584d5c1c8b24e69629b',1,'mlx::core::array::strides() const'],['../classmlx_1_1core_1_1array.html#a919f850ca087d1c40aa68f854cb30be2',1,'mlx::core::array::strides(int dim) const']]], + ['submit_123',['submit',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4',1,'pocketfft::detail::threading::thread_pool']]], + ['subtract_124',['Subtract',['../classmlx_1_1core_1_1_subtract.html#a834854757394f8de7082af65bf86ed9c',1,'mlx::core::Subtract']]], + ['subtract_125',['subtract',['../group__ops.html#ga196c240d3d0fcbb4713802c485e15133',1,'mlx::core']]], + ['sum_126',['sum',['../group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6',1,'mlx::core::sum(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga3627754d7868487bdab1bd83f05d9c81',1,'mlx::core::sum(const array &a, StreamOrDevice s={})'],['../group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4',1,'mlx::core::sum(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d',1,'mlx::core::sum(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], + ['svd_127',['SVD',['../classmlx_1_1core_1_1_s_v_d.html#ae89ff583e34fa894cccb8e7a475ee6d1',1,'mlx::core::SVD']]], + ['svd_128',['svd',['../namespacemlx_1_1core_1_1linalg.html#a64364b880e99914cf47bf756fa8dbaf0',1,'mlx::core::linalg']]], + ['swapaxes_129',['swapaxes',['../group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb',1,'mlx::core']]], + ['swizzle_130',['swizzle',['../structmlx_1_1steel_1_1_block_swizzle.html#a98e558d63826d2aaa06d3e65a06d2760',1,'mlx::steel::BlockSwizzle']]], + ['synchronize_131',['synchronize',['../namespacemlx_1_1core.html#a14287949d82ffefad0306cef5eb5f9e4',1,'mlx::core::synchronize()'],['../namespacemlx_1_1core.html#a6648a71937b055e5ff513d98056c2fb5',1,'mlx::core::synchronize(Stream)']]] ]; diff --git a/docs/build/html/search/functions_14.js b/docs/build/html/search/functions_14.js index c71c4eaca6..72ef570f24 100644 --- a/docs/build/html/search/functions_14.js +++ b/docs/build/html/search/functions_14.js @@ -23,26 +23,29 @@ var searchData= ['ternary_5fop_5fgpu_5finplace_20',['ternary_op_gpu_inplace',['../namespacemlx_1_1core.html#a37645c0adccb3eb46844115def1a68d7',1,'mlx::core']]], ['ternary_5fops_21',['ternary_ops',['../namespacemlx_1_1core_1_1metal.html#a11b593b07e9a33e5f78fe4695fb99ec9',1,'mlx::core::metal']]], ['ternary_5fv_22',['ternary_v',['../metal_2kernels_2ternary_8h.html#a83f93644d21ee774e06e8190d0725ccb',1,'ternary.h']]], - ['thread_5fcount_23',['thread_count',['../structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87',1,'pocketfft::detail::util']]], - ['thread_5ffn_24',['thread_fn',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a06a62c21c1174e4eb4d242e50aad7adf',1,'mlx::core::scheduler::StreamThread']]], - ['thread_5fid_25',['thread_id',['../namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b',1,'pocketfft::detail::threading']]], - ['thread_5fmap_26',['thread_map',['../namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e',1,'pocketfft::detail::threading']]], - ['thread_5fpool_27',['thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560',1,'pocketfft::detail::threading::thread_pool::thread_pool(size_t nthreads)'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0',1,'pocketfft::detail::threading::thread_pool::thread_pool()']]], - ['thread_5fswap_28',['thread_swap',['../sort_8h.html#a6e8c2da4975a8001fd5ddf211a3058b7',1,'sort.h']]], - ['threefry2x32_5fhash_29',['threefry2x32_hash',['../namespacemlx_1_1core_1_1random.html#ac7e92c89a2bac1b0bed922a3d4c3c66b',1,'mlx::core::random']]], - ['tile_30',['tile',['../group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361',1,'mlx::core']]], - ['to_5fbnns_5fdtype_31',['to_bnns_dtype',['../namespacemlx_1_1core.html#a7a4193f37b1de9c33c31d1da09c77edb',1,'mlx::core']]], - ['to_5fstream_32',['to_stream',['../namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9',1,'mlx::core']]], - ['topk_33',['topk',['../group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a',1,'mlx::core::topk(const array &a, int k, StreamOrDevice s={})'],['../group__ops.html#ga35b8436c79ff953f6c809598b646f498',1,'mlx::core::topk(const array &a, int k, int axis, StreamOrDevice s={})']]], - ['trace_34',['trace',['../group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, StreamOrDevice s={})'],['../group__ops.html#gaf25c00108feaafaa6350a4434cb0062e',1,'mlx::core::trace(const array &a, StreamOrDevice s={})']]], - ['transformadd_35',['TransformAdd',['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd']]], - ['transformaxpby_36',['TransformAxpby',['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby']]], - ['transpose_37',['Transpose',['../classmlx_1_1core_1_1_transpose.html#a1a9ba023584c61c7ac93d6dce536760a',1,'mlx::core::Transpose']]], - ['transpose_38',['transpose',['../group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b',1,'mlx::core::transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5',1,'mlx::core::transpose(const array &a, std::initializer_list< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a',1,'mlx::core::transpose(const array &a, StreamOrDevice s={})']]], - ['tri_39',['tri',['../group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78',1,'mlx::core::tri(int n, int m, int k, Dtype type, StreamOrDevice s={})'],['../group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5',1,'mlx::core::tri(int n, Dtype type, StreamOrDevice s={})']]], - ['tril_40',['tril',['../group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2',1,'mlx::core']]], - ['triu_41',['triu',['../group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412',1,'mlx::core']]], - ['trunc_42',['trunc',['../namespacemetal.html#a93cb75a11a362bfc8310ea19c554c887',1,'metal::trunc()'],['../namespacemetal_1_1fast.html#aa62e1075e86c626d97038f16e9433415',1,'metal::fast::trunc()'],['../namespacemetal_1_1precise.html#a334183e7a2dd49b983d072d1e8ee2b27',1,'metal::precise::trunc()']]], - ['truncated_5fnormal_43',['truncated_normal',['../namespacemlx_1_1core_1_1random.html#a00aa5746bac6d729d2ba9465153bb279',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a39663eda0fd7b274d01499a7b1c9035f',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['try_5fpop_44',['try_pop',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43',1,'pocketfft::detail::threading::concurrent_queue']]] + ['ternary_5fv2_23',['ternary_v2',['../metal_2kernels_2ternary_8h.html#a3e610f3b01966bdbf23fdfebe5d2c508',1,'ternary.h']]], + ['thread_5fcount_24',['thread_count',['../structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87',1,'pocketfft::detail::util']]], + ['thread_5ffn_25',['thread_fn',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a06a62c21c1174e4eb4d242e50aad7adf',1,'mlx::core::scheduler::StreamThread']]], + ['thread_5fid_26',['thread_id',['../namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b',1,'pocketfft::detail::threading']]], + ['thread_5fmap_27',['thread_map',['../namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e',1,'pocketfft::detail::threading']]], + ['thread_5fpool_28',['thread_pool',['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560',1,'pocketfft::detail::threading::thread_pool::thread_pool(size_t nthreads)'],['../classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0',1,'pocketfft::detail::threading::thread_pool::thread_pool()']]], + ['thread_5fswap_29',['thread_swap',['../sort_8h.html#a6e8c2da4975a8001fd5ddf211a3058b7',1,'sort.h']]], + ['threefry2x32_5fhash_30',['threefry2x32_hash',['../namespacemlx_1_1core_1_1random.html#ac7e92c89a2bac1b0bed922a3d4c3c66b',1,'mlx::core::random']]], + ['tile_31',['tile',['../group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361',1,'mlx::core']]], + ['to_5fbnns_5fdtype_32',['to_bnns_dtype',['../namespacemlx_1_1core.html#a7a4193f37b1de9c33c31d1da09c77edb',1,'mlx::core']]], + ['to_5fstream_33',['to_stream',['../namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9',1,'mlx::core']]], + ['topk_34',['topk',['../group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a',1,'mlx::core::topk(const array &a, int k, StreamOrDevice s={})'],['../group__ops.html#ga35b8436c79ff953f6c809598b646f498',1,'mlx::core::topk(const array &a, int k, int axis, StreamOrDevice s={})']]], + ['trace_35',['trace',['../group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd',1,'mlx::core::trace(const array &a, int offset, int axis1, int axis2, StreamOrDevice s={})'],['../group__ops.html#gaf25c00108feaafaa6350a4434cb0062e',1,'mlx::core::trace(const array &a, StreamOrDevice s={})']]], + ['transformadd_36',['TransformAdd',['../structmlx_1_1steel_1_1_transform_add.html#a7c1b7292910b74281e5296b3dac157ae',1,'mlx::steel::TransformAdd']]], + ['transformaxpby_37',['TransformAxpby',['../structmlx_1_1steel_1_1_transform_axpby.html#ad7d11c53de13646b725921391d15bbe9',1,'mlx::steel::TransformAxpby']]], + ['transpose_38',['Transpose',['../classmlx_1_1core_1_1_transpose.html#a1a9ba023584c61c7ac93d6dce536760a',1,'mlx::core::Transpose']]], + ['transpose_39',['transpose',['../group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b',1,'mlx::core::transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5',1,'mlx::core::transpose(const array &a, std::initializer_list< int > axes, StreamOrDevice s={})'],['../group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a',1,'mlx::core::transpose(const array &a, StreamOrDevice s={})']]], + ['tri_40',['tri',['../group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78',1,'mlx::core::tri(int n, int m, int k, Dtype type, StreamOrDevice s={})'],['../group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5',1,'mlx::core::tri(int n, Dtype type, StreamOrDevice s={})']]], + ['tri_5finv_41',['tri_inv',['../namespacemlx_1_1core_1_1linalg.html#aba1994571326326717b5b5e38c2e0661',1,'mlx::core::linalg']]], + ['tril_42',['tril',['../group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2',1,'mlx::core']]], + ['triu_43',['triu',['../group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412',1,'mlx::core']]], + ['trunc_44',['trunc',['../namespacemetal.html#a93cb75a11a362bfc8310ea19c554c887',1,'metal::trunc()'],['../namespacemetal_1_1fast.html#aa62e1075e86c626d97038f16e9433415',1,'metal::fast::trunc()'],['../namespacemetal_1_1precise.html#a334183e7a2dd49b983d072d1e8ee2b27',1,'metal::precise::trunc()']]], + ['truncated_5fnormal_45',['truncated_normal',['../namespacemlx_1_1core_1_1random.html#a00aa5746bac6d729d2ba9465153bb279',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a39663eda0fd7b274d01499a7b1c9035f',1,'mlx::core::random::truncated_normal(const array &lower, const array &upper, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['try_5fpop_46',['try_pop',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43',1,'pocketfft::detail::threading::concurrent_queue']]], + ['type_5fto_5fname_47',['type_to_name',['../namespacemlx_1_1core.html#af1fdfdaa5644394362e6baba30701bae',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/functions_15.js b/docs/build/html/search/functions_15.js index 40f99e8abf..afc15e7cab 100644 --- a/docs/build/html/search/functions_15.js +++ b/docs/build/html/search/functions_15.js @@ -6,9 +6,10 @@ var searchData= ['unary_5fop_5fgpu_5finplace_3',['unary_op_gpu_inplace',['../namespacemlx_1_1core.html#a668fde2bd280a88f63a68b68a343d375',1,'mlx::core']]], ['unary_5fops_4',['unary_ops',['../namespacemlx_1_1core_1_1metal.html#a17b471fa52ea5f24ee63e081f46528f5',1,'mlx::core::metal']]], ['unary_5fv_5',['unary_v',['../metal_2kernels_2unary_8h.html#ada52df4976078091cd7b47c152f326fb',1,'unary.h']]], - ['unaryprimitive_6',['UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html#a189f6d4ed369f82a4b724a29eb056d4e',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(Stream stream)'],['../classmlx_1_1core_1_1_unary_primitive.html#a9935cffc4f246d3d883bc3d26c5163f2',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a780281fb04e2daf1be630c124bd605e3',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(UnaryPrimitive &&other)=delete']]], - ['uniform_7',['Uniform',['../classmlx_1_1core_1_1_uniform.html#a626aa1091aa77b4a32c02290106b85e1',1,'mlx::core::Uniform']]], - ['uniform_8',['uniform',['../namespacemlx_1_1core_1_1random.html#adaa626cf75ab891978954bd1eb79a38b',1,'mlx::core::random::uniform(const array &low, const array &high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ac0dd00f7e37377d621f9f5bfb5a3f8e4',1,'mlx::core::random::uniform(T low, U high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2922e133d9f82dcf925bae0a784cc4a7',1,'mlx::core::random::uniform(const std::vector< int > &shape, Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a133d2855ff4d8daf41029cffdf43cdf9',1,'mlx::core::random::uniform(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['update_9',['update',['../struct_and.html#a514334a0fd865164d9c83fc07d499783',1,'And::update()'],['../struct_or.html#a942628b1b3c6be9f486c15081a0cba8e',1,'Or::update()']]], - ['utils_10',['utils',['../namespacemlx_1_1core_1_1metal.html#a529dc6c2d4a37ba544b66b2c3cd792cc',1,'mlx::core::metal']]] + ['unary_5fv2_6',['unary_v2',['../metal_2kernels_2unary_8h.html#a95f7b9c0112cdc84ae10465fb74d29dd',1,'unary.h']]], + ['unaryprimitive_7',['UnaryPrimitive',['../classmlx_1_1core_1_1_unary_primitive.html#a189f6d4ed369f82a4b724a29eb056d4e',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(Stream stream)'],['../classmlx_1_1core_1_1_unary_primitive.html#a9935cffc4f246d3d883bc3d26c5163f2',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(const UnaryPrimitive &other)=delete'],['../classmlx_1_1core_1_1_unary_primitive.html#a780281fb04e2daf1be630c124bd605e3',1,'mlx::core::UnaryPrimitive::UnaryPrimitive(UnaryPrimitive &&other)=delete']]], + ['uniform_8',['Uniform',['../classmlx_1_1core_1_1_uniform.html#a626aa1091aa77b4a32c02290106b85e1',1,'mlx::core::Uniform']]], + ['uniform_9',['uniform',['../namespacemlx_1_1core_1_1random.html#adaa626cf75ab891978954bd1eb79a38b',1,'mlx::core::random::uniform(const array &low, const array &high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#ac0dd00f7e37377d621f9f5bfb5a3f8e4',1,'mlx::core::random::uniform(T low, U high, const std::vector< int > &shape, Dtype dtype=float32, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a2922e133d9f82dcf925bae0a784cc4a7',1,'mlx::core::random::uniform(const std::vector< int > &shape, Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a133d2855ff4d8daf41029cffdf43cdf9',1,'mlx::core::random::uniform(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['update_10',['update',['../struct_and.html#a514334a0fd865164d9c83fc07d499783',1,'And::update()'],['../struct_or.html#a942628b1b3c6be9f486c15081a0cba8e',1,'Or::update()']]], + ['utils_11',['utils',['../namespacemlx_1_1core_1_1metal.html#a529dc6c2d4a37ba544b66b2c3cd792cc',1,'mlx::core::metal']]] ]; diff --git a/docs/build/html/search/functions_16.js b/docs/build/html/search/functions_16.js index c8d9c75ddc..7fb14508d0 100644 --- a/docs/build/html/search/functions_16.js +++ b/docs/build/html/search/functions_16.js @@ -4,11 +4,10 @@ var searchData= ['value_1',['value',['../classmlx_1_1core_1_1_event.html#ab71c7baee3d1d02ad6a2001bbf90b970',1,'mlx::core::Event']]], ['value_5fand_5fgrad_2',['value_and_grad',['../namespacemlx_1_1core.html#abf49b337a00997231c0f7fd389efa8f3',1,'mlx::core::value_and_grad(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#a7b987f404b8699de00f9e0099ab6b1b0',1,'mlx::core::value_and_grad(const std::function< std::vector< array >(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a7620f1ae298127cb6181db9162f012a7',1,'mlx::core::value_and_grad(const std::function< array(const std::vector< array > &)> &fun, const std::vector< int > &argnums)'],['../namespacemlx_1_1core.html#a2f69ffc30d66b1fca8f24b65be161a51',1,'mlx::core::value_and_grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)']]], ['var_3',['var',['../group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9',1,'mlx::core::var(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga7d7b38d118fa2613214078ef0f7d5a42',1,'mlx::core::var(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga78ddeb966cbe7a5b0aa17e1de43025f2',1,'mlx::core::var(const array &a, const std::vector< int > &axes, bool keepdims=false, int ddof=0, StreamOrDevice s={})'],['../group__ops.html#ga4fbf3e3f98f2e4956faf87af320aa9d0',1,'mlx::core::var(const array &a, int axis, bool keepdims=false, int ddof=0, StreamOrDevice s={})']]], - ['vectordescriptor_4',['vectorDescriptor',['../class_m_p_s_1_1_vector_descriptor.html#a2bac55ff9bffcebba0755abcca09ec15',1,'MPS::VectorDescriptor::vectorDescriptor(NS::UInteger length, NS::UInteger dataType)'],['../class_m_p_s_1_1_vector_descriptor.html#ab6534f1c9536613d8e78ed72fe12b275',1,'MPS::VectorDescriptor::vectorDescriptor(NS::UInteger length, NS::UInteger vectors, NS::UInteger vectorBytes, NS::UInteger dataType)']]], - ['view_5',['View',['../classmlx_1_1core_1_1_view.html#ad7eed156c308e9a29a8b41f965ec941e',1,'mlx::core::View']]], - ['view_6',['view',['../group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02',1,'mlx::core']]], - ['vjp_7',['vjp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abbf6d1d63dcda207ad7d9eeb4fc36225',1,'mlx::core::distributed::AllReduce::vjp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#aa5eff6fc128b71220899aab8ab9116fb',1,'mlx::core::distributed::AllGather::vjp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a74be4bcd0382f7f6400bf73fd5569c91',1,'mlx::core::fast::Custom::vjp()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#aacfbbbc15fcee0a5ce4f519ca3cca5eb',1,'mlx::core::fast::RMSNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#ae5e1b5df0705a6b1d141691a4396b0b6',1,'mlx::core::fast::LayerNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#ad999105414badd66c8fd9e069454a533',1,'mlx::core::fast::RoPE::vjp()'],['../classmlx_1_1core_1_1_primitive.html#a1dcb6807326eeab62474c6a0e3836d42',1,'mlx::core::Primitive::vjp()'],['../classmlx_1_1core_1_1_abs.html#aa2dd8ec0989e716b77394ac349b34592',1,'mlx::core::Abs::vjp()'],['../classmlx_1_1core_1_1_add.html#ac28e581862880e24ed2b99bb6a916607',1,'mlx::core::Add::vjp()'],['../classmlx_1_1core_1_1_add_m_m.html#ac1562a37cec6928e01281926ebeb47c6',1,'mlx::core::AddMM::vjp()'],['../classmlx_1_1core_1_1_arc_cos.html#a78e73e5e639d1249c7fe9614bf157c92',1,'mlx::core::ArcCos::vjp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a856c677f16e2b3f2edd2491e35db2d26',1,'mlx::core::ArcCosh::vjp()'],['../classmlx_1_1core_1_1_arc_sin.html#ab4057cd5ef1a8359f97493018e10d3a1',1,'mlx::core::ArcSin::vjp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a7988ee5b9e1e7e498dcab73d61ba147e',1,'mlx::core::ArcSinh::vjp()'],['../classmlx_1_1core_1_1_arc_tan.html#a5fefc3634b96a67ff8ae011a8ee180c2',1,'mlx::core::ArcTan::vjp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a99840c282e37b2b2a9c312e6e8ade1d2',1,'mlx::core::ArcTan2::vjp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a07da5797f7aaf3dfe43bf24e8562ac72',1,'mlx::core::ArcTanh::vjp()'],['../classmlx_1_1core_1_1_as_type.html#ac38a4f889311a3b5e5be9a67dcb93e18',1,'mlx::core::AsType::vjp()'],['../classmlx_1_1core_1_1_as_strided.html#a34783284c9b2f5b4a62c3c3ee5dd4062',1,'mlx::core::AsStrided::vjp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6131ed1c317ff8700a3e9b13fdaa9d61',1,'mlx::core::BitwiseBinary::vjp()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a1adf20087ee2f685bf39c2724b8e7120',1,'mlx::core::BlockMaskedMM::vjp()'],['../classmlx_1_1core_1_1_gather_m_m.html#a76c9f27c57354f6230b43944882e1bda',1,'mlx::core::GatherMM::vjp()'],['../classmlx_1_1core_1_1_broadcast.html#a0318847c9be40f00b23907ad56037d18',1,'mlx::core::Broadcast::vjp()'],['../classmlx_1_1core_1_1_ceil.html#ac2f5a2bd84b8f013e5ce688419a88acb',1,'mlx::core::Ceil::vjp()'],['../classmlx_1_1core_1_1_compiled.html#a32462e65c52f84b708188130cc508133',1,'mlx::core::Compiled::vjp()'],['../classmlx_1_1core_1_1_concatenate.html#a8155db9100ec3b8bd0bc94baeaeee3b0',1,'mlx::core::Concatenate::vjp()'],['../classmlx_1_1core_1_1_convolution.html#af8eb9c0c055ad20aa74b547016917690',1,'mlx::core::Convolution::vjp()'],['../classmlx_1_1core_1_1_copy.html#a6c4dee582001e9983e9517485ee37efd',1,'mlx::core::Copy::vjp()'],['../classmlx_1_1core_1_1_cos.html#a51d84113728e651ef9d4a1fe671c4d00',1,'mlx::core::Cos::vjp()'],['../classmlx_1_1core_1_1_cosh.html#a0791abd4305a333fb3b181a5357ce0f4',1,'mlx::core::Cosh::vjp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa1da36cef632df767cd9809d6cf06209',1,'mlx::core::CustomTransforms::vjp()'],['../classmlx_1_1core_1_1_depends.html#a02996fa45f01f7cb9f37074d5f8ccab0',1,'mlx::core::Depends::vjp()'],['../classmlx_1_1core_1_1_divide.html#ad3af7c70cad22c1a1a75b4a78ef793b6',1,'mlx::core::Divide::vjp()'],['../classmlx_1_1core_1_1_div_mod.html#a8c914a07f666a1d9377a27ed5d55e7c1',1,'mlx::core::DivMod::vjp()'],['../classmlx_1_1core_1_1_select.html#a9b522487b78fceeca7f827cd1c29a9a3',1,'mlx::core::Select::vjp()'],['../classmlx_1_1core_1_1_remainder.html#ab18f7bca1027ae71847a50da0933cec6',1,'mlx::core::Remainder::vjp()'],['../classmlx_1_1core_1_1_equal.html#af3c1bfcd1bf50922fc00e302bb193736',1,'mlx::core::Equal::vjp()'],['../classmlx_1_1core_1_1_erf.html#a1f529e95a42a2d69a8b18979d3ee2909',1,'mlx::core::Erf::vjp()'],['../classmlx_1_1core_1_1_erf_inv.html#a48afff12a58ddefae7ae0245c3580189',1,'mlx::core::ErfInv::vjp()'],['../classmlx_1_1core_1_1_exp.html#a94b9b7d137c3640d290b96c5e8b7e1a8',1,'mlx::core::Exp::vjp()'],['../classmlx_1_1core_1_1_expm1.html#af6ce416169190479c9792bb9cdbe2f43',1,'mlx::core::Expm1::vjp()'],['../classmlx_1_1core_1_1_f_f_t.html#aafc895614a6e368c0e6d64af20d01090',1,'mlx::core::FFT::vjp()'],['../classmlx_1_1core_1_1_floor.html#a589e2cf99b6fd1a5ba85534a2a31338e',1,'mlx::core::Floor::vjp()'],['../classmlx_1_1core_1_1_full.html#a49e76e7a8641f990701abc1b3bd49969',1,'mlx::core::Full::vjp()'],['../classmlx_1_1core_1_1_gather.html#aacf612a8f5f1cdbbfd19707d8d33c426',1,'mlx::core::Gather::vjp()'],['../classmlx_1_1core_1_1_greater.html#a341766a8a7e41d2a1160d35d4e781679',1,'mlx::core::Greater::vjp()'],['../classmlx_1_1core_1_1_greater_equal.html#a62f07a4ac54c708307c82aac0e5693ee',1,'mlx::core::GreaterEqual::vjp()'],['../classmlx_1_1core_1_1_hadamard.html#af4134775427b8998d66f489468b98656',1,'mlx::core::Hadamard::vjp()'],['../classmlx_1_1core_1_1_less.html#aaf205d389b5e602e0814b68f66de8f50',1,'mlx::core::Less::vjp()'],['../classmlx_1_1core_1_1_less_equal.html#aab2aab7590c299885e815c18eedd1028',1,'mlx::core::LessEqual::vjp()'],['../classmlx_1_1core_1_1_log.html#a40885dccfbf928c4d035881be1d49280',1,'mlx::core::Log::vjp()'],['../classmlx_1_1core_1_1_log1p.html#a3113c1d2b4c5e73d0b470f42dc48a880',1,'mlx::core::Log1p::vjp()'],['../classmlx_1_1core_1_1_logical_not.html#af2c3c241cf3910fbaba013c69d052a50',1,'mlx::core::LogicalNot::vjp()'],['../classmlx_1_1core_1_1_logical_and.html#ae42f8fc454577b0fd6410cae9d5f3b54',1,'mlx::core::LogicalAnd::vjp()'],['../classmlx_1_1core_1_1_logical_or.html#a51aed488f52d5031998689af9cb17847',1,'mlx::core::LogicalOr::vjp()'],['../classmlx_1_1core_1_1_log_add_exp.html#ae231af0ed24a93eb647ee58c2d2b20b4',1,'mlx::core::LogAddExp::vjp()'],['../classmlx_1_1core_1_1_matmul.html#a524136cca481598ea20894d85ca66bb0',1,'mlx::core::Matmul::vjp()'],['../classmlx_1_1core_1_1_maximum.html#a7de15d7b28784e24bbfc7e85ddcbcff3',1,'mlx::core::Maximum::vjp()'],['../classmlx_1_1core_1_1_minimum.html#a48a0cbe3a6c4f7473c00e343f63b5204',1,'mlx::core::Minimum::vjp()'],['../classmlx_1_1core_1_1_multiply.html#a74b7556ec03e2c3d3f971666d06f5db1',1,'mlx::core::Multiply::vjp()'],['../classmlx_1_1core_1_1_negative.html#a889585f056d33bda30c30311257af52a',1,'mlx::core::Negative::vjp()'],['../classmlx_1_1core_1_1_not_equal.html#a0361f29f4ae1235bdf3f3304527e2d4b',1,'mlx::core::NotEqual::vjp()'],['../classmlx_1_1core_1_1_pad.html#ad8a7e547644f2717a24322968e971038',1,'mlx::core::Pad::vjp()'],['../classmlx_1_1core_1_1_partition.html#a7110772b6cd2d430a2b825cf5c952ca9',1,'mlx::core::Partition::vjp()'],['../classmlx_1_1core_1_1_power.html#a1453bb8307d6ff33134f1e00263bf082',1,'mlx::core::Power::vjp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#acb975e272b4a88ab232ef7f7c3a2bf26',1,'mlx::core::QuantizedMatmul::vjp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#ae08a4b7d28902d46f39e66beeb0e23ab',1,'mlx::core::GatherQMM::vjp()'],['../classmlx_1_1core_1_1_reshape.html#ab17294ecc6b5d4e89626fb48c7516365',1,'mlx::core::Reshape::vjp()'],['../classmlx_1_1core_1_1_reduce.html#a684883d2a96315f548ca769510e28e4e',1,'mlx::core::Reduce::vjp()'],['../classmlx_1_1core_1_1_round.html#af8f085e08b7fa8840c52a20b12ca35ce',1,'mlx::core::Round::vjp()'],['../classmlx_1_1core_1_1_scan.html#aaf13f72620b4b5d6a20e1228930e848e',1,'mlx::core::Scan::vjp()'],['../classmlx_1_1core_1_1_scatter.html#a0b51287fba789bb139ed61d40a0c636a',1,'mlx::core::Scatter::vjp()'],['../classmlx_1_1core_1_1_sigmoid.html#aac2f56a4c8362e36a28e232758ca52cf',1,'mlx::core::Sigmoid::vjp()'],['../classmlx_1_1core_1_1_sign.html#aa60ac52edd739fbdf388a997acd01bce',1,'mlx::core::Sign::vjp()'],['../classmlx_1_1core_1_1_sin.html#aedefe550ab4b0687858981bc0bcfbfa0',1,'mlx::core::Sin::vjp()'],['../classmlx_1_1core_1_1_sinh.html#a6b39fdd429bbb4de389e7c904fd561f0',1,'mlx::core::Sinh::vjp()'],['../classmlx_1_1core_1_1_slice.html#a291746a527ff991b66249fb2b54b685f',1,'mlx::core::Slice::vjp()'],['../classmlx_1_1core_1_1_slice_update.html#aedcdc60a0477997a96306c02b66d3f77',1,'mlx::core::SliceUpdate::vjp()'],['../classmlx_1_1core_1_1_softmax.html#abb68c311c45ee422a7c966accde9041b',1,'mlx::core::Softmax::vjp()'],['../classmlx_1_1core_1_1_sort.html#a3a8900dce53ee4eb7a1b83806e629358',1,'mlx::core::Sort::vjp()'],['../classmlx_1_1core_1_1_split.html#a7e8730f9cffa9872fff6f8d577031674',1,'mlx::core::Split::vjp()'],['../classmlx_1_1core_1_1_square.html#abcd9516da7f02dc906368c23b0bca263',1,'mlx::core::Square::vjp()'],['../classmlx_1_1core_1_1_sqrt.html#a08a21bd2c3a016f042d95aca294e68f3',1,'mlx::core::Sqrt::vjp()'],['../classmlx_1_1core_1_1_subtract.html#a3a3322be7c3bcaa0397cf099091df16b',1,'mlx::core::Subtract::vjp()'],['../classmlx_1_1core_1_1_tan.html#a4639836cff03d73c769387d6943e92d7',1,'mlx::core::Tan::vjp()'],['../classmlx_1_1core_1_1_tanh.html#afe7b05e2b36b99c3a1b66f0cd3544e95',1,'mlx::core::Tanh::vjp()'],['../classmlx_1_1core_1_1_transpose.html#ac7805aa29b34afdf8852554f1e759f80',1,'mlx::core::Transpose::vjp()'],['../namespacemlx_1_1core.html#a1b33e2c2e3471420490cf0be2de6de18',1,'mlx::core::vjp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &cotangents)'],['../namespacemlx_1_1core.html#a2065a11249c3f4356ffd69b7a8c487ff',1,'mlx::core::vjp(const std::function< array(const array &)> &fun, const array &primal, const array &cotangent)']]], - ['vmap_8',['vmap',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a3f2dc71859847ca675ec4bfbe125035a',1,'mlx::core::distributed::AllReduce::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ad532d1d51f089dec3c84799b724ea031',1,'mlx::core::distributed::AllGather::vmap()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a7f4c3a4c48c6807faa36fb31e39dad8d',1,'mlx::core::fast::Custom::vmap()'],['../classmlx_1_1core_1_1_primitive.html#ac632b9619dd7a6a0f177bd36202e8103',1,'mlx::core::Primitive::vmap()'],['../classmlx_1_1core_1_1_abs.html#a4c9c98f1d71432fd3752ad9a6a8e7f2f',1,'mlx::core::Abs::vmap()'],['../classmlx_1_1core_1_1_add.html#a0e557d4d896153f84a25532562e4c646',1,'mlx::core::Add::vmap()'],['../classmlx_1_1core_1_1_add_m_m.html#a73ce80b3a37ec2523943028d50ebce81',1,'mlx::core::AddMM::vmap()'],['../classmlx_1_1core_1_1_arc_cos.html#a7548e23ace6827674aa6d284d44ccf83',1,'mlx::core::ArcCos::vmap()'],['../classmlx_1_1core_1_1_arc_cosh.html#af8ff78e910a9e485a203e1d3347bd461',1,'mlx::core::ArcCosh::vmap()'],['../classmlx_1_1core_1_1_arc_sin.html#a7cabb1e5a2bda44944378822c671ec82',1,'mlx::core::ArcSin::vmap()'],['../classmlx_1_1core_1_1_arc_sinh.html#a9e72b9751939387c333b5d4e19a37f6d',1,'mlx::core::ArcSinh::vmap()'],['../classmlx_1_1core_1_1_arc_tan.html#a1fb921554544a56498bc54f82e4a0556',1,'mlx::core::ArcTan::vmap()'],['../classmlx_1_1core_1_1_arc_tan2.html#ae02cb9fbf25e93dc1d7fbc9e3fb28634',1,'mlx::core::ArcTan2::vmap()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6ddcae68873559211cb91e7740dfc040',1,'mlx::core::ArcTanh::vmap()'],['../classmlx_1_1core_1_1_arg_partition.html#a441093795bcc31495ab5fbc9957b740a',1,'mlx::core::ArgPartition::vmap()'],['../classmlx_1_1core_1_1_arg_reduce.html#abfec42fa06ea15edaf393593751fb1ba',1,'mlx::core::ArgReduce::vmap()'],['../classmlx_1_1core_1_1_arg_sort.html#a3522bbbe4626a467394c1a8a9d7ac34e',1,'mlx::core::ArgSort::vmap()'],['../classmlx_1_1core_1_1_as_type.html#a7ebaf86fd6cad4a1ecfd7cde1ee0b0cc',1,'mlx::core::AsType::vmap()'],['../classmlx_1_1core_1_1_bitwise_binary.html#aa10be55f05bc1868bf4b375dc475f965',1,'mlx::core::BitwiseBinary::vmap()'],['../classmlx_1_1core_1_1_broadcast.html#aee4c71c2588ad01eb57e10f346cd666f',1,'mlx::core::Broadcast::vmap()'],['../classmlx_1_1core_1_1_ceil.html#ae86819990b43bdb0c2b3a25719b3a7a4',1,'mlx::core::Ceil::vmap()'],['../classmlx_1_1core_1_1_compiled.html#a732e7548f53977b4513bb7f30a04c30d',1,'mlx::core::Compiled::vmap()'],['../classmlx_1_1core_1_1_concatenate.html#a58c54dcf8e4b045d25edd3afc2caffc1',1,'mlx::core::Concatenate::vmap()'],['../classmlx_1_1core_1_1_conjugate.html#a2c7632c8ae0ca07777e23a0a79344e60',1,'mlx::core::Conjugate::vmap()'],['../classmlx_1_1core_1_1_copy.html#a669b10253c15b769d90058d1ad7d0e61',1,'mlx::core::Copy::vmap()'],['../classmlx_1_1core_1_1_cos.html#aec9460daf0131156734013d03b230cd6',1,'mlx::core::Cos::vmap()'],['../classmlx_1_1core_1_1_cosh.html#a1ab2386e7d96219b6e4a525f7dac0406',1,'mlx::core::Cosh::vmap()'],['../classmlx_1_1core_1_1_custom_transforms.html#a906a2ff30d9c5281fbf1fa927e4c021b',1,'mlx::core::CustomTransforms::vmap()'],['../classmlx_1_1core_1_1_divide.html#a83e7da52831165b3a026e97b63770242',1,'mlx::core::Divide::vmap()'],['../classmlx_1_1core_1_1_div_mod.html#ae709e0fdd83994bd1d156e0d0e6a7942',1,'mlx::core::DivMod::vmap()'],['../classmlx_1_1core_1_1_select.html#a84e80361c8cf02536b4b98098793550f',1,'mlx::core::Select::vmap()'],['../classmlx_1_1core_1_1_remainder.html#a79867e1099a2e3c2d3e87407b2ab6e3d',1,'mlx::core::Remainder::vmap()'],['../classmlx_1_1core_1_1_equal.html#aea9cc3c88924ac824d72c39c2e83b0ca',1,'mlx::core::Equal::vmap()'],['../classmlx_1_1core_1_1_erf.html#abe554f553356654a3e800ba368108aaa',1,'mlx::core::Erf::vmap()'],['../classmlx_1_1core_1_1_erf_inv.html#ad5d7634e8568af8cc4a54a558a48d0e9',1,'mlx::core::ErfInv::vmap()'],['../classmlx_1_1core_1_1_exp.html#a0fcd579fe148b4c3dbc72e514b81bb37',1,'mlx::core::Exp::vmap()'],['../classmlx_1_1core_1_1_expm1.html#aa4caa848b2ea97e71ee3dd33de039296',1,'mlx::core::Expm1::vmap()'],['../classmlx_1_1core_1_1_f_f_t.html#ac32d6cc9b67289124f855ea68a61ede1',1,'mlx::core::FFT::vmap()'],['../classmlx_1_1core_1_1_floor.html#aea4dc79a65774990e775ad49519a5d10',1,'mlx::core::Floor::vmap()'],['../classmlx_1_1core_1_1_full.html#afc57ab6bd9ebdbbf042af54a59785d95',1,'mlx::core::Full::vmap()'],['../classmlx_1_1core_1_1_gather.html#abab0c4c204e66489825ce80d2194a275',1,'mlx::core::Gather::vmap()'],['../classmlx_1_1core_1_1_greater.html#a6d8267411fc4951de781f9e8e6c53aa0',1,'mlx::core::Greater::vmap()'],['../classmlx_1_1core_1_1_greater_equal.html#ab0e1be93eb01b0ce7fa83e953f5e3e1d',1,'mlx::core::GreaterEqual::vmap()'],['../classmlx_1_1core_1_1_hadamard.html#a9f1a172e6246859e813002abe9b8f99c',1,'mlx::core::Hadamard::vmap()'],['../classmlx_1_1core_1_1_less.html#a5fee5956cf087d8405359121aa62ba7e',1,'mlx::core::Less::vmap()'],['../classmlx_1_1core_1_1_less_equal.html#a3d5df21db184f2b7620cda9da1684480',1,'mlx::core::LessEqual::vmap()'],['../classmlx_1_1core_1_1_log.html#a007ddbcf911093231f607a8b9ed5cd49',1,'mlx::core::Log::vmap()'],['../classmlx_1_1core_1_1_log1p.html#a7122576f95ce479926bbbbc690891f71',1,'mlx::core::Log1p::vmap()'],['../classmlx_1_1core_1_1_logical_not.html#a5308a271619ee74df561b0aaf525915d',1,'mlx::core::LogicalNot::vmap()'],['../classmlx_1_1core_1_1_logical_and.html#aacc5f6f53ffc327b7771485e3da2a4e5',1,'mlx::core::LogicalAnd::vmap()'],['../classmlx_1_1core_1_1_logical_or.html#a6e2e77e6aaf47872b2e96b151c32daf3',1,'mlx::core::LogicalOr::vmap()'],['../classmlx_1_1core_1_1_log_add_exp.html#a82190aa1421a9734b6e9480debffac78',1,'mlx::core::LogAddExp::vmap()'],['../classmlx_1_1core_1_1_matmul.html#a3a1c6e70bac300240760fe41a58340c2',1,'mlx::core::Matmul::vmap()'],['../classmlx_1_1core_1_1_maximum.html#ab664918e0d71cfec1318a9879e78c5d3',1,'mlx::core::Maximum::vmap()'],['../classmlx_1_1core_1_1_minimum.html#adab0f31acf68075a0be908d8eb882980',1,'mlx::core::Minimum::vmap()'],['../classmlx_1_1core_1_1_multiply.html#ae7e82c8fc8cbaf4e00c27eb54fac7dbf',1,'mlx::core::Multiply::vmap()'],['../classmlx_1_1core_1_1_negative.html#a1f8a6079e272f1a0599f88a1a8419cf0',1,'mlx::core::Negative::vmap()'],['../classmlx_1_1core_1_1_not_equal.html#ab8b57932f03c8eee664bf89adeaa43b5',1,'mlx::core::NotEqual::vmap()'],['../classmlx_1_1core_1_1_number_of_elements.html#a977d83eae845b8bd8c0b98b48cb1c6c2',1,'mlx::core::NumberOfElements::vmap()'],['../classmlx_1_1core_1_1_pad.html#a85658812a0f3275ba3eb74b7c75686cf',1,'mlx::core::Pad::vmap()'],['../classmlx_1_1core_1_1_partition.html#aa0cc55e4d4d2cb5d129d32832321df2c',1,'mlx::core::Partition::vmap()'],['../classmlx_1_1core_1_1_power.html#a5e22749592413a9adbdc877b03b87c8f',1,'mlx::core::Power::vmap()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a3434394140177b285f971c9ffe7e8763',1,'mlx::core::QuantizedMatmul::vmap()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a13ce5e138ebddb8780a034452f68892f',1,'mlx::core::GatherQMM::vmap()'],['../classmlx_1_1core_1_1_random_bits.html#a0dc12f053c6492f934bc18031412c415',1,'mlx::core::RandomBits::vmap()'],['../classmlx_1_1core_1_1_reshape.html#ae239dd3c6cab147e4af572dc58204f9d',1,'mlx::core::Reshape::vmap()'],['../classmlx_1_1core_1_1_reduce.html#abab1b5aa01ccad44f213f510c3596b38',1,'mlx::core::Reduce::vmap()'],['../classmlx_1_1core_1_1_round.html#a6fad8799a7982e1ccbe05be7cc38a7fd',1,'mlx::core::Round::vmap()'],['../classmlx_1_1core_1_1_scan.html#a297c7cc89c9bf9d186ebdebb634c7804',1,'mlx::core::Scan::vmap()'],['../classmlx_1_1core_1_1_sigmoid.html#a12712c23037e38192cbccd2d4b14cc85',1,'mlx::core::Sigmoid::vmap()'],['../classmlx_1_1core_1_1_sign.html#aa7296045907015b4e0ae8a93e5e6e295',1,'mlx::core::Sign::vmap()'],['../classmlx_1_1core_1_1_sin.html#a45533996f3d72d9dd97d4c61cd684fba',1,'mlx::core::Sin::vmap()'],['../classmlx_1_1core_1_1_sinh.html#ae171df22bc34c32e31b8135dc4caa788',1,'mlx::core::Sinh::vmap()'],['../classmlx_1_1core_1_1_slice.html#ae33583b0db22fcfeae34dfe1c0e3eaa2',1,'mlx::core::Slice::vmap()'],['../classmlx_1_1core_1_1_slice_update.html#adbf1c76de6ab2f986758530d351d6fa3',1,'mlx::core::SliceUpdate::vmap()'],['../classmlx_1_1core_1_1_softmax.html#ad22d3dcc71054d3dba529cf2dc981e19',1,'mlx::core::Softmax::vmap()'],['../classmlx_1_1core_1_1_sort.html#abfabb9e625cc0cb9335c7454ed27505c',1,'mlx::core::Sort::vmap()'],['../classmlx_1_1core_1_1_split.html#ab7c40e02a842e83bdb4698608472c7a6',1,'mlx::core::Split::vmap()'],['../classmlx_1_1core_1_1_square.html#a55bf43f878d4741c57a08d5fef472ea5',1,'mlx::core::Square::vmap()'],['../classmlx_1_1core_1_1_sqrt.html#a9d30e306ce08980c27d98c898577017e',1,'mlx::core::Sqrt::vmap()'],['../classmlx_1_1core_1_1_stop_gradient.html#aca680c8befef81da414c4375b11b16b0',1,'mlx::core::StopGradient::vmap()'],['../classmlx_1_1core_1_1_subtract.html#aa98f960e621a767c8a03624fd292f098',1,'mlx::core::Subtract::vmap()'],['../classmlx_1_1core_1_1_tan.html#ae2f67ca2adc83b10009cf28498bf58b7',1,'mlx::core::Tan::vmap()'],['../classmlx_1_1core_1_1_tanh.html#a32df3564c1ecb858c1ba9f855376762f',1,'mlx::core::Tanh::vmap()'],['../classmlx_1_1core_1_1_uniform.html#ad795037d5b1820e98f4268f166609926',1,'mlx::core::Uniform::vmap()'],['../classmlx_1_1core_1_1_view.html#a2230d3e5f434fb2b888de50b529ac121',1,'mlx::core::View::vmap()'],['../classmlx_1_1core_1_1_transpose.html#a5ef848b69def9a246665b67e6e3ffdfe',1,'mlx::core::Transpose::vmap()'],['../classmlx_1_1core_1_1_s_v_d.html#a0366c958f6cdac8d1d9e1a4eda53fae8',1,'mlx::core::SVD::vmap()'],['../classmlx_1_1core_1_1_inverse.html#a98419b9f0b8a6c9185fe012d523552c2',1,'mlx::core::Inverse::vmap()'],['../classmlx_1_1core_1_1_cholesky.html#ab5c3f6199ec3b399c91243a05d116aa5',1,'mlx::core::Cholesky::vmap()'],['../namespacemlx_1_1core.html#a8481a3bb4c12c2b7dc6ba576c2be3d0d',1,'mlx::core::vmap(const std::function< array(const array &, const array &)> &fun, int in_axis_a=0, int in_axis_b=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a95a7757e8d18fced38acfc6a3e8d686a',1,'mlx::core::vmap(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &in_axes={}, const std::vector< int > &out_axes={})']]], - ['vmap_5freplace_9',['vmap_replace',['../namespacemlx_1_1core_1_1detail.html#a31a5582530faea230eb8acafc0f7e154',1,'mlx::core::detail']]], - ['vmap_5ftrace_10',['vmap_trace',['../namespacemlx_1_1core_1_1detail.html#a5ba794afe1a557e0505887cfb481c515',1,'mlx::core::detail']]] + ['view_4',['View',['../classmlx_1_1core_1_1_view.html#ad7eed156c308e9a29a8b41f965ec941e',1,'mlx::core::View']]], + ['view_5',['view',['../group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02',1,'mlx::core']]], + ['vjp_6',['vjp',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#abbf6d1d63dcda207ad7d9eeb4fc36225',1,'mlx::core::distributed::AllReduce::vjp()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#aa5eff6fc128b71220899aab8ab9116fb',1,'mlx::core::distributed::AllGather::vjp()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a74be4bcd0382f7f6400bf73fd5569c91',1,'mlx::core::fast::Custom::vjp()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#aacfbbbc15fcee0a5ce4f519ca3cca5eb',1,'mlx::core::fast::RMSNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#ae5e1b5df0705a6b1d141691a4396b0b6',1,'mlx::core::fast::LayerNorm::vjp()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#ad999105414badd66c8fd9e069454a533',1,'mlx::core::fast::RoPE::vjp()'],['../classmlx_1_1core_1_1_primitive.html#a1dcb6807326eeab62474c6a0e3836d42',1,'mlx::core::Primitive::vjp()'],['../classmlx_1_1core_1_1_abs.html#aa2dd8ec0989e716b77394ac349b34592',1,'mlx::core::Abs::vjp()'],['../classmlx_1_1core_1_1_add.html#ac28e581862880e24ed2b99bb6a916607',1,'mlx::core::Add::vjp()'],['../classmlx_1_1core_1_1_add_m_m.html#ac1562a37cec6928e01281926ebeb47c6',1,'mlx::core::AddMM::vjp()'],['../classmlx_1_1core_1_1_arc_cos.html#a78e73e5e639d1249c7fe9614bf157c92',1,'mlx::core::ArcCos::vjp()'],['../classmlx_1_1core_1_1_arc_cosh.html#a856c677f16e2b3f2edd2491e35db2d26',1,'mlx::core::ArcCosh::vjp()'],['../classmlx_1_1core_1_1_arc_sin.html#ab4057cd5ef1a8359f97493018e10d3a1',1,'mlx::core::ArcSin::vjp()'],['../classmlx_1_1core_1_1_arc_sinh.html#a7988ee5b9e1e7e498dcab73d61ba147e',1,'mlx::core::ArcSinh::vjp()'],['../classmlx_1_1core_1_1_arc_tan.html#a5fefc3634b96a67ff8ae011a8ee180c2',1,'mlx::core::ArcTan::vjp()'],['../classmlx_1_1core_1_1_arc_tan2.html#a99840c282e37b2b2a9c312e6e8ade1d2',1,'mlx::core::ArcTan2::vjp()'],['../classmlx_1_1core_1_1_arc_tanh.html#a07da5797f7aaf3dfe43bf24e8562ac72',1,'mlx::core::ArcTanh::vjp()'],['../classmlx_1_1core_1_1_as_type.html#ac38a4f889311a3b5e5be9a67dcb93e18',1,'mlx::core::AsType::vjp()'],['../classmlx_1_1core_1_1_as_strided.html#a34783284c9b2f5b4a62c3c3ee5dd4062',1,'mlx::core::AsStrided::vjp()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a6131ed1c317ff8700a3e9b13fdaa9d61',1,'mlx::core::BitwiseBinary::vjp()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#a1adf20087ee2f685bf39c2724b8e7120',1,'mlx::core::BlockMaskedMM::vjp()'],['../classmlx_1_1core_1_1_gather_m_m.html#a76c9f27c57354f6230b43944882e1bda',1,'mlx::core::GatherMM::vjp()'],['../classmlx_1_1core_1_1_broadcast.html#a0318847c9be40f00b23907ad56037d18',1,'mlx::core::Broadcast::vjp()'],['../classmlx_1_1core_1_1_ceil.html#ac2f5a2bd84b8f013e5ce688419a88acb',1,'mlx::core::Ceil::vjp()'],['../classmlx_1_1core_1_1_compiled.html#a32462e65c52f84b708188130cc508133',1,'mlx::core::Compiled::vjp()'],['../classmlx_1_1core_1_1_concatenate.html#a8155db9100ec3b8bd0bc94baeaeee3b0',1,'mlx::core::Concatenate::vjp()'],['../classmlx_1_1core_1_1_convolution.html#af8eb9c0c055ad20aa74b547016917690',1,'mlx::core::Convolution::vjp()'],['../classmlx_1_1core_1_1_copy.html#a6c4dee582001e9983e9517485ee37efd',1,'mlx::core::Copy::vjp()'],['../classmlx_1_1core_1_1_cos.html#a51d84113728e651ef9d4a1fe671c4d00',1,'mlx::core::Cos::vjp()'],['../classmlx_1_1core_1_1_cosh.html#a0791abd4305a333fb3b181a5357ce0f4',1,'mlx::core::Cosh::vjp()'],['../classmlx_1_1core_1_1_custom_transforms.html#aa1da36cef632df767cd9809d6cf06209',1,'mlx::core::CustomTransforms::vjp()'],['../classmlx_1_1core_1_1_depends.html#a02996fa45f01f7cb9f37074d5f8ccab0',1,'mlx::core::Depends::vjp()'],['../classmlx_1_1core_1_1_divide.html#ad3af7c70cad22c1a1a75b4a78ef793b6',1,'mlx::core::Divide::vjp()'],['../classmlx_1_1core_1_1_div_mod.html#a8c914a07f666a1d9377a27ed5d55e7c1',1,'mlx::core::DivMod::vjp()'],['../classmlx_1_1core_1_1_select.html#a9b522487b78fceeca7f827cd1c29a9a3',1,'mlx::core::Select::vjp()'],['../classmlx_1_1core_1_1_remainder.html#ab18f7bca1027ae71847a50da0933cec6',1,'mlx::core::Remainder::vjp()'],['../classmlx_1_1core_1_1_equal.html#af3c1bfcd1bf50922fc00e302bb193736',1,'mlx::core::Equal::vjp()'],['../classmlx_1_1core_1_1_erf.html#a1f529e95a42a2d69a8b18979d3ee2909',1,'mlx::core::Erf::vjp()'],['../classmlx_1_1core_1_1_erf_inv.html#a48afff12a58ddefae7ae0245c3580189',1,'mlx::core::ErfInv::vjp()'],['../classmlx_1_1core_1_1_exp.html#a94b9b7d137c3640d290b96c5e8b7e1a8',1,'mlx::core::Exp::vjp()'],['../classmlx_1_1core_1_1_expm1.html#af6ce416169190479c9792bb9cdbe2f43',1,'mlx::core::Expm1::vjp()'],['../classmlx_1_1core_1_1_f_f_t.html#aafc895614a6e368c0e6d64af20d01090',1,'mlx::core::FFT::vjp()'],['../classmlx_1_1core_1_1_floor.html#a589e2cf99b6fd1a5ba85534a2a31338e',1,'mlx::core::Floor::vjp()'],['../classmlx_1_1core_1_1_full.html#a49e76e7a8641f990701abc1b3bd49969',1,'mlx::core::Full::vjp()'],['../classmlx_1_1core_1_1_gather.html#aacf612a8f5f1cdbbfd19707d8d33c426',1,'mlx::core::Gather::vjp()'],['../classmlx_1_1core_1_1_greater.html#a341766a8a7e41d2a1160d35d4e781679',1,'mlx::core::Greater::vjp()'],['../classmlx_1_1core_1_1_greater_equal.html#a62f07a4ac54c708307c82aac0e5693ee',1,'mlx::core::GreaterEqual::vjp()'],['../classmlx_1_1core_1_1_hadamard.html#af4134775427b8998d66f489468b98656',1,'mlx::core::Hadamard::vjp()'],['../classmlx_1_1core_1_1_less.html#aaf205d389b5e602e0814b68f66de8f50',1,'mlx::core::Less::vjp()'],['../classmlx_1_1core_1_1_less_equal.html#aab2aab7590c299885e815c18eedd1028',1,'mlx::core::LessEqual::vjp()'],['../classmlx_1_1core_1_1_log.html#a40885dccfbf928c4d035881be1d49280',1,'mlx::core::Log::vjp()'],['../classmlx_1_1core_1_1_log1p.html#a3113c1d2b4c5e73d0b470f42dc48a880',1,'mlx::core::Log1p::vjp()'],['../classmlx_1_1core_1_1_logical_not.html#af2c3c241cf3910fbaba013c69d052a50',1,'mlx::core::LogicalNot::vjp()'],['../classmlx_1_1core_1_1_logical_and.html#ae42f8fc454577b0fd6410cae9d5f3b54',1,'mlx::core::LogicalAnd::vjp()'],['../classmlx_1_1core_1_1_logical_or.html#a51aed488f52d5031998689af9cb17847',1,'mlx::core::LogicalOr::vjp()'],['../classmlx_1_1core_1_1_log_add_exp.html#ae231af0ed24a93eb647ee58c2d2b20b4',1,'mlx::core::LogAddExp::vjp()'],['../classmlx_1_1core_1_1_matmul.html#a524136cca481598ea20894d85ca66bb0',1,'mlx::core::Matmul::vjp()'],['../classmlx_1_1core_1_1_maximum.html#a7de15d7b28784e24bbfc7e85ddcbcff3',1,'mlx::core::Maximum::vjp()'],['../classmlx_1_1core_1_1_minimum.html#a48a0cbe3a6c4f7473c00e343f63b5204',1,'mlx::core::Minimum::vjp()'],['../classmlx_1_1core_1_1_multiply.html#a74b7556ec03e2c3d3f971666d06f5db1',1,'mlx::core::Multiply::vjp()'],['../classmlx_1_1core_1_1_negative.html#a889585f056d33bda30c30311257af52a',1,'mlx::core::Negative::vjp()'],['../classmlx_1_1core_1_1_not_equal.html#a0361f29f4ae1235bdf3f3304527e2d4b',1,'mlx::core::NotEqual::vjp()'],['../classmlx_1_1core_1_1_pad.html#ad8a7e547644f2717a24322968e971038',1,'mlx::core::Pad::vjp()'],['../classmlx_1_1core_1_1_partition.html#a7110772b6cd2d430a2b825cf5c952ca9',1,'mlx::core::Partition::vjp()'],['../classmlx_1_1core_1_1_power.html#a1453bb8307d6ff33134f1e00263bf082',1,'mlx::core::Power::vjp()'],['../classmlx_1_1core_1_1_quantized_matmul.html#acb975e272b4a88ab232ef7f7c3a2bf26',1,'mlx::core::QuantizedMatmul::vjp()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#ae08a4b7d28902d46f39e66beeb0e23ab',1,'mlx::core::GatherQMM::vjp()'],['../classmlx_1_1core_1_1_reshape.html#ab17294ecc6b5d4e89626fb48c7516365',1,'mlx::core::Reshape::vjp()'],['../classmlx_1_1core_1_1_reduce.html#a684883d2a96315f548ca769510e28e4e',1,'mlx::core::Reduce::vjp()'],['../classmlx_1_1core_1_1_round.html#af8f085e08b7fa8840c52a20b12ca35ce',1,'mlx::core::Round::vjp()'],['../classmlx_1_1core_1_1_scan.html#aaf13f72620b4b5d6a20e1228930e848e',1,'mlx::core::Scan::vjp()'],['../classmlx_1_1core_1_1_scatter.html#a0b51287fba789bb139ed61d40a0c636a',1,'mlx::core::Scatter::vjp()'],['../classmlx_1_1core_1_1_sigmoid.html#aac2f56a4c8362e36a28e232758ca52cf',1,'mlx::core::Sigmoid::vjp()'],['../classmlx_1_1core_1_1_sign.html#aa60ac52edd739fbdf388a997acd01bce',1,'mlx::core::Sign::vjp()'],['../classmlx_1_1core_1_1_sin.html#aedefe550ab4b0687858981bc0bcfbfa0',1,'mlx::core::Sin::vjp()'],['../classmlx_1_1core_1_1_sinh.html#a6b39fdd429bbb4de389e7c904fd561f0',1,'mlx::core::Sinh::vjp()'],['../classmlx_1_1core_1_1_slice.html#a291746a527ff991b66249fb2b54b685f',1,'mlx::core::Slice::vjp()'],['../classmlx_1_1core_1_1_slice_update.html#aedcdc60a0477997a96306c02b66d3f77',1,'mlx::core::SliceUpdate::vjp()'],['../classmlx_1_1core_1_1_softmax.html#abb68c311c45ee422a7c966accde9041b',1,'mlx::core::Softmax::vjp()'],['../classmlx_1_1core_1_1_sort.html#a3a8900dce53ee4eb7a1b83806e629358',1,'mlx::core::Sort::vjp()'],['../classmlx_1_1core_1_1_split.html#a7e8730f9cffa9872fff6f8d577031674',1,'mlx::core::Split::vjp()'],['../classmlx_1_1core_1_1_square.html#abcd9516da7f02dc906368c23b0bca263',1,'mlx::core::Square::vjp()'],['../classmlx_1_1core_1_1_sqrt.html#a08a21bd2c3a016f042d95aca294e68f3',1,'mlx::core::Sqrt::vjp()'],['../classmlx_1_1core_1_1_subtract.html#a3a3322be7c3bcaa0397cf099091df16b',1,'mlx::core::Subtract::vjp()'],['../classmlx_1_1core_1_1_tan.html#a4639836cff03d73c769387d6943e92d7',1,'mlx::core::Tan::vjp()'],['../classmlx_1_1core_1_1_tanh.html#afe7b05e2b36b99c3a1b66f0cd3544e95',1,'mlx::core::Tanh::vjp()'],['../classmlx_1_1core_1_1_transpose.html#ac7805aa29b34afdf8852554f1e759f80',1,'mlx::core::Transpose::vjp()'],['../namespacemlx_1_1core.html#a1b33e2c2e3471420490cf0be2de6de18',1,'mlx::core::vjp(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &primals, const std::vector< array > &cotangents)'],['../namespacemlx_1_1core.html#a2065a11249c3f4356ffd69b7a8c487ff',1,'mlx::core::vjp(const std::function< array(const array &)> &fun, const array &primal, const array &cotangent)']]], + ['vmap_7',['vmap',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#a3f2dc71859847ca675ec4bfbe125035a',1,'mlx::core::distributed::AllReduce::vmap()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ad532d1d51f089dec3c84799b724ea031',1,'mlx::core::distributed::AllGather::vmap()'],['../classmlx_1_1core_1_1fast_1_1_custom.html#a7f4c3a4c48c6807faa36fb31e39dad8d',1,'mlx::core::fast::Custom::vmap()'],['../classmlx_1_1core_1_1_primitive.html#ac632b9619dd7a6a0f177bd36202e8103',1,'mlx::core::Primitive::vmap()'],['../classmlx_1_1core_1_1_abs.html#a4c9c98f1d71432fd3752ad9a6a8e7f2f',1,'mlx::core::Abs::vmap()'],['../classmlx_1_1core_1_1_add.html#a0e557d4d896153f84a25532562e4c646',1,'mlx::core::Add::vmap()'],['../classmlx_1_1core_1_1_add_m_m.html#a73ce80b3a37ec2523943028d50ebce81',1,'mlx::core::AddMM::vmap()'],['../classmlx_1_1core_1_1_arc_cos.html#a7548e23ace6827674aa6d284d44ccf83',1,'mlx::core::ArcCos::vmap()'],['../classmlx_1_1core_1_1_arc_cosh.html#af8ff78e910a9e485a203e1d3347bd461',1,'mlx::core::ArcCosh::vmap()'],['../classmlx_1_1core_1_1_arc_sin.html#a7cabb1e5a2bda44944378822c671ec82',1,'mlx::core::ArcSin::vmap()'],['../classmlx_1_1core_1_1_arc_sinh.html#a9e72b9751939387c333b5d4e19a37f6d',1,'mlx::core::ArcSinh::vmap()'],['../classmlx_1_1core_1_1_arc_tan.html#a1fb921554544a56498bc54f82e4a0556',1,'mlx::core::ArcTan::vmap()'],['../classmlx_1_1core_1_1_arc_tan2.html#ae02cb9fbf25e93dc1d7fbc9e3fb28634',1,'mlx::core::ArcTan2::vmap()'],['../classmlx_1_1core_1_1_arc_tanh.html#a6ddcae68873559211cb91e7740dfc040',1,'mlx::core::ArcTanh::vmap()'],['../classmlx_1_1core_1_1_arg_partition.html#a441093795bcc31495ab5fbc9957b740a',1,'mlx::core::ArgPartition::vmap()'],['../classmlx_1_1core_1_1_arg_reduce.html#abfec42fa06ea15edaf393593751fb1ba',1,'mlx::core::ArgReduce::vmap()'],['../classmlx_1_1core_1_1_arg_sort.html#a3522bbbe4626a467394c1a8a9d7ac34e',1,'mlx::core::ArgSort::vmap()'],['../classmlx_1_1core_1_1_as_type.html#a7ebaf86fd6cad4a1ecfd7cde1ee0b0cc',1,'mlx::core::AsType::vmap()'],['../classmlx_1_1core_1_1_bitwise_binary.html#aa10be55f05bc1868bf4b375dc475f965',1,'mlx::core::BitwiseBinary::vmap()'],['../classmlx_1_1core_1_1_broadcast.html#aee4c71c2588ad01eb57e10f346cd666f',1,'mlx::core::Broadcast::vmap()'],['../classmlx_1_1core_1_1_ceil.html#ae86819990b43bdb0c2b3a25719b3a7a4',1,'mlx::core::Ceil::vmap()'],['../classmlx_1_1core_1_1_compiled.html#a732e7548f53977b4513bb7f30a04c30d',1,'mlx::core::Compiled::vmap()'],['../classmlx_1_1core_1_1_concatenate.html#a58c54dcf8e4b045d25edd3afc2caffc1',1,'mlx::core::Concatenate::vmap()'],['../classmlx_1_1core_1_1_conjugate.html#a2c7632c8ae0ca07777e23a0a79344e60',1,'mlx::core::Conjugate::vmap()'],['../classmlx_1_1core_1_1_copy.html#a669b10253c15b769d90058d1ad7d0e61',1,'mlx::core::Copy::vmap()'],['../classmlx_1_1core_1_1_cos.html#aec9460daf0131156734013d03b230cd6',1,'mlx::core::Cos::vmap()'],['../classmlx_1_1core_1_1_cosh.html#a1ab2386e7d96219b6e4a525f7dac0406',1,'mlx::core::Cosh::vmap()'],['../classmlx_1_1core_1_1_custom_transforms.html#a906a2ff30d9c5281fbf1fa927e4c021b',1,'mlx::core::CustomTransforms::vmap()'],['../classmlx_1_1core_1_1_divide.html#a83e7da52831165b3a026e97b63770242',1,'mlx::core::Divide::vmap()'],['../classmlx_1_1core_1_1_div_mod.html#ae709e0fdd83994bd1d156e0d0e6a7942',1,'mlx::core::DivMod::vmap()'],['../classmlx_1_1core_1_1_select.html#a84e80361c8cf02536b4b98098793550f',1,'mlx::core::Select::vmap()'],['../classmlx_1_1core_1_1_remainder.html#a79867e1099a2e3c2d3e87407b2ab6e3d',1,'mlx::core::Remainder::vmap()'],['../classmlx_1_1core_1_1_equal.html#aea9cc3c88924ac824d72c39c2e83b0ca',1,'mlx::core::Equal::vmap()'],['../classmlx_1_1core_1_1_erf.html#abe554f553356654a3e800ba368108aaa',1,'mlx::core::Erf::vmap()'],['../classmlx_1_1core_1_1_erf_inv.html#ad5d7634e8568af8cc4a54a558a48d0e9',1,'mlx::core::ErfInv::vmap()'],['../classmlx_1_1core_1_1_exp.html#a0fcd579fe148b4c3dbc72e514b81bb37',1,'mlx::core::Exp::vmap()'],['../classmlx_1_1core_1_1_expm1.html#aa4caa848b2ea97e71ee3dd33de039296',1,'mlx::core::Expm1::vmap()'],['../classmlx_1_1core_1_1_f_f_t.html#ac32d6cc9b67289124f855ea68a61ede1',1,'mlx::core::FFT::vmap()'],['../classmlx_1_1core_1_1_floor.html#aea4dc79a65774990e775ad49519a5d10',1,'mlx::core::Floor::vmap()'],['../classmlx_1_1core_1_1_full.html#afc57ab6bd9ebdbbf042af54a59785d95',1,'mlx::core::Full::vmap()'],['../classmlx_1_1core_1_1_gather.html#abab0c4c204e66489825ce80d2194a275',1,'mlx::core::Gather::vmap()'],['../classmlx_1_1core_1_1_greater.html#a6d8267411fc4951de781f9e8e6c53aa0',1,'mlx::core::Greater::vmap()'],['../classmlx_1_1core_1_1_greater_equal.html#ab0e1be93eb01b0ce7fa83e953f5e3e1d',1,'mlx::core::GreaterEqual::vmap()'],['../classmlx_1_1core_1_1_hadamard.html#a9f1a172e6246859e813002abe9b8f99c',1,'mlx::core::Hadamard::vmap()'],['../classmlx_1_1core_1_1_less.html#a5fee5956cf087d8405359121aa62ba7e',1,'mlx::core::Less::vmap()'],['../classmlx_1_1core_1_1_less_equal.html#a3d5df21db184f2b7620cda9da1684480',1,'mlx::core::LessEqual::vmap()'],['../classmlx_1_1core_1_1_log.html#a007ddbcf911093231f607a8b9ed5cd49',1,'mlx::core::Log::vmap()'],['../classmlx_1_1core_1_1_log1p.html#a7122576f95ce479926bbbbc690891f71',1,'mlx::core::Log1p::vmap()'],['../classmlx_1_1core_1_1_logical_not.html#a5308a271619ee74df561b0aaf525915d',1,'mlx::core::LogicalNot::vmap()'],['../classmlx_1_1core_1_1_logical_and.html#aacc5f6f53ffc327b7771485e3da2a4e5',1,'mlx::core::LogicalAnd::vmap()'],['../classmlx_1_1core_1_1_logical_or.html#a6e2e77e6aaf47872b2e96b151c32daf3',1,'mlx::core::LogicalOr::vmap()'],['../classmlx_1_1core_1_1_log_add_exp.html#a82190aa1421a9734b6e9480debffac78',1,'mlx::core::LogAddExp::vmap()'],['../classmlx_1_1core_1_1_matmul.html#a3a1c6e70bac300240760fe41a58340c2',1,'mlx::core::Matmul::vmap()'],['../classmlx_1_1core_1_1_maximum.html#ab664918e0d71cfec1318a9879e78c5d3',1,'mlx::core::Maximum::vmap()'],['../classmlx_1_1core_1_1_minimum.html#adab0f31acf68075a0be908d8eb882980',1,'mlx::core::Minimum::vmap()'],['../classmlx_1_1core_1_1_multiply.html#ae7e82c8fc8cbaf4e00c27eb54fac7dbf',1,'mlx::core::Multiply::vmap()'],['../classmlx_1_1core_1_1_negative.html#a1f8a6079e272f1a0599f88a1a8419cf0',1,'mlx::core::Negative::vmap()'],['../classmlx_1_1core_1_1_not_equal.html#ab8b57932f03c8eee664bf89adeaa43b5',1,'mlx::core::NotEqual::vmap()'],['../classmlx_1_1core_1_1_number_of_elements.html#a977d83eae845b8bd8c0b98b48cb1c6c2',1,'mlx::core::NumberOfElements::vmap()'],['../classmlx_1_1core_1_1_pad.html#a85658812a0f3275ba3eb74b7c75686cf',1,'mlx::core::Pad::vmap()'],['../classmlx_1_1core_1_1_partition.html#aa0cc55e4d4d2cb5d129d32832321df2c',1,'mlx::core::Partition::vmap()'],['../classmlx_1_1core_1_1_power.html#a5e22749592413a9adbdc877b03b87c8f',1,'mlx::core::Power::vmap()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a3434394140177b285f971c9ffe7e8763',1,'mlx::core::QuantizedMatmul::vmap()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a13ce5e138ebddb8780a034452f68892f',1,'mlx::core::GatherQMM::vmap()'],['../classmlx_1_1core_1_1_random_bits.html#a0dc12f053c6492f934bc18031412c415',1,'mlx::core::RandomBits::vmap()'],['../classmlx_1_1core_1_1_reshape.html#ae239dd3c6cab147e4af572dc58204f9d',1,'mlx::core::Reshape::vmap()'],['../classmlx_1_1core_1_1_reduce.html#abab1b5aa01ccad44f213f510c3596b38',1,'mlx::core::Reduce::vmap()'],['../classmlx_1_1core_1_1_round.html#a6fad8799a7982e1ccbe05be7cc38a7fd',1,'mlx::core::Round::vmap()'],['../classmlx_1_1core_1_1_scan.html#a297c7cc89c9bf9d186ebdebb634c7804',1,'mlx::core::Scan::vmap()'],['../classmlx_1_1core_1_1_scatter.html#a696c38b373a7a7c71bc112bd1117e322',1,'mlx::core::Scatter::vmap()'],['../classmlx_1_1core_1_1_sigmoid.html#a12712c23037e38192cbccd2d4b14cc85',1,'mlx::core::Sigmoid::vmap()'],['../classmlx_1_1core_1_1_sign.html#aa7296045907015b4e0ae8a93e5e6e295',1,'mlx::core::Sign::vmap()'],['../classmlx_1_1core_1_1_sin.html#a45533996f3d72d9dd97d4c61cd684fba',1,'mlx::core::Sin::vmap()'],['../classmlx_1_1core_1_1_sinh.html#ae171df22bc34c32e31b8135dc4caa788',1,'mlx::core::Sinh::vmap()'],['../classmlx_1_1core_1_1_slice.html#ae33583b0db22fcfeae34dfe1c0e3eaa2',1,'mlx::core::Slice::vmap()'],['../classmlx_1_1core_1_1_slice_update.html#adbf1c76de6ab2f986758530d351d6fa3',1,'mlx::core::SliceUpdate::vmap()'],['../classmlx_1_1core_1_1_softmax.html#ad22d3dcc71054d3dba529cf2dc981e19',1,'mlx::core::Softmax::vmap()'],['../classmlx_1_1core_1_1_sort.html#abfabb9e625cc0cb9335c7454ed27505c',1,'mlx::core::Sort::vmap()'],['../classmlx_1_1core_1_1_split.html#ab7c40e02a842e83bdb4698608472c7a6',1,'mlx::core::Split::vmap()'],['../classmlx_1_1core_1_1_square.html#a55bf43f878d4741c57a08d5fef472ea5',1,'mlx::core::Square::vmap()'],['../classmlx_1_1core_1_1_sqrt.html#a9d30e306ce08980c27d98c898577017e',1,'mlx::core::Sqrt::vmap()'],['../classmlx_1_1core_1_1_stop_gradient.html#aca680c8befef81da414c4375b11b16b0',1,'mlx::core::StopGradient::vmap()'],['../classmlx_1_1core_1_1_subtract.html#aa98f960e621a767c8a03624fd292f098',1,'mlx::core::Subtract::vmap()'],['../classmlx_1_1core_1_1_tan.html#ae2f67ca2adc83b10009cf28498bf58b7',1,'mlx::core::Tan::vmap()'],['../classmlx_1_1core_1_1_tanh.html#a32df3564c1ecb858c1ba9f855376762f',1,'mlx::core::Tanh::vmap()'],['../classmlx_1_1core_1_1_uniform.html#ad795037d5b1820e98f4268f166609926',1,'mlx::core::Uniform::vmap()'],['../classmlx_1_1core_1_1_view.html#a2230d3e5f434fb2b888de50b529ac121',1,'mlx::core::View::vmap()'],['../classmlx_1_1core_1_1_transpose.html#a5ef848b69def9a246665b67e6e3ffdfe',1,'mlx::core::Transpose::vmap()'],['../classmlx_1_1core_1_1_s_v_d.html#a0366c958f6cdac8d1d9e1a4eda53fae8',1,'mlx::core::SVD::vmap()'],['../classmlx_1_1core_1_1_inverse.html#a98419b9f0b8a6c9185fe012d523552c2',1,'mlx::core::Inverse::vmap()'],['../classmlx_1_1core_1_1_cholesky.html#ab5c3f6199ec3b399c91243a05d116aa5',1,'mlx::core::Cholesky::vmap()'],['../namespacemlx_1_1core.html#a8481a3bb4c12c2b7dc6ba576c2be3d0d',1,'mlx::core::vmap(const std::function< array(const array &, const array &)> &fun, int in_axis_a=0, int in_axis_b=0, int out_axis=0)'],['../namespacemlx_1_1core.html#a95a7757e8d18fced38acfc6a3e8d686a',1,'mlx::core::vmap(const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< int > &in_axes={}, const std::vector< int > &out_axes={})']]], + ['vmap_5freplace_8',['vmap_replace',['../namespacemlx_1_1core_1_1detail.html#a31a5582530faea230eb8acafc0f7e154',1,'mlx::core::detail']]], + ['vmap_5ftrace_9',['vmap_trace',['../namespacemlx_1_1core_1_1detail.html#a5ba794afe1a557e0505887cfb481c515',1,'mlx::core::detail']]] ]; diff --git a/docs/build/html/search/functions_2.js b/docs/build/html/search/functions_2.js index 14fcd34f83..0fe844a598 100644 --- a/docs/build/html/search/functions_2.js +++ b/docs/build/html/search/functions_2.js @@ -9,38 +9,41 @@ var searchData= ['binary_5fg_5fnd1_6',['binary_g_nd1',['../metal_2kernels_2binary_8h.html#a6808bfb006cb5473da087a2758d0d867',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, constant const size_t &a_stride, constant const size_t &b_stride, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ad1fad37c168192b212a4294f4cf78133',1,'binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const size_t &a_stride, constant const size_t &b_stride, uint index): binary_two.h']]], ['binary_5fg_5fnd2_7',['binary_g_nd2',['../metal_2kernels_2binary_8h.html#a8cd5989852ec704c6fd132ae28f4fc14',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a03f7c15a1607576755abb65c542ae347',1,'binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim): binary_two.h']]], ['binary_5fg_5fnd3_8',['binary_g_nd3',['../metal_2kernels_2binary_8h.html#ac4979e60b993f7ffb602bcb91cd68bc9',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a43e5943460996c43060d1f3aa1309ba6',1,'binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim): binary_two.h']]], - ['binary_5fop_5fgpu_9',['binary_op_gpu',['../namespacemlx_1_1core.html#a63f3c166ea4916faa5d038668cfe16b1',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s)'],['../namespacemlx_1_1core.html#a6437ced86c227e0807352ae6b16da796',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)']]], - ['binary_5fop_5fgpu_5finplace_10',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#afee8898eaceab6dd5e90357f4a0613c4',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string op, const Stream &s)'],['../namespacemlx_1_1core.html#ad53410af1488f83d74035e3132fa16af',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string op, const Stream &s)']]], + ['binary_5fop_5fgpu_9',['binary_op_gpu',['../namespacemlx_1_1core.html#ad884f4a36308b5b4f8a5d990d2e086df',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a094876ea5a2a2445ab64efc8222da202',1,'mlx::core::binary_op_gpu(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], + ['binary_5fop_5fgpu_5finplace_10',['binary_op_gpu_inplace',['../namespacemlx_1_1core.html#a8616c0b7b0fc118a75400bc86404c367',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, std::vector< array > &outputs, const std::string &op, const Stream &s)'],['../namespacemlx_1_1core.html#a7e6af6624e322e7ad60a3873a66e18a3',1,'mlx::core::binary_op_gpu_inplace(const std::vector< array > &inputs, array &out, const std::string &op, const Stream &s)']]], ['binary_5fops_11',['binary_ops',['../namespacemlx_1_1core_1_1metal.html#a8db7f9cc781d4bfb08423a401665f322',1,'mlx::core::metal']]], ['binary_5fss_12',['binary_ss',['../metal_2kernels_2binary_8h.html#a242b8b29a852c255467e50628c6dccf5',1,'binary_ss(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#af8a791ac7ca88d32cd8f4e9ac0f9ab4f',1,'binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], ['binary_5fsv_13',['binary_sv',['../metal_2kernels_2binary_8h.html#a4116c35f2e4632366d1611d5a95ba141',1,'binary_sv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab18c6ecf5065275c93701efd095c916c',1,'binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5ftwo_14',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]], - ['binary_5fvs_15',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['binary_5fvv_16',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], - ['bits_17',['bits',['../namespacemlx_1_1core_1_1random.html#abb895baa477f5a06b5f88e69245f1825',1,'mlx::core::random::bits(const std::vector< int > &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a548ffed4ba3107b89885ff850ffce5f4',1,'mlx::core::random::bits(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], - ['bits_5fto_5fbfloat_18',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]], - ['bitwise_5fand_19',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]], - ['bitwise_5for_20',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]], - ['bitwise_5fxor_21',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]], - ['bitwisebinary_22',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary']]], - ['block_5fmasked_5fgemm_23',['block_masked_gemm',['../steel__gemm__masked_8h.html#af805e998b2046ee30c2b4be813e3af97',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#a477932e2ae9d49366f7ede6db63f9cac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]], - ['block_5fmasked_5fmm_24',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]], - ['block_5fsort_25',['block_sort',['../struct_kernel_merge_sort.html#a56b644ec66f7fb5c01b280f124304be9',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a322ed2eac315a561e0fd90af2fd577eb',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]], - ['block_5fsort_5fnc_26',['block_sort_nc',['../sort_8h.html#a29229399f51e5c440ffe5c9b99b27598',1,'sort.h']]], - ['blockloader_27',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader']]], - ['blockmaskedmm_28',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM']]], - ['blockmma_29',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA']]], - ['bluestein_5ffft_30',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]], - ['broadcast_31',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html#accbab8433c93e281608a268d11afaefb',1,'mlx::core::Broadcast']]], - ['broadcast_5farrays_32',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]], - ['broadcast_5fshapes_33',['broadcast_shapes',['../namespacemlx_1_1core.html#a075e07def338cd9d815182d0e6a656c0',1,'mlx::core']]], - ['broadcast_5fto_34',['broadcast_to',['../group__ops.html#gad256e86cc1a6e6b3832e392baa90318d',1,'mlx::core']]], - ['bs_5fqmm_5fn_35',['bs_qmm_n',['../quantized_8h.html#acfc43edcf0c742ddc090d22cb7229284',1,'quantized.h']]], - ['bs_5fqmm_5ft_36',['bs_qmm_t',['../quantized_8h.html#a3acc2ace766cb855f13da2d1834e7dc7',1,'quantized.h']]], - ['bs_5fqmv_37',['bs_qmv',['../quantized_8h.html#a03ba4a4a5fe9955319b0aa477d2d7d98',1,'quantized.h']]], - ['bs_5fqmv_5ffast_38',['bs_qmv_fast',['../quantized_8h.html#a6f92bb9f3d29d707bfc680bebe1c80f7',1,'quantized.h']]], - ['bs_5fqvm_39',['bs_qvm',['../quantized_8h.html#a3af1c89416632c9275b8218a33cb8a04',1,'quantized.h']]], - ['buffer_40',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer']]], - ['buffer_41',['buffer',['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]], - ['build_5flib_5fname_42',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]] + ['binary_5fsv2_14',['binary_sv2',['../metal_2kernels_2binary_8h.html#aa8c48b1b21d8f5a181f5443de2346589',1,'binary_sv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a08822ff98ea6f61a98b49a9e9a38b891',1,'binary_sv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5ftwo_15',['binary_two',['../namespacemlx_1_1core_1_1metal.html#aed047eec38b030ec5f29b9da54abf8cb',1,'mlx::core::metal']]], + ['binary_5fvs_16',['binary_vs',['../metal_2kernels_2binary_8h.html#a649851d133358dd5832a73b1061b3313',1,'binary_vs(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca',1,'binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fvs2_17',['binary_vs2',['../metal_2kernels_2binary_8h.html#a48bd82eb10f9c623ce7d28daec4fa512',1,'binary_vs2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a273d2f31691f2c64623c2a97eab344be',1,'binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['binary_5fvv_18',['binary_vv',['../metal_2kernels_2binary_8h.html#add6a9aeee3cb0ba909574f27fa9ecd5b',1,'binary_vv(device const T *a, device const T *b, device U *c, uint index): binary.h'],['../metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9',1,'binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index): binary_two.h']]], + ['binary_5fvv2_19',['binary_vv2',['../metal_2kernels_2binary_8h.html#a19dbbf8fea68b64bdd25dc8d36865171',1,'binary_vv2(device const T *a, device const T *b, device U *c, uint2 index, uint2 grid_dim): binary.h'],['../metal_2kernels_2binary__two_8h.html#a12e80730e43dfaa4c79ce8d5f99edc50',1,'binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim): binary_two.h']]], + ['bits_20',['bits',['../namespacemlx_1_1core_1_1random.html#abb895baa477f5a06b5f88e69245f1825',1,'mlx::core::random::bits(const std::vector< int > &shape, int width, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a548ffed4ba3107b89885ff850ffce5f4',1,'mlx::core::random::bits(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], + ['bits_5fto_5fbfloat_21',['bits_to_bfloat',['../struct___m_l_x___b_float16.html#a91ccb774773b65f8d4c1aea3f1c6e1ca',1,'_MLX_BFloat16']]], + ['bitwise_5fand_22',['bitwise_and',['../group__ops.html#ga752fd2707dabb05d0308ba3d55346ada',1,'mlx::core']]], + ['bitwise_5for_23',['bitwise_or',['../group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6',1,'mlx::core']]], + ['bitwise_5fxor_24',['bitwise_xor',['../group__ops.html#ga3188638fba3a60e264baf69956a1e08b',1,'mlx::core']]], + ['bitwisebinary_25',['BitwiseBinary',['../classmlx_1_1core_1_1_bitwise_binary.html#a0d8b3a94951621ffcdebc6fda748a172',1,'mlx::core::BitwiseBinary']]], + ['block_5fmasked_5fgemm_26',['block_masked_gemm',['../steel__gemm__masked_8h.html#af805e998b2046ee30c2b4be813e3af97',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h'],['../steel__gemm__masked_8h.html#a477932e2ae9d49366f7ede6db63f9cac',1,'block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device bool *out_mask, const device bool *lhs_mask, const device bool *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid): steel_gemm_masked.h']]], + ['block_5fmasked_5fmm_27',['block_masked_mm',['../group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6',1,'mlx::core']]], + ['block_5fsort_28',['block_sort',['../struct_kernel_merge_sort.html#a56b644ec66f7fb5c01b280f124304be9',1,'KernelMergeSort::block_sort()'],['../struct_kernel_multi_block_merge_sort.html#a322ed2eac315a561e0fd90af2fd577eb',1,'KernelMultiBlockMergeSort::block_sort()'],['../sort_8h.html#a93f14092416169c4449141043ac45ffd',1,'block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid): sort.h']]], + ['block_5fsort_5fnc_29',['block_sort_nc',['../sort_8h.html#a29229399f51e5c440ffe5c9b99b27598',1,'sort.h']]], + ['blockloader_30',['BlockLoader',['../structmlx_1_1steel_1_1_block_loader.html#a37aca066e63dff238865b5923a2d4335',1,'mlx::steel::BlockLoader']]], + ['blockmaskedmm_31',['BlockMaskedMM',['../classmlx_1_1core_1_1_block_masked_m_m.html#ad26509deb5306d0c5eb72477e9a57477',1,'mlx::core::BlockMaskedMM']]], + ['blockmma_32',['BlockMMA',['../structmlx_1_1steel_1_1_block_m_m_a.html#aa14406b7298456ac45d23dd3c4642dd8',1,'mlx::steel::BlockMMA']]], + ['bluestein_5ffft_33',['bluestein_fft',['../backend_2metal_2kernels_2fft_8h.html#a0abc609e9756475800e996775a96a87e',1,'fft.h']]], + ['broadcast_34',['Broadcast',['../classmlx_1_1core_1_1_broadcast.html#accbab8433c93e281608a268d11afaefb',1,'mlx::core::Broadcast']]], + ['broadcast_5farrays_35',['broadcast_arrays',['../group__ops.html#gab783890428b596f715dc7dd2057eae99',1,'mlx::core']]], + ['broadcast_5fshapes_36',['broadcast_shapes',['../namespacemlx_1_1core.html#a075e07def338cd9d815182d0e6a656c0',1,'mlx::core']]], + ['broadcast_5fto_37',['broadcast_to',['../group__ops.html#gad256e86cc1a6e6b3832e392baa90318d',1,'mlx::core']]], + ['bs_5fqmm_5fn_38',['bs_qmm_n',['../quantized_8h.html#acfc43edcf0c742ddc090d22cb7229284',1,'quantized.h']]], + ['bs_5fqmm_5ft_39',['bs_qmm_t',['../quantized_8h.html#a3acc2ace766cb855f13da2d1834e7dc7',1,'quantized.h']]], + ['bs_5fqmv_40',['bs_qmv',['../quantized_8h.html#a03ba4a4a5fe9955319b0aa477d2d7d98',1,'quantized.h']]], + ['bs_5fqmv_5ffast_41',['bs_qmv_fast',['../quantized_8h.html#a6f92bb9f3d29d707bfc680bebe1c80f7',1,'quantized.h']]], + ['bs_5fqvm_42',['bs_qvm',['../quantized_8h.html#a3af1c89416632c9275b8218a33cb8a04',1,'quantized.h']]], + ['buffer_43',['Buffer',['../classmlx_1_1core_1_1allocator_1_1_buffer.html#ac4fc2cc6aa1368cfb74aff329d9a1300',1,'mlx::core::allocator::Buffer']]], + ['buffer_44',['buffer',['../classmlx_1_1core_1_1array.html#ab3daf04c27c4593d9d73c397b8484a08',1,'mlx::core::array::buffer()'],['../classmlx_1_1core_1_1array.html#a634466ce661485394f2fdc3bd6796bcd',1,'mlx::core::array::buffer() const']]], + ['build_5flib_5fname_45',['build_lib_name',['../namespacemlx_1_1core.html#a3ef23f334cb9f68a2c50524bc67c913b',1,'mlx::core']]] ]; diff --git a/docs/build/html/search/functions_3.js b/docs/build/html/search/functions_3.js index 58cdabd29f..9d4dec85ab 100644 --- a/docs/build/html/search/functions_3.js +++ b/docs/build/html/search/functions_3.js @@ -11,81 +11,84 @@ var searchData= ['check_5fshape_5fdim_8',['check_shape_dim',['../namespacemlx_1_1core.html#a8b5f1435b7628a094a38b62e403e1540',1,'mlx::core']]], ['cholesky_9',['Cholesky',['../classmlx_1_1core_1_1_cholesky.html#a6ae2e30b85f99f4f0d7f14c7949818ab',1,'mlx::core::Cholesky']]], ['cholesky_10',['cholesky',['../namespacemlx_1_1core_1_1linalg.html#a46c8a4f806f0a97a4323e91189aa512b',1,'mlx::core::linalg']]], - ['clear_5fcache_11',['clear_cache',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a447c1eb38c00d2e8e521675297f4a9b1',1,'mlx::core::metal::MetalAllocator::clear_cache()'],['../namespacemlx_1_1core_1_1metal.html#a22b3384ebd17f2fca198f81b9f1b6dc3',1,'mlx::core::metal::clear_cache()']]], - ['clip_12',['clip',['../group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8',1,'mlx::core']]], - ['cmplx_13',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c',1,'pocketfft::detail::cmplx::cmplx()'],['../structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a',1,'pocketfft::detail::cmplx::cmplx(T r_, T i_)']]], - ['cndarr_14',['cndarr',['../classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129',1,'pocketfft::detail::cndarr']]], - ['col_5freduce_5fgeneral_15',['col_reduce_general',['../reduce__col_8h.html#a3b14333e195c0a07b70069bebf85d5c3',1,'reduce_col.h']]], - ['col_5freduce_5fgeneral_5fno_5fatomics_16',['col_reduce_general_no_atomics',['../reduce__col_8h.html#aef03a4131c710ba8f94a666f58719eb7',1,'reduce_col.h']]], - ['col_5freduce_5fsmall_17',['col_reduce_small',['../reduce__col_8h.html#a7da96ffb66e1fda27bad2852c2285b94',1,'reduce_col.h']]], - ['collapse_5fcontiguous_5fdims_18',['collapse_contiguous_dims',['../namespacemlx_1_1core.html#a9d151ba3e138be1954d2f51f85806b0c',1,'mlx::core::collapse_contiguous_dims(const std::vector< int > &shape, const std::vector< std::vector< stride_t > > strides)'],['../namespacemlx_1_1core.html#a8430e0baac3f6d8a2ab22428f9c0b7e2',1,'mlx::core::collapse_contiguous_dims(const std::vector< array > &xs)'],['../namespacemlx_1_1core.html#ac813412cce77fc1340dcfefc6e099276',1,'mlx::core::collapse_contiguous_dims(Arrays &&... xs)']]], - ['commandencoder_19',['CommandEncoder',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::metal::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::metal::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete']]], - ['commit_5fcommand_5fbuffer_20',['commit_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a95248f1387824067fd4fed23ace5ac0c',1,'mlx::core::metal::Device']]], - ['communication_5fstream_21',['communication_stream',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6',1,'mlx::core::distributed::detail']]], - ['compile_5favailable_5ffor_5fdevice_22',['compile_available_for_device',['../namespacemlx_1_1core_1_1detail.html#aeeff2ba6ec3d9d4ed090de6d2681dbc2',1,'mlx::core::detail']]], - ['compile_5fclear_5fcache_23',['compile_clear_cache',['../namespacemlx_1_1core_1_1detail.html#a3fb927c209b946aefebb195993fbe4cf',1,'mlx::core::detail']]], - ['compile_5ferase_24',['compile_erase',['../namespacemlx_1_1core_1_1detail.html#a69eb76a14f845ca000f1ccb2edda0175',1,'mlx::core::detail']]], - ['compiled_25',['Compiled',['../classmlx_1_1core_1_1_compiled.html#a2d8cefff835c419a48a077d306b8e051',1,'mlx::core::Compiled']]], - ['compiled_5fallocate_5foutputs_26',['compiled_allocate_outputs',['../namespacemlx_1_1core.html#ab8c3c4fc05745f586de922c8266f4fce',1,'mlx::core']]], - ['compiled_5fcheck_5fcontiguity_27',['compiled_check_contiguity',['../namespacemlx_1_1core.html#a3b900ab319948c5a01a3ecd30a709027',1,'mlx::core']]], - ['complex128_5ft_28',['complex128_t',['../structmlx_1_1core_1_1complex128__t.html#aa15d0b805f8790f7c7b76fc7b9d677e0',1,'mlx::core::complex128_t::complex128_t(double v, double u)'],['../structmlx_1_1core_1_1complex128__t.html#abf2842253b874f9f13f39ea68a89e5b6',1,'mlx::core::complex128_t::complex128_t(std::complex< double > v)'],['../structmlx_1_1core_1_1complex128__t.html#a526fba96d7e815360cb4226af085a1bf',1,'mlx::core::complex128_t::complex128_t(T x)']]], - ['complex64_5ft_29',['complex64_t',['../structcomplex64__t.html#adbd392a5e92d31997380ad0a38be4be8',1,'complex64_t::complex64_t(float real, float imag)'],['../structcomplex64__t.html#a33a2452eb33b5ed53655773539c357a5',1,'complex64_t::complex64_t(T x) thread'],['../structcomplex64__t.html#a89b65ace8588b7bf215355f705eb23d9',1,'complex64_t::complex64_t(T x) threadgroup'],['../structcomplex64__t.html#ac81b486f642fb3b26c5d659917bdbcd0',1,'complex64_t::complex64_t(T x) device'],['../structcomplex64__t.html#a0a27a41206400f1e62b60ceb56960c93',1,'complex64_t::complex64_t(T x) const ant'],['../structmlx_1_1core_1_1complex64__t.html#a697cc973ae27d63c8e00d830e780bd8c',1,'mlx::core::complex64_t::complex64_t(float v, float u)'],['../structmlx_1_1core_1_1complex64__t.html#ae065e39938f9c4374b4116f4c67d4d09',1,'mlx::core::complex64_t::complex64_t(std::complex< float > v)'],['../structmlx_1_1core_1_1complex64__t.html#a2232cbbe591a9d2bc228cb23fac38b50',1,'mlx::core::complex64_t::complex64_t(T x)']]], - ['complex_5fmul_30',['complex_mul',['../radix_8h.html#a5bfc53b531214c9ce277bebc18aa67d6',1,'radix.h']]], - ['complex_5fmul_5fconj_31',['complex_mul_conj',['../radix_8h.html#a0e2dfd3d1dda09f47ccc64eec35629f3',1,'radix.h']]], - ['compute_5fstrided_5findices_32',['compute_strided_indices',['../struct_read_writer.html#a7c903fbb8b85a856ba5564d7df537cdf',1,'ReadWriter']]], - ['concatenate_33',['Concatenate',['../classmlx_1_1core_1_1_concatenate.html#acff07853de2d31faeec7c4ca40ce0888',1,'mlx::core::Concatenate']]], - ['concatenate_34',['concatenate',['../group__ops.html#gabdc36fa65697d0361c8d67495de77129',1,'mlx::core::concatenate(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#gaa95c34ca3a8877f2c50cb60e7fa312b8',1,'mlx::core::concatenate(const std::vector< array > &arrays, StreamOrDevice s={})']]], - ['concatenate_5fgpu_35',['concatenate_gpu',['../namespacemlx_1_1core.html#a050299d0d366ca5c9d09d1004dcc3e7d',1,'mlx::core']]], - ['concurrentcontext_36',['ConcurrentContext',['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::metal::CommandEncoder::ConcurrentContext']]], - ['conj_37',['conj',['../namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3',1,'pocketfft::detail']]], - ['conjugate_38',['Conjugate',['../classmlx_1_1core_1_1_conjugate.html#a627f9e6a8729fb3ffb3ca3228d007c87',1,'mlx::core::Conjugate']]], - ['conjugate_39',['conjugate',['../group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23',1,'mlx::core']]], - ['contiguous_5fscan_40',['contiguous_scan',['../kernels_2scan_8h.html#a1f9c197dc09641f3fbb2e886c85b0675',1,'scan.h']]], - ['conv_41',['conv',['../namespacemlx_1_1core_1_1metal.html#ab1704e853394c725668c06752ebb5c24',1,'mlx::core::metal']]], - ['conv1d_42',['conv1d',['../group__ops.html#ga30d47e08093c03a3676f235f9f559411',1,'mlx::core']]], - ['conv2d_43',['conv2d',['../group__ops.html#ga73b02833229678786e7f302d458d5a83',1,'mlx::core']]], - ['conv2dinputblockloadergeneral_44',['Conv2DInputBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1d83af561a483432bf8dcb42e734b23b',1,'mlx::steel::Conv2DInputBlockLoaderGeneral']]], - ['conv2dinputblockloaderlargefilter_45',['Conv2DInputBlockLoaderLargeFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8755116a535539744e4947bc69f9c50f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter']]], - ['conv2dinputblockloadersmallchannels_46',['Conv2DInputBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ab9fd3fdeab94470dde3326f1dd5c455a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels']]], - ['conv2dinputblockloadersmallfilter_47',['Conv2DInputBlockLoaderSmallFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0a2cbf57c51cd928722e3f06aafcf933',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], - ['conv2dweightblockloader_48',['Conv2DWeightBlockLoader',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a9a7dca3512b64cffb6eac305d795831c',1,'mlx::steel::Conv2DWeightBlockLoader']]], - ['conv2dweightblockloadergeneral_49',['Conv2DWeightBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ad0550fabbdc9297559381a5b488e9af1',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], - ['conv2dweightblockloadersmallchannels_50',['Conv2DWeightBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae1806ea1c19713819dee83a38ab35fa6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels']]], - ['conv3d_51',['conv3d',['../group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca',1,'mlx::core']]], - ['conv_5fgeneral_52',['conv_general',['../group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e',1,'mlx::core::conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})'],['../group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d',1,'mlx::core::conv_general(const array &input, const array &weight, std::vector< int > stride={}, std::vector< int > padding={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})']]], - ['convolution_53',['Convolution',['../classmlx_1_1core_1_1_convolution.html#a6f1de77b719bb13217b0d8c64cabb8ef',1,'mlx::core::Convolution']]], - ['copy_54',['Copy',['../classmlx_1_1core_1_1_copy.html#a6243e044af119105ffaaed7d405cd584',1,'mlx::core::Copy']]], - ['copy_55',['copy',['../namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2',1,'mlx::core::copy()'],['../namespacemlx_1_1core_1_1metal.html#aa215e631e2680f04a591b88d91571719',1,'mlx::core::metal::copy()'],['../group__ops.html#gae306e93af12f774bd80bad6c231b09d6',1,'mlx::core::copy()']]], - ['copy_5fg_56',['copy_g',['../metal_2kernels_2copy_8h.html#a1ea80f8edb842dd0ed8ffe553a2238b2',1,'copy.h']]], - ['copy_5fg_5fnd_57',['copy_g_nd',['../metal_2kernels_2copy_8h.html#a07336a7a024125b4f6d3e8556d04058f',1,'copy.h']]], - ['copy_5fg_5fnd1_58',['copy_g_nd1',['../metal_2kernels_2copy_8h.html#aba4530a7db6a61ca36f50e4f5e58fb77',1,'copy.h']]], - ['copy_5fg_5fnd2_59',['copy_g_nd2',['../metal_2kernels_2copy_8h.html#aee678c7c31119f3e609685589f37490c',1,'copy.h']]], - ['copy_5fg_5fnd3_60',['copy_g_nd3',['../metal_2kernels_2copy_8h.html#a821f8f3f3891159a295c66fc25aed1ff',1,'copy.h']]], - ['copy_5fgg_61',['copy_gg',['../metal_2kernels_2copy_8h.html#a9fd1d98f0dcf976089754cd47f55784d',1,'copy.h']]], - ['copy_5fgg_5fnd_62',['copy_gg_nd',['../metal_2kernels_2copy_8h.html#a3197d34e9c68e7059419ae0635667059',1,'copy.h']]], - ['copy_5fgg_5fnd1_63',['copy_gg_nd1',['../metal_2kernels_2copy_8h.html#a3278d9c999718bee3ccbe2922f501bf1',1,'copy.h']]], - ['copy_5fgg_5fnd2_64',['copy_gg_nd2',['../metal_2kernels_2copy_8h.html#a3e2d3cc7f34f56170409b6735f51a950',1,'copy.h']]], - ['copy_5fgg_5fnd3_65',['copy_gg_nd3',['../metal_2kernels_2copy_8h.html#a59f43b5bffed936d7559ceb06a10aabd',1,'copy.h']]], - ['copy_5fgpu_66',['copy_gpu',['../namespacemlx_1_1core.html#addaa46a13ac2deb1d9ce621338320e0e',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a6a6f4e46c8fc44fdc74c50ace02bcf38',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype)']]], - ['copy_5fgpu_5finplace_67',['copy_gpu_inplace',['../namespacemlx_1_1core.html#a69e30f5d30a6d72ac0ffe4886f24b7ba',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a8e1ccb0ed9387b0a789311d9f8964803',1,'mlx::core::copy_gpu_inplace(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#ae55b801b09ccf55cba96278163a9b1ef',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const std::vector< int64_t > &istride, int64_t ioffset, CopyType ctype, const Stream &s)']]], - ['copy_5fhartley_68',['copy_hartley',['../namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], - ['copy_5finplace_69',['copy_inplace',['../namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64',1,'mlx::core::copy_inplace(const array &src, array &dst, CopyType ctype)'],['../namespacemlx_1_1core.html#aad636e2d0b2f882cadd1b438f4daa9ed',1,'mlx::core::copy_inplace(const array &src, array &dst, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)']]], - ['copy_5finput_70',['copy_input',['../namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)'],['../namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, vtype_t< T > *dst)'],['../namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, T *dst)']]], - ['copy_5foutput_71',['copy_output',['../namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)'],['../namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], - ['copy_5fs_72',['copy_s',['../metal_2kernels_2copy_8h.html#aef09f9b9475345b1bba121d037d222ea',1,'copy.h']]], - ['copy_5fshared_5fbuffer_73',['copy_shared_buffer',['../classmlx_1_1core_1_1array.html#a28df7a333d90a311c49bc4bce7a1ad6d',1,'mlx::core::array::copy_shared_buffer(const array &other, const std::vector< size_t > &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a92974c656c35a972ad241f80584bbd29',1,'mlx::core::array::copy_shared_buffer(const array &other)']]], - ['copy_5fv_74',['copy_v',['../metal_2kernels_2copy_8h.html#ae26a13e0c8e6c15f7b10078e65970659',1,'copy.h']]], - ['cos_75',['Cos',['../classmlx_1_1core_1_1_cos.html#a2acb9fcf0901462189c476756fd99995',1,'mlx::core::Cos']]], - ['cos_76',['cos',['../namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d',1,'pocketfft::detail::cos()'],['../namespacemetal.html#a2fa4778a6fe2fa43253ea724e5a608a3',1,'metal::cos()'],['../namespacemetal_1_1fast.html#a75b6bb32fa3870eda46a7bfc9f481f88',1,'metal::fast::cos()'],['../namespacemetal_1_1precise.html#ac4941f62e7d8ab9d7cabbd967aa9f220',1,'metal::precise::cos()'],['../group__ops.html#ga39dfdf72b556012aa35ff27a94116e74',1,'mlx::core::cos()']]], - ['cosh_77',['Cosh',['../classmlx_1_1core_1_1_cosh.html#a44e8ac2e09a55ec32e9dc6641eedc8f1',1,'mlx::core::Cosh']]], - ['cosh_78',['cosh',['../namespacemetal.html#a8a68a88cc110830d057dbd71431b93c0',1,'metal::cosh()'],['../namespacemetal_1_1fast.html#a31544ad9de28012a4ddda86e3966a77e',1,'metal::fast::cosh()'],['../namespacemetal_1_1precise.html#a72d86d508300a9b58f4ccbbe70da4fbc',1,'metal::precise::cosh()'],['../group__ops.html#ga2181b71cda88007a3092be4795ff0715',1,'mlx::core::cosh()']]], - ['cospi_79',['cospi',['../namespacemetal.html#a5c2f37939ad705ddea4409d3bedb8ce1',1,'metal::cospi()'],['../namespacemetal_1_1fast.html#a9906b41f75319b384ffb570cc94d67ce',1,'metal::fast::cospi()'],['../namespacemetal_1_1precise.html#a2392b78bd196efdbbac65901c4ab20e7',1,'metal::precise::cospi()']]], - ['cost_5fguess_80',['cost_guess',['../structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0',1,'pocketfft::detail::util']]], - ['count_5fdown_81',['count_down',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342',1,'pocketfft::detail::threading::latch']]], - ['cummax_82',['cummax',['../group__ops.html#gaee37cac8476e8f8d666bcded5bc59143',1,'mlx::core']]], - ['cummin_83',['cummin',['../group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8',1,'mlx::core']]], - ['cumprod_84',['cumprod',['../group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f',1,'mlx::core']]], - ['cumsum_85',['cumsum',['../group__ops.html#gaddc825a5c173e195ab0fda83ad630420',1,'mlx::core']]], - ['custom_86',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html#a4186fea23f7156c38960426821fca313',1,'mlx::core::fast::Custom']]], - ['customtransforms_87',['CustomTransforms',['../classmlx_1_1core_1_1_custom_transforms.html#ab52abadb9c6f6db83d087c7b751be488',1,'mlx::core::CustomTransforms']]] + ['cholesky_5finv_11',['cholesky_inv',['../namespacemlx_1_1core_1_1linalg.html#aef0fe4894c5cf98792d59859c6d20511',1,'mlx::core::linalg']]], + ['clear_5fcache_12',['clear_cache',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a447c1eb38c00d2e8e521675297f4a9b1',1,'mlx::core::metal::MetalAllocator::clear_cache()'],['../namespacemlx_1_1core_1_1metal.html#a22b3384ebd17f2fca198f81b9f1b6dc3',1,'mlx::core::metal::clear_cache()']]], + ['clip_13',['clip',['../group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8',1,'mlx::core']]], + ['cmplx_14',['cmplx',['../structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c',1,'pocketfft::detail::cmplx::cmplx()'],['../structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a',1,'pocketfft::detail::cmplx::cmplx(T r_, T i_)']]], + ['cndarr_15',['cndarr',['../classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129',1,'pocketfft::detail::cndarr']]], + ['col_5freduce_5fgeneral_16',['col_reduce_general',['../reduce__col_8h.html#a3b14333e195c0a07b70069bebf85d5c3',1,'reduce_col.h']]], + ['col_5freduce_5fgeneral_5fno_5fatomics_17',['col_reduce_general_no_atomics',['../reduce__col_8h.html#aef03a4131c710ba8f94a666f58719eb7',1,'reduce_col.h']]], + ['col_5freduce_5fsmall_18',['col_reduce_small',['../reduce__col_8h.html#a7da96ffb66e1fda27bad2852c2285b94',1,'reduce_col.h']]], + ['collapse_5fcontiguous_5fdims_19',['collapse_contiguous_dims',['../namespacemlx_1_1core.html#a9d151ba3e138be1954d2f51f85806b0c',1,'mlx::core::collapse_contiguous_dims(const std::vector< int > &shape, const std::vector< std::vector< stride_t > > strides)'],['../namespacemlx_1_1core.html#a8430e0baac3f6d8a2ab22428f9c0b7e2',1,'mlx::core::collapse_contiguous_dims(const std::vector< array > &xs)'],['../namespacemlx_1_1core.html#ac813412cce77fc1340dcfefc6e099276',1,'mlx::core::collapse_contiguous_dims(Arrays &&... xs)']]], + ['commandencoder_20',['CommandEncoder',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a2334774486f447213ee997e55c2e52a3',1,'mlx::core::metal::CommandEncoder::CommandEncoder(MTL::CommandBuffer *cbuf)'],['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#ac68ca977b5bde5434284ce7979647f14',1,'mlx::core::metal::CommandEncoder::CommandEncoder(const CommandEncoder &)=delete']]], + ['commit_5fcommand_5fbuffer_21',['commit_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a95248f1387824067fd4fed23ace5ac0c',1,'mlx::core::metal::Device']]], + ['communication_5fstream_22',['communication_stream',['../namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6',1,'mlx::core::distributed::detail']]], + ['compile_5favailable_5ffor_5fdevice_23',['compile_available_for_device',['../namespacemlx_1_1core_1_1detail.html#aeeff2ba6ec3d9d4ed090de6d2681dbc2',1,'mlx::core::detail']]], + ['compile_5fclear_5fcache_24',['compile_clear_cache',['../namespacemlx_1_1core_1_1detail.html#a3fb927c209b946aefebb195993fbe4cf',1,'mlx::core::detail']]], + ['compile_5ferase_25',['compile_erase',['../namespacemlx_1_1core_1_1detail.html#a69eb76a14f845ca000f1ccb2edda0175',1,'mlx::core::detail']]], + ['compiled_26',['Compiled',['../classmlx_1_1core_1_1_compiled.html#a2d8cefff835c419a48a077d306b8e051',1,'mlx::core::Compiled']]], + ['compiled_5fallocate_5foutputs_27',['compiled_allocate_outputs',['../namespacemlx_1_1core.html#ab8c3c4fc05745f586de922c8266f4fce',1,'mlx::core']]], + ['compiled_5fcheck_5fcontiguity_28',['compiled_check_contiguity',['../namespacemlx_1_1core.html#a3b900ab319948c5a01a3ecd30a709027',1,'mlx::core']]], + ['complex128_5ft_29',['complex128_t',['../structmlx_1_1core_1_1complex128__t.html#aa15d0b805f8790f7c7b76fc7b9d677e0',1,'mlx::core::complex128_t::complex128_t(double v, double u)'],['../structmlx_1_1core_1_1complex128__t.html#abf2842253b874f9f13f39ea68a89e5b6',1,'mlx::core::complex128_t::complex128_t(std::complex< double > v)'],['../structmlx_1_1core_1_1complex128__t.html#a526fba96d7e815360cb4226af085a1bf',1,'mlx::core::complex128_t::complex128_t(T x)']]], + ['complex64_5ft_30',['complex64_t',['../structcomplex64__t.html#adbd392a5e92d31997380ad0a38be4be8',1,'complex64_t::complex64_t(float real, float imag)'],['../structcomplex64__t.html#a33a2452eb33b5ed53655773539c357a5',1,'complex64_t::complex64_t(T x) thread'],['../structcomplex64__t.html#a89b65ace8588b7bf215355f705eb23d9',1,'complex64_t::complex64_t(T x) threadgroup'],['../structcomplex64__t.html#ac81b486f642fb3b26c5d659917bdbcd0',1,'complex64_t::complex64_t(T x) device'],['../structcomplex64__t.html#a0a27a41206400f1e62b60ceb56960c93',1,'complex64_t::complex64_t(T x) const ant'],['../structmlx_1_1core_1_1complex64__t.html#a697cc973ae27d63c8e00d830e780bd8c',1,'mlx::core::complex64_t::complex64_t(float v, float u)'],['../structmlx_1_1core_1_1complex64__t.html#ae065e39938f9c4374b4116f4c67d4d09',1,'mlx::core::complex64_t::complex64_t(std::complex< float > v)'],['../structmlx_1_1core_1_1complex64__t.html#a2232cbbe591a9d2bc228cb23fac38b50',1,'mlx::core::complex64_t::complex64_t(T x)']]], + ['complex_5fmul_31',['complex_mul',['../radix_8h.html#a5bfc53b531214c9ce277bebc18aa67d6',1,'radix.h']]], + ['complex_5fmul_5fconj_32',['complex_mul_conj',['../radix_8h.html#a0e2dfd3d1dda09f47ccc64eec35629f3',1,'radix.h']]], + ['compute_5fstrided_5findices_33',['compute_strided_indices',['../struct_read_writer.html#a7c903fbb8b85a856ba5564d7df537cdf',1,'ReadWriter']]], + ['concatenate_34',['Concatenate',['../classmlx_1_1core_1_1_concatenate.html#acff07853de2d31faeec7c4ca40ce0888',1,'mlx::core::Concatenate']]], + ['concatenate_35',['concatenate',['../group__ops.html#gabdc36fa65697d0361c8d67495de77129',1,'mlx::core::concatenate(const std::vector< array > &arrays, int axis, StreamOrDevice s={})'],['../group__ops.html#gaa95c34ca3a8877f2c50cb60e7fa312b8',1,'mlx::core::concatenate(const std::vector< array > &arrays, StreamOrDevice s={})']]], + ['concatenate_5fgpu_36',['concatenate_gpu',['../namespacemlx_1_1core.html#a050299d0d366ca5c9d09d1004dcc3e7d',1,'mlx::core']]], + ['concurrentcontext_37',['ConcurrentContext',['../structmlx_1_1core_1_1metal_1_1_command_encoder_1_1_concurrent_context.html#aee044d7729739c96e845823f9ecc5174',1,'mlx::core::metal::CommandEncoder::ConcurrentContext']]], + ['conj_38',['conj',['../namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3',1,'pocketfft::detail']]], + ['conjugate_39',['Conjugate',['../classmlx_1_1core_1_1_conjugate.html#a627f9e6a8729fb3ffb3ca3228d007c87',1,'mlx::core::Conjugate']]], + ['conjugate_40',['conjugate',['../group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23',1,'mlx::core']]], + ['contiguous_5fscan_41',['contiguous_scan',['../kernels_2scan_8h.html#a1f9c197dc09641f3fbb2e886c85b0675',1,'scan.h']]], + ['conv_42',['conv',['../namespacemlx_1_1core_1_1metal.html#ab1704e853394c725668c06752ebb5c24',1,'mlx::core::metal']]], + ['conv1d_43',['conv1d',['../group__ops.html#ga30d47e08093c03a3676f235f9f559411',1,'mlx::core']]], + ['conv2d_44',['conv2d',['../group__ops.html#ga73b02833229678786e7f302d458d5a83',1,'mlx::core']]], + ['conv2dinputblockloadergeneral_45',['Conv2DInputBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a1d83af561a483432bf8dcb42e734b23b',1,'mlx::steel::Conv2DInputBlockLoaderGeneral']]], + ['conv2dinputblockloaderlargefilter_46',['Conv2DInputBlockLoaderLargeFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a8755116a535539744e4947bc69f9c50f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter']]], + ['conv2dinputblockloadersmallchannels_47',['Conv2DInputBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ab9fd3fdeab94470dde3326f1dd5c455a',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels']]], + ['conv2dinputblockloadersmallfilter_48',['Conv2DInputBlockLoaderSmallFilter',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a0a2cbf57c51cd928722e3f06aafcf933',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter']]], + ['conv2dweightblockloader_49',['Conv2DWeightBlockLoader',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a9a7dca3512b64cffb6eac305d795831c',1,'mlx::steel::Conv2DWeightBlockLoader']]], + ['conv2dweightblockloadergeneral_50',['Conv2DWeightBlockLoaderGeneral',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#ad0550fabbdc9297559381a5b488e9af1',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral']]], + ['conv2dweightblockloadersmallchannels_51',['Conv2DWeightBlockLoaderSmallChannels',['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae1806ea1c19713819dee83a38ab35fa6',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels']]], + ['conv3d_52',['conv3d',['../group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca',1,'mlx::core']]], + ['conv_5fgeneral_53',['conv_general',['../group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e',1,'mlx::core::conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})'],['../group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d',1,'mlx::core::conv_general(const array &input, const array &weight, std::vector< int > stride={}, std::vector< int > padding={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})']]], + ['convolution_54',['Convolution',['../classmlx_1_1core_1_1_convolution.html#a6f1de77b719bb13217b0d8c64cabb8ef',1,'mlx::core::Convolution']]], + ['copy_55',['Copy',['../classmlx_1_1core_1_1_copy.html#a6243e044af119105ffaaed7d405cd584',1,'mlx::core::Copy']]], + ['copy_56',['copy',['../namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2',1,'mlx::core::copy()'],['../namespacemlx_1_1core_1_1metal.html#aa215e631e2680f04a591b88d91571719',1,'mlx::core::metal::copy()'],['../group__ops.html#gae306e93af12f774bd80bad6c231b09d6',1,'mlx::core::copy()']]], + ['copy_5fg_57',['copy_g',['../metal_2kernels_2copy_8h.html#a1ea80f8edb842dd0ed8ffe553a2238b2',1,'copy.h']]], + ['copy_5fg_5fnd_58',['copy_g_nd',['../metal_2kernels_2copy_8h.html#a07336a7a024125b4f6d3e8556d04058f',1,'copy.h']]], + ['copy_5fg_5fnd1_59',['copy_g_nd1',['../metal_2kernels_2copy_8h.html#aba4530a7db6a61ca36f50e4f5e58fb77',1,'copy.h']]], + ['copy_5fg_5fnd2_60',['copy_g_nd2',['../metal_2kernels_2copy_8h.html#aee678c7c31119f3e609685589f37490c',1,'copy.h']]], + ['copy_5fg_5fnd3_61',['copy_g_nd3',['../metal_2kernels_2copy_8h.html#a821f8f3f3891159a295c66fc25aed1ff',1,'copy.h']]], + ['copy_5fgg_62',['copy_gg',['../metal_2kernels_2copy_8h.html#a9fd1d98f0dcf976089754cd47f55784d',1,'copy.h']]], + ['copy_5fgg_5fnd_63',['copy_gg_nd',['../metal_2kernels_2copy_8h.html#a3197d34e9c68e7059419ae0635667059',1,'copy.h']]], + ['copy_5fgg_5fnd1_64',['copy_gg_nd1',['../metal_2kernels_2copy_8h.html#a3278d9c999718bee3ccbe2922f501bf1',1,'copy.h']]], + ['copy_5fgg_5fnd2_65',['copy_gg_nd2',['../metal_2kernels_2copy_8h.html#a3e2d3cc7f34f56170409b6735f51a950',1,'copy.h']]], + ['copy_5fgg_5fnd3_66',['copy_gg_nd3',['../metal_2kernels_2copy_8h.html#a59f43b5bffed936d7559ceb06a10aabd',1,'copy.h']]], + ['copy_5fgpu_67',['copy_gpu',['../namespacemlx_1_1core.html#addaa46a13ac2deb1d9ce621338320e0e',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a6a6f4e46c8fc44fdc74c50ace02bcf38',1,'mlx::core::copy_gpu(const array &src, array &out, CopyType ctype)']]], + ['copy_5fgpu_5finplace_68',['copy_gpu_inplace',['../namespacemlx_1_1core.html#a69e30f5d30a6d72ac0ffe4886f24b7ba',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#a8e1ccb0ed9387b0a789311d9f8964803',1,'mlx::core::copy_gpu_inplace(const array &src, array &out, CopyType ctype, const Stream &s)'],['../namespacemlx_1_1core.html#ae55b801b09ccf55cba96278163a9b1ef',1,'mlx::core::copy_gpu_inplace(const array &in, array &out, const std::vector< int64_t > &istride, int64_t ioffset, CopyType ctype, const Stream &s)']]], + ['copy_5fhartley_69',['copy_hartley',['../namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed',1,'pocketfft::detail::copy_hartley(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], + ['copy_5finplace_70',['copy_inplace',['../namespacemlx_1_1core.html#a98495894a796b2cc6d022e7a03432c64',1,'mlx::core::copy_inplace(const array &src, array &dst, CopyType ctype)'],['../namespacemlx_1_1core.html#aad636e2d0b2f882cadd1b438f4daa9ed',1,'mlx::core::copy_inplace(const array &src, array &dst, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)']]], + ['copy_5finput_71',['copy_input',['../namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)'],['../namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, vtype_t< T > *dst)'],['../namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4',1,'pocketfft::detail::copy_input(const multi_iter< vlen > &it, const cndarr< T > &src, T *dst)']]], + ['copy_5foutput_72',['copy_output',['../namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)'],['../namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)'],['../namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0',1,'pocketfft::detail::copy_output(const multi_iter< vlen > &it, const T *src, ndarr< T > &dst)']]], + ['copy_5fs_73',['copy_s',['../metal_2kernels_2copy_8h.html#aef09f9b9475345b1bba121d037d222ea',1,'copy.h']]], + ['copy_5fs2_74',['copy_s2',['../metal_2kernels_2copy_8h.html#a8023e9335cc5334847a8d315042be3a3',1,'copy.h']]], + ['copy_5fshared_5fbuffer_75',['copy_shared_buffer',['../classmlx_1_1core_1_1array.html#a28df7a333d90a311c49bc4bce7a1ad6d',1,'mlx::core::array::copy_shared_buffer(const array &other, const std::vector< size_t > &strides, Flags flags, size_t data_size, size_t offset=0)'],['../classmlx_1_1core_1_1array.html#a92974c656c35a972ad241f80584bbd29',1,'mlx::core::array::copy_shared_buffer(const array &other)']]], + ['copy_5fv_76',['copy_v',['../metal_2kernels_2copy_8h.html#ae26a13e0c8e6c15f7b10078e65970659',1,'copy.h']]], + ['copy_5fv2_77',['copy_v2',['../metal_2kernels_2copy_8h.html#aee14a5326f53d9b30b0b38e27d180ef3',1,'copy.h']]], + ['cos_78',['Cos',['../classmlx_1_1core_1_1_cos.html#a2acb9fcf0901462189c476756fd99995',1,'mlx::core::Cos']]], + ['cos_79',['cos',['../namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d',1,'pocketfft::detail::cos()'],['../namespacemetal.html#a2fa4778a6fe2fa43253ea724e5a608a3',1,'metal::cos()'],['../namespacemetal_1_1fast.html#a75b6bb32fa3870eda46a7bfc9f481f88',1,'metal::fast::cos()'],['../namespacemetal_1_1precise.html#ac4941f62e7d8ab9d7cabbd967aa9f220',1,'metal::precise::cos()'],['../group__ops.html#ga39dfdf72b556012aa35ff27a94116e74',1,'mlx::core::cos()']]], + ['cosh_80',['Cosh',['../classmlx_1_1core_1_1_cosh.html#a44e8ac2e09a55ec32e9dc6641eedc8f1',1,'mlx::core::Cosh']]], + ['cosh_81',['cosh',['../namespacemetal.html#a8a68a88cc110830d057dbd71431b93c0',1,'metal::cosh()'],['../namespacemetal_1_1fast.html#a31544ad9de28012a4ddda86e3966a77e',1,'metal::fast::cosh()'],['../namespacemetal_1_1precise.html#a72d86d508300a9b58f4ccbbe70da4fbc',1,'metal::precise::cosh()'],['../group__ops.html#ga2181b71cda88007a3092be4795ff0715',1,'mlx::core::cosh()']]], + ['cospi_82',['cospi',['../namespacemetal.html#a5c2f37939ad705ddea4409d3bedb8ce1',1,'metal::cospi()'],['../namespacemetal_1_1fast.html#a9906b41f75319b384ffb570cc94d67ce',1,'metal::fast::cospi()'],['../namespacemetal_1_1precise.html#a2392b78bd196efdbbac65901c4ab20e7',1,'metal::precise::cospi()']]], + ['cost_5fguess_83',['cost_guess',['../structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0',1,'pocketfft::detail::util']]], + ['count_5fdown_84',['count_down',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342',1,'pocketfft::detail::threading::latch']]], + ['cummax_85',['cummax',['../group__ops.html#gaee37cac8476e8f8d666bcded5bc59143',1,'mlx::core']]], + ['cummin_86',['cummin',['../group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8',1,'mlx::core']]], + ['cumprod_87',['cumprod',['../group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f',1,'mlx::core']]], + ['cumsum_88',['cumsum',['../group__ops.html#gaddc825a5c173e195ab0fda83ad630420',1,'mlx::core']]], + ['custom_89',['Custom',['../classmlx_1_1core_1_1fast_1_1_custom.html#a4186fea23f7156c38960426821fca313',1,'mlx::core::fast::Custom']]], + ['customtransforms_90',['CustomTransforms',['../classmlx_1_1core_1_1_custom_transforms.html#ab52abadb9c6f6db83d087c7b751be488',1,'mlx::core::CustomTransforms']]] ]; diff --git a/docs/build/html/search/functions_4.js b/docs/build/html/search/functions_4.js index 5135ce6c6f..a3d2e98ddd 100644 --- a/docs/build/html/search/functions_4.js +++ b/docs/build/html/search/functions_4.js @@ -6,35 +6,35 @@ var searchData= ['data_5fsize_3',['data_size',['../classmlx_1_1core_1_1array.html#afaf2a370fa35d96af1b27a4b814e3bfd',1,'mlx::core::array']]], ['dct_4',['dct',['../namespacepocketfft_1_1detail.html#a60615f5b685314c658346c309d5ef2ba',1,'pocketfft::detail']]], ['deallocate_5',['deallocate',['../structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a2a99b8e296d26b255e9937ba5f30e76f',1,'pocketfft::detail::threading::aligned_allocator']]], - ['decompose_5fhadamard_6',['decompose_hadamard',['../namespacemlx_1_1core.html#a3a8fe7ba84714dbb5fdc81e93a07abc8',1,'mlx::core']]], - ['default_5f_7',['default_',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#ab5993daeed822c6b970caddab7e3fd90',1,'mlx::core::random::KeySequence']]], - ['default_5fdevice_8',['default_device',['../namespacemlx_1_1core.html#a0196171cfe6ee2953113abce597dc815',1,'mlx::core']]], - ['default_5fstream_9',['default_stream',['../namespacemlx_1_1core.html#ac198b7e282957c724c84a435e8f1215e',1,'mlx::core']]], - ['defaultcontiguousreduce_10',['DefaultContiguousReduce',['../structmlx_1_1core_1_1_default_contiguous_reduce.html#aeb4fb7fa1a4c8e7d1da1f450ce95c57f',1,'mlx::core::DefaultContiguousReduce']]], - ['defaultstridedreduce_11',['DefaultStridedReduce',['../structmlx_1_1core_1_1_default_strided_reduce.html#a477e5dd0dd33071e48769d11d19a13fb',1,'mlx::core::DefaultStridedReduce']]], - ['define_5fprint_12',['DEFINE_PRINT',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a8af1e90d4aa56f31ec40ad152ebd2421',1,'mlx::core::distributed::AllGather::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae6eea81b5e3789c2f6f376cc07f0a47c',1,'mlx::core::fast::RMSNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a9895733eab845e11484d86cf6ecedced',1,'mlx::core::fast::RMSNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a467fcf02b3ddf1d8b6d476b244ae3568',1,'mlx::core::fast::LayerNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a5ab3eb5402c7e8060916056eb2b7887f',1,'mlx::core::fast::LayerNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a2b06fe64fa8feca65140632087065e16',1,'mlx::core::fast::RoPE::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6cc2092fa5b8e7585921b8e0f3ec3db7',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_PRINT()']]], - ['degrees_13',['degrees',['../group__ops.html#ga3a70569b50e1083c5ded199d73fb960c',1,'mlx::core']]], - ['denorm_5fmin_14',['denorm_min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a6a9dbcba4dd79cad50876dda506b9eed',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['depends_15',['Depends',['../classmlx_1_1core_1_1_depends.html#a4ccb792c99f5d8d133d3fac29f7d3f62',1,'mlx::core::Depends']]], - ['depends_16',['depends',['../group__ops.html#gac4a51a68fbe1725436b026d2fbb95759',1,'mlx::core']]], - ['dequantize_17',['dequantize',['../quantized_8h.html#aecff265b63566d0d5689cfc4e5b037d2',1,'dequantize(): quantized.h'],['../group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077',1,'mlx::core::dequantize()']]], - ['detach_18',['detach',['../classmlx_1_1core_1_1array.html#a84948c29df8c957904919c8602692bd2',1,'mlx::core::array']]], - ['device_19',['Device',['../classmlx_1_1core_1_1metal_1_1_device.html#ae0db74570eb4b19d8cf19774db91bfd6',1,'mlx::core::metal::Device::Device()'],['../classmlx_1_1core_1_1metal_1_1_device.html#abf59a4addb5473f9e814e3651ba85f06',1,'mlx::core::metal::Device::Device(const Device &)=delete'],['../structmlx_1_1core_1_1_device.html#a481ccfb94d689994396bd353e966b489',1,'mlx::core::Device::Device()']]], - ['device_20',['device',['../class_m_p_s_1_1_kernel.html#a46793d1c80a4df313c067103fcae6024',1,'MPS::Kernel::device()'],['../classmlx_1_1core_1_1_primitive.html#a8ae61e3289c4134232a69295268f8261',1,'mlx::core::Primitive::device()'],['../namespacemlx_1_1core_1_1metal.html#a910797b74824e6ee576fbb533dee8b57',1,'mlx::core::metal::device(mlx::core::Device)']]], - ['device_5finfo_21',['device_info',['../namespacemlx_1_1core_1_1metal.html#a6ad19c44efabb7423f973407926ead61',1,'mlx::core::metal']]], - ['diag_22',['diag',['../group__ops.html#ga11af511875640e1fa88e0ca87e199344',1,'mlx::core']]], - ['diagonal_23',['diagonal',['../group__ops.html#ga9236b085a88ead3128ed8079d009cac6',1,'mlx::core']]], - ['disable_5fcompile_24',['disable_compile',['../namespacemlx_1_1core.html#a5f5fea955057bb3842b271b037909e66',1,'mlx::core']]], - ['dispatchthreadgroups_25',['dispatchThreadgroups',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a74bcd8e35f80f5a62db48c4a2bb0173e',1,'mlx::core::metal::CommandEncoder']]], - ['dispatchthreads_26',['dispatchThreads',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a1e41477f2f489e38499f7830a91c9810',1,'mlx::core::metal::CommandEncoder']]], - ['distprimitive_27',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a9e4b0b92776aca0c1331450ae4955cef',1,'mlx::core::distributed::DistPrimitive']]], - ['divide_28',['Divide',['../classmlx_1_1core_1_1_divide.html#a62fc71e8998be65ff18285dbbd21eedb',1,'mlx::core::Divide']]], - ['divide_29',['divide',['../namespacemetal.html#a2aea493fc1a874970b77ed0031e965df',1,'metal::divide()'],['../namespacemetal_1_1fast.html#ae70bc2185e4649369cf7b15f5e1d48be',1,'metal::fast::divide()'],['../namespacemetal_1_1precise.html#aec0982cdb96a08b61f51129150d82e9d',1,'metal::precise::divide()'],['../group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f',1,'mlx::core::divide()']]], - ['divmod_30',['DivMod',['../classmlx_1_1core_1_1_div_mod.html#a859e3b6149cdceab1c7ccfd2246fb826',1,'mlx::core::DivMod']]], - ['divmod_31',['divmod',['../group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894',1,'mlx::core']]], - ['dst_32',['dst',['../namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::detail']]], - ['dtype_33',['dtype',['../classmlx_1_1core_1_1array.html#ae29e7d6fbfbea1e5e321a8d1ea3cfacd',1,'mlx::core::array']]], - ['dtype_34',['Dtype',['../structmlx_1_1core_1_1_dtype.html#aec17f0a4a51729e5ac40b62f0aa765d1',1,'mlx::core::Dtype']]], - ['dtype_5ffrom_5farray_5fprotocol_35',['dtype_from_array_protocol',['../namespacemlx_1_1core.html#a14e0dfe96b41a04be91553799eb07594',1,'mlx::core']]], - ['dtype_5fto_5farray_5fprotocol_36',['dtype_to_array_protocol',['../namespacemlx_1_1core.html#a80509820694d2f74f0d262d13a90774b',1,'mlx::core']]] + ['debug_5fset_5fprimitive_5fbuffer_5flabel_6',['debug_set_primitive_buffer_label',['../namespacemlx_1_1core.html#a489e45b3a5cd8b46e8ea56b9132eb230',1,'mlx::core']]], + ['debug_5fset_5fstream_5fqueue_5flabel_7',['debug_set_stream_queue_label',['../namespacemlx_1_1core.html#a79817d2432e782e596c9c49a08b93be2',1,'mlx::core']]], + ['decompose_5fhadamard_8',['decompose_hadamard',['../namespacemlx_1_1core.html#a3a8fe7ba84714dbb5fdc81e93a07abc8',1,'mlx::core']]], + ['default_5f_9',['default_',['../classmlx_1_1core_1_1random_1_1_key_sequence.html#ab5993daeed822c6b970caddab7e3fd90',1,'mlx::core::random::KeySequence']]], + ['default_5fdevice_10',['default_device',['../namespacemlx_1_1core.html#a0196171cfe6ee2953113abce597dc815',1,'mlx::core']]], + ['default_5fstream_11',['default_stream',['../namespacemlx_1_1core.html#ac198b7e282957c724c84a435e8f1215e',1,'mlx::core']]], + ['defaultcontiguousreduce_12',['DefaultContiguousReduce',['../structmlx_1_1core_1_1_default_contiguous_reduce.html#aeb4fb7fa1a4c8e7d1da1f450ce95c57f',1,'mlx::core::DefaultContiguousReduce']]], + ['defaultstridedreduce_13',['DefaultStridedReduce',['../structmlx_1_1core_1_1_default_strided_reduce.html#a477e5dd0dd33071e48769d11d19a13fb',1,'mlx::core::DefaultStridedReduce']]], + ['define_5fprint_14',['DEFINE_PRINT',['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#a8af1e90d4aa56f31ec40ad152ebd2421',1,'mlx::core::distributed::AllGather::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae6eea81b5e3789c2f6f376cc07f0a47c',1,'mlx::core::fast::RMSNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a9895733eab845e11484d86cf6ecedced',1,'mlx::core::fast::RMSNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a467fcf02b3ddf1d8b6d476b244ae3568',1,'mlx::core::fast::LayerNorm::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a5ab3eb5402c7e8060916056eb2b7887f',1,'mlx::core::fast::LayerNormVJP::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a2b06fe64fa8feca65140632087065e16',1,'mlx::core::fast::RoPE::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a6cc2092fa5b8e7585921b8e0f3ec3db7',1,'mlx::core::fast::ScaledDotProductAttention::DEFINE_PRINT()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a4b8f1b1f633002c8ca6fa8f0ef4dd587',1,'mlx::core::fast::AffineQuantize::DEFINE_PRINT()']]], + ['degrees_15',['degrees',['../group__ops.html#ga3a70569b50e1083c5ded199d73fb960c',1,'mlx::core']]], + ['denorm_5fmin_16',['denorm_min',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a6a9dbcba4dd79cad50876dda506b9eed',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['depends_17',['Depends',['../classmlx_1_1core_1_1_depends.html#a4ccb792c99f5d8d133d3fac29f7d3f62',1,'mlx::core::Depends']]], + ['depends_18',['depends',['../group__ops.html#gac4a51a68fbe1725436b026d2fbb95759',1,'mlx::core']]], + ['dequantize_19',['dequantize',['../quantized_8h.html#aecff265b63566d0d5689cfc4e5b037d2',1,'dequantize(): quantized.h'],['../group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077',1,'mlx::core::dequantize()']]], + ['detach_20',['detach',['../classmlx_1_1core_1_1array.html#a84948c29df8c957904919c8602692bd2',1,'mlx::core::array']]], + ['device_21',['Device',['../classmlx_1_1core_1_1metal_1_1_device.html#ae0db74570eb4b19d8cf19774db91bfd6',1,'mlx::core::metal::Device::Device()'],['../classmlx_1_1core_1_1metal_1_1_device.html#abf59a4addb5473f9e814e3651ba85f06',1,'mlx::core::metal::Device::Device(const Device &)=delete'],['../structmlx_1_1core_1_1_device.html#a481ccfb94d689994396bd353e966b489',1,'mlx::core::Device::Device()']]], + ['device_22',['device',['../classmlx_1_1core_1_1_primitive.html#a8ae61e3289c4134232a69295268f8261',1,'mlx::core::Primitive::device()'],['../namespacemlx_1_1core_1_1metal.html#a910797b74824e6ee576fbb533dee8b57',1,'mlx::core::metal::device(mlx::core::Device)']]], + ['device_5finfo_23',['device_info',['../namespacemlx_1_1core_1_1metal.html#a6ad19c44efabb7423f973407926ead61',1,'mlx::core::metal']]], + ['diag_24',['diag',['../group__ops.html#ga11af511875640e1fa88e0ca87e199344',1,'mlx::core']]], + ['diagonal_25',['diagonal',['../group__ops.html#ga9236b085a88ead3128ed8079d009cac6',1,'mlx::core']]], + ['disable_5fcompile_26',['disable_compile',['../namespacemlx_1_1core.html#a5f5fea955057bb3842b271b037909e66',1,'mlx::core']]], + ['dispatchthreadgroups_27',['dispatchThreadgroups',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a74bcd8e35f80f5a62db48c4a2bb0173e',1,'mlx::core::metal::CommandEncoder']]], + ['dispatchthreads_28',['dispatchThreads',['../structmlx_1_1core_1_1metal_1_1_command_encoder.html#a1e41477f2f489e38499f7830a91c9810',1,'mlx::core::metal::CommandEncoder']]], + ['distprimitive_29',['DistPrimitive',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a9e4b0b92776aca0c1331450ae4955cef',1,'mlx::core::distributed::DistPrimitive']]], + ['divide_30',['Divide',['../classmlx_1_1core_1_1_divide.html#a62fc71e8998be65ff18285dbbd21eedb',1,'mlx::core::Divide']]], + ['divide_31',['divide',['../namespacemetal.html#a2aea493fc1a874970b77ed0031e965df',1,'metal::divide()'],['../namespacemetal_1_1fast.html#ae70bc2185e4649369cf7b15f5e1d48be',1,'metal::fast::divide()'],['../namespacemetal_1_1precise.html#aec0982cdb96a08b61f51129150d82e9d',1,'metal::precise::divide()'],['../group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f',1,'mlx::core::divide()']]], + ['divmod_32',['DivMod',['../classmlx_1_1core_1_1_div_mod.html#a859e3b6149cdceab1c7ccfd2246fb826',1,'mlx::core::DivMod']]], + ['divmod_33',['divmod',['../group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894',1,'mlx::core']]], + ['dst_34',['dst',['../namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb',1,'pocketfft::detail']]], + ['dtype_35',['Dtype',['../structmlx_1_1core_1_1_dtype.html#aec17f0a4a51729e5ac40b62f0aa765d1',1,'mlx::core::Dtype']]], + ['dtype_36',['dtype',['../classmlx_1_1core_1_1array.html#ae29e7d6fbfbea1e5e321a8d1ea3cfacd',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/functions_5.js b/docs/build/html/search/functions_5.js index 9977d9d220..c635785cb1 100644 --- a/docs/build/html/search/functions_5.js +++ b/docs/build/html/search/functions_5.js @@ -12,33 +12,32 @@ var searchData= ['elem_5fto_5floc_5fnd_9',['elem_to_loc_nd',['../backend_2metal_2kernels_2utils_8h.html#aa25c926e32ba8f05de765c662326d955',1,'elem_to_loc_nd(uint elem, device const int *shape, device const size_t *strides): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a0d04f0d0718d0a5796ce5ca1a289d942',1,'elem_to_loc_nd(uint3 elem, constant const int shape[NDIM], constant const size_t strides[NDIM]): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#ac7d74fb6d5fed31513b6b7defcf45921',1,'elem_to_loc_nd(uint elem, constant const int shape[NDIM], constant const int64_t strides[NDIM]): utils.h'],['../backend_2metal_2kernels_2utils_8h.html#a4fec636fff34a288ccd56ce202703232',1,'elem_to_loc_nd(uint3 elem, constant const int shape[NDIM], constant const int64_t strides[NDIM]): utils.h']]], ['empty_10',['empty',['../classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#a1269e5da40c3f5145c895cee3641879a',1,'pocketfft::detail::threading::concurrent_queue']]], ['enable_5fcompile_11',['enable_compile',['../namespacemlx_1_1core.html#a1983a2466bff3bae4d23cf34bd0946c9',1,'mlx::core']]], - ['encodetocommandbuffer_12',['encodeToCommandBuffer',['../class_m_p_s_1_1_matrix_multiplication.html#ad30480456e66b79fa47086a3411c4921',1,'MPS::MatrixMultiplication::encodeToCommandBuffer()'],['../class_m_p_s_1_1_matrix_vector_multiplication.html#adefd2759c843035c73da2d9aad8ffb8a',1,'MPS::MatrixVectorMultiplication::encodeToCommandBuffer()']]], - ['end_13',['end',['../classmlx_1_1core_1_1array.html#a5daf64552fb450825c9b382f3a5fa2d4',1,'mlx::core::array']]], - ['end_5fencoding_14',['end_encoding',['../classmlx_1_1core_1_1metal_1_1_device.html#a60689f97347811b27e8c5ca23e0372bf',1,'mlx::core::metal::Device']]], - ['enqueue_15',['enqueue',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4918720319cf224a1b4208568964c286',1,'mlx::core::scheduler::StreamThread::enqueue()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a03809c783bd1866362dc7cb9118abbcc',1,'mlx::core::scheduler::Scheduler::enqueue()'],['../namespacemlx_1_1core_1_1scheduler.html#aa2d4eacf5d5cbc778a51aafd4fd8e4d7',1,'mlx::core::scheduler::enqueue()']]], - ['epsilon_16',['epsilon',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a96c4197e3076f0aa9065370b8ece49ca',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['equal_17',['Equal',['../classmlx_1_1core_1_1_equal.html#a4af81cf2dd071db5bbf8ce1df95fdf36',1,'mlx::core::Equal']]], - ['equal_18',['equal',['../group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde',1,'mlx::core']]], - ['erf_19',['Erf',['../classmlx_1_1core_1_1_erf.html#a702f76f848928d8d7d3d0881ac6e4c82',1,'mlx::core::Erf']]], - ['erf_20',['erf',['../erf_8h.html#a6ce199ee56105c67adbf8c48c019a8b2',1,'erf(): erf.h'],['../group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0',1,'mlx::core::erf()']]], - ['erfinv_21',['ErfInv',['../classmlx_1_1core_1_1_erf_inv.html#a5d0279247b67da4592311559f04e1478',1,'mlx::core::ErfInv']]], - ['erfinv_22',['erfinv',['../erf_8h.html#a1846e0d683c7aff826bb32addcc3b885',1,'erfinv(): erf.h'],['../group__ops.html#ga76fb9062c64264e34d2e07013390557c',1,'mlx::core::erfinv()']]], - ['eval_23',['eval',['../classmlx_1_1core_1_1array.html#a2820c45188071a22175e9fa42e10a49a',1,'mlx::core::array::eval()'],['../namespacemlx_1_1core.html#a7d6e097d8effed52f4713672e471f299',1,'mlx::core::eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#adb14f689c9f75f7901edb196c2bfb971',1,'mlx::core::eval(Arrays &&... outputs)']]], - ['eval_5fcpu_24',['eval_cpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#acdc1965ad64ee9ee6328fe150a97902e',1,'mlx::core::distributed::AllReduce::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ab721fe0072fffbddbc3c4334dd033ba5',1,'mlx::core::distributed::AllGather::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a7da6e0cfd630958d9633b2e2bd97a54f',1,'mlx::core::fast::RMSNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#adfc1d52bc266466ab29ee45fd8fab439',1,'mlx::core::fast::RMSNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a5d7a4c1c9ee84e327d1c371733108c05',1,'mlx::core::fast::LayerNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a0d8c4c6e7462befc38f7e08244fa1c2b',1,'mlx::core::fast::LayerNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a05a7d595c6b9dadf7ddfd6e3fd402f0e',1,'mlx::core::fast::RoPE::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ae20851e002f7fcb6d4f97817596f6328',1,'mlx::core::fast::ScaledDotProductAttention::eval_cpu()'],['../classmlx_1_1core_1_1_primitive.html#a1596dc50b910538eae14878e98f07575',1,'mlx::core::Primitive::eval_cpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a7e8f6f5d6ae0a33f6abc0f5a46e0b132',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#aa0ed6e32c36200a3ff9bc592c9b300db',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0d3e697496ef8e842d21195cb3c14e60',1,'mlx::core::Abs::eval_cpu()'],['../classmlx_1_1core_1_1_add.html#a5bacfc51dfa2a5a931bad2dd7bdc7a5f',1,'mlx::core::Add::eval_cpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a15694e3bf2ed5c193237b2b9ca00867c',1,'mlx::core::AddMM::eval_cpu()'],['../classmlx_1_1core_1_1_arange.html#aba44432491cbd599bf72712f5f4267a1',1,'mlx::core::Arange::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a58dcba9e706cb12bab062bb7fa5fa006',1,'mlx::core::ArcCos::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#a0f6d989bcbbc38f15ef17a136879a9c9',1,'mlx::core::ArcCosh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sin.html#ab3542492c14021329788de8f2a9be1e4',1,'mlx::core::ArcSin::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a52574b24d8d16839c58673f51f8ac066',1,'mlx::core::ArcSinh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a1211bc31241227528f04435239ddb9a3',1,'mlx::core::ArcTan::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a13094e6b702769928ca0da468f5ce45c',1,'mlx::core::ArcTan2::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a5af9224e1f1ffec412b0baa0af7e1ecd',1,'mlx::core::ArcTanh::eval_cpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a896f75c5325798ac3f9093f6a4581828',1,'mlx::core::ArgPartition::eval_cpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#ad8d48725623ede1ff654fa13eccf2287',1,'mlx::core::ArgReduce::eval_cpu()'],['../classmlx_1_1core_1_1_arg_sort.html#a022079683774bfeb531b3a002cff16fa',1,'mlx::core::ArgSort::eval_cpu()'],['../classmlx_1_1core_1_1_as_type.html#aa89dbf4d73b00c6a44cffd04d5bb228d',1,'mlx::core::AsType::eval_cpu()'],['../classmlx_1_1core_1_1_as_strided.html#acdd4705e4503ff0b124215c4676b4193',1,'mlx::core::AsStrided::eval_cpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a2194bf585213bda1b2966aa02d2fe283',1,'mlx::core::BitwiseBinary::eval_cpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aa85da478cdc6d4a97be06e5d4abee1f2',1,'mlx::core::BlockMaskedMM::eval_cpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#a62352074a480df0e1f879b0bae425730',1,'mlx::core::GatherMM::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast.html#a53d48d9778e2d4c24a124cd767900780',1,'mlx::core::Broadcast::eval_cpu()'],['../classmlx_1_1core_1_1_ceil.html#a9791801fff3f8b79944e15ac2a45a035',1,'mlx::core::Ceil::eval_cpu()'],['../classmlx_1_1core_1_1_compiled.html#ac45b1d0fedd85feefbff7ce7e168b151',1,'mlx::core::Compiled::eval_cpu()'],['../classmlx_1_1core_1_1_concatenate.html#a609e76bede7fc5581ec84ddcb727a258',1,'mlx::core::Concatenate::eval_cpu()'],['../classmlx_1_1core_1_1_conjugate.html#ae39643e2178f442ffba05139f8609d61',1,'mlx::core::Conjugate::eval_cpu()'],['../classmlx_1_1core_1_1_convolution.html#ac74256068da01730629109fa4fa8432b',1,'mlx::core::Convolution::eval_cpu()'],['../classmlx_1_1core_1_1_copy.html#af4a0ebec423e84ffe8083a5e9ed0d70c',1,'mlx::core::Copy::eval_cpu()'],['../classmlx_1_1core_1_1_cos.html#a061fc446268fe56237ae6b20ccf78152',1,'mlx::core::Cos::eval_cpu()'],['../classmlx_1_1core_1_1_cosh.html#ae8702df7e8f0e20cbeccb2a548961d3d',1,'mlx::core::Cosh::eval_cpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#adba1c40c77a2138df6b5f75483f62184',1,'mlx::core::CustomTransforms::eval_cpu()'],['../classmlx_1_1core_1_1_depends.html#a0c7ea6db97337591fa53c6e6bde41e5e',1,'mlx::core::Depends::eval_cpu()'],['../classmlx_1_1core_1_1_divide.html#a823443c2a8e8b81bbcaeee6ddbcdbf49',1,'mlx::core::Divide::eval_cpu()'],['../classmlx_1_1core_1_1_div_mod.html#ae350b7b93ad128e3133ee14f247193b3',1,'mlx::core::DivMod::eval_cpu()'],['../classmlx_1_1core_1_1_select.html#aa51aa36e0adbd69e0d23d7c7adf88de2',1,'mlx::core::Select::eval_cpu()'],['../classmlx_1_1core_1_1_remainder.html#ac6c6c86a0bf02e6e529eb87f6e617ccc',1,'mlx::core::Remainder::eval_cpu()'],['../classmlx_1_1core_1_1_equal.html#aabb8aa61fa581defddcdca1274b1b454',1,'mlx::core::Equal::eval_cpu()'],['../classmlx_1_1core_1_1_erf.html#a84ea16e43d5b7f83bbc2d5ece78a3fb6',1,'mlx::core::Erf::eval_cpu()'],['../classmlx_1_1core_1_1_erf_inv.html#af579627402af3249565134884701d39e',1,'mlx::core::ErfInv::eval_cpu()'],['../classmlx_1_1core_1_1_exp.html#a47934c5a5023bc7ae7ae89bff45ebb2c',1,'mlx::core::Exp::eval_cpu()'],['../classmlx_1_1core_1_1_expm1.html#ab9c8b7aa50fe4592d55f8957baac647a',1,'mlx::core::Expm1::eval_cpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a6bc262a0c2b5d4fe655e3e2e0ff28635',1,'mlx::core::FFT::eval_cpu()'],['../classmlx_1_1core_1_1_floor.html#a1a7dc5f571b7b73e7ef3cbdc1dd1fcf7',1,'mlx::core::Floor::eval_cpu()'],['../classmlx_1_1core_1_1_full.html#a3dccd3756599d7fd018b2af0093b082c',1,'mlx::core::Full::eval_cpu()'],['../classmlx_1_1core_1_1_gather.html#a9ed5587f0d04b59a2b9186c0aac21290',1,'mlx::core::Gather::eval_cpu()'],['../classmlx_1_1core_1_1_greater.html#abe1c03f311d0e0b610f3392a6566f2ae',1,'mlx::core::Greater::eval_cpu()'],['../classmlx_1_1core_1_1_greater_equal.html#a15469125b9bea89b64bfeac01590c075',1,'mlx::core::GreaterEqual::eval_cpu()'],['../classmlx_1_1core_1_1_hadamard.html#ab27d6a9df42b3aab41ace3073a4c880d',1,'mlx::core::Hadamard::eval_cpu()'],['../classmlx_1_1core_1_1_less.html#a32624124ffece066f496b3299056bcef',1,'mlx::core::Less::eval_cpu()'],['../classmlx_1_1core_1_1_less_equal.html#a55d1352b0e97841a92503bc57c19ed16',1,'mlx::core::LessEqual::eval_cpu()'],['../classmlx_1_1core_1_1_load.html#ada026ac30566f3109d8182e35d307c0a',1,'mlx::core::Load::eval_cpu()'],['../classmlx_1_1core_1_1_log.html#aadc7bb4cb24f3ecbbb9ed54a699ab74f',1,'mlx::core::Log::eval_cpu()'],['../classmlx_1_1core_1_1_log1p.html#a8192e5438de99c4cda056987935cba23',1,'mlx::core::Log1p::eval_cpu()'],['../classmlx_1_1core_1_1_logical_not.html#acf3f7b3b20ca69533536e0e0a05725b3',1,'mlx::core::LogicalNot::eval_cpu()'],['../classmlx_1_1core_1_1_logical_and.html#adbe1c1785af1a8b827289d22b0d170b3',1,'mlx::core::LogicalAnd::eval_cpu()'],['../classmlx_1_1core_1_1_logical_or.html#a13cd4cbf26589287e85aeaaca42d7f62',1,'mlx::core::LogicalOr::eval_cpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#abef17fb590b1a8d356f2a580e45d41f0',1,'mlx::core::LogAddExp::eval_cpu()'],['../classmlx_1_1core_1_1_matmul.html#a357a7f57a2a220a91977f810a69413fc',1,'mlx::core::Matmul::eval_cpu()'],['../classmlx_1_1core_1_1_maximum.html#a62b38fbe5f96db58c2b60165ac4eadcf',1,'mlx::core::Maximum::eval_cpu()'],['../classmlx_1_1core_1_1_minimum.html#a6b93f493ee87089943a8085fe59dfc6e',1,'mlx::core::Minimum::eval_cpu()'],['../classmlx_1_1core_1_1_multiply.html#a624fce06c047cdc4dfdbdcaaddb25f34',1,'mlx::core::Multiply::eval_cpu()'],['../classmlx_1_1core_1_1_negative.html#af43553dc418c8ebe75fa9cdcba103c3b',1,'mlx::core::Negative::eval_cpu()'],['../classmlx_1_1core_1_1_not_equal.html#a8f95f8b5873850b875b1641df8196047',1,'mlx::core::NotEqual::eval_cpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#acc328321cf5300874ee884367cbede3f',1,'mlx::core::NumberOfElements::eval_cpu()'],['../classmlx_1_1core_1_1_pad.html#aaf82dd163cd536fbf97304f8b29080cb',1,'mlx::core::Pad::eval_cpu()'],['../classmlx_1_1core_1_1_partition.html#a784596ab567f9f3cb4fe1a69466523d8',1,'mlx::core::Partition::eval_cpu()'],['../classmlx_1_1core_1_1_power.html#a6783da16fb6ff393aaa57737f1973206',1,'mlx::core::Power::eval_cpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ab3dfa73b74d8f4f2e9ab4f0eb016b0e3',1,'mlx::core::QuantizedMatmul::eval_cpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a89aae98bfbdd6563df44ef7d70f0bf8c',1,'mlx::core::GatherQMM::eval_cpu()'],['../classmlx_1_1core_1_1_random_bits.html#a5752d051cd16cf5f8d4754c0a656f0d2',1,'mlx::core::RandomBits::eval_cpu()'],['../classmlx_1_1core_1_1_reshape.html#a658de2c5f710991b48e14b2bd19b229f',1,'mlx::core::Reshape::eval_cpu()'],['../classmlx_1_1core_1_1_reduce.html#aeb8a58b560c0a09ae3a695df7829acfa',1,'mlx::core::Reduce::eval_cpu()'],['../classmlx_1_1core_1_1_round.html#ad066b0944b437f64ab546025efa00007',1,'mlx::core::Round::eval_cpu()'],['../classmlx_1_1core_1_1_scan.html#a15676d9fd066e935782a923fba3e940b',1,'mlx::core::Scan::eval_cpu()'],['../classmlx_1_1core_1_1_scatter.html#a7623f590f8b77167b5ebb4f14bc9dc97',1,'mlx::core::Scatter::eval_cpu()'],['../classmlx_1_1core_1_1_sigmoid.html#aa930ce05734cca529ebcb8d0ca8e1255',1,'mlx::core::Sigmoid::eval_cpu()'],['../classmlx_1_1core_1_1_sign.html#a7498ec993b66879be30c5d9762c45a97',1,'mlx::core::Sign::eval_cpu()'],['../classmlx_1_1core_1_1_sin.html#ab34f9cebc2aed55a0b6ab4c991f02eb5',1,'mlx::core::Sin::eval_cpu()'],['../classmlx_1_1core_1_1_sinh.html#ab6d5f6f40d177f6435f6a51c71b939dd',1,'mlx::core::Sinh::eval_cpu()'],['../classmlx_1_1core_1_1_slice.html#a4b13503f5b2f5c6a90d394b020f9b3f2',1,'mlx::core::Slice::eval_cpu()'],['../classmlx_1_1core_1_1_slice_update.html#ad82ca0e3ab88a0e086431050deea831b',1,'mlx::core::SliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_softmax.html#ac9ebc2eab1683b682e689ed8f4622b79',1,'mlx::core::Softmax::eval_cpu()'],['../classmlx_1_1core_1_1_sort.html#a459769a0241b2620e55bedaba19827cd',1,'mlx::core::Sort::eval_cpu()'],['../classmlx_1_1core_1_1_split.html#aff2889cb9074f0fda53edf8fa40b1fd4',1,'mlx::core::Split::eval_cpu()'],['../classmlx_1_1core_1_1_square.html#a1f4d327a705950616da63b83c2829e59',1,'mlx::core::Square::eval_cpu()'],['../classmlx_1_1core_1_1_sqrt.html#a5a64ecc4eef1e30a2963435dca7cefd5',1,'mlx::core::Sqrt::eval_cpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a56207714d374b08f60e4d9cdbc7340b2',1,'mlx::core::StopGradient::eval_cpu()'],['../classmlx_1_1core_1_1_subtract.html#a47574258b6c95f8ad260c114d6d36a12',1,'mlx::core::Subtract::eval_cpu()'],['../classmlx_1_1core_1_1_tan.html#a9c9a731158fa60eef30067fe0da9f3e9',1,'mlx::core::Tan::eval_cpu()'],['../classmlx_1_1core_1_1_tanh.html#af7ed4345f622da069e5b0284067923f5',1,'mlx::core::Tanh::eval_cpu()'],['../classmlx_1_1core_1_1_uniform.html#a037a2c96b79b70a64f2b637c9f1a432f',1,'mlx::core::Uniform::eval_cpu()'],['../classmlx_1_1core_1_1_view.html#a0ad6deb11914a242f10e8039fcb02497',1,'mlx::core::View::eval_cpu()'],['../classmlx_1_1core_1_1_transpose.html#a1fbcfcca43f9ec06c63a3c14708c30f8',1,'mlx::core::Transpose::eval_cpu()'],['../classmlx_1_1core_1_1_q_r_f.html#a48493887395d65a27f04de1804d277d2',1,'mlx::core::QRF::eval_cpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a637f5c39fa8b10722c04a066f6c1ada6',1,'mlx::core::SVD::eval_cpu()'],['../classmlx_1_1core_1_1_inverse.html#aeb1d8dc9bc4052a616023f65b3c7bb81',1,'mlx::core::Inverse::eval_cpu()'],['../classmlx_1_1core_1_1_cholesky.html#a4bdec36c1cc99aadf9a4a39d4c57bea5',1,'mlx::core::Cholesky::eval_cpu()']]], - ['eval_5fgpu_25',['eval_gpu',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a611ac8dfd9b4f49617ee8976ff3e15e9',1,'mlx::core::distributed::DistPrimitive::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae7955e8d43c097eecae264e804b4d8ca',1,'mlx::core::fast::RMSNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a48efb8fa84c4ba6cc9fb560ebbe01560',1,'mlx::core::fast::RMSNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a77abda7f47bffa2c037a5d60cccc1528',1,'mlx::core::fast::LayerNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a954a003a4a27c8c4c60a5a14142a9cc3',1,'mlx::core::fast::LayerNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a913b6b00fc518b25ac3947e4e15790f2',1,'mlx::core::fast::RoPE::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a505f38ba93a3499895f5312e0112e73d',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ad51666e69f670e286293aff96eb435a9',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, array &out)'],['../classmlx_1_1core_1_1_primitive.html#ad217376dcf5eff691d731566faec2ba2',1,'mlx::core::Primitive::eval_gpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a6b7f80abaf038d53ec6ffbb0dfac6adb',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#a971fe9ad47f6569118879ce1d0f41447',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0a976e636dd8505b473fbdddf949f514',1,'mlx::core::Abs::eval_gpu()'],['../classmlx_1_1core_1_1_add.html#aa0aacbc1e26b95a2f040f62aa4f69c3d',1,'mlx::core::Add::eval_gpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a5f933be14baebc32a0be0f9a69148aa9',1,'mlx::core::AddMM::eval_gpu()'],['../classmlx_1_1core_1_1_arange.html#a7a2e9787c6c3a78b4a6df91206974031',1,'mlx::core::Arange::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a46f72d4af89b0a0f5f203783fb44589c',1,'mlx::core::ArcCos::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#aa6a2587485a0e015ac2d5211d7d045fc',1,'mlx::core::ArcCosh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sin.html#a7fa4ae7a85bc8bed97ea258ae30762f3',1,'mlx::core::ArcSin::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79f648a86de4c10386a1ce3b5e38e8ac',1,'mlx::core::ArcSinh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a77866feb27028865d844070447c9a254',1,'mlx::core::ArcTan::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a76d3f0c29e0ff4642b8d39dac90d3f50',1,'mlx::core::ArcTan2::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a10566b9d3b2c7d090895b46d9040bc1d',1,'mlx::core::ArcTanh::eval_gpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a9a60995eaf85f63c877e86b23cbc15fc',1,'mlx::core::ArgPartition::eval_gpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#aafa982ce2abc0cd9e81e43aa2c823d29',1,'mlx::core::ArgReduce::eval_gpu()'],['../classmlx_1_1core_1_1_arg_sort.html#abc2d730850ec4ee8d7968b7417911709',1,'mlx::core::ArgSort::eval_gpu()'],['../classmlx_1_1core_1_1_as_type.html#a5b111b9d74c60d27b4a7ebaa49f96e0b',1,'mlx::core::AsType::eval_gpu()'],['../classmlx_1_1core_1_1_as_strided.html#ab6771a208323994927ca162ba7bb10ed',1,'mlx::core::AsStrided::eval_gpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#ac831a29fc46701b00bbe63ee33832afd',1,'mlx::core::BitwiseBinary::eval_gpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ab372b6df4de00a33795a052a23bb1df9',1,'mlx::core::BlockMaskedMM::eval_gpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#ad754c35f460a055cc383ad93a5f72da1',1,'mlx::core::GatherMM::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast.html#ab9bd9dbcedcefc9b29c84911b5ce69fe',1,'mlx::core::Broadcast::eval_gpu()'],['../classmlx_1_1core_1_1_ceil.html#abe178e0058e44b6618be414215e96887',1,'mlx::core::Ceil::eval_gpu()'],['../classmlx_1_1core_1_1_compiled.html#aa3d5ff0f2b3554ad48fbbf2a0f3336d5',1,'mlx::core::Compiled::eval_gpu()'],['../classmlx_1_1core_1_1_concatenate.html#a309a1c50e97f9925866433ee2841c474',1,'mlx::core::Concatenate::eval_gpu()'],['../classmlx_1_1core_1_1_conjugate.html#aff0a802166e3724db88ab5d3feb2d3de',1,'mlx::core::Conjugate::eval_gpu()'],['../classmlx_1_1core_1_1_convolution.html#a30b64109eeb1778f002b99447dff9dd2',1,'mlx::core::Convolution::eval_gpu()'],['../classmlx_1_1core_1_1_copy.html#a1eda7b2ea771a168f67421f0d384b3a1',1,'mlx::core::Copy::eval_gpu()'],['../classmlx_1_1core_1_1_cos.html#a5ef41aafad595f6cdd8c535e36e12060',1,'mlx::core::Cos::eval_gpu()'],['../classmlx_1_1core_1_1_cosh.html#a23f71b43792934c3ec0ebe9b74f32559',1,'mlx::core::Cosh::eval_gpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#a7b3538681acbb20af3ed37b0877f6667',1,'mlx::core::CustomTransforms::eval_gpu()'],['../classmlx_1_1core_1_1_depends.html#ae5057f65e69490ad0add8eeda2b75e28',1,'mlx::core::Depends::eval_gpu()'],['../classmlx_1_1core_1_1_divide.html#abffda0ce37221ddc28dc9eea794f6bc7',1,'mlx::core::Divide::eval_gpu()'],['../classmlx_1_1core_1_1_div_mod.html#a003117c9ecf3c06a27248f72a76348dc',1,'mlx::core::DivMod::eval_gpu()'],['../classmlx_1_1core_1_1_select.html#a2a82b6cba4c386b2b87f225a4b08ea9b',1,'mlx::core::Select::eval_gpu()'],['../classmlx_1_1core_1_1_remainder.html#a7919ea9b84e42522d51bf0d5a396e161',1,'mlx::core::Remainder::eval_gpu()'],['../classmlx_1_1core_1_1_equal.html#ac3757001fec42ceb5ece2954df42161c',1,'mlx::core::Equal::eval_gpu()'],['../classmlx_1_1core_1_1_erf.html#ad8551be664d767dccc3c0d8cc1eca008',1,'mlx::core::Erf::eval_gpu()'],['../classmlx_1_1core_1_1_erf_inv.html#a4a2413d0634db1f3dae1806ddfa632db',1,'mlx::core::ErfInv::eval_gpu()'],['../classmlx_1_1core_1_1_exp.html#a7d63695a97a14760fd33b5d4e6590822',1,'mlx::core::Exp::eval_gpu()'],['../classmlx_1_1core_1_1_expm1.html#a82930071f4b77d883b300f77966aff5f',1,'mlx::core::Expm1::eval_gpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a1c21b26d1e9ad7c4da78ae845721b2dd',1,'mlx::core::FFT::eval_gpu()'],['../classmlx_1_1core_1_1_floor.html#aaa29c83538099eb8f951c95a41f2eb65',1,'mlx::core::Floor::eval_gpu()'],['../classmlx_1_1core_1_1_full.html#aa54f99bb4cba12a551392dea56003872',1,'mlx::core::Full::eval_gpu()'],['../classmlx_1_1core_1_1_gather.html#aec48ee529cb2449915a7b27a3c4361e8',1,'mlx::core::Gather::eval_gpu()'],['../classmlx_1_1core_1_1_greater.html#ae8957cccf4c924d941f57a1bb751c878',1,'mlx::core::Greater::eval_gpu()'],['../classmlx_1_1core_1_1_greater_equal.html#ac246263b4548126c3d4ab7e392575d24',1,'mlx::core::GreaterEqual::eval_gpu()'],['../classmlx_1_1core_1_1_hadamard.html#a2470feb690f5463138490763c38b5733',1,'mlx::core::Hadamard::eval_gpu()'],['../classmlx_1_1core_1_1_less.html#a353335ce06ddbe8498d86d129c835917',1,'mlx::core::Less::eval_gpu()'],['../classmlx_1_1core_1_1_less_equal.html#acf035a82b11e6f63742143ea540fedac',1,'mlx::core::LessEqual::eval_gpu()'],['../classmlx_1_1core_1_1_load.html#a06933e887ea94a4d01d81195c5e07a3d',1,'mlx::core::Load::eval_gpu()'],['../classmlx_1_1core_1_1_log.html#aaaa49e9455f3a197bc319646b5ca6390',1,'mlx::core::Log::eval_gpu()'],['../classmlx_1_1core_1_1_log1p.html#a1b97decae7338d46874e736c95fa7431',1,'mlx::core::Log1p::eval_gpu()'],['../classmlx_1_1core_1_1_logical_not.html#a1d0d2bc93f935eca6c85ef7bf67f2d6a',1,'mlx::core::LogicalNot::eval_gpu()'],['../classmlx_1_1core_1_1_logical_and.html#a132b2eedaa3978de5a5350da3c2ca40f',1,'mlx::core::LogicalAnd::eval_gpu()'],['../classmlx_1_1core_1_1_logical_or.html#a3be1da328f0f8620de2e4fc1d22a077a',1,'mlx::core::LogicalOr::eval_gpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#acace355b62ec00df649f9f99e8f2eb7a',1,'mlx::core::LogAddExp::eval_gpu()'],['../classmlx_1_1core_1_1_matmul.html#a8707a4e9b75c769e8f1dbca15c6a1ae7',1,'mlx::core::Matmul::eval_gpu()'],['../classmlx_1_1core_1_1_maximum.html#ade0f721b10a6b3a12bdadd34c48f72a7',1,'mlx::core::Maximum::eval_gpu()'],['../classmlx_1_1core_1_1_minimum.html#aadc68afa0afbe2103f19d161f5e0a2ba',1,'mlx::core::Minimum::eval_gpu()'],['../classmlx_1_1core_1_1_multiply.html#a634fcb4e981d8d3f4d94252caf25bee0',1,'mlx::core::Multiply::eval_gpu()'],['../classmlx_1_1core_1_1_negative.html#a97f1b316eace0c6d9e576d766940c75b',1,'mlx::core::Negative::eval_gpu()'],['../classmlx_1_1core_1_1_not_equal.html#a61179747e34e203150e9c660dfddb5f2',1,'mlx::core::NotEqual::eval_gpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#a2c98c42915fb2bfe12f5c99ea553eff5',1,'mlx::core::NumberOfElements::eval_gpu()'],['../classmlx_1_1core_1_1_pad.html#aefd4d3a5bd8b6b35b266c9e558ada153',1,'mlx::core::Pad::eval_gpu()'],['../classmlx_1_1core_1_1_partition.html#a8eca1be21ae9ccfda46e6f3e85f506ef',1,'mlx::core::Partition::eval_gpu()'],['../classmlx_1_1core_1_1_power.html#a80577d4c0853c24027777c90a1ec7e11',1,'mlx::core::Power::eval_gpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a2812ad007d695ed1aaf9cf706fb9c4b3',1,'mlx::core::QuantizedMatmul::eval_gpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a86eb048afc95646b2e96ec5493e3d887',1,'mlx::core::GatherQMM::eval_gpu()'],['../classmlx_1_1core_1_1_random_bits.html#a578756866665358577418e4cdd94aa3a',1,'mlx::core::RandomBits::eval_gpu()'],['../classmlx_1_1core_1_1_reshape.html#aa1e85f28471875750c47351520b56059',1,'mlx::core::Reshape::eval_gpu()'],['../classmlx_1_1core_1_1_reduce.html#ae9caaf42edadfe73ea208d98f526890f',1,'mlx::core::Reduce::eval_gpu()'],['../classmlx_1_1core_1_1_round.html#af7fe5ff8f3db166c203b4be4b07f13ec',1,'mlx::core::Round::eval_gpu()'],['../classmlx_1_1core_1_1_scan.html#aef22c6fc2b2cb2a907cd8965c7413dde',1,'mlx::core::Scan::eval_gpu()'],['../classmlx_1_1core_1_1_scatter.html#ab304345db3d8cfeea15e27461ae2e678',1,'mlx::core::Scatter::eval_gpu()'],['../classmlx_1_1core_1_1_sigmoid.html#a7a6bd0222d51d7f25f2719a91ccdfeca',1,'mlx::core::Sigmoid::eval_gpu()'],['../classmlx_1_1core_1_1_sign.html#afa2b48b99a194106006b44af69ffda8b',1,'mlx::core::Sign::eval_gpu()'],['../classmlx_1_1core_1_1_sin.html#a6b59f1156cf8bdad8d45acd1d825cb5e',1,'mlx::core::Sin::eval_gpu()'],['../classmlx_1_1core_1_1_sinh.html#a5a1af2399f166d5b228b5e83a1837c75',1,'mlx::core::Sinh::eval_gpu()'],['../classmlx_1_1core_1_1_slice.html#aa53c21ff06a7c659e889af6b97d10a4a',1,'mlx::core::Slice::eval_gpu()'],['../classmlx_1_1core_1_1_slice_update.html#aac1a1d122e5697be057d63552141032b',1,'mlx::core::SliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_softmax.html#a35dac69ddcc7e2ec0e1a76fe93db85af',1,'mlx::core::Softmax::eval_gpu()'],['../classmlx_1_1core_1_1_sort.html#a4141c48f0e8670c728663f3722675382',1,'mlx::core::Sort::eval_gpu()'],['../classmlx_1_1core_1_1_split.html#a78ddda89c4daee73c74cfbc1e44656df',1,'mlx::core::Split::eval_gpu()'],['../classmlx_1_1core_1_1_square.html#a0ea2a78a5bb52daa4103263bf2f98045',1,'mlx::core::Square::eval_gpu()'],['../classmlx_1_1core_1_1_sqrt.html#a6d205e679a593d1ba20206c5c47ba501',1,'mlx::core::Sqrt::eval_gpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a907b96f0a1ce608e211d87ccf2b9ca89',1,'mlx::core::StopGradient::eval_gpu()'],['../classmlx_1_1core_1_1_subtract.html#a69021b23daf061764d97fabbc0f4f06c',1,'mlx::core::Subtract::eval_gpu()'],['../classmlx_1_1core_1_1_tan.html#aca7dbb4836507005a2032ac957a04d3f',1,'mlx::core::Tan::eval_gpu()'],['../classmlx_1_1core_1_1_tanh.html#a48df896599ae93dbce84a5c0f50cf761',1,'mlx::core::Tanh::eval_gpu()'],['../classmlx_1_1core_1_1_uniform.html#a5f88cbf2495f24f87cefd99aaaebe4d0',1,'mlx::core::Uniform::eval_gpu()'],['../classmlx_1_1core_1_1_view.html#add6e12ff1e476fe1db7718b14f21b075',1,'mlx::core::View::eval_gpu()'],['../classmlx_1_1core_1_1_transpose.html#a38d25739c08aa594a6775015a1d7d92e',1,'mlx::core::Transpose::eval_gpu()'],['../classmlx_1_1core_1_1_q_r_f.html#ae5fa3482192f4713605cd07e7fc1c6c9',1,'mlx::core::QRF::eval_gpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a7067b2207f826a25549d571856b94e83',1,'mlx::core::SVD::eval_gpu()'],['../classmlx_1_1core_1_1_inverse.html#a086fbbc947ad232e01686ad063a78ed2',1,'mlx::core::Inverse::eval_gpu()'],['../classmlx_1_1core_1_1_cholesky.html#a8c918594bf129888044ef37fcae56795',1,'mlx::core::Cholesky::eval_gpu()']]], - ['event_26',['Event',['../classmlx_1_1core_1_1_event.html#a833506419b2110ad1abd89b2dd238b4d',1,'mlx::core::Event::Event()=default'],['../classmlx_1_1core_1_1_event.html#a13e4835f2ffb2cc22e29148a448ea184',1,'mlx::core::Event::Event(const Stream &steam)']]], - ['event_27',['event',['../classmlx_1_1core_1_1array.html#a0a8e4d6e67e739a712876bb36f88f9bf',1,'mlx::core::array']]], - ['exec_28',['exec',['../classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2',1,'pocketfft::detail::cfftp::exec()'],['../classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac',1,'pocketfft::detail::rfftp::exec()'],['../classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c',1,'pocketfft::detail::fftblue::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d',1,'pocketfft::detail::pocketfft_c::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33',1,'pocketfft::detail::pocketfft_r::exec()'],['../classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058',1,'pocketfft::detail::T_dct1::exec()'],['../classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26',1,'pocketfft::detail::T_dst1::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac',1,'pocketfft::detail::T_dcst23::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708',1,'pocketfft::detail::T_dcst4::exec()']]], - ['exec_5fr_29',['exec_r',['../classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00',1,'pocketfft::detail::fftblue']]], - ['exp_30',['Exp',['../classmlx_1_1core_1_1_exp.html#a1d0a618cbb91ab29ef53b57ff6ed6e06',1,'mlx::core::Exp']]], - ['exp_31',['exp',['../namespacemetal.html#ac2a0b3618d922ac014baac8189d44650',1,'metal::exp()'],['../namespacemetal_1_1fast.html#ad3dbd387b63373c29e3449609f763ede',1,'metal::fast::exp()'],['../namespacemetal_1_1precise.html#a8d8d2d5700ce432b33cf47cf22528e8f',1,'metal::precise::exp()'],['../group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125',1,'mlx::core::exp()']]], - ['exp10_32',['exp10',['../namespacemetal.html#a4c63707d13c89364496a48906631c204',1,'metal::exp10()'],['../namespacemetal_1_1fast.html#a453122f982485cbb4e471b3ac282ee5e',1,'metal::fast::exp10()'],['../namespacemetal_1_1precise.html#af9addb343c967da3a83e9e123a8521fd',1,'metal::precise::exp10()']]], - ['exp2_33',['exp2',['../namespacemetal.html#a228201c20777848804a4d0589c1d33e7',1,'metal::exp2()'],['../namespacemetal_1_1fast.html#ac092b65a46720adaf22f6266671d2d71',1,'metal::fast::exp2()'],['../namespacemetal_1_1precise.html#a92a880bd2197efc0da0f8f0f7ec1e4c9',1,'metal::precise::exp2()']]], - ['expand_5fdims_34',['expand_dims',['../group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133',1,'mlx::core::expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a',1,'mlx::core::expand_dims(const array &a, int axis, StreamOrDevice s={})']]], - ['expm1_35',['Expm1',['../classmlx_1_1core_1_1_expm1.html#a47c2a1b2a4ef6bb07ba77c55ddddaec2',1,'mlx::core::Expm1']]], - ['expm1_36',['expm1',['../group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0',1,'mlx::core']]], - ['expm1f_37',['expm1f',['../expm1f_8h.html#a87f66d30e185950f42ce3641783cdc40',1,'expm1f.h']]], - ['expm1f_5fscaled_5funchecked_38',['expm1f_scaled_unchecked',['../expm1f_8h.html#adf20e03405fba634ca8d01acac24592e',1,'expm1f.h']]], - ['export_5fto_5fdot_39',['export_to_dot',['../namespacemlx_1_1core.html#a57395bdf43d9c5c134e610c169222cca',1,'mlx::core::export_to_dot(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a839f94dbad44f0d37333006fc876b42e',1,'mlx::core::export_to_dot(std::ostream &os, Arrays &&... outputs)']]], - ['eye_40',['eye',['../group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3',1,'mlx::core::eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd',1,'mlx::core::eye(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga61657db78ef35d41112d362c869c25d2',1,'mlx::core::eye(int n, int m, StreamOrDevice s={})'],['../group__ops.html#ga908a15b42834be498a46856c99dfc779',1,'mlx::core::eye(int n, int m, int k, StreamOrDevice s={})'],['../group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f',1,'mlx::core::eye(int n, StreamOrDevice s={})']]] + ['end_12',['end',['../classmlx_1_1core_1_1array.html#a5daf64552fb450825c9b382f3a5fa2d4',1,'mlx::core::array']]], + ['end_5fencoding_13',['end_encoding',['../classmlx_1_1core_1_1metal_1_1_device.html#a60689f97347811b27e8c5ca23e0372bf',1,'mlx::core::metal::Device']]], + ['enqueue_14',['enqueue',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a4918720319cf224a1b4208568964c286',1,'mlx::core::scheduler::StreamThread::enqueue()'],['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a03809c783bd1866362dc7cb9118abbcc',1,'mlx::core::scheduler::Scheduler::enqueue()'],['../namespacemlx_1_1core_1_1scheduler.html#aa2d4eacf5d5cbc778a51aafd4fd8e4d7',1,'mlx::core::scheduler::enqueue()']]], + ['epsilon_15',['epsilon',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a96c4197e3076f0aa9065370b8ece49ca',1,'metal::_numeric_limits_impl< bfloat16_t >']]], + ['equal_16',['Equal',['../classmlx_1_1core_1_1_equal.html#a4af81cf2dd071db5bbf8ce1df95fdf36',1,'mlx::core::Equal']]], + ['equal_17',['equal',['../group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde',1,'mlx::core']]], + ['erf_18',['Erf',['../classmlx_1_1core_1_1_erf.html#a702f76f848928d8d7d3d0881ac6e4c82',1,'mlx::core::Erf']]], + ['erf_19',['erf',['../erf_8h.html#a6ce199ee56105c67adbf8c48c019a8b2',1,'erf(): erf.h'],['../group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0',1,'mlx::core::erf()']]], + ['erfinv_20',['ErfInv',['../classmlx_1_1core_1_1_erf_inv.html#a5d0279247b67da4592311559f04e1478',1,'mlx::core::ErfInv']]], + ['erfinv_21',['erfinv',['../erf_8h.html#a1846e0d683c7aff826bb32addcc3b885',1,'erfinv(): erf.h'],['../group__ops.html#ga76fb9062c64264e34d2e07013390557c',1,'mlx::core::erfinv()']]], + ['eval_22',['eval',['../classmlx_1_1core_1_1array.html#a2820c45188071a22175e9fa42e10a49a',1,'mlx::core::array::eval()'],['../namespacemlx_1_1core.html#a7d6e097d8effed52f4713672e471f299',1,'mlx::core::eval(std::vector< array > outputs)'],['../namespacemlx_1_1core.html#adb14f689c9f75f7901edb196c2bfb971',1,'mlx::core::eval(Arrays &&... outputs)']]], + ['eval_5fcpu_23',['eval_cpu',['../classmlx_1_1core_1_1distributed_1_1_all_reduce.html#acdc1965ad64ee9ee6328fe150a97902e',1,'mlx::core::distributed::AllReduce::eval_cpu()'],['../classmlx_1_1core_1_1distributed_1_1_all_gather.html#ab721fe0072fffbddbc3c4334dd033ba5',1,'mlx::core::distributed::AllGather::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#a7da6e0cfd630958d9633b2e2bd97a54f',1,'mlx::core::fast::RMSNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#adfc1d52bc266466ab29ee45fd8fab439',1,'mlx::core::fast::RMSNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a5d7a4c1c9ee84e327d1c371733108c05',1,'mlx::core::fast::LayerNorm::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a0d8c4c6e7462befc38f7e08244fa1c2b',1,'mlx::core::fast::LayerNormVJP::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a05a7d595c6b9dadf7ddfd6e3fd402f0e',1,'mlx::core::fast::RoPE::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ae20851e002f7fcb6d4f97817596f6328',1,'mlx::core::fast::ScaledDotProductAttention::eval_cpu()'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a3b5d628628d245b38911118d4a0ff9fd',1,'mlx::core::fast::AffineQuantize::eval_cpu()'],['../classmlx_1_1core_1_1_primitive.html#a1596dc50b910538eae14878e98f07575',1,'mlx::core::Primitive::eval_cpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a7e8f6f5d6ae0a33f6abc0f5a46e0b132',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#aa0ed6e32c36200a3ff9bc592c9b300db',1,'mlx::core::UnaryPrimitive::eval_cpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0d3e697496ef8e842d21195cb3c14e60',1,'mlx::core::Abs::eval_cpu()'],['../classmlx_1_1core_1_1_add.html#a5bacfc51dfa2a5a931bad2dd7bdc7a5f',1,'mlx::core::Add::eval_cpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a15694e3bf2ed5c193237b2b9ca00867c',1,'mlx::core::AddMM::eval_cpu()'],['../classmlx_1_1core_1_1_arange.html#aba44432491cbd599bf72712f5f4267a1',1,'mlx::core::Arange::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a58dcba9e706cb12bab062bb7fa5fa006',1,'mlx::core::ArcCos::eval_cpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#a0f6d989bcbbc38f15ef17a136879a9c9',1,'mlx::core::ArcCosh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sin.html#ab3542492c14021329788de8f2a9be1e4',1,'mlx::core::ArcSin::eval_cpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a52574b24d8d16839c58673f51f8ac066',1,'mlx::core::ArcSinh::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a1211bc31241227528f04435239ddb9a3',1,'mlx::core::ArcTan::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a13094e6b702769928ca0da468f5ce45c',1,'mlx::core::ArcTan2::eval_cpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a5af9224e1f1ffec412b0baa0af7e1ecd',1,'mlx::core::ArcTanh::eval_cpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a896f75c5325798ac3f9093f6a4581828',1,'mlx::core::ArgPartition::eval_cpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#ad8d48725623ede1ff654fa13eccf2287',1,'mlx::core::ArgReduce::eval_cpu()'],['../classmlx_1_1core_1_1_arg_sort.html#a022079683774bfeb531b3a002cff16fa',1,'mlx::core::ArgSort::eval_cpu()'],['../classmlx_1_1core_1_1_as_type.html#aa89dbf4d73b00c6a44cffd04d5bb228d',1,'mlx::core::AsType::eval_cpu()'],['../classmlx_1_1core_1_1_as_strided.html#acdd4705e4503ff0b124215c4676b4193',1,'mlx::core::AsStrided::eval_cpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a2194bf585213bda1b2966aa02d2fe283',1,'mlx::core::BitwiseBinary::eval_cpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aa85da478cdc6d4a97be06e5d4abee1f2',1,'mlx::core::BlockMaskedMM::eval_cpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#a62352074a480df0e1f879b0bae425730',1,'mlx::core::GatherMM::eval_cpu()'],['../classmlx_1_1core_1_1_broadcast.html#a53d48d9778e2d4c24a124cd767900780',1,'mlx::core::Broadcast::eval_cpu()'],['../classmlx_1_1core_1_1_ceil.html#a9791801fff3f8b79944e15ac2a45a035',1,'mlx::core::Ceil::eval_cpu()'],['../classmlx_1_1core_1_1_compiled.html#ac45b1d0fedd85feefbff7ce7e168b151',1,'mlx::core::Compiled::eval_cpu()'],['../classmlx_1_1core_1_1_concatenate.html#a609e76bede7fc5581ec84ddcb727a258',1,'mlx::core::Concatenate::eval_cpu()'],['../classmlx_1_1core_1_1_conjugate.html#ae39643e2178f442ffba05139f8609d61',1,'mlx::core::Conjugate::eval_cpu()'],['../classmlx_1_1core_1_1_convolution.html#ac74256068da01730629109fa4fa8432b',1,'mlx::core::Convolution::eval_cpu()'],['../classmlx_1_1core_1_1_copy.html#af4a0ebec423e84ffe8083a5e9ed0d70c',1,'mlx::core::Copy::eval_cpu()'],['../classmlx_1_1core_1_1_cos.html#a061fc446268fe56237ae6b20ccf78152',1,'mlx::core::Cos::eval_cpu()'],['../classmlx_1_1core_1_1_cosh.html#ae8702df7e8f0e20cbeccb2a548961d3d',1,'mlx::core::Cosh::eval_cpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#adba1c40c77a2138df6b5f75483f62184',1,'mlx::core::CustomTransforms::eval_cpu()'],['../classmlx_1_1core_1_1_depends.html#a0c7ea6db97337591fa53c6e6bde41e5e',1,'mlx::core::Depends::eval_cpu()'],['../classmlx_1_1core_1_1_divide.html#a823443c2a8e8b81bbcaeee6ddbcdbf49',1,'mlx::core::Divide::eval_cpu()'],['../classmlx_1_1core_1_1_div_mod.html#ae350b7b93ad128e3133ee14f247193b3',1,'mlx::core::DivMod::eval_cpu()'],['../classmlx_1_1core_1_1_select.html#aa51aa36e0adbd69e0d23d7c7adf88de2',1,'mlx::core::Select::eval_cpu()'],['../classmlx_1_1core_1_1_remainder.html#ac6c6c86a0bf02e6e529eb87f6e617ccc',1,'mlx::core::Remainder::eval_cpu()'],['../classmlx_1_1core_1_1_equal.html#aabb8aa61fa581defddcdca1274b1b454',1,'mlx::core::Equal::eval_cpu()'],['../classmlx_1_1core_1_1_erf.html#a84ea16e43d5b7f83bbc2d5ece78a3fb6',1,'mlx::core::Erf::eval_cpu()'],['../classmlx_1_1core_1_1_erf_inv.html#af579627402af3249565134884701d39e',1,'mlx::core::ErfInv::eval_cpu()'],['../classmlx_1_1core_1_1_exp.html#a47934c5a5023bc7ae7ae89bff45ebb2c',1,'mlx::core::Exp::eval_cpu()'],['../classmlx_1_1core_1_1_expm1.html#ab9c8b7aa50fe4592d55f8957baac647a',1,'mlx::core::Expm1::eval_cpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a6bc262a0c2b5d4fe655e3e2e0ff28635',1,'mlx::core::FFT::eval_cpu()'],['../classmlx_1_1core_1_1_floor.html#a1a7dc5f571b7b73e7ef3cbdc1dd1fcf7',1,'mlx::core::Floor::eval_cpu()'],['../classmlx_1_1core_1_1_full.html#a3dccd3756599d7fd018b2af0093b082c',1,'mlx::core::Full::eval_cpu()'],['../classmlx_1_1core_1_1_gather.html#a9ed5587f0d04b59a2b9186c0aac21290',1,'mlx::core::Gather::eval_cpu()'],['../classmlx_1_1core_1_1_greater.html#abe1c03f311d0e0b610f3392a6566f2ae',1,'mlx::core::Greater::eval_cpu()'],['../classmlx_1_1core_1_1_greater_equal.html#a15469125b9bea89b64bfeac01590c075',1,'mlx::core::GreaterEqual::eval_cpu()'],['../classmlx_1_1core_1_1_hadamard.html#ab27d6a9df42b3aab41ace3073a4c880d',1,'mlx::core::Hadamard::eval_cpu()'],['../classmlx_1_1core_1_1_less.html#a32624124ffece066f496b3299056bcef',1,'mlx::core::Less::eval_cpu()'],['../classmlx_1_1core_1_1_less_equal.html#a55d1352b0e97841a92503bc57c19ed16',1,'mlx::core::LessEqual::eval_cpu()'],['../classmlx_1_1core_1_1_load.html#ada026ac30566f3109d8182e35d307c0a',1,'mlx::core::Load::eval_cpu()'],['../classmlx_1_1core_1_1_log.html#aadc7bb4cb24f3ecbbb9ed54a699ab74f',1,'mlx::core::Log::eval_cpu()'],['../classmlx_1_1core_1_1_log1p.html#a8192e5438de99c4cda056987935cba23',1,'mlx::core::Log1p::eval_cpu()'],['../classmlx_1_1core_1_1_logical_not.html#acf3f7b3b20ca69533536e0e0a05725b3',1,'mlx::core::LogicalNot::eval_cpu()'],['../classmlx_1_1core_1_1_logical_and.html#adbe1c1785af1a8b827289d22b0d170b3',1,'mlx::core::LogicalAnd::eval_cpu()'],['../classmlx_1_1core_1_1_logical_or.html#a13cd4cbf26589287e85aeaaca42d7f62',1,'mlx::core::LogicalOr::eval_cpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#abef17fb590b1a8d356f2a580e45d41f0',1,'mlx::core::LogAddExp::eval_cpu()'],['../classmlx_1_1core_1_1_matmul.html#a357a7f57a2a220a91977f810a69413fc',1,'mlx::core::Matmul::eval_cpu()'],['../classmlx_1_1core_1_1_maximum.html#a62b38fbe5f96db58c2b60165ac4eadcf',1,'mlx::core::Maximum::eval_cpu()'],['../classmlx_1_1core_1_1_minimum.html#a6b93f493ee87089943a8085fe59dfc6e',1,'mlx::core::Minimum::eval_cpu()'],['../classmlx_1_1core_1_1_multiply.html#a624fce06c047cdc4dfdbdcaaddb25f34',1,'mlx::core::Multiply::eval_cpu()'],['../classmlx_1_1core_1_1_negative.html#af43553dc418c8ebe75fa9cdcba103c3b',1,'mlx::core::Negative::eval_cpu()'],['../classmlx_1_1core_1_1_not_equal.html#a8f95f8b5873850b875b1641df8196047',1,'mlx::core::NotEqual::eval_cpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#acc328321cf5300874ee884367cbede3f',1,'mlx::core::NumberOfElements::eval_cpu()'],['../classmlx_1_1core_1_1_pad.html#aaf82dd163cd536fbf97304f8b29080cb',1,'mlx::core::Pad::eval_cpu()'],['../classmlx_1_1core_1_1_partition.html#a784596ab567f9f3cb4fe1a69466523d8',1,'mlx::core::Partition::eval_cpu()'],['../classmlx_1_1core_1_1_power.html#a6783da16fb6ff393aaa57737f1973206',1,'mlx::core::Power::eval_cpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#ab3dfa73b74d8f4f2e9ab4f0eb016b0e3',1,'mlx::core::QuantizedMatmul::eval_cpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a89aae98bfbdd6563df44ef7d70f0bf8c',1,'mlx::core::GatherQMM::eval_cpu()'],['../classmlx_1_1core_1_1_random_bits.html#a5752d051cd16cf5f8d4754c0a656f0d2',1,'mlx::core::RandomBits::eval_cpu()'],['../classmlx_1_1core_1_1_reshape.html#a658de2c5f710991b48e14b2bd19b229f',1,'mlx::core::Reshape::eval_cpu()'],['../classmlx_1_1core_1_1_reduce.html#aeb8a58b560c0a09ae3a695df7829acfa',1,'mlx::core::Reduce::eval_cpu()'],['../classmlx_1_1core_1_1_round.html#ad066b0944b437f64ab546025efa00007',1,'mlx::core::Round::eval_cpu()'],['../classmlx_1_1core_1_1_scan.html#a15676d9fd066e935782a923fba3e940b',1,'mlx::core::Scan::eval_cpu()'],['../classmlx_1_1core_1_1_scatter.html#a7623f590f8b77167b5ebb4f14bc9dc97',1,'mlx::core::Scatter::eval_cpu()'],['../classmlx_1_1core_1_1_sigmoid.html#aa930ce05734cca529ebcb8d0ca8e1255',1,'mlx::core::Sigmoid::eval_cpu()'],['../classmlx_1_1core_1_1_sign.html#a7498ec993b66879be30c5d9762c45a97',1,'mlx::core::Sign::eval_cpu()'],['../classmlx_1_1core_1_1_sin.html#ab34f9cebc2aed55a0b6ab4c991f02eb5',1,'mlx::core::Sin::eval_cpu()'],['../classmlx_1_1core_1_1_sinh.html#ab6d5f6f40d177f6435f6a51c71b939dd',1,'mlx::core::Sinh::eval_cpu()'],['../classmlx_1_1core_1_1_slice.html#a4b13503f5b2f5c6a90d394b020f9b3f2',1,'mlx::core::Slice::eval_cpu()'],['../classmlx_1_1core_1_1_slice_update.html#ad82ca0e3ab88a0e086431050deea831b',1,'mlx::core::SliceUpdate::eval_cpu()'],['../classmlx_1_1core_1_1_softmax.html#ac9ebc2eab1683b682e689ed8f4622b79',1,'mlx::core::Softmax::eval_cpu()'],['../classmlx_1_1core_1_1_sort.html#a459769a0241b2620e55bedaba19827cd',1,'mlx::core::Sort::eval_cpu()'],['../classmlx_1_1core_1_1_split.html#aff2889cb9074f0fda53edf8fa40b1fd4',1,'mlx::core::Split::eval_cpu()'],['../classmlx_1_1core_1_1_square.html#a1f4d327a705950616da63b83c2829e59',1,'mlx::core::Square::eval_cpu()'],['../classmlx_1_1core_1_1_sqrt.html#a5a64ecc4eef1e30a2963435dca7cefd5',1,'mlx::core::Sqrt::eval_cpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a56207714d374b08f60e4d9cdbc7340b2',1,'mlx::core::StopGradient::eval_cpu()'],['../classmlx_1_1core_1_1_subtract.html#a47574258b6c95f8ad260c114d6d36a12',1,'mlx::core::Subtract::eval_cpu()'],['../classmlx_1_1core_1_1_tan.html#a9c9a731158fa60eef30067fe0da9f3e9',1,'mlx::core::Tan::eval_cpu()'],['../classmlx_1_1core_1_1_tanh.html#af7ed4345f622da069e5b0284067923f5',1,'mlx::core::Tanh::eval_cpu()'],['../classmlx_1_1core_1_1_uniform.html#a037a2c96b79b70a64f2b637c9f1a432f',1,'mlx::core::Uniform::eval_cpu()'],['../classmlx_1_1core_1_1_view.html#a0ad6deb11914a242f10e8039fcb02497',1,'mlx::core::View::eval_cpu()'],['../classmlx_1_1core_1_1_transpose.html#a1fbcfcca43f9ec06c63a3c14708c30f8',1,'mlx::core::Transpose::eval_cpu()'],['../classmlx_1_1core_1_1_q_r_f.html#a48493887395d65a27f04de1804d277d2',1,'mlx::core::QRF::eval_cpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a637f5c39fa8b10722c04a066f6c1ada6',1,'mlx::core::SVD::eval_cpu()'],['../classmlx_1_1core_1_1_inverse.html#aeb1d8dc9bc4052a616023f65b3c7bb81',1,'mlx::core::Inverse::eval_cpu()'],['../classmlx_1_1core_1_1_cholesky.html#a4bdec36c1cc99aadf9a4a39d4c57bea5',1,'mlx::core::Cholesky::eval_cpu()']]], + ['eval_5fgpu_24',['eval_gpu',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a611ac8dfd9b4f49617ee8976ff3e15e9',1,'mlx::core::distributed::DistPrimitive::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm.html#ae7955e8d43c097eecae264e804b4d8ca',1,'mlx::core::fast::RMSNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_r_m_s_norm_v_j_p.html#a48efb8fa84c4ba6cc9fb560ebbe01560',1,'mlx::core::fast::RMSNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm.html#a77abda7f47bffa2c037a5d60cccc1528',1,'mlx::core::fast::LayerNorm::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_layer_norm_v_j_p.html#a954a003a4a27c8c4c60a5a14142a9cc3',1,'mlx::core::fast::LayerNormVJP::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_ro_p_e.html#a913b6b00fc518b25ac3947e4e15790f2',1,'mlx::core::fast::RoPE::eval_gpu()'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#a505f38ba93a3499895f5312e0112e73d',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#ad51666e69f670e286293aff96eb435a9',1,'mlx::core::fast::ScaledDotProductAttention::eval_gpu(const std::vector< array > &inputs, array &out)'],['../classmlx_1_1core_1_1fast_1_1_affine_quantize.html#a63812b2abaf26ad7e7fa4c9e82db1628',1,'mlx::core::fast::AffineQuantize::eval_gpu()'],['../classmlx_1_1core_1_1_primitive.html#ad217376dcf5eff691d731566faec2ba2',1,'mlx::core::Primitive::eval_gpu()'],['../classmlx_1_1core_1_1_unary_primitive.html#a6b7f80abaf038d53ec6ffbb0dfac6adb',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, array &output)=0'],['../classmlx_1_1core_1_1_unary_primitive.html#a971fe9ad47f6569118879ce1d0f41447',1,'mlx::core::UnaryPrimitive::eval_gpu(const std::vector< array > &inputs, std::vector< array > &outputs) override'],['../classmlx_1_1core_1_1_abs.html#a0a976e636dd8505b473fbdddf949f514',1,'mlx::core::Abs::eval_gpu()'],['../classmlx_1_1core_1_1_add.html#aa0aacbc1e26b95a2f040f62aa4f69c3d',1,'mlx::core::Add::eval_gpu()'],['../classmlx_1_1core_1_1_add_m_m.html#a5f933be14baebc32a0be0f9a69148aa9',1,'mlx::core::AddMM::eval_gpu()'],['../classmlx_1_1core_1_1_arange.html#a7a2e9787c6c3a78b4a6df91206974031',1,'mlx::core::Arange::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cos.html#a46f72d4af89b0a0f5f203783fb44589c',1,'mlx::core::ArcCos::eval_gpu()'],['../classmlx_1_1core_1_1_arc_cosh.html#aa6a2587485a0e015ac2d5211d7d045fc',1,'mlx::core::ArcCosh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sin.html#a7fa4ae7a85bc8bed97ea258ae30762f3',1,'mlx::core::ArcSin::eval_gpu()'],['../classmlx_1_1core_1_1_arc_sinh.html#a79f648a86de4c10386a1ce3b5e38e8ac',1,'mlx::core::ArcSinh::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan.html#a77866feb27028865d844070447c9a254',1,'mlx::core::ArcTan::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tan2.html#a76d3f0c29e0ff4642b8d39dac90d3f50',1,'mlx::core::ArcTan2::eval_gpu()'],['../classmlx_1_1core_1_1_arc_tanh.html#a10566b9d3b2c7d090895b46d9040bc1d',1,'mlx::core::ArcTanh::eval_gpu()'],['../classmlx_1_1core_1_1_arg_partition.html#a9a60995eaf85f63c877e86b23cbc15fc',1,'mlx::core::ArgPartition::eval_gpu()'],['../classmlx_1_1core_1_1_arg_reduce.html#aafa982ce2abc0cd9e81e43aa2c823d29',1,'mlx::core::ArgReduce::eval_gpu()'],['../classmlx_1_1core_1_1_arg_sort.html#abc2d730850ec4ee8d7968b7417911709',1,'mlx::core::ArgSort::eval_gpu()'],['../classmlx_1_1core_1_1_as_type.html#a5b111b9d74c60d27b4a7ebaa49f96e0b',1,'mlx::core::AsType::eval_gpu()'],['../classmlx_1_1core_1_1_as_strided.html#ab6771a208323994927ca162ba7bb10ed',1,'mlx::core::AsStrided::eval_gpu()'],['../classmlx_1_1core_1_1_bitwise_binary.html#ac831a29fc46701b00bbe63ee33832afd',1,'mlx::core::BitwiseBinary::eval_gpu()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#ab372b6df4de00a33795a052a23bb1df9',1,'mlx::core::BlockMaskedMM::eval_gpu()'],['../classmlx_1_1core_1_1_gather_m_m.html#ad754c35f460a055cc383ad93a5f72da1',1,'mlx::core::GatherMM::eval_gpu()'],['../classmlx_1_1core_1_1_broadcast.html#ab9bd9dbcedcefc9b29c84911b5ce69fe',1,'mlx::core::Broadcast::eval_gpu()'],['../classmlx_1_1core_1_1_ceil.html#abe178e0058e44b6618be414215e96887',1,'mlx::core::Ceil::eval_gpu()'],['../classmlx_1_1core_1_1_compiled.html#aa3d5ff0f2b3554ad48fbbf2a0f3336d5',1,'mlx::core::Compiled::eval_gpu()'],['../classmlx_1_1core_1_1_concatenate.html#a309a1c50e97f9925866433ee2841c474',1,'mlx::core::Concatenate::eval_gpu()'],['../classmlx_1_1core_1_1_conjugate.html#aff0a802166e3724db88ab5d3feb2d3de',1,'mlx::core::Conjugate::eval_gpu()'],['../classmlx_1_1core_1_1_convolution.html#a30b64109eeb1778f002b99447dff9dd2',1,'mlx::core::Convolution::eval_gpu()'],['../classmlx_1_1core_1_1_copy.html#a1eda7b2ea771a168f67421f0d384b3a1',1,'mlx::core::Copy::eval_gpu()'],['../classmlx_1_1core_1_1_cos.html#a5ef41aafad595f6cdd8c535e36e12060',1,'mlx::core::Cos::eval_gpu()'],['../classmlx_1_1core_1_1_cosh.html#a23f71b43792934c3ec0ebe9b74f32559',1,'mlx::core::Cosh::eval_gpu()'],['../classmlx_1_1core_1_1_custom_transforms.html#a7b3538681acbb20af3ed37b0877f6667',1,'mlx::core::CustomTransforms::eval_gpu()'],['../classmlx_1_1core_1_1_depends.html#ae5057f65e69490ad0add8eeda2b75e28',1,'mlx::core::Depends::eval_gpu()'],['../classmlx_1_1core_1_1_divide.html#abffda0ce37221ddc28dc9eea794f6bc7',1,'mlx::core::Divide::eval_gpu()'],['../classmlx_1_1core_1_1_div_mod.html#a003117c9ecf3c06a27248f72a76348dc',1,'mlx::core::DivMod::eval_gpu()'],['../classmlx_1_1core_1_1_select.html#a2a82b6cba4c386b2b87f225a4b08ea9b',1,'mlx::core::Select::eval_gpu()'],['../classmlx_1_1core_1_1_remainder.html#a7919ea9b84e42522d51bf0d5a396e161',1,'mlx::core::Remainder::eval_gpu()'],['../classmlx_1_1core_1_1_equal.html#ac3757001fec42ceb5ece2954df42161c',1,'mlx::core::Equal::eval_gpu()'],['../classmlx_1_1core_1_1_erf.html#ad8551be664d767dccc3c0d8cc1eca008',1,'mlx::core::Erf::eval_gpu()'],['../classmlx_1_1core_1_1_erf_inv.html#a4a2413d0634db1f3dae1806ddfa632db',1,'mlx::core::ErfInv::eval_gpu()'],['../classmlx_1_1core_1_1_exp.html#a7d63695a97a14760fd33b5d4e6590822',1,'mlx::core::Exp::eval_gpu()'],['../classmlx_1_1core_1_1_expm1.html#a82930071f4b77d883b300f77966aff5f',1,'mlx::core::Expm1::eval_gpu()'],['../classmlx_1_1core_1_1_f_f_t.html#a1c21b26d1e9ad7c4da78ae845721b2dd',1,'mlx::core::FFT::eval_gpu()'],['../classmlx_1_1core_1_1_floor.html#aaa29c83538099eb8f951c95a41f2eb65',1,'mlx::core::Floor::eval_gpu()'],['../classmlx_1_1core_1_1_full.html#aa54f99bb4cba12a551392dea56003872',1,'mlx::core::Full::eval_gpu()'],['../classmlx_1_1core_1_1_gather.html#aec48ee529cb2449915a7b27a3c4361e8',1,'mlx::core::Gather::eval_gpu()'],['../classmlx_1_1core_1_1_greater.html#ae8957cccf4c924d941f57a1bb751c878',1,'mlx::core::Greater::eval_gpu()'],['../classmlx_1_1core_1_1_greater_equal.html#ac246263b4548126c3d4ab7e392575d24',1,'mlx::core::GreaterEqual::eval_gpu()'],['../classmlx_1_1core_1_1_hadamard.html#a2470feb690f5463138490763c38b5733',1,'mlx::core::Hadamard::eval_gpu()'],['../classmlx_1_1core_1_1_less.html#a353335ce06ddbe8498d86d129c835917',1,'mlx::core::Less::eval_gpu()'],['../classmlx_1_1core_1_1_less_equal.html#acf035a82b11e6f63742143ea540fedac',1,'mlx::core::LessEqual::eval_gpu()'],['../classmlx_1_1core_1_1_load.html#a06933e887ea94a4d01d81195c5e07a3d',1,'mlx::core::Load::eval_gpu()'],['../classmlx_1_1core_1_1_log.html#aaaa49e9455f3a197bc319646b5ca6390',1,'mlx::core::Log::eval_gpu()'],['../classmlx_1_1core_1_1_log1p.html#a1b97decae7338d46874e736c95fa7431',1,'mlx::core::Log1p::eval_gpu()'],['../classmlx_1_1core_1_1_logical_not.html#a1d0d2bc93f935eca6c85ef7bf67f2d6a',1,'mlx::core::LogicalNot::eval_gpu()'],['../classmlx_1_1core_1_1_logical_and.html#a132b2eedaa3978de5a5350da3c2ca40f',1,'mlx::core::LogicalAnd::eval_gpu()'],['../classmlx_1_1core_1_1_logical_or.html#a3be1da328f0f8620de2e4fc1d22a077a',1,'mlx::core::LogicalOr::eval_gpu()'],['../classmlx_1_1core_1_1_log_add_exp.html#acace355b62ec00df649f9f99e8f2eb7a',1,'mlx::core::LogAddExp::eval_gpu()'],['../classmlx_1_1core_1_1_matmul.html#a8707a4e9b75c769e8f1dbca15c6a1ae7',1,'mlx::core::Matmul::eval_gpu()'],['../classmlx_1_1core_1_1_maximum.html#ade0f721b10a6b3a12bdadd34c48f72a7',1,'mlx::core::Maximum::eval_gpu()'],['../classmlx_1_1core_1_1_minimum.html#aadc68afa0afbe2103f19d161f5e0a2ba',1,'mlx::core::Minimum::eval_gpu()'],['../classmlx_1_1core_1_1_multiply.html#a634fcb4e981d8d3f4d94252caf25bee0',1,'mlx::core::Multiply::eval_gpu()'],['../classmlx_1_1core_1_1_negative.html#a97f1b316eace0c6d9e576d766940c75b',1,'mlx::core::Negative::eval_gpu()'],['../classmlx_1_1core_1_1_not_equal.html#a61179747e34e203150e9c660dfddb5f2',1,'mlx::core::NotEqual::eval_gpu()'],['../classmlx_1_1core_1_1_number_of_elements.html#a2c98c42915fb2bfe12f5c99ea553eff5',1,'mlx::core::NumberOfElements::eval_gpu()'],['../classmlx_1_1core_1_1_pad.html#aefd4d3a5bd8b6b35b266c9e558ada153',1,'mlx::core::Pad::eval_gpu()'],['../classmlx_1_1core_1_1_partition.html#a8eca1be21ae9ccfda46e6f3e85f506ef',1,'mlx::core::Partition::eval_gpu()'],['../classmlx_1_1core_1_1_power.html#a80577d4c0853c24027777c90a1ec7e11',1,'mlx::core::Power::eval_gpu()'],['../classmlx_1_1core_1_1_quantized_matmul.html#a2812ad007d695ed1aaf9cf706fb9c4b3',1,'mlx::core::QuantizedMatmul::eval_gpu()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a86eb048afc95646b2e96ec5493e3d887',1,'mlx::core::GatherQMM::eval_gpu()'],['../classmlx_1_1core_1_1_random_bits.html#a578756866665358577418e4cdd94aa3a',1,'mlx::core::RandomBits::eval_gpu()'],['../classmlx_1_1core_1_1_reshape.html#aa1e85f28471875750c47351520b56059',1,'mlx::core::Reshape::eval_gpu()'],['../classmlx_1_1core_1_1_reduce.html#ae9caaf42edadfe73ea208d98f526890f',1,'mlx::core::Reduce::eval_gpu()'],['../classmlx_1_1core_1_1_round.html#af7fe5ff8f3db166c203b4be4b07f13ec',1,'mlx::core::Round::eval_gpu()'],['../classmlx_1_1core_1_1_scan.html#aef22c6fc2b2cb2a907cd8965c7413dde',1,'mlx::core::Scan::eval_gpu()'],['../classmlx_1_1core_1_1_scatter.html#ab304345db3d8cfeea15e27461ae2e678',1,'mlx::core::Scatter::eval_gpu()'],['../classmlx_1_1core_1_1_sigmoid.html#a7a6bd0222d51d7f25f2719a91ccdfeca',1,'mlx::core::Sigmoid::eval_gpu()'],['../classmlx_1_1core_1_1_sign.html#afa2b48b99a194106006b44af69ffda8b',1,'mlx::core::Sign::eval_gpu()'],['../classmlx_1_1core_1_1_sin.html#a6b59f1156cf8bdad8d45acd1d825cb5e',1,'mlx::core::Sin::eval_gpu()'],['../classmlx_1_1core_1_1_sinh.html#a5a1af2399f166d5b228b5e83a1837c75',1,'mlx::core::Sinh::eval_gpu()'],['../classmlx_1_1core_1_1_slice.html#aa53c21ff06a7c659e889af6b97d10a4a',1,'mlx::core::Slice::eval_gpu()'],['../classmlx_1_1core_1_1_slice_update.html#aac1a1d122e5697be057d63552141032b',1,'mlx::core::SliceUpdate::eval_gpu()'],['../classmlx_1_1core_1_1_softmax.html#a35dac69ddcc7e2ec0e1a76fe93db85af',1,'mlx::core::Softmax::eval_gpu()'],['../classmlx_1_1core_1_1_sort.html#a4141c48f0e8670c728663f3722675382',1,'mlx::core::Sort::eval_gpu()'],['../classmlx_1_1core_1_1_split.html#a78ddda89c4daee73c74cfbc1e44656df',1,'mlx::core::Split::eval_gpu()'],['../classmlx_1_1core_1_1_square.html#a0ea2a78a5bb52daa4103263bf2f98045',1,'mlx::core::Square::eval_gpu()'],['../classmlx_1_1core_1_1_sqrt.html#a6d205e679a593d1ba20206c5c47ba501',1,'mlx::core::Sqrt::eval_gpu()'],['../classmlx_1_1core_1_1_stop_gradient.html#a907b96f0a1ce608e211d87ccf2b9ca89',1,'mlx::core::StopGradient::eval_gpu()'],['../classmlx_1_1core_1_1_subtract.html#a69021b23daf061764d97fabbc0f4f06c',1,'mlx::core::Subtract::eval_gpu()'],['../classmlx_1_1core_1_1_tan.html#aca7dbb4836507005a2032ac957a04d3f',1,'mlx::core::Tan::eval_gpu()'],['../classmlx_1_1core_1_1_tanh.html#a48df896599ae93dbce84a5c0f50cf761',1,'mlx::core::Tanh::eval_gpu()'],['../classmlx_1_1core_1_1_uniform.html#a5f88cbf2495f24f87cefd99aaaebe4d0',1,'mlx::core::Uniform::eval_gpu()'],['../classmlx_1_1core_1_1_view.html#add6e12ff1e476fe1db7718b14f21b075',1,'mlx::core::View::eval_gpu()'],['../classmlx_1_1core_1_1_transpose.html#a38d25739c08aa594a6775015a1d7d92e',1,'mlx::core::Transpose::eval_gpu()'],['../classmlx_1_1core_1_1_q_r_f.html#ae5fa3482192f4713605cd07e7fc1c6c9',1,'mlx::core::QRF::eval_gpu()'],['../classmlx_1_1core_1_1_s_v_d.html#a7067b2207f826a25549d571856b94e83',1,'mlx::core::SVD::eval_gpu()'],['../classmlx_1_1core_1_1_inverse.html#a086fbbc947ad232e01686ad063a78ed2',1,'mlx::core::Inverse::eval_gpu()'],['../classmlx_1_1core_1_1_cholesky.html#a8c918594bf129888044ef37fcae56795',1,'mlx::core::Cholesky::eval_gpu()']]], + ['event_25',['Event',['../classmlx_1_1core_1_1_event.html#a833506419b2110ad1abd89b2dd238b4d',1,'mlx::core::Event::Event()=default'],['../classmlx_1_1core_1_1_event.html#a13e4835f2ffb2cc22e29148a448ea184',1,'mlx::core::Event::Event(const Stream &steam)']]], + ['event_26',['event',['../classmlx_1_1core_1_1array.html#a0a8e4d6e67e739a712876bb36f88f9bf',1,'mlx::core::array']]], + ['exec_27',['exec',['../classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2',1,'pocketfft::detail::cfftp::exec()'],['../classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac',1,'pocketfft::detail::rfftp::exec()'],['../classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c',1,'pocketfft::detail::fftblue::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d',1,'pocketfft::detail::pocketfft_c::exec()'],['../classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33',1,'pocketfft::detail::pocketfft_r::exec()'],['../classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058',1,'pocketfft::detail::T_dct1::exec()'],['../classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26',1,'pocketfft::detail::T_dst1::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac',1,'pocketfft::detail::T_dcst23::exec()'],['../classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708',1,'pocketfft::detail::T_dcst4::exec()']]], + ['exec_5fr_28',['exec_r',['../classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00',1,'pocketfft::detail::fftblue']]], + ['exp_29',['Exp',['../classmlx_1_1core_1_1_exp.html#a1d0a618cbb91ab29ef53b57ff6ed6e06',1,'mlx::core::Exp']]], + ['exp_30',['exp',['../namespacemetal.html#ac2a0b3618d922ac014baac8189d44650',1,'metal::exp()'],['../namespacemetal_1_1fast.html#ad3dbd387b63373c29e3449609f763ede',1,'metal::fast::exp()'],['../namespacemetal_1_1precise.html#a8d8d2d5700ce432b33cf47cf22528e8f',1,'metal::precise::exp()'],['../group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125',1,'mlx::core::exp()']]], + ['exp10_31',['exp10',['../namespacemetal.html#a4c63707d13c89364496a48906631c204',1,'metal::exp10()'],['../namespacemetal_1_1fast.html#a453122f982485cbb4e471b3ac282ee5e',1,'metal::fast::exp10()'],['../namespacemetal_1_1precise.html#af9addb343c967da3a83e9e123a8521fd',1,'metal::precise::exp10()']]], + ['exp2_32',['exp2',['../namespacemetal.html#a228201c20777848804a4d0589c1d33e7',1,'metal::exp2()'],['../namespacemetal_1_1fast.html#ac092b65a46720adaf22f6266671d2d71',1,'metal::fast::exp2()'],['../namespacemetal_1_1precise.html#a92a880bd2197efc0da0f8f0f7ec1e4c9',1,'metal::precise::exp2()']]], + ['expand_5fdims_33',['expand_dims',['../group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133',1,'mlx::core::expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a',1,'mlx::core::expand_dims(const array &a, int axis, StreamOrDevice s={})']]], + ['expm1_34',['Expm1',['../classmlx_1_1core_1_1_expm1.html#a47c2a1b2a4ef6bb07ba77c55ddddaec2',1,'mlx::core::Expm1']]], + ['expm1_35',['expm1',['../group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0',1,'mlx::core']]], + ['expm1f_36',['expm1f',['../expm1f_8h.html#a87f66d30e185950f42ce3641783cdc40',1,'expm1f.h']]], + ['expm1f_5fscaled_5funchecked_37',['expm1f_scaled_unchecked',['../expm1f_8h.html#adf20e03405fba634ca8d01acac24592e',1,'expm1f.h']]], + ['export_5fto_5fdot_38',['export_to_dot',['../namespacemlx_1_1core.html#a57395bdf43d9c5c134e610c169222cca',1,'mlx::core::export_to_dot(std::ostream &os, const std::vector< array > &outputs)'],['../namespacemlx_1_1core.html#a839f94dbad44f0d37333006fc876b42e',1,'mlx::core::export_to_dot(std::ostream &os, Arrays &&... outputs)']]], + ['eye_39',['eye',['../group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3',1,'mlx::core::eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd',1,'mlx::core::eye(int n, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga61657db78ef35d41112d362c869c25d2',1,'mlx::core::eye(int n, int m, StreamOrDevice s={})'],['../group__ops.html#ga908a15b42834be498a46856c99dfc779',1,'mlx::core::eye(int n, int m, int k, StreamOrDevice s={})'],['../group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f',1,'mlx::core::eye(int n, StreamOrDevice s={})']]] ]; diff --git a/docs/build/html/search/functions_7.js b/docs/build/html/search/functions_7.js index d586d79749..c56c7f833f 100644 --- a/docs/build/html/search/functions_7.js +++ b/docs/build/html/search/functions_7.js @@ -12,59 +12,64 @@ var searchData= ['gemm_5fsplitk_9',['gemm_splitk',['../steel__gemm__splitk_8h.html#a3be6e095a0a026d3ecf57a3e67f76188',1,'steel_gemm_splitk.h']]], ['gemm_5fsplitk_5faccum_10',['gemm_splitk_accum',['../steel__gemm__splitk_8h.html#abeb921bf1dc7941125188ddd390b0907',1,'steel_gemm_splitk.h']]], ['gemm_5fsplitk_5faccum_5faxpby_11',['gemm_splitk_accum_axpby',['../steel__gemm__splitk_8h.html#acc33fdfaaf3eb3a0629b3d52c7043dc1',1,'steel_gemm_splitk.h']]], - ['general_5fc2r_12',['general_c2r',['../namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc',1,'pocketfft::detail']]], - ['general_5fnd_13',['general_nd',['../namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0',1,'pocketfft::detail']]], - ['general_5fr2c_14',['general_r2c',['../namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed',1,'pocketfft::detail']]], - ['get_5factive_5fmemory_15',['get_active_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a7a3ad4e33d57a47474c98e2f88e775d7',1,'mlx::core::metal::MetalAllocator::get_active_memory()'],['../namespacemlx_1_1core_1_1metal.html#a7b75c2639016ac4d350fa6c9da386667',1,'mlx::core::metal::get_active_memory()']]], - ['get_5farange_5fkernel_16',['get_arange_kernel',['../namespacemlx_1_1core.html#a76f614e9956a6ca05a9be4db5a483446',1,'mlx::core']]], - ['get_5fbinary_5fkernel_17',['get_binary_kernel',['../namespacemlx_1_1core.html#a4decd4a07d91487e6903f6e3c8b7513a',1,'mlx::core']]], - ['get_5fbinary_5ftwo_5fkernel_18',['get_binary_two_kernel',['../namespacemlx_1_1core.html#a4e809746f48e5dcf7fa63215d3f5e33e',1,'mlx::core']]], - ['get_5fcache_5fmemory_19',['get_cache_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ad3cabbe638917ca4114eb74dcabe381f',1,'mlx::core::metal::MetalAllocator::get_cache_memory()'],['../namespacemlx_1_1core_1_1metal.html#a43307654f62ed7c58e014be7fb03909c',1,'mlx::core::metal::get_cache_memory()']]], - ['get_5fcolocated_5fmtllib_5fpath_20',['get_colocated_mtllib_path',['../namespacemlx_1_1core_1_1metal.html#a5fd6ba2040e53a254b9d71ae7ebd315f',1,'mlx::core::metal']]], - ['get_5fcommand_5fbuffer_21',['get_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a5fe3970fbe92ccc55fce4241ffbe5210',1,'mlx::core::metal::Device']]], - ['get_5fcommand_5fbuffer_5fops_22',['get_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a064e1cb6a16de7a0619f6447622350f8',1,'mlx::core::metal::Device']]], - ['get_5fcommand_5fencoder_23',['get_command_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#affa682ef612def4890f5152f81ffb7e6',1,'mlx::core::metal::Device']]], - ['get_5fcopy_5fkernel_24',['get_copy_kernel',['../namespacemlx_1_1core.html#a05a220cff45f12439fde775983c6df78',1,'mlx::core']]], - ['get_5fdefault_5fstream_25',['get_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a2366c7b888e433608e203752edc92282',1,'mlx::core::scheduler::Scheduler']]], - ['get_5ffft_5fkernel_26',['get_fft_kernel',['../namespacemlx_1_1core.html#a1d4cffc3c78067b3d9a62d64f3fb686f',1,'mlx::core']]], - ['get_5ffunction_27',['get_function',['../classmlx_1_1core_1_1metal_1_1_device.html#abf4ecf035e8be2ff00d2a7e95475d7ac',1,'mlx::core::metal::Device::get_function(const std::string &base_name, MTL::Library *mtl_lib, const std::string &specialized_name="", const MTLFCList &func_consts={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#ac083535ead07161cb27ad31f8f64fb84',1,'mlx::core::metal::Device::get_function(const std::string &base_name, const std::string &lib_name="mlx", const std::string &specialized_name="", const MTLFCList &func_consts={})']]], - ['get_5fkernel_28',['get_kernel',['../classmlx_1_1core_1_1metal_1_1_device.html#a6810c4dcbcfbf93fc51d42aa5ff0fc3a',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#afa0cac9d800c21a8a7f6cb224256abaf',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, const std::string &lib_name="mlx", const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})']]], - ['get_5fkernel_5fpreamble_29',['get_kernel_preamble',['../compiled__preamble_8h.html#a1dfa17a0369fb90ff615c7461f5013f3',1,'compiled_preamble.h']]], - ['get_5flibrary_30',['get_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a86297d66de43ee77d0697f69c6bd0b87',1,'mlx::core::metal::Device::get_library(const std::string &name)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a961d3cdb4bfb14c1a0b0aeb05609f56e',1,'mlx::core::metal::Device::get_library(const std::string &name, const std::string &source_string, bool cache=true)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a8d8d9e28ffeea10d687b91ea480fb4bd',1,'mlx::core::metal::Device::get_library(const std::string &name, const MTL::StitchedLibraryDescriptor *desc, bool cache=true)']]], - ['get_5fmb_5fsort_5fkernel_31',['get_mb_sort_kernel',['../namespacemlx_1_1core.html#afb57825bb763050cc9a9d194aa41ac36',1,'mlx::core']]], - ['get_5fname_32',['get_name',['../structmlx_1_1core_1_1_node_namer.html#a1690dd38de288c0aee2bb53156eb770e',1,'mlx::core::NodeNamer']]], - ['get_5fpeak_5fmemory_33',['get_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ac7972a3fe58e69489de775a0f152da17',1,'mlx::core::metal::MetalAllocator::get_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#a4b67d680cefa95f0ed5801f0e14e48ce',1,'mlx::core::metal::get_peak_memory()']]], - ['get_5fplan_34',['get_plan',['../namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b',1,'pocketfft::detail']]], - ['get_5fpool_35',['get_pool',['../namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a',1,'pocketfft::detail::threading']]], - ['get_5fquantized_5fkernel_36',['get_quantized_kernel',['../namespacemlx_1_1core.html#aa3faeae5378bfaafe3ce3432a051e43e',1,'mlx::core']]], - ['get_5freduce_5finit_5fkernel_37',['get_reduce_init_kernel',['../namespacemlx_1_1core.html#a51c4bb09230348bd0252e22bfdc9bc89',1,'mlx::core']]], - ['get_5freduce_5fkernel_38',['get_reduce_kernel',['../namespacemlx_1_1core.html#a5d6843dcbdbd4d12851fc2edbafbc9e6',1,'mlx::core']]], - ['get_5freduction_5fplan_39',['get_reduction_plan',['../namespacemlx_1_1core.html#a1555dc378c5254e79199421761f26f2b',1,'mlx::core']]], - ['get_5fscan_5fkernel_40',['get_scan_kernel',['../namespacemlx_1_1core.html#aeefaff208444d3fa61ecc0946fe1de5f',1,'mlx::core']]], - ['get_5fshape_41',['get_shape',['../namespacemlx_1_1core.html#aab0d8a256957984acc1e3615c65c898e',1,'mlx::core']]], - ['get_5fsoftmax_5fkernel_42',['get_softmax_kernel',['../namespacemlx_1_1core.html#a35a412f688d79eb47e42d20a7c8650ee',1,'mlx::core']]], - ['get_5fsort_5fkernel_43',['get_sort_kernel',['../namespacemlx_1_1core.html#a84ebe6275218070f0ea320f126f64e22',1,'mlx::core']]], - ['get_5fsteel_5fconv_5fgeneral_5fkernel_44',['get_steel_conv_general_kernel',['../namespacemlx_1_1core.html#abce2b67044ee06a7bbe7a91ec7c8c48d',1,'mlx::core']]], - ['get_5fsteel_5fconv_5fkernel_45',['get_steel_conv_kernel',['../namespacemlx_1_1core.html#adce79d220672f5f3c65cc31d145ca9c4',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5ffused_5fkernel_46',['get_steel_gemm_fused_kernel',['../namespacemlx_1_1core.html#a84fa8e0aee321a9d614433a0b933103b',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fmasked_5fkernel_47',['get_steel_gemm_masked_kernel',['../namespacemlx_1_1core.html#ab5f60614e965144b451930fdf935e08d',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fsplitk_5faccum_5fkernel_48',['get_steel_gemm_splitk_accum_kernel',['../namespacemlx_1_1core.html#a195b86cad5bb99aa1bcd23952305af6b',1,'mlx::core']]], - ['get_5fsteel_5fgemm_5fsplitk_5fkernel_49',['get_steel_gemm_splitk_kernel',['../namespacemlx_1_1core.html#af48c6f2f72b61dbd6766e4f5fea85df5',1,'mlx::core']]], - ['get_5ftemplate_5fdefinition_50',['get_template_definition',['../namespacemlx_1_1core.html#aae0d19f0acdef2accd2428fb84c8a032',1,'mlx::core']]], - ['get_5fternary_5fkernel_51',['get_ternary_kernel',['../namespacemlx_1_1core.html#a54eb3b65375022428aab5f810e40624b',1,'mlx::core']]], - ['get_5ftwiddle_52',['get_twiddle',['../radix_8h.html#ac5cf950316b9445296ee9ecfc56a56bd',1,'radix.h']]], - ['get_5ftype_5fstring_53',['get_type_string',['../namespacemlx_1_1core.html#af776fd91dd60594dcfebbafd17f19068',1,'mlx::core']]], - ['get_5funary_5fkernel_54',['get_unary_kernel',['../namespacemlx_1_1core.html#a15175e8e2b1e26726c63393e4d68b628',1,'mlx::core']]], - ['gguf_5fload_5fquantized_55',['gguf_load_quantized',['../namespacemlx_1_1core.html#a65dd68163bdaef3631e3724327782498',1,'mlx::core']]], - ['good_56',['good',['../classmlx_1_1core_1_1io_1_1_reader.html#a005d0b52c1f34866f7412b7f41dabec3',1,'mlx::core::io::Reader::good()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a0b050c2c27487007e250e2e19560ffe4',1,'mlx::core::io::Writer::good()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a4ecb046dfa7dab346d096adab3a1c318',1,'mlx::core::io::FileReader::good()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9ec4934b26fb358d699ddce1482b2d54',1,'mlx::core::io::FileWriter::good()']]], - ['good_5fsize_5fcmplx_57',['good_size_cmplx',['../structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624',1,'pocketfft::detail::util']]], - ['good_5fsize_5freal_58',['good_size_real',['../structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa',1,'pocketfft::detail::util']]], - ['grad_59',['grad',['../namespacemlx_1_1core.html#af482f6c64acd77c57ef5bb4b7be9726c',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a64bc619876b0f8cc81a2637ca81c99f7',1,'mlx::core::grad(const std::function< array(const array &)> &fun)']]], - ['greater_60',['Greater',['../classmlx_1_1core_1_1_greater.html#a1d5992a66c020cd97a70e8e3d8cd1a1b',1,'mlx::core::Greater']]], - ['greater_61',['greater',['../group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d',1,'mlx::core']]], - ['greater_5fequal_62',['greater_equal',['../group__ops.html#ga7153071bcfff6faad21332163fb9a430',1,'mlx::core']]], - ['greaterequal_63',['GreaterEqual',['../classmlx_1_1core_1_1_greater_equal.html#a19a3c49d5a9b40e17da0e56ef6908527',1,'mlx::core::GreaterEqual']]], - ['group_64',['group',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8831cb61ac633431b78b5fb99c0ea9ff',1,'mlx::core::distributed::DistPrimitive']]], - ['group_65',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html#a6f84accc8d6734989b2757bf6cdd0152',1,'mlx::core::distributed::Group']]], - ['gumbel_66',['gumbel',['../namespacemlx_1_1core_1_1random.html#aa849b765cd794306997bcbb9936d3d84',1,'mlx::core::random']]] + ['gemv_5fmasked_12',['gemv_masked',['../namespacemlx_1_1core_1_1metal.html#abc055b75e6a059618f279c35f8de36e7',1,'mlx::core::metal::gemv_masked()'],['../kernels_2gemv__masked_8h.html#ab3070d14cdecb1dd7dc220a551da6b7b',1,'gemv_masked(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const constant int &batch_ndim, const constant int *batch_shape, const constant size_t *vector_batch_stride, const constant size_t *matrix_batch_stride, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, const constant size_t *mask_batch_strides, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid): gemv_masked.h']]], + ['gemv_5ft_5fmasked_13',['gemv_t_masked',['../kernels_2gemv__masked_8h.html#a0c8d353fc453e448b2d0ed9a19431b63',1,'gemv_masked.h']]], + ['general_5fc2r_14',['general_c2r',['../namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc',1,'pocketfft::detail']]], + ['general_5fnd_15',['general_nd',['../namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0',1,'pocketfft::detail']]], + ['general_5fr2c_16',['general_r2c',['../namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed',1,'pocketfft::detail']]], + ['get_5f2d_5fgrid_5fdims_17',['get_2d_grid_dims',['../namespacemlx_1_1core.html#a8dc169474a51a1f4f761d5752819bd7c',1,'mlx::core']]], + ['get_5factive_5fmemory_18',['get_active_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a7a3ad4e33d57a47474c98e2f88e775d7',1,'mlx::core::metal::MetalAllocator::get_active_memory()'],['../namespacemlx_1_1core_1_1metal.html#a7b75c2639016ac4d350fa6c9da386667',1,'mlx::core::metal::get_active_memory()']]], + ['get_5farange_5fkernel_19',['get_arange_kernel',['../namespacemlx_1_1core.html#a76f614e9956a6ca05a9be4db5a483446',1,'mlx::core']]], + ['get_5fbinary_5fkernel_20',['get_binary_kernel',['../namespacemlx_1_1core.html#a4decd4a07d91487e6903f6e3c8b7513a',1,'mlx::core']]], + ['get_5fbinary_5ftwo_5fkernel_21',['get_binary_two_kernel',['../namespacemlx_1_1core.html#a4e809746f48e5dcf7fa63215d3f5e33e',1,'mlx::core']]], + ['get_5fblock_5fdims_22',['get_block_dims',['../namespacemlx_1_1core.html#a7b3f872f6046bdd4d429ee1f1c31cd40',1,'mlx::core']]], + ['get_5fcache_5fmemory_23',['get_cache_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ad3cabbe638917ca4114eb74dcabe381f',1,'mlx::core::metal::MetalAllocator::get_cache_memory()'],['../namespacemlx_1_1core_1_1metal.html#a43307654f62ed7c58e014be7fb03909c',1,'mlx::core::metal::get_cache_memory()']]], + ['get_5fcommand_5fbuffer_24',['get_command_buffer',['../classmlx_1_1core_1_1metal_1_1_device.html#a5fe3970fbe92ccc55fce4241ffbe5210',1,'mlx::core::metal::Device']]], + ['get_5fcommand_5fbuffer_5fops_25',['get_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a064e1cb6a16de7a0619f6447622350f8',1,'mlx::core::metal::Device']]], + ['get_5fcommand_5fencoder_26',['get_command_encoder',['../classmlx_1_1core_1_1metal_1_1_device.html#affa682ef612def4890f5152f81ffb7e6',1,'mlx::core::metal::Device']]], + ['get_5fcopy_5fkernel_27',['get_copy_kernel',['../namespacemlx_1_1core.html#a05a220cff45f12439fde775983c6df78',1,'mlx::core']]], + ['get_5fdefault_5fstream_28',['get_default_stream',['../classmlx_1_1core_1_1scheduler_1_1_scheduler.html#a2366c7b888e433608e203752edc92282',1,'mlx::core::scheduler::Scheduler']]], + ['get_5ffft_5fkernel_29',['get_fft_kernel',['../namespacemlx_1_1core.html#a1d4cffc3c78067b3d9a62d64f3fb686f',1,'mlx::core']]], + ['get_5ffunction_30',['get_function',['../classmlx_1_1core_1_1metal_1_1_device.html#abf4ecf035e8be2ff00d2a7e95475d7ac',1,'mlx::core::metal::Device::get_function(const std::string &base_name, MTL::Library *mtl_lib, const std::string &specialized_name="", const MTLFCList &func_consts={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#ac083535ead07161cb27ad31f8f64fb84',1,'mlx::core::metal::Device::get_function(const std::string &base_name, const std::string &lib_name="mlx", const std::string &specialized_name="", const MTLFCList &func_consts={})']]], + ['get_5fgemv_5fmasked_5fkernel_31',['get_gemv_masked_kernel',['../namespacemlx_1_1core.html#a90c24e0d0b99b68fad9deefcf4d3e818',1,'mlx::core']]], + ['get_5fkernel_32',['get_kernel',['../classmlx_1_1core_1_1metal_1_1_device.html#a6810c4dcbcfbf93fc51d42aa5ff0fc3a',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, MTL::Library *mtl_lib, const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})'],['../classmlx_1_1core_1_1metal_1_1_device.html#afa0cac9d800c21a8a7f6cb224256abaf',1,'mlx::core::metal::Device::get_kernel(const std::string &base_name, const std::string &lib_name="mlx", const std::string &hash_name="", const MTLFCList &func_consts={}, const std::vector< MTL::Function * > &linked_functions={})']]], + ['get_5fkernel_5fpreamble_33',['get_kernel_preamble',['../compiled__preamble_8h.html#a1dfa17a0369fb90ff615c7461f5013f3',1,'compiled_preamble.h']]], + ['get_5flibrary_34',['get_library',['../classmlx_1_1core_1_1metal_1_1_device.html#a86297d66de43ee77d0697f69c6bd0b87',1,'mlx::core::metal::Device::get_library(const std::string &name)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a961d3cdb4bfb14c1a0b0aeb05609f56e',1,'mlx::core::metal::Device::get_library(const std::string &name, const std::string &source_string, bool cache=true)'],['../classmlx_1_1core_1_1metal_1_1_device.html#a8d8d9e28ffeea10d687b91ea480fb4bd',1,'mlx::core::metal::Device::get_library(const std::string &name, const MTL::StitchedLibraryDescriptor *desc, bool cache=true)']]], + ['get_5fmb_5fsort_5fkernel_35',['get_mb_sort_kernel',['../namespacemlx_1_1core.html#afb57825bb763050cc9a9d194aa41ac36',1,'mlx::core']]], + ['get_5fname_36',['get_name',['../structmlx_1_1core_1_1_node_namer.html#a1690dd38de288c0aee2bb53156eb770e',1,'mlx::core::NodeNamer']]], + ['get_5fpeak_5fmemory_37',['get_peak_memory',['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#ac7972a3fe58e69489de775a0f152da17',1,'mlx::core::metal::MetalAllocator::get_peak_memory()'],['../namespacemlx_1_1core_1_1metal.html#a4b67d680cefa95f0ed5801f0e14e48ce',1,'mlx::core::metal::get_peak_memory()']]], + ['get_5fplan_38',['get_plan',['../namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b',1,'pocketfft::detail']]], + ['get_5fpool_39',['get_pool',['../namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a',1,'pocketfft::detail::threading']]], + ['get_5fprimitive_5fstring_40',['get_primitive_string',['../namespacemlx_1_1core.html#ad4be35b310a252edd80d9cf04f094a60',1,'mlx::core']]], + ['get_5fquantized_5fkernel_41',['get_quantized_kernel',['../namespacemlx_1_1core.html#aa3faeae5378bfaafe3ce3432a051e43e',1,'mlx::core']]], + ['get_5freduce_5finit_5fkernel_42',['get_reduce_init_kernel',['../namespacemlx_1_1core.html#a51c4bb09230348bd0252e22bfdc9bc89',1,'mlx::core']]], + ['get_5freduce_5fkernel_43',['get_reduce_kernel',['../namespacemlx_1_1core.html#a5d6843dcbdbd4d12851fc2edbafbc9e6',1,'mlx::core']]], + ['get_5freduction_5fplan_44',['get_reduction_plan',['../namespacemlx_1_1core.html#a1555dc378c5254e79199421761f26f2b',1,'mlx::core']]], + ['get_5fscan_5fkernel_45',['get_scan_kernel',['../namespacemlx_1_1core.html#aeefaff208444d3fa61ecc0946fe1de5f',1,'mlx::core']]], + ['get_5fshape_46',['get_shape',['../namespacemlx_1_1core.html#aab0d8a256957984acc1e3615c65c898e',1,'mlx::core']]], + ['get_5fsoftmax_5fkernel_47',['get_softmax_kernel',['../namespacemlx_1_1core.html#a35a412f688d79eb47e42d20a7c8650ee',1,'mlx::core']]], + ['get_5fsort_5fkernel_48',['get_sort_kernel',['../namespacemlx_1_1core.html#a84ebe6275218070f0ea320f126f64e22',1,'mlx::core']]], + ['get_5fsteel_5fconv_5fgeneral_5fkernel_49',['get_steel_conv_general_kernel',['../namespacemlx_1_1core.html#abce2b67044ee06a7bbe7a91ec7c8c48d',1,'mlx::core']]], + ['get_5fsteel_5fconv_5fkernel_50',['get_steel_conv_kernel',['../namespacemlx_1_1core.html#adce79d220672f5f3c65cc31d145ca9c4',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5ffused_5fkernel_51',['get_steel_gemm_fused_kernel',['../namespacemlx_1_1core.html#a84fa8e0aee321a9d614433a0b933103b',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fmasked_5fkernel_52',['get_steel_gemm_masked_kernel',['../namespacemlx_1_1core.html#ab5f60614e965144b451930fdf935e08d',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fsplitk_5faccum_5fkernel_53',['get_steel_gemm_splitk_accum_kernel',['../namespacemlx_1_1core.html#a195b86cad5bb99aa1bcd23952305af6b',1,'mlx::core']]], + ['get_5fsteel_5fgemm_5fsplitk_5fkernel_54',['get_steel_gemm_splitk_kernel',['../namespacemlx_1_1core.html#af48c6f2f72b61dbd6766e4f5fea85df5',1,'mlx::core']]], + ['get_5ftemplate_5fdefinition_55',['get_template_definition',['../namespacemlx_1_1core.html#aae0d19f0acdef2accd2428fb84c8a032',1,'mlx::core']]], + ['get_5fternary_5fkernel_56',['get_ternary_kernel',['../namespacemlx_1_1core.html#a54eb3b65375022428aab5f810e40624b',1,'mlx::core']]], + ['get_5ftwiddle_57',['get_twiddle',['../radix_8h.html#ac5cf950316b9445296ee9ecfc56a56bd',1,'radix.h']]], + ['get_5ftype_5fstring_58',['get_type_string',['../namespacemlx_1_1core.html#af776fd91dd60594dcfebbafd17f19068',1,'mlx::core']]], + ['get_5funary_5fkernel_59',['get_unary_kernel',['../namespacemlx_1_1core.html#a15175e8e2b1e26726c63393e4d68b628',1,'mlx::core']]], + ['gguf_5fload_5fquantized_60',['gguf_load_quantized',['../namespacemlx_1_1core.html#a65dd68163bdaef3631e3724327782498',1,'mlx::core']]], + ['good_61',['good',['../classmlx_1_1core_1_1io_1_1_reader.html#a005d0b52c1f34866f7412b7f41dabec3',1,'mlx::core::io::Reader::good()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a0b050c2c27487007e250e2e19560ffe4',1,'mlx::core::io::Writer::good()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a4ecb046dfa7dab346d096adab3a1c318',1,'mlx::core::io::FileReader::good()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a9ec4934b26fb358d699ddce1482b2d54',1,'mlx::core::io::FileWriter::good()']]], + ['good_5fsize_5fcmplx_62',['good_size_cmplx',['../structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624',1,'pocketfft::detail::util']]], + ['good_5fsize_5freal_63',['good_size_real',['../structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa',1,'pocketfft::detail::util']]], + ['grad_64',['grad',['../namespacemlx_1_1core.html#af482f6c64acd77c57ef5bb4b7be9726c',1,'mlx::core::grad(const std::function< array(const std::vector< array > &)> &fun, int argnum=0)'],['../namespacemlx_1_1core.html#a64bc619876b0f8cc81a2637ca81c99f7',1,'mlx::core::grad(const std::function< array(const array &)> &fun)']]], + ['greater_65',['Greater',['../classmlx_1_1core_1_1_greater.html#a1d5992a66c020cd97a70e8e3d8cd1a1b',1,'mlx::core::Greater']]], + ['greater_66',['greater',['../group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d',1,'mlx::core']]], + ['greater_5fequal_67',['greater_equal',['../group__ops.html#ga7153071bcfff6faad21332163fb9a430',1,'mlx::core']]], + ['greaterequal_68',['GreaterEqual',['../classmlx_1_1core_1_1_greater_equal.html#a19a3c49d5a9b40e17da0e56ef6908527',1,'mlx::core::GreaterEqual']]], + ['group_69',['Group',['../structmlx_1_1core_1_1distributed_1_1_group.html#a6f84accc8d6734989b2757bf6cdd0152',1,'mlx::core::distributed::Group']]], + ['group_70',['group',['../classmlx_1_1core_1_1distributed_1_1_dist_primitive.html#a8831cb61ac633431b78b5fb99c0ea9ff',1,'mlx::core::distributed::DistPrimitive']]], + ['gumbel_71',['gumbel',['../namespacemlx_1_1core_1_1random.html#aa849b765cd794306997bcbb9936d3d84',1,'mlx::core::random']]] ]; diff --git a/docs/build/html/search/functions_9.js b/docs/build/html/search/functions_9.js index d69ae2ea79..07ca8be157 100644 --- a/docs/build/html/search/functions_9.js +++ b/docs/build/html/search/functions_9.js @@ -10,34 +10,33 @@ var searchData= ['in_5ftracing_7',['in_tracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#ac52b8e2c3f808d3076c4e1ebaf9dc63d',1,'mlx::core::detail::InTracing']]], ['increment_5fcommand_5fbuffer_5fops_8',['increment_command_buffer_ops',['../classmlx_1_1core_1_1metal_1_1_device.html#a7a33d4d601423a3d3c23d5ad7072abb6',1,'mlx::core::metal::Device']]], ['infinity_9',['infinity',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a61eb741e7af49046beb863abf023b206',1,'metal::_numeric_limits_impl< bfloat16_t >']]], - ['init_10',['init',['../class_m_p_s_1_1_matrix.html#a2398221610a7c610c85cf7fb3f18281b',1,'MPS::Matrix::init(MTL::Buffer *buffer, MatrixDescriptor *descriptor)'],['../class_m_p_s_1_1_matrix.html#a40771ae44d119abd56bd32b7cc38feda',1,'MPS::Matrix::init(const MTL::Buffer *buffer, MatrixDescriptor *descriptor)'],['../class_m_p_s_1_1_matrix_multiplication.html#a35b1a908807eb8ececd0241531dea823',1,'MPS::MatrixMultiplication::init()'],['../class_m_p_s_1_1_vector.html#a2ea0c0575489e4689d19640d1da1b179',1,'MPS::Vector::init(MTL::Buffer *buffer, VectorDescriptor *descriptor)'],['../class_m_p_s_1_1_vector.html#aa95af1607e9978c2757f8a5503a95244',1,'MPS::Vector::init(const MTL::Buffer *buffer, VectorDescriptor *descriptor)'],['../class_m_p_s_1_1_matrix_vector_multiplication.html#a62c065490befe32c6a59b888b1347c80',1,'MPS::MatrixVectorMultiplication::init()'],['../namespacemlx_1_1core_1_1distributed.html#a33633c058c7ec82cca4f237243c6810d',1,'mlx::core::distributed::init()']]], + ['init_10',['init',['../namespacemlx_1_1core_1_1distributed.html#a33633c058c7ec82cca4f237243c6810d',1,'mlx::core::distributed']]], ['init_5freduce_11',['init_reduce',['../reduce__init_8h.html#a0088604ac2eaa6940689ff12c4ba5fc2',1,'reduce_init.h']]], ['inner_12',['inner',['../group__ops.html#ga654fec16a9746b390916697a2ab2546e',1,'mlx::core']]], ['inputs_13',['inputs',['../classmlx_1_1core_1_1array.html#a2913abcdf71826827c8457f529825fff',1,'mlx::core::array::inputs() const'],['../classmlx_1_1core_1_1array.html#acffb082177f9b78f0c52e406adff972f',1,'mlx::core::array::inputs()']]], ['intracing_14',['InTracing',['../structmlx_1_1core_1_1detail_1_1_in_tracing.html#a7a77f19391498afa5dcea3509d241a70',1,'mlx::core::detail::InTracing']]], ['inv_15',['inv',['../namespacemlx_1_1core_1_1linalg.html#ad966a0b6bff176c9f933534ed62389a2',1,'mlx::core::linalg']]], - ['inverse_16',['Inverse',['../classmlx_1_1core_1_1_inverse.html#a4fcdda66a176b520fe4655ab91862aad',1,'mlx::core::Inverse']]], + ['inverse_16',['Inverse',['../classmlx_1_1core_1_1_inverse.html#a71467681e523abb725724490bfeb76ad',1,'mlx::core::Inverse']]], ['iofs_17',['iofs',['../classpocketfft_1_1detail_1_1multi__iter.html#ad33360d4a8ab8e6d72efadc6f9cb5bfa',1,'pocketfft::detail::multi_iter::iofs(size_t i) const'],['../classpocketfft_1_1detail_1_1multi__iter.html#a97462d97bdca6419d8d2f37c2031fe83',1,'pocketfft::detail::multi_iter::iofs(size_t j, size_t i) const']]], ['irfft_18',['irfft',['../namespacemlx_1_1core_1_1fft.html#aafa721d0492e9f74913a6e86b4896ad8',1,'mlx::core::fft::irfft(const array &a, int n, int axis, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#afbd0035a3cf91f428838de1fcf01a3a3',1,'mlx::core::fft::irfft(const array &a, int axis=-1, StreamOrDevice s={})']]], ['irfft2_19',['irfft2',['../namespacemlx_1_1core_1_1fft.html#a35754b00e98d7ef37ce8230c8887a933',1,'mlx::core::fft::irfft2(const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#af7c7bbbbce26c2775a77473502a8de02',1,'mlx::core::fft::irfft2(const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})']]], ['irfftn_20',['irfftn',['../namespacemlx_1_1core_1_1fft.html#a33f2973ea1b621e67064e46136d2960f',1,'mlx::core::fft::irfftn(const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#a1c9ad11121c5879d5c04bbde2ee238c3',1,'mlx::core::fft::irfftn(const array &a, const std::vector< int > &axes, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1fft.html#aaf5a7ef93b3426b94c2363a23a5a5b36',1,'mlx::core::fft::irfftn(const array &a, StreamOrDevice s={})']]], ['is_5favailable_21',['is_available',['../classmlx_1_1core_1_1array.html#aebed1f37c19197be76105161102a8a40',1,'mlx::core::array::is_available()'],['../namespacemlx_1_1core_1_1metal.html#a0cdf2c08c7bc0927a86070adc206987f',1,'mlx::core::metal::is_available()'],['../namespacemlx_1_1core_1_1distributed.html#a95655473cd0032c06e5fe3fca85aeef3',1,'mlx::core::distributed::is_available()']]], - ['is_5fbig_5fendian_22',['is_big_endian',['../namespacemlx_1_1core.html#a625ed440df3fa57318017c1f2c589efe',1,'mlx::core']]], - ['is_5fdonatable_23',['is_donatable',['../classmlx_1_1core_1_1array.html#a4677a404b5d191af20b52649225de087',1,'mlx::core::array']]], - ['is_5fequivalent_24',['is_equivalent',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#af08b1294f3f93505a96fdfa85b1edd62',1,'mlx::core::fast::ScaledDotProductAttention::is_equivalent()'],['../classmlx_1_1core_1_1_primitive.html#a6140a502af4c2bbbc776ab26e9afebcd',1,'mlx::core::Primitive::is_equivalent()'],['../classmlx_1_1core_1_1_abs.html#ab6f0ec56bc7c048382297e12dabadc67',1,'mlx::core::Abs::is_equivalent()'],['../classmlx_1_1core_1_1_add.html#aba0a35410c3aac53d0f7a0c283d9ee3f',1,'mlx::core::Add::is_equivalent()'],['../classmlx_1_1core_1_1_add_m_m.html#a6e37c6882dba995a63fb6d8dfb01754f',1,'mlx::core::AddMM::is_equivalent()'],['../classmlx_1_1core_1_1_arange.html#a7b6a45cf9c4b109d4e0373f3fe576c35',1,'mlx::core::Arange::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cos.html#a39557461e3235801886675a9b7d25bf5',1,'mlx::core::ArcCos::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6928e827b9ac2e86e7d5b02b78150eee',1,'mlx::core::ArcCosh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sin.html#a13b5e39eeccaf32d94b8eb85b3b753ab',1,'mlx::core::ArcSin::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sinh.html#a63c7a765c7906242dc3371deec094f0f',1,'mlx::core::ArcSinh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan.html#a0e5b5fc7218143ecd0a8666d9137c34c',1,'mlx::core::ArcTan::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan2.html#aeaee58cd803d3ebf0b76574a409682cc',1,'mlx::core::ArcTan2::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tanh.html#ac8ecdd640043dab0461d49d7650679a2',1,'mlx::core::ArcTanh::is_equivalent()'],['../classmlx_1_1core_1_1_arg_partition.html#ad87509ce70b51fb75dfb9c3a05a5b31a',1,'mlx::core::ArgPartition::is_equivalent()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03b81a670dcb1e39bf7279e4d4583b97',1,'mlx::core::ArgReduce::is_equivalent()'],['../classmlx_1_1core_1_1_arg_sort.html#a048cd09c557d29d1111726f97010a845',1,'mlx::core::ArgSort::is_equivalent()'],['../classmlx_1_1core_1_1_as_type.html#a8e6c8b2428ab15c4fb43f2e3a8fb38af',1,'mlx::core::AsType::is_equivalent()'],['../classmlx_1_1core_1_1_as_strided.html#a1738c6aa0a3a3eb68530f0d5b436e094',1,'mlx::core::AsStrided::is_equivalent()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8cd6b916b4838a6c329cf4df8530c3b8',1,'mlx::core::BitwiseBinary::is_equivalent()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aef1c303955f9b8f445296372cf181160',1,'mlx::core::BlockMaskedMM::is_equivalent()'],['../classmlx_1_1core_1_1_gather_m_m.html#a163f17f6ce2c002f22e81b302777342b',1,'mlx::core::GatherMM::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast.html#a0e27692b0090ec451954649a36042616',1,'mlx::core::Broadcast::is_equivalent()'],['../classmlx_1_1core_1_1_ceil.html#aacd90acb56eb0649c1cef807aa21df52',1,'mlx::core::Ceil::is_equivalent()'],['../classmlx_1_1core_1_1_compiled.html#a63e5016458887813b4a59dee5a0a3f10',1,'mlx::core::Compiled::is_equivalent()'],['../classmlx_1_1core_1_1_concatenate.html#aaf8a72a0c30114460caf519580cc35d2',1,'mlx::core::Concatenate::is_equivalent()'],['../classmlx_1_1core_1_1_conjugate.html#af42f00a790c6bc5572bd8fe9e5b36c5e',1,'mlx::core::Conjugate::is_equivalent()'],['../classmlx_1_1core_1_1_convolution.html#afb87708a5e3aab2e9e663daa9d8863de',1,'mlx::core::Convolution::is_equivalent()'],['../classmlx_1_1core_1_1_copy.html#afcfa39465015f638e294aa954ea0f3da',1,'mlx::core::Copy::is_equivalent()'],['../classmlx_1_1core_1_1_cos.html#ab611ca38c987915659f7ffcce0370417',1,'mlx::core::Cos::is_equivalent()'],['../classmlx_1_1core_1_1_cosh.html#ae0bacccaf501f5349db0c13cca776ff9',1,'mlx::core::Cosh::is_equivalent()'],['../classmlx_1_1core_1_1_divide.html#a3dda091f05c4164c29bb8129e9712650',1,'mlx::core::Divide::is_equivalent()'],['../classmlx_1_1core_1_1_div_mod.html#af5fcf8ec8515d46844cbeeab6dafb38a',1,'mlx::core::DivMod::is_equivalent()'],['../classmlx_1_1core_1_1_select.html#afc3c333fac7f902c98839921ef2874c8',1,'mlx::core::Select::is_equivalent()'],['../classmlx_1_1core_1_1_remainder.html#a802039faaa2ed7b763ec3d7debcce814',1,'mlx::core::Remainder::is_equivalent()'],['../classmlx_1_1core_1_1_equal.html#a58c1c5003e43f47dc0788c1851deaa02',1,'mlx::core::Equal::is_equivalent()'],['../classmlx_1_1core_1_1_erf.html#abe99dfbc2954c3a7d5dec56ab165ee82',1,'mlx::core::Erf::is_equivalent()'],['../classmlx_1_1core_1_1_erf_inv.html#aaac9e3b454ba564f9c6e804ab6562832',1,'mlx::core::ErfInv::is_equivalent()'],['../classmlx_1_1core_1_1_exp.html#ac6e44bffe7a643ab4ca51e74c7328357',1,'mlx::core::Exp::is_equivalent()'],['../classmlx_1_1core_1_1_f_f_t.html#a0ede3bc8b6d77d560c0a750b68fddc06',1,'mlx::core::FFT::is_equivalent()'],['../classmlx_1_1core_1_1_floor.html#a24b64feb026c4fcd02fc481cffdb1c94',1,'mlx::core::Floor::is_equivalent()'],['../classmlx_1_1core_1_1_full.html#afafcbcae1e28597fe8f7fde289105792',1,'mlx::core::Full::is_equivalent()'],['../classmlx_1_1core_1_1_gather.html#a23ff1406dbf0c770e75ad47440b467aa',1,'mlx::core::Gather::is_equivalent()'],['../classmlx_1_1core_1_1_greater.html#a6877a6888614a618dc64296763ccabb1',1,'mlx::core::Greater::is_equivalent()'],['../classmlx_1_1core_1_1_greater_equal.html#a3daef8596b963026b602019bc56fc5fc',1,'mlx::core::GreaterEqual::is_equivalent()'],['../classmlx_1_1core_1_1_hadamard.html#a8a528d8d69a7343bdfd704a3e74230b8',1,'mlx::core::Hadamard::is_equivalent()'],['../classmlx_1_1core_1_1_less.html#a7d6ed6353a0dcefebd008026dbd3cd63',1,'mlx::core::Less::is_equivalent()'],['../classmlx_1_1core_1_1_less_equal.html#a76ee1438cf4bd109eae4e0b3472b26af',1,'mlx::core::LessEqual::is_equivalent()'],['../classmlx_1_1core_1_1_log.html#a2fc58ea4ca744db493b947d1136d05f8',1,'mlx::core::Log::is_equivalent()'],['../classmlx_1_1core_1_1_logical_not.html#aba53675da351cd9b71a73d475b4bbe99',1,'mlx::core::LogicalNot::is_equivalent()'],['../classmlx_1_1core_1_1_logical_and.html#a9572c35f72e0db2f7f86bbf42438a6be',1,'mlx::core::LogicalAnd::is_equivalent()'],['../classmlx_1_1core_1_1_logical_or.html#a9c8b10a5cf5c69fdc2362390197e4e71',1,'mlx::core::LogicalOr::is_equivalent()'],['../classmlx_1_1core_1_1_log_add_exp.html#a3cf9a202c05aff39919d713d6e2b32e4',1,'mlx::core::LogAddExp::is_equivalent()'],['../classmlx_1_1core_1_1_matmul.html#aab372b59eae0840fc4f75ef5719a2630',1,'mlx::core::Matmul::is_equivalent()'],['../classmlx_1_1core_1_1_maximum.html#a21fe93fbd7799682f481260aee8bdb46',1,'mlx::core::Maximum::is_equivalent()'],['../classmlx_1_1core_1_1_minimum.html#a56c54ee3293cc2cd84462b9ec7ac36b4',1,'mlx::core::Minimum::is_equivalent()'],['../classmlx_1_1core_1_1_multiply.html#ae288159fa2d6d35087a85aca8eafa9b2',1,'mlx::core::Multiply::is_equivalent()'],['../classmlx_1_1core_1_1_negative.html#ac2a4d8159c548639d6289980c8975823',1,'mlx::core::Negative::is_equivalent()'],['../classmlx_1_1core_1_1_not_equal.html#ac12fd6b3e2f2e7e4e622b59badf2c73d',1,'mlx::core::NotEqual::is_equivalent()'],['../classmlx_1_1core_1_1_number_of_elements.html#ad6a32565ccc64499e368e15bba0b438f',1,'mlx::core::NumberOfElements::is_equivalent()'],['../classmlx_1_1core_1_1_pad.html#aad7c3bfecafe435d6a8e807de4c7ea9b',1,'mlx::core::Pad::is_equivalent()'],['../classmlx_1_1core_1_1_partition.html#aabdf6ef4f2159b2bfe93e0e87d4772f8',1,'mlx::core::Partition::is_equivalent()'],['../classmlx_1_1core_1_1_power.html#a76b4ec9d1ff07f06189e414480453d68',1,'mlx::core::Power::is_equivalent()'],['../classmlx_1_1core_1_1_quantized_matmul.html#af28b36e3f40ea41785387800326cc8e1',1,'mlx::core::QuantizedMatmul::is_equivalent()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a6a7da6bcf657fcdb157c45bf35fdec11',1,'mlx::core::GatherQMM::is_equivalent()'],['../classmlx_1_1core_1_1_random_bits.html#a72ec915debf5823e7c0463045b2894e6',1,'mlx::core::RandomBits::is_equivalent()'],['../classmlx_1_1core_1_1_reshape.html#abd07c53af476777a04307e0423784cf3',1,'mlx::core::Reshape::is_equivalent()'],['../classmlx_1_1core_1_1_reduce.html#abe8f3327d617d0dd7438f066497ae08e',1,'mlx::core::Reduce::is_equivalent()'],['../classmlx_1_1core_1_1_round.html#aeb3d8607bbba7345a3142d4cbd4e6927',1,'mlx::core::Round::is_equivalent()'],['../classmlx_1_1core_1_1_scan.html#a54445a4d677ca4fe2a58d08eb5223ac6',1,'mlx::core::Scan::is_equivalent()'],['../classmlx_1_1core_1_1_scatter.html#a0208172562abdc90472e6eb5f84c987f',1,'mlx::core::Scatter::is_equivalent()'],['../classmlx_1_1core_1_1_sigmoid.html#a04814ba1b0edf8299d5ca1bcb8749d8e',1,'mlx::core::Sigmoid::is_equivalent()'],['../classmlx_1_1core_1_1_sign.html#a8c0934acbcc4b146e5aacd35a8c445bb',1,'mlx::core::Sign::is_equivalent()'],['../classmlx_1_1core_1_1_sin.html#af00b0e5516f884996ce7a97e6c1e3e6a',1,'mlx::core::Sin::is_equivalent()'],['../classmlx_1_1core_1_1_sinh.html#adcb1878996fd4902cd550042dd6ad70d',1,'mlx::core::Sinh::is_equivalent()'],['../classmlx_1_1core_1_1_slice.html#a43202c3b8966ae1db9ab82072e4918b0',1,'mlx::core::Slice::is_equivalent()'],['../classmlx_1_1core_1_1_slice_update.html#a60f588acced42391e6e5615ae8d16119',1,'mlx::core::SliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_softmax.html#a9215ed7bd36bc11276c58dfb9808d728',1,'mlx::core::Softmax::is_equivalent()'],['../classmlx_1_1core_1_1_sort.html#ae48f07cf641d54234fc4fb6529a33511',1,'mlx::core::Sort::is_equivalent()'],['../classmlx_1_1core_1_1_split.html#af25a0cc259573b9dce60d285eee18345',1,'mlx::core::Split::is_equivalent()'],['../classmlx_1_1core_1_1_square.html#a6abc881d44071019aa15481e5ea75ab2',1,'mlx::core::Square::is_equivalent()'],['../classmlx_1_1core_1_1_sqrt.html#ab871c2b8ab4a27a3f782a005d0e87c46',1,'mlx::core::Sqrt::is_equivalent()'],['../classmlx_1_1core_1_1_stop_gradient.html#a327539298b21d800d26482b94fce41b3',1,'mlx::core::StopGradient::is_equivalent()'],['../classmlx_1_1core_1_1_subtract.html#af1c05e1e3f703ba916d54f8ccbbd102b',1,'mlx::core::Subtract::is_equivalent()'],['../classmlx_1_1core_1_1_tan.html#afdf46288e7f60ea7f878688347dff7e4',1,'mlx::core::Tan::is_equivalent()'],['../classmlx_1_1core_1_1_tanh.html#a0692a1de2373b86eb394252ed4fecfda',1,'mlx::core::Tanh::is_equivalent()'],['../classmlx_1_1core_1_1_uniform.html#abb6048807a7c5b2e35a77e06a17f801b',1,'mlx::core::Uniform::is_equivalent()'],['../classmlx_1_1core_1_1_view.html#a7cb8403a96a47cb258caac4e3b850f64',1,'mlx::core::View::is_equivalent()'],['../classmlx_1_1core_1_1_transpose.html#a799ec3c3fa9f1b9e6177c755252a3eab',1,'mlx::core::Transpose::is_equivalent()']]], - ['is_5fopen_25',['is_open',['../classmlx_1_1core_1_1io_1_1_reader.html#a780f504058bd9c80cb3d105046a9f985',1,'mlx::core::io::Reader::is_open()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a85aa36bdb0dbfb8c5b6cfd955b03417a',1,'mlx::core::io::Writer::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a97ccdbd79886d9b95b32e5cd4741d150',1,'mlx::core::io::FileReader::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#ad5d2ee671a81700cb1658c41309d6676',1,'mlx::core::io::FileWriter::is_open()']]], - ['is_5fpower_5fof_5f2_26',['is_power_of_2',['../namespacemlx_1_1core.html#adacbc4526e8964b267a8ec3eb1bc1a32',1,'mlx::core']]], - ['is_5fready_27',['is_ready',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3',1,'pocketfft::detail::threading::latch']]], - ['is_5fsame_5fshape_28',['is_same_shape',['../namespacemlx_1_1core.html#ad4b664de4a4abd305827b30879b9da33',1,'mlx::core']]], - ['is_5fscalar_29',['is_scalar',['../namespacemlx_1_1core.html#a985c60929757190e0b4ec51f57c767d0',1,'mlx::core']]], - ['is_5fstatic_5fcast_30',['is_static_cast',['../namespacemlx_1_1core.html#afd9e740e567f9d7c28e00113caf46d5f',1,'mlx::core']]], - ['is_5ftracer_31',['is_tracer',['../classmlx_1_1core_1_1array.html#af9acb115019b995354d366c4ac6b968c',1,'mlx::core::array']]], - ['isclose_32',['isclose',['../group__ops.html#ga51eac95c04400921c54716de14b52491',1,'mlx::core']]], - ['isinf_33',['isinf',['../group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844',1,'mlx::core']]], - ['isnan_34',['isnan',['../namespacemetal.html#a83320ba983d90dd1fa5847b6940dc0bb',1,'metal::isnan()'],['../group__ops.html#ga175592792471b0ffb45196dca4711ba6',1,'mlx::core::isnan(const array &a, StreamOrDevice s={})']]], - ['isneginf_35',['isneginf',['../group__ops.html#ga1940523da381ed7be50656a3bc465ff3',1,'mlx::core']]], - ['isposinf_36',['isposinf',['../group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993',1,'mlx::core']]], - ['issubdtype_37',['issubdtype',['../namespacemlx_1_1core.html#aca9e69b06f4212eba44bf0ce6711d5f7',1,'mlx::core::issubdtype(const Dtype &a, const Dtype &b)'],['../namespacemlx_1_1core.html#aca39f224c1d17bde35dfcb9088430704',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype &b)'],['../namespacemlx_1_1core.html#ae9ee4a7c205df061c1caa7e62b7504e8',1,'mlx::core::issubdtype(const Dtype &a, const Dtype::Category &b)'],['../namespacemlx_1_1core.html#ab5b1a5a3d545a5de00c3117f76d71a1d',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype::Category &b)']]], - ['item_38',['item',['../classmlx_1_1core_1_1array.html#a90c5afddc2fa3028c0f8099bd64c8a99',1,'mlx::core::array::item()'],['../classmlx_1_1core_1_1array.html#a8650a99a6b7549bc823b03ad92590ff7',1,'mlx::core::array::item() const']]], - ['itemsize_39',['itemsize',['../classmlx_1_1core_1_1array.html#af329d9432c92de87cbaa2de8454eefc0',1,'mlx::core::array']]] + ['is_5fdonatable_22',['is_donatable',['../classmlx_1_1core_1_1array.html#a4677a404b5d191af20b52649225de087',1,'mlx::core::array']]], + ['is_5fequivalent_23',['is_equivalent',['../classmlx_1_1core_1_1fast_1_1_scaled_dot_product_attention.html#af08b1294f3f93505a96fdfa85b1edd62',1,'mlx::core::fast::ScaledDotProductAttention::is_equivalent()'],['../classmlx_1_1core_1_1_primitive.html#a6140a502af4c2bbbc776ab26e9afebcd',1,'mlx::core::Primitive::is_equivalent()'],['../classmlx_1_1core_1_1_abs.html#ab6f0ec56bc7c048382297e12dabadc67',1,'mlx::core::Abs::is_equivalent()'],['../classmlx_1_1core_1_1_add.html#aba0a35410c3aac53d0f7a0c283d9ee3f',1,'mlx::core::Add::is_equivalent()'],['../classmlx_1_1core_1_1_add_m_m.html#a6e37c6882dba995a63fb6d8dfb01754f',1,'mlx::core::AddMM::is_equivalent()'],['../classmlx_1_1core_1_1_arange.html#a7b6a45cf9c4b109d4e0373f3fe576c35',1,'mlx::core::Arange::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cos.html#a39557461e3235801886675a9b7d25bf5',1,'mlx::core::ArcCos::is_equivalent()'],['../classmlx_1_1core_1_1_arc_cosh.html#a6928e827b9ac2e86e7d5b02b78150eee',1,'mlx::core::ArcCosh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sin.html#a13b5e39eeccaf32d94b8eb85b3b753ab',1,'mlx::core::ArcSin::is_equivalent()'],['../classmlx_1_1core_1_1_arc_sinh.html#a63c7a765c7906242dc3371deec094f0f',1,'mlx::core::ArcSinh::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan.html#a0e5b5fc7218143ecd0a8666d9137c34c',1,'mlx::core::ArcTan::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tan2.html#aeaee58cd803d3ebf0b76574a409682cc',1,'mlx::core::ArcTan2::is_equivalent()'],['../classmlx_1_1core_1_1_arc_tanh.html#ac8ecdd640043dab0461d49d7650679a2',1,'mlx::core::ArcTanh::is_equivalent()'],['../classmlx_1_1core_1_1_arg_partition.html#ad87509ce70b51fb75dfb9c3a05a5b31a',1,'mlx::core::ArgPartition::is_equivalent()'],['../classmlx_1_1core_1_1_arg_reduce.html#a03b81a670dcb1e39bf7279e4d4583b97',1,'mlx::core::ArgReduce::is_equivalent()'],['../classmlx_1_1core_1_1_arg_sort.html#a048cd09c557d29d1111726f97010a845',1,'mlx::core::ArgSort::is_equivalent()'],['../classmlx_1_1core_1_1_as_type.html#a8e6c8b2428ab15c4fb43f2e3a8fb38af',1,'mlx::core::AsType::is_equivalent()'],['../classmlx_1_1core_1_1_as_strided.html#a1738c6aa0a3a3eb68530f0d5b436e094',1,'mlx::core::AsStrided::is_equivalent()'],['../classmlx_1_1core_1_1_bitwise_binary.html#a8cd6b916b4838a6c329cf4df8530c3b8',1,'mlx::core::BitwiseBinary::is_equivalent()'],['../classmlx_1_1core_1_1_block_masked_m_m.html#aef1c303955f9b8f445296372cf181160',1,'mlx::core::BlockMaskedMM::is_equivalent()'],['../classmlx_1_1core_1_1_gather_m_m.html#a163f17f6ce2c002f22e81b302777342b',1,'mlx::core::GatherMM::is_equivalent()'],['../classmlx_1_1core_1_1_broadcast.html#a0e27692b0090ec451954649a36042616',1,'mlx::core::Broadcast::is_equivalent()'],['../classmlx_1_1core_1_1_ceil.html#aacd90acb56eb0649c1cef807aa21df52',1,'mlx::core::Ceil::is_equivalent()'],['../classmlx_1_1core_1_1_compiled.html#a63e5016458887813b4a59dee5a0a3f10',1,'mlx::core::Compiled::is_equivalent()'],['../classmlx_1_1core_1_1_concatenate.html#aaf8a72a0c30114460caf519580cc35d2',1,'mlx::core::Concatenate::is_equivalent()'],['../classmlx_1_1core_1_1_conjugate.html#af42f00a790c6bc5572bd8fe9e5b36c5e',1,'mlx::core::Conjugate::is_equivalent()'],['../classmlx_1_1core_1_1_convolution.html#afb87708a5e3aab2e9e663daa9d8863de',1,'mlx::core::Convolution::is_equivalent()'],['../classmlx_1_1core_1_1_copy.html#afcfa39465015f638e294aa954ea0f3da',1,'mlx::core::Copy::is_equivalent()'],['../classmlx_1_1core_1_1_cos.html#ab611ca38c987915659f7ffcce0370417',1,'mlx::core::Cos::is_equivalent()'],['../classmlx_1_1core_1_1_cosh.html#ae0bacccaf501f5349db0c13cca776ff9',1,'mlx::core::Cosh::is_equivalent()'],['../classmlx_1_1core_1_1_divide.html#a3dda091f05c4164c29bb8129e9712650',1,'mlx::core::Divide::is_equivalent()'],['../classmlx_1_1core_1_1_div_mod.html#af5fcf8ec8515d46844cbeeab6dafb38a',1,'mlx::core::DivMod::is_equivalent()'],['../classmlx_1_1core_1_1_select.html#afc3c333fac7f902c98839921ef2874c8',1,'mlx::core::Select::is_equivalent()'],['../classmlx_1_1core_1_1_remainder.html#a802039faaa2ed7b763ec3d7debcce814',1,'mlx::core::Remainder::is_equivalent()'],['../classmlx_1_1core_1_1_equal.html#a58c1c5003e43f47dc0788c1851deaa02',1,'mlx::core::Equal::is_equivalent()'],['../classmlx_1_1core_1_1_erf.html#abe99dfbc2954c3a7d5dec56ab165ee82',1,'mlx::core::Erf::is_equivalent()'],['../classmlx_1_1core_1_1_erf_inv.html#aaac9e3b454ba564f9c6e804ab6562832',1,'mlx::core::ErfInv::is_equivalent()'],['../classmlx_1_1core_1_1_exp.html#ac6e44bffe7a643ab4ca51e74c7328357',1,'mlx::core::Exp::is_equivalent()'],['../classmlx_1_1core_1_1_f_f_t.html#a0ede3bc8b6d77d560c0a750b68fddc06',1,'mlx::core::FFT::is_equivalent()'],['../classmlx_1_1core_1_1_floor.html#a24b64feb026c4fcd02fc481cffdb1c94',1,'mlx::core::Floor::is_equivalent()'],['../classmlx_1_1core_1_1_full.html#afafcbcae1e28597fe8f7fde289105792',1,'mlx::core::Full::is_equivalent()'],['../classmlx_1_1core_1_1_gather.html#a23ff1406dbf0c770e75ad47440b467aa',1,'mlx::core::Gather::is_equivalent()'],['../classmlx_1_1core_1_1_greater.html#a6877a6888614a618dc64296763ccabb1',1,'mlx::core::Greater::is_equivalent()'],['../classmlx_1_1core_1_1_greater_equal.html#a3daef8596b963026b602019bc56fc5fc',1,'mlx::core::GreaterEqual::is_equivalent()'],['../classmlx_1_1core_1_1_hadamard.html#a8a528d8d69a7343bdfd704a3e74230b8',1,'mlx::core::Hadamard::is_equivalent()'],['../classmlx_1_1core_1_1_less.html#a7d6ed6353a0dcefebd008026dbd3cd63',1,'mlx::core::Less::is_equivalent()'],['../classmlx_1_1core_1_1_less_equal.html#a76ee1438cf4bd109eae4e0b3472b26af',1,'mlx::core::LessEqual::is_equivalent()'],['../classmlx_1_1core_1_1_log.html#a2fc58ea4ca744db493b947d1136d05f8',1,'mlx::core::Log::is_equivalent()'],['../classmlx_1_1core_1_1_logical_not.html#aba53675da351cd9b71a73d475b4bbe99',1,'mlx::core::LogicalNot::is_equivalent()'],['../classmlx_1_1core_1_1_logical_and.html#a9572c35f72e0db2f7f86bbf42438a6be',1,'mlx::core::LogicalAnd::is_equivalent()'],['../classmlx_1_1core_1_1_logical_or.html#a9c8b10a5cf5c69fdc2362390197e4e71',1,'mlx::core::LogicalOr::is_equivalent()'],['../classmlx_1_1core_1_1_log_add_exp.html#a3cf9a202c05aff39919d713d6e2b32e4',1,'mlx::core::LogAddExp::is_equivalent()'],['../classmlx_1_1core_1_1_matmul.html#aab372b59eae0840fc4f75ef5719a2630',1,'mlx::core::Matmul::is_equivalent()'],['../classmlx_1_1core_1_1_maximum.html#a21fe93fbd7799682f481260aee8bdb46',1,'mlx::core::Maximum::is_equivalent()'],['../classmlx_1_1core_1_1_minimum.html#a56c54ee3293cc2cd84462b9ec7ac36b4',1,'mlx::core::Minimum::is_equivalent()'],['../classmlx_1_1core_1_1_multiply.html#ae288159fa2d6d35087a85aca8eafa9b2',1,'mlx::core::Multiply::is_equivalent()'],['../classmlx_1_1core_1_1_negative.html#ac2a4d8159c548639d6289980c8975823',1,'mlx::core::Negative::is_equivalent()'],['../classmlx_1_1core_1_1_not_equal.html#ac12fd6b3e2f2e7e4e622b59badf2c73d',1,'mlx::core::NotEqual::is_equivalent()'],['../classmlx_1_1core_1_1_number_of_elements.html#ad6a32565ccc64499e368e15bba0b438f',1,'mlx::core::NumberOfElements::is_equivalent()'],['../classmlx_1_1core_1_1_pad.html#aad7c3bfecafe435d6a8e807de4c7ea9b',1,'mlx::core::Pad::is_equivalent()'],['../classmlx_1_1core_1_1_partition.html#aabdf6ef4f2159b2bfe93e0e87d4772f8',1,'mlx::core::Partition::is_equivalent()'],['../classmlx_1_1core_1_1_power.html#a76b4ec9d1ff07f06189e414480453d68',1,'mlx::core::Power::is_equivalent()'],['../classmlx_1_1core_1_1_quantized_matmul.html#af28b36e3f40ea41785387800326cc8e1',1,'mlx::core::QuantizedMatmul::is_equivalent()'],['../classmlx_1_1core_1_1_gather_q_m_m.html#a6a7da6bcf657fcdb157c45bf35fdec11',1,'mlx::core::GatherQMM::is_equivalent()'],['../classmlx_1_1core_1_1_random_bits.html#a72ec915debf5823e7c0463045b2894e6',1,'mlx::core::RandomBits::is_equivalent()'],['../classmlx_1_1core_1_1_reshape.html#abd07c53af476777a04307e0423784cf3',1,'mlx::core::Reshape::is_equivalent()'],['../classmlx_1_1core_1_1_reduce.html#abe8f3327d617d0dd7438f066497ae08e',1,'mlx::core::Reduce::is_equivalent()'],['../classmlx_1_1core_1_1_round.html#aeb3d8607bbba7345a3142d4cbd4e6927',1,'mlx::core::Round::is_equivalent()'],['../classmlx_1_1core_1_1_scan.html#a54445a4d677ca4fe2a58d08eb5223ac6',1,'mlx::core::Scan::is_equivalent()'],['../classmlx_1_1core_1_1_scatter.html#a0208172562abdc90472e6eb5f84c987f',1,'mlx::core::Scatter::is_equivalent()'],['../classmlx_1_1core_1_1_sigmoid.html#a04814ba1b0edf8299d5ca1bcb8749d8e',1,'mlx::core::Sigmoid::is_equivalent()'],['../classmlx_1_1core_1_1_sign.html#a8c0934acbcc4b146e5aacd35a8c445bb',1,'mlx::core::Sign::is_equivalent()'],['../classmlx_1_1core_1_1_sin.html#af00b0e5516f884996ce7a97e6c1e3e6a',1,'mlx::core::Sin::is_equivalent()'],['../classmlx_1_1core_1_1_sinh.html#adcb1878996fd4902cd550042dd6ad70d',1,'mlx::core::Sinh::is_equivalent()'],['../classmlx_1_1core_1_1_slice.html#a43202c3b8966ae1db9ab82072e4918b0',1,'mlx::core::Slice::is_equivalent()'],['../classmlx_1_1core_1_1_slice_update.html#a60f588acced42391e6e5615ae8d16119',1,'mlx::core::SliceUpdate::is_equivalent()'],['../classmlx_1_1core_1_1_softmax.html#a9215ed7bd36bc11276c58dfb9808d728',1,'mlx::core::Softmax::is_equivalent()'],['../classmlx_1_1core_1_1_sort.html#ae48f07cf641d54234fc4fb6529a33511',1,'mlx::core::Sort::is_equivalent()'],['../classmlx_1_1core_1_1_split.html#af25a0cc259573b9dce60d285eee18345',1,'mlx::core::Split::is_equivalent()'],['../classmlx_1_1core_1_1_square.html#a6abc881d44071019aa15481e5ea75ab2',1,'mlx::core::Square::is_equivalent()'],['../classmlx_1_1core_1_1_sqrt.html#ab871c2b8ab4a27a3f782a005d0e87c46',1,'mlx::core::Sqrt::is_equivalent()'],['../classmlx_1_1core_1_1_stop_gradient.html#a327539298b21d800d26482b94fce41b3',1,'mlx::core::StopGradient::is_equivalent()'],['../classmlx_1_1core_1_1_subtract.html#af1c05e1e3f703ba916d54f8ccbbd102b',1,'mlx::core::Subtract::is_equivalent()'],['../classmlx_1_1core_1_1_tan.html#afdf46288e7f60ea7f878688347dff7e4',1,'mlx::core::Tan::is_equivalent()'],['../classmlx_1_1core_1_1_tanh.html#a0692a1de2373b86eb394252ed4fecfda',1,'mlx::core::Tanh::is_equivalent()'],['../classmlx_1_1core_1_1_uniform.html#abb6048807a7c5b2e35a77e06a17f801b',1,'mlx::core::Uniform::is_equivalent()'],['../classmlx_1_1core_1_1_view.html#a7cb8403a96a47cb258caac4e3b850f64',1,'mlx::core::View::is_equivalent()'],['../classmlx_1_1core_1_1_transpose.html#a799ec3c3fa9f1b9e6177c755252a3eab',1,'mlx::core::Transpose::is_equivalent()']]], + ['is_5fopen_24',['is_open',['../classmlx_1_1core_1_1io_1_1_reader.html#a780f504058bd9c80cb3d105046a9f985',1,'mlx::core::io::Reader::is_open()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a85aa36bdb0dbfb8c5b6cfd955b03417a',1,'mlx::core::io::Writer::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a97ccdbd79886d9b95b32e5cd4741d150',1,'mlx::core::io::FileReader::is_open()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#ad5d2ee671a81700cb1658c41309d6676',1,'mlx::core::io::FileWriter::is_open()']]], + ['is_5fpower_5fof_5f2_25',['is_power_of_2',['../namespacemlx_1_1core.html#adacbc4526e8964b267a8ec3eb1bc1a32',1,'mlx::core']]], + ['is_5fready_26',['is_ready',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3',1,'pocketfft::detail::threading::latch']]], + ['is_5fsame_5fshape_27',['is_same_shape',['../namespacemlx_1_1core.html#ad4b664de4a4abd305827b30879b9da33',1,'mlx::core']]], + ['is_5fscalar_28',['is_scalar',['../namespacemlx_1_1core.html#a985c60929757190e0b4ec51f57c767d0',1,'mlx::core']]], + ['is_5fstatic_5fcast_29',['is_static_cast',['../namespacemlx_1_1core.html#afd9e740e567f9d7c28e00113caf46d5f',1,'mlx::core']]], + ['is_5ftracer_30',['is_tracer',['../classmlx_1_1core_1_1array.html#af9acb115019b995354d366c4ac6b968c',1,'mlx::core::array']]], + ['isclose_31',['isclose',['../group__ops.html#ga51eac95c04400921c54716de14b52491',1,'mlx::core']]], + ['isinf_32',['isinf',['../group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844',1,'mlx::core']]], + ['isnan_33',['isnan',['../namespacemetal.html#a83320ba983d90dd1fa5847b6940dc0bb',1,'metal::isnan()'],['../group__ops.html#ga175592792471b0ffb45196dca4711ba6',1,'mlx::core::isnan(const array &a, StreamOrDevice s={})']]], + ['isneginf_34',['isneginf',['../group__ops.html#ga1940523da381ed7be50656a3bc465ff3',1,'mlx::core']]], + ['isposinf_35',['isposinf',['../group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993',1,'mlx::core']]], + ['issubdtype_36',['issubdtype',['../namespacemlx_1_1core.html#aca9e69b06f4212eba44bf0ce6711d5f7',1,'mlx::core::issubdtype(const Dtype &a, const Dtype &b)'],['../namespacemlx_1_1core.html#aca39f224c1d17bde35dfcb9088430704',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype &b)'],['../namespacemlx_1_1core.html#ae9ee4a7c205df061c1caa7e62b7504e8',1,'mlx::core::issubdtype(const Dtype &a, const Dtype::Category &b)'],['../namespacemlx_1_1core.html#ab5b1a5a3d545a5de00c3117f76d71a1d',1,'mlx::core::issubdtype(const Dtype::Category &a, const Dtype::Category &b)']]], + ['item_37',['item',['../classmlx_1_1core_1_1array.html#a90c5afddc2fa3028c0f8099bd64c8a99',1,'mlx::core::array::item()'],['../classmlx_1_1core_1_1array.html#a8650a99a6b7549bc823b03ad92590ff7',1,'mlx::core::array::item() const']]], + ['itemsize_38',['itemsize',['../classmlx_1_1core_1_1array.html#af329d9432c92de87cbaa2de8454eefc0',1,'mlx::core::array']]] ]; diff --git a/docs/build/html/search/functions_c.js b/docs/build/html/search/functions_c.js index 08b9da6e46..1abc1aba22 100644 --- a/docs/build/html/search/functions_c.js +++ b/docs/build/html/search/functions_c.js @@ -1,6 +1,6 @@ var searchData= [ - ['label_0',['label',['../class_m_p_s_1_1_kernel.html#aaf646bce46266bc11c419464ec72cdfe',1,'MPS::Kernel::label()'],['../classmlx_1_1core_1_1io_1_1_reader.html#a8c244daf52fd5ebb9a2c7e5f4ae793cd',1,'mlx::core::io::Reader::label()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a828125a9adcb7e90c8bcaba0fe47f854',1,'mlx::core::io::Writer::label()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a7415da4cb14c328ed07901ff1c266221',1,'mlx::core::io::FileReader::label()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a5093dce80ff0c51ea036a87e3e5fb456',1,'mlx::core::io::FileWriter::label()']]], + ['label_0',['label',['../classmlx_1_1core_1_1io_1_1_reader.html#a8c244daf52fd5ebb9a2c7e5f4ae793cd',1,'mlx::core::io::Reader::label()'],['../classmlx_1_1core_1_1io_1_1_writer.html#a828125a9adcb7e90c8bcaba0fe47f854',1,'mlx::core::io::Writer::label()'],['../classmlx_1_1core_1_1io_1_1_file_reader.html#a7415da4cb14c328ed07901ff1c266221',1,'mlx::core::io::FileReader::label()'],['../classmlx_1_1core_1_1io_1_1_file_writer.html#a5093dce80ff0c51ea036a87e3e5fb456',1,'mlx::core::io::FileWriter::label()']]], ['laplace_1',['laplace',['../namespacemlx_1_1core_1_1random.html#ad52c7cb7709542c5a464df12bfc6b0b7',1,'mlx::core::random::laplace(const std::vector< int > &shape, Dtype dtype, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a0d0c393dd53d6fdd02e46844b3cf45d1',1,'mlx::core::random::laplace(const std::vector< int > &shape, const float loc, const float scale, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a0f73eb7d337529708937eef0080c0998',1,'mlx::core::random::laplace(const std::vector< int > &shape, const Dtype dtype, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})'],['../namespacemlx_1_1core_1_1random.html#a96b6f40efbee7045a35218296ba12a37',1,'mlx::core::random::laplace(const std::vector< int > &shape, const std::optional< array > &key=std::nullopt, StreamOrDevice s={})']]], ['largest_5fprime_5ffactor_2',['largest_prime_factor',['../structpocketfft_1_1detail_1_1util.html#a0e9b92280f25063d1ebef942dd0d233c',1,'pocketfft::detail::util']]], ['latch_3',['latch',['../classpocketfft_1_1detail_1_1threading_1_1latch.html#a9260c871bb927525f7f53caa23a19c15',1,'pocketfft::detail::threading::latch']]], @@ -23,10 +23,10 @@ var searchData= ['load_20',['load',['../namespacemlx_1_1core.html#abada9bfa834d7423959362386720f3db',1,'mlx::core::load(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#ac71a08bf4c052ae3c77e9e89cbea071d',1,'mlx::core::load(std::string file, StreamOrDevice s={})']]], ['load_5fgguf_21',['load_gguf',['../namespacemlx_1_1core.html#a2aa12b351ce559deb14cda0a5292c2ce',1,'mlx::core']]], ['load_5fpadded_22',['load_padded',['../struct_read_writer.html#add5bd3f647793a5a19d63197a19df73c',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#af3ce6bbb1a8dfb3bab1ae18d3eb45bc0',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const'],['../struct_read_writer.html#ab116f4569bb9dc6eaef0d8d08472e239',1,'ReadWriter::load_padded(int length, const device float2 *w_k) const']]], - ['load_5fsafe_23',['load_safe',['../struct_quantized_block_loader.html#a699dc9aa284b8fbf870310bbb224465b',1,'QuantizedBlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../kernels_2scan_8h.html#ae8eb101e538b85f8a4bcf451489ae0ac',1,'load_safe(): scan.h']]], + ['load_5fsafe_23',['load_safe',['../struct_g_e_m_v_kernel.html#a04bb72da9a93d6d1eba468fa311bbba7',1,'GEMVKernel::load_safe()'],['../struct_quantized_block_loader.html#a699dc9aa284b8fbf870310bbb224465b',1,'QuantizedBlockLoader::load_safe()'],['../structmlx_1_1steel_1_1_block_loader.html#abb0f4f66ec8b123627beb8eb4fbb609d',1,'mlx::steel::BlockLoader::load_safe()'],['../kernels_2scan_8h.html#ae8eb101e538b85f8a4bcf451489ae0ac',1,'load_safe(): scan.h']]], ['load_5fsafetensors_24',['load_safetensors',['../namespacemlx_1_1core.html#a96cc40e1af8c4626c813ce4859f70a5c',1,'mlx::core::load_safetensors(std::shared_ptr< io::Reader > in_stream, StreamOrDevice s={})'],['../namespacemlx_1_1core.html#af7eea1682a38d363c56a066321e6d526',1,'mlx::core::load_safetensors(const std::string &file, StreamOrDevice s={})']]], ['load_5fstrided_25',['load_strided',['../struct_read_writer.html#a998ef484bade81f726b9edfc6b878197',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a3d9c8cbc582cad6b5218339d0f721559',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a795a71a8e1f154a5af415ebe1b3f0713',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a0935b946b8bf2e769427fcbf2da2f7be',1,'ReadWriter::load_strided(int stride, int overall_n)'],['../struct_read_writer.html#a7d45368c74a8b7c632659504b3273a13',1,'ReadWriter::load_strided(int stride, int overall_n)']]], - ['load_5funsafe_26',['load_unsafe',['../struct_quantized_block_loader.html#a86009527cb4b53e4c21fd6b1f78cfefc',1,'QuantizedBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a961836be363409744e48e595d5e0c2ec',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8034abc10483487fc94313e3674d1111',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a69e2f7c9814d1cc1c5c267be8618dc55',1,'mlx::steel::Conv2DWeightBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#aa11d1a142bc868df462f48a7102147f3',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a0e262b003ac0e7ee6272585eac921704',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3859ca11b5991ef6ee9b99afdc3ea30a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8f078982186421f5b484c0b53af9c655',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader.html#a6c9e27f11f48b34580ed2c7e9cad9a27',1,'mlx::steel::BlockLoader::load_unsafe()'],['../kernels_2scan_8h.html#a9c415d07921f3961bad0a00a34f4a9a3',1,'load_unsafe(U values[N_READS], const device T *input): scan.h']]], + ['load_5funsafe_26',['load_unsafe',['../struct_g_e_m_v_kernel.html#a6013e9c5b2f72fa1311dd038172df0ce',1,'GEMVKernel::load_unsafe()'],['../struct_quantized_block_loader.html#a86009527cb4b53e4c21fd6b1f78cfefc',1,'QuantizedBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a961836be363409744e48e595d5e0c2ec',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a8034abc10483487fc94313e3674d1111',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a69e2f7c9814d1cc1c5c267be8618dc55',1,'mlx::steel::Conv2DWeightBlockLoader::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#aa11d1a142bc868df462f48a7102147f3',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a0e262b003ac0e7ee6272585eac921704',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a3859ca11b5991ef6ee9b99afdc3ea30a',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a8f078982186421f5b484c0b53af9c655',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::load_unsafe()'],['../structmlx_1_1steel_1_1_block_loader.html#a6c9e27f11f48b34580ed2c7e9cad9a27',1,'mlx::steel::BlockLoader::load_unsafe()'],['../kernels_2scan_8h.html#a9c415d07921f3961bad0a00a34f4a9a3',1,'load_unsafe(U values[N_READS], const device T *input): scan.h']]], ['load_5fvector_27',['load_vector',['../quantized_8h.html#a8dbace41de9e1e21dd59d016db11b3e9',1,'quantized.h']]], ['load_5fvector_5fsafe_28',['load_vector_safe',['../quantized_8h.html#aa69e143d646fad332c1a53e8c9b337b7',1,'quantized.h']]], ['log_29',['Log',['../classmlx_1_1core_1_1_log.html#a663e54790c60b56eb0ff09f4f6635fb9',1,'mlx::core::Log']]], diff --git a/docs/build/html/search/functions_d.js b/docs/build/html/search/functions_d.js index 94d49d6c75..9b55f4c1e1 100644 --- a/docs/build/html/search/functions_d.js +++ b/docs/build/html/search/functions_d.js @@ -2,13 +2,13 @@ var searchData= [ ['make_5farrays_0',['make_arrays',['../classmlx_1_1core_1_1array.html#a1173db4e23f5a8230911cb8fba45d5e6',1,'mlx::core::array']]], ['make_5fcontiguous_5fstrides_1',['make_contiguous_strides',['../namespacemlx_1_1core.html#ac9fb1286a1a00395e901dbff80560895',1,'mlx::core']]], - ['make_5fsynchronize_5ftask_2',['make_synchronize_task',['../namespacemlx_1_1core_1_1metal.html#ab31abdda3052162d59f6590a89e38337',1,'mlx::core::metal']]], - ['make_5ftask_3',['make_task',['../namespacemlx_1_1core_1_1metal.html#a4552b7ccdfa7f3cc9895c09799d8048e',1,'mlx::core::metal']]], - ['malloc_4',['malloc',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a9a17d2c7a97772bf4a15e6c74af34ca4',1,'mlx::core::allocator::Allocator::malloc()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#a4f3d5de6b8c0eba22e9403b28a5ef3f0',1,'mlx::core::allocator::CommonAllocator::malloc()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a6c0feb9b1ff9977f76c69745393944bc',1,'mlx::core::metal::MetalAllocator::malloc()'],['../namespacemlx_1_1core_1_1allocator.html#a560d10a166e3c294f3757166f9bd6801',1,'mlx::core::allocator::malloc(size_t size)']]], - ['malloc_5for_5fwait_5',['malloc_or_wait',['../namespacemlx_1_1core_1_1allocator.html#a86ac0a11ff78f21e717f641716c34abc',1,'mlx::core::allocator']]], - ['matmul_6',['Matmul',['../classmlx_1_1core_1_1_matmul.html#adef92f30ab35e540ccb316ea6b94e6f7',1,'mlx::core::Matmul']]], - ['matmul_7',['matmul',['../group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a',1,'mlx::core']]], - ['matrixdescriptor_8',['matrixDescriptor',['../class_m_p_s_1_1_matrix_descriptor.html#a7a99042b3c8afa037719ca316500a37b',1,'MPS::MatrixDescriptor::matrixDescriptor(NS::UInteger rows, NS::UInteger columns, NS::UInteger rowBytes, NS::UInteger dataType)'],['../class_m_p_s_1_1_matrix_descriptor.html#a67e5df7d48fc175efd5124b3b43b6366',1,'MPS::MatrixDescriptor::matrixDescriptor(NS::UInteger rows, NS::UInteger columns, NS::UInteger matrices, NS::UInteger rowBytes, NS::UInteger matrixBytes, NS::UInteger dataType)']]], + ['make_5fstring_2',['make_string',['../namespacemlx_1_1core.html#aed148d95e7b5221f1312473deded0d27',1,'mlx::core']]], + ['make_5fsynchronize_5ftask_3',['make_synchronize_task',['../namespacemlx_1_1core_1_1metal.html#ab31abdda3052162d59f6590a89e38337',1,'mlx::core::metal']]], + ['make_5ftask_4',['make_task',['../namespacemlx_1_1core_1_1metal.html#a4552b7ccdfa7f3cc9895c09799d8048e',1,'mlx::core::metal']]], + ['malloc_5',['malloc',['../classmlx_1_1core_1_1allocator_1_1_allocator.html#a9a17d2c7a97772bf4a15e6c74af34ca4',1,'mlx::core::allocator::Allocator::malloc()'],['../classmlx_1_1core_1_1allocator_1_1_common_allocator.html#a4f3d5de6b8c0eba22e9403b28a5ef3f0',1,'mlx::core::allocator::CommonAllocator::malloc()'],['../classmlx_1_1core_1_1metal_1_1_metal_allocator.html#a6c0feb9b1ff9977f76c69745393944bc',1,'mlx::core::metal::MetalAllocator::malloc()'],['../namespacemlx_1_1core_1_1allocator.html#a560d10a166e3c294f3757166f9bd6801',1,'mlx::core::allocator::malloc(size_t size)']]], + ['malloc_5for_5fwait_6',['malloc_or_wait',['../namespacemlx_1_1core_1_1allocator.html#a86ac0a11ff78f21e717f641716c34abc',1,'mlx::core::allocator']]], + ['matmul_7',['Matmul',['../classmlx_1_1core_1_1_matmul.html#adef92f30ab35e540ccb316ea6b94e6f7',1,'mlx::core::Matmul']]], + ['matmul_8',['matmul',['../group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a',1,'mlx::core']]], ['max_9',['max',['../structmetal_1_1__numeric__limits__impl_3_01bfloat16__t_01_4.html#a92320d40a58218e40cc414986ac95c50',1,'metal::_numeric_limits_impl< bfloat16_t >::max()'],['../namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b',1,'metal::max()'],['../namespacemetal_1_1fast.html#a747e2e58092a27fb8b4dd3d16934fb52',1,'metal::fast::max()'],['../namespacemetal_1_1precise.html#a6a954a4e4e3753303d1dc734855a185f',1,'metal::precise::max()'],['../group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7',1,'mlx::core::max(const array &a, bool keepdims, StreamOrDevice s={})'],['../group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc',1,'mlx::core::max(const array &a, StreamOrDevice s={})'],['../group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f',1,'mlx::core::max(const array &a, const std::vector< int > &axes, bool keepdims=false, StreamOrDevice s={})'],['../group__ops.html#ga7b638050e03a93f2896c981bc2850a47',1,'mlx::core::max(const array &a, int axis, bool keepdims=false, StreamOrDevice s={})']]], ['max3_10',['max3',['../namespacemetal.html#a00f9c0ad66d969794614f56912eed9c9',1,'metal::max3()'],['../namespacemetal_1_1fast.html#a6fc2cf18ffa8149561864c86dba0f803',1,'metal::fast::max3()'],['../namespacemetal_1_1precise.html#ac490e8614ebd2c9343af1ae6c0d4e82c',1,'metal::precise::max3()']]], ['maximum_11',['Maximum',['../classmlx_1_1core_1_1_maximum.html#a28389307e385efe1b2955b86b115e816',1,'mlx::core::Maximum']]], diff --git a/docs/build/html/search/functions_f.js b/docs/build/html/search/functions_f.js index 5bafadf957..b229f47934 100644 --- a/docs/build/html/search/functions_f.js +++ b/docs/build/html/search/functions_f.js @@ -5,7 +5,7 @@ var searchData= ['ones_2',['ones',['../group__ops.html#ga54eeed455321a54c8e72e16552a978f2',1,'mlx::core::ones(const std::vector< int > &shape, Dtype dtype, StreamOrDevice s={})'],['../group__ops.html#ga6cf4b5e8580e4436302c519d05897dab',1,'mlx::core::ones(const std::vector< int > &shape, StreamOrDevice s={})']]], ['ones_5flike_3',['ones_like',['../group__ops.html#ga94f8d3b1906fee99da9cbe39f7be7d42',1,'mlx::core']]], ['oofs_4',['oofs',['../classpocketfft_1_1detail_1_1multi__iter.html#aae63e67caac095d474ddd32daa5ffa34',1,'pocketfft::detail::multi_iter::oofs(size_t i) const'],['../classpocketfft_1_1detail_1_1multi__iter.html#a9236047e7419e5d21379cbf95eb3a78e',1,'pocketfft::detail::multi_iter::oofs(size_t j, size_t i) const']]], - ['operator_20bool_5',['operator bool',['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant']]], + ['operator_20bool_5',['operator bool',['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant'],['../struct___no_mask.html#ad3723c1e70e46beefd283ce6317416cb',1,'_NoMask::operator bool()'],['../struct___no_mask.html#aafbf8a3201e1cc1abf74dd1f1b7272cd',1,'_NoMask::operator bool() const threadgroup'],['../struct___no_mask.html#a73e9612a619885cbc97cbd8f40df71e7',1,'_NoMask::operator bool() const device'],['../struct___no_mask.html#a4bf336d472bc677028250f76b9cdc08c',1,'_NoMask::operator bool() const constant']]], ['operator_20dtype_6',['operator Dtype',['../structmlx_1_1core_1_1_type_to_dtype.html#aefdd0fd6a5bbf0197a3996ccd4adea13',1,'mlx::core::TypeToDtype']]], ['operator_20float_7',['operator float',['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aaae72e5340ce91325f1925be36ba46cb',1,'mlx::core::_MLX_BFloat16::operator float()'],['../structmlx_1_1core_1_1complex128__t.html#a3e2faf180c0b785646a0e4296f709a5e',1,'mlx::core::complex128_t::operator float()'],['../structmlx_1_1core_1_1complex64__t.html#a90d224dd37308345086bb9cc882ef6fc',1,'mlx::core::complex64_t::operator float()'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a363de5054f3673bddc90293fc3c9bb99',1,'mlx::core::_MLX_Float16::operator float()']]], ['operator_20t_8',['operator T',['../struct___m_l_x___b_float16.html#aa7dfefdf0d15e102d2b8258c9ab01836',1,'_MLX_BFloat16::operator T() const thread'],['../struct___m_l_x___b_float16.html#a2546a8afa77e14ed5b3c5da79a281260',1,'_MLX_BFloat16::operator T() const threadgroup'],['../struct___m_l_x___b_float16.html#a1d523f87740fcb852db6ab57896c245a',1,'_MLX_BFloat16::operator T() const device'],['../struct___m_l_x___b_float16.html#a95acd29283024d7093a0bc58c9468a0a',1,'_MLX_BFloat16::operator T() const constant'],['../structcomplex64__t.html#a70e9b16031eeaff3baa601f400023fcd',1,'complex64_t::operator T() const thread'],['../structcomplex64__t.html#a4f3beea7ab6001189b782a74d1746b67',1,'complex64_t::operator T() const threadgroup'],['../structcomplex64__t.html#a9f4f7eca89ffe6c8d126a4145df6d9f2',1,'complex64_t::operator T() const device'],['../structcomplex64__t.html#ac33e2e5263fec76a4fb4418c6e1d8d14',1,'complex64_t::operator T() const constant']]], diff --git a/docs/build/html/search/namespaces_0.js b/docs/build/html/search/namespaces_0.js index 7390c5c94d..b24736856c 100644 --- a/docs/build/html/search/namespaces_0.js +++ b/docs/build/html/search/namespaces_0.js @@ -16,10 +16,5 @@ var searchData= ['mlx_3a_3acore_3a_3ametal_13',['metal',['../namespacemlx_1_1core_1_1metal.html',1,'mlx::core']]], ['mlx_3a_3acore_3a_3arandom_14',['random',['../namespacemlx_1_1core_1_1random.html',1,'mlx::core']]], ['mlx_3a_3acore_3a_3ascheduler_15',['scheduler',['../namespacemlx_1_1core_1_1scheduler.html',1,'mlx::core']]], - ['mlx_3a_3asteel_16',['steel',['../namespacemlx_1_1steel.html',1,'mlx']]], - ['mps_17',['MPS',['../namespace_m_p_s.html',1,'']]], - ['mtl_18',['MTL',['../namespace_m_t_l.html',1,'']]], - ['mtl_3a_3aprivate_19',['Private',['../namespace_m_t_l_1_1_private.html',1,'MTL']]], - ['mtl_3a_3aprivate_3a_3aclass_20',['Class',['../namespace_m_t_l_1_1_private_1_1_class.html',1,'MTL::Private']]], - ['mtl_3a_3aprivate_3a_3aselector_21',['Selector',['../namespace_m_t_l_1_1_private_1_1_selector.html',1,'MTL::Private']]] + ['mlx_3a_3asteel_16',['steel',['../namespacemlx_1_1steel.html',1,'mlx']]] ]; diff --git a/docs/build/html/search/typedefs_2.js b/docs/build/html/search/typedefs_2.js index 013d4a2360..8bf5190141 100644 --- a/docs/build/html/search/typedefs_2.js +++ b/docs/build/html/search/typedefs_2.js @@ -1,6 +1,5 @@ var searchData= [ - ['datatype_0',['DataType',['../namespace_m_p_s.html#a96effe6e5c9d287c6b86491509c78be2',1,'MPS']]], - ['deleter_5ft_1',['deleter_t',['../namespacemlx_1_1core.html#a1e6cec03ebd80fd2d6b12b288367bfa8',1,'mlx::core']]], - ['difference_5ftype_2',['difference_type',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#adcee44c77980fc2370a2c31e203aead5',1,'mlx::core::array::ArrayIterator']]] + ['deleter_5ft_0',['deleter_t',['../namespacemlx_1_1core.html#a1e6cec03ebd80fd2d6b12b288367bfa8',1,'mlx::core']]], + ['difference_5ftype_1',['difference_type',['../structmlx_1_1core_1_1array_1_1_array_iterator.html#adcee44c77980fc2370a2c31e203aead5',1,'mlx::core::array::ArrayIterator']]] ]; diff --git a/docs/build/html/search/typedefs_9.js b/docs/build/html/search/typedefs_9.js index 7ff5deb198..cb3ea6733e 100644 --- a/docs/build/html/search/typedefs_9.js +++ b/docs/build/html/search/typedefs_9.js @@ -1,4 +1,4 @@ var searchData= [ - ['nomask_5ft_0',['nomask_t',['../steel__gemm__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'steel_gemm_masked.h']]] + ['nomask_5ft_0',['nomask_t',['../kernels_2gemv__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'nomask_t: gemv_masked.h'],['../steel__gemm__masked_8h.html#a1480c8cdff1cae1462a5a71632969bca',1,'nomask_t: steel_gemm_masked.h']]] ]; diff --git a/docs/build/html/search/variables_1.js b/docs/build/html/search/variables_1.js index afa75c6235..ea5e5fa5e9 100644 --- a/docs/build/html/search/variables_1.js +++ b/docs/build/html/search/variables_1.js @@ -22,11 +22,13 @@ var searchData= ['biases_19',['biases',['../struct_quantized_block_loader.html#a17d01a6aba0833b073586ef2c09d0fbd',1,'QuantizedBlockLoader']]], ['bits_5f_20',['bits_',['../struct___m_l_x___b_float16.html#a4113263b63e3757ea8334cc4f0f5c3c8',1,'_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___b_float16.html#aca48963f820065c3d8ecab24265ab3fc',1,'mlx::core::_MLX_BFloat16::bits_'],['../structmlx_1_1core_1_1___m_l_x___float16.html#a5203fe52424fd32bce6eb7917dd9288b',1,'mlx::core::_MLX_Float16::bits_']]], ['bj_21',['bj',['../struct_quantized_block_loader.html#ae2add92b2aaf3414e91f0470b9b0cc00',1,'QuantizedBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a7ae9e41f50c0c63c35b63086a1c22cc3',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a6fd3dd7b74d91609fa9dd61c657a0e32',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a6f2fdcaf5a67567cca38ae3d8120ab37',1,'mlx::steel::Conv2DWeightBlockLoader::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a7cf448573d41fbc67f8dfc65b7aef2b2',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#adaa261fc2e8e694aedab4ebd60b52e5e',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::bj'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#ace16704025bc6e6204c306a357f3a8b8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acec010e10d5733654963407af38d4f67',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::bj'],['../structmlx_1_1steel_1_1_block_loader.html#a78c326e75ee35a484685771143047cd4',1,'mlx::steel::BlockLoader::bj']]], - ['bool_5f_22',['bool_',['../namespacemlx_1_1core.html#a113d2bac7e4aa6a4cb4a5c3242527b82',1,'mlx::core']]], - ['brows_23',['BROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ac070c6bd5be85b1ae805e18890db4fd4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a10591ea957605a9c662f93d59ff3410d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae9b86b05b23153ea1abaeead456c491c',1,'mlx::steel::Conv2DWeightBlockLoader::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a343984fb74ec579a4404278dbbc7e7b5',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acc8140aae84694f62e6324dbb6a614a4',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aba1e1c8012e4e50f0e9bcfb9486c1781',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a015a0c56de74a0c4d51953a7e94fbba8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BROWS']]], - ['bs_5foffset_24',['Bs_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92f6aeee432f53638447eac842f43eca',1,'mlx::steel::BlockMMA']]], - ['bsimd_25',['Bsimd',['../structmlx_1_1steel_1_1_block_m_m_a.html#a3de506004b428d3487bae05ad9e757e6',1,'mlx::steel::BlockMMA']]], - ['buf_26',['buf',['../struct_read_writer.html#a23bac3c96dd0265ddbee1f256be45ff5',1,'ReadWriter::buf'],['../backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697',1,'buf: allocator.h']]], - ['buffer_27',['buffer',['../structmlx_1_1core_1_1array_1_1_data.html#a9a51e2d12ba505027cc0fca86bdd39ad',1,'mlx::core::array::Data']]], - ['buffers_28',['buffers',['../struct_indices.html#ad705070a740579c07d109ae4f3d86e76',1,'Indices']]] + ['blockm_22',['blockM',['../struct_g_e_m_v_kernel.html#a7281520100658811076400060663903c',1,'GEMVKernel::blockM'],['../struct_g_e_m_v_t_kernel.html#a2ae8ce535d59cccf453381b4485a77f0',1,'GEMVTKernel::blockM']]], + ['blockn_23',['blockN',['../struct_g_e_m_v_kernel.html#a2fef17f9c9aa0bdf530ad3554fb0988b',1,'GEMVKernel::blockN'],['../struct_g_e_m_v_t_kernel.html#a60be87666006ba0bf88bc8e6902da42a',1,'GEMVTKernel::blockN']]], + ['bool_5f_24',['bool_',['../namespacemlx_1_1core.html#a113d2bac7e4aa6a4cb4a5c3242527b82',1,'mlx::core']]], + ['brows_25',['BROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ac070c6bd5be85b1ae805e18890db4fd4',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a10591ea957605a9c662f93d59ff3410d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ae9b86b05b23153ea1abaeead456c491c',1,'mlx::steel::Conv2DWeightBlockLoader::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a343984fb74ec579a4404278dbbc7e7b5',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acc8140aae84694f62e6324dbb6a614a4',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#aba1e1c8012e4e50f0e9bcfb9486c1781',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::BROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a015a0c56de74a0c4d51953a7e94fbba8',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::BROWS']]], + ['bs_5foffset_26',['Bs_offset',['../structmlx_1_1steel_1_1_block_m_m_a.html#a92f6aeee432f53638447eac842f43eca',1,'mlx::steel::BlockMMA']]], + ['bsimd_27',['Bsimd',['../structmlx_1_1steel_1_1_block_m_m_a.html#a3de506004b428d3487bae05ad9e757e6',1,'mlx::steel::BlockMMA']]], + ['buf_28',['buf',['../struct_read_writer.html#a23bac3c96dd0265ddbee1f256be45ff5',1,'ReadWriter::buf'],['../backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697',1,'buf: allocator.h']]], + ['buffer_29',['buffer',['../structmlx_1_1core_1_1array_1_1_data.html#a9a51e2d12ba505027cc0fca86bdd39ad',1,'mlx::core::array::Data']]], + ['buffers_30',['buffers',['../struct_indices.html#ad705070a740579c07d109ae4f3d86e76',1,'Indices']]] ]; diff --git a/docs/build/html/search/variables_13.js b/docs/build/html/search/variables_13.js index 665b14e568..1500be8a1c 100644 --- a/docs/build/html/search/variables_13.js +++ b/docs/build/html/search/variables_13.js @@ -1,7 +1,7 @@ var searchData= [ ['tcols_0',['TCOLS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a0b5303f3258e0a21862dead8e3f5401e',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a5adbd51e9adb6f7853724d83de4ff755',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a80cb90674f839d5d4ecfde384fa0a7a2',1,'mlx::steel::Conv2DWeightBlockLoader::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#ad2508cd5cdb51b2f611057e743b8fc6f',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acd54132d0928d0f6fb15b2f367e5d5e8',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#ae25c676b7318d78462ee89bcd80dc805',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::TCOLS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aff021a6fae860b4ac01fb593b2720457',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::TCOLS']]], - ['tgp_5fmem_5fsize_1',['tgp_mem_size',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a1ec583584e69dcbbb72106390a4fc5da',1,'mlx::steel::GEMMKernel']]], + ['tgp_5fmem_5fsize_1',['tgp_mem_size',['../struct_g_e_m_v_kernel.html#a9ef4d0e62094d7033069f5dda5efb236',1,'GEMVKernel::tgp_mem_size'],['../struct_g_e_m_v_t_kernel.html#a48a09a21d7b822f380d040c752b785d7',1,'GEMVTKernel::tgp_mem_size'],['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a1ec583584e69dcbbb72106390a4fc5da',1,'mlx::steel::GEMMKernel::tgp_mem_size']]], ['tgp_5fmem_5fsize_5fa_2',['tgp_mem_size_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ac00b149d76a903c2f91b0f477dc5037f',1,'mlx::steel::GEMMKernel']]], ['tgp_5fmem_5fsize_5fb_3',['tgp_mem_size_b',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#a105af1069668028c6f1bc6d6dd162298',1,'mlx::steel::GEMMKernel']]], ['tgp_5fpadding_5fa_4',['tgp_padding_a',['../structmlx_1_1steel_1_1_g_e_m_m_kernel.html#ad547704ccbff6c2076abeffa6628c5a0',1,'mlx::steel::GEMMKernel']]], @@ -10,17 +10,19 @@ var searchData= ['thread_7',['thread',['../structmlx_1_1core_1_1scheduler_1_1_stream_thread.html#a449de02bf2ac80d8fe2f208fa7eac359',1,'mlx::core::scheduler::StreamThread']]], ['thread_5fidx_8',['thread_idx',['../struct_quantized_block_loader.html#a50821537ea747bc03295a09bb0eef475',1,'QuantizedBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a70da26a715135d973f88371a70255be9',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#ac18de37cde1459595bfe18b0d5ef146d',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ab1cb2ade639787243e0325dcd3dc0a11',1,'mlx::steel::Conv2DWeightBlockLoader::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a9642399b8066e29123524f36ebc7b482',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acacdac168004c87fee27c8554ac905a7',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a401f0c7cf1588552556603c7ffba2316',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::thread_idx'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#a08a517bc50caf41155b98be0690bfe44',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::thread_idx'],['../structmlx_1_1steel_1_1_block_loader.html#a064e2cc77e0b1cf0f8027929e031775b',1,'mlx::steel::BlockLoader::thread_idx']]], ['threads_5fper_5ftg_9',['threads_per_tg',['../struct_read_writer.html#a64c58e358da22358df3075448ea23893',1,'ReadWriter']]], - ['tile_5fstride_10',['tile_stride',['../struct_quantized_block_loader.html#ac3f651c1a645291d1037a2cc8ded2320',1,'QuantizedBlockLoader::tile_stride'],['../structmlx_1_1steel_1_1_block_loader.html#ab87876699d55473620c7ea99f9da911d',1,'mlx::steel::BlockLoader::tile_stride']]], - ['tile_5fstride_5fa_11',['tile_stride_a',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8fddaa78913cdc8eea5e1cf7d2776330',1,'mlx::steel::BlockMMA']]], - ['tile_5fstride_5fb_12',['tile_stride_b',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae3f35453b3afbaac9df64ad5966b34a4',1,'mlx::steel::BlockMMA']]], - ['tiles_5fm_13',['tiles_m',['../struct_m_l_x_fast_attention_params.html#a0df159c839fc27b9426b8ac4336cc0ad',1,'MLXFastAttentionParams::tiles_m'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a4c5e33edf70be99cf93ac5723c12eb24',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#ad23a5a7f74cd5859741a36e4bc7823ca',1,'mlx::steel::GEMMParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a0970989624e17088d5326c2e198cb95b',1,'mlx::steel::GEMMSpiltKParams::tiles_m']]], - ['tiles_5fn_14',['tiles_n',['../struct_m_l_x_fast_attention_params.html#a608aa256216ac6d80af00209303d2029',1,'MLXFastAttentionParams::tiles_n'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a76f9f381e7187a993d65128b9b681b2d',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0e6b8b629232f1b43fbce9a395174bed',1,'mlx::steel::GEMMParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a5b46dfb9cee3606efa05d217349a20a6',1,'mlx::steel::GEMMSpiltKParams::tiles_n']]], - ['tm_15',['tm',['../structmlx_1_1steel_1_1_block_m_m_a.html#ab84d70540d09ce115794f343849f016f',1,'mlx::steel::BlockMMA']]], - ['tm_16',['TM',['../structmlx_1_1steel_1_1_block_m_m_a.html#aba5f749fdf32d8bd9d9e29f2a9ae4591',1,'mlx::steel::BlockMMA']]], - ['tm_5fstride_17',['TM_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a5b0029866f493363942133b55bff7307',1,'mlx::steel::BlockMMA']]], - ['tn_18',['TN',['../structmlx_1_1steel_1_1_block_m_m_a.html#a706ae779c1f8d2eb18f19c248567d424',1,'mlx::steel::BlockMMA']]], - ['tn_19',['tn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ad7a2033983cfbb474c50c2878057d8f1',1,'mlx::steel::BlockMMA']]], - ['tn_5fstride_20',['TN_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8b3690b383afd26563efb38f9c375e50',1,'mlx::steel::BlockMMA']]], - ['trows_21',['TROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a2528ff5ed472e4ed35415ada42276b07',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3957fb263fe040fe70683fd1d7b06487',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a94f0ce5bb7d87bc1fb6a7c2ba2b892d4',1,'mlx::steel::Conv2DWeightBlockLoader::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a789683f9ac9d9309d07c05f3bdedd2fd',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a5cefb1285ed13ad3490198e9303453de',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a2aff22af70f685f858adea73f5575cf7',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acbc28f364381166faaeec2783dc88e10',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::TROWS']]], - ['type_22',['type',['../structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea',1,'pocketfft::detail::ExecDcst::type'],['../structmlx_1_1core_1_1_reduction_plan.html#a24e407f13d4d02156380ecc1a6748a76',1,'mlx::core::ReductionPlan::type'],['../structmlx_1_1core_1_1_device.html#a763264ec90f7f23c5dced36c3f0db2e5',1,'mlx::core::Device::type']]] + ['threadsm_10',['threadsM',['../struct_g_e_m_v_kernel.html#a1dd943fcbf5e7be435fc36bed589a641',1,'GEMVKernel::threadsM'],['../struct_g_e_m_v_t_kernel.html#a4a53e73a581aa8881b1f86ce653519e6',1,'GEMVTKernel::threadsM']]], + ['threadsn_11',['threadsN',['../struct_g_e_m_v_kernel.html#a47bfab7d21dd18760d3e0937ad36b19d',1,'GEMVKernel::threadsN'],['../struct_g_e_m_v_t_kernel.html#ade6f15a9744616de9dd71498ad7e758d',1,'GEMVTKernel::threadsN']]], + ['tile_5fstride_12',['tile_stride',['../struct_quantized_block_loader.html#ac3f651c1a645291d1037a2cc8ded2320',1,'QuantizedBlockLoader::tile_stride'],['../structmlx_1_1steel_1_1_block_loader.html#ab87876699d55473620c7ea99f9da911d',1,'mlx::steel::BlockLoader::tile_stride']]], + ['tile_5fstride_5fa_13',['tile_stride_a',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8fddaa78913cdc8eea5e1cf7d2776330',1,'mlx::steel::BlockMMA']]], + ['tile_5fstride_5fb_14',['tile_stride_b',['../structmlx_1_1steel_1_1_block_m_m_a.html#ae3f35453b3afbaac9df64ad5966b34a4',1,'mlx::steel::BlockMMA']]], + ['tiles_5fm_15',['tiles_m',['../struct_m_l_x_fast_attention_params.html#a0df159c839fc27b9426b8ac4336cc0ad',1,'MLXFastAttentionParams::tiles_m'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a4c5e33edf70be99cf93ac5723c12eb24',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#ad23a5a7f74cd5859741a36e4bc7823ca',1,'mlx::steel::GEMMParams::tiles_m'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a0970989624e17088d5326c2e198cb95b',1,'mlx::steel::GEMMSpiltKParams::tiles_m']]], + ['tiles_5fn_16',['tiles_n',['../struct_m_l_x_fast_attention_params.html#a608aa256216ac6d80af00209303d2029',1,'MLXFastAttentionParams::tiles_n'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a76f9f381e7187a993d65128b9b681b2d',1,'mlx::steel::ImplicitGemmConv2DParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a0e6b8b629232f1b43fbce9a395174bed',1,'mlx::steel::GEMMParams::tiles_n'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a5b46dfb9cee3606efa05d217349a20a6',1,'mlx::steel::GEMMSpiltKParams::tiles_n']]], + ['tm_17',['TM',['../structmlx_1_1steel_1_1_block_m_m_a.html#aba5f749fdf32d8bd9d9e29f2a9ae4591',1,'mlx::steel::BlockMMA']]], + ['tm_18',['tm',['../structmlx_1_1steel_1_1_block_m_m_a.html#ab84d70540d09ce115794f343849f016f',1,'mlx::steel::BlockMMA']]], + ['tm_5fstride_19',['TM_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a5b0029866f493363942133b55bff7307',1,'mlx::steel::BlockMMA']]], + ['tn_20',['tn',['../structmlx_1_1steel_1_1_block_m_m_a.html#ad7a2033983cfbb474c50c2878057d8f1',1,'mlx::steel::BlockMMA']]], + ['tn_21',['TN',['../structmlx_1_1steel_1_1_block_m_m_a.html#a706ae779c1f8d2eb18f19c248567d424',1,'mlx::steel::BlockMMA']]], + ['tn_5fstride_22',['TN_stride',['../structmlx_1_1steel_1_1_block_m_m_a.html#a8b3690b383afd26563efb38f9c375e50',1,'mlx::steel::BlockMMA']]], + ['trows_23',['TROWS',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a2528ff5ed472e4ed35415ada42276b07',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3957fb263fe040fe70683fd1d7b06487',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a94f0ce5bb7d87bc1fb6a7c2ba2b892d4',1,'mlx::steel::Conv2DWeightBlockLoader::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a789683f9ac9d9309d07c05f3bdedd2fd',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a5cefb1285ed13ad3490198e9303453de',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#a2aff22af70f685f858adea73f5575cf7',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::TROWS'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#acbc28f364381166faaeec2783dc88e10',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::TROWS']]], + ['type_24',['type',['../structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea',1,'pocketfft::detail::ExecDcst::type'],['../structmlx_1_1core_1_1_reduction_plan.html#a24e407f13d4d02156380ecc1a6748a76',1,'mlx::core::ReductionPlan::type'],['../structmlx_1_1core_1_1_device.html#a763264ec90f7f23c5dced36c3f0db2e5',1,'mlx::core::Device::type']]] ]; diff --git a/docs/build/html/search/variables_6.js b/docs/build/html/search/variables_6.js index a0c1e1450e..48661e8eec 100644 --- a/docs/build/html/search/variables_6.js +++ b/docs/build/html/search/variables_6.js @@ -7,13 +7,14 @@ var searchData= ['gemm_5fn_5fiterations_5faligned_4',['gemm_n_iterations_aligned',['../struct_m_l_x_fast_attention_params.html#ab56b3db8fc6a938ce9c739ee78a7b803',1,'MLXFastAttentionParams']]], ['gemm_5fparams_5',['gemm_params',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#ab0724eb3ef52ee773b6607f6433b9f2c',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::gemm_params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#acc778b3c0b7ec38a43e8ea943df8704c',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::gemm_params'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#af59f9d356c4c3ec5627dc5a263d239d4',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::gemm_params']]], ['gemm_5fsv_5fm_5fblock_5fiterations_6',['gemm_sv_m_block_iterations',['../struct_m_l_x_fast_attention_params.html#a2799a2f219441fef7f351374f4cbc67c',1,'MLXFastAttentionParams']]], - ['generic_7',['generic',['../namespacemlx_1_1core.html#a34d69c4d46aa9b2a4a79dba7aba093d2',1,'mlx::core']]], - ['global_5fformatter_8',['global_formatter',['../namespacemlx_1_1core.html#af5a408a78cc934717dd711ddfda58ea6',1,'mlx::core']]], - ['gpu_9',['gpu',['../structmlx_1_1core_1_1_device.html#a45ed081b56ae5d4ddd39c83a5d8a1616',1,'mlx::core::Device']]], - ['grad_10',['grad',['../namespacemlx_1_1core.html#a99994806f81e83f98746c9e5ba4c3673',1,'mlx::core']]], - ['grid_11',['grid',['../struct_read_writer.html#ac7a957f99873d3797081f5d620f3b2c8',1,'ReadWriter']]], - ['group_5fstep_5fcnt_12',['group_step_cnt',['../struct_quantized_block_loader.html#a234feacde36a4afc0d740332a3769fb6',1,'QuantizedBlockLoader']]], - ['group_5fsteps_13',['group_steps',['../struct_quantized_block_loader.html#a31e14175f3d4902d9fe5ab5a219f61ba',1,'QuantizedBlockLoader']]], - ['group_5fstride_14',['group_stride',['../struct_quantized_block_loader.html#a0ace7e3762ecfa5a4106e7dee7e1b6ab',1,'QuantizedBlockLoader']]], - ['groups_15',['groups',['../struct_m_l_x_conv_params.html#af7a5590ac0974c7841c7f8b9fda0cbed',1,'MLXConvParams']]] + ['gemv_5fmasked_5fkernel_7',['gemv_masked_kernel',['../jit_2gemv__masked_8h.html#a933f06c211f86c37673dee329ed6901f',1,'gemv_masked.h']]], + ['generic_8',['generic',['../namespacemlx_1_1core.html#a34d69c4d46aa9b2a4a79dba7aba093d2',1,'mlx::core']]], + ['global_5fformatter_9',['global_formatter',['../namespacemlx_1_1core.html#af5a408a78cc934717dd711ddfda58ea6',1,'mlx::core']]], + ['gpu_10',['gpu',['../structmlx_1_1core_1_1_device.html#a45ed081b56ae5d4ddd39c83a5d8a1616',1,'mlx::core::Device']]], + ['grad_11',['grad',['../namespacemlx_1_1core.html#a99994806f81e83f98746c9e5ba4c3673',1,'mlx::core']]], + ['grid_12',['grid',['../struct_read_writer.html#ac7a957f99873d3797081f5d620f3b2c8',1,'ReadWriter']]], + ['group_5fstep_5fcnt_13',['group_step_cnt',['../struct_quantized_block_loader.html#a234feacde36a4afc0d740332a3769fb6',1,'QuantizedBlockLoader']]], + ['group_5fsteps_14',['group_steps',['../struct_quantized_block_loader.html#a31e14175f3d4902d9fe5ab5a219f61ba',1,'QuantizedBlockLoader']]], + ['group_5fstride_15',['group_stride',['../struct_quantized_block_loader.html#a0ace7e3762ecfa5a4106e7dee7e1b6ab',1,'QuantizedBlockLoader']]], + ['groups_16',['groups',['../struct_m_l_x_conv_params.html#af7a5590ac0974c7841c7f8b9fda0cbed',1,'MLXConvParams']]] ]; diff --git a/docs/build/html/search/variables_7.js b/docs/build/html/search/variables_7.js index 4ca064be03..33bcc018d6 100644 --- a/docs/build/html/search/variables_7.js +++ b/docs/build/html/search/variables_7.js @@ -3,5 +3,9 @@ var searchData= ['h12_0',['h12',['../namespacemlx_1_1core.html#a4beeeec4413be7adcfb14feaa9cf0e2e',1,'mlx::core']]], ['h20_1',['h20',['../namespacemlx_1_1core.html#a862c6b94fec384c34a699ced64d01404',1,'mlx::core']]], ['h28_2',['h28',['../namespacemlx_1_1core.html#ac447ad59592dd06435adca7df37e33ad',1,'mlx::core']]], - ['has_5fbatch_3',['has_batch',['../steel__gemm__fused_8h.html#adffcdc900c19ff97f1523e43f1a5a6cc',1,'steel_gemm_fused.h']]] + ['has_5fbatch_3',['has_batch',['../steel__gemm__fused_8h.html#adffcdc900c19ff97f1523e43f1a5a6cc',1,'steel_gemm_fused.h']]], + ['has_5fmul_5foperand_5fmask_4',['has_mul_operand_mask',['../struct_g_e_m_v_kernel.html#ad47223ee49b3cb7bf3746a2cec45f883',1,'GEMVKernel::has_mul_operand_mask'],['../struct_g_e_m_v_t_kernel.html#a8db6f01f96a36b216acd801c34a96ef5',1,'GEMVTKernel::has_mul_operand_mask']]], + ['has_5fmul_5foutput_5fmask_5',['has_mul_output_mask',['../struct_g_e_m_v_kernel.html#a0edbf2dd6a6563e7afa6dab6b670615c',1,'GEMVKernel::has_mul_output_mask'],['../struct_g_e_m_v_t_kernel.html#a8eb06f6569e4042e24fee220b11fa10d',1,'GEMVTKernel::has_mul_output_mask']]], + ['has_5foperand_5fmask_6',['has_operand_mask',['../struct_g_e_m_v_kernel.html#ab00784dff1512a7b0919fcb4cfa5d50e',1,'GEMVKernel::has_operand_mask'],['../struct_g_e_m_v_t_kernel.html#a6729d6e63e76a1e9c7c8e78d9aac4869',1,'GEMVTKernel::has_operand_mask']]], + ['has_5foutput_5fmask_7',['has_output_mask',['../struct_g_e_m_v_kernel.html#ab8b64c94f4c8f6f09c0777415589b487',1,'GEMVKernel::has_output_mask'],['../struct_g_e_m_v_t_kernel.html#aaefdf8f023da255bbb70a0c3e3408626',1,'GEMVTKernel::has_output_mask']]] ]; diff --git a/docs/build/html/search/variables_d.js b/docs/build/html/search/variables_d.js index 07fb1e2d27..56ed5f9a92 100644 --- a/docs/build/html/search/variables_d.js +++ b/docs/build/html/search/variables_d.js @@ -1,7 +1,7 @@ var searchData= [ - ['n_0',['N',['../struct_m_l_x_fast_attention_params.html#ab42c792a80388002e34992cbd837a167',1,'MLXFastAttentionParams::N'],['../struct_m_l_x_conv_params.html#ae6b7054dc3cffa8e6aedeb29fa7da932',1,'MLXConvParams::N'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a213f5ea4018120d8b61ab82754aaba83',1,'mlx::steel::ImplicitGemmConv2DParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a174626ab98515d89923b2841a664b9a1',1,'mlx::steel::GEMMParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a1103e79fb8962812b9a3c9d5c902ff86',1,'mlx::steel::GEMMSpiltKParams::N']]], - ['n_1',['n',['../struct_read_writer.html#a655346c9ebfc33a69da3f1c1d4238dfb',1,'ReadWriter']]], + ['n_0',['n',['../struct_read_writer.html#a655346c9ebfc33a69da3f1c1d4238dfb',1,'ReadWriter']]], + ['n_1',['N',['../struct_m_l_x_fast_attention_params.html#ab42c792a80388002e34992cbd837a167',1,'MLXFastAttentionParams::N'],['../struct_m_l_x_conv_params.html#ae6b7054dc3cffa8e6aedeb29fa7da932',1,'MLXConvParams::N'],['../structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html#a213f5ea4018120d8b61ab82754aaba83',1,'mlx::steel::ImplicitGemmConv2DParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_params.html#a174626ab98515d89923b2841a664b9a1',1,'mlx::steel::GEMMParams::N'],['../structmlx_1_1steel_1_1_g_e_m_m_spilt_k_params.html#a1103e79fb8962812b9a3c9d5c902ff86',1,'mlx::steel::GEMMSpiltKParams::N']]], ['n_5fchannels_2',['n_channels',['../structmlx_1_1steel_1_1_channel_helper.html#aa476bd0fcb38494c268547fc9820fc0a',1,'mlx::steel::ChannelHelper::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_011_01_4.html#a06c2fb9c93660e8f6916228cd77f9494',1,'mlx::steel::ChannelHelper< 1 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_012_01_4.html#ac66ff37bc2cf78d96667192a6cca73b5',1,'mlx::steel::ChannelHelper< 2 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_013_01_4.html#a071c015713b7bab09930661165517eff',1,'mlx::steel::ChannelHelper< 3 >::n_channels'],['../structmlx_1_1steel_1_1_channel_helper_3_014_01_4.html#a167b00a84adf93b60e3d7a943d5eb977',1,'mlx::steel::ChannelHelper< 4 >::n_channels']]], ['n_5fkv_5fheads_3',['N_KV_HEADS',['../struct_m_l_x_scaled_dot_product_attention_params.html#a68a292b9986c20560aca88394f82e9f7',1,'MLXScaledDotProductAttentionParams']]], ['n_5fper_5fblock_4',['N_PER_BLOCK',['../struct_kernel_merge_sort.html#a959aaf5bfb70796a525fed318f7ae8ab',1,'KernelMergeSort::N_PER_BLOCK'],['../struct_kernel_multi_block_merge_sort.html#ae5113ca5852d11999ae932439af95a5c',1,'KernelMultiBlockMergeSort::N_PER_BLOCK']]], @@ -10,7 +10,8 @@ var searchData= ['n_5frows_7',['n_rows',['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_large_filter.html#a097c48a23e1bd7d8cf3e9d531397602f',1,'mlx::steel::Conv2DInputBlockLoaderLargeFilter::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_filter.html#a3ec8a92c9e6643c1d5bf8af278026fe8',1,'mlx::steel::Conv2DInputBlockLoaderSmallFilter::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a593ec140370d53f8c968f6240116d38b',1,'mlx::steel::Conv2DWeightBlockLoader::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_small_channels.html#a8b6c0936c9ad2766242664f034d1115f',1,'mlx::steel::Conv2DInputBlockLoaderSmallChannels::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae905e56c1129606e93dbbcd7baed8f0f',1,'mlx::steel::Conv2DWeightBlockLoaderSmallChannels::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_input_block_loader_general.html#abff29c5d96645d9113314c9a997dd7a8',1,'mlx::steel::Conv2DInputBlockLoaderGeneral::n_rows'],['../structmlx_1_1steel_1_1_conv2_d_weight_block_loader_general.html#aaebb6da2cac9961f5edf52d16c18de7d',1,'mlx::steel::Conv2DWeightBlockLoaderGeneral::n_rows'],['../structmlx_1_1steel_1_1_block_loader.html#a973804e5b1d418c98c90861cda1a6fb5',1,'mlx::steel::BlockLoader::n_rows']]], ['names_8',['names',['../structmlx_1_1core_1_1_node_namer.html#a57823f9a2cdc60b2f06f857b36019277',1,'mlx::core::NodeNamer']]], ['ndim_9',['ndim',['../struct_indices.html#a7dec359e91d0eb2b64e5461b54308313',1,'Indices']]], - ['new_5fscoped_5fmemory_5fpool_10',['new_scoped_memory_pool',['../namespacemlx_1_1core_1_1metal.html#a6c52659362f6da31629216b65c297407',1,'mlx::core::metal']]], - ['next_11',['next',['../backend_2metal_2allocator_8h.html#ae704ab07eac590091daa5fc4aec7bddb',1,'allocator.h']]], - ['number_12',['number',['../namespacemlx_1_1core.html#a069c0aab6b36aef34419534ec4a4310d',1,'mlx::core']]] + ['needs_5ftgp_5freduction_10',['needs_tgp_reduction',['../struct_g_e_m_v_kernel.html#ae8113fddf6fb637acfd12efd978b704c',1,'GEMVKernel::needs_tgp_reduction'],['../struct_g_e_m_v_t_kernel.html#a67be7ec69c3791f02e97ccdb00ae0e03',1,'GEMVTKernel::needs_tgp_reduction']]], + ['new_5fscoped_5fmemory_5fpool_11',['new_scoped_memory_pool',['../namespacemlx_1_1core_1_1metal.html#a6c52659362f6da31629216b65c297407',1,'mlx::core::metal']]], + ['next_12',['next',['../backend_2metal_2allocator_8h.html#ae704ab07eac590091daa5fc4aec7bddb',1,'allocator.h']]], + ['number_13',['number',['../namespacemlx_1_1core.html#a069c0aab6b36aef34419534ec4a4310d',1,'mlx::core']]] ]; diff --git a/docs/build/html/searchindex.js b/docs/build/html/searchindex.js index aa250b071c..a30b4a9de1 100644 --- a/docs/build/html/searchindex.js +++ b/docs/build/html/searchindex.js @@ -1 +1 @@ -Search.setIndex({"docnames": ["cpp/ops", "dev/extensions", "dev/metal_debugger", "examples/linear_regression", "examples/llama-inference", "examples/mlp", "index", "install", "python/_autosummary/mlx.core.Device", "python/_autosummary/mlx.core.Dtype", "python/_autosummary/mlx.core.DtypeCategory", "python/_autosummary/mlx.core.abs", "python/_autosummary/mlx.core.add", "python/_autosummary/mlx.core.addmm", "python/_autosummary/mlx.core.all", "python/_autosummary/mlx.core.allclose", "python/_autosummary/mlx.core.any", "python/_autosummary/mlx.core.arange", "python/_autosummary/mlx.core.arccos", "python/_autosummary/mlx.core.arccosh", "python/_autosummary/mlx.core.arcsin", "python/_autosummary/mlx.core.arcsinh", "python/_autosummary/mlx.core.arctan", "python/_autosummary/mlx.core.arctan2", "python/_autosummary/mlx.core.arctanh", "python/_autosummary/mlx.core.argmax", "python/_autosummary/mlx.core.argmin", "python/_autosummary/mlx.core.argpartition", "python/_autosummary/mlx.core.argsort", "python/_autosummary/mlx.core.array", "python/_autosummary/mlx.core.array.T", "python/_autosummary/mlx.core.array.abs", "python/_autosummary/mlx.core.array.all", "python/_autosummary/mlx.core.array.any", "python/_autosummary/mlx.core.array.argmax", "python/_autosummary/mlx.core.array.argmin", "python/_autosummary/mlx.core.array.astype", "python/_autosummary/mlx.core.array.at", "python/_autosummary/mlx.core.array.conj", "python/_autosummary/mlx.core.array.cos", "python/_autosummary/mlx.core.array.cummax", "python/_autosummary/mlx.core.array.cummin", "python/_autosummary/mlx.core.array.cumprod", "python/_autosummary/mlx.core.array.cumsum", "python/_autosummary/mlx.core.array.diag", "python/_autosummary/mlx.core.array.diagonal", "python/_autosummary/mlx.core.array.dtype", "python/_autosummary/mlx.core.array.exp", "python/_autosummary/mlx.core.array.flatten", "python/_autosummary/mlx.core.array.item", "python/_autosummary/mlx.core.array.itemsize", "python/_autosummary/mlx.core.array.log", "python/_autosummary/mlx.core.array.log10", "python/_autosummary/mlx.core.array.log1p", "python/_autosummary/mlx.core.array.log2", "python/_autosummary/mlx.core.array.logsumexp", "python/_autosummary/mlx.core.array.max", "python/_autosummary/mlx.core.array.mean", "python/_autosummary/mlx.core.array.min", "python/_autosummary/mlx.core.array.moveaxis", "python/_autosummary/mlx.core.array.nbytes", "python/_autosummary/mlx.core.array.ndim", "python/_autosummary/mlx.core.array.prod", "python/_autosummary/mlx.core.array.reciprocal", "python/_autosummary/mlx.core.array.reshape", "python/_autosummary/mlx.core.array.round", "python/_autosummary/mlx.core.array.rsqrt", "python/_autosummary/mlx.core.array.shape", "python/_autosummary/mlx.core.array.sin", "python/_autosummary/mlx.core.array.size", "python/_autosummary/mlx.core.array.split", "python/_autosummary/mlx.core.array.sqrt", "python/_autosummary/mlx.core.array.square", "python/_autosummary/mlx.core.array.squeeze", "python/_autosummary/mlx.core.array.sum", "python/_autosummary/mlx.core.array.swapaxes", "python/_autosummary/mlx.core.array.tolist", "python/_autosummary/mlx.core.array.transpose", "python/_autosummary/mlx.core.array.var", "python/_autosummary/mlx.core.array.view", "python/_autosummary/mlx.core.array_equal", "python/_autosummary/mlx.core.as_strided", "python/_autosummary/mlx.core.atleast_1d", "python/_autosummary/mlx.core.atleast_2d", "python/_autosummary/mlx.core.atleast_3d", "python/_autosummary/mlx.core.bitwise_and", "python/_autosummary/mlx.core.bitwise_or", "python/_autosummary/mlx.core.bitwise_xor", "python/_autosummary/mlx.core.block_masked_mm", "python/_autosummary/mlx.core.broadcast_to", "python/_autosummary/mlx.core.ceil", "python/_autosummary/mlx.core.clip", "python/_autosummary/mlx.core.compile", "python/_autosummary/mlx.core.concatenate", "python/_autosummary/mlx.core.conj", "python/_autosummary/mlx.core.conjugate", "python/_autosummary/mlx.core.conv1d", "python/_autosummary/mlx.core.conv2d", "python/_autosummary/mlx.core.conv_general", "python/_autosummary/mlx.core.convolve", "python/_autosummary/mlx.core.cos", "python/_autosummary/mlx.core.cosh", "python/_autosummary/mlx.core.cummax", "python/_autosummary/mlx.core.cummin", "python/_autosummary/mlx.core.cumprod", "python/_autosummary/mlx.core.cumsum", "python/_autosummary/mlx.core.custom_function", "python/_autosummary/mlx.core.default_device", "python/_autosummary/mlx.core.default_stream", "python/_autosummary/mlx.core.degrees", "python/_autosummary/mlx.core.dequantize", "python/_autosummary/mlx.core.diag", "python/_autosummary/mlx.core.diagonal", "python/_autosummary/mlx.core.disable_compile", "python/_autosummary/mlx.core.distributed.Group", "python/_autosummary/mlx.core.distributed.all_gather", "python/_autosummary/mlx.core.distributed.all_sum", "python/_autosummary/mlx.core.distributed.init", "python/_autosummary/mlx.core.distributed.is_available", "python/_autosummary/mlx.core.divide", "python/_autosummary/mlx.core.divmod", "python/_autosummary/mlx.core.einsum", "python/_autosummary/mlx.core.einsum_path", "python/_autosummary/mlx.core.enable_compile", "python/_autosummary/mlx.core.equal", "python/_autosummary/mlx.core.erf", "python/_autosummary/mlx.core.erfinv", "python/_autosummary/mlx.core.eval", "python/_autosummary/mlx.core.exp", "python/_autosummary/mlx.core.expand_dims", "python/_autosummary/mlx.core.expm1", "python/_autosummary/mlx.core.eye", "python/_autosummary/mlx.core.fast.layer_norm", "python/_autosummary/mlx.core.fast.rms_norm", "python/_autosummary/mlx.core.fast.rope", "python/_autosummary/mlx.core.fast.scaled_dot_product_attention", "python/_autosummary/mlx.core.fft.fft", "python/_autosummary/mlx.core.fft.fft2", "python/_autosummary/mlx.core.fft.fftn", "python/_autosummary/mlx.core.fft.ifft", "python/_autosummary/mlx.core.fft.ifft2", "python/_autosummary/mlx.core.fft.ifftn", "python/_autosummary/mlx.core.fft.irfft", "python/_autosummary/mlx.core.fft.irfft2", "python/_autosummary/mlx.core.fft.irfftn", "python/_autosummary/mlx.core.fft.rfft", "python/_autosummary/mlx.core.fft.rfft2", "python/_autosummary/mlx.core.fft.rfftn", "python/_autosummary/mlx.core.flatten", "python/_autosummary/mlx.core.floor", "python/_autosummary/mlx.core.floor_divide", "python/_autosummary/mlx.core.full", "python/_autosummary/mlx.core.gather_mm", "python/_autosummary/mlx.core.gather_qmm", "python/_autosummary/mlx.core.grad", "python/_autosummary/mlx.core.greater", "python/_autosummary/mlx.core.greater_equal", "python/_autosummary/mlx.core.hadamard_transform", "python/_autosummary/mlx.core.identity", "python/_autosummary/mlx.core.inner", "python/_autosummary/mlx.core.isclose", "python/_autosummary/mlx.core.isinf", "python/_autosummary/mlx.core.isnan", "python/_autosummary/mlx.core.isneginf", "python/_autosummary/mlx.core.isposinf", "python/_autosummary/mlx.core.issubdtype", "python/_autosummary/mlx.core.jvp", "python/_autosummary/mlx.core.left_shift", "python/_autosummary/mlx.core.less", "python/_autosummary/mlx.core.less_equal", "python/_autosummary/mlx.core.linalg.cholesky", "python/_autosummary/mlx.core.linalg.inv", "python/_autosummary/mlx.core.linalg.norm", "python/_autosummary/mlx.core.linalg.qr", "python/_autosummary/mlx.core.linalg.svd", "python/_autosummary/mlx.core.linspace", "python/_autosummary/mlx.core.load", "python/_autosummary/mlx.core.log", "python/_autosummary/mlx.core.log10", "python/_autosummary/mlx.core.log1p", "python/_autosummary/mlx.core.log2", "python/_autosummary/mlx.core.logaddexp", "python/_autosummary/mlx.core.logical_and", "python/_autosummary/mlx.core.logical_not", "python/_autosummary/mlx.core.logical_or", "python/_autosummary/mlx.core.logsumexp", "python/_autosummary/mlx.core.matmul", "python/_autosummary/mlx.core.max", "python/_autosummary/mlx.core.maximum", "python/_autosummary/mlx.core.mean", "python/_autosummary/mlx.core.meshgrid", "python/_autosummary/mlx.core.metal.clear_cache", "python/_autosummary/mlx.core.metal.device_info", "python/_autosummary/mlx.core.metal.get_active_memory", "python/_autosummary/mlx.core.metal.get_cache_memory", "python/_autosummary/mlx.core.metal.get_peak_memory", "python/_autosummary/mlx.core.metal.is_available", "python/_autosummary/mlx.core.metal.reset_peak_memory", "python/_autosummary/mlx.core.metal.set_cache_limit", "python/_autosummary/mlx.core.metal.set_memory_limit", "python/_autosummary/mlx.core.metal.start_capture", "python/_autosummary/mlx.core.metal.stop_capture", "python/_autosummary/mlx.core.min", "python/_autosummary/mlx.core.minimum", "python/_autosummary/mlx.core.moveaxis", "python/_autosummary/mlx.core.multiply", "python/_autosummary/mlx.core.nan_to_num", "python/_autosummary/mlx.core.negative", "python/_autosummary/mlx.core.new_stream", "python/_autosummary/mlx.core.not_equal", "python/_autosummary/mlx.core.ones", "python/_autosummary/mlx.core.ones_like", "python/_autosummary/mlx.core.outer", "python/_autosummary/mlx.core.pad", "python/_autosummary/mlx.core.partition", "python/_autosummary/mlx.core.power", "python/_autosummary/mlx.core.prod", "python/_autosummary/mlx.core.quantize", "python/_autosummary/mlx.core.quantized_matmul", "python/_autosummary/mlx.core.radians", "python/_autosummary/mlx.core.random.bernoulli", "python/_autosummary/mlx.core.random.categorical", "python/_autosummary/mlx.core.random.gumbel", "python/_autosummary/mlx.core.random.key", "python/_autosummary/mlx.core.random.laplace", "python/_autosummary/mlx.core.random.multivariate_normal", "python/_autosummary/mlx.core.random.normal", "python/_autosummary/mlx.core.random.randint", "python/_autosummary/mlx.core.random.seed", "python/_autosummary/mlx.core.random.split", "python/_autosummary/mlx.core.random.truncated_normal", "python/_autosummary/mlx.core.random.uniform", "python/_autosummary/mlx.core.reciprocal", "python/_autosummary/mlx.core.remainder", "python/_autosummary/mlx.core.repeat", "python/_autosummary/mlx.core.reshape", "python/_autosummary/mlx.core.right_shift", "python/_autosummary/mlx.core.round", "python/_autosummary/mlx.core.rsqrt", "python/_autosummary/mlx.core.save", "python/_autosummary/mlx.core.save_gguf", "python/_autosummary/mlx.core.save_safetensors", "python/_autosummary/mlx.core.savez", "python/_autosummary/mlx.core.savez_compressed", "python/_autosummary/mlx.core.set_default_device", "python/_autosummary/mlx.core.set_default_stream", "python/_autosummary/mlx.core.sigmoid", "python/_autosummary/mlx.core.sign", "python/_autosummary/mlx.core.sin", "python/_autosummary/mlx.core.sinh", "python/_autosummary/mlx.core.softmax", "python/_autosummary/mlx.core.sort", "python/_autosummary/mlx.core.split", "python/_autosummary/mlx.core.sqrt", "python/_autosummary/mlx.core.square", "python/_autosummary/mlx.core.squeeze", "python/_autosummary/mlx.core.stack", "python/_autosummary/mlx.core.std", "python/_autosummary/mlx.core.stop_gradient", "python/_autosummary/mlx.core.stream", "python/_autosummary/mlx.core.subtract", "python/_autosummary/mlx.core.sum", "python/_autosummary/mlx.core.swapaxes", "python/_autosummary/mlx.core.synchronize", "python/_autosummary/mlx.core.take", "python/_autosummary/mlx.core.take_along_axis", "python/_autosummary/mlx.core.tan", "python/_autosummary/mlx.core.tanh", "python/_autosummary/mlx.core.tensordot", "python/_autosummary/mlx.core.tile", "python/_autosummary/mlx.core.topk", "python/_autosummary/mlx.core.trace", "python/_autosummary/mlx.core.transpose", "python/_autosummary/mlx.core.tri", "python/_autosummary/mlx.core.tril", "python/_autosummary/mlx.core.triu", "python/_autosummary/mlx.core.value_and_grad", "python/_autosummary/mlx.core.var", "python/_autosummary/mlx.core.view", "python/_autosummary/mlx.core.vjp", "python/_autosummary/mlx.core.vmap", "python/_autosummary/mlx.core.where", "python/_autosummary/mlx.core.zeros", "python/_autosummary/mlx.core.zeros_like", "python/_autosummary/mlx.nn.quantize", "python/_autosummary/mlx.nn.value_and_grad", "python/_autosummary/mlx.optimizers.clip_grad_norm", "python/_autosummary/mlx.utils.tree_flatten", "python/_autosummary/mlx.utils.tree_map", "python/_autosummary/mlx.utils.tree_map_with_path", "python/_autosummary/mlx.utils.tree_reduce", "python/_autosummary/mlx.utils.tree_unflatten", "python/_autosummary/stream_class", "python/array", "python/data_types", "python/devices_and_streams", "python/distributed", "python/fast", "python/fft", "python/linalg", "python/metal", "python/nn", "python/nn/_autosummary/mlx.nn.ALiBi", "python/nn/_autosummary/mlx.nn.AvgPool1d", "python/nn/_autosummary/mlx.nn.AvgPool2d", "python/nn/_autosummary/mlx.nn.BatchNorm", "python/nn/_autosummary/mlx.nn.Conv1d", "python/nn/_autosummary/mlx.nn.Conv2d", "python/nn/_autosummary/mlx.nn.Conv3d", "python/nn/_autosummary/mlx.nn.Dropout", "python/nn/_autosummary/mlx.nn.Dropout2d", "python/nn/_autosummary/mlx.nn.Dropout3d", "python/nn/_autosummary/mlx.nn.Embedding", "python/nn/_autosummary/mlx.nn.GELU", "python/nn/_autosummary/mlx.nn.GLU", "python/nn/_autosummary/mlx.nn.GRU", "python/nn/_autosummary/mlx.nn.GroupNorm", "python/nn/_autosummary/mlx.nn.HardShrink", "python/nn/_autosummary/mlx.nn.HardTanh", "python/nn/_autosummary/mlx.nn.Hardswish", "python/nn/_autosummary/mlx.nn.InstanceNorm", "python/nn/_autosummary/mlx.nn.LSTM", "python/nn/_autosummary/mlx.nn.LayerNorm", "python/nn/_autosummary/mlx.nn.LeakyReLU", "python/nn/_autosummary/mlx.nn.Linear", "python/nn/_autosummary/mlx.nn.MaxPool1d", "python/nn/_autosummary/mlx.nn.MaxPool2d", "python/nn/_autosummary/mlx.nn.Mish", "python/nn/_autosummary/mlx.nn.Module.apply", "python/nn/_autosummary/mlx.nn.Module.apply_to_modules", "python/nn/_autosummary/mlx.nn.Module.children", "python/nn/_autosummary/mlx.nn.Module.eval", "python/nn/_autosummary/mlx.nn.Module.filter_and_map", "python/nn/_autosummary/mlx.nn.Module.freeze", "python/nn/_autosummary/mlx.nn.Module.leaf_modules", "python/nn/_autosummary/mlx.nn.Module.load_weights", "python/nn/_autosummary/mlx.nn.Module.modules", "python/nn/_autosummary/mlx.nn.Module.named_modules", "python/nn/_autosummary/mlx.nn.Module.parameters", "python/nn/_autosummary/mlx.nn.Module.save_weights", "python/nn/_autosummary/mlx.nn.Module.set_dtype", "python/nn/_autosummary/mlx.nn.Module.state", "python/nn/_autosummary/mlx.nn.Module.train", "python/nn/_autosummary/mlx.nn.Module.trainable_parameters", "python/nn/_autosummary/mlx.nn.Module.training", "python/nn/_autosummary/mlx.nn.Module.unfreeze", "python/nn/_autosummary/mlx.nn.Module.update", "python/nn/_autosummary/mlx.nn.Module.update_modules", "python/nn/_autosummary/mlx.nn.MultiHeadAttention", "python/nn/_autosummary/mlx.nn.PReLU", "python/nn/_autosummary/mlx.nn.QuantizedEmbedding", "python/nn/_autosummary/mlx.nn.QuantizedLinear", "python/nn/_autosummary/mlx.nn.RMSNorm", "python/nn/_autosummary/mlx.nn.RNN", "python/nn/_autosummary/mlx.nn.ReLU", "python/nn/_autosummary/mlx.nn.ReLU6", "python/nn/_autosummary/mlx.nn.RoPE", "python/nn/_autosummary/mlx.nn.SELU", "python/nn/_autosummary/mlx.nn.Sequential", "python/nn/_autosummary/mlx.nn.SiLU", "python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding", "python/nn/_autosummary/mlx.nn.Softmax", "python/nn/_autosummary/mlx.nn.Softmin", "python/nn/_autosummary/mlx.nn.Softplus", "python/nn/_autosummary/mlx.nn.Softshrink", "python/nn/_autosummary/mlx.nn.Softsign", "python/nn/_autosummary/mlx.nn.Step", "python/nn/_autosummary/mlx.nn.Tanh", "python/nn/_autosummary/mlx.nn.Transformer", "python/nn/_autosummary/mlx.nn.Upsample", "python/nn/_autosummary/mlx.nn.init.constant", "python/nn/_autosummary/mlx.nn.init.glorot_normal", "python/nn/_autosummary/mlx.nn.init.glorot_uniform", "python/nn/_autosummary/mlx.nn.init.he_normal", "python/nn/_autosummary/mlx.nn.init.he_uniform", "python/nn/_autosummary/mlx.nn.init.identity", "python/nn/_autosummary/mlx.nn.init.normal", "python/nn/_autosummary/mlx.nn.init.uniform", "python/nn/_autosummary_functions/mlx.nn.elu", "python/nn/_autosummary_functions/mlx.nn.gelu", "python/nn/_autosummary_functions/mlx.nn.gelu_approx", "python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx", "python/nn/_autosummary_functions/mlx.nn.glu", "python/nn/_autosummary_functions/mlx.nn.hard_shrink", "python/nn/_autosummary_functions/mlx.nn.hard_tanh", "python/nn/_autosummary_functions/mlx.nn.hardswish", "python/nn/_autosummary_functions/mlx.nn.leaky_relu", "python/nn/_autosummary_functions/mlx.nn.log_sigmoid", "python/nn/_autosummary_functions/mlx.nn.log_softmax", "python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy", "python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss", "python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy", "python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss", "python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss", "python/nn/_autosummary_functions/mlx.nn.losses.huber_loss", "python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss", "python/nn/_autosummary_functions/mlx.nn.losses.l1_loss", "python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss", "python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss", "python/nn/_autosummary_functions/mlx.nn.losses.mse_loss", "python/nn/_autosummary_functions/mlx.nn.losses.nll_loss", "python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss", "python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss", "python/nn/_autosummary_functions/mlx.nn.mish", "python/nn/_autosummary_functions/mlx.nn.prelu", "python/nn/_autosummary_functions/mlx.nn.relu", "python/nn/_autosummary_functions/mlx.nn.relu6", "python/nn/_autosummary_functions/mlx.nn.selu", "python/nn/_autosummary_functions/mlx.nn.sigmoid", "python/nn/_autosummary_functions/mlx.nn.silu", "python/nn/_autosummary_functions/mlx.nn.softmax", "python/nn/_autosummary_functions/mlx.nn.softmin", "python/nn/_autosummary_functions/mlx.nn.softplus", "python/nn/_autosummary_functions/mlx.nn.softshrink", "python/nn/_autosummary_functions/mlx.nn.step", "python/nn/_autosummary_functions/mlx.nn.tanh", "python/nn/functions", "python/nn/init", "python/nn/layers", "python/nn/losses", "python/nn/module", "python/ops", "python/optimizers", "python/optimizers/_autosummary/mlx.optimizers.AdaDelta", "python/optimizers/_autosummary/mlx.optimizers.Adafactor", "python/optimizers/_autosummary/mlx.optimizers.Adagrad", "python/optimizers/_autosummary/mlx.optimizers.Adam", "python/optimizers/_autosummary/mlx.optimizers.AdamW", "python/optimizers/_autosummary/mlx.optimizers.Adamax", "python/optimizers/_autosummary/mlx.optimizers.Lion", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.init", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.state", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.update", "python/optimizers/_autosummary/mlx.optimizers.RMSprop", "python/optimizers/_autosummary/mlx.optimizers.SGD", "python/optimizers/_autosummary/mlx.optimizers.cosine_decay", "python/optimizers/_autosummary/mlx.optimizers.exponential_decay", "python/optimizers/_autosummary/mlx.optimizers.join_schedules", "python/optimizers/_autosummary/mlx.optimizers.linear_schedule", "python/optimizers/_autosummary/mlx.optimizers.step_decay", "python/optimizers/common_optimizers", "python/optimizers/optimizer", "python/optimizers/schedulers", "python/random", "python/transforms", "python/tree_utils", "usage/compile", "usage/distributed", "usage/function_transforms", "usage/indexing", "usage/lazy_evaluation", "usage/numpy", "usage/quick_start", "usage/saving_and_loading", "usage/unified_memory", "usage/using_streams"], "filenames": ["cpp/ops.rst", "dev/extensions.rst", "dev/metal_debugger.rst", "examples/linear_regression.rst", "examples/llama-inference.rst", "examples/mlp.rst", "index.rst", "install.rst", "python/_autosummary/mlx.core.Device.rst", "python/_autosummary/mlx.core.Dtype.rst", "python/_autosummary/mlx.core.DtypeCategory.rst", "python/_autosummary/mlx.core.abs.rst", "python/_autosummary/mlx.core.add.rst", "python/_autosummary/mlx.core.addmm.rst", "python/_autosummary/mlx.core.all.rst", "python/_autosummary/mlx.core.allclose.rst", "python/_autosummary/mlx.core.any.rst", "python/_autosummary/mlx.core.arange.rst", "python/_autosummary/mlx.core.arccos.rst", "python/_autosummary/mlx.core.arccosh.rst", "python/_autosummary/mlx.core.arcsin.rst", "python/_autosummary/mlx.core.arcsinh.rst", "python/_autosummary/mlx.core.arctan.rst", "python/_autosummary/mlx.core.arctan2.rst", "python/_autosummary/mlx.core.arctanh.rst", "python/_autosummary/mlx.core.argmax.rst", "python/_autosummary/mlx.core.argmin.rst", "python/_autosummary/mlx.core.argpartition.rst", "python/_autosummary/mlx.core.argsort.rst", "python/_autosummary/mlx.core.array.rst", "python/_autosummary/mlx.core.array.T.rst", "python/_autosummary/mlx.core.array.abs.rst", "python/_autosummary/mlx.core.array.all.rst", "python/_autosummary/mlx.core.array.any.rst", "python/_autosummary/mlx.core.array.argmax.rst", "python/_autosummary/mlx.core.array.argmin.rst", "python/_autosummary/mlx.core.array.astype.rst", "python/_autosummary/mlx.core.array.at.rst", "python/_autosummary/mlx.core.array.conj.rst", "python/_autosummary/mlx.core.array.cos.rst", "python/_autosummary/mlx.core.array.cummax.rst", "python/_autosummary/mlx.core.array.cummin.rst", "python/_autosummary/mlx.core.array.cumprod.rst", "python/_autosummary/mlx.core.array.cumsum.rst", "python/_autosummary/mlx.core.array.diag.rst", "python/_autosummary/mlx.core.array.diagonal.rst", "python/_autosummary/mlx.core.array.dtype.rst", "python/_autosummary/mlx.core.array.exp.rst", "python/_autosummary/mlx.core.array.flatten.rst", "python/_autosummary/mlx.core.array.item.rst", "python/_autosummary/mlx.core.array.itemsize.rst", "python/_autosummary/mlx.core.array.log.rst", "python/_autosummary/mlx.core.array.log10.rst", "python/_autosummary/mlx.core.array.log1p.rst", "python/_autosummary/mlx.core.array.log2.rst", "python/_autosummary/mlx.core.array.logsumexp.rst", "python/_autosummary/mlx.core.array.max.rst", "python/_autosummary/mlx.core.array.mean.rst", "python/_autosummary/mlx.core.array.min.rst", "python/_autosummary/mlx.core.array.moveaxis.rst", "python/_autosummary/mlx.core.array.nbytes.rst", "python/_autosummary/mlx.core.array.ndim.rst", "python/_autosummary/mlx.core.array.prod.rst", "python/_autosummary/mlx.core.array.reciprocal.rst", "python/_autosummary/mlx.core.array.reshape.rst", "python/_autosummary/mlx.core.array.round.rst", "python/_autosummary/mlx.core.array.rsqrt.rst", "python/_autosummary/mlx.core.array.shape.rst", "python/_autosummary/mlx.core.array.sin.rst", "python/_autosummary/mlx.core.array.size.rst", "python/_autosummary/mlx.core.array.split.rst", "python/_autosummary/mlx.core.array.sqrt.rst", "python/_autosummary/mlx.core.array.square.rst", "python/_autosummary/mlx.core.array.squeeze.rst", "python/_autosummary/mlx.core.array.sum.rst", "python/_autosummary/mlx.core.array.swapaxes.rst", "python/_autosummary/mlx.core.array.tolist.rst", "python/_autosummary/mlx.core.array.transpose.rst", "python/_autosummary/mlx.core.array.var.rst", "python/_autosummary/mlx.core.array.view.rst", "python/_autosummary/mlx.core.array_equal.rst", "python/_autosummary/mlx.core.as_strided.rst", "python/_autosummary/mlx.core.atleast_1d.rst", "python/_autosummary/mlx.core.atleast_2d.rst", "python/_autosummary/mlx.core.atleast_3d.rst", "python/_autosummary/mlx.core.bitwise_and.rst", "python/_autosummary/mlx.core.bitwise_or.rst", "python/_autosummary/mlx.core.bitwise_xor.rst", "python/_autosummary/mlx.core.block_masked_mm.rst", "python/_autosummary/mlx.core.broadcast_to.rst", "python/_autosummary/mlx.core.ceil.rst", "python/_autosummary/mlx.core.clip.rst", "python/_autosummary/mlx.core.compile.rst", "python/_autosummary/mlx.core.concatenate.rst", "python/_autosummary/mlx.core.conj.rst", "python/_autosummary/mlx.core.conjugate.rst", "python/_autosummary/mlx.core.conv1d.rst", "python/_autosummary/mlx.core.conv2d.rst", "python/_autosummary/mlx.core.conv_general.rst", "python/_autosummary/mlx.core.convolve.rst", "python/_autosummary/mlx.core.cos.rst", "python/_autosummary/mlx.core.cosh.rst", "python/_autosummary/mlx.core.cummax.rst", "python/_autosummary/mlx.core.cummin.rst", "python/_autosummary/mlx.core.cumprod.rst", "python/_autosummary/mlx.core.cumsum.rst", "python/_autosummary/mlx.core.custom_function.rst", "python/_autosummary/mlx.core.default_device.rst", "python/_autosummary/mlx.core.default_stream.rst", "python/_autosummary/mlx.core.degrees.rst", "python/_autosummary/mlx.core.dequantize.rst", "python/_autosummary/mlx.core.diag.rst", "python/_autosummary/mlx.core.diagonal.rst", "python/_autosummary/mlx.core.disable_compile.rst", "python/_autosummary/mlx.core.distributed.Group.rst", "python/_autosummary/mlx.core.distributed.all_gather.rst", "python/_autosummary/mlx.core.distributed.all_sum.rst", "python/_autosummary/mlx.core.distributed.init.rst", "python/_autosummary/mlx.core.distributed.is_available.rst", "python/_autosummary/mlx.core.divide.rst", "python/_autosummary/mlx.core.divmod.rst", "python/_autosummary/mlx.core.einsum.rst", "python/_autosummary/mlx.core.einsum_path.rst", "python/_autosummary/mlx.core.enable_compile.rst", "python/_autosummary/mlx.core.equal.rst", "python/_autosummary/mlx.core.erf.rst", "python/_autosummary/mlx.core.erfinv.rst", "python/_autosummary/mlx.core.eval.rst", "python/_autosummary/mlx.core.exp.rst", "python/_autosummary/mlx.core.expand_dims.rst", "python/_autosummary/mlx.core.expm1.rst", "python/_autosummary/mlx.core.eye.rst", "python/_autosummary/mlx.core.fast.layer_norm.rst", "python/_autosummary/mlx.core.fast.rms_norm.rst", "python/_autosummary/mlx.core.fast.rope.rst", "python/_autosummary/mlx.core.fast.scaled_dot_product_attention.rst", "python/_autosummary/mlx.core.fft.fft.rst", "python/_autosummary/mlx.core.fft.fft2.rst", "python/_autosummary/mlx.core.fft.fftn.rst", "python/_autosummary/mlx.core.fft.ifft.rst", "python/_autosummary/mlx.core.fft.ifft2.rst", "python/_autosummary/mlx.core.fft.ifftn.rst", "python/_autosummary/mlx.core.fft.irfft.rst", "python/_autosummary/mlx.core.fft.irfft2.rst", "python/_autosummary/mlx.core.fft.irfftn.rst", "python/_autosummary/mlx.core.fft.rfft.rst", "python/_autosummary/mlx.core.fft.rfft2.rst", "python/_autosummary/mlx.core.fft.rfftn.rst", "python/_autosummary/mlx.core.flatten.rst", "python/_autosummary/mlx.core.floor.rst", "python/_autosummary/mlx.core.floor_divide.rst", "python/_autosummary/mlx.core.full.rst", "python/_autosummary/mlx.core.gather_mm.rst", "python/_autosummary/mlx.core.gather_qmm.rst", "python/_autosummary/mlx.core.grad.rst", "python/_autosummary/mlx.core.greater.rst", "python/_autosummary/mlx.core.greater_equal.rst", "python/_autosummary/mlx.core.hadamard_transform.rst", "python/_autosummary/mlx.core.identity.rst", "python/_autosummary/mlx.core.inner.rst", "python/_autosummary/mlx.core.isclose.rst", "python/_autosummary/mlx.core.isinf.rst", "python/_autosummary/mlx.core.isnan.rst", "python/_autosummary/mlx.core.isneginf.rst", "python/_autosummary/mlx.core.isposinf.rst", "python/_autosummary/mlx.core.issubdtype.rst", "python/_autosummary/mlx.core.jvp.rst", "python/_autosummary/mlx.core.left_shift.rst", "python/_autosummary/mlx.core.less.rst", "python/_autosummary/mlx.core.less_equal.rst", "python/_autosummary/mlx.core.linalg.cholesky.rst", "python/_autosummary/mlx.core.linalg.inv.rst", "python/_autosummary/mlx.core.linalg.norm.rst", "python/_autosummary/mlx.core.linalg.qr.rst", "python/_autosummary/mlx.core.linalg.svd.rst", "python/_autosummary/mlx.core.linspace.rst", "python/_autosummary/mlx.core.load.rst", "python/_autosummary/mlx.core.log.rst", "python/_autosummary/mlx.core.log10.rst", "python/_autosummary/mlx.core.log1p.rst", "python/_autosummary/mlx.core.log2.rst", "python/_autosummary/mlx.core.logaddexp.rst", "python/_autosummary/mlx.core.logical_and.rst", "python/_autosummary/mlx.core.logical_not.rst", "python/_autosummary/mlx.core.logical_or.rst", "python/_autosummary/mlx.core.logsumexp.rst", "python/_autosummary/mlx.core.matmul.rst", "python/_autosummary/mlx.core.max.rst", "python/_autosummary/mlx.core.maximum.rst", "python/_autosummary/mlx.core.mean.rst", "python/_autosummary/mlx.core.meshgrid.rst", "python/_autosummary/mlx.core.metal.clear_cache.rst", "python/_autosummary/mlx.core.metal.device_info.rst", "python/_autosummary/mlx.core.metal.get_active_memory.rst", "python/_autosummary/mlx.core.metal.get_cache_memory.rst", "python/_autosummary/mlx.core.metal.get_peak_memory.rst", "python/_autosummary/mlx.core.metal.is_available.rst", "python/_autosummary/mlx.core.metal.reset_peak_memory.rst", "python/_autosummary/mlx.core.metal.set_cache_limit.rst", "python/_autosummary/mlx.core.metal.set_memory_limit.rst", "python/_autosummary/mlx.core.metal.start_capture.rst", "python/_autosummary/mlx.core.metal.stop_capture.rst", "python/_autosummary/mlx.core.min.rst", "python/_autosummary/mlx.core.minimum.rst", "python/_autosummary/mlx.core.moveaxis.rst", "python/_autosummary/mlx.core.multiply.rst", "python/_autosummary/mlx.core.nan_to_num.rst", "python/_autosummary/mlx.core.negative.rst", "python/_autosummary/mlx.core.new_stream.rst", "python/_autosummary/mlx.core.not_equal.rst", "python/_autosummary/mlx.core.ones.rst", "python/_autosummary/mlx.core.ones_like.rst", "python/_autosummary/mlx.core.outer.rst", "python/_autosummary/mlx.core.pad.rst", "python/_autosummary/mlx.core.partition.rst", "python/_autosummary/mlx.core.power.rst", "python/_autosummary/mlx.core.prod.rst", "python/_autosummary/mlx.core.quantize.rst", "python/_autosummary/mlx.core.quantized_matmul.rst", "python/_autosummary/mlx.core.radians.rst", "python/_autosummary/mlx.core.random.bernoulli.rst", "python/_autosummary/mlx.core.random.categorical.rst", "python/_autosummary/mlx.core.random.gumbel.rst", "python/_autosummary/mlx.core.random.key.rst", "python/_autosummary/mlx.core.random.laplace.rst", "python/_autosummary/mlx.core.random.multivariate_normal.rst", "python/_autosummary/mlx.core.random.normal.rst", "python/_autosummary/mlx.core.random.randint.rst", "python/_autosummary/mlx.core.random.seed.rst", "python/_autosummary/mlx.core.random.split.rst", "python/_autosummary/mlx.core.random.truncated_normal.rst", "python/_autosummary/mlx.core.random.uniform.rst", "python/_autosummary/mlx.core.reciprocal.rst", "python/_autosummary/mlx.core.remainder.rst", "python/_autosummary/mlx.core.repeat.rst", "python/_autosummary/mlx.core.reshape.rst", "python/_autosummary/mlx.core.right_shift.rst", "python/_autosummary/mlx.core.round.rst", "python/_autosummary/mlx.core.rsqrt.rst", "python/_autosummary/mlx.core.save.rst", "python/_autosummary/mlx.core.save_gguf.rst", "python/_autosummary/mlx.core.save_safetensors.rst", "python/_autosummary/mlx.core.savez.rst", "python/_autosummary/mlx.core.savez_compressed.rst", "python/_autosummary/mlx.core.set_default_device.rst", "python/_autosummary/mlx.core.set_default_stream.rst", "python/_autosummary/mlx.core.sigmoid.rst", "python/_autosummary/mlx.core.sign.rst", "python/_autosummary/mlx.core.sin.rst", "python/_autosummary/mlx.core.sinh.rst", "python/_autosummary/mlx.core.softmax.rst", "python/_autosummary/mlx.core.sort.rst", "python/_autosummary/mlx.core.split.rst", "python/_autosummary/mlx.core.sqrt.rst", "python/_autosummary/mlx.core.square.rst", "python/_autosummary/mlx.core.squeeze.rst", "python/_autosummary/mlx.core.stack.rst", "python/_autosummary/mlx.core.std.rst", "python/_autosummary/mlx.core.stop_gradient.rst", "python/_autosummary/mlx.core.stream.rst", "python/_autosummary/mlx.core.subtract.rst", "python/_autosummary/mlx.core.sum.rst", "python/_autosummary/mlx.core.swapaxes.rst", "python/_autosummary/mlx.core.synchronize.rst", "python/_autosummary/mlx.core.take.rst", "python/_autosummary/mlx.core.take_along_axis.rst", "python/_autosummary/mlx.core.tan.rst", "python/_autosummary/mlx.core.tanh.rst", "python/_autosummary/mlx.core.tensordot.rst", "python/_autosummary/mlx.core.tile.rst", "python/_autosummary/mlx.core.topk.rst", "python/_autosummary/mlx.core.trace.rst", "python/_autosummary/mlx.core.transpose.rst", "python/_autosummary/mlx.core.tri.rst", "python/_autosummary/mlx.core.tril.rst", "python/_autosummary/mlx.core.triu.rst", "python/_autosummary/mlx.core.value_and_grad.rst", "python/_autosummary/mlx.core.var.rst", "python/_autosummary/mlx.core.view.rst", "python/_autosummary/mlx.core.vjp.rst", "python/_autosummary/mlx.core.vmap.rst", "python/_autosummary/mlx.core.where.rst", "python/_autosummary/mlx.core.zeros.rst", "python/_autosummary/mlx.core.zeros_like.rst", "python/_autosummary/mlx.nn.quantize.rst", "python/_autosummary/mlx.nn.value_and_grad.rst", "python/_autosummary/mlx.optimizers.clip_grad_norm.rst", "python/_autosummary/mlx.utils.tree_flatten.rst", "python/_autosummary/mlx.utils.tree_map.rst", "python/_autosummary/mlx.utils.tree_map_with_path.rst", "python/_autosummary/mlx.utils.tree_reduce.rst", "python/_autosummary/mlx.utils.tree_unflatten.rst", "python/_autosummary/stream_class.rst", "python/array.rst", "python/data_types.rst", "python/devices_and_streams.rst", "python/distributed.rst", "python/fast.rst", "python/fft.rst", "python/linalg.rst", "python/metal.rst", "python/nn.rst", "python/nn/_autosummary/mlx.nn.ALiBi.rst", "python/nn/_autosummary/mlx.nn.AvgPool1d.rst", "python/nn/_autosummary/mlx.nn.AvgPool2d.rst", "python/nn/_autosummary/mlx.nn.BatchNorm.rst", "python/nn/_autosummary/mlx.nn.Conv1d.rst", "python/nn/_autosummary/mlx.nn.Conv2d.rst", "python/nn/_autosummary/mlx.nn.Conv3d.rst", "python/nn/_autosummary/mlx.nn.Dropout.rst", "python/nn/_autosummary/mlx.nn.Dropout2d.rst", "python/nn/_autosummary/mlx.nn.Dropout3d.rst", "python/nn/_autosummary/mlx.nn.Embedding.rst", "python/nn/_autosummary/mlx.nn.GELU.rst", "python/nn/_autosummary/mlx.nn.GLU.rst", "python/nn/_autosummary/mlx.nn.GRU.rst", "python/nn/_autosummary/mlx.nn.GroupNorm.rst", "python/nn/_autosummary/mlx.nn.HardShrink.rst", "python/nn/_autosummary/mlx.nn.HardTanh.rst", "python/nn/_autosummary/mlx.nn.Hardswish.rst", "python/nn/_autosummary/mlx.nn.InstanceNorm.rst", "python/nn/_autosummary/mlx.nn.LSTM.rst", "python/nn/_autosummary/mlx.nn.LayerNorm.rst", "python/nn/_autosummary/mlx.nn.LeakyReLU.rst", "python/nn/_autosummary/mlx.nn.Linear.rst", "python/nn/_autosummary/mlx.nn.MaxPool1d.rst", "python/nn/_autosummary/mlx.nn.MaxPool2d.rst", "python/nn/_autosummary/mlx.nn.Mish.rst", "python/nn/_autosummary/mlx.nn.Module.apply.rst", "python/nn/_autosummary/mlx.nn.Module.apply_to_modules.rst", "python/nn/_autosummary/mlx.nn.Module.children.rst", "python/nn/_autosummary/mlx.nn.Module.eval.rst", "python/nn/_autosummary/mlx.nn.Module.filter_and_map.rst", "python/nn/_autosummary/mlx.nn.Module.freeze.rst", "python/nn/_autosummary/mlx.nn.Module.leaf_modules.rst", "python/nn/_autosummary/mlx.nn.Module.load_weights.rst", "python/nn/_autosummary/mlx.nn.Module.modules.rst", "python/nn/_autosummary/mlx.nn.Module.named_modules.rst", "python/nn/_autosummary/mlx.nn.Module.parameters.rst", "python/nn/_autosummary/mlx.nn.Module.save_weights.rst", "python/nn/_autosummary/mlx.nn.Module.set_dtype.rst", "python/nn/_autosummary/mlx.nn.Module.state.rst", "python/nn/_autosummary/mlx.nn.Module.train.rst", "python/nn/_autosummary/mlx.nn.Module.trainable_parameters.rst", "python/nn/_autosummary/mlx.nn.Module.training.rst", "python/nn/_autosummary/mlx.nn.Module.unfreeze.rst", "python/nn/_autosummary/mlx.nn.Module.update.rst", "python/nn/_autosummary/mlx.nn.Module.update_modules.rst", "python/nn/_autosummary/mlx.nn.MultiHeadAttention.rst", "python/nn/_autosummary/mlx.nn.PReLU.rst", "python/nn/_autosummary/mlx.nn.QuantizedEmbedding.rst", "python/nn/_autosummary/mlx.nn.QuantizedLinear.rst", "python/nn/_autosummary/mlx.nn.RMSNorm.rst", "python/nn/_autosummary/mlx.nn.RNN.rst", "python/nn/_autosummary/mlx.nn.ReLU.rst", "python/nn/_autosummary/mlx.nn.ReLU6.rst", "python/nn/_autosummary/mlx.nn.RoPE.rst", "python/nn/_autosummary/mlx.nn.SELU.rst", "python/nn/_autosummary/mlx.nn.Sequential.rst", "python/nn/_autosummary/mlx.nn.SiLU.rst", "python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.rst", "python/nn/_autosummary/mlx.nn.Softmax.rst", "python/nn/_autosummary/mlx.nn.Softmin.rst", "python/nn/_autosummary/mlx.nn.Softplus.rst", "python/nn/_autosummary/mlx.nn.Softshrink.rst", "python/nn/_autosummary/mlx.nn.Softsign.rst", "python/nn/_autosummary/mlx.nn.Step.rst", "python/nn/_autosummary/mlx.nn.Tanh.rst", "python/nn/_autosummary/mlx.nn.Transformer.rst", "python/nn/_autosummary/mlx.nn.Upsample.rst", "python/nn/_autosummary/mlx.nn.init.constant.rst", "python/nn/_autosummary/mlx.nn.init.glorot_normal.rst", "python/nn/_autosummary/mlx.nn.init.glorot_uniform.rst", "python/nn/_autosummary/mlx.nn.init.he_normal.rst", "python/nn/_autosummary/mlx.nn.init.he_uniform.rst", "python/nn/_autosummary/mlx.nn.init.identity.rst", "python/nn/_autosummary/mlx.nn.init.normal.rst", "python/nn/_autosummary/mlx.nn.init.uniform.rst", "python/nn/_autosummary_functions/mlx.nn.elu.rst", "python/nn/_autosummary_functions/mlx.nn.gelu.rst", "python/nn/_autosummary_functions/mlx.nn.gelu_approx.rst", "python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.rst", "python/nn/_autosummary_functions/mlx.nn.glu.rst", "python/nn/_autosummary_functions/mlx.nn.hard_shrink.rst", "python/nn/_autosummary_functions/mlx.nn.hard_tanh.rst", "python/nn/_autosummary_functions/mlx.nn.hardswish.rst", "python/nn/_autosummary_functions/mlx.nn.leaky_relu.rst", "python/nn/_autosummary_functions/mlx.nn.log_sigmoid.rst", "python/nn/_autosummary_functions/mlx.nn.log_softmax.rst", "python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.rst", "python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.rst", "python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.rst", "python/nn/_autosummary_functions/mlx.nn.mish.rst", "python/nn/_autosummary_functions/mlx.nn.prelu.rst", "python/nn/_autosummary_functions/mlx.nn.relu.rst", "python/nn/_autosummary_functions/mlx.nn.relu6.rst", "python/nn/_autosummary_functions/mlx.nn.selu.rst", "python/nn/_autosummary_functions/mlx.nn.sigmoid.rst", "python/nn/_autosummary_functions/mlx.nn.silu.rst", "python/nn/_autosummary_functions/mlx.nn.softmax.rst", "python/nn/_autosummary_functions/mlx.nn.softmin.rst", "python/nn/_autosummary_functions/mlx.nn.softplus.rst", "python/nn/_autosummary_functions/mlx.nn.softshrink.rst", "python/nn/_autosummary_functions/mlx.nn.step.rst", "python/nn/_autosummary_functions/mlx.nn.tanh.rst", "python/nn/functions.rst", "python/nn/init.rst", "python/nn/layers.rst", "python/nn/losses.rst", "python/nn/module.rst", "python/ops.rst", "python/optimizers.rst", "python/optimizers/_autosummary/mlx.optimizers.AdaDelta.rst", "python/optimizers/_autosummary/mlx.optimizers.Adafactor.rst", "python/optimizers/_autosummary/mlx.optimizers.Adagrad.rst", "python/optimizers/_autosummary/mlx.optimizers.Adam.rst", "python/optimizers/_autosummary/mlx.optimizers.AdamW.rst", "python/optimizers/_autosummary/mlx.optimizers.Adamax.rst", "python/optimizers/_autosummary/mlx.optimizers.Lion.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.rst", "python/optimizers/_autosummary/mlx.optimizers.RMSprop.rst", "python/optimizers/_autosummary/mlx.optimizers.SGD.rst", "python/optimizers/_autosummary/mlx.optimizers.cosine_decay.rst", "python/optimizers/_autosummary/mlx.optimizers.exponential_decay.rst", "python/optimizers/_autosummary/mlx.optimizers.join_schedules.rst", "python/optimizers/_autosummary/mlx.optimizers.linear_schedule.rst", "python/optimizers/_autosummary/mlx.optimizers.step_decay.rst", "python/optimizers/common_optimizers.rst", "python/optimizers/optimizer.rst", "python/optimizers/schedulers.rst", "python/random.rst", "python/transforms.rst", "python/tree_utils.rst", "usage/compile.rst", "usage/distributed.rst", "usage/function_transforms.rst", "usage/indexing.rst", "usage/lazy_evaluation.rst", "usage/numpy.rst", "usage/quick_start.rst", "usage/saving_and_loading.rst", "usage/unified_memory.rst", "usage/using_streams.rst"], "titles": ["Operations", "Custom Extensions in MLX", "Metal Debugger", "Linear Regression", "LLM inference", "Multi-Layer Perceptron", "MLX", "Build and Install", "mlx.core.Device", "mlx.core.Dtype", "mlx.core.DtypeCategory", "mlx.core.abs", "mlx.core.add", "mlx.core.addmm", "mlx.core.all", "mlx.core.allclose", "mlx.core.any", "mlx.core.arange", "mlx.core.arccos", "mlx.core.arccosh", "mlx.core.arcsin", "mlx.core.arcsinh", "mlx.core.arctan", "mlx.core.arctan2", "mlx.core.arctanh", "mlx.core.argmax", "mlx.core.argmin", "mlx.core.argpartition", "mlx.core.argsort", "mlx.core.array", "mlx.core.array.T", "mlx.core.array.abs", "mlx.core.array.all", "mlx.core.array.any", "mlx.core.array.argmax", "mlx.core.array.argmin", "mlx.core.array.astype", "mlx.core.array.at", "mlx.core.array.conj", "mlx.core.array.cos", "mlx.core.array.cummax", "mlx.core.array.cummin", "mlx.core.array.cumprod", "mlx.core.array.cumsum", "mlx.core.array.diag", "mlx.core.array.diagonal", "mlx.core.array.dtype", "mlx.core.array.exp", "mlx.core.array.flatten", "mlx.core.array.item", "mlx.core.array.itemsize", "mlx.core.array.log", "mlx.core.array.log10", "mlx.core.array.log1p", "mlx.core.array.log2", "mlx.core.array.logsumexp", "mlx.core.array.max", "mlx.core.array.mean", "mlx.core.array.min", "mlx.core.array.moveaxis", "mlx.core.array.nbytes", "mlx.core.array.ndim", "mlx.core.array.prod", "mlx.core.array.reciprocal", "mlx.core.array.reshape", "mlx.core.array.round", "mlx.core.array.rsqrt", "mlx.core.array.shape", "mlx.core.array.sin", "mlx.core.array.size", "mlx.core.array.split", "mlx.core.array.sqrt", "mlx.core.array.square", "mlx.core.array.squeeze", "mlx.core.array.sum", "mlx.core.array.swapaxes", "mlx.core.array.tolist", "mlx.core.array.transpose", "mlx.core.array.var", "mlx.core.array.view", "mlx.core.array_equal", "mlx.core.as_strided", "mlx.core.atleast_1d", "mlx.core.atleast_2d", "mlx.core.atleast_3d", "mlx.core.bitwise_and", "mlx.core.bitwise_or", "mlx.core.bitwise_xor", "mlx.core.block_masked_mm", "mlx.core.broadcast_to", "mlx.core.ceil", "mlx.core.clip", "mlx.core.compile", "mlx.core.concatenate", "mlx.core.conj", "mlx.core.conjugate", "mlx.core.conv1d", "mlx.core.conv2d", "mlx.core.conv_general", "mlx.core.convolve", "mlx.core.cos", "mlx.core.cosh", "mlx.core.cummax", "mlx.core.cummin", "mlx.core.cumprod", "mlx.core.cumsum", "mlx.core.custom_function", "mlx.core.default_device", "mlx.core.default_stream", "mlx.core.degrees", "mlx.core.dequantize", "mlx.core.diag", "mlx.core.diagonal", "mlx.core.disable_compile", "mlx.core.distributed.Group", "mlx.core.distributed.all_gather", "mlx.core.distributed.all_sum", "mlx.core.distributed.init", "mlx.core.distributed.is_available", "mlx.core.divide", "mlx.core.divmod", "mlx.core.einsum", "mlx.core.einsum_path", "mlx.core.enable_compile", "mlx.core.equal", "mlx.core.erf", "mlx.core.erfinv", "mlx.core.eval", "mlx.core.exp", "mlx.core.expand_dims", "mlx.core.expm1", "mlx.core.eye", "mlx.core.fast.layer_norm", "mlx.core.fast.rms_norm", "mlx.core.fast.rope", "mlx.core.fast.scaled_dot_product_attention", "mlx.core.fft.fft", "mlx.core.fft.fft2", "mlx.core.fft.fftn", "mlx.core.fft.ifft", "mlx.core.fft.ifft2", "mlx.core.fft.ifftn", "mlx.core.fft.irfft", "mlx.core.fft.irfft2", "mlx.core.fft.irfftn", "mlx.core.fft.rfft", "mlx.core.fft.rfft2", "mlx.core.fft.rfftn", "mlx.core.flatten", "mlx.core.floor", "mlx.core.floor_divide", "mlx.core.full", "mlx.core.gather_mm", "mlx.core.gather_qmm", "mlx.core.grad", "mlx.core.greater", "mlx.core.greater_equal", "mlx.core.hadamard_transform", "mlx.core.identity", "mlx.core.inner", "mlx.core.isclose", "mlx.core.isinf", "mlx.core.isnan", "mlx.core.isneginf", "mlx.core.isposinf", "mlx.core.issubdtype", "mlx.core.jvp", "mlx.core.left_shift", "mlx.core.less", "mlx.core.less_equal", "mlx.core.linalg.cholesky", "mlx.core.linalg.inv", "mlx.core.linalg.norm", "mlx.core.linalg.qr", "mlx.core.linalg.svd", "mlx.core.linspace", "mlx.core.load", "mlx.core.log", "mlx.core.log10", "mlx.core.log1p", "mlx.core.log2", "mlx.core.logaddexp", "mlx.core.logical_and", "mlx.core.logical_not", "mlx.core.logical_or", "mlx.core.logsumexp", "mlx.core.matmul", "mlx.core.max", "mlx.core.maximum", "mlx.core.mean", "mlx.core.meshgrid", "mlx.core.metal.clear_cache", "mlx.core.metal.device_info", "mlx.core.metal.get_active_memory", "mlx.core.metal.get_cache_memory", "mlx.core.metal.get_peak_memory", "mlx.core.metal.is_available", "mlx.core.metal.reset_peak_memory", "mlx.core.metal.set_cache_limit", "mlx.core.metal.set_memory_limit", "mlx.core.metal.start_capture", "mlx.core.metal.stop_capture", "mlx.core.min", "mlx.core.minimum", "mlx.core.moveaxis", "mlx.core.multiply", "mlx.core.nan_to_num", "mlx.core.negative", "mlx.core.new_stream", "mlx.core.not_equal", "mlx.core.ones", "mlx.core.ones_like", "mlx.core.outer", "mlx.core.pad", "mlx.core.partition", "mlx.core.power", "mlx.core.prod", "mlx.core.quantize", "mlx.core.quantized_matmul", "mlx.core.radians", "mlx.core.random.bernoulli", "mlx.core.random.categorical", "mlx.core.random.gumbel", "mlx.core.random.key", "mlx.core.random.laplace", "mlx.core.random.multivariate_normal", "mlx.core.random.normal", "mlx.core.random.randint", "mlx.core.random.seed", "mlx.core.random.split", "mlx.core.random.truncated_normal", "mlx.core.random.uniform", "mlx.core.reciprocal", "mlx.core.remainder", "mlx.core.repeat", "mlx.core.reshape", "mlx.core.right_shift", "mlx.core.round", "mlx.core.rsqrt", "mlx.core.save", "mlx.core.save_gguf", "mlx.core.save_safetensors", "mlx.core.savez", "mlx.core.savez_compressed", "mlx.core.set_default_device", "mlx.core.set_default_stream", "mlx.core.sigmoid", "mlx.core.sign", "mlx.core.sin", "mlx.core.sinh", "mlx.core.softmax", "mlx.core.sort", "mlx.core.split", "mlx.core.sqrt", "mlx.core.square", "mlx.core.squeeze", "mlx.core.stack", "mlx.core.std", "mlx.core.stop_gradient", "mlx.core.stream", "mlx.core.subtract", "mlx.core.sum", "mlx.core.swapaxes", "mlx.core.synchronize", "mlx.core.take", "mlx.core.take_along_axis", "mlx.core.tan", "mlx.core.tanh", "mlx.core.tensordot", "mlx.core.tile", "mlx.core.topk", "mlx.core.trace", "mlx.core.transpose", "mlx.core.tri", "mlx.core.tril", "mlx.core.triu", "mlx.core.value_and_grad", "mlx.core.var", "mlx.core.view", "mlx.core.vjp", "mlx.core.vmap", "mlx.core.where", "mlx.core.zeros", "mlx.core.zeros_like", "mlx.nn.quantize", "mlx.nn.value_and_grad", "mlx.optimizers.clip_grad_norm", "mlx.utils.tree_flatten", "mlx.utils.tree_map", "mlx.utils.tree_map_with_path", "mlx.utils.tree_reduce", "mlx.utils.tree_unflatten", "mlx.core.Stream", "Array", "Data Types", "Devices and Streams", "Distributed Communication", "Fast", "FFT", "Linear Algebra", "Metal", "Neural Networks", "mlx.nn.ALiBi", "mlx.nn.AvgPool1d", "mlx.nn.AvgPool2d", "mlx.nn.BatchNorm", "mlx.nn.Conv1d", "mlx.nn.Conv2d", "mlx.nn.Conv3d", "mlx.nn.Dropout", "mlx.nn.Dropout2d", "mlx.nn.Dropout3d", "mlx.nn.Embedding", "mlx.nn.GELU", "mlx.nn.GLU", "mlx.nn.GRU", "mlx.nn.GroupNorm", "mlx.nn.HardShrink", "mlx.nn.HardTanh", "mlx.nn.Hardswish", "mlx.nn.InstanceNorm", "mlx.nn.LSTM", "mlx.nn.LayerNorm", "mlx.nn.LeakyReLU", "mlx.nn.Linear", "mlx.nn.MaxPool1d", "mlx.nn.MaxPool2d", "mlx.nn.Mish", "mlx.nn.Module.apply", "mlx.nn.Module.apply_to_modules", "mlx.nn.Module.children", "mlx.nn.Module.eval", "mlx.nn.Module.filter_and_map", "mlx.nn.Module.freeze", "mlx.nn.Module.leaf_modules", "mlx.nn.Module.load_weights", "mlx.nn.Module.modules", "mlx.nn.Module.named_modules", "mlx.nn.Module.parameters", "mlx.nn.Module.save_weights", "mlx.nn.Module.set_dtype", "mlx.nn.Module.state", "mlx.nn.Module.train", "mlx.nn.Module.trainable_parameters", "mlx.nn.Module.training", "mlx.nn.Module.unfreeze", "mlx.nn.Module.update", "mlx.nn.Module.update_modules", "mlx.nn.MultiHeadAttention", "mlx.nn.PReLU", "mlx.nn.QuantizedEmbedding", "mlx.nn.QuantizedLinear", "mlx.nn.RMSNorm", "mlx.nn.RNN", "mlx.nn.ReLU", "mlx.nn.ReLU6", "mlx.nn.RoPE", "mlx.nn.SELU", "mlx.nn.Sequential", "mlx.nn.SiLU", "mlx.nn.SinusoidalPositionalEncoding", "mlx.nn.Softmax", "mlx.nn.Softmin", "mlx.nn.Softplus", "mlx.nn.Softshrink", "mlx.nn.Softsign", "mlx.nn.Step", "mlx.nn.Tanh", "mlx.nn.Transformer", "mlx.nn.Upsample", "mlx.nn.init.constant", "mlx.nn.init.glorot_normal", "mlx.nn.init.glorot_uniform", "mlx.nn.init.he_normal", "mlx.nn.init.he_uniform", "mlx.nn.init.identity", "mlx.nn.init.normal", "mlx.nn.init.uniform", "mlx.nn.elu", "mlx.nn.gelu", "mlx.nn.gelu_approx", "mlx.nn.gelu_fast_approx", "mlx.nn.glu", "mlx.nn.hard_shrink", "mlx.nn.hard_tanh", "mlx.nn.hardswish", "mlx.nn.leaky_relu", "mlx.nn.log_sigmoid", "mlx.nn.log_softmax", "mlx.nn.losses.binary_cross_entropy", "mlx.nn.losses.cosine_similarity_loss", "mlx.nn.losses.cross_entropy", "mlx.nn.losses.gaussian_nll_loss", "mlx.nn.losses.hinge_loss", "mlx.nn.losses.huber_loss", "mlx.nn.losses.kl_div_loss", "mlx.nn.losses.l1_loss", "mlx.nn.losses.log_cosh_loss", "mlx.nn.losses.margin_ranking_loss", "mlx.nn.losses.mse_loss", "mlx.nn.losses.nll_loss", "mlx.nn.losses.smooth_l1_loss", "mlx.nn.losses.triplet_loss", "mlx.nn.mish", "mlx.nn.prelu", "mlx.nn.relu", "mlx.nn.relu6", "mlx.nn.selu", "mlx.nn.sigmoid", "mlx.nn.silu", "mlx.nn.softmax", "mlx.nn.softmin", "mlx.nn.softplus", "mlx.nn.softshrink", "mlx.nn.step", "mlx.nn.tanh", "Functions", "Initializers", "Layers", "Loss Functions", "Module", "Operations", "Optimizers", "mlx.optimizers.AdaDelta", "mlx.optimizers.Adafactor", "mlx.optimizers.Adagrad", "mlx.optimizers.Adam", "mlx.optimizers.AdamW", "mlx.optimizers.Adamax", "mlx.optimizers.Lion", "mlx.optimizers.Optimizer.apply_gradients", "mlx.optimizers.Optimizer.init", "mlx.optimizers.Optimizer.state", "mlx.optimizers.Optimizer.update", "mlx.optimizers.RMSprop", "mlx.optimizers.SGD", "mlx.optimizers.cosine_decay", "mlx.optimizers.exponential_decay", "mlx.optimizers.join_schedules", "mlx.optimizers.linear_schedule", "mlx.optimizers.step_decay", "Common Optimizers", "Optimizer", "Schedulers", "Random", "Transforms", "Tree Utils", "Compilation", "Distributed Communication", "Function Transforms", "Indexing Arrays", "Lazy Evaluation", "Conversion to NumPy and Other Frameworks", "Quick Start Guide", "Saving and Loading Arrays", "Unified Memory", "Using Streams"], "terms": {"arrai": [0, 1, 4, 5, 6, 9, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 115, 116, 119, 120, 121, 122, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 202, 203, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 260, 261, 262, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 286, 301, 305, 321, 328, 335, 338, 343, 349, 369, 370, 371, 372, 373, 374, 375, 376, 377, 382, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 414, 417, 420, 423, 424, 425, 426, 427, 428, 429, 434, 435, 436, 437, 438, 439, 440, 447, 448, 449, 451, 452, 453, 455], "arang": [0, 172, 294, 369, 450, 452], "doubl": [0, 4], "start": [0, 1, 3, 4, 6, 7, 17, 134, 175, 200, 252, 290, 447, 450, 455], "stop": [0, 1, 4, 17, 175, 201, 258, 449, 450], "step": [0, 2, 4, 5, 17, 301, 315, 321, 353, 424, 431, 436, 438, 439, 440, 447, 448], "dtype": [0, 1, 4, 10, 17, 29, 36, 37, 76, 79, 131, 148, 151, 158, 165, 172, 173, 175, 210, 222, 224, 225, 226, 227, 230, 231, 271, 273, 278, 282, 294, 340, 369, 370, 371, 372, 373, 374, 375, 376, 377, 389, 391, 398, 436, 437, 438, 439, 440, 447, 448, 449, 450, 452, 453, 454], "streamordevic": [0, 1], "": [0, 1, 3, 4, 5, 46, 50, 61, 92, 108, 110, 137, 138, 140, 141, 143, 144, 146, 147, 154, 172, 174, 176, 189, 212, 217, 221, 237, 240, 241, 257, 259, 276, 277, 278, 280, 285, 301, 304, 315, 321, 326, 332, 333, 335, 339, 340, 341, 345, 353, 422, 431, 432, 444, 447, 449, 451, 452, 453, 454, 455], "A": [0, 1, 4, 6, 7, 8, 67, 80, 92, 132, 133, 135, 154, 166, 172, 173, 174, 176, 185, 186, 187, 192, 202, 217, 220, 221, 222, 224, 225, 226, 227, 230, 231, 252, 256, 259, 276, 279, 280, 284, 285, 286, 287, 288, 289, 290, 291, 292, 301, 305, 310, 315, 316, 320, 322, 332, 336, 337, 340, 346, 347, 352, 358, 360, 368, 371, 372, 374, 381, 402, 403, 420, 422, 426, 428, 430, 431, 433, 438, 447, 448, 449, 451, 452], "1d": [0, 96, 99, 240, 265], "number": [0, 1, 10, 17, 60, 69, 92, 97, 98, 110, 131, 153, 154, 158, 166, 175, 206, 213, 217, 218, 221, 224, 226, 229, 231, 234, 237, 268, 269, 273, 276, 279, 280, 284, 301, 305, 306, 307, 308, 310, 311, 316, 320, 348, 349, 368, 369, 371, 372, 373, 374, 436, 438, 439, 444, 447, 449, 456], "option": [0, 2, 4, 13, 14, 16, 17, 25, 26, 27, 28, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 81, 82, 83, 84, 88, 92, 93, 96, 97, 98, 99, 102, 103, 104, 105, 106, 110, 111, 112, 117, 131, 132, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 151, 152, 153, 154, 157, 158, 163, 164, 170, 171, 172, 173, 174, 175, 176, 185, 187, 189, 190, 199, 202, 206, 210, 213, 214, 216, 217, 218, 220, 221, 222, 224, 225, 226, 227, 229, 230, 231, 234, 235, 250, 251, 252, 255, 256, 257, 261, 263, 264, 268, 270, 271, 272, 273, 274, 275, 276, 277, 280, 282, 284, 287, 288, 289, 290, 303, 304, 305, 306, 307, 308, 315, 321, 324, 325, 326, 328, 332, 333, 335, 340, 345, 348, 350, 351, 353, 356, 360, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 423, 424, 425, 426, 427, 428, 429, 431, 434, 435, 436, 444, 447, 454, 456], "int": [0, 1, 4, 5, 8, 14, 16, 17, 25, 26, 27, 28, 32, 33, 34, 35, 40, 41, 42, 43, 44, 45, 48, 55, 56, 57, 58, 59, 62, 65, 67, 70, 73, 74, 75, 76, 78, 81, 88, 89, 93, 96, 97, 98, 102, 103, 104, 105, 110, 111, 112, 122, 129, 131, 134, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 151, 153, 154, 158, 165, 172, 175, 185, 187, 189, 192, 193, 194, 195, 198, 199, 202, 204, 210, 213, 214, 216, 217, 218, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 234, 235, 237, 250, 251, 252, 255, 256, 257, 261, 262, 264, 265, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 280, 282, 284, 292, 301, 303, 304, 305, 306, 307, 308, 312, 314, 315, 316, 320, 321, 322, 324, 325, 326, 348, 350, 351, 352, 353, 356, 360, 368, 382, 390, 391, 395, 400, 402, 420, 436, 438, 439, 440], "linspac": 0, "num": [0, 4, 175, 229], "50": [0, 175], "float32": [0, 1, 10, 17, 131, 135, 157, 158, 165, 172, 173, 175, 210, 222, 224, 225, 226, 230, 231, 273, 282, 294, 369, 370, 371, 372, 373, 374, 375, 376, 377, 389, 391, 398, 436, 437, 438, 439, 440, 447, 448, 449, 450, 451, 452, 453, 454], "evenli": [0, 175], "space": [0, 1, 175, 389, 400], "rang": [0, 1, 2, 3, 4, 5, 7, 17, 148, 152, 175, 372, 374, 380, 381, 422, 436, 437, 438, 439, 440, 444, 447, 449, 451, 455], "astyp": [0, 1, 4, 328, 452], "convert": [0, 1, 76, 82, 83, 84, 109, 148, 219, 350, 351, 451, 452, 453], "an": [0, 1, 2, 4, 5, 7, 9, 14, 16, 29, 82, 83, 84, 89, 96, 97, 98, 114, 127, 131, 132, 135, 148, 151, 158, 170, 172, 176, 199, 204, 210, 211, 213, 216, 217, 218, 234, 235, 237, 252, 255, 262, 264, 265, 268, 269, 273, 280, 282, 283, 287, 288, 289, 290, 301, 303, 304, 309, 316, 321, 322, 324, 325, 326, 328, 348, 349, 351, 353, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 380, 404, 417, 422, 423, 433, 437, 442, 444, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456], "given": [0, 1, 7, 14, 16, 27, 37, 81, 89, 91, 93, 102, 103, 104, 105, 110, 112, 122, 127, 129, 136, 137, 138, 139, 140, 141, 145, 146, 147, 151, 152, 172, 185, 187, 189, 198, 202, 206, 208, 216, 225, 227, 237, 245, 250, 252, 257, 261, 263, 269, 270, 271, 273, 274, 275, 277, 292, 303, 304, 309, 325, 326, 332, 348, 390, 392, 398], "data": [0, 1, 5, 6, 9, 17, 131, 145, 146, 151, 158, 175, 206, 210, 230, 271, 273, 278, 282, 311, 370, 371, 372, 373, 374, 375, 376, 377, 447, 448, 450, 452], "type": [0, 1, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 36, 67, 76, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 115, 116, 117, 119, 120, 121, 122, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 192, 198, 199, 202, 203, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 260, 261, 262, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 286, 287, 290, 301, 340, 368, 370, 371, 372, 373, 374, 375, 376, 377, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 447, 450], "as_strid": 0, "std": [0, 1, 376], "vector": [0, 1, 3, 6, 159, 166, 172, 264, 279, 280, 312, 391, 453], "shape": [0, 1, 2, 4, 5, 64, 80, 81, 88, 89, 92, 96, 97, 98, 112, 115, 135, 136, 139, 142, 145, 146, 147, 151, 152, 157, 166, 171, 186, 210, 211, 220, 221, 222, 224, 225, 226, 227, 230, 231, 235, 265, 278, 279, 281, 282, 283, 301, 303, 304, 305, 306, 307, 308, 310, 311, 315, 320, 321, 324, 325, 326, 335, 353, 370, 371, 372, 373, 374, 375, 376, 377, 391, 402, 422, 447, 449, 450, 453, 455], "size_t": [0, 1], "stride": [0, 1, 81, 96, 97, 98, 303, 304, 306, 307, 308, 325, 326, 356, 450], "offset": [0, 1, 4, 45, 81, 112, 132, 134, 271], "creat": [0, 1, 4, 7, 81, 117, 131, 158, 259, 301, 420, 422, 438, 447, 450, 452], "view": [0, 2, 81, 452], "copi": [0, 1, 4, 6, 214, 251, 452], "anoth": [0, 91, 165, 186, 260, 281, 294, 301, 328, 447, 449, 450, 455], "full": [0, 1, 5, 64, 77, 99, 250, 346, 347, 392, 447, 448, 451], "val": [0, 29, 151], "fill": [0, 1, 151, 211, 273, 283, 370, 371, 372, 373, 374, 376, 377], "valu": [0, 3, 4, 11, 15, 17, 25, 26, 49, 76, 80, 91, 131, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 151, 160, 172, 174, 175, 192, 206, 213, 220, 221, 222, 224, 225, 226, 227, 230, 231, 240, 264, 265, 276, 280, 285, 287, 288, 289, 290, 294, 304, 309, 310, 311, 317, 320, 324, 326, 332, 348, 349, 364, 366, 368, 370, 389, 390, 391, 392, 393, 394, 396, 397, 398, 399, 400, 401, 414, 420, 424, 427, 436, 437, 439, 440, 449], "templat": [0, 1], "typenam": [0, 1], "t": [0, 1, 4, 7, 125, 135, 153, 170, 218, 276, 301, 303, 315, 321, 325, 353, 423, 424, 425, 426, 427, 428, 429, 434, 435, 447, 449, 455], "zero": [0, 131, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 190, 197, 273, 274, 275, 283, 301, 303, 304, 309, 310, 311, 335, 370, 371, 372, 373, 374, 375, 376, 377, 417, 422, 424, 450], "const": [0, 1, 392], "inlin": 0, "zeros_lik": 0, "ones": [0, 1, 4, 211, 242, 273, 346, 347, 422, 448, 450], "ones_lik": 0, "ey": [0, 4, 171], "n": [0, 1, 4, 29, 88, 96, 97, 98, 131, 136, 138, 139, 141, 142, 145, 147, 157, 158, 225, 257, 273, 277, 303, 304, 305, 306, 307, 308, 310, 311, 315, 321, 325, 326, 353, 369, 397, 402, 448], "m": [0, 1, 4, 7, 88, 131, 152, 157, 172, 273, 303, 304, 325, 326, 423, 447], "k": [0, 4, 44, 88, 111, 131, 135, 152, 157, 270, 273, 274, 275, 303, 324, 325, 333], "specifi": [0, 1, 17, 36, 97, 98, 112, 143, 144, 151, 154, 172, 175, 204, 210, 221, 234, 262, 264, 265, 268, 271, 272, 276, 280, 282, 305, 366, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 414, 448, 449, 455], "diagon": [0, 44, 111, 131, 271, 273, 274, 275], "everywher": 0, "els": [0, 1, 4, 301, 333, 448, 451], "ident": [0, 106, 131, 258, 301, 342], "squar": [0, 3, 4, 133, 158, 171, 238, 253, 276, 288, 301, 352, 399, 401, 423, 424, 426, 427, 428, 449, 452], "matrix": [0, 3, 13, 44, 88, 110, 111, 131, 152, 153, 157, 158, 170, 171, 172, 173, 174, 186, 190, 217, 218, 225, 350, 351, 375, 417], "major": [0, 1], "tri": 0, "tril": 0, "x": [0, 1, 3, 4, 5, 37, 88, 106, 115, 116, 125, 130, 132, 133, 153, 157, 158, 172, 218, 222, 237, 242, 246, 274, 275, 281, 288, 290, 301, 303, 304, 305, 313, 314, 316, 320, 322, 323, 324, 325, 326, 327, 328, 349, 352, 354, 360, 366, 369, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 401, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 420, 422, 429, 447, 448, 449, 450, 451, 452, 453, 455], "0": [0, 1, 3, 4, 5, 7, 8, 13, 17, 37, 44, 45, 48, 65, 70, 78, 81, 93, 96, 97, 98, 111, 112, 131, 135, 148, 152, 154, 171, 172, 173, 191, 198, 206, 213, 220, 224, 226, 231, 234, 237, 252, 256, 257, 271, 273, 274, 275, 276, 277, 280, 286, 287, 289, 290, 301, 303, 304, 305, 306, 307, 308, 309, 310, 311, 313, 316, 317, 320, 322, 323, 325, 326, 349, 354, 356, 360, 364, 366, 368, 370, 371, 372, 373, 374, 375, 376, 377, 378, 380, 381, 383, 384, 385, 386, 389, 391, 393, 394, 398, 401, 402, 404, 405, 406, 407, 413, 414, 417, 420, 423, 424, 426, 427, 428, 429, 431, 434, 435, 436, 437, 438, 439, 440, 444, 447, 448, 449, 450, 451, 452, 453, 454], "triu": 0, "reshap": [0, 4, 172, 369, 450], "flatten": [0, 27, 28, 102, 103, 104, 105, 172, 212, 214, 234, 251, 264, 265, 270, 287], "start_axi": [0, 48, 148], "end_axi": [0, 48, 148], "1": [0, 1, 2, 4, 5, 13, 17, 27, 28, 37, 45, 48, 96, 97, 98, 111, 112, 130, 135, 136, 137, 139, 140, 142, 143, 144, 145, 146, 147, 148, 157, 159, 165, 172, 173, 186, 190, 199, 212, 214, 217, 221, 224, 225, 226, 231, 246, 251, 264, 270, 271, 276, 286, 289, 290, 294, 301, 303, 304, 305, 306, 307, 308, 309, 310, 311, 313, 314, 315, 316, 320, 321, 322, 324, 325, 326, 349, 352, 353, 356, 360, 366, 369, 371, 372, 373, 374, 375, 376, 377, 378, 380, 381, 382, 384, 387, 388, 389, 390, 391, 392, 393, 394, 395, 397, 398, 400, 401, 402, 407, 408, 410, 411, 412, 414, 417, 420, 422, 423, 424, 425, 426, 427, 428, 429, 431, 434, 435, 436, 437, 438, 439, 440, 447, 448, 449, 450, 452, 453, 454, 455], "dimens": [0, 1, 4, 14, 16, 25, 26, 61, 67, 76, 82, 83, 84, 92, 97, 98, 112, 129, 134, 143, 144, 146, 147, 148, 152, 153, 159, 170, 171, 172, 173, 174, 185, 186, 187, 189, 202, 216, 217, 221, 229, 257, 261, 265, 268, 272, 277, 305, 306, 307, 308, 310, 311, 314, 315, 316, 320, 321, 322, 348, 352, 353, 356, 368, 369, 382, 391, 447, 449], "hadamard_transform": 0, "float": [0, 1, 10, 13, 15, 17, 76, 132, 133, 134, 135, 150, 151, 157, 160, 165, 172, 206, 218, 220, 224, 226, 286, 294, 305, 309, 310, 311, 316, 320, 322, 328, 340, 352, 356, 360, 366, 368, 369, 370, 371, 372, 373, 374, 376, 377, 390, 391, 392, 394, 398, 401, 402, 413, 414, 423, 424, 425, 426, 427, 428, 429, 434, 435, 436, 437, 439, 440], "scale": [0, 1, 4, 13, 110, 132, 133, 134, 135, 153, 157, 217, 218, 224, 226, 286, 310, 311, 322, 348, 356, 357, 360, 369, 407, 424], "nullopt": 0, "multipli": [0, 1, 37, 153, 217, 218, 309, 360, 369], "hadamard": [0, 157], "correspond": [0, 1, 14, 16, 76, 91, 110, 112, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 185, 187, 202, 216, 261, 268, 280, 288, 449], "size": [0, 1, 4, 5, 50, 67, 88, 97, 110, 129, 132, 133, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 151, 153, 157, 158, 165, 172, 194, 199, 217, 218, 221, 235, 252, 255, 278, 284, 301, 303, 304, 306, 307, 308, 312, 320, 325, 326, 350, 351, 369, 424, 448, 451, 452], "squeez": [0, 369, 447], "ax": [0, 1, 14, 16, 25, 26, 77, 106, 129, 137, 138, 140, 141, 143, 144, 146, 147, 148, 159, 172, 185, 187, 189, 202, 213, 216, 250, 255, 257, 261, 262, 268, 272, 277, 449], "remov": [0, 112, 186, 221, 255, 391], "singleton": [0, 14, 16, 25, 26, 117, 185, 186, 187, 189, 202, 216, 257, 261, 277], "axi": [0, 1, 4, 5, 14, 16, 25, 26, 27, 28, 32, 33, 34, 35, 40, 41, 42, 43, 55, 56, 57, 58, 62, 70, 73, 74, 78, 93, 102, 103, 104, 105, 112, 115, 129, 132, 133, 136, 139, 142, 143, 144, 145, 146, 147, 148, 157, 172, 185, 187, 189, 202, 204, 213, 214, 216, 221, 234, 250, 251, 252, 255, 256, 257, 261, 262, 264, 265, 269, 270, 271, 272, 277, 278, 280, 303, 304, 314, 325, 326, 353, 382, 388, 390, 391, 395, 400, 402, 410, 411, 450], "all": [0, 1, 2, 5, 7, 15, 27, 37, 82, 83, 84, 92, 97, 98, 106, 115, 116, 117, 131, 138, 141, 144, 147, 152, 153, 174, 186, 213, 214, 255, 284, 301, 328, 329, 333, 336, 337, 338, 343, 345, 348, 360, 368, 369, 417, 420, 442, 444, 447, 450, 451, 453, 456], "expand_dim": 0, "add": [0, 1, 2, 4, 13, 37, 129, 181, 213, 217, 306, 307, 308, 449, 455], "slice": [0, 450], "each": [0, 1, 67, 110, 127, 134, 153, 165, 170, 171, 174, 186, 190, 213, 217, 218, 221, 234, 242, 243, 252, 269, 272, 278, 280, 281, 310, 311, 312, 315, 316, 321, 353, 356, 368, 389, 391, 444, 447, 448, 451], "slice_upd": 0, "src": 0, "updat": [0, 1, 3, 4, 5, 37, 92, 284, 288, 290, 305, 328, 329, 335, 340, 341, 342, 347, 422, 424, 427, 429, 430, 431, 435, 436, 437, 438, 439, 440, 447, 448, 451], "from": [0, 1, 4, 5, 6, 81, 109, 110, 112, 115, 116, 143, 144, 146, 147, 151, 152, 157, 172, 176, 186, 190, 195, 198, 211, 217, 219, 220, 221, 222, 223, 224, 227, 230, 242, 255, 258, 260, 264, 265, 270, 271, 281, 283, 287, 288, 289, 290, 291, 301, 324, 333, 335, 348, 371, 372, 373, 374, 376, 377, 392, 401, 417, 422, 446, 447, 448, 449, 451, 452, 453, 454, 455], "sourc": [0, 1, 2, 59, 204, 272, 448], "split": [0, 314, 316, 382], "num_split": 0, "sub": [0, 5, 112, 229, 271, 284], "along": [0, 1, 25, 26, 92, 93, 102, 103, 104, 105, 115, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 152, 153, 157, 172, 234, 250, 252, 256, 264, 265, 268, 269, 270, 271, 278, 301, 314, 353, 382], "indic": [0, 1, 15, 25, 26, 27, 28, 37, 152, 153, 154, 160, 161, 162, 163, 164, 174, 252, 264, 265, 276, 342, 344, 391, 398, 438, 450], "meshgrid": 0, "bool": [0, 1, 14, 15, 16, 25, 26, 32, 33, 34, 35, 40, 41, 42, 43, 55, 56, 57, 58, 62, 74, 76, 78, 80, 92, 98, 102, 103, 104, 105, 117, 118, 134, 153, 160, 165, 170, 172, 176, 185, 187, 189, 190, 196, 199, 202, 216, 218, 257, 261, 277, 305, 306, 307, 308, 315, 316, 320, 321, 322, 324, 328, 332, 333, 335, 340, 342, 345, 348, 351, 353, 356, 360, 368, 369, 389, 392, 424, 435], "spars": [0, 190], "fals": [0, 1, 4, 14, 15, 16, 25, 26, 32, 33, 34, 35, 40, 41, 42, 43, 55, 56, 57, 58, 62, 74, 78, 80, 92, 98, 102, 103, 104, 105, 117, 160, 165, 170, 172, 176, 185, 187, 189, 190, 199, 202, 216, 257, 261, 277, 281, 284, 287, 288, 289, 290, 294, 316, 320, 322, 324, 333, 335, 345, 348, 351, 356, 360, 368, 369, 389, 392, 424, 435, 452], "string": [0, 1, 122, 192, 452, 454], "index": [0, 1, 6, 8, 27, 37, 129, 131, 154, 190, 214, 264, 265, 276, 292], "xy": [0, 190], "coordin": [0, 190], "clip": [0, 286, 389, 424], "a_min": [0, 91], "a_max": [0, 91], "limit": [0, 1, 91, 198, 199, 450], "concaten": [0, 4, 115], "stack": [0, 447], "new": [0, 1, 5, 89, 112, 204, 208, 235, 256, 272, 278, 288, 289, 340, 348, 420, 422, 433, 438, 447, 450, 451, 452], "repeat": [0, 269], "arr": [0, 239, 450], "tile": [0, 135], "rep": [0, 269], "transpos": [0, 4, 30, 153, 218], "permut": [0, 5], "accord": [0, 222, 281, 284, 348, 371, 372, 373, 374], "initializer_list": 0, "swapax": [0, 106], "axis1": [0, 45, 75, 112, 262, 271], "axis2": [0, 45, 75, 112, 262, 271], "swap": [0, 99, 199, 262, 347], "two": [0, 1, 12, 13, 15, 23, 80, 83, 85, 86, 87, 88, 112, 119, 124, 137, 140, 146, 152, 153, 155, 156, 160, 168, 169, 170, 171, 173, 174, 181, 186, 188, 203, 205, 209, 212, 262, 290, 304, 314, 321, 326, 382, 390, 447, 448, 449, 450, 455], "moveaxi": 0, "destin": [0, 1, 59, 204], "move": [0, 1, 204, 455], "pad": [0, 96, 97, 98, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 303, 304, 306, 307, 308, 325, 326], "low_pad_s": 0, "high_pad_s": 0, "pad_valu": 0, "constant": [0, 1, 4, 7, 132, 133, 213, 301, 305, 316, 322, 352, 392, 402, 434, 436, 447, 452], "pair": [0, 1, 213, 335, 356], "pad_width": [0, 213], "revers": [0, 1, 40, 41, 42, 43, 81, 102, 103, 104, 105, 272, 360], "order": [0, 27, 81, 98, 122, 172, 214, 217, 270, 301, 316, 346, 358, 431, 447, 449], "broadcast_to": 0, "broadcast": [0, 1, 12, 15, 85, 86, 87, 89, 91, 119, 120, 124, 151, 155, 156, 160, 167, 168, 169, 181, 186, 188, 203, 205, 209, 215, 220, 221, 225, 230, 231, 233, 236, 260, 265, 281, 348], "broadcast_arrai": [0, 1], "input": [0, 1, 3, 4, 11, 12, 13, 14, 15, 16, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 109, 111, 112, 115, 116, 119, 120, 121, 122, 124, 125, 126, 128, 129, 130, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 152, 153, 154, 155, 156, 157, 159, 160, 161, 162, 163, 164, 166, 167, 168, 169, 170, 171, 172, 173, 174, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 202, 203, 204, 205, 206, 207, 209, 211, 212, 213, 214, 215, 216, 217, 218, 219, 229, 232, 233, 234, 235, 236, 237, 238, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 260, 261, 262, 264, 265, 266, 267, 268, 269, 270, 271, 272, 274, 275, 276, 277, 278, 280, 281, 283, 303, 304, 305, 306, 307, 308, 310, 311, 312, 314, 315, 316, 320, 321, 322, 324, 325, 326, 348, 351, 352, 353, 356, 366, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 382, 389, 390, 392, 393, 394, 395, 397, 398, 400, 402, 414, 417, 447, 449, 450, 453, 454], "against": 0, "one": [0, 1, 4, 7, 37, 76, 82, 91, 97, 98, 129, 131, 132, 133, 172, 179, 186, 218, 221, 255, 260, 294, 345, 369, 391, 448, 455], "equal": [0, 15, 27, 80, 131, 156, 160, 169, 209, 214, 227, 252, 320, 324], "b": [0, 1, 2, 4, 12, 13, 15, 23, 80, 85, 86, 87, 88, 119, 120, 124, 150, 152, 155, 156, 159, 160, 167, 168, 169, 172, 181, 182, 184, 186, 188, 203, 205, 209, 212, 215, 217, 224, 233, 236, 260, 268, 276, 289, 290, 314, 324, 353, 369, 382, 449, 450, 451, 452, 453, 454, 455], "return": [0, 1, 3, 4, 5, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 36, 49, 67, 76, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 109, 110, 111, 112, 115, 116, 117, 119, 120, 121, 122, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 194, 198, 199, 202, 203, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 301, 315, 321, 328, 329, 330, 332, 333, 334, 335, 336, 337, 338, 342, 343, 345, 346, 347, 353, 370, 371, 372, 373, 374, 375, 376, 377, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 417, 420, 430, 446, 447, 448, 449, 450, 451, 452, 454, 455], "element": [0, 1, 11, 12, 18, 19, 20, 21, 22, 23, 24, 27, 69, 81, 85, 86, 87, 90, 100, 101, 102, 103, 104, 105, 110, 119, 120, 124, 125, 126, 128, 130, 131, 149, 150, 153, 155, 156, 160, 161, 162, 163, 164, 167, 168, 169, 177, 178, 179, 180, 181, 182, 183, 184, 188, 190, 203, 205, 207, 209, 214, 215, 217, 218, 232, 233, 234, 236, 238, 246, 247, 248, 249, 253, 254, 260, 264, 266, 267, 270, 276, 278, 281, 309, 310, 311, 315, 319, 321, 327, 349, 353, 356, 384, 385, 387, 388, 403, 404, 406, 409, 410, 411, 412, 447, 449], "wise": [0, 1, 11, 12, 18, 19, 20, 21, 22, 23, 24, 85, 86, 87, 90, 100, 101, 119, 120, 124, 125, 126, 128, 130, 149, 150, 155, 156, 160, 167, 168, 169, 177, 178, 179, 180, 181, 182, 183, 184, 188, 203, 205, 207, 209, 215, 232, 233, 236, 238, 246, 247, 248, 249, 253, 254, 260, 266, 267, 310, 311, 319, 327, 349, 384, 385, 387, 388, 403, 404, 406, 409, 410, 411, 412, 447], "not_equ": 0, "greater": [0, 4, 27, 130, 156, 214, 286, 366, 414], "greater_equ": 0, "less": [0, 4, 27, 169, 214, 356, 401], "less_equ": 0, "array_equ": [0, 15, 160], "equal_nan": [0, 15, 80, 160], "true": [0, 1, 3, 4, 15, 40, 41, 42, 43, 80, 92, 102, 103, 104, 105, 134, 153, 160, 165, 170, 172, 176, 190, 199, 218, 250, 281, 284, 287, 288, 289, 290, 294, 301, 305, 306, 307, 308, 315, 316, 320, 321, 322, 324, 332, 333, 335, 342, 345, 351, 353, 356, 360, 368, 369, 389, 397, 424], "have": [0, 1, 4, 7, 15, 80, 82, 83, 84, 88, 115, 143, 144, 146, 147, 153, 160, 186, 200, 221, 278, 287, 321, 348, 358, 429, 431, 446, 447, 448, 450, 451, 455], "same": [0, 1, 4, 7, 15, 37, 80, 89, 92, 97, 98, 99, 115, 132, 133, 142, 145, 146, 147, 153, 154, 160, 166, 213, 221, 237, 278, 279, 281, 289, 301, 304, 305, 309, 316, 320, 326, 350, 370, 371, 372, 373, 374, 375, 376, 377, 391, 402, 420, 430, 444, 447, 448, 450, 455], "isnan": 0, "isinf": 0, "isposinf": 0, "isneginf": 0, "where": [0, 5, 131, 160, 217, 276, 280, 303, 304, 305, 306, 307, 308, 309, 310, 311, 313, 315, 316, 320, 321, 322, 324, 325, 326, 332, 349, 352, 353, 366, 373, 374, 378, 379, 381, 392, 398, 404, 407, 409, 414, 431, 448, 449, 450], "condit": [0, 1, 281, 455], "y": [0, 1, 3, 4, 5, 37, 106, 157, 281, 301, 305, 310, 316, 320, 322, 324, 352, 393, 398, 401, 422, 425, 447, 448, 449, 451, 452], "select": [0, 2, 7, 270, 281, 328, 332, 340], "depend": [0, 1, 2, 3, 7, 76, 172, 315, 321, 353, 448, 450, 454, 455], "nan_to_num": 0, "nan": [0, 15, 80, 160, 162, 206], "0f": 0, "posinf": [0, 206], "neginf": [0, 206], "replac": [0, 4, 206, 346, 347, 368, 401], "infin": [0, 161, 163, 164, 206, 325, 326, 428], "finit": [0, 206], "keepdim": [0, 14, 16, 25, 26, 32, 33, 34, 35, 55, 56, 57, 58, 62, 74, 78, 172, 185, 187, 189, 202, 216, 250, 257, 261, 277], "ar": [0, 1, 3, 4, 5, 6, 7, 15, 17, 80, 88, 89, 91, 92, 98, 99, 106, 112, 127, 131, 137, 138, 140, 141, 143, 144, 146, 147, 148, 153, 154, 160, 161, 162, 163, 164, 165, 166, 172, 173, 176, 186, 199, 212, 213, 214, 217, 218, 220, 221, 222, 227, 230, 231, 242, 243, 255, 256, 264, 276, 279, 280, 284, 287, 288, 294, 305, 306, 307, 308, 309, 310, 311, 316, 320, 322, 324, 335, 348, 351, 369, 389, 391, 392, 416, 420, 422, 429, 431, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455], "non": [0, 1, 7, 190, 343, 353, 403, 420], "allclos": 0, "rtol": [0, 15, 160], "1e": [0, 3, 5, 15, 160, 305, 316, 320, 322, 323, 352, 390, 392, 402, 422, 423, 424, 425, 426, 427, 428, 431, 434, 436, 437, 438, 439, 440], "5": [0, 1, 3, 4, 7, 172, 199, 220, 290, 303, 305, 309, 310, 311, 313, 317, 320, 325, 364, 369, 370, 373, 374, 380, 383, 401, 413, 417, 422, 434, 436, 437, 447, 449, 450], "atol": [0, 15, 160], "8": [0, 1, 4, 7, 172, 217, 294, 304, 320, 326, 368, 390, 423, 424, 425, 426, 427, 428, 434, 447, 450, 453, 455], "within": [0, 2, 27, 160], "toler": [0, 15, 160], "isclos": 0, "boolean": [0, 15, 80, 160, 161, 162, 163, 164, 182, 183, 184, 294, 344, 450], "reduc": [0, 7, 14, 16, 25, 26, 116, 185, 187, 189, 202, 216, 257, 261, 277, 290, 305, 368, 397], "output": [0, 1, 4, 7, 14, 15, 16, 17, 27, 81, 88, 89, 92, 94, 95, 102, 103, 104, 105, 106, 121, 131, 132, 133, 134, 135, 142, 145, 146, 147, 151, 152, 154, 157, 158, 160, 172, 175, 185, 187, 189, 190, 202, 206, 210, 211, 214, 216, 220, 221, 222, 224, 225, 226, 227, 230, 231, 242, 243, 250, 255, 257, 261, 265, 271, 273, 276, 277, 278, 279, 280, 281, 282, 283, 303, 304, 305, 306, 307, 308, 320, 324, 325, 326, 348, 351, 366, 368, 369, 371, 372, 373, 374, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 414, 417, 447, 448, 449, 450, 451, 452, 453, 454, 455], "i": [0, 1, 2, 4, 5, 6, 7, 15, 17, 27, 36, 76, 81, 91, 97, 98, 99, 102, 103, 104, 105, 106, 111, 112, 115, 116, 118, 120, 127, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 150, 151, 152, 153, 157, 160, 165, 166, 170, 171, 172, 173, 174, 176, 181, 185, 186, 190, 196, 199, 213, 214, 217, 218, 225, 226, 235, 239, 240, 241, 246, 250, 252, 257, 258, 263, 264, 265, 268, 271, 272, 276, 277, 278, 279, 280, 281, 284, 286, 287, 288, 289, 290, 294, 296, 301, 303, 304, 305, 306, 307, 308, 309, 310, 311, 313, 315, 316, 320, 321, 322, 324, 325, 326, 332, 333, 339, 341, 342, 344, 345, 347, 348, 349, 351, 352, 353, 356, 360, 366, 368, 369, 373, 374, 379, 381, 389, 390, 392, 397, 398, 401, 402, 404, 409, 414, 420, 422, 424, 427, 429, 430, 431, 436, 438, 439, 444, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456], "ani": [0, 1, 4, 6, 17, 92, 287, 288, 289, 290, 291, 301, 313, 328, 329, 332, 341, 351, 368, 369, 417, 439, 446, 447, 449, 451, 453, 454, 455], "sum": [0, 1, 3, 12, 105, 116, 159, 172, 185, 250, 268, 271, 301, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 448, 450, 452], "mean": [0, 3, 4, 5, 133, 224, 225, 226, 276, 301, 305, 316, 333, 352, 376, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 447, 449, 452], "comput": [0, 1, 3, 4, 5, 6, 7, 102, 103, 104, 105, 106, 110, 122, 130, 134, 154, 166, 170, 171, 172, 181, 189, 212, 217, 233, 250, 257, 258, 268, 276, 277, 279, 285, 301, 305, 315, 316, 320, 321, 322, 333, 346, 351, 352, 356, 368, 371, 372, 373, 374, 380, 381, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 422, 423, 424, 426, 427, 428, 429, 433, 447, 448, 449, 453, 455], "var": [0, 305, 316, 320, 322, 392], "ddof": [0, 78, 257, 277], "varianc": [0, 257, 277, 305, 316, 392], "standard": [0, 5, 49, 76, 186, 222, 226, 257, 368, 371, 373, 376, 448, 453], "deviat": [0, 226, 257, 371, 373, 376], "deviatoin": 0, "prod": 0, "The": [0, 1, 2, 4, 5, 6, 7, 11, 12, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 36, 46, 50, 60, 61, 67, 76, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 115, 116, 117, 119, 120, 121, 122, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 166, 167, 168, 169, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 194, 195, 198, 199, 200, 202, 203, 204, 205, 207, 209, 210, 211, 212, 213, 214, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 229, 230, 231, 232, 233, 234, 235, 236, 240, 241, 246, 247, 248, 249, 250, 251, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 294, 296, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 314, 315, 316, 320, 321, 322, 324, 325, 326, 328, 329, 333, 335, 339, 340, 341, 342, 345, 346, 347, 348, 350, 351, 352, 353, 356, 358, 360, 366, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 382, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 414, 417, 420, 422, 423, 424, 425, 426, 427, 428, 429, 432, 434, 435, 436, 439, 442, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456], "product": [0, 1, 13, 81, 104, 159, 166, 186, 212, 216, 268, 279, 348, 453], "max": [0, 1, 172, 188, 325, 326, 349, 384, 385, 390, 392, 393, 398, 402, 404, 406, 424, 428, 447, 449, 455], "maximum": [0, 5, 25, 37, 91, 102, 195, 199, 286, 301, 323, 354, 360, 380, 381, 386, 405, 420, 451], "min": [0, 1, 172, 203, 349, 384, 385, 404, 406], "minimum": [0, 26, 37, 91, 103, 360, 389, 390], "argmin": 0, "argmax": [0, 5], "sort": [0, 27, 28, 214, 270], "argsort": 0, "partit": [0, 27], "kth": [0, 27, 214], "smaller": [0, 7, 214, 429, 447], "first": [0, 1, 2, 3, 4, 5, 7, 112, 115, 148, 154, 167, 174, 182, 184, 186, 214, 229, 236, 262, 268, 271, 276, 287, 289, 290, 301, 304, 316, 326, 369, 390, 398, 424, 426, 427, 428, 431, 447, 449, 452, 455], "argpartit": 0, "topk": 0, "logsumexp": 0, "ab": [0, 15, 160, 172, 276, 316, 320, 322, 327, 352, 381, 403, 447], "absolut": [0, 11, 15, 160, 380, 381, 401], "neg": [0, 112, 148, 163, 206, 271, 323, 325, 326, 348, 392, 400, 402, 450], "negat": [0, 207], "sign": [0, 15, 160, 294, 429], "logical_not": 0, "logic": [0, 1, 182, 183, 184], "logical_and": 0, "logical_or": 0, "reciproc": [0, 238], "subtract": [0, 37], "divid": [0, 1, 37, 150, 217, 233, 448], "divmod": 0, "quotient": [0, 119, 120, 150], "remaind": [0, 120], "floor_divid": 0, "integ": [0, 10, 150, 152, 153, 165, 172, 192, 213, 217, 218, 220, 227, 252, 268, 280, 294, 312, 340, 438, 450], "divis": [0, 119, 150, 217, 233], "equival": [0, 1, 30, 64, 77, 120, 150, 153, 157, 264, 313, 317, 318, 319, 347, 349, 351, 354, 355, 357, 359, 361, 362, 363, 364, 365, 367], "do": [0, 1, 4, 7, 278, 301, 334, 345, 417, 420, 427, 447, 448, 449, 451], "floor": [0, 150], "between": [0, 1, 6, 91, 148, 368, 390, 393, 394, 397, 438, 448, 451, 452, 455], "ceil": 0, "exp": [0, 130, 181, 185, 222, 250, 378, 395, 407, 408, 412, 447, 455], "exponenti": [0, 128, 130, 357, 378, 407, 437], "sin": [0, 106, 360, 449, 453], "sine": [0, 20, 21, 248, 249, 449], "co": [0, 1, 106, 360, 449], "cosin": [0, 18, 19, 100, 101, 390, 436, 438, 449], "tan": 0, "tangent": [0, 1, 22, 23, 24, 106, 166, 266, 267, 367, 415], "arcsin": 0, "arc": 0, "arcco": 0, "arctan": 0, "arctan2": 0, "invers": [0, 18, 19, 20, 21, 22, 23, 24, 126, 139, 140, 141, 142, 143, 144, 171], "ratio": [0, 23], "sinh": 0, "hyperbol": [0, 19, 21, 24, 101, 249, 267, 367, 415], "cosh": [0, 397], "tanh": [0, 301, 313, 315, 321, 327, 353, 380, 403], "arcsinh": 0, "arccosh": 0, "arctanh": 0, "degre": [0, 219, 402], "radian": [0, 109], "log": [0, 179, 181, 185, 387, 388, 389, 392, 395, 397, 400, 412], "natur": [0, 177, 179, 451], "logarithm": [0, 177, 178, 179, 180], "log2": 0, "base": [0, 1, 134, 172, 178, 180, 215, 356, 368, 420, 422, 428, 442, 444, 447, 450], "2": [0, 1, 3, 4, 5, 37, 97, 111, 112, 125, 137, 140, 142, 143, 144, 145, 146, 147, 148, 157, 165, 170, 171, 172, 173, 174, 180, 186, 217, 225, 229, 268, 271, 273, 274, 275, 286, 290, 294, 301, 303, 304, 307, 313, 323, 325, 326, 352, 360, 369, 370, 371, 372, 373, 374, 375, 376, 377, 380, 391, 392, 394, 401, 402, 417, 420, 422, 423, 425, 426, 427, 431, 434, 447, 448, 449, 450, 451, 452, 453, 454, 455], "log10": 0, "10": [0, 2, 4, 5, 178, 237, 242, 288, 301, 335, 417, 438, 440, 447, 448, 450], "log1p": 0, "plu": [0, 179], "logaddexp": 0, "sigmoid": [0, 4, 301, 359, 381, 387, 389, 409], "logist": [0, 3, 246, 381, 409], "erf": [0, 126, 447], "error": [0, 1, 7, 117, 125, 126, 199, 252, 313, 379, 380, 381, 397, 399, 449, 452], "function": [0, 1, 2, 3, 4, 5, 6, 15, 81, 92, 106, 120, 125, 126, 154, 160, 166, 170, 171, 172, 173, 174, 186, 246, 276, 279, 280, 285, 286, 288, 289, 290, 301, 313, 314, 317, 318, 319, 327, 329, 333, 340, 345, 349, 353, 354, 355, 357, 358, 359, 361, 362, 363, 364, 365, 366, 367, 368, 379, 380, 381, 382, 383, 384, 385, 387, 388, 389, 403, 408, 410, 411, 412, 413, 414, 415, 417, 422, 431, 444, 446, 448, 450, 451, 452, 454], "erfinv": 0, "expm1": 0, "stop_gradi": [0, 449], "flow": [0, 258, 451], "gradient": [0, 3, 5, 106, 154, 258, 276, 285, 286, 301, 333, 346, 351, 368, 397, 420, 422, 423, 424, 426, 427, 428, 429, 430, 433, 435, 447, 448, 449, 450, 451, 452, 453], "round": [0, 217], "decim": [0, 65, 237], "point": [0, 1, 3, 4, 7, 81, 150, 218, 294], "matmul": [0, 152, 455], "multipl": [0, 7, 13, 88, 132, 133, 152, 153, 186, 205, 217, 218, 348, 360, 437, 438, 440, 447, 451, 454], "gather": [0, 115, 152, 153], "slice_s": 0, "entri": [0, 310, 311], "take": [0, 1, 4, 5, 85, 86, 87, 92, 152, 154, 166, 188, 203, 211, 218, 265, 276, 279, 280, 283, 289, 290, 348, 389, 444, 448, 449, 450, 454, 455, 456], "treat": [0, 1, 143, 144, 146, 147, 264, 369, 447], "take_along_axi": [0, 450], "scatter": 0, "paramet": [0, 1, 3, 4, 5, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 36, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 115, 116, 117, 119, 120, 121, 122, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 198, 199, 200, 202, 203, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 320, 321, 322, 323, 324, 325, 326, 328, 329, 332, 333, 335, 340, 341, 342, 345, 346, 347, 348, 349, 350, 351, 352, 353, 356, 358, 360, 364, 366, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 382, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 414, 416, 417, 420, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 433, 434, 435, 436, 437, 438, 439, 440, 442, 447, 448, 449, 451], "determin": [0, 1, 112, 225, 290, 294, 339, 454], "locat": [0, 1, 81, 346, 347, 448, 455], "assum": [0, 1, 4, 88, 173, 288, 301, 303, 304, 316, 325, 326], "d": [0, 1, 4, 111, 112, 159, 172, 186, 190, 212, 264, 271, 273, 274, 275, 291, 308, 311, 315, 321, 353, 423, 426, 428, 455], "simplic": 0, "which": [0, 1, 4, 5, 6, 7, 17, 36, 81, 92, 98, 112, 127, 134, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 154, 161, 162, 163, 164, 166, 170, 171, 172, 173, 174, 176, 190, 200, 217, 221, 222, 234, 235, 239, 240, 241, 242, 243, 255, 256, 264, 271, 276, 279, 280, 284, 304, 310, 311, 313, 326, 328, 332, 356, 389, 391, 394, 398, 401, 417, 430, 431, 444, 447, 448, 449, 450, 451, 455, 456], "appli": [0, 37, 134, 135, 152, 174, 288, 289, 290, 301, 303, 304, 305, 306, 307, 308, 310, 311, 313, 314, 316, 317, 318, 319, 320, 322, 323, 324, 325, 326, 327, 329, 342, 349, 351, 352, 353, 354, 355, 357, 359, 361, 362, 363, 364, 365, 366, 367, 369, 378, 379, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 417, 430, 433, 439, 442, 447, 448], "note": [0, 1, 4, 7, 15, 81, 88, 92, 106, 135, 143, 144, 153, 160, 172, 193, 217, 221, 278, 284, 301, 352, 369, 422, 452, 454], "assign": [0, 1, 37, 420], "henc": [0, 1, 217, 447], "If": [0, 1, 4, 7, 14, 15, 16, 17, 25, 26, 27, 28, 76, 80, 81, 91, 93, 99, 102, 103, 104, 105, 111, 112, 115, 116, 117, 127, 132, 134, 145, 146, 147, 150, 151, 154, 160, 170, 172, 176, 185, 186, 187, 189, 190, 198, 199, 202, 206, 210, 213, 214, 216, 221, 225, 234, 250, 251, 252, 257, 261, 263, 264, 265, 268, 270, 271, 276, 277, 280, 282, 284, 288, 290, 305, 306, 307, 308, 316, 322, 324, 333, 335, 345, 351, 353, 356, 358, 360, 369, 389, 391, 402, 424, 447, 448, 449, 451, 454, 455, 456], "provid": [0, 1, 4, 81, 110, 154, 268, 276, 288, 290, 296, 301, 328, 333, 335, 345, 346, 347, 350, 351, 368, 369, 416, 420, 448, 454, 456], "rank": [0, 398, 448], "must": [0, 1, 2, 7, 88, 91, 151, 153, 172, 220, 221, 225, 227, 230, 231, 281, 369, 452], "In": [0, 1, 4, 5, 37, 186, 217, 288, 301, 310, 316, 420, 423, 425, 426, 428, 429, 430, 446, 447, 448, 449, 451, 454, 455], "other": [0, 1, 4, 6, 165, 172, 301, 334, 420, 429, 447, 448, 450, 451, 453], "word": 0, "ndim": [0, 1, 148, 172, 174, 369], "lead": [0, 17, 81, 447], "remain": [0, 4, 276, 289, 309, 310, 311, 448], "For": [0, 1, 4, 7, 37, 135, 152, 165, 172, 217, 291, 301, 305, 310, 328, 333, 342, 345, 351, 356, 360, 369, 371, 372, 373, 374, 389, 417, 422, 444, 447, 448, 449, 450, 451, 452, 453, 454, 455], "exampl": [0, 2, 3, 4, 5, 7, 17, 37, 106, 148, 172, 173, 259, 264, 286, 289, 290, 301, 303, 304, 305, 320, 325, 326, 333, 335, 342, 345, 369, 370, 371, 372, 373, 374, 375, 376, 377, 389, 391, 398, 417, 422, 431, 436, 437, 438, 439, 440, 444, 449, 450, 451, 452, 453, 454], "auto": [0, 1, 7], "4": [0, 1, 4, 110, 148, 153, 172, 217, 218, 242, 284, 290, 294, 303, 304, 305, 320, 325, 326, 350, 351, 368, 369, 371, 372, 373, 389, 447, 448, 450, 453, 455], "3": [0, 1, 4, 7, 148, 165, 172, 173, 286, 290, 308, 313, 369, 372, 374, 380, 385, 424, 429, 444, 447, 450, 452, 453], "out": [0, 1, 7, 88, 303, 304, 310, 311, 325, 326, 342, 447, 448, 449, 450], "produc": [0, 1, 7, 92, 348, 417], "thi": [0, 1, 4, 5, 7, 14, 15, 16, 17, 25, 26, 27, 28, 81, 106, 123, 152, 153, 157, 160, 166, 170, 171, 172, 173, 174, 181, 185, 186, 187, 189, 191, 193, 202, 214, 216, 221, 245, 250, 251, 252, 257, 261, 264, 270, 277, 286, 289, 290, 301, 309, 310, 311, 314, 315, 321, 329, 330, 332, 333, 336, 337, 338, 343, 345, 346, 347, 348, 351, 353, 366, 371, 372, 373, 374, 380, 381, 382, 389, 397, 414, 420, 431, 446, 447, 448, 449, 451, 452, 454], "row": [0, 1, 81, 131, 158, 217, 273], "posit": [0, 4, 27, 112, 134, 148, 154, 164, 170, 204, 206, 214, 225, 271, 276, 288, 301, 306, 307, 308, 348, 356, 360, 392, 402], "ad": [0, 1, 3, 7, 132, 320, 420, 423, 424, 425, 426, 427, 428, 434, 448, 451, 454], "we": [0, 1, 3, 4, 5, 110, 153, 217, 218, 301, 312, 350, 358, 427, 429, 444, 446, 447, 448, 449, 451, 455], "also": [0, 1, 4, 5, 6, 7, 10, 12, 85, 86, 87, 113, 119, 120, 124, 138, 141, 144, 147, 155, 156, 167, 168, 169, 181, 188, 203, 205, 209, 215, 217, 233, 236, 260, 284, 285, 296, 301, 332, 346, 348, 350, 351, 359, 379, 407, 409, 416, 422, 447, 448, 449, 450, 451, 452, 453, 456], "To": [0, 1, 2, 3, 4, 5, 7, 198, 301, 417, 422, 447, 448, 449, 453], "control": [0, 323, 444, 451], "addit": [0, 1, 4, 7, 12, 13, 132, 133, 135, 176, 305, 316, 322, 348, 352, 420, 449], "item": [0, 1, 3, 4, 5, 288, 422, 451, 452, 453], "togeth": [0, 1, 5, 217, 288, 289, 448], "doe": [0, 1, 2, 4, 7, 193, 278, 286, 301, 447, 450, 451, 452], "perform": [0, 1, 2, 4, 6, 13, 88, 98, 102, 103, 104, 105, 121, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 152, 153, 157, 186, 218, 237, 250, 264, 301, 316, 368, 373, 374, 422, 447, 448, 450, 451, 455], "bound": [0, 227, 230, 231, 313, 377, 447, 450, 455], "check": [0, 1, 7, 80, 118, 165, 196, 335, 449, 450], "access": [0, 4, 49, 301, 420, 431, 448, 451, 455], "undefin": [0, 27, 106, 170, 214, 225, 450], "typic": [0, 135, 312, 422, 447, 451], "result": [0, 4, 13, 17, 37, 76, 81, 92, 132, 133, 153, 172, 186, 218, 225, 234, 256, 288, 289, 290, 360, 389, 447, 449, 452], "unintend": 0, "invalid": [0, 81], "memori": [0, 1, 6, 81, 191, 193, 194, 195, 197, 198, 199, 368, 420, 424, 447, 451, 452], "write": [0, 1, 4, 301, 452], "scatter_add": 0, "scatter_prod": 0, "scatter_max": 0, "linear": [0, 1, 4, 5, 6, 284, 288, 301, 313, 314, 323, 335, 351, 353, 354, 355, 357, 359, 369, 378, 379, 380, 381, 382, 386, 405, 406, 407, 409, 417, 420, 431, 439, 447], "scatter_min": 0, "sqrt": [0, 4, 125, 135, 157, 305, 313, 316, 320, 322, 324, 352, 360, 371, 372, 373, 374, 380, 423, 425, 426, 427, 434, 447], "root": [0, 4, 133, 238, 253, 352], "rsqrt": 0, "softmax": [0, 4, 135, 301, 388, 391], "precis": [0, 1, 4, 130, 135, 301, 313, 352, 389, 430, 447], "power": [0, 449, 452], "rais": [0, 4, 172, 199, 215, 252, 335], "cumsum": 0, "inclus": [0, 40, 41, 42, 43, 102, 103, 104, 105, 148], "cumul": [0, 81, 102, 103, 104, 105], "cumprod": 0, "cummax": 0, "cummin": 0, "conv_gener": 0, "weight": [0, 3, 96, 97, 98, 132, 133, 288, 301, 335, 339, 350, 351, 389, 391, 420, 424, 427, 429, 431, 435, 449, 451], "padding_lo": 0, "padding_hi": 0, "kernel_dil": [0, 98], "input_dil": [0, 98], "group": [0, 96, 97, 98, 110, 115, 116, 117, 135, 153, 217, 218, 278, 284, 316, 350, 351, 448], "flip": [0, 98, 99], "gener": [0, 1, 2, 3, 10, 17, 98, 131, 143, 144, 175, 190, 220, 225, 226, 227, 230, 231, 368, 444, 447, 450, 451, 456], "convolut": [0, 96, 97, 98, 99, 306, 307, 308, 310, 311], "filter": [0, 99, 306, 307, 308, 328, 332], "conv1d": [0, 301], "dilat": [0, 96, 97, 98, 306, 307], "conv2d": [0, 301], "2d": [0, 97, 112, 217, 305, 310], "conv3d": [0, 301], "tupl": [0, 29, 64, 67, 77, 93, 97, 98, 120, 122, 127, 129, 166, 172, 173, 174, 213, 217, 235, 255, 276, 279, 287, 288, 289, 290, 291, 303, 304, 307, 308, 325, 326, 335, 337, 358, 369, 424, 426, 427, 428, 429, 446, 449], "3d": [0, 1, 305, 311, 369], "quantized_matmul": 0, "w": [0, 3, 97, 110, 153, 217, 218, 276, 289, 304, 305, 307, 308, 310, 311, 324, 326, 422, 435, 449], "bias": [0, 110, 153, 217, 218, 315, 321, 333, 345, 348], "group_siz": [0, 110, 153, 217, 218, 284, 350, 351], "64": [0, 88, 110, 153, 217, 218, 284, 294, 350, 351], "bit": [0, 110, 153, 167, 217, 218, 236, 284, 294, 328, 350, 351, 352], "quantiz": [0, 110, 153, 176, 218, 350, 351], "its": [0, 1, 7, 186, 214, 229, 273, 285, 291, 301, 351, 422, 426, 427, 428, 448, 452, 455], "last": [0, 4, 28, 76, 132, 133, 138, 141, 143, 144, 146, 147, 148, 152, 153, 159, 170, 171, 173, 174, 186, 195, 221, 251, 268, 278, 306, 307, 308, 310, 311, 316, 369, 452], "dequant": [0, 217], "gather_qmm": 0, "lhs_indic": [0, 152, 153], "rhs_indic": [0, 152, 153], "level": [0, 152, 153, 373, 374], "tensordot": 0, "contract": [0, 122], "over": [0, 1, 4, 5, 14, 16, 25, 26, 27, 28, 96, 97, 98, 102, 103, 104, 105, 138, 141, 144, 147, 159, 172, 174, 175, 185, 187, 189, 202, 214, 216, 238, 250, 251, 257, 261, 268, 270, 277, 305, 306, 307, 308, 316, 322, 352, 391, 436, 439, 448, 449], "axes_a": 0, "axes_b": 0, "outer": [0, 447, 451], "inner": [0, 447], "addmm": 0, "c": [0, 1, 4, 13, 172, 303, 304, 305, 306, 307, 308, 310, 311, 320, 321, 325, 326, 452, 453, 455], "alpha": [0, 1, 13, 217, 378, 402, 404, 407, 427, 434], "f": [0, 1, 3, 5, 106, 172, 301, 321, 427, 447, 452], "beta": [0, 1, 13, 110, 217, 305, 316, 320, 322, 401, 422, 426, 427, 428, 429], "block_masked_mm": 0, "block_siz": [0, 88], "mask_out": [0, 88], "mask_lh": [0, 88], "mask_rh": [0, 88], "block": [0, 1, 4, 88, 368], "mask": [0, 4, 88, 135, 342, 348, 450], "gather_mm": [0, 153], "extract": [0, 4, 44, 111, 112, 301, 332, 420, 448], "construct": [0, 1, 5, 44, 111, 151, 210, 269, 282], "diag": [0, 174], "trace": [0, 2, 447], "implement": [0, 3, 5, 134, 135, 172, 312, 332, 348, 356, 358, 360, 366, 368, 369, 414, 423, 424, 425, 426, 428, 429, 430, 442, 447, 449], "allow": [0, 1, 165, 286, 301, 347, 420, 442, 448, 450, 453], "inject": 0, "ensur": [0, 1, 7, 286, 397, 448], "been": [0, 1, 4, 194, 451], "when": [0, 1, 4, 6, 7, 92, 98, 106, 170, 171, 172, 174, 176, 306, 307, 308, 369, 373, 374, 389, 395, 401, 420, 422, 438, 444, 447, 448, 455], "atleast_1d": 0, "atleast": 0, "atleast_2d": 0, "atleast_3d": 0, "number_of_el": 0, "invert": 0, "int32": [0, 10, 17, 37, 148, 165, 172, 227, 294, 369, 450, 453], "some": [0, 1, 3, 4, 5, 333, 345, 422, 431, 447, 449, 451], "scalar": [0, 1, 12, 13, 15, 29, 49, 76, 80, 85, 86, 87, 88, 89, 91, 119, 120, 124, 150, 151, 154, 155, 156, 157, 160, 167, 168, 169, 172, 175, 181, 182, 183, 184, 186, 188, 203, 205, 206, 209, 213, 215, 220, 227, 230, 231, 233, 236, 240, 260, 276, 278, 281, 285, 402, 449, 451, 453], "us": [0, 2, 3, 4, 5, 6, 7, 17, 37, 81, 106, 110, 113, 115, 116, 120, 134, 148, 153, 167, 170, 171, 172, 173, 174, 186, 193, 194, 195, 198, 217, 218, 234, 235, 236, 263, 287, 290, 294, 296, 301, 304, 310, 312, 313, 315, 321, 324, 326, 328, 332, 339, 346, 348, 350, 351, 353, 356, 360, 368, 369, 373, 374, 380, 381, 390, 417, 420, 422, 423, 424, 426, 427, 428, 429, 430, 431, 444, 446, 447, 448, 449, 450, 453, 455], "shapeless": [0, 92], "compil": [0, 2, 6, 7, 113, 123, 448, 449, 451], "pun": 0, "intend": [0, 447], "conjug": [0, 94], "bitwise_and": 0, "bitwis": [0, 85, 86, 87, 167, 236], "bitwise_or": 0, "bitwise_xor": 0, "exclus": [0, 81, 87], "left_shift": 0, "shift": [0, 167, 236, 305], "left": [0, 4, 134, 167, 172, 217, 303, 304, 313, 325, 326, 356, 369, 380, 381, 392, 394, 402], "right_shift": 0, "right": [0, 1, 7, 217, 236, 303, 304, 313, 325, 326, 369, 380, 381, 392, 394, 402], "you": [1, 2, 4, 5, 6, 7, 301, 360, 368, 417, 444, 447, 448, 449, 450, 452, 454, 455], "can": [1, 2, 4, 6, 7, 12, 17, 64, 77, 81, 85, 86, 87, 92, 112, 113, 114, 119, 120, 124, 127, 155, 156, 167, 168, 169, 172, 181, 188, 203, 205, 209, 215, 220, 221, 227, 230, 231, 233, 236, 240, 260, 271, 276, 290, 301, 304, 312, 326, 332, 345, 350, 358, 369, 391, 417, 420, 422, 430, 431, 444, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456], "extend": [1, 213], "guid": [1, 6], "explain": 1, "how": [1, 4, 5, 301, 303, 304, 306, 307, 308, 312, 325, 326, 350, 369, 430, 447, 450, 455], "simpl": [1, 4, 5, 301, 312, 416, 422, 447, 448, 449, 451], "let": [1, 3, 4, 447, 449, 451, 452], "sai": [1, 4, 417, 451], "would": [1, 4, 369, 448, 450, 451, 452, 455], "like": [1, 4, 6, 165, 211, 283, 311, 397, 431, 433, 447, 448, 449, 451, 452, 453, 455], "them": [1, 4, 115, 301, 333, 345, 448, 455], "both": [1, 12, 85, 86, 87, 119, 120, 124, 155, 156, 165, 167, 168, 169, 172, 181, 188, 203, 205, 209, 215, 221, 233, 236, 260, 284, 303, 304, 320, 321, 325, 326, 422, 447, 448, 449, 453, 455], "coeffici": [1, 423, 424, 426, 427, 428, 429], "respect": [1, 3, 5, 132, 133, 152, 153, 154, 217, 276, 288, 301, 305, 313, 316, 320, 322, 420, 449, 453], "get": [1, 3, 5, 7, 97, 98, 107, 108, 192, 193, 194, 195, 223, 301, 447, 449, 451, 455], "z": [1, 315, 447, 451], "directli": [1, 4, 81], "import": [1, 2, 3, 4, 5, 7, 106, 157, 172, 242, 276, 287, 288, 289, 290, 291, 301, 303, 304, 305, 320, 325, 326, 335, 369, 389, 391, 398, 417, 420, 422, 447, 448, 449, 450, 451, 452, 453], "core": [1, 2, 3, 4, 5, 284, 301, 303, 304, 305, 320, 325, 326, 335, 338, 340, 343, 369, 370, 371, 372, 373, 374, 375, 376, 377, 389, 391, 398, 417, 420, 422, 447, 448, 452, 453], "mx": [1, 2, 3, 4, 5, 37, 94, 95, 106, 117, 148, 165, 172, 173, 176, 242, 276, 286, 301, 303, 304, 305, 320, 323, 325, 326, 328, 335, 339, 354, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 386, 389, 390, 391, 395, 398, 405, 415, 417, 420, 422, 444, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456], "def": [1, 3, 4, 5, 106, 276, 301, 420, 447, 448, 449, 450, 451, 452, 455], "simple_axpbi": 1, "while": [1, 2, 4, 7, 235, 356, 451, 452], "leav": [1, 127, 288, 289, 290], "howev": [1, 106, 301, 313, 316, 431, 444, 447, 448, 451, 452], "mai": [1, 172, 284, 310, 448, 449, 450], "need": [1, 4, 5, 6, 7, 80, 217, 301, 346, 347, 360, 368, 444, 448, 449, 451, 452, 453, 455], "underli": [1, 278], "perhap": [1, 4], "make": [1, 2, 4, 5, 7, 186, 208, 245, 301, 436, 437, 439, 440, 447, 451, 453, 455], "faster": [1, 120, 379, 389, 447, 449], "differenti": [1, 6], "tutori": 1, "go": [1, 4, 449], "through": [1, 258, 368, 429, 447, 449, 452], "It": [1, 4, 7, 154, 245, 276, 286, 290, 301, 347, 351, 430, 442, 452, 454], "cover": 1, "structur": [1, 430, 449], "librari": [1, 7, 296, 301], "redirect": 1, "acceler": [1, 305], "appropri": [1, 447], "metal": [1, 6], "vjp": [1, 106, 453], "jvp": [1, 106, 453], "graph": [1, 4, 5, 6, 449], "rule": [1, 422], "evalu": [1, 4, 5, 6, 127, 166, 279, 301, 331, 342, 420, 422, 447, 453], "discuss": 1, "more": [1, 2, 5, 9, 76, 112, 152, 170, 171, 174, 186, 198, 199, 240, 241, 294, 301, 305, 310, 356, 360, 368, 369, 371, 372, 373, 374, 389, 444, 447, 448, 449, 450, 453, 455], "detail": [1, 9, 198, 301, 310, 356, 360, 369, 371, 372, 373, 374, 423, 425, 426, 428, 429, 450, 453], "front": 1, "thei": [1, 3, 4, 7, 15, 99, 153, 160, 358, 393, 420, 429, 446, 447, 448, 451, 453, 454, 455], "defin": [1, 3, 4, 5, 7, 106, 153, 172, 218, 284, 287, 452], "api": [1, 448, 449], "axpbi": 1, "follow": [1, 4, 5, 6, 7, 17, 99, 110, 152, 172, 217, 289, 301, 380, 381, 395, 423, 424, 425, 426, 427, 428, 429, 435, 444, 447, 448, 449, 455], "numpi": [1, 4, 5, 6, 12, 15, 17, 85, 86, 87, 89, 119, 120, 124, 155, 156, 160, 167, 168, 169, 181, 186, 188, 203, 205, 209, 215, 233, 236, 260, 451, 453, 454], "style": [1, 12, 15, 85, 86, 87, 119, 120, 124, 155, 156, 160, 167, 168, 169, 181, 186, 188, 203, 205, 209, 215, 233, 236, 260], "upcast": 1, "factor": [1, 13, 157, 170, 173, 369, 391, 437, 440], "stream": [1, 6, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 108, 109, 110, 111, 112, 119, 120, 121, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 224, 225, 226, 227, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 277, 278, 281, 282, 283, 448, 455], "schedul": [1, 199, 422, 436, 437, 438, 439, 440, 442, 455], "simplest": [1, 301, 448], "wai": [1, 4, 7, 301, 369, 447, 448, 449, 450], "term": [1, 392, 423, 424, 425, 426, 427, 428, 434], "exist": [1, 2, 4, 333, 345], "themselv": [1, 447], "contain": [1, 4, 7, 27, 28, 67, 92, 112, 122, 142, 143, 144, 152, 153, 172, 182, 183, 184, 217, 252, 281, 286, 301, 332, 334, 335, 341, 368, 398, 417, 420, 447, 448, 449], "act": [1, 397], "nor": [1, 154, 276], "rather": [1, 449, 455], "easi": [1, 301, 448], "interfac": 1, "part": [1, 449, 450], "further": [1, 7, 449], "ha": [1, 2, 4, 5, 6, 7, 76, 92, 112, 142, 143, 145, 146, 147, 154, 170, 171, 174, 190, 194, 221, 305, 315, 321, 324, 353, 420, 422, 447, 450, 451, 453, 455], "method": [1, 4, 8, 9, 10, 29, 106, 114, 284, 292, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 339, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 420, 423, 424, 425, 426, 427, 428, 429, 431, 434, 435, 442], "run": [1, 2, 4, 5, 6, 7, 8, 212, 292, 305, 328, 423, 424, 426, 427, 428, 447, 448, 451, 455, 456], "our": [1, 4, 5, 358, 423, 424, 425, 426, 428, 429, 448], "concret": [1, 315, 321, 324, 353, 451, 455], "class": [1, 4, 5, 8, 9, 10, 29, 106, 114, 292, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 420, 423, 424, 425, 426, 427, 428, 429, 434, 435, 442], "public": [1, 301], "explicit": [1, 431, 444, 452], "alpha_": 1, "beta_": 1, "know": [1, 4], "itself": [1, 284, 431], "popul": 1, "avoid": [1, 340, 447], "unnecessari": [1, 4], "alloc": [1, 194, 198, 199, 420], "respons": 1, "void": 1, "eval_cpu": 1, "overrid": [1, 123], "eval_gpu": 1, "jacobian": [1, 166, 279, 453], "primal": [1, 106, 166, 279], "argnum": [1, 154, 276, 449], "cotan": [1, 106], "across": [1, 316, 448], "repres": [1, 4, 114, 117, 153, 398, 402, 452], "virtual": 1, "vmap": [1, 106, 449, 451, 453], "print": [1, 3, 4, 5, 7, 286, 287, 288, 289, 291, 301, 444, 447, 448, 449, 450, 451, 452, 453], "ostream": 1, "o": [1, 7, 135, 321], "is_equival": 1, "privat": 1, "fall": [1, 106], "eval": [1, 2, 3, 4, 5, 301, 420, 422, 447, 448, 449, 451, 453], "deriv": [1, 449, 451], "pass": [1, 4, 5, 64, 77, 212, 213, 276, 285, 287, 288, 289, 301, 333, 345, 346, 347, 358, 447, 448, 451], "reimplement": 1, "now": [1, 4, 7, 351, 447, 448, 452], "promot": 1, "promoted_dtyp": 1, "promote_typ": 1, "out_dtyp": 1, "is_floating_point": 1, "cast": [1, 36, 145, 146, 147, 176, 328, 340, 452], "up": [1, 4, 106, 447], "x_cast": 1, "y_cast": 1, "broadcasted_input": 1, "out_shap": 1, "unique_ptr": 1, "make_shar": 1, "to_stream": 1, "handl": [1, 301, 447], "resolv": 1, "No": [1, 4], "happen": [1, 4, 132, 368, 422, 447, 451], "call": [1, 2, 4, 5, 30, 150, 191, 195, 301, 312, 333, 345, 350, 358, 420, 422, 431, 447, 448, 449, 451], "alon": [1, 452], "onli": [1, 4, 6, 7, 80, 88, 97, 98, 172, 217, 225, 278, 301, 332, 333, 335, 340, 342, 345, 346, 347, 420, 447, 448, 449, 454, 455], "execut": [1, 7, 82, 83, 84, 195, 452, 455], "devic": [1, 6, 7, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 107, 108, 109, 110, 111, 112, 119, 120, 121, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 192, 199, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 224, 225, 226, 227, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 277, 278, 281, 282, 283, 292, 455, 456], "user": [1, 4, 301], "naiv": [1, 449], "version": [1, 7, 110, 181, 185, 217, 250, 280, 444, 449, 450], "declar": 1, "member": [1, 301, 338, 343], "earlier": 1, "find": [1, 3, 7, 448], "captur": [1, 2, 92, 200, 201, 301, 447], "axpby_impl": 1, "readi": 1, "malloc_or_wait": 1, "synchron": [1, 447], "avail": [1, 3, 4, 5, 7, 9, 118, 196, 296, 455], "There": [1, 301, 369, 447], "wait": [1, 4, 199], "here": [1, 4, 422, 447, 449, 451, 454, 455], "request": 1, "under": [1, 172], "pressur": 1, "set_data": 1, "nbyte": 1, "collect": [1, 288, 289, 446], "pointer": 1, "x_ptr": 1, "y_ptr": 1, "out_ptr": 1, "relev": 1, "static_cast": 1, "out_idx": 1, "map": [1, 5, 37, 176, 288, 312, 328], "x_offset": 1, "elem_to_loc": 1, "y_offset": 1, "contigu": [1, 81], "regularli": 1, "default": [1, 7, 13, 14, 15, 16, 17, 25, 26, 27, 28, 80, 81, 88, 92, 93, 96, 97, 98, 106, 107, 108, 110, 111, 112, 115, 116, 117, 131, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 152, 153, 154, 157, 158, 160, 170, 171, 172, 173, 174, 175, 176, 185, 187, 189, 190, 198, 199, 202, 206, 210, 214, 216, 217, 218, 220, 221, 222, 224, 225, 226, 227, 229, 230, 231, 234, 235, 237, 244, 245, 251, 252, 255, 256, 257, 259, 261, 263, 268, 270, 271, 272, 273, 274, 275, 276, 277, 280, 282, 284, 294, 303, 304, 305, 306, 307, 308, 314, 315, 317, 320, 321, 323, 324, 325, 326, 328, 333, 335, 340, 342, 345, 348, 349, 350, 351, 353, 356, 360, 364, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 382, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 420, 423, 424, 425, 426, 427, 428, 429, 434, 435, 436, 444, 446, 447, 449, 452, 454, 456], "doesn": [1, 301], "should": [1, 3, 4, 5, 7, 81, 112, 115, 132, 133, 135, 166, 191, 200, 217, 265, 271, 276, 279, 284, 287, 301, 306, 307, 308, 310, 311, 342, 348, 358, 391, 393, 398, 420, 446, 447, 448, 449, 451, 452, 456], "work": [1, 2, 4, 7, 199, 447, 448, 449, 450, 451], "incom": 1, "accordingli": 1, "dispatch": 1, "float16": [1, 10, 157, 176, 294, 328, 451, 452], "bfloat16": [1, 10, 157, 294, 452], "complex64": [1, 294], "throw": [1, 92, 117], "encount": [1, 449], "unexpect": [1, 17], "correct": [1, 7, 426, 427, 428, 450, 451], "float16_t": 1, "bfloat16_t": 1, "complex64_t": 1, "runtime_error": 1, "support": [1, 4, 6, 7, 15, 88, 135, 148, 157, 160, 170, 171, 173, 174, 176, 186, 217, 225, 448, 449, 450, 452, 454], "good": [1, 7, 422, 447, 448, 455], "fallback": 1, "routin": 1, "framework": [1, 6], "certain": [1, 342, 447], "case": [1, 4, 117, 138, 141, 142, 144, 145, 146, 147, 148, 170, 171, 173, 174, 186, 235, 255, 304, 310, 326, 366, 383, 401, 407, 413, 414, 430, 431, 447, 449, 453, 454, 455, 456], "half": [1, 17, 227, 231, 356, 451], "fix": [1, 4, 7, 451], "direct": [1, 4, 330, 429, 455], "column": [1, 131, 158, 217], "place": [1, 4, 37, 237, 284, 448, 451, 452], "expect": [1, 4, 306, 307, 308, 309, 310, 311, 360, 368, 392, 447, 450], "catlas_saxpbi": 1, "axpby_impl_acceler": 1, "special": 1, "copy_inplac": 1, "copytyp": 1, "inplac": 1, "incx": 1, "inci": 1, "fit": [1, 217, 455], "criteria": 1, "With": 1, "mind": [1, 4], "finish": 1, "assert": 1, "singl": [1, 5, 127, 166, 176, 190, 213, 279, 304, 326, 447, 450, 454], "flag": [1, 7, 447, 452], "row_contigu": 1, "col_contigu": 1, "common": [1, 422, 447, 451], "just": [1, 5, 322, 447, 450], "much": [1, 4, 303, 304, 325, 326, 447, 451], "enough": [1, 451], "plan": [1, 447], "enjoi": 1, "speed": 1, "appl": [1, 4, 6, 7, 455], "silicon": [1, 4, 6, 7, 455], "address": 1, "shade": 1, "languag": 1, "kernel": [1, 7, 96, 97, 98, 303, 304, 325, 326, 447, 450], "written": 1, "help": [1, 4, 447, 455], "resourc": 1, "walkthrough": 1, "pipelin": 1, "document": [1, 2, 64, 77, 240, 241, 294, 447, 449, 450], "specif": [1, 7, 448, 449], "cpp": 1, "keep": [1, 14, 16, 25, 26, 185, 187, 189, 202, 216, 257, 261, 277, 301, 332, 449, 451], "launch": [1, 117, 448, 450], "exactli": [1, 4, 335, 449], "mani": [1, 81, 252, 306, 307, 308, 312, 350, 447, 448, 451], "thread": 1, "pick": 1, "axpby_gener": 1, "buffer": [1, 193, 452], "x_stride": 1, "6": [1, 4, 172, 242, 355, 368, 372, 380, 381, 385, 392, 402, 406, 434, 447, 450, 453], "y_stride": 1, "7": [1, 4, 172, 217, 450], "uint": 1, "thread_position_in_grid": 1, "instanti": [1, 5, 451], "give": [1, 4, 5, 27, 447], "uniqu": [1, 444], "host": 1, "name": [1, 153, 176, 217, 218, 240, 241, 242, 243, 301, 316, 332, 335, 337, 448, 450, 454], "so": [1, 4, 7, 154, 157, 276, 309, 369, 422, 447, 448, 451, 455], "identifi": [1, 287, 446], "instantiate_axpbi": 1, "type_nam": 1, "host_nam": 1, "axpby_general_": 1, "set": [1, 4, 5, 7, 92, 106, 113, 115, 116, 117, 123, 132, 134, 192, 198, 199, 244, 245, 259, 313, 322, 324, 331, 333, 340, 341, 342, 345, 346, 351, 356, 366, 390, 402, 414, 420, 422, 424, 431, 444, 449, 451], "grid": [1, 190], "shown": 1, "below": [1, 7, 172, 273, 275, 294, 369, 451], "prepar": [1, 4], "carri": 1, "ostringstream": 1, "kname": 1, "axpby_": 1, "general_": 1, "type_to_nam": 1, "sure": [1, 2, 4, 7, 301, 447], "look": [1, 4, 448], "folder": 1, "register_librari": 1, "mlx_ext": 1, "get_colocated_mtllib_path": 1, "get_kernel": 1, "str": [1, 99, 121, 122, 154, 172, 176, 190, 192, 200, 239, 240, 241, 242, 243, 276, 287, 291, 328, 329, 332, 333, 335, 337, 339, 345, 369, 373, 374, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402], "encod": [1, 134, 356, 360, 368, 391], "compute_encod": 1, "get_command_encod": 1, "setcomputepipelinest": 1, "regist": [1, 5], "those": [1, 4, 301], "nelem": 1, "set_input_arrai": 1, "set_output_arrai": 1, "setbyt": 1, "sizeof": 1, "threadgroup": 1, "higher": [1, 159, 398, 449], "than": [1, 4, 76, 99, 112, 120, 134, 152, 155, 156, 168, 169, 170, 171, 174, 186, 198, 286, 288, 356, 366, 369, 398, 401, 414, 424, 429, 447, 449, 455], "tgp_size": 1, "maxtotalthreadsperthreadgroup": 1, "mtl": 1, "group_dim": 1, "grid_dim": 1, "among": 1, "dispatchthread": 1, "few": [1, 4, 5, 6, 7, 448, 451, 453], "thing": [1, 4, 448], "about": [1, 4, 5, 122, 192, 451, 455], "befor": [1, 4, 7, 27, 214, 332, 368, 431, 448, 450, 451], "track": [1, 301, 305], "activ": [1, 7, 193, 310, 366, 368, 383, 403, 413, 414, 416, 447], "command_buff": 1, "mtlcommandbuff": 1, "associ": [1, 242, 243, 451], "reli": 1, "u": [1, 170, 174, 324, 347, 442, 451], "command": [1, 2, 7, 448], "instead": [1, 7, 106, 301, 347, 360, 448, 449, 451], "end_encod": 1, "until": [1, 451, 453], "hit": 1, "flush": 1, "next": [1, 4, 5, 198], "These": [1, 92, 265, 391, 455], "built": [1, 7, 451], "top": [1, 270, 324, 369], "includ": [1, 102, 103, 104, 105, 193, 194, 199, 322, 329, 341, 351, 392, 422, 447, 449, 450, 453, 454, 456], "forward": [1, 276, 447, 451], "mode": [1, 99, 331, 342, 344, 369, 373, 374, 448], "diff": 1, "push": 1, "op": [1, 212, 278, 333, 451], "similarli": [1, 7, 186, 449, 451], "scale_arr": 1, "contribut": 1, "tangent_x": 1, "tangent_i": 1, "cotang": [1, 106, 279], "unus": 1, "arg": [1, 4, 9, 10, 114, 127, 242, 243], "push_back": 1, "fulli": [1, 6, 447, 448, 452, 455], "overal": 1, "directori": [1, 4, 7], "h": [1, 96, 97, 172, 304, 305, 307, 308, 310, 311, 315, 321, 326, 353, 449, 451], "mlx_sample_extens": 1, "__init__": [1, 4, 5, 8, 9, 10, 29, 106, 114, 292, 301, 420], "py": [1, 4, 7, 448], "cmakelist": 1, "txt": 1, "setup": [1, 3, 5, 7, 447], "packag": [1, 3, 5, 296, 417, 448], "hold": [1, 4, 9, 10, 172, 447], "instal": 1, "nanobind": [1, 7, 368], "sinc": [1, 4, 5, 153, 195, 420, 429, 438, 452, 455], "compon": [1, 4], "etc": [1, 217, 301, 369, 448], "alreadi": [1, 2, 4, 448], "nb_modul": 1, "_ext": 1, "doc": [1, 5, 448], "sampl": [1, 3, 4, 175, 220, 221, 222, 224, 225, 227, 230, 231, 371, 372, 373, 374, 376, 377, 392, 398, 402, 444, 447], "_a": 1, "nb": 1, "kw_onli": 1, "none": [1, 4, 8, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 109, 110, 111, 112, 113, 115, 116, 119, 120, 121, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 197, 200, 201, 202, 203, 204, 205, 206, 207, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 241, 242, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 280, 281, 282, 283, 284, 287, 288, 289, 290, 292, 303, 304, 313, 325, 326, 328, 332, 333, 340, 345, 348, 353, 360, 368, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 424, 442, 450], "r": [1, 4, 173, 276, 310, 315], "most": [1, 221, 301, 433, 447, 448, 449, 450, 451], "complex": [1, 94, 95, 143, 144, 145, 146, 147, 287, 294, 301, 347, 447, 449], "abov": [1, 4, 217, 274, 301, 369, 427, 448, 449, 450, 451, 455], "come": [1, 4, 448, 449], "bell": 1, "whistl": 1, "liter": [1, 369, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402], "modul": [1, 4, 5, 284, 285, 358, 368, 417, 433, 446, 447, 451], "caster": 1, "requir": [1, 4, 301, 448, 451, 452], "find_packag": 1, "config": [1, 448], "link": [1, 7], "your": [1, 4, 7, 420, 449, 451], "add_librari": 1, "target_sourc": 1, "cmake_current_list_dir": 1, "header": 1, "target_include_directori": 1, "target_link_librari": 1, "attach": 1, "conveni": [1, 5, 165], "mlx_build_metallib": 1, "metallib": [1, 7], "target": [1, 276, 389, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 447], "automat": [1, 6, 176, 448, 453, 454, 455], "what": [1, 4, 288], "practic": [1, 447], "mlx_build_met": [1, 7], "mlx_ext_metallib": 1, "titl": 1, "include_dir": 1, "project_source_dir": 1, "mlx_include_dir": 1, "output_directori": 1, "cmake_library_output_directori": 1, "add_depend": 1, "endif": 1, "final": [1, 3, 4, 5, 157, 436, 439], "nanobind_add_modul": 1, "nb_static": 1, "stable_abi": 1, "lto": 1, "nomins": 1, "nb_domain": 1, "build_shared_lib": [1, 7], "target_link_opt": 1, "wl": 1, "rpath": 1, "loader_path": 1, "onc": [1, 7, 447], "describ": [1, 451], "util": [1, 4, 6, 7, 242, 301, 422, 448], "__name__": [1, 4], "__main__": [1, 4], "descript": [1, 4, 294], "ext_modul": 1, "cmakeextens": 1, "cmdclass": 1, "build_ext": 1, "cmakebuild": 1, "package_data": 1, "dylib": 1, "extras_requir": 1, "dev": [1, 7], "zip_saf": 1, "python_requir": 1, "even": [1, 4, 92, 447, 451, 452], "though": [1, 4, 447, 451, 452], "pip": [1, 7], "develop": [1, 7], "j8": 1, "libmlx_ext": 1, "cpython": 1, "3x": 1, "darwin": 1, "try": [1, 7], "after": [1, 4, 5, 27, 148, 150, 153, 191, 214, 217, 305, 316, 322, 328, 329, 333, 335, 342, 345, 346, 347, 348, 368, 401, 447, 455], "abl": [1, 217], "simpli": [1, 4, 7, 323, 354, 378, 386, 405, 415, 420, 447, 448, 449], "plai": [1, 4], "quick": [1, 6], "benchmark": [1, 447], "see": [1, 4, 5, 7, 9, 10, 31, 32, 33, 34, 35, 38, 39, 40, 41, 42, 43, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 172, 198, 240, 241, 284, 294, 301, 305, 310, 313, 317, 318, 319, 331, 349, 350, 351, 354, 355, 356, 357, 359, 360, 361, 362, 363, 364, 365, 367, 369, 371, 372, 373, 374, 379, 380, 381, 407, 447, 448, 449, 450, 453, 455], "compar": [1, 80, 447], "time": [1, 4, 7, 199, 269, 301, 303, 304, 315, 321, 325, 326, 353, 447, 449, 451, 455], "set_default_devic": 1, "256": [1, 5], "512": [1, 2, 4, 368, 455], "random": [1, 2, 3, 4, 5, 6, 303, 304, 305, 320, 325, 326, 335, 342, 447, 449, 455, 456], "normal": [1, 3, 4, 132, 133, 225, 230, 301, 303, 304, 305, 316, 320, 322, 325, 326, 352, 368, 371, 373, 452, 455], "bench": 1, "warm": [1, 447], "100": [1, 3, 4, 389, 439, 447, 449, 451, 455], "5000": 1, "e": [1, 5, 7, 106, 125, 152, 153, 166, 246, 305, 306, 307, 308, 310, 311, 316, 320, 322, 333, 352, 387, 388, 410, 411, 416, 422, 425, 447, 451, 456], "simple_tim": 1, "custom_tim": 1, "3f": [1, 5, 447], "114": 1, "109": 1, "modest": 1, "improv": [1, 2, 4, 389, 423, 424, 425, 426, 427, 428, 434, 447, 448], "awai": [1, 4], "nn": [1, 4, 5, 242, 288, 301, 417, 420, 422, 431, 433, 447, 451], "grad": [1, 3, 5, 276, 286, 422, 430, 447, 448, 449, 450, 451, 453], "profil": 2, "kei": [2, 4, 135, 192, 220, 221, 222, 224, 225, 226, 227, 229, 230, 231, 287, 288, 332, 333, 345, 348, 431, 444, 446, 449], "optim": [2, 3, 5, 6, 346, 447, 448, 449, 451], "build": [2, 4, 6, 373, 420, 447], "mlx": [2, 3, 4, 5, 7, 296, 301, 417, 420, 422, 444, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455], "mlx_metal_debug": [2, 7], "debug": [2, 448], "record": [2, 195, 451], "dure": [2, 92, 309, 310, 311, 369, 452], "later": [2, 7, 422], "inspect": [2, 447, 453], "label": [2, 3, 391, 398], "object": [2, 9, 29, 49, 76, 92, 165, 242, 280, 287, 288, 289, 290, 294, 310, 368, 446], "queue": 2, "readabl": 2, "enabl": [2, 4, 7, 92, 123, 435], "python": [2, 4, 49, 67, 76, 127, 287, 288, 289, 290, 291, 420, 430, 431, 433, 446, 448, 449, 452], "prepend": [2, 186], "cmake_arg": 2, "dmlx_metal_debug": 2, "ON": [2, 7], "start_captur": 2, "initi": [2, 3, 4, 117, 290, 301, 305, 316, 320, 322, 324, 349, 352, 370, 371, 372, 373, 374, 375, 376, 377, 420, 431, 436, 437, 439, 440, 447, 448, 451], "gpu": [2, 6, 7, 192, 447, 450, 455], "applic": [2, 7], "mtl_capture_en": 2, "uniform": [2, 301, 324, 335, 372, 374, 417, 444, 447, 449, 455], "trace_fil": 2, "mlx_trace": 2, "gputrac": [2, 200], "path": [2, 7, 122, 200, 242, 243, 284, 289, 335, 448], "_": [2, 3, 4, 289, 301, 436, 437, 438, 439, 440, 444, 447, 451, 455], "stop_captur": 2, "open": [2, 7, 17, 227, 231], "replai": 2, "great": 2, "overview": 2, "oper": [2, 4, 6, 8, 36, 82, 83, 84, 98, 135, 152, 153, 215, 250, 258, 265, 292, 301, 368, 429, 447, 448, 449, 450, 451, 452, 453, 455, 456], "checkout": [2, 447], "inform": [2, 4, 5, 7, 122, 192, 240, 241, 294, 301, 305, 313, 348, 449, 455], "skip": [2, 81], "save": [2, 4, 6, 176, 200, 217, 240, 241, 242, 243, 339, 451], "project": [2, 4, 348], "cmake": [2, 7], "mkdir": [2, 7], "cd": [2, 7], "g": [2, 7, 106, 172, 217, 321, 416, 434, 435, 451, 456], "xcodeproj": 2, "metal_captur": 2, "schema": 2, "basic": [3, 237, 449], "model": [3, 5, 6, 242, 284, 285, 288, 289, 301, 328, 331, 333, 335, 339, 342, 344, 345, 346, 348, 368, 417, 420, 422, 430, 431, 433, 447, 448, 451], "learn": [3, 5, 6, 305, 316, 320, 322, 349, 352, 422, 423, 424, 425, 426, 427, 428, 429, 434, 435], "problem": [3, 5, 301], "metadata": [3, 176, 240, 241], "num_featur": [3, 305], "num_exampl": 3, "1_000": 3, "num_it": 3, "10_000": 3, "iter": [3, 5, 174, 288, 289, 444, 447, 451], "sgd": [3, 5, 422, 429, 431, 436, 437, 440, 447], "lr": [3, 429], "01": [3, 323, 386, 427], "rate": [3, 422, 423, 424, 425, 426, 427, 428, 429, 434, 435], "ll": [3, 5, 394, 447, 449], "synthet": 3, "dataset": [3, 448, 451], "design": [3, 6, 444, 455], "ground": [3, 4, 391, 401], "truth": [3, 391, 401], "w_star": 3, "gaussian": [3, 313, 379, 380, 381, 392], "nois": 3, "noisi": 3, "ep": [3, 132, 133, 305, 316, 320, 322, 352, 390, 392, 402, 422, 423, 424, 425, 426, 427, 428, 434], "loss": [3, 5, 276, 301, 422, 447, 448, 449, 451], "loss_fn": [3, 5, 422, 447, 449], "grad_fn": [3, 447, 449], "randomli": [3, 4, 309, 310, 311], "Then": [3, 7], "repeatedli": 3, "verifi": [3, 7], "close": [3, 6, 7, 15, 160], "error_norm": 3, "5f": 3, "someth": [3, 4, 450], "00005": 3, "00364": 3, "complet": [3, 4, 7, 199, 346, 347, 449, 455], "github": [3, 5, 7, 447], "repo": [3, 5, 7, 447], "effici": [4, 6, 152, 310, 356, 451, 453], "larg": [4, 301, 348, 397, 447, 448, 451], "ish": 4, "transform": [4, 6, 106, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 157, 285, 301, 305, 316, 322, 324, 332, 333, 345, 351, 356, 450], "without": [4, 6, 258, 348, 416, 446, 447, 448, 451, 452, 455], "compromis": 4, "eas": 4, "llama": 4, "famili": 4, "200": [4, 438], "line": [4, 448, 451, 452], "neural": [4, 6, 312, 371, 372, 403, 417, 420, 434], "network": [4, 6, 305, 310, 312, 371, 372, 417, 420, 434, 448], "concis": 4, "architectur": [4, 7, 192, 301, 347, 455], "notabl": [4, 6], "rope": [4, 301], "cach": [4, 7, 191, 193, 194, 198, 447], "llamaattent": 4, "self": [4, 5, 8, 29, 30, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 49, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 106, 292, 301, 403, 420], "dim": [4, 134, 135, 312, 316, 320, 322, 348, 350, 352, 356, 360, 368], "num_head": [4, 348, 368], "super": [4, 5, 301, 420], "tradit": [4, 134, 310, 311, 356], "query_proj": 4, "bia": [4, 110, 132, 153, 217, 218, 288, 301, 306, 307, 308, 315, 321, 322, 324, 333, 335, 345, 348, 351, 353, 426, 427, 428, 431, 449], "key_proj": 4, "value_proj": 4, "out_proj": [4, 420], "__call__": [4, 5, 301, 420], "queri": [4, 135, 348], "l": [4, 5, 170, 301, 303, 305, 306, 315, 321, 325, 353, 401], "combin": [4, 174, 290], "key_cach": 4, "value_cach": 4, "math": [4, 402, 447], "score": [4, 135, 398], "values_hat": 4, "possibli": [4, 13, 88, 152, 186, 286], "rm": [4, 7, 133, 424], "swiglu": 4, "rmsnorm": [4, 301], "llamaencoderlay": 4, "mlp_dim": [4, 368], "norm1": 4, "norm2": 4, "linear1": 4, "linear2": 4, "linear3": 4, "instanc": [4, 37, 106, 217, 291, 301, 320, 328, 329, 330, 333, 335, 336, 337, 342, 345, 346, 347, 358, 420, 452], "embed": [4, 284, 301, 350, 356, 360, 390], "emb": [4, 312, 350, 360], "token": [4, 312, 350], "num_lay": [4, 5, 422], "vocab_s": 4, "norm": [4, 133, 286, 316, 402, 428, 429], "multiheadattent": [4, 301], "create_additive_causal_mask": 4, "list": [4, 9, 14, 16, 29, 70, 76, 81, 82, 83, 84, 89, 92, 93, 98, 122, 127, 137, 138, 140, 141, 143, 144, 146, 147, 151, 154, 166, 172, 185, 187, 189, 190, 202, 210, 213, 216, 220, 221, 222, 224, 225, 226, 227, 230, 231, 240, 250, 252, 256, 257, 261, 268, 269, 272, 276, 277, 279, 282, 287, 290, 291, 301, 333, 335, 336, 337, 338, 343, 345, 346, 347, 420, 422, 426, 427, 428, 429, 438, 446, 447, 448, 449, 451], "still": [4, 7, 172, 447, 451], "consid": [4, 15, 80, 160, 287, 288, 289, 316, 446], "train": [4, 5, 301, 305, 309, 310, 311, 331, 333, 345, 371, 372], "process": [4, 98, 99, 114, 115, 116, 117, 288, 289, 311, 312, 368, 446, 448], "ignor": [4, 37, 91, 92, 127, 424], "whatsoev": 4, "rest": [4, 134, 288, 289, 356], "subsect": 4, "prompt": 4, "autoregress": 4, "yield": [4, 5, 444], "temp": 4, "causal": 4, "append": [4, 186, 447, 451], "store": 4, "per": [4, 5, 110, 153, 217, 218, 284, 305, 316, 320, 322, 352, 442, 447, 448, 451], "care": [4, 451], "logit": [4, 221, 389, 391, 447], "categor": 4, "lazili": [4, 301], "noth": [4, 301, 451], "yet": [4, 172, 301, 420, 431, 449, 450, 451, 453], "forc": [4, 5, 301, 448, 453], "choos": [4, 134, 356], "pars": 4, "feed": 4, "back": [4, 106, 196, 452], "loop": [4, 5, 447, 448, 449, 451], "unsqueez": 4, "sequenc": [4, 14, 16, 32, 33, 55, 56, 57, 58, 62, 70, 73, 74, 78, 81, 89, 98, 129, 137, 138, 140, 141, 143, 144, 146, 147, 151, 185, 187, 189, 202, 210, 216, 220, 221, 222, 224, 225, 226, 227, 230, 231, 235, 250, 252, 255, 257, 261, 268, 269, 272, 277, 282, 305, 306, 315, 321, 353, 368, 444, 455], "length": [4, 255, 305, 306, 315, 321, 353, 438], "len": [4, 138, 141, 144, 147, 157, 438], "overwrit": 4, "discard": [4, 287], "old": 4, "moment": [4, 424, 426, 427, 428], "anymor": 4, "everyth": [4, 448], "small": [4, 130, 132, 133, 305, 316, 322, 352, 392, 397, 402, 447, 448, 455], "12": [4, 157, 438], "8192": [4, 157], "1024": 4, "actual": [4, 17, 335, 420, 451], "materi": [4, 6], "could": [4, 301], "20_000": 4, "machin": [4, 6, 7, 434, 448], "8gb": 4, "ram": 4, "32": [4, 5, 88, 217, 218, 294, 304, 326, 352, 447], "44": 4, "bracket": 4, "becaus": [4, 193, 301, 451], "batch": [4, 13, 88, 152, 153, 186, 225, 305, 306, 307, 308, 310, 311, 315, 321, 348, 353, 369, 451], "zip": [4, 5], "haven": 4, "anyth": [4, 276, 451], "veri": [4, 348, 448, 451, 455], "similar": [4, 153, 165, 288, 346, 347, 348, 390, 452, 454], "runtim": [4, 117, 296, 447, 448], "section": [4, 7, 252, 402, 447, 448, 449], "origin": [4, 112, 286, 305, 341, 371, 372, 373, 374, 423, 424, 425, 426, 428, 429, 452], "sentencepiec": 4, "pytorch": [4, 6, 316, 449], "compat": [4, 221, 225, 454], "npz": [4, 176, 242, 243, 335, 339, 454], "file": [4, 7, 176, 239, 240, 241, 242, 243, 335, 339, 448, 449, 454], "argpars": 4, "itertool": [4, 288], "starmap": [4, 288], "np": [4, 5, 448, 452, 453], "torch": [4, 452], "map_torch_to_mlx": 4, "tok_embed": 4, "elif": 4, "attention_norm": 4, "ffn_norm": 4, "wq": 4, "wk": 4, "wv": 4, "wo": 4, "w1": [4, 286], "w2": [4, 286], "w3": 4, "ffn": 4, "separ": [4, 64, 77, 316, 398], "submodul": [4, 5, 301, 329, 333, 334, 345, 347], "feed_forward": 4, "parser": 4, "argumentpars": 4, "add_argu": 4, "torch_weight": 4, "output_fil": 4, "parse_arg": 4, "state": [4, 5, 301, 315, 321, 353, 422, 431, 444, 447], "savez": [4, 339, 454], "v": [4, 99, 135, 301, 333, 452], "disk": 4, "text": [4, 303, 304, 313, 315, 321, 325, 326, 327, 353, 366, 371, 372, 373, 374, 380, 383, 384, 385, 392, 393, 394, 397, 398, 401, 403, 404, 407, 408, 413, 414, 424, 429], "format": [4, 176, 239, 240, 241, 242, 243, 452], "dictionari": [4, 92, 176, 192, 240, 241, 286, 287, 290, 301, 332, 341, 346, 347, 432, 446, 454], "represent": [4, 217, 278, 287, 291], "tree_unflatten": [4, 422], "helper": [4, 447], "weight_fil": 4, "incur": [4, 7], "sever": [4, 7, 96, 97, 98, 242, 243, 447, 454], "futur": [4, 351, 450, 451], "pth": 4, "current": [4, 6, 7, 81, 88, 192, 194, 217, 290, 301, 424, 448, 451], "around": 4, "m1": [4, 447, 449, 455], "ultra": 4, "7b": 4, "me": 4, "ishmael": 4, "year": 4, "ago": 4, "never": [4, 451], "long": 4, "info": [4, 7], "247": 4, "press": [4, 172], "enter": 4, "littl": 4, "monei": 4, "my": [4, 7], "purs": 4, "consequ": 4, "walk": 4, "down": [4, 286], "gower": 4, "street": 4, "afternoon": 4, "heavi": 4, "rain": 4, "saw": [4, 449], "off": [4, 7, 451], "man": 4, "rag": 4, "who": 4, "sat": 4, "upon": [4, 288, 289], "hi": [4, 321], "bundl": 4, "hard": 4, "wet": 4, "he": [4, 373, 374], "were": [4, 455], "cry": 4, "watch": [4, 447], "him": 4, "observ": 4, "numer": [4, 132, 133, 172, 181, 185, 250, 305, 316, 320, 322, 352, 389, 390, 392, 402, 423, 424, 425, 426, 427, 428, 434, 447, 451], "crowd": 4, "wa": [4, 81, 448, 451], "hurri": 4, "437": 4, "330": 4, "second": [4, 7, 112, 167, 182, 184, 186, 236, 262, 271, 276, 304, 326, 390, 398, 424, 426, 427, 428, 449, 455], "spent": 4, "amount": [4, 195, 303, 325], "39": 4, "By": [4, 284, 340, 389, 449, 452], "bigger": [4, 424], "well": [4, 301, 333, 345, 348, 451], "almost": 4, "nobodi": 4, "took": 4, "least": [4, 82, 83, 84, 91, 170, 171, 173, 174, 217], "notic": [4, 449, 454], "distanc": [4, 402], "had": 4, "doubt": 4, "minut": 4, "straight": 4, "slowli": 4, "speak": [4, 172], "resum": 4, "postur": 4, "stood": 4, "feel": 4, "pain": 4, "heart": 4, "said": 4, "smile": 4, "face": 4, "am": 4, "someon": 4, "three": [4, 84, 369], "quarter": 4, "hour": 4, "made": [4, 296], "immedi": [4, 328], "repli": 4, "again": [4, 7, 301, 447], "hand": [4, 449, 451], "did": 4, "accustom": 4, "thu": [4, 301], "question": [4, 451], "reason": [4, 450], "tell": [4, 447, 452], "understand": [4, 371, 372], "579": 4, "690": 4, "500": [4, 455], "628": 4, "went": 4, "nervou": 4, "trembl": 4, "told": 4, "why": 4, "And": [4, 369], "surpris": 4, "matter": [4, 301], "shall": 4, "anyhow": 4, "friend": 4, "ye": 4, "slight": [4, 451], "kind": 4, "want": [4, 448, 449, 455], "longer": [4, 99, 449], "soon": 4, "unless": [4, 15, 160, 172, 420], "unlik": [4, 15, 160, 310, 311, 341], "strang": 4, "amus": 4, "That": 4, "secret": 4, "disappoint": 4, "mine": 4, "cannot": [4, 91, 450, 452], "happi": 4, "ask": [4, 448], "shop": 4, "bui": 4, "food": 4, "633": 4, "21": [4, 440], "475": 4, "su": 4, "j": [4, 7, 172, 310, 425, 426, 428], "lu": 4, "pan": 4, "murtadha": 4, "wen": 4, "liu": 4, "2021": 4, "roform": [4, 356], "enhanc": [4, 356, 451], "rotari": [4, 134, 356], "arxiv": [4, 316, 320, 322, 327, 352, 381, 403, 423, 429], "preprint": [4, 423, 429], "2104": 4, "09864": 4, "zhang": 4, "sennrich": 4, "2019": [4, 427], "advanc": [4, 447], "system": [4, 7, 192, 193, 194], "shazeer": 4, "2020": 4, "glu": [4, 301], "variant": [4, 401, 428], "2002": 4, "05202": 4, "classifi": 5, "mnist": 5, "As": [5, 37, 264, 301, 447], "mlp": [5, 301, 368, 422], "inherit": [5, 446], "idiom": [5, 447], "input_dim": [5, 301, 324, 351], "hidden_dim": [5, 420, 422], "output_dim": [5, 301, 324, 351], "layer_s": 5, "idim": 5, "odim": 5, "cross": [5, 98, 389, 391], "entropi": [5, 389, 391], "commonli": [5, 346, 417, 447], "cross_entropi": [5, 301], "accuraci": 5, "valid": [5, 99, 148, 280, 287, 333, 345, 446], "eval_fn": 5, "load": [5, 6, 296, 335, 448], "loader": 5, "num_class": [5, 422], "batch_siz": [5, 422], "num_epoch": [5, 422], "learning_r": [5, 422, 423, 424, 425, 426, 427, 428, 429, 431, 434, 435, 436, 437, 438, 439, 440, 447], "train_imag": [5, 422], "train_label": [5, 422], "test_imag": 5, "test_label": 5, "re": [5, 7, 417], "shuffl": 5, "minibatch": 5, "batch_iter": [5, 422], "perm": 5, "id": [5, 7], "put": [5, 447, 448], "trainabl": [5, 285, 301, 420], "loss_and_grad_fn": [5, 422, 447, 449], "value_and_grad": [5, 106, 301, 346, 420, 422, 433, 447, 449, 452, 453], "epoch": 5, "test": [5, 7, 448], "confus": 5, "decent": 5, "95": 5, "flexibl": 6, "brought": 6, "research": 6, "except": [6, 131, 142, 143, 145, 146, 147, 316, 335, 450, 452], "featur": [6, 96, 97, 98, 134, 305, 315, 316, 320, 321, 322, 324, 351, 352, 353, 356, 368, 369, 447, 451], "main": [6, 112, 131, 271, 288, 289, 301, 448], "differ": [6, 165, 260, 278, 401, 449], "compos": [6, 301, 447, 449, 453], "lazi": [6, 420, 453], "multi": [6, 135, 306, 307, 308, 450, 452], "cpu": [6, 7, 173, 447, 455], "inspir": 6, "jax": [6, 444], "arrayfir": 6, "unifi": 6, "live": [6, 455], "share": [6, 110, 153, 217, 218, 278, 448], "convers": 6, "distribut": [6, 7, 220, 221, 222, 224, 225, 226, 230, 231, 324, 371, 372, 373, 374, 376, 377, 392, 395, 400, 402, 417], "commun": [6, 114, 117, 118], "regress": [6, 397], "layer": [6, 132, 284, 301, 303, 304, 310, 311, 315, 316, 321, 322, 324, 325, 326, 342, 347, 350, 351, 353, 358, 368, 416, 420], "perceptron": 6, "llm": 6, "infer": [6, 151, 176, 271, 448], "fast": [6, 313, 381, 448, 455], "fft": 6, "algebra": 6, "tree": [6, 92, 127, 154, 276, 280, 287, 288, 289, 290, 291, 430, 431, 433, 442, 449], "custom": [6, 106, 368], "extens": [6, 176, 200, 339, 454], "debugg": 6, "pypi": 7, "own": [7, 452], "meet": 7, "seri": 7, "chip": 7, "nativ": 7, "maco": 7, "13": 7, "highli": 7, "recommend": [7, 199, 429], "14": 7, "sonoma": 7, "conda": [7, 448], "forg": 7, "match": [7, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 193, 335, 369, 391, 448, 450, 452], "probabl": [7, 227, 309, 310, 311, 351, 389, 391, 395, 455], "platform": 7, "processor": 7, "arm": 7, "i386": 7, "switch": 7, "17": 7, "clang": 7, "24": 7, "xcode": 7, "15": [7, 172, 290, 447], "sdk": 7, "environ": [7, 113, 123, 448], "via": [7, 106, 430, 433, 448, 451, 452], "rosetta": 7, "unam": 7, "p": [7, 220, 301, 309, 310, 311, 402, 426, 428], "clone": 7, "git": 7, "com": [7, 448], "ml": 7, "explor": 7, "http": [7, 316, 320, 322, 327, 352, 381, 403], "wjakob": 7, "2f04eac452a6d9142dedb957701bdb20125561e4": 7, "env": 7, "cmake_build_parallel_level": 7, "edit": [7, 347], "unittest": 7, "discov": 7, "stub": 7, "generate_stub": 7, "either": [7, 12, 64, 76, 77, 85, 86, 87, 91, 119, 120, 124, 150, 155, 156, 167, 168, 169, 172, 181, 186, 188, 203, 205, 209, 215, 233, 236, 260, 276, 304, 326, 358, 369, 373, 374, 452], "static": 7, "libmlx": 7, "preprocessor": 7, "metal_path": 7, "mlx_build_test": 7, "mlx_build_exampl": 7, "mlx_build_benchmark": 7, "mlx_build_cpu": 7, "mlx_build_python_bind": 7, "mlx_build_safetensor": 7, "mlx_build_gguf": 7, "mlx_metal_jit": 7, "wish": 7, "variabl": [7, 92, 113, 123, 154, 166, 276, 279, 280, 448], "export": 7, "developer_dir": 7, "app": 7, "content": [7, 332, 447], "xcrun": 7, "macosx": 7, "show": [7, 294, 447], "cmake_build_typ": 7, "minsizerel": 7, "don": [7, 447, 455], "backend": [7, 117, 118], "safetensor": [7, 176, 241, 335, 339, 422, 451, 454], "gguf": [7, 176, 240, 454], "dcmake_build_typ": 7, "dbuild_shared_lib": 7, "dmlx_build_cpu": 7, "dmlx_build_safetensor": 7, "dmlx_build_gguf": 7, "dmlx_metal_jit": 7, "THE": 7, "pre": [7, 135, 389], "substanti": 7, "cold": 7, "cost": [7, 424, 448, 451], "anwywher": 7, "hundr": 7, "millisecond": [7, 447, 455], "persist": 7, "accross": 7, "reboot": 7, "unabl": 7, "tool": 7, "sudo": 7, "ouptut": 7, "finder": 7, "iterm": 7, "termin": 7, "click": 7, "uncheck": 7, "window": [7, 303, 304, 325, 326], "restart": 7, "grep": 7, "cmake_host_system_processor": 7, "arm64": 7, "x86_64": 7, "wipe": 7, "cahc": 7, "rf": 7, "devicetyp": 8, "attribut": [8, 9, 29, 292, 341, 420, 442], "kwarg": [9, 10, 114, 242, 243, 456], "categori": [10, 294], "bool_": [10, 294], "unsignedinteg": 10, "uint8": [10, 294], "uint16": [10, 294], "uint32": [10, 25, 26, 27, 28, 221, 294], "uint64": [10, 294], "signedinteg": [10, 165], "int8": [10, 294], "int64": [10, 294], "inexact": [10, 165], "complexflo": 10, "complex128": 10, "issubdtyp": [10, 294], "semant": [12, 85, 86, 87, 89, 119, 120, 124, 155, 156, 167, 168, 169, 181, 186, 188, 203, 205, 209, 215, 233, 236, 260, 455], "reduct": [14, 16, 116, 185, 187, 202, 216, 290, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 448], "unspecifi": [14, 16, 17, 25, 26, 27, 28, 93, 102, 103, 104, 105, 151, 185, 187, 189, 202, 210, 214, 216, 234, 250, 251, 257, 261, 264, 270, 271, 277, 282, 456], "entir": [14, 16, 25, 26, 185, 187, 189, 202, 216, 257, 261, 277, 310, 311], "05": [15, 160, 305, 316, 320, 322, 352], "08": [15, 160, 390, 425, 426, 427, 428, 434], "approxim": [15, 313, 379, 380, 381], "comparison": [15, 124, 155, 156, 168, 169, 209], "infinit": [15, 160], "rel": [15, 160, 424, 447], "interv": [17, 175, 227, 231], "increment": 17, "otherwis": [17, 98, 117, 199, 284, 287, 288, 289, 290, 333, 335, 345, 366, 368, 369, 383, 389, 394, 401, 413, 414, 451, 452], "convent": [17, 99, 121, 122, 369, 427], "fraction": 17, "integr": [17, 264, 451], "dimension": [29, 132, 133, 136, 137, 138, 139, 140, 141, 145, 146, 147, 303, 304, 305, 306, 307, 308, 312, 324, 325, 326, 350, 351, 360, 450, 452], "ndarrai": [29, 450, 451, 453], "properti": [30, 37, 46, 50, 60, 61, 67, 69, 341, 344, 432, 449], "argument": [30, 64, 77, 92, 127, 154, 276, 288, 289, 290, 301, 369, 444, 448, 449, 454, 455, 456], "union": [31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 82, 83, 84, 88, 121, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 157, 163, 164, 192, 229, 230, 240, 259, 271], "regular": [37, 310, 403, 427, 447, 450], "idx": [37, 450], "correctli": 37, "syntax": [37, 450], "datatyp": 50, "byte": [50, 60, 193, 194, 195, 198, 199, 294], "indices_or_sect": [70, 252], "nest": [76, 92, 290, 301, 420, 446, 449], "alwai": [81, 193, 287, 449], "regardless": [81, 135], "storag": 81, "caution": 81, "chang": [81, 92, 245, 278, 346, 351, 369, 394, 401, 447, 452], "crash": [81, 447], "begin": [81, 195, 217, 304, 315, 321, 326, 366, 383, 394, 401, 407, 413, 414], "ari": [82, 83, 84], "xor": 87, "lhs_mask": 88, "lceil": 88, "rceil": 88, "rhs_mask": 88, "out_mask": 88, "edg": [91, 213, 369, 447], "At": 91, "fun": [92, 154, 166, 276, 279, 280, 447, 450, 451, 455], "callabl": [92, 106, 154, 166, 276, 279, 280, 284, 285, 287, 288, 289, 290, 328, 329, 332, 340, 353, 358, 368, 370, 371, 372, 373, 374, 375, 376, 377, 423, 424, 425, 426, 427, 428, 429, 434, 435, 436, 437, 438, 439, 440], "dict": [92, 127, 176, 192, 240, 241, 242, 286, 338, 343, 346, 347, 420, 422, 430, 431, 433, 446, 449, 454], "arbitrarili": [92, 301, 446, 449, 453], "leaf": [92, 284, 287, 288, 289, 290, 332], "node": [92, 127, 280, 289, 290], "recompil": [92, 447], "Not": [92, 209, 447], "attempt": 92, "elementwis": [94, 95], "alia": [94, 95], "conj": 95, "channel": [96, 97, 98, 305, 306, 307, 308, 310, 311], "c_in": [96, 97, 98], "c_out": [96, 97, 98], "convolv": [96, 97, 98], "spatial": [97, 98, 303, 316, 325, 369], "symmetr": [97, 170], "correl": [98, 310], "discret": [99, 136, 137, 138, 139, 140, 141, 145, 146, 147, 312, 350], "conv": 99, "signal": [99, 369], "th": [102, 103, 104, 105, 111, 131, 438], "definit": [106, 170, 225], "meant": 106, "decor": 106, "behav": 106, "wrap": [106, 301], "behaviour": [106, 170], "usag": [106, 368, 447], "f_vjp": 106, "f_jvp": 106, "dx": 106, "dy": 106, "f_vmap": 106, "ai": 106, "angl": [109, 219, 323], "configur": [110, 448], "formal": [110, 217], "notat": [110, 287, 337], "w_i": [110, 217], "hat": [110, 217], "occupi": [110, 153, 217, 218], "subarrai": [112, 252], "insert": [112, 129, 455], "end": [112, 196, 217, 304, 315, 321, 326, 366, 383, 394, 401, 407, 413, 414, 436, 439], "taken": [112, 264, 271], "global": [113, 115, 116, 117, 123, 228, 286, 444, 447], "disabl": [113, 198, 447], "mlx_disable_compil": [113, 123, 447], "independ": [114, 310, 311], "particip": [115, 116], "strict": [117, 155, 168, 333, 335, 345], "is_avail": 117, "fuction": 120, "subscript": [121, 122], "operand": [121, 122, 152], "einstein": [121, 122], "summat": [121, 122], "equat": [121, 122], "einsum": 122, "chosen": 122, "mathrm": [125, 246, 320], "frac": [125, 217, 246, 303, 304, 305, 309, 310, 311, 316, 320, 322, 324, 325, 326, 352, 371, 372, 373, 374, 390, 392, 394, 397, 408, 410, 411, 423, 425, 426, 427, 428, 434], "pi": [125, 313, 360, 380, 449], "int_0": 125, "dt": 125, "minu": 130, "whose": [131, 284, 285], "translat": [132, 322], "stabil": [132, 133, 305, 316, 320, 322, 352, 389, 390, 392, 423, 424, 425, 426, 427, 428, 434], "rotat": [134, 356], "larger": [134, 356, 429], "unchang": [134, 258, 356], "consecut": [134, 217, 356], "angular": [134, 356], "frequenc": [134, 356, 360], "q": [135, 173], "head": [135, 348, 368], "attent": [135, 333, 348, 360, 368], "One": [136, 139, 145, 238, 447, 449], "fourier": [136, 137, 138, 139, 140, 141, 145, 146, 147], "truncat": [136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 230], "dft": [136, 137, 138, 139, 140, 141, 145, 146, 147], "rfft": 142, "real": [142, 143, 144, 145, 146, 147, 170], "rfft2": 143, "rfftn": 144, "silent": [145, 146, 147], "outsid": 148, "clamp": 148, "explicitli": [152, 301, 444], "flat": [152, 153, 287, 291], "a1": 152, "a2": 152, "AS": 152, "b1": 152, "b2": 152, "pack": [153, 217, 218], "unsign": [153, 217, 218, 294], "whether": [153, 218, 315, 321, 332, 348, 353, 389, 392, 398], "argnam": [154, 276], "neither": [154, 276], "keyword": [154, 242, 243, 276, 288, 301, 444, 454, 456], "walsh": 157, "scipi": 157, "linalg": 157, "20": [157, 172], "28": 157, "16384": 157, "orthonorm": 157, "ordinari": 159, "inifn": 161, "dtypecategori": [165, 294], "subtyp": [165, 294], "subdtyp": 165, "float64": 165, "too": [165, 447, 451], "upper": [170, 217, 227, 230, 231, 377], "decomposit": [170, 174], "semi": [170, 225], "triangular": 170, "lower": [170, 217, 227, 230, 231, 273, 377], "trinagular": 170, "dot": [170, 171, 268, 287, 337, 348], "ainv": 171, "ord": 172, "tabl": [172, 294, 312], "frobeniu": 172, "matric": [172, 173, 174], "strictli": 172, "mathemat": 172, "variou": 172, "purpos": 172, "calcul": [172, 389, 392, 398, 424], "fro": 172, "inf": [172, 206, 348], "largest": [172, 206, 270], "sing": 172, "smallest": 172, "singular": [172, 174], "nuclear": 172, "_f": 172, "sum_": [172, 303, 304, 397], "a_": 172, "valueerror": [172, 335, 449], "refer": [172, 320, 327, 341, 371, 372, 373, 374, 381, 403, 450], "golub": 172, "van": 172, "loan": 172, "baltimor": 172, "md": 172, "john": 172, "hopkin": 172, "univers": 172, "1985": 172, "pg": 172, "la": 172, "9": [172, 391, 423, 426, 427, 428, 429, 431, 437, 440, 452], "74597": 172, "84804": 172, "41421": 172, "23607": [172, 173], "74166": 172, "24264": 172, "11": 172, "225": 172, "894427": 173, "447214": 173, "57771": 173, "vt": 174, "return_metadata": 176, "binari": [176, 239, 240, 241, 242, 243, 278, 366, 389, 414, 447], "npy": [176, 239, 454], "matadata": 176, "unsupport": 176, "tensor": [176, 268, 303, 304, 325, 326, 402, 452], "stabl": [181, 185, 250, 397], "multidimension": 190, "dens": [190, 455], "cartesian": 190, "ij": 190, "clear": 191, "get_cache_memori": 191, "max_buffer_s": 192, "max_recommended_working_set_s": 192, "memory_s": 192, "report": [193, 199], "peak": [195, 197], "program": [195, 448], "reset_peak_memori": 195, "reset": 197, "free": 198, "reclaim": 198, "set_memory_limit": 198, "previou": [198, 199], "relax": 199, "task": [199, 397], "exceed": 199, "potenti": 199, "beforehand": 212, "constant_valu": 213, "before_1": 213, "after_1": 213, "before_2": 213, "after_2": 213, "before_n": 213, "after_n": 213, "before_i": 213, "after_i": 213, "side": [213, 303, 304, 325, 326, 447], "everi": [217, 288, 422, 440, 449], "particular": [217, 316], "w_1": 217, "w_g": 217, "align": [217, 304, 315, 321, 326], "max_i": 217, "min_i": 217, "textrm": [217, 313, 314, 379, 382], "1st": 217, "signific": 217, "2nd": 217, "w_q": 217, "prng": [220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 444], "num_sampl": 221, "unnorm": [221, 389, 391], "draw": 221, "cdf": [222, 313, 379], "seed": 223, "loc": [224, 226], "cov": 225, "jointli": 225, "covari": [225, 305], "behavior": [225, 397, 450, 451], "empti": 225, "low": [227, 231, 377, 417], "high": [227, 231, 301, 312, 377, 417], "roadcast": 227, "domain": [230, 448], "uniformli": 231, "repetit": 234, "preserv": [235, 449], "obj": 240, "uncompress": 242, "my_path": 242, "tree_flatten": [242, 288, 291, 301, 422], "transformerencod": 242, "128": [242, 301], "flat_param": 242, "compress": 243, "possibl": [252, 312, 350, 447, 448, 450, 455], "divisor": [257, 277], "being": [258, 301], "prevent": [258, 402, 452], "streamcontext": 259, "context": 259, "manag": [259, 444, 448, 455], "prior": [264, 265], "exclud": 265, "necessarili": 270, "elsewher": [273, 450], "col": 273, "triangl": 273, "mse": 276, "param": [276, 301, 417, 449], "lvalu": 276, "dlvalu": 276, "dparam": 276, "lasso": 276, "l1": [276, 394, 396, 397, 401], "impli": 278, "gaurante": 278, "in_ax": [280, 449], "out_ax": [280, 449], "prefix": [280, 287], "class_pred": 284, "predic": [284, 340], "to_quant": 284, "receiv": [284, 438, 452], "fn": [285, 288, 289, 290, 453], "wrt": 285, "max_norm": 286, "exce": 286, "proportion": 286, "clipped_grad": 286, "total_norm": 286, "rescal": 286, "is_leaf": [287, 288, 289, 290], "arbitrari": [287, 420], "depth": [287, 308, 311, 449], "hello": [287, 291], "charact": 287, "superset": [288, 430], "extra": [288, 289], "closer": 288, "decid": [288, 332], "constitut": 288, "dict_kei": [288, 431], "lambda": [288, 289, 290, 301, 317, 328, 333, 340, 364, 383, 407, 413, 423, 424, 425, 426, 427, 428, 429, 434, 435, 447, 448, 449], "tree_map": [289, 301, 448], "new_tre": 289, "accumul": [290, 352], "acc": 290, "recreat": [291, 422], "world": [291, 448], "42": 291, "16": [294, 303, 320, 325, 328, 420], "int16": 294, "brain": 294, "e8": 294, "m7": 294, "ieee": 294, "e5": 294, "m10": 294, "hierarchi": 294, "mpi": 296, "done": [301, 309, 352, 447, 448, 451, 452], "manual": 301, "solv": 301, "intuit": 301, "freez": [301, 345, 420], "finetun": 301, "in_dim": [301, 420], "out_dim": [301, 420], "enumer": 301, "caus": [301, 447, 451], "local": [301, 310, 448], "scope": 301, "l2_loss": 301, "y_hat": 301, "trainable_paramet": [301, 332, 431], "loss_and_grad": 301, "workhors": 301, "Its": 301, "recurs": [301, 332, 333, 338, 343, 345, 420], "frozen": [301, 333, 343, 345, 351, 420], "individu": [301, 310, 311], "subset": [301, 332], "action": 301, "displai": 301, "count": [301, 438], "num_param": 301, "preclud": 301, "pure": [301, 422], "pattern": [301, 451], "achiev": [301, 448], "other_input": 301, "necessari": 301, "apply_to_modul": [301, 333], "children": 301, "filter_and_map": 301, "leaf_modul": 301, "load_weight": [301, 451], "named_modul": 301, "save_weight": 301, "set_dtyp": 301, "unfreez": [301, 333], "update_modul": 301, "alibi": 301, "avgpool1d": 301, "avgpool2d": 301, "batchnorm": 301, "dropout": [301, 310, 311, 342, 368, 447], "dropout2d": 301, "dropout3d": 301, "gelu": [301, 380, 381, 447], "groupnorm": 301, "gru": 301, "hardshrink": [301, 383], "hardtanh": [301, 384], "hardswish": 301, "instancenorm": 301, "layernorm": 301, "leakyrelu": 301, "lstm": 301, "maxpool1d": 301, "maxpool2d": 301, "mish": 301, "prelu": 301, "quantizedembed": 301, "quantizedlinear": 301, "relu": [301, 349, 368, 404, 417], "relu6": 301, "rnn": [301, 315], "selu": 301, "sequenti": [301, 417], "silu": 301, "sinusoidalpositionalencod": 301, "softmin": 301, "softshrink": 301, "softsign": 301, "softplu": [301, 327, 403], "upsampl": 301, "elu": [301, 407], "gelu_approx": [301, 313, 379], "gelu_fast_approx": [301, 313, 379], "hard_shrink": [301, 317], "hard_tanh": [301, 318], "leaky_relu": 301, "log_sigmoid": 301, "log_softmax": 301, "binary_cross_entropi": [301, 447], "cosine_similarity_loss": 301, "gaussian_nll_loss": 301, "hinge_loss": 301, "huber_loss": 301, "kl_div_loss": 301, "l1_loss": 301, "log_cosh_loss": 301, "margin_ranking_loss": 301, "mse_loss": 301, "nll_loss": 301, "smooth_l1_loss": 301, "triplet_loss": 301, "init": [301, 349, 417, 422, 436, 437, 439, 440, 448], "glorot_norm": 301, "glorot_uniform": 301, "he_norm": 301, "he_uniform": 301, "kernel_s": [303, 304, 306, 307, 308, 325, 326], "averag": [303, 304, 423, 424, 426, 427, 428, 448], "pool": [303, 304, 325, 326, 455], "l_": [303, 325, 394], "n_i": [303, 304, 325, 326], "c_j": [303, 304, 325, 326], "ldot": [303, 304, 325, 326], "lfloor": [303, 304, 325, 326], "_size": [303, 304, 325, 326], "rfloor": [303, 304, 325, 326], "k_h": [304, 326], "k_w": [304, 326], "h_": [304, 315, 321, 326, 353], "w_": [304, 315, 321, 326, 353, 423, 424, 425, 426, 427, 428, 429, 434, 435], "height": [304, 305, 307, 308, 310, 311, 326], "width": [304, 305, 307, 308, 310, 311, 326, 350, 351], "momentum": [305, 429, 431, 435, 447], "affin": [305, 316, 320, 322, 324, 351], "track_running_stat": 305, "epsilon": [305, 316, 320, 322, 352, 390, 392, 423, 425, 426, 427, 428, 434], "gamma": [305, 316, 320, 322, 352, 371, 372, 373, 374], "nc": 305, "nlc": [305, 306], "four": 305, "nhwc": [305, 307], "paper": [305, 360, 423, 424, 425, 426, 428, 429], "deep": [305, 371, 372, 373, 374], "intern": 305, "bn": 305, "in_channel": [306, 307, 308], "out_channel": [306, 307, 308], "learnabl": [306, 307, 308, 358], "imag": [307, 308, 310, 311, 369], "ndhwc": [308, 311], "portion": 309, "nwhc": 310, "whc": 310, "maintain": [310, 311, 429], "benefici": [310, 311, 451], "earli": 310, "adjac": 310, "pixel": 310, "effect": [310, 447, 451], "thompson": 310, "goroshin": 310, "jain": 310, "lecun": 310, "bregler": 310, "2015": [310, 426, 428], "cvpr": 310, "dhwc": 311, "often": 311, "medic": 311, "video": 311, "num_embed": [312, 350], "lookup": 312, "usual": [312, 350, 446, 451], "vocabulari": [312, 350], "approx": 313, "unit": [313, 314, 315, 323, 354, 355, 357, 359, 371, 372, 373, 374, 378, 379, 380, 381, 382, 386, 405, 406, 407, 409], "phi": [313, 379], "geluapprox": 313, "044715": [313, 380], "gelufast": 313, "sigma": [313, 314, 315, 321, 371, 372, 373, 374, 381, 382, 387, 408, 409], "773": 313, "regard": 313, "gate": [314, 315, 382], "halv": [314, 382], "input_s": [315, 321, 353], "hidden_s": [315, 321, 353], "recurr": [315, 321, 353], "nld": [315, 321, 353], "ld": [315, 321, 353], "r_t": 315, "xr": 315, "x_t": [315, 321, 353], "hr": 315, "h_t": [315, 321, 353], "b_": [315, 321], "z_t": 315, "xz": 315, "hz": 315, "n_t": 315, "xn": 315, "odot": [315, 321], "hn": 315, "hidden": [315, 321, 353, 368], "nh": [315, 321, 353], "nlh": [315, 321, 353], "lh": [315, 321, 353], "num_group": 316, "pytorch_compat": 316, "preced": 316, "org": [316, 320, 322, 327, 352, 381, 403], "1803": 316, "08494": 316, "lambd": [317, 364, 383, 413], "denomin": [320, 390, 423, 425, 426, 427, 428, 434], "inorm": 320, "1607": [320, 322], "08022": 320, "i_t": 321, "xi": 321, "f_t": 321, "xf": 321, "hf": 321, "g_t": [321, 423, 425, 426, 427, 428, 429, 434, 435], "xg": 321, "hg": 321, "o_t": 321, "xo": 321, "ho": 321, "c_": [321, 429], "c_t": [321, 429], "cell": 321, "06450": 322, "realli": 322, "negative_slop": [323, 386], "leaki": [323, 386], "rectifi": [323, 354, 355, 373, 374, 386, 405, 406], "slope": 323, "mathcal": 324, "d_i": 324, "max_": [325, 326], "1908": [327, 403], "08681": [327, 403], "map_fn": [328, 332], "filter_fn": [328, 332], "valid_parameter_filt": 328, "apply_fn": 329, "descend": 330, "is_leaf_fn": 332, "found": 332, "drop": 332, "idempot": [333, 345], "endswith": 333, "file_or_weight": 335, "miss": [335, 454], "ok": [335, 449], "save_safetensor": [339, 422, 454], "reflect": [341, 447, 450, 452], "ie": [345, 448], "noop": 345, "unfrozen": 345, "tracer": 346, "partial": [346, 347, 447, 451], "child": 347, "flexibli": 347, "programmat": 347, "query_input_dim": 348, "key_input_dim": 348, "value_input_dim": 348, "value_dim": 348, "value_output_dim": 348, "aggreg": 348, "linearli": 348, "attend": 348, "num_paramet": 349, "25": [349, 369], "parametr": [349, 404], "from_embed": 350, "classmethod": [350, 351], "from_linear": 351, "1910": 352, "07467": 352, "nonlinear": [353, 447], "elman": 353, "ih": 353, "hh": 353, "func": 353, "10000": 356, "slightli": [356, 455], "plain": 358, "known": [359, 409], "swish": [359, 409], "min_freq": 360, "0001": 360, "max_freq": 360, "cos_first": 360, "full_turn": 360, "sinusoid": 360, "threshold": [366, 394, 401, 414], "geq": [366, 414], "num_encoder_lay": 368, "num_decoder_lay": 368, "nb_func": 368, "custom_encod": 368, "custom_decod": 368, "norm_first": 368, "checkpoint": [368, 422], "decod": 368, "interact": 368, "mechan": 368, "chekpoint": 368, "expens": 368, "scale_factor": 369, "nearest": 369, "align_corn": 369, "audio": 369, "4d": 369, "forth": 369, "algorithm": [369, 429], "neighbor": 369, "interpol": 369, "cubic": 369, "bilinear": 369, "trilinear": 369, "bicub": 369, "corner": 369, "bottom": 369, "75": 369, "33333": 369, "66667": 369, "init_fn": [370, 371, 372, 373, 374, 375, 376, 377, 417], "glorot": [371, 372], "fan_in": [371, 372, 373, 374], "fan_out": [371, 372, 373, 374], "fan": [371, 372, 373, 374], "_in": [371, 372], "_out": [371, 372], "difficulti": [371, 372], "feedforward": [371, 372], "191107": 371, "61278": 371, "150594": 371, "363207": 371, "gain": [371, 372, 373, 374], "89613": 371, "53947": 371, "48095": 371, "995016": 371, "223404": 372, "890597": 372, "379159": 372, "776856": 372, "90041": 372, "02264": 372, "912766": 372, "12451": 372, "delv": [373, 374], "surpass": [373, 374], "human": [373, 374], "imagenet": [373, 374], "classif": [373, 374], "25211": 373, "458835": 373, "177208": 373, "0137595": 373, "6967": 373, "02765": 373, "15268": 373, "75787": 373, "kaim": 374, "0300242": 374, "0184009": 374, "793615": 374, "666329": 374, "64331": 374, "16506": 374, "08619": 374, "79854": 374, "982273": 376, "534422": 376, "380709": 376, "0645099": 376, "883935": 377, "863726": 377, "617261": 377, "417497": 377, "exact": [380, 381], "0005": 380, "015": 381, "702": 381, "cdot": [381, 390, 393, 409], "hendryck": 381, "1606": 381, "08415": 381, "min_val": 384, "max_val": 384, "_val": 384, "sum_i": 388, "x_i": [388, 410, 411], "with_logit": 389, "predict": [389, 392, 393, 394, 395, 396, 397, 399, 400, 401], "105361": 389, "223144": 389, "20397": 389, "916291": 389, "539245": 389, "prob": 389, "510826": 389, "x1": 390, "x2": 390, "x_1": [390, 398], "x_2": [390, 398], "label_smooth": 391, "hot": 391, "smooth": [391, 401, 434], "0485873": 391, "348587": 391, "06": [392, 402, 423], "likelihood": [392, 400], "nll": [392, 400], "hing": 393, "y_": [393, 397], "pred": [393, 397], "delta": [394, 423], "huber": 394, "leq": [394, 407], "l2": [394, 397, 435], "kullback": 395, "leibler": 395, "diverg": 395, "logcosh": 397, "sensit": 397, "outlier": 397, "dual": 397, "offer": 397, "balanc": 397, "robust": 397, "approach": [397, 449], "inputs1": 398, "inputs2": 398, "margin": [398, 402], "573409": 398, "765166": 398, "0638": 398, "75596": 398, "225763": 398, "256995": 398, "773433": 398, "formula": 401, "anchor": 402, "triplet": 402, "_p": 402, "pairwis": 402, "instabl": 402, "monoton": 403, "0507": 407, "67326": 407, "sum_j": [410, 411], "x_j": [410, 411], "subclass": 420, "concept": 420, "mymlp": 420, "in_proj": 420, "involv": [422, 447], "subsequ": 422, "far": 422, "serial": 422, "adam": [422, 428, 429, 438, 439], "configu": 422, "thumb": 422, "apply_gradi": 422, "rmsprop": 422, "adagrad": 422, "adafactor": 422, "adadelta": 422, "adamw": [422, 429], "adamax": 422, "lion": 422, "cosine_decai": [422, 438], "exponential_decai": 422, "join_schedul": 422, "linear_schedul": [422, 438], "step_decai": 422, "rho": 423, "zeiler": 423, "2012": [423, 434], "adapt": [423, 424, 425, 448], "1212": 423, "5701": 423, "v_": [423, 425, 426, 427, 428, 434, 435], "v_t": [423, 425, 426, 427, 428, 434, 435], "u_t": 423, "u_": 423, "w_t": [423, 425, 426, 427, 428, 429, 434, 435], "30": 424, "001": 424, "clip_threshold": 424, "decay_r": [424, 437, 440], "beta_1": [424, 426, 427, 428, 429], "weight_decai": [424, 427, 429, 435], "scale_paramet": 424, "relative_step": 424, "warmup_init": 424, "sublinear": 424, "epsilon_1": 424, "epsilon_2": 424, "parameter_scal": 424, "unscal": 424, "decai": [424, 427, 429, 435, 436, 437, 440], "duchi": 425, "hazan": 425, "singer": 425, "2011": 425, "subgradi": 425, "onlin": 425, "stochast": [425, 426, 428, 435, 451], "jmlr": 425, "999": [426, 427, 428], "omit": [426, 428, 448], "estim": [426, 428], "kingma": [426, 428], "ba": [426, 428], "iclr": [426, 427, 428], "m_": [426, 427, 428, 429], "m_t": [426, 427, 428, 429], "beta_2": [426, 427, 428, 429], "contrast": 427, "loshchilov": 427, "hutter": 427, "decoupl": 427, "99": [429, 434], "tend": 429, "10x": 429, "strength": [429, 435], "wd": 429, "chen": 429, "symbol": 429, "discoveri": 429, "2302": 429, "06675": 429, "eta": 429, "opt": [430, 448], "tieleman": 434, "hinton": 434, "lectur": 434, "coursera": 434, "dampen": 435, "nesterov": 435, "descent": [435, 447, 451], "mu": 435, "tau": 435, "penalti": 435, "decay_step": 436, "beyond": [436, 439], "lr_schedul": [436, 437, 438, 440], "1000": [436, 447], "0999961": 436, "06561": 437, "boundari": 438, "join": 438, "transit": 438, "warmup": [438, 439], "0999938": 438, "101": 439, "step_siz": 440, "081": 440, "basi": 442, "implicit": [444, 447, 449], "fine": [444, 451], "grain": 444, "pseudo": 444, "altern": 444, "splittabl": 444, "threefri": 444, "counter": 444, "cycl": 446, "merg": 447, "fuse": 447, "big": 447, "awar": [447, 451], "36788": 447, "compiled_fun": 447, "code": [447, 448, 451], "slow": 447, "rerun": [447, 451], "frequent": [447, 451], "destroi": 447, "anonym": 447, "unari": 447, "overhead": [447, 451, 455], "bandwidth": [447, 448], "fusibl": 447, "consider": 447, "versu": 447, "timeit": [447, 449], "tic": 447, "perf_count": 447, "toc": 447, "tpi": 447, "1e3": 447, "4096": [447, 449, 455], "On": [447, 449, 451], "five": 447, "latest": 447, "won": 447, "placehold": 447, "insid": 447, "disable_compil": 447, "okai": [447, 451], "deal": 447, "pretti": [447, 451], "inconveni": 447, "functool": 447, "particularli": 447, "backward": [447, 449], "compiled_grad_fn": 447, "71828": 447, "opportun": 447, "physic": 448, "lot": [448, 449], "figur": 448, "best": 448, "mac": 448, "abil": 448, "talk": 448, "minim": 448, "all_sum": 448, "mpirun": 448, "mpiexec": 448, "send": 448, "homebrew": 448, "anaconda": 448, "openmpi": 448, "libmpi": 448, "dyld": 448, "dyld_library_path": 448, "lib": 448, "connect": 448, "ssh": 448, "checklist": 448, "issu": [448, 449, 452], "hostnam": 448, "password": 448, "confirm": 448, "foo": 448, "bar": 448, "host1": 448, "host2": 448, "qualifi": 448, "ip": 448, "slot": 448, "hostfil": 448, "parallel": [448, 455], "snippet": 448, "loss_grad_fn": 448, "all_avg": 448, "all_reduce_grad": 448, "latenc": 448, "mca": 448, "btl_tcp_link": 448, "tcp": 448, "idea": [449, 451], "behind": 449, "dfdx": [449, 450], "d2fdx2": 449, "zero_grad": 449, "detach": 449, "requires_grad": 449, "dloss_dw": 449, "dloss_dx": 449, "redund": 449, "continu": 449, "suppos": [449, 455], "nice": [449, 451], "propag": [449, 450], "autom": 449, "contriv": [449, 455], "sake": 449, "clariti": 449, "quit": [449, 452], "difficult": 449, "primit": 449, "priorit": 449, "naive_add": 449, "vmap_add": 449, "total": 449, "390": 449, "wherea": 449, "025": 449, "ten": [449, 451], "Of": 449, "cours": 449, "better": [449, 455], "handi": 449, "ellipsi": 450, "mix": 450, "lack": 450, "extrem": [450, 451], "ineffici": [450, 451], "nonzero": 450, "dynam": 451, "easier": 451, "worri": 451, "fun1": 451, "expensive_fun": 451, "consum": 451, "eager": 451, "thank": 451, "weights_fp16": 451, "trade": 451, "bad": 451, "grow": 451, "computation": 451, "costli": 451, "luckili": 451, "wide": 451, "thousand": 451, "value_and_grad_fn": 451, "implicitli": 451, "anytim": 451, "memoryview": [451, 452], "perfectli": 451, "first_lay": 451, "second_layer_a": 451, "second_layer_b": 451, "protocol": 452, "dlpack": 452, "pep": 452, "3118": 452, "a_view": 452, "owndata": 452, "extern": 452, "demonstr": 452, "x_view": 452, "modifi": 452, "df": 452, "x\u00b2": 452, "2x": 452, "indirectli": 452, "modif": 452, "seen": 452, "occur": 452, "incorpor": 452, "aris": 452, "incorrect": 452, "experiment": 452, "break": 452, "advis": 452, "intermedi": 452, "jnp": 452, "tf": 452, "page": 453, "composit": 453, "archiv": 454, "savez_compress": 454, "save_gguf": 454, "arr_0": 454, "advantag": 455, "race": 455, "interest": 455, "albeit": 455, "d1": 455, "d2": 455, "But": 455, "twice": 455, "measur": 455, "default_stream": 456, "default_devic": 456, "my_devic": 456}, "objects": {"": [[0, 0, 1, "_CPPv43absRK5array14StreamOrDevice", "abs"], [0, 1, 1, "_CPPv43absRK5array14StreamOrDevice", "abs::a"], [0, 1, 1, "_CPPv43absRK5array14StreamOrDevice", "abs::s"], [0, 0, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::a"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::b"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::s"], [0, 0, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::a"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::alpha"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::b"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::beta"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::c"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::s"], [0, 0, 1, "_CPPv43allRK5array14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all"], [0, 1, 1, "_CPPv43allRK5array14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::axes"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::axis"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5array14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::s"], [0, 0, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::a"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::atol"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::b"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::equal_nan"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::rtol"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::s"], [0, 0, 1, "_CPPv43anyRK5array14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any"], [0, 1, 1, "_CPPv43anyRK5array14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::axes"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::axis"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5array14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::s"], [0, 0, 1, "_CPPv46aranged14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangedd14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeddd14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangei14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeii14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeiii14StreamOrDevice", "arange"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46aranged14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangei14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46aranged14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangei14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::stop"], [0, 0, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos"], [0, 1, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos::a"], [0, 1, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos::s"], [0, 0, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh"], [0, 1, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh::a"], [0, 1, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh::s"], [0, 0, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin"], [0, 1, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin::a"], [0, 1, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin::s"], [0, 0, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh"], [0, 1, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh::a"], [0, 1, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh::s"], [0, 0, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan"], [0, 0, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::a"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::b"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::s"], [0, 1, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan::a"], [0, 1, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan::s"], [0, 0, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh"], [0, 1, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh::a"], [0, 1, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh::s"], [0, 0, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax"], [0, 0, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax"], [0, 0, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax"], [0, 1, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::axis"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::keepdims"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::keepdims"], [0, 1, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax::s"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::s"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::s"], [0, 0, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin"], [0, 0, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin"], [0, 0, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin"], [0, 1, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::axis"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::keepdims"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::keepdims"], [0, 1, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin::s"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::s"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::s"], [0, 0, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition"], [0, 0, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::a"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::a"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::axis"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::kth"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::kth"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::s"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::s"], [0, 0, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort"], [0, 0, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort"], [0, 1, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort::a"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::a"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::axis"], [0, 1, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort::s"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::s"], [0, 0, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal"], [0, 0, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::a"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::a"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::b"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::b"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::equal_nan"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::s"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::s"], [0, 0, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::a"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::offset"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::s"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::shape"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::strides"], [0, 0, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::a"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::dtype"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::s"], [0, 0, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d"], [0, 0, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d"], [0, 1, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d::a"], [0, 1, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d::a"], [0, 1, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d::s"], [0, 1, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d::s"], [0, 0, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d"], [0, 0, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d"], [0, 1, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d::a"], [0, 1, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d::a"], [0, 1, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d::s"], [0, 1, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d::s"], [0, 0, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d"], [0, 0, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d"], [0, 1, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d::a"], [0, 1, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d::a"], [0, 1, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d::s"], [0, 1, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d::s"], [0, 0, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::a"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::b"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::s"], [0, 0, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::a"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::b"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::s"], [0, 0, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::a"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::b"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::s"], [0, 0, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::a"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::b"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::block_size"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_lhs"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_out"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_rhs"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::s"], [0, 0, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays"], [0, 1, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays::inputs"], [0, 1, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays::s"], [0, 0, 1, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "broadcast_to"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "broadcast_to::a"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "broadcast_to::s"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "broadcast_to::shape"], [0, 0, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil"], [0, 1, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil::a"], [0, 1, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil::s"], [0, 0, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a_max"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a_min"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::s"], [0, 0, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice", "concatenate"], [0, 0, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::axis"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::s"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::s"], [0, 0, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::a"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::s"], [0, 0, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::dilation"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::groups"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::input"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::padding"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::s"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::stride"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::weight"], [0, 0, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::dilation"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::groups"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::input"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::padding"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::s"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::stride"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::weight"], [0, 0, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::dilation"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::groups"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::input"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::padding"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::s"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::stride"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::weight"], [0, 0, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general"], [0, 0, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::flip"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::flip"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::groups"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::groups"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input_dilation"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::kernel_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::kernel_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding_hi"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding_lo"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::s"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::s"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::stride"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::stride"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::weight"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::weight"], [0, 0, 1, "_CPPv44copy5array14StreamOrDevice", "copy"], [0, 1, 1, "_CPPv44copy5array14StreamOrDevice", "copy::a"], [0, 1, 1, "_CPPv44copy5array14StreamOrDevice", "copy::s"], [0, 0, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos"], [0, 1, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos::a"], [0, 1, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos::s"], [0, 0, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh"], [0, 1, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh::a"], [0, 1, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh::s"], [0, 0, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::a"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::axis"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::inclusive"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::reverse"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::s"], [0, 0, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::a"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::axis"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::inclusive"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::reverse"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::s"], [0, 0, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::a"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::axis"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::inclusive"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::reverse"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::s"], [0, 0, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::a"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::axis"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::inclusive"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::reverse"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::s"], [0, 0, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees"], [0, 1, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees::a"], [0, 1, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees::s"], [0, 0, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends"], [0, 1, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends::dependencies"], [0, 1, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends::inputs"], [0, 0, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::biases"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::bits"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::group_size"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::s"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::scales"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::w"], [0, 0, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::a"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::k"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::s"], [0, 0, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::a"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::axis1"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::axis2"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::offset"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::s"], [0, 0, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::a"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::b"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::s"], [0, 0, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::a"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::b"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::s"], [0, 0, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::a"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::b"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::s"], [0, 0, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf"], [0, 1, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf::a"], [0, 1, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf::s"], [0, 0, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv"], [0, 1, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv::a"], [0, 1, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv::s"], [0, 0, 1, "_CPPv43expRK5array14StreamOrDevice", "exp"], [0, 1, 1, "_CPPv43expRK5array14StreamOrDevice", "exp::a"], [0, 1, 1, "_CPPv43expRK5array14StreamOrDevice", "exp::s"], [0, 0, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims"], [0, 0, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::a"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::a"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::axes"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::axis"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::s"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::s"], [0, 0, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1"], [0, 1, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1::a"], [0, 1, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1::s"], [0, 0, 1, "_CPPv43eyei14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeii14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeiii14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::dtype"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::dtype"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::k"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::k"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyei14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyei14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::s"], [0, 0, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten"], [0, 0, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten"], [0, 1, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten::a"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::a"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::end_axis"], [0, 1, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten::s"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::s"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::start_axis"], [0, 0, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor"], [0, 1, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor::a"], [0, 1, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor::s"], [0, 0, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::a"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::b"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::s"], [0, 0, 1, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice", "full"], [0, 0, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full"], [0, 2, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full::T"], [0, 2, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::T"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice", "full::vals"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full::vals"], [0, 0, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather"], [0, 0, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::axes"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::axis"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::slice_sizes"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::slice_sizes"], [0, 0, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::a"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::b"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::lhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::rhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::s"], [0, 0, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::biases"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::bits"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::group_size"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::lhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::rhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::s"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::scales"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::transpose"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::w"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::x"], [0, 0, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::a"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::b"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::s"], [0, 0, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::a"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::b"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::s"], [0, 0, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::a"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::s"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::scale"], [0, 0, 1, "_CPPv48identityi14StreamOrDevice", "identity"], [0, 0, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::dtype"], [0, 1, 1, "_CPPv48identityi14StreamOrDevice", "identity::n"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::n"], [0, 1, 1, "_CPPv48identityi14StreamOrDevice", "identity::s"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::s"], [0, 0, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::a"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::b"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::s"], [0, 0, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::a"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::atol"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::b"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::equal_nan"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::rtol"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::s"], [0, 0, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf"], [0, 1, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf::a"], [0, 1, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf::s"], [0, 0, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan"], [0, 1, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan::a"], [0, 1, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan::s"], [0, 0, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf"], [0, 1, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf::a"], [0, 1, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf::s"], [0, 0, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf"], [0, 1, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf::a"], [0, 1, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf::s"], [0, 0, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::a"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::b"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::s"], [0, 0, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::a"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::b"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::s"], [0, 0, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::a"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::b"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::s"], [0, 0, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::dtype"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::num"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::s"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::start"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::stop"], [0, 0, 1, "_CPPv43logRK5array14StreamOrDevice", "log"], [0, 0, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10"], [0, 1, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10::a"], [0, 1, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10::s"], [0, 0, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p"], [0, 1, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p::a"], [0, 1, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p::s"], [0, 0, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2"], [0, 1, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2::a"], [0, 1, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2::s"], [0, 1, 1, "_CPPv43logRK5array14StreamOrDevice", "log::a"], [0, 1, 1, "_CPPv43logRK5array14StreamOrDevice", "log::s"], [0, 0, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::a"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::b"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::s"], [0, 0, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::a"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::b"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::s"], [0, 0, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not"], [0, 1, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not::a"], [0, 1, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not::s"], [0, 0, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::a"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::b"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::s"], [0, 0, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp"], [0, 1, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::axes"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::axis"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::s"], [0, 0, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::a"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::b"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::s"], [0, 0, 1, "_CPPv43maxRK5array14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max"], [0, 1, 1, "_CPPv43maxRK5array14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::axes"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::axis"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5array14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::s"], [0, 0, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::a"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::b"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::s"], [0, 0, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean"], [0, 1, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::axes"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::axis"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::s"], [0, 0, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::arrays"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::indexing"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::s"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::sparse"], [0, 0, 1, "_CPPv43minRK5array14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min"], [0, 1, 1, "_CPPv43minRK5array14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::axes"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::axis"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5array14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::s"], [0, 0, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::a"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::b"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::s"], [0, 0, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::a"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::destination"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::s"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::source"], [0, 0, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::a"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::b"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::s"], [0, 0, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::a"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::nan"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::neginf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::posinf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::s"], [0, 0, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative"], [0, 1, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative::a"], [0, 1, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative::s"], [0, 0, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::a"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::b"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::s"], [0, 0, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::a"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::axes"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::dtype"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::inverted"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::s"], [0, 0, 1, "_CPPv44onesRKNSt6vectorIiEE14StreamOrDevice", "ones"], [0, 0, 1, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice", "ones"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice", "ones::dtype"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE14StreamOrDevice", "ones::shape"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice", "ones::shape"], [0, 0, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like"], [0, 1, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like::a"], [0, 1, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like::s"], [0, 0, 1, "_CPPv4I0Ene5array1TRK5array", "operator!="], [0, 0, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!="], [0, 0, 1, "_CPPv4neRK5arrayRK5array", "operator!="], [0, 2, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::T"], [0, 2, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::T"], [0, 1, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::a"], [0, 1, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::a"], [0, 1, 1, "_CPPv4neRK5arrayRK5array", "operator!=::a"], [0, 1, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::b"], [0, 1, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::b"], [0, 1, 1, "_CPPv4neRK5arrayRK5array", "operator!=::b"], [0, 0, 1, "_CPPv4I0Erm5array1TRK5array", "operator%"], [0, 0, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%"], [0, 0, 1, "_CPPv4rmRK5arrayRK5array", "operator%"], [0, 2, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::T"], [0, 2, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::T"], [0, 1, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::a"], [0, 1, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::a"], [0, 1, 1, "_CPPv4rmRK5arrayRK5array", "operator%::a"], [0, 1, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::b"], [0, 1, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::b"], [0, 1, 1, "_CPPv4rmRK5arrayRK5array", "operator%::b"], [0, 0, 1, "_CPPv4anRK5arrayRK5array", "operator&"], [0, 0, 1, "_CPPv4aaRK5arrayRK5array", "operator&&"], [0, 1, 1, "_CPPv4aaRK5arrayRK5array", "operator&&::a"], [0, 1, 1, "_CPPv4aaRK5arrayRK5array", "operator&&::b"], [0, 1, 1, "_CPPv4anRK5arrayRK5array", "operator&::a"], [0, 1, 1, "_CPPv4anRK5arrayRK5array", "operator&::b"], [0, 0, 1, "_CPPv4I0Eml5array1TRK5array", "operator*"], [0, 0, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*"], [0, 0, 1, "_CPPv4mlRK5arrayRK5array", "operator*"], [0, 2, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::T"], [0, 2, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::T"], [0, 1, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::a"], [0, 1, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::a"], [0, 1, 1, "_CPPv4mlRK5arrayRK5array", "operator*::a"], [0, 1, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::b"], [0, 1, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::b"], [0, 1, 1, "_CPPv4mlRK5arrayRK5array", "operator*::b"], [0, 0, 1, "_CPPv4I0Epl5array1TRK5array", "operator+"], [0, 0, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+"], [0, 0, 1, "_CPPv4plRK5arrayRK5array", "operator+"], [0, 2, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::T"], [0, 2, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::T"], [0, 1, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::a"], [0, 1, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::a"], [0, 1, 1, "_CPPv4plRK5arrayRK5array", "operator+::a"], [0, 1, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::b"], [0, 1, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::b"], [0, 1, 1, "_CPPv4plRK5arrayRK5array", "operator+::b"], [0, 0, 1, "_CPPv4I0Emi5array1TRK5array", "operator-"], [0, 0, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-"], [0, 0, 1, "_CPPv4miRK5array", "operator-"], [0, 0, 1, "_CPPv4miRK5arrayRK5array", "operator-"], [0, 2, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::T"], [0, 2, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::T"], [0, 1, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::a"], [0, 1, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::a"], [0, 1, 1, "_CPPv4miRK5array", "operator-::a"], [0, 1, 1, "_CPPv4miRK5arrayRK5array", "operator-::a"], [0, 1, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::b"], [0, 1, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::b"], [0, 1, 1, "_CPPv4miRK5arrayRK5array", "operator-::b"], [0, 0, 1, "_CPPv4dvRK5arrayRK5array", "operator/"], [0, 0, 1, "_CPPv4dvRK5arrayd", "operator/"], [0, 0, 1, "_CPPv4dvdRK5array", "operator/"], [0, 1, 1, "_CPPv4dvRK5arrayRK5array", "operator/::a"], [0, 1, 1, "_CPPv4dvRK5arrayd", "operator/::a"], [0, 1, 1, "_CPPv4dvdRK5array", "operator/::a"], [0, 1, 1, "_CPPv4dvRK5arrayRK5array", "operator/::b"], [0, 1, 1, "_CPPv4dvRK5arrayd", "operator/::b"], [0, 1, 1, "_CPPv4dvdRK5array", "operator/::b"], [0, 0, 1, "_CPPv4I0Elt5array1TRK5array", "operator<"], [0, 0, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<"], [0, 0, 1, "_CPPv4ltRK5arrayRK5array", "operator<"], [0, 2, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::T"], [0, 2, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::T"], [0, 1, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::a"], [0, 1, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::a"], [0, 1, 1, "_CPPv4ltRK5arrayRK5array", "operator<::a"], [0, 1, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::b"], [0, 1, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::b"], [0, 1, 1, "_CPPv4ltRK5arrayRK5array", "operator<::b"], [0, 0, 1, "_CPPv4lsRK5arrayRK5array", "operator<<"], [0, 1, 1, "_CPPv4lsRK5arrayRK5array", "operator<<::a"], [0, 1, 1, "_CPPv4lsRK5arrayRK5array", "operator<<::b"], [0, 0, 1, "_CPPv4I0Ele5array1TRK5array", "operator<="], [0, 0, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<="], [0, 0, 1, "_CPPv4leRK5arrayRK5array", "operator<="], [0, 2, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::T"], [0, 2, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::T"], [0, 1, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::a"], [0, 1, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::a"], [0, 1, 1, "_CPPv4leRK5arrayRK5array", "operator<=::a"], [0, 1, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::b"], [0, 1, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::b"], [0, 1, 1, "_CPPv4leRK5arrayRK5array", "operator<=::b"], [0, 0, 1, "_CPPv4I0Eeq5array1TRK5array", "operator=="], [0, 0, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator=="], [0, 0, 1, "_CPPv4eqRK5arrayRK5array", "operator=="], [0, 2, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::T"], [0, 2, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::T"], [0, 1, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::a"], [0, 1, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::a"], [0, 1, 1, "_CPPv4eqRK5arrayRK5array", "operator==::a"], [0, 1, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::b"], [0, 1, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::b"], [0, 1, 1, "_CPPv4eqRK5arrayRK5array", "operator==::b"], [0, 0, 1, "_CPPv4I0Egt5array1TRK5array", "operator>"], [0, 0, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>"], [0, 0, 1, "_CPPv4gtRK5arrayRK5array", "operator>"], [0, 2, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::T"], [0, 2, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::T"], [0, 1, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::a"], [0, 1, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::a"], [0, 1, 1, "_CPPv4gtRK5arrayRK5array", "operator>::a"], [0, 1, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::b"], [0, 1, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::b"], [0, 1, 1, "_CPPv4gtRK5arrayRK5array", "operator>::b"], [0, 0, 1, "_CPPv4I0Ege5array1TRK5array", "operator>="], [0, 0, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>="], [0, 0, 1, "_CPPv4geRK5arrayRK5array", "operator>="], [0, 2, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::T"], [0, 2, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::T"], [0, 1, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::a"], [0, 1, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::a"], [0, 1, 1, "_CPPv4geRK5arrayRK5array", "operator>=::a"], [0, 1, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::b"], [0, 1, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::b"], [0, 1, 1, "_CPPv4geRK5arrayRK5array", "operator>=::b"], [0, 0, 1, "_CPPv4rsRK5arrayRK5array", "operator>>"], [0, 1, 1, "_CPPv4rsRK5arrayRK5array", "operator>>::a"], [0, 1, 1, "_CPPv4rsRK5arrayRK5array", "operator>>::b"], [0, 0, 1, "_CPPv4eoRK5arrayRK5array", "operator^"], [0, 1, 1, "_CPPv4eoRK5arrayRK5array", "operator^::a"], [0, 1, 1, "_CPPv4eoRK5arrayRK5array", "operator^::b"], [0, 0, 1, "_CPPv4orRK5arrayRK5array", "operator|"], [0, 1, 1, "_CPPv4orRK5arrayRK5array", "operator|::a"], [0, 1, 1, "_CPPv4orRK5arrayRK5array", "operator|::b"], [0, 0, 1, "_CPPv4ooRK5arrayRK5array", "operator||"], [0, 1, 1, "_CPPv4ooRK5arrayRK5array", "operator||::a"], [0, 1, 1, "_CPPv4ooRK5arrayRK5array", "operator||::b"], [0, 0, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::a"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::b"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::s"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5array14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5array14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5array14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayiRK5array14StreamOrDevice", "pad"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5array14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5array14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5array14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayiRK5array14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5array14StreamOrDevice", "pad::axes"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5array14StreamOrDevice", "pad::high_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5array14StreamOrDevice", "pad::low_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5array14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5array14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5array14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayiRK5array14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5array14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5array14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayiRK5array14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5array14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5array14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5array14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayiRK5array14StreamOrDevice", "pad::s"], [0, 0, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition"], [0, 0, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::a"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::a"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::axis"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::kth"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::kth"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::s"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::s"], [0, 0, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::a"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::b"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::s"], [0, 0, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod"], [0, 1, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::axes"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::axis"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::s"], [0, 0, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::bits"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::group_size"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::s"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::w"], [0, 0, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::biases"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::bits"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::group_size"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::s"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::scales"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::transpose"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::w"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::x"], [0, 0, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::a"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::s"], [0, 0, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal"], [0, 1, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal::a"], [0, 1, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal::s"], [0, 0, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::a"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::b"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::s"], [0, 0, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat"], [0, 0, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::arr"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::arr"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::axis"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::repeats"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::repeats"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::s"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::s"], [0, 0, 1, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice", "reshape"], [0, 1, 1, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice", "reshape::a"], [0, 1, 1, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice", "reshape::s"], [0, 1, 1, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice", "reshape::shape"], [0, 0, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::a"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::b"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::s"], [0, 0, 1, "_CPPv45roundRK5array14StreamOrDevice", "round"], [0, 0, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round"], [0, 1, 1, "_CPPv45roundRK5array14StreamOrDevice", "round::a"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::a"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::decimals"], [0, 1, 1, "_CPPv45roundRK5array14StreamOrDevice", "round::s"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::s"], [0, 0, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt"], [0, 1, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt::a"], [0, 1, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt::s"], [0, 0, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter"], [0, 0, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::a"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::a"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::axes"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::axis"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::indices"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::indices"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::s"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::s"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::updates"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::updates"], [0, 0, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add"], [0, 0, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::a"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::a"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::axes"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::axis"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::indices"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::indices"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::s"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::s"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::updates"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::updates"], [0, 0, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max"], [0, 0, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::a"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::a"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::axes"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::axis"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::indices"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::indices"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::s"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::s"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::updates"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::updates"], [0, 0, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min"], [0, 0, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::a"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::a"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::axes"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::axis"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::indices"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::indices"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::s"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::s"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::updates"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::updates"], [0, 0, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod"], [0, 0, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::a"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::a"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::axes"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::axis"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::indices"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::indices"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::s"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::s"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::updates"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::updates"], [0, 0, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid"], [0, 1, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid::a"], [0, 1, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid::s"], [0, 0, 1, "_CPPv44signRK5array14StreamOrDevice", "sign"], [0, 1, 1, "_CPPv44signRK5array14StreamOrDevice", "sign::a"], [0, 1, 1, "_CPPv44signRK5array14StreamOrDevice", "sign::s"], [0, 0, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin"], [0, 1, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin::a"], [0, 1, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin::s"], [0, 0, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh"], [0, 1, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh::a"], [0, 1, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh::s"], [0, 0, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::strides"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::strides"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::update"], [0, 0, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax"], [0, 0, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax"], [0, 0, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::axes"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::axis"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::s"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::s"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::s"], [0, 0, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort"], [0, 0, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort"], [0, 1, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort::a"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::a"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::axis"], [0, 1, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort::s"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::s"], [0, 0, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::s"], [0, 0, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt"], [0, 1, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt::a"], [0, 1, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt::s"], [0, 0, 1, "_CPPv46squareRK5array14StreamOrDevice", "square"], [0, 1, 1, "_CPPv46squareRK5array14StreamOrDevice", "square::a"], [0, 1, 1, "_CPPv46squareRK5array14StreamOrDevice", "square::s"], [0, 0, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze"], [0, 0, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze"], [0, 0, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze"], [0, 1, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::axes"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::axis"], [0, 1, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze::s"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::s"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::s"], [0, 0, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack"], [0, 0, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack::arrays"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::arrays"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::axis"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack::s"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::s"], [0, 0, 1, "_CPPv4StRK5array14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std"], [0, 1, 1, "_CPPv4StRK5array14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::axes"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::axis"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5array14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::s"], [0, 0, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient"], [0, 1, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient::a"], [0, 1, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient::s"], [0, 0, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::a"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::b"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::s"], [0, 0, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum"], [0, 1, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::axes"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::axis"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::s"], [0, 0, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::a"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::axis1"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::axis2"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::s"], [0, 0, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::indices"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::indices"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::s"], [0, 0, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::a"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::axis"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::indices"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::s"], [0, 0, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan"], [0, 1, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan::a"], [0, 1, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan::s"], [0, 0, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh"], [0, 1, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh::a"], [0, 1, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh::s"], [0, 0, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot"], [0, 0, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::axes_a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::axes_b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::axis"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::s"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::s"], [0, 0, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::arr"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::reps"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::s"], [0, 0, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk"], [0, 0, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::a"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::a"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::axis"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::k"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::k"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::s"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::s"], [0, 0, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace"], [0, 0, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace"], [0, 0, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace"], [0, 1, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::axis1"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::axis1"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::axis2"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::axis2"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::dtype"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::offset"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::offset"], [0, 1, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace::s"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::s"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::s"], [0, 0, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose"], [0, 0, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose"], [0, 0, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose"], [0, 1, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::axes"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::axes"], [0, 1, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose::s"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::s"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::s"], [0, 0, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri"], [0, 0, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::k"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::m"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::n"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::n"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::s"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::s"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::type"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::type"], [0, 0, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::k"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::s"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::x"], [0, 0, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::k"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::s"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::x"], [0, 0, 1, "_CPPv43varRK5array14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var"], [0, 1, 1, "_CPPv43varRK5array14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::axes"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::axis"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5array14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::s"], [0, 0, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::a"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::dtype"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::s"], [0, 0, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::condition"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::s"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::x"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::y"], [0, 0, 1, "_CPPv45zerosRKNSt6vectorIiEE14StreamOrDevice", "zeros"], [0, 0, 1, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice", "zeros"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice", "zeros::dtype"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE14StreamOrDevice", "zeros::shape"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice", "zeros::shape"], [0, 0, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like"], [0, 1, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like::a"], [0, 1, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like::s"]], "mlx.core": [[8, 3, 1, "", "Device"], [9, 3, 1, "", "Dtype"], [10, 3, 1, "", "DtypeCategory"], [292, 3, 1, "", "Stream"], [11, 5, 1, "", "abs"], [12, 5, 1, "", "add"], [13, 5, 1, "", "addmm"], [14, 5, 1, "", "all"], [15, 5, 1, "", "allclose"], [16, 5, 1, "", "any"], [17, 5, 1, "", "arange"], [18, 5, 1, "", "arccos"], [19, 5, 1, "", "arccosh"], [20, 5, 1, "", "arcsin"], [21, 5, 1, "", "arcsinh"], [22, 5, 1, "", "arctan"], [23, 5, 1, "", "arctan2"], [24, 5, 1, "", "arctanh"], [25, 5, 1, "", "argmax"], [26, 5, 1, "", "argmin"], [27, 5, 1, "", "argpartition"], [28, 5, 1, "", "argsort"], [29, 3, 1, "", "array"], [80, 5, 1, "", "array_equal"], [81, 5, 1, "", "as_strided"], [82, 5, 1, "", "atleast_1d"], [83, 5, 1, "", "atleast_2d"], [84, 5, 1, "", "atleast_3d"], [85, 5, 1, "", "bitwise_and"], [86, 5, 1, "", "bitwise_or"], [87, 5, 1, "", "bitwise_xor"], [88, 5, 1, "", "block_masked_mm"], [89, 5, 1, "", "broadcast_to"], [90, 5, 1, "", "ceil"], [91, 5, 1, "", "clip"], [92, 5, 1, "", "compile"], [93, 5, 1, "", "concatenate"], [94, 5, 1, "", "conj"], [95, 5, 1, "", "conjugate"], [96, 5, 1, "", "conv1d"], [97, 5, 1, "", "conv2d"], [98, 5, 1, "", "conv_general"], [99, 5, 1, "", "convolve"], [100, 5, 1, "", "cos"], [101, 5, 1, "", "cosh"], [102, 5, 1, "", "cummax"], [103, 5, 1, "", "cummin"], [104, 5, 1, "", "cumprod"], [105, 5, 1, "", "cumsum"], [106, 3, 1, "", "custom_function"], [107, 5, 1, "", "default_device"], [108, 5, 1, "", "default_stream"], [109, 5, 1, "", "degrees"], [110, 5, 1, "", "dequantize"], [111, 5, 1, "", "diag"], [112, 5, 1, "", "diagonal"], [113, 5, 1, "", "disable_compile"], [119, 5, 1, "", "divide"], [120, 5, 1, "", "divmod"], [121, 5, 1, "", "einsum"], [122, 5, 1, "", "einsum_path"], [123, 5, 1, "", "enable_compile"], [124, 5, 1, "", "equal"], [125, 5, 1, "", "erf"], [126, 5, 1, "", "erfinv"], [127, 5, 1, "", "eval"], [128, 5, 1, "", "exp"], [129, 5, 1, "", "expand_dims"], [130, 5, 1, "", "expm1"], [131, 5, 1, "", "eye"], [148, 5, 1, "", "flatten"], [149, 5, 1, "", "floor"], [150, 5, 1, "", "floor_divide"], [151, 5, 1, "", "full"], [152, 5, 1, "", "gather_mm"], [153, 5, 1, "", "gather_qmm"], [154, 5, 1, "", "grad"], [155, 5, 1, "", "greater"], [156, 5, 1, "", "greater_equal"], [157, 5, 1, "", "hadamard_transform"], [158, 5, 1, "", "identity"], [159, 5, 1, "", "inner"], [160, 5, 1, "", "isclose"], [161, 5, 1, "", "isinf"], [162, 5, 1, "", "isnan"], [163, 5, 1, "", "isneginf"], [164, 5, 1, "", "isposinf"], [165, 5, 1, "", "issubdtype"], [166, 5, 1, "", "jvp"], [167, 5, 1, "", "left_shift"], [168, 5, 1, "", "less"], [169, 5, 1, "", "less_equal"], [175, 5, 1, "", "linspace"], [176, 5, 1, "", "load"], [177, 5, 1, "", "log"], [178, 5, 1, "", "log10"], [179, 5, 1, "", "log1p"], [180, 5, 1, "", "log2"], [181, 5, 1, "", "logaddexp"], [182, 5, 1, "", "logical_and"], [183, 5, 1, "", "logical_not"], [184, 5, 1, "", "logical_or"], [185, 5, 1, "", "logsumexp"], [186, 5, 1, "", "matmul"], [187, 5, 1, "", "max"], [188, 5, 1, "", "maximum"], [189, 5, 1, "", "mean"], [190, 5, 1, "", "meshgrid"], [202, 5, 1, "", "min"], [203, 5, 1, "", "minimum"], [204, 5, 1, "", "moveaxis"], [205, 5, 1, "", "multiply"], [206, 5, 1, "", "nan_to_num"], [207, 5, 1, "", "negative"], [208, 5, 1, "", "new_stream"], [209, 5, 1, "", "not_equal"], [210, 5, 1, "", "ones"], [211, 5, 1, "", "ones_like"], [212, 5, 1, "", "outer"], [213, 5, 1, "", "pad"], [214, 5, 1, "", "partition"], [215, 5, 1, "", "power"], [216, 5, 1, "", "prod"], [217, 5, 1, "", "quantize"], [218, 5, 1, "", "quantized_matmul"], [219, 5, 1, "", "radians"], [232, 5, 1, "", "reciprocal"], [233, 5, 1, "", "remainder"], [234, 5, 1, "", "repeat"], [235, 5, 1, "", "reshape"], [236, 5, 1, "", "right_shift"], [237, 5, 1, "", "round"], [238, 5, 1, "", "rsqrt"], [239, 5, 1, "", "save"], [240, 5, 1, "", "save_gguf"], [241, 5, 1, "", "save_safetensors"], [242, 5, 1, "", "savez"], [243, 5, 1, "", "savez_compressed"], [244, 5, 1, "", "set_default_device"], [245, 5, 1, "", "set_default_stream"], [246, 5, 1, "", "sigmoid"], [247, 5, 1, "", "sign"], [248, 5, 1, "", "sin"], [249, 5, 1, "", "sinh"], [250, 5, 1, "", "softmax"], [251, 5, 1, "", "sort"], [252, 5, 1, "", "split"], [253, 5, 1, "", "sqrt"], [254, 5, 1, "", "square"], [255, 5, 1, "", "squeeze"], [256, 5, 1, "", "stack"], [257, 5, 1, "", "std"], [258, 5, 1, "", "stop_gradient"], [259, 5, 1, "", "stream"], [260, 5, 1, "", "subtract"], [261, 5, 1, "", "sum"], [262, 5, 1, "", "swapaxes"], [263, 5, 1, "", "synchronize"], [264, 5, 1, "", "take"], [265, 5, 1, "", "take_along_axis"], [266, 5, 1, "", "tan"], [267, 5, 1, "", "tanh"], [268, 5, 1, "", "tensordot"], [269, 5, 1, "", "tile"], [270, 5, 1, "", "topk"], [271, 5, 1, "", "trace"], [272, 5, 1, "", "transpose"], [273, 5, 1, "", "tri"], [274, 5, 1, "", "tril"], [275, 5, 1, "", "triu"], [276, 5, 1, "", "value_and_grad"], [277, 5, 1, "", "var"], [278, 5, 1, "", "view"], [279, 5, 1, "", "vjp"], [280, 5, 1, "", "vmap"], [281, 5, 1, "", "where"], [282, 5, 1, "", "zeros"], [283, 5, 1, "", "zeros_like"]], "mlx.core.Device": [[8, 4, 1, "", "__init__"]], "mlx.core.Dtype": [[9, 4, 1, "", "__init__"]], "mlx.core.DtypeCategory": [[10, 4, 1, "", "__init__"]], "mlx.core.Stream": [[292, 4, 1, "", "__init__"]], "mlx.core.array": [[30, 6, 1, "", "T"], [29, 4, 1, "", "__init__"], [31, 4, 1, "", "abs"], [32, 4, 1, "", "all"], [33, 4, 1, "", "any"], [34, 4, 1, "", "argmax"], [35, 4, 1, "", "argmin"], [36, 4, 1, "", "astype"], [37, 6, 1, "", "at"], [38, 4, 1, "", "conj"], [39, 4, 1, "", "cos"], [40, 4, 1, "", "cummax"], [41, 4, 1, "", "cummin"], [42, 4, 1, "", "cumprod"], [43, 4, 1, "", "cumsum"], [44, 4, 1, "", "diag"], [45, 4, 1, "", "diagonal"], [46, 6, 1, "", "dtype"], [47, 4, 1, "", "exp"], [48, 4, 1, "", "flatten"], [49, 4, 1, "", "item"], [50, 6, 1, "", "itemsize"], [51, 4, 1, "", "log"], [52, 4, 1, "", "log10"], [53, 4, 1, "", "log1p"], [54, 4, 1, "", "log2"], [55, 4, 1, "", "logsumexp"], [56, 4, 1, "", "max"], [57, 4, 1, "", "mean"], [58, 4, 1, "", "min"], [59, 4, 1, "", "moveaxis"], [60, 6, 1, "", "nbytes"], [61, 6, 1, "", "ndim"], [62, 4, 1, "", "prod"], [63, 4, 1, "", "reciprocal"], [64, 4, 1, "", "reshape"], [65, 4, 1, "", "round"], [66, 4, 1, "", "rsqrt"], [67, 6, 1, "", "shape"], [68, 4, 1, "", "sin"], [69, 6, 1, "", "size"], [70, 4, 1, "", "split"], [71, 4, 1, "", "sqrt"], [72, 4, 1, "", "square"], [73, 4, 1, "", "squeeze"], [74, 4, 1, "", "sum"], [75, 4, 1, "", "swapaxes"], [76, 4, 1, "", "tolist"], [77, 4, 1, "", "transpose"], [78, 4, 1, "", "var"], [79, 4, 1, "", "view"]], "mlx.core.custom_function": [[106, 4, 1, "", "__init__"]], "mlx.core.distributed": [[114, 3, 1, "", "Group"], [115, 5, 1, "", "all_gather"], [116, 5, 1, "", "all_sum"], [117, 5, 1, "", "init"], [118, 5, 1, "", "is_available"]], "mlx.core.distributed.Group": [[114, 4, 1, "", "__init__"]], "mlx.core.fast": [[132, 5, 1, "", "layer_norm"], [133, 5, 1, "", "rms_norm"], [134, 5, 1, "", "rope"], [135, 5, 1, "", "scaled_dot_product_attention"]], "mlx.core.fft": [[136, 5, 1, "", "fft"], [137, 5, 1, "", "fft2"], [138, 5, 1, "", "fftn"], [139, 5, 1, "", "ifft"], [140, 5, 1, "", "ifft2"], [141, 5, 1, "", "ifftn"], [142, 5, 1, "", "irfft"], [143, 5, 1, "", "irfft2"], [144, 5, 1, "", "irfftn"], [145, 5, 1, "", "rfft"], [146, 5, 1, "", "rfft2"], [147, 5, 1, "", "rfftn"]], "mlx.core.linalg": [[170, 5, 1, "", "cholesky"], [171, 5, 1, "", "inv"], [172, 5, 1, "", "norm"], [173, 5, 1, "", "qr"], [174, 5, 1, "", "svd"]], "mlx.core.metal": [[191, 5, 1, "", "clear_cache"], [192, 5, 1, "", "device_info"], [193, 5, 1, "", "get_active_memory"], [194, 5, 1, "", "get_cache_memory"], [195, 5, 1, "", "get_peak_memory"], [196, 5, 1, "", "is_available"], [197, 5, 1, "", "reset_peak_memory"], [198, 5, 1, "", "set_cache_limit"], [199, 5, 1, "", "set_memory_limit"], [200, 5, 1, "", "start_capture"], [201, 5, 1, "", "stop_capture"]], "mlx.core.random": [[220, 5, 1, "", "bernoulli"], [221, 5, 1, "", "categorical"], [222, 5, 1, "", "gumbel"], [223, 5, 1, "", "key"], [224, 5, 1, "", "laplace"], [225, 5, 1, "", "multivariate_normal"], [226, 5, 1, "", "normal"], [227, 5, 1, "", "randint"], [228, 5, 1, "", "seed"], [229, 5, 1, "", "split"], [230, 5, 1, "", "truncated_normal"], [231, 5, 1, "", "uniform"]], "mlx.nn": [[302, 3, 1, "", "ALiBi"], [303, 3, 1, "", "AvgPool1d"], [304, 3, 1, "", "AvgPool2d"], [305, 3, 1, "", "BatchNorm"], [306, 3, 1, "", "Conv1d"], [307, 3, 1, "", "Conv2d"], [308, 3, 1, "", "Conv3d"], [309, 3, 1, "", "Dropout"], [310, 3, 1, "", "Dropout2d"], [311, 3, 1, "", "Dropout3d"], [312, 3, 1, "", "Embedding"], [313, 3, 1, "", "GELU"], [314, 3, 1, "", "GLU"], [315, 3, 1, "", "GRU"], [316, 3, 1, "", "GroupNorm"], [317, 3, 1, "", "HardShrink"], [318, 3, 1, "", "HardTanh"], [319, 3, 1, "", "Hardswish"], [320, 3, 1, "", "InstanceNorm"], [321, 3, 1, "", "LSTM"], [322, 3, 1, "", "LayerNorm"], [323, 3, 1, "", "LeakyReLU"], [324, 3, 1, "", "Linear"], [325, 3, 1, "", "MaxPool1d"], [326, 3, 1, "", "MaxPool2d"], [327, 3, 1, "", "Mish"], [420, 3, 1, "", "Module"], [348, 3, 1, "", "MultiHeadAttention"], [349, 3, 1, "", "PReLU"], [350, 3, 1, "", "QuantizedEmbedding"], [351, 3, 1, "", "QuantizedLinear"], [352, 3, 1, "", "RMSNorm"], [353, 3, 1, "", "RNN"], [354, 3, 1, "", "ReLU"], [355, 3, 1, "", "ReLU6"], [356, 3, 1, "", "RoPE"], [357, 3, 1, "", "SELU"], [358, 3, 1, "", "Sequential"], [359, 3, 1, "", "SiLU"], [360, 3, 1, "", "SinusoidalPositionalEncoding"], [361, 3, 1, "", "Softmax"], [362, 3, 1, "", "Softmin"], [363, 3, 1, "", "Softplus"], [364, 3, 1, "", "Softshrink"], [365, 3, 1, "", "Softsign"], [366, 3, 1, "", "Step"], [367, 3, 1, "", "Tanh"], [368, 3, 1, "", "Transformer"], [369, 3, 1, "", "Upsample"], [378, 3, 1, "", "elu"], [379, 3, 1, "", "gelu"], [380, 3, 1, "", "gelu_approx"], [381, 3, 1, "", "gelu_fast_approx"], [382, 3, 1, "", "glu"], [383, 3, 1, "", "hard_shrink"], [384, 3, 1, "", "hard_tanh"], [385, 3, 1, "", "hardswish"], [386, 3, 1, "", "leaky_relu"], [387, 3, 1, "", "log_sigmoid"], [388, 3, 1, "", "log_softmax"], [403, 3, 1, "", "mish"], [404, 3, 1, "", "prelu"], [284, 5, 1, "", "quantize"], [405, 3, 1, "", "relu"], [406, 3, 1, "", "relu6"], [407, 3, 1, "", "selu"], [408, 3, 1, "", "sigmoid"], [409, 3, 1, "", "silu"], [410, 3, 1, "", "softmax"], [411, 3, 1, "", "softmin"], [412, 3, 1, "", "softplus"], [413, 3, 1, "", "softshrink"], [414, 3, 1, "", "step"], [415, 3, 1, "", "tanh"], [285, 5, 1, "", "value_and_grad"]], "mlx.nn.Module": [[328, 4, 1, "", "apply"], [329, 4, 1, "", "apply_to_modules"], [330, 4, 1, "", "children"], [331, 4, 1, "", "eval"], [332, 4, 1, "", "filter_and_map"], [333, 4, 1, "", "freeze"], [334, 4, 1, "", "leaf_modules"], [335, 4, 1, "", "load_weights"], [336, 4, 1, "", "modules"], [337, 4, 1, "", "named_modules"], [338, 4, 1, "", "parameters"], [339, 4, 1, "", "save_weights"], [340, 4, 1, "", "set_dtype"], [341, 6, 1, "", "state"], [342, 4, 1, "", "train"], [343, 4, 1, "", "trainable_parameters"], [344, 6, 1, "", "training"], [345, 4, 1, "", "unfreeze"], [346, 4, 1, "", "update"], [347, 4, 1, "", "update_modules"]], "mlx.nn.init": [[370, 5, 1, "", "constant"], [371, 5, 1, "", "glorot_normal"], [372, 5, 1, "", "glorot_uniform"], [373, 5, 1, "", "he_normal"], [374, 5, 1, "", "he_uniform"], [375, 5, 1, "", "identity"], [376, 5, 1, "", "normal"], [377, 5, 1, "", "uniform"]], "mlx.nn.losses": [[389, 3, 1, "", "binary_cross_entropy"], [390, 3, 1, "", "cosine_similarity_loss"], [391, 3, 1, "", "cross_entropy"], [392, 3, 1, "", "gaussian_nll_loss"], [393, 3, 1, "", "hinge_loss"], [394, 3, 1, "", "huber_loss"], [395, 3, 1, "", "kl_div_loss"], [396, 3, 1, "", "l1_loss"], [397, 3, 1, "", "log_cosh_loss"], [398, 3, 1, "", "margin_ranking_loss"], [399, 3, 1, "", "mse_loss"], [400, 3, 1, "", "nll_loss"], [401, 3, 1, "", "smooth_l1_loss"], [402, 3, 1, "", "triplet_loss"]], "mlx.optimizers": [[423, 3, 1, "", "AdaDelta"], [424, 3, 1, "", "Adafactor"], [425, 3, 1, "", "Adagrad"], [426, 3, 1, "", "Adam"], [427, 3, 1, "", "AdamW"], [428, 3, 1, "", "Adamax"], [429, 3, 1, "", "Lion"], [442, 3, 1, "", "Optimizer"], [434, 3, 1, "", "RMSprop"], [435, 3, 1, "", "SGD"], [286, 5, 1, "", "clip_grad_norm"], [436, 5, 1, "", "cosine_decay"], [437, 5, 1, "", "exponential_decay"], [438, 5, 1, "", "join_schedules"], [439, 5, 1, "", "linear_schedule"], [440, 5, 1, "", "step_decay"]], "mlx.optimizers.Optimizer": [[430, 4, 1, "", "apply_gradients"], [431, 4, 1, "", "init"], [432, 6, 1, "", "state"], [433, 4, 1, "", "update"]], "mlx.utils": [[287, 5, 1, "", "tree_flatten"], [288, 5, 1, "", "tree_map"], [289, 5, 1, "", "tree_map_with_path"], [290, 5, 1, "", "tree_reduce"], [291, 5, 1, "", "tree_unflatten"]]}, "objtypes": {"0": "cpp:function", "1": "cpp:functionParam", "2": "cpp:templateParam", "3": "py:class", "4": "py:method", "5": "py:function", "6": "py:property"}, "objnames": {"0": ["cpp", "function", "C++ function"], "1": ["cpp", "functionParam", "C++ function parameter"], "2": ["cpp", "templateParam", "C++ template parameter"], "3": ["py", "class", "Python class"], "4": ["py", "method", "Python method"], "5": ["py", "function", "Python function"], "6": ["py", "property", "Python property"]}, "titleterms": {"oper": [0, 1, 421], "custom": 1, "extens": 1, "mlx": [1, 6, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440], "introduc": 1, "exampl": [1, 6, 447, 448, 455], "primit": 1, "us": [1, 451, 456], "implement": [1, 4], "cpu": 1, "back": 1, "end": 1, "gpu": 1, "transform": [1, 368, 445, 447, 449, 451, 453], "build": [1, 7], "bind": 1, "python": [1, 6, 7], "cmake": 1, "setuptool": 1, "usag": [1, 6], "result": 1, "script": [1, 4], "download": [1, 4], "code": [1, 4], "metal": [2, 7, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 300], "debugg": 2, "xcode": 2, "workflow": 2, "linear": [3, 299, 324], "regress": 3, "llm": 4, "infer": 4, "model": 4, "attent": 4, "layer": [4, 5, 418], "encod": 4, "full": [4, 151], "gener": 4, "put": 4, "all": [4, 14, 32, 448], "togeth": 4, "convert": 4, "weight": 4, "load": [4, 176, 422, 454], "benchmark": 4, "multi": 5, "perceptron": 5, "instal": [6, 7, 448], "api": [6, 7], "refer": 6, "c": [6, 7], "further": 6, "read": 6, "troubleshoot": 7, "from": [7, 450], "sourc": 7, "requir": 7, "option": 7, "binari": 7, "size": [7, 69], "minim": 7, "found": 7, "x86": 7, "shell": 7, "core": [8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 292], "devic": [8, 295], "dtype": [9, 46], "dtypecategori": 10, "ab": [11, 31], "add": 12, "addmm": 13, "allclos": 15, "ani": [16, 33], "arang": 17, "arcco": 18, "arccosh": 19, "arcsin": 20, "arcsinh": 21, "arctan": 22, "arctan2": 23, "arctanh": 24, "argmax": [25, 34], "argmin": [26, 35], "argpartit": 27, "argsort": 28, "arrai": [29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 293, 450, 454], "t": 30, "astyp": 36, "conj": [38, 94], "co": [39, 100], "cummax": [40, 102], "cummin": [41, 103], "cumprod": [42, 104], "cumsum": [43, 105], "diag": [44, 111], "diagon": [45, 112], "exp": [47, 128], "flatten": [48, 148], "item": 49, "items": 50, "log": [51, 177], "log10": [52, 178], "log1p": [53, 179], "log2": [54, 180], "logsumexp": [55, 185], "max": [56, 187], "mean": [57, 189], "min": [58, 202], "moveaxi": [59, 204], "nbyte": 60, "ndim": 61, "prod": [62, 216], "reciproc": [63, 232], "reshap": [64, 235], "round": [65, 237], "rsqrt": [66, 238], "shape": 67, "sin": [68, 248], "split": [70, 229, 252], "sqrt": [71, 253], "squar": [72, 254], "squeez": [73, 255], "sum": [74, 261], "swapax": [75, 262], "tolist": 76, "transpos": [77, 272], "var": [78, 277], "view": [79, 278], "array_equ": 80, "as_strid": 81, "atleast_1d": 82, "atleast_2d": 83, "atleast_3d": 84, "bitwise_and": 85, "bitwise_or": 86, "bitwise_xor": 87, "block_masked_mm": 88, "broadcast_to": 89, "ceil": 90, "clip": 91, "compil": [92, 447], "concaten": 93, "conjug": 95, "conv1d": [96, 306], "conv2d": [97, 307], "conv_gener": 98, "convolv": 99, "cosh": 101, "custom_funct": 106, "default_devic": 107, "default_stream": 108, "degre": 109, "dequant": 110, "disable_compil": 113, "distribut": [114, 115, 116, 117, 118, 296, 448], "group": 114, "all_gath": 115, "all_sum": 116, "init": [117, 370, 371, 372, 373, 374, 375, 376, 377, 431], "is_avail": [118, 196], "divid": 119, "divmod": 120, "einsum": 121, "einsum_path": 122, "enable_compil": 123, "equal": 124, "erf": 125, "erfinv": 126, "eval": [127, 331], "expand_dim": 129, "expm1": 130, "ey": 131, "fast": [132, 133, 134, 135, 297], "layer_norm": 132, "rms_norm": 133, "rope": [134, 356], "scaled_dot_product_attent": 135, "fft": [136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 298], "fft2": 137, "fftn": 138, "ifft": 139, "ifft2": 140, "ifftn": 141, "irfft": 142, "irfft2": 143, "irfftn": 144, "rfft": 145, "rfft2": 146, "rfftn": 147, "floor": 149, "floor_divid": 150, "gather_mm": 152, "gather_qmm": 153, "grad": [154, 301], "greater": 155, "greater_equ": 156, "hadamard_transform": 157, "ident": [158, 375], "inner": 159, "isclos": 160, "isinf": 161, "isnan": 162, "isneginf": 163, "isposinf": 164, "issubdtyp": 165, "jvp": 166, "left_shift": 167, "less": 168, "less_equ": 169, "linalg": [170, 171, 172, 173, 174], "choleski": 170, "inv": 171, "norm": 172, "qr": 173, "svd": 174, "linspac": 175, "logaddexp": 181, "logical_and": 182, "logical_not": 183, "logical_or": 184, "matmul": 186, "maximum": 188, "meshgrid": 190, "clear_cach": 191, "device_info": 192, "get_active_memori": 193, "get_cache_memori": 194, "get_peak_memori": 195, "reset_peak_memori": 197, "set_cache_limit": 198, "set_memory_limit": 199, "start_captur": 200, "stop_captur": 201, "minimum": 203, "multipli": 205, "nan_to_num": 206, "neg": 207, "new_stream": 208, "not_equ": 209, "ones": 210, "ones_lik": 211, "outer": 212, "pad": 213, "partit": 214, "power": 215, "quantiz": [217, 284], "quantized_matmul": 218, "radian": 219, "random": [220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 444], "bernoulli": 220, "categor": 221, "gumbel": 222, "kei": 223, "laplac": 224, "multivariate_norm": 225, "normal": [226, 376], "randint": 227, "seed": 228, "truncated_norm": 230, "uniform": [231, 377], "remaind": 233, "repeat": 234, "right_shift": 236, "save": [239, 422, 454], "save_gguf": 240, "save_safetensor": 241, "savez": 242, "savez_compress": 243, "set_default_devic": 244, "set_default_stream": 245, "sigmoid": [246, 408], "sign": 247, "sinh": 249, "softmax": [250, 361, 410], "sort": 251, "stack": 256, "std": 257, "stop_gradi": 258, "stream": [259, 292, 295, 456], "subtract": 260, "synchron": 263, "take": 264, "take_along_axi": 265, "tan": 266, "tanh": [267, 367, 415], "tensordot": 268, "tile": 269, "topk": 270, "trace": 271, "tri": 273, "tril": 274, "triu": 275, "value_and_grad": [276, 285], "vjp": 279, "vmap": 280, "where": 281, "zero": 282, "zeros_lik": 283, "nn": [284, 285, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415], "optim": [286, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442], "clip_grad_norm": 286, "util": [287, 288, 289, 290, 291, 446], "tree_flatten": 287, "tree_map": 288, "tree_map_with_path": 289, "tree_reduc": 290, "tree_unflatten": 291, "data": 294, "type": 294, "support": 294, "commun": [296, 448], "algebra": 299, "neural": 301, "network": 301, "quick": [301, 453], "start": [301, 448, 453], "The": 301, "modul": [301, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 420], "class": 301, "paramet": [301, 338], "updat": [301, 346, 433, 450], "inspect": 301, "valu": 301, "alibi": 302, "avgpool1d": 303, "avgpool2d": 304, "batchnorm": 305, "conv3d": 308, "dropout": 309, "dropout2d": 310, "dropout3d": 311, "embed": 312, "gelu": [313, 379], "glu": [314, 382], "gru": 315, "groupnorm": 316, "hardshrink": 317, "hardtanh": 318, "hardswish": [319, 385], "instancenorm": 320, "lstm": 321, "layernorm": 322, "leakyrelu": 323, "maxpool1d": 325, "maxpool2d": 326, "mish": [327, 403], "appli": 328, "apply_to_modul": 329, "children": 330, "filter_and_map": 332, "freez": 333, "leaf_modul": 334, "load_weight": 335, "named_modul": 337, "save_weight": 339, "set_dtyp": 340, "state": [341, 432], "train": [342, 344, 447, 448], "trainable_paramet": 343, "unfreez": 345, "update_modul": 347, "multiheadattent": 348, "prelu": [349, 404], "quantizedembed": 350, "quantizedlinear": 351, "rmsnorm": 352, "rnn": 353, "relu": [354, 405], "relu6": [355, 406], "selu": [357, 407], "sequenti": 358, "silu": [359, 409], "sinusoidalpositionalencod": 360, "softmin": [362, 411], "softplu": [363, 412], "softshrink": [364, 413], "softsign": 365, "step": [366, 414], "upsampl": 369, "constant": 370, "glorot_norm": 371, "glorot_uniform": 372, "he_norm": 373, "he_uniform": 374, "elu": 378, "gelu_approx": 380, "gelu_fast_approx": 381, "hard_shrink": 383, "hard_tanh": 384, "leaky_relu": 386, "log_sigmoid": 387, "log_softmax": 388, "loss": [389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 419], "binary_cross_entropi": 389, "cosine_similarity_loss": 390, "cross_entropi": 391, "gaussian_nll_loss": 392, "hinge_loss": 393, "huber_loss": 394, "kl_div_loss": 395, "l1_loss": 396, "log_cosh_loss": 397, "margin_ranking_loss": 398, "mse_loss": 399, "nll_loss": 400, "smooth_l1_loss": 401, "triplet_loss": 402, "function": [416, 419, 447, 449, 453], "initi": 417, "adadelta": 423, "adafactor": 424, "adagrad": 425, "adam": 426, "adamw": 427, "adamax": 428, "lion": 429, "apply_gradi": 430, "rmsprop": 434, "sgd": 435, "cosine_decai": 436, "exponential_decai": 437, "join_schedul": 438, "linear_schedul": 439, "step_decai": 440, "common": 441, "schedul": 443, "tree": 446, "basic": [447, 453], "speedup": 447, "debug": 447, "pure": 447, "graph": [447, 451, 453], "get": 448, "mpi": 448, "set": 448, "up": 448, "remot": 448, "host": 448, "tune": 448, "reduc": 448, "automat": 449, "differenti": 449, "vector": 449, "index": 450, "differ": 450, "numpi": [450, 452], "In": 450, "place": 450, "lazi": 451, "evalu": 451, "why": 451, "comput": 451, "onli": 451, "what": 451, "you": 451, "when": 451, "convers": 452, "other": 452, "framework": 452, "pytorch": 452, "jax": 452, "tensorflow": 452, "guid": 453, "serial": 454, "format": 454, "unifi": 455, "memori": 455, "A": 455, "simpl": 455, "specifi": 456}, "envversion": {"sphinx.domains.c": 3, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 9, "sphinx.domains.index": 1, "sphinx.domains.javascript": 3, "sphinx.domains.math": 2, "sphinx.domains.python": 4, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.intersphinx": 1, "sphinx": 60}, "alltitles": {"Operations": [[0, "operations"], [1, "operations"], [421, "operations"]], "Custom Extensions in MLX": [[1, "custom-extensions-in-mlx"]], "Introducing the Example": [[1, "introducing-the-example"]], "Operations and Primitives": [[1, "operations-and-primitives"]], "Primitives": [[1, "primitives"]], "Using the Primitive": [[1, "using-the-primitive"]], "Implementing the Primitive": [[1, "implementing-the-primitive"]], "Implementing the CPU Back-end": [[1, "implementing-the-cpu-back-end"]], "Implementing the GPU Back-end": [[1, "implementing-the-gpu-back-end"]], "Primitive Transforms": [[1, "primitive-transforms"]], "Building and Binding": [[1, "building-and-binding"]], "Binding to Python": [[1, "binding-to-python"]], "Building with CMake": [[1, "building-with-cmake"]], "Building with setuptools": [[1, "building-with-setuptools"]], "Usage": [[1, "usage"], [6, null]], "Results": [[1, "results"]], "Scripts": [[1, "scripts"], [4, "scripts"]], "Download the code": [[1, null], [4, null]], "Metal Debugger": [[2, "metal-debugger"]], "Xcode Workflow": [[2, "xcode-workflow"]], "Linear Regression": [[3, "linear-regression"]], "LLM inference": [[4, "llm-inference"]], "Implementing the model": [[4, "implementing-the-model"]], "Attention layer": [[4, "attention-layer"]], "Encoder layer": [[4, "encoder-layer"]], "Full model": [[4, "full-model"]], "Generation": [[4, "generation"]], "Putting it all together": [[4, "putting-it-all-together"]], "Converting the weights": [[4, "converting-the-weights"]], "Weight loading and benchmarking": [[4, "weight-loading-and-benchmarking"]], "Multi-Layer Perceptron": [[5, "multi-layer-perceptron"]], "MLX": [[6, "mlx"]], "Install": [[6, null]], "Examples": [[6, null]], "Python API Reference": [[6, null]], "C++ API Reference": [[6, null]], "Further Reading": [[6, null]], "Build and Install": [[7, "build-and-install"]], "Python Installation": [[7, "python-installation"]], "Troubleshooting": [[7, "troubleshooting"], [7, "id2"]], "Build from source": [[7, "build-from-source"]], "Build Requirements": [[7, "build-requirements"]], "Python API": [[7, "python-api"]], "C++ API": [[7, "c-api"]], "Build Options": [[7, "id3"]], "Binary Size Minimization": [[7, "binary-size-minimization"]], "Metal not found": [[7, "metal-not-found"]], "x86 Shell": [[7, "x86-shell"]], "mlx.core.Device": [[8, "mlx-core-device"]], "mlx.core.Dtype": [[9, "mlx-core-dtype"]], "mlx.core.DtypeCategory": [[10, "mlx-core-dtypecategory"]], "mlx.core.abs": [[11, "mlx-core-abs"]], "mlx.core.add": [[12, "mlx-core-add"]], "mlx.core.addmm": [[13, "mlx-core-addmm"]], "mlx.core.all": [[14, "mlx-core-all"]], "mlx.core.allclose": [[15, "mlx-core-allclose"]], "mlx.core.any": [[16, "mlx-core-any"]], "mlx.core.arange": [[17, "mlx-core-arange"]], "mlx.core.arccos": [[18, "mlx-core-arccos"]], "mlx.core.arccosh": [[19, "mlx-core-arccosh"]], "mlx.core.arcsin": [[20, "mlx-core-arcsin"]], "mlx.core.arcsinh": [[21, "mlx-core-arcsinh"]], "mlx.core.arctan": [[22, "mlx-core-arctan"]], "mlx.core.arctan2": [[23, "mlx-core-arctan2"]], "mlx.core.arctanh": [[24, "mlx-core-arctanh"]], "mlx.core.argmax": [[25, "mlx-core-argmax"]], "mlx.core.argmin": [[26, "mlx-core-argmin"]], "mlx.core.argpartition": [[27, "mlx-core-argpartition"]], "mlx.core.argsort": [[28, "mlx-core-argsort"]], "mlx.core.array": [[29, "mlx-core-array"]], "mlx.core.array.T": [[30, "mlx-core-array-t"]], "mlx.core.array.abs": [[31, "mlx-core-array-abs"]], "mlx.core.array.all": [[32, "mlx-core-array-all"]], "mlx.core.array.any": [[33, "mlx-core-array-any"]], "mlx.core.array.argmax": [[34, "mlx-core-array-argmax"]], "mlx.core.array.argmin": [[35, "mlx-core-array-argmin"]], "mlx.core.array.astype": [[36, "mlx-core-array-astype"]], "mlx.core.array.at": [[37, "mlx-core-array-at"]], "mlx.core.array.conj": [[38, "mlx-core-array-conj"]], "mlx.core.array.cos": [[39, "mlx-core-array-cos"]], "mlx.core.array.cummax": [[40, "mlx-core-array-cummax"]], "mlx.core.array.cummin": [[41, "mlx-core-array-cummin"]], "mlx.core.array.cumprod": [[42, "mlx-core-array-cumprod"]], "mlx.core.array.cumsum": [[43, "mlx-core-array-cumsum"]], "mlx.core.array.diag": [[44, "mlx-core-array-diag"]], "mlx.core.array.diagonal": [[45, "mlx-core-array-diagonal"]], "mlx.core.array.dtype": [[46, "mlx-core-array-dtype"]], "mlx.core.array.exp": [[47, "mlx-core-array-exp"]], "mlx.core.array.flatten": [[48, "mlx-core-array-flatten"]], "mlx.core.array.item": [[49, "mlx-core-array-item"]], "mlx.core.array.itemsize": [[50, "mlx-core-array-itemsize"]], "mlx.core.array.log": [[51, "mlx-core-array-log"]], "mlx.core.array.log10": [[52, "mlx-core-array-log10"]], "mlx.core.array.log1p": [[53, "mlx-core-array-log1p"]], "mlx.core.array.log2": [[54, "mlx-core-array-log2"]], "mlx.core.array.logsumexp": [[55, "mlx-core-array-logsumexp"]], "mlx.core.array.max": [[56, "mlx-core-array-max"]], "mlx.core.array.mean": [[57, "mlx-core-array-mean"]], "mlx.core.array.min": [[58, "mlx-core-array-min"]], "mlx.core.array.moveaxis": [[59, "mlx-core-array-moveaxis"]], "mlx.core.array.nbytes": [[60, "mlx-core-array-nbytes"]], "mlx.core.array.ndim": [[61, "mlx-core-array-ndim"]], "mlx.core.array.prod": [[62, "mlx-core-array-prod"]], "mlx.core.array.reciprocal": [[63, "mlx-core-array-reciprocal"]], "mlx.core.array.reshape": [[64, "mlx-core-array-reshape"]], "mlx.core.array.round": [[65, "mlx-core-array-round"]], "mlx.core.array.rsqrt": [[66, "mlx-core-array-rsqrt"]], "mlx.core.array.shape": [[67, "mlx-core-array-shape"]], "mlx.core.array.sin": [[68, "mlx-core-array-sin"]], "mlx.core.array.size": [[69, "mlx-core-array-size"]], "mlx.core.array.split": [[70, "mlx-core-array-split"]], "mlx.core.array.sqrt": [[71, "mlx-core-array-sqrt"]], "mlx.core.array.square": [[72, "mlx-core-array-square"]], "mlx.core.array.squeeze": [[73, "mlx-core-array-squeeze"]], "mlx.core.array.sum": [[74, "mlx-core-array-sum"]], "mlx.core.array.swapaxes": [[75, "mlx-core-array-swapaxes"]], "mlx.core.array.tolist": [[76, "mlx-core-array-tolist"]], "mlx.core.array.transpose": [[77, "mlx-core-array-transpose"]], "mlx.core.array.var": [[78, "mlx-core-array-var"]], "mlx.core.array.view": [[79, "mlx-core-array-view"]], "mlx.core.array_equal": [[80, "mlx-core-array-equal"]], "mlx.core.as_strided": [[81, "mlx-core-as-strided"]], "mlx.core.atleast_1d": [[82, "mlx-core-atleast-1d"]], "mlx.core.atleast_2d": [[83, "mlx-core-atleast-2d"]], "mlx.core.atleast_3d": [[84, "mlx-core-atleast-3d"]], "mlx.core.bitwise_and": [[85, "mlx-core-bitwise-and"]], "mlx.core.bitwise_or": [[86, "mlx-core-bitwise-or"]], "mlx.core.bitwise_xor": [[87, "mlx-core-bitwise-xor"]], "mlx.core.block_masked_mm": [[88, "mlx-core-block-masked-mm"]], "mlx.core.broadcast_to": [[89, "mlx-core-broadcast-to"]], "mlx.core.ceil": [[90, "mlx-core-ceil"]], "mlx.core.clip": [[91, "mlx-core-clip"]], "mlx.core.compile": [[92, "mlx-core-compile"]], "mlx.core.concatenate": [[93, "mlx-core-concatenate"]], "mlx.core.conj": [[94, "mlx-core-conj"]], "mlx.core.conjugate": [[95, "mlx-core-conjugate"]], "mlx.core.conv1d": [[96, "mlx-core-conv1d"]], "mlx.core.conv2d": [[97, "mlx-core-conv2d"]], "mlx.core.conv_general": [[98, "mlx-core-conv-general"]], "mlx.core.convolve": [[99, "mlx-core-convolve"]], "mlx.core.cos": [[100, "mlx-core-cos"]], "mlx.core.cosh": [[101, "mlx-core-cosh"]], "mlx.core.cummax": [[102, "mlx-core-cummax"]], "mlx.core.cummin": [[103, "mlx-core-cummin"]], "mlx.core.cumprod": [[104, "mlx-core-cumprod"]], "mlx.core.cumsum": [[105, "mlx-core-cumsum"]], "mlx.core.custom_function": [[106, "mlx-core-custom-function"]], "mlx.core.default_device": [[107, "mlx-core-default-device"]], "mlx.core.default_stream": [[108, "mlx-core-default-stream"]], "mlx.core.degrees": [[109, "mlx-core-degrees"]], "mlx.core.dequantize": [[110, "mlx-core-dequantize"]], "mlx.core.diag": [[111, "mlx-core-diag"]], "mlx.core.diagonal": [[112, "mlx-core-diagonal"]], "mlx.core.disable_compile": [[113, "mlx-core-disable-compile"]], "mlx.core.distributed.Group": [[114, "mlx-core-distributed-group"]], "mlx.core.distributed.all_gather": [[115, "mlx-core-distributed-all-gather"]], "mlx.core.distributed.all_sum": [[116, "mlx-core-distributed-all-sum"]], "mlx.core.distributed.init": [[117, "mlx-core-distributed-init"]], "mlx.core.distributed.is_available": [[118, "mlx-core-distributed-is-available"]], "mlx.core.divide": [[119, "mlx-core-divide"]], "mlx.core.divmod": [[120, "mlx-core-divmod"]], "mlx.core.einsum": [[121, "mlx-core-einsum"]], "mlx.core.einsum_path": [[122, "mlx-core-einsum-path"]], "mlx.core.enable_compile": [[123, "mlx-core-enable-compile"]], "mlx.core.equal": [[124, "mlx-core-equal"]], "mlx.core.erf": [[125, "mlx-core-erf"]], "mlx.core.erfinv": [[126, "mlx-core-erfinv"]], "mlx.core.eval": [[127, "mlx-core-eval"]], "mlx.core.exp": [[128, "mlx-core-exp"]], "mlx.core.expand_dims": [[129, "mlx-core-expand-dims"]], "mlx.core.expm1": [[130, "mlx-core-expm1"]], "mlx.core.eye": [[131, "mlx-core-eye"]], "mlx.core.fast.layer_norm": [[132, "mlx-core-fast-layer-norm"]], "mlx.core.fast.rms_norm": [[133, "mlx-core-fast-rms-norm"]], "mlx.core.fast.rope": [[134, "mlx-core-fast-rope"]], "mlx.core.fast.scaled_dot_product_attention": [[135, "mlx-core-fast-scaled-dot-product-attention"]], "mlx.core.fft.fft": [[136, "mlx-core-fft-fft"]], "mlx.core.fft.fft2": [[137, "mlx-core-fft-fft2"]], "mlx.core.fft.fftn": [[138, "mlx-core-fft-fftn"]], "mlx.core.fft.ifft": [[139, "mlx-core-fft-ifft"]], "mlx.core.fft.ifft2": [[140, "mlx-core-fft-ifft2"]], "mlx.core.fft.ifftn": [[141, "mlx-core-fft-ifftn"]], "mlx.core.fft.irfft": [[142, "mlx-core-fft-irfft"]], "mlx.core.fft.irfft2": [[143, "mlx-core-fft-irfft2"]], "mlx.core.fft.irfftn": [[144, "mlx-core-fft-irfftn"]], "mlx.core.fft.rfft": [[145, "mlx-core-fft-rfft"]], "mlx.core.fft.rfft2": [[146, "mlx-core-fft-rfft2"]], "mlx.core.fft.rfftn": [[147, "mlx-core-fft-rfftn"]], "mlx.core.flatten": [[148, "mlx-core-flatten"]], "mlx.core.floor": [[149, "mlx-core-floor"]], "mlx.core.floor_divide": [[150, "mlx-core-floor-divide"]], "mlx.core.full": [[151, "mlx-core-full"]], "mlx.core.gather_mm": [[152, "mlx-core-gather-mm"]], "mlx.core.gather_qmm": [[153, "mlx-core-gather-qmm"]], "mlx.core.grad": [[154, "mlx-core-grad"]], "mlx.core.greater": [[155, "mlx-core-greater"]], "mlx.core.greater_equal": [[156, "mlx-core-greater-equal"]], "mlx.core.hadamard_transform": [[157, "mlx-core-hadamard-transform"]], "mlx.core.identity": [[158, "mlx-core-identity"]], "mlx.core.inner": [[159, "mlx-core-inner"]], "mlx.core.isclose": [[160, "mlx-core-isclose"]], "mlx.core.isinf": [[161, "mlx-core-isinf"]], "mlx.core.isnan": [[162, "mlx-core-isnan"]], "mlx.core.isneginf": [[163, "mlx-core-isneginf"]], "mlx.core.isposinf": [[164, "mlx-core-isposinf"]], "mlx.core.issubdtype": [[165, "mlx-core-issubdtype"]], "mlx.core.jvp": [[166, "mlx-core-jvp"]], "mlx.core.left_shift": [[167, "mlx-core-left-shift"]], "mlx.core.less": [[168, "mlx-core-less"]], "mlx.core.less_equal": [[169, "mlx-core-less-equal"]], "mlx.core.linalg.cholesky": [[170, "mlx-core-linalg-cholesky"]], "mlx.core.linalg.inv": [[171, "mlx-core-linalg-inv"]], "mlx.core.linalg.norm": [[172, "mlx-core-linalg-norm"]], "mlx.core.linalg.qr": [[173, "mlx-core-linalg-qr"]], "mlx.core.linalg.svd": [[174, "mlx-core-linalg-svd"]], "mlx.core.linspace": [[175, "mlx-core-linspace"]], "mlx.core.load": [[176, "mlx-core-load"]], "mlx.core.log": [[177, "mlx-core-log"]], "mlx.core.log10": [[178, "mlx-core-log10"]], "mlx.core.log1p": [[179, "mlx-core-log1p"]], "mlx.core.log2": [[180, "mlx-core-log2"]], "mlx.core.logaddexp": [[181, "mlx-core-logaddexp"]], "mlx.core.logical_and": [[182, "mlx-core-logical-and"]], "mlx.core.logical_not": [[183, "mlx-core-logical-not"]], "mlx.core.logical_or": [[184, "mlx-core-logical-or"]], "mlx.core.logsumexp": [[185, "mlx-core-logsumexp"]], "mlx.core.matmul": [[186, "mlx-core-matmul"]], "mlx.core.max": [[187, "mlx-core-max"]], "mlx.core.maximum": [[188, "mlx-core-maximum"]], "mlx.core.mean": [[189, "mlx-core-mean"]], "mlx.core.meshgrid": [[190, "mlx-core-meshgrid"]], "mlx.core.metal.clear_cache": [[191, "mlx-core-metal-clear-cache"]], "mlx.core.metal.device_info": [[192, "mlx-core-metal-device-info"]], "mlx.core.metal.get_active_memory": [[193, "mlx-core-metal-get-active-memory"]], "mlx.core.metal.get_cache_memory": [[194, "mlx-core-metal-get-cache-memory"]], "mlx.core.metal.get_peak_memory": [[195, "mlx-core-metal-get-peak-memory"]], "mlx.core.metal.is_available": [[196, "mlx-core-metal-is-available"]], "mlx.core.metal.reset_peak_memory": [[197, "mlx-core-metal-reset-peak-memory"]], "mlx.core.metal.set_cache_limit": [[198, "mlx-core-metal-set-cache-limit"]], "mlx.core.metal.set_memory_limit": [[199, "mlx-core-metal-set-memory-limit"]], "mlx.core.metal.start_capture": [[200, "mlx-core-metal-start-capture"]], "mlx.core.metal.stop_capture": [[201, "mlx-core-metal-stop-capture"]], "mlx.core.min": [[202, "mlx-core-min"]], "mlx.core.minimum": [[203, "mlx-core-minimum"]], "mlx.core.moveaxis": [[204, "mlx-core-moveaxis"]], "mlx.core.multiply": [[205, "mlx-core-multiply"]], "mlx.core.nan_to_num": [[206, "mlx-core-nan-to-num"]], "mlx.core.negative": [[207, "mlx-core-negative"]], "mlx.core.new_stream": [[208, "mlx-core-new-stream"]], "mlx.core.not_equal": [[209, "mlx-core-not-equal"]], "mlx.core.ones": [[210, "mlx-core-ones"]], "mlx.core.ones_like": [[211, "mlx-core-ones-like"]], "mlx.core.outer": [[212, "mlx-core-outer"]], "mlx.core.pad": [[213, "mlx-core-pad"]], "mlx.core.partition": [[214, "mlx-core-partition"]], "mlx.core.power": [[215, "mlx-core-power"]], "mlx.core.prod": [[216, "mlx-core-prod"]], "mlx.core.quantize": [[217, "mlx-core-quantize"]], "mlx.core.quantized_matmul": [[218, "mlx-core-quantized-matmul"]], "mlx.core.radians": [[219, "mlx-core-radians"]], "mlx.core.random.bernoulli": [[220, "mlx-core-random-bernoulli"]], "mlx.core.random.categorical": [[221, "mlx-core-random-categorical"]], "mlx.core.random.gumbel": [[222, "mlx-core-random-gumbel"]], "mlx.core.random.key": [[223, "mlx-core-random-key"]], "mlx.core.random.laplace": [[224, "mlx-core-random-laplace"]], "mlx.core.random.multivariate_normal": [[225, "mlx-core-random-multivariate-normal"]], "mlx.core.random.normal": [[226, "mlx-core-random-normal"]], "mlx.core.random.randint": [[227, "mlx-core-random-randint"]], "mlx.core.random.seed": [[228, "mlx-core-random-seed"]], "mlx.core.random.split": [[229, "mlx-core-random-split"]], "mlx.core.random.truncated_normal": [[230, "mlx-core-random-truncated-normal"]], "mlx.core.random.uniform": [[231, "mlx-core-random-uniform"]], "mlx.core.reciprocal": [[232, "mlx-core-reciprocal"]], "mlx.core.remainder": [[233, "mlx-core-remainder"]], "mlx.core.repeat": [[234, "mlx-core-repeat"]], "mlx.core.reshape": [[235, "mlx-core-reshape"]], "mlx.core.right_shift": [[236, "mlx-core-right-shift"]], "mlx.core.round": [[237, "mlx-core-round"]], "mlx.core.rsqrt": [[238, "mlx-core-rsqrt"]], "mlx.core.save": [[239, "mlx-core-save"]], "mlx.core.save_gguf": [[240, "mlx-core-save-gguf"]], "mlx.core.save_safetensors": [[241, "mlx-core-save-safetensors"]], "mlx.core.savez": [[242, "mlx-core-savez"]], "mlx.core.savez_compressed": [[243, "mlx-core-savez-compressed"]], "mlx.core.set_default_device": [[244, "mlx-core-set-default-device"]], "mlx.core.set_default_stream": [[245, "mlx-core-set-default-stream"]], "mlx.core.sigmoid": [[246, "mlx-core-sigmoid"]], "mlx.core.sign": [[247, "mlx-core-sign"]], "mlx.core.sin": [[248, "mlx-core-sin"]], "mlx.core.sinh": [[249, "mlx-core-sinh"]], "mlx.core.softmax": [[250, "mlx-core-softmax"]], "mlx.core.sort": [[251, "mlx-core-sort"]], "mlx.core.split": [[252, "mlx-core-split"]], "mlx.core.sqrt": [[253, "mlx-core-sqrt"]], "mlx.core.square": [[254, "mlx-core-square"]], "mlx.core.squeeze": [[255, "mlx-core-squeeze"]], "mlx.core.stack": [[256, "mlx-core-stack"]], "mlx.core.std": [[257, "mlx-core-std"]], "mlx.core.stop_gradient": [[258, "mlx-core-stop-gradient"]], "mlx.core.stream": [[259, "mlx-core-stream"]], "mlx.core.subtract": [[260, "mlx-core-subtract"]], "mlx.core.sum": [[261, "mlx-core-sum"]], "mlx.core.swapaxes": [[262, "mlx-core-swapaxes"]], "mlx.core.synchronize": [[263, "mlx-core-synchronize"]], "mlx.core.take": [[264, "mlx-core-take"]], "mlx.core.take_along_axis": [[265, "mlx-core-take-along-axis"]], "mlx.core.tan": [[266, "mlx-core-tan"]], "mlx.core.tanh": [[267, "mlx-core-tanh"]], "mlx.core.tensordot": [[268, "mlx-core-tensordot"]], "mlx.core.tile": [[269, "mlx-core-tile"]], "mlx.core.topk": [[270, "mlx-core-topk"]], "mlx.core.trace": [[271, "mlx-core-trace"]], "mlx.core.transpose": [[272, "mlx-core-transpose"]], "mlx.core.tri": [[273, "mlx-core-tri"]], "mlx.core.tril": [[274, "mlx-core-tril"]], "mlx.core.triu": [[275, "mlx-core-triu"]], "mlx.core.value_and_grad": [[276, "mlx-core-value-and-grad"]], "mlx.core.var": [[277, "mlx-core-var"]], "mlx.core.view": [[278, "mlx-core-view"]], "mlx.core.vjp": [[279, "mlx-core-vjp"]], "mlx.core.vmap": [[280, "mlx-core-vmap"]], "mlx.core.where": [[281, "mlx-core-where"]], "mlx.core.zeros": [[282, "mlx-core-zeros"]], "mlx.core.zeros_like": [[283, "mlx-core-zeros-like"]], "mlx.nn.quantize": [[284, "mlx-nn-quantize"]], "mlx.nn.value_and_grad": [[285, "mlx-nn-value-and-grad"]], "mlx.optimizers.clip_grad_norm": [[286, "mlx-optimizers-clip-grad-norm"]], "mlx.utils.tree_flatten": [[287, "mlx-utils-tree-flatten"]], "mlx.utils.tree_map": [[288, "mlx-utils-tree-map"]], "mlx.utils.tree_map_with_path": [[289, "mlx-utils-tree-map-with-path"]], "mlx.utils.tree_reduce": [[290, "mlx-utils-tree-reduce"]], "mlx.utils.tree_unflatten": [[291, "mlx-utils-tree-unflatten"]], "mlx.core.Stream": [[292, "mlx-core-stream"]], "Array": [[293, "array"]], "Data Types": [[294, "data-types"]], "Supported Data Types": [[294, "id2"]], "Devices and Streams": [[295, "devices-and-streams"]], "Distributed Communication": [[296, "distributed-communication"], [448, "distributed-communication"]], "Fast": [[297, "fast"]], "FFT": [[298, "fft"]], "Linear Algebra": [[299, "linear-algebra"]], "Metal": [[300, "metal"]], "Neural Networks": [[301, "neural-networks"]], "Quick Start with Neural Networks": [[301, "quick-start-with-neural-networks"]], "The Module Class": [[301, "the-module-class"]], "Parameters": [[301, "parameters"]], "Updating the Parameters": [[301, "updating-the-parameters"]], "Inspecting Modules": [[301, "inspecting-modules"]], "Value and Grad": [[301, "value-and-grad"]], "mlx.nn.ALiBi": [[302, "mlx-nn-alibi"]], "mlx.nn.AvgPool1d": [[303, "mlx-nn-avgpool1d"]], "mlx.nn.AvgPool2d": [[304, "mlx-nn-avgpool2d"]], "mlx.nn.BatchNorm": [[305, "mlx-nn-batchnorm"]], "mlx.nn.Conv1d": [[306, "mlx-nn-conv1d"]], "mlx.nn.Conv2d": [[307, "mlx-nn-conv2d"]], "mlx.nn.Conv3d": [[308, "mlx-nn-conv3d"]], "mlx.nn.Dropout": [[309, "mlx-nn-dropout"]], "mlx.nn.Dropout2d": [[310, "mlx-nn-dropout2d"]], "mlx.nn.Dropout3d": [[311, "mlx-nn-dropout3d"]], "mlx.nn.Embedding": [[312, "mlx-nn-embedding"]], "mlx.nn.GELU": [[313, "mlx-nn-gelu"]], "mlx.nn.GLU": [[314, "mlx-nn-glu"]], "mlx.nn.GRU": [[315, "mlx-nn-gru"]], "mlx.nn.GroupNorm": [[316, "mlx-nn-groupnorm"]], "mlx.nn.HardShrink": [[317, "mlx-nn-hardshrink"]], "mlx.nn.HardTanh": [[318, "mlx-nn-hardtanh"]], "mlx.nn.Hardswish": [[319, "mlx-nn-hardswish"]], "mlx.nn.InstanceNorm": [[320, "mlx-nn-instancenorm"]], "mlx.nn.LSTM": [[321, "mlx-nn-lstm"]], "mlx.nn.LayerNorm": [[322, "mlx-nn-layernorm"]], "mlx.nn.LeakyReLU": [[323, "mlx-nn-leakyrelu"]], "mlx.nn.Linear": [[324, "mlx-nn-linear"]], "mlx.nn.MaxPool1d": [[325, "mlx-nn-maxpool1d"]], "mlx.nn.MaxPool2d": [[326, "mlx-nn-maxpool2d"]], "mlx.nn.Mish": [[327, "mlx-nn-mish"]], "mlx.nn.Module.apply": [[328, "mlx-nn-module-apply"]], "mlx.nn.Module.apply_to_modules": [[329, "mlx-nn-module-apply-to-modules"]], "mlx.nn.Module.children": [[330, "mlx-nn-module-children"]], "mlx.nn.Module.eval": [[331, "mlx-nn-module-eval"]], "mlx.nn.Module.filter_and_map": [[332, "mlx-nn-module-filter-and-map"]], "mlx.nn.Module.freeze": [[333, "mlx-nn-module-freeze"]], "mlx.nn.Module.leaf_modules": [[334, "mlx-nn-module-leaf-modules"]], "mlx.nn.Module.load_weights": [[335, "mlx-nn-module-load-weights"]], "mlx.nn.Module.modules": [[336, "mlx-nn-module-modules"]], "mlx.nn.Module.named_modules": [[337, "mlx-nn-module-named-modules"]], "mlx.nn.Module.parameters": [[338, "mlx-nn-module-parameters"]], "mlx.nn.Module.save_weights": [[339, "mlx-nn-module-save-weights"]], "mlx.nn.Module.set_dtype": [[340, "mlx-nn-module-set-dtype"]], "mlx.nn.Module.state": [[341, "mlx-nn-module-state"]], "mlx.nn.Module.train": [[342, "mlx-nn-module-train"]], "mlx.nn.Module.trainable_parameters": [[343, "mlx-nn-module-trainable-parameters"]], "mlx.nn.Module.training": [[344, "mlx-nn-module-training"]], "mlx.nn.Module.unfreeze": [[345, "mlx-nn-module-unfreeze"]], "mlx.nn.Module.update": [[346, "mlx-nn-module-update"]], "mlx.nn.Module.update_modules": [[347, "mlx-nn-module-update-modules"]], "mlx.nn.MultiHeadAttention": [[348, "mlx-nn-multiheadattention"]], "mlx.nn.PReLU": [[349, "mlx-nn-prelu"]], "mlx.nn.QuantizedEmbedding": [[350, "mlx-nn-quantizedembedding"]], "mlx.nn.QuantizedLinear": [[351, "mlx-nn-quantizedlinear"]], "mlx.nn.RMSNorm": [[352, "mlx-nn-rmsnorm"]], "mlx.nn.RNN": [[353, "mlx-nn-rnn"]], "mlx.nn.ReLU": [[354, "mlx-nn-relu"]], "mlx.nn.ReLU6": [[355, "mlx-nn-relu6"]], "mlx.nn.RoPE": [[356, "mlx-nn-rope"]], "mlx.nn.SELU": [[357, "mlx-nn-selu"]], "mlx.nn.Sequential": [[358, "mlx-nn-sequential"]], "mlx.nn.SiLU": [[359, "mlx-nn-silu"]], "mlx.nn.SinusoidalPositionalEncoding": [[360, "mlx-nn-sinusoidalpositionalencoding"]], "mlx.nn.Softmax": [[361, "mlx-nn-softmax"]], "mlx.nn.Softmin": [[362, "mlx-nn-softmin"]], "mlx.nn.Softplus": [[363, "mlx-nn-softplus"]], "mlx.nn.Softshrink": [[364, "mlx-nn-softshrink"]], "mlx.nn.Softsign": [[365, "mlx-nn-softsign"]], "mlx.nn.Step": [[366, "mlx-nn-step"]], "mlx.nn.Tanh": [[367, "mlx-nn-tanh"]], "mlx.nn.Transformer": [[368, "mlx-nn-transformer"]], "mlx.nn.Upsample": [[369, "mlx-nn-upsample"]], "mlx.nn.init.constant": [[370, "mlx-nn-init-constant"]], "mlx.nn.init.glorot_normal": [[371, "mlx-nn-init-glorot-normal"]], "mlx.nn.init.glorot_uniform": [[372, "mlx-nn-init-glorot-uniform"]], "mlx.nn.init.he_normal": [[373, "mlx-nn-init-he-normal"]], "mlx.nn.init.he_uniform": [[374, "mlx-nn-init-he-uniform"]], "mlx.nn.init.identity": [[375, "mlx-nn-init-identity"]], "mlx.nn.init.normal": [[376, "mlx-nn-init-normal"]], "mlx.nn.init.uniform": [[377, "mlx-nn-init-uniform"]], "mlx.nn.elu": [[378, "mlx-nn-elu"]], "mlx.nn.gelu": [[379, "mlx-nn-gelu"]], "mlx.nn.gelu_approx": [[380, "mlx-nn-gelu-approx"]], "mlx.nn.gelu_fast_approx": [[381, "mlx-nn-gelu-fast-approx"]], "mlx.nn.glu": [[382, "mlx-nn-glu"]], "mlx.nn.hard_shrink": [[383, "mlx-nn-hard-shrink"]], "mlx.nn.hard_tanh": [[384, "mlx-nn-hard-tanh"]], "mlx.nn.hardswish": [[385, "mlx-nn-hardswish"]], "mlx.nn.leaky_relu": [[386, "mlx-nn-leaky-relu"]], "mlx.nn.log_sigmoid": [[387, "mlx-nn-log-sigmoid"]], "mlx.nn.log_softmax": [[388, "mlx-nn-log-softmax"]], "mlx.nn.losses.binary_cross_entropy": [[389, "mlx-nn-losses-binary-cross-entropy"]], "mlx.nn.losses.cosine_similarity_loss": [[390, "mlx-nn-losses-cosine-similarity-loss"]], "mlx.nn.losses.cross_entropy": [[391, "mlx-nn-losses-cross-entropy"]], "mlx.nn.losses.gaussian_nll_loss": [[392, "mlx-nn-losses-gaussian-nll-loss"]], "mlx.nn.losses.hinge_loss": [[393, "mlx-nn-losses-hinge-loss"]], "mlx.nn.losses.huber_loss": [[394, "mlx-nn-losses-huber-loss"]], "mlx.nn.losses.kl_div_loss": [[395, "mlx-nn-losses-kl-div-loss"]], "mlx.nn.losses.l1_loss": [[396, "mlx-nn-losses-l1-loss"]], "mlx.nn.losses.log_cosh_loss": [[397, "mlx-nn-losses-log-cosh-loss"]], "mlx.nn.losses.margin_ranking_loss": [[398, "mlx-nn-losses-margin-ranking-loss"]], "mlx.nn.losses.mse_loss": [[399, "mlx-nn-losses-mse-loss"]], "mlx.nn.losses.nll_loss": [[400, "mlx-nn-losses-nll-loss"]], "mlx.nn.losses.smooth_l1_loss": [[401, "mlx-nn-losses-smooth-l1-loss"]], "mlx.nn.losses.triplet_loss": [[402, "mlx-nn-losses-triplet-loss"]], "mlx.nn.mish": [[403, "mlx-nn-mish"]], "mlx.nn.prelu": [[404, "mlx-nn-prelu"]], "mlx.nn.relu": [[405, "mlx-nn-relu"]], "mlx.nn.relu6": [[406, "mlx-nn-relu6"]], "mlx.nn.selu": [[407, "mlx-nn-selu"]], "mlx.nn.sigmoid": [[408, "mlx-nn-sigmoid"]], "mlx.nn.silu": [[409, "mlx-nn-silu"]], "mlx.nn.softmax": [[410, "mlx-nn-softmax"]], "mlx.nn.softmin": [[411, "mlx-nn-softmin"]], "mlx.nn.softplus": [[412, "mlx-nn-softplus"]], "mlx.nn.softshrink": [[413, "mlx-nn-softshrink"]], "mlx.nn.step": [[414, "mlx-nn-step"]], "mlx.nn.tanh": [[415, "mlx-nn-tanh"]], "Functions": [[416, "functions"]], "Initializers": [[417, "initializers"]], "Layers": [[418, "layers"]], "Loss Functions": [[419, "loss-functions"]], "Module": [[420, "module"]], "Optimizers": [[422, "optimizers"]], "Saving and Loading": [[422, "saving-and-loading"]], "mlx.optimizers.AdaDelta": [[423, "mlx-optimizers-adadelta"]], "mlx.optimizers.Adafactor": [[424, "mlx-optimizers-adafactor"]], "mlx.optimizers.Adagrad": [[425, "mlx-optimizers-adagrad"]], "mlx.optimizers.Adam": [[426, "mlx-optimizers-adam"]], "mlx.optimizers.AdamW": [[427, "mlx-optimizers-adamw"]], "mlx.optimizers.Adamax": [[428, "mlx-optimizers-adamax"]], "mlx.optimizers.Lion": [[429, "mlx-optimizers-lion"]], "mlx.optimizers.Optimizer.apply_gradients": [[430, "mlx-optimizers-optimizer-apply-gradients"]], "mlx.optimizers.Optimizer.init": [[431, "mlx-optimizers-optimizer-init"]], "mlx.optimizers.Optimizer.state": [[432, "mlx-optimizers-optimizer-state"]], "mlx.optimizers.Optimizer.update": [[433, "mlx-optimizers-optimizer-update"]], "mlx.optimizers.RMSprop": [[434, "mlx-optimizers-rmsprop"]], "mlx.optimizers.SGD": [[435, "mlx-optimizers-sgd"]], "mlx.optimizers.cosine_decay": [[436, "mlx-optimizers-cosine-decay"]], "mlx.optimizers.exponential_decay": [[437, "mlx-optimizers-exponential-decay"]], "mlx.optimizers.join_schedules": [[438, "mlx-optimizers-join-schedules"]], "mlx.optimizers.linear_schedule": [[439, "mlx-optimizers-linear-schedule"]], "mlx.optimizers.step_decay": [[440, "mlx-optimizers-step-decay"]], "Common Optimizers": [[441, "common-optimizers"]], "Optimizer": [[442, "optimizer"]], "Schedulers": [[443, "schedulers"]], "Random": [[444, "random"]], "Transforms": [[445, "transforms"]], "Tree Utils": [[446, "tree-utils"]], "Compilation": [[447, "compilation"]], "Basics of Compile": [[447, "basics-of-compile"]], "Example Speedup": [[447, "example-speedup"]], "Debugging": [[447, "debugging"]], "Pure Functions": [[447, "pure-functions"]], "Compiling Training Graphs": [[447, "compiling-training-graphs"]], "Transformations with Compile": [[447, "transformations-with-compile"]], "Getting Started": [[448, "getting-started"]], "Installing MPI": [[448, "installing-mpi"]], "Setting up Remote Hosts": [[448, "setting-up-remote-hosts"]], "Training Example": [[448, "training-example"]], "Tuning All Reduce": [[448, "tuning-all-reduce"]], "Function Transforms": [[449, "function-transforms"]], "Automatic Differentiation": [[449, "automatic-differentiation"]], "Automatic Vectorization": [[449, "automatic-vectorization"]], "Indexing Arrays": [[450, "indexing-arrays"]], "Differences from NumPy": [[450, "differences-from-numpy"]], "In Place Updates": [[450, "in-place-updates"]], "Lazy Evaluation": [[451, "lazy-evaluation"]], "Why Lazy Evaluation": [[451, "why-lazy-evaluation"]], "Transforming Compute Graphs": [[451, "transforming-compute-graphs"]], "Only Compute What You Use": [[451, "only-compute-what-you-use"]], "When to Evaluate": [[451, "when-to-evaluate"]], "Conversion to NumPy and Other Frameworks": [[452, "conversion-to-numpy-and-other-frameworks"]], "PyTorch": [[452, "pytorch"]], "JAX": [[452, "jax"]], "TensorFlow": [[452, "tensorflow"]], "Quick Start Guide": [[453, "quick-start-guide"]], "Basics": [[453, "basics"]], "Function and Graph Transformations": [[453, "function-and-graph-transformations"]], "Saving and Loading Arrays": [[454, "saving-and-loading-arrays"]], "Serialization Formats": [[454, "id1"]], "Unified Memory": [[455, "unified-memory"]], "A Simple Example": [[455, "a-simple-example"]], "Using Streams": [[456, "using-streams"]], "Specifying the Stream": [[456, "specifying-the-stream"]]}, "indexentries": {"abs (c++ function)": [[0, "_CPPv43absRK5array14StreamOrDevice"]], "add (c++ function)": [[0, "_CPPv43addRK5arrayRK5array14StreamOrDevice"]], "addmm (c++ function)": [[0, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice"]], "all (c++ function)": [[0, "_CPPv43allRK5array14StreamOrDevice"], [0, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43allRK5arrayb14StreamOrDevice"], [0, "_CPPv43allRK5arrayib14StreamOrDevice"]], "allclose (c++ function)": [[0, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice"]], "any (c++ function)": [[0, "_CPPv43anyRK5array14StreamOrDevice"], [0, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43anyRK5arrayb14StreamOrDevice"], [0, "_CPPv43anyRK5arrayib14StreamOrDevice"]], "arange (c++ function)": [[0, "_CPPv46aranged14StreamOrDevice"], [0, "_CPPv46aranged5Dtype14StreamOrDevice"], [0, "_CPPv46arangedd14StreamOrDevice"], [0, "_CPPv46arangedd5Dtype14StreamOrDevice"], [0, "_CPPv46arangeddd14StreamOrDevice"], [0, "_CPPv46arangeddd5Dtype14StreamOrDevice"], [0, "_CPPv46arangei14StreamOrDevice"], [0, "_CPPv46arangeii14StreamOrDevice"], [0, "_CPPv46arangeiii14StreamOrDevice"]], "arccos (c++ function)": [[0, "_CPPv46arccosRK5array14StreamOrDevice"]], "arccosh (c++ function)": [[0, "_CPPv47arccoshRK5array14StreamOrDevice"]], "arcsin (c++ function)": [[0, "_CPPv46arcsinRK5array14StreamOrDevice"]], "arcsinh (c++ function)": [[0, "_CPPv47arcsinhRK5array14StreamOrDevice"]], "arctan (c++ function)": [[0, "_CPPv46arctanRK5array14StreamOrDevice"]], "arctan2 (c++ function)": [[0, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice"]], "arctanh (c++ function)": [[0, "_CPPv47arctanhRK5array14StreamOrDevice"]], "argmax (c++ function)": [[0, "_CPPv46argmaxRK5array14StreamOrDevice"], [0, "_CPPv46argmaxRK5arrayb14StreamOrDevice"], [0, "_CPPv46argmaxRK5arrayib14StreamOrDevice"]], "argmin (c++ function)": [[0, "_CPPv46argminRK5array14StreamOrDevice"], [0, "_CPPv46argminRK5arrayb14StreamOrDevice"], [0, "_CPPv46argminRK5arrayib14StreamOrDevice"]], "argpartition (c++ function)": [[0, "_CPPv412argpartitionRK5arrayi14StreamOrDevice"], [0, "_CPPv412argpartitionRK5arrayii14StreamOrDevice"]], "argsort (c++ function)": [[0, "_CPPv47argsortRK5array14StreamOrDevice"], [0, "_CPPv47argsortRK5arrayi14StreamOrDevice"]], "array_equal (c++ function)": [[0, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice"], [0, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice"]], "as_strided (c++ function)": [[0, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice"]], "astype (c++ function)": [[0, "_CPPv46astype5array5Dtype14StreamOrDevice"]], "atleast_1d (c++ function)": [[0, "_CPPv410atleast_1dRK5array14StreamOrDevice"], [0, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice"]], "atleast_2d (c++ function)": [[0, "_CPPv410atleast_2dRK5array14StreamOrDevice"], [0, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice"]], "atleast_3d (c++ function)": [[0, "_CPPv410atleast_3dRK5array14StreamOrDevice"], [0, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice"]], "bitwise_and (c++ function)": [[0, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice"]], "bitwise_or (c++ function)": [[0, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice"]], "bitwise_xor (c++ function)": [[0, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice"]], "block_masked_mm (c++ function)": [[0, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice"]], "broadcast_arrays (c++ function)": [[0, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice"]], "broadcast_to (c++ function)": [[0, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "ceil (c++ function)": [[0, "_CPPv44ceilRK5array14StreamOrDevice"]], "clip (c++ function)": [[0, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice"]], "concatenate (c++ function)": [[0, "_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice"], [0, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice"]], "conjugate (c++ function)": [[0, "_CPPv49conjugateRK5array14StreamOrDevice"]], "conv1d (c++ function)": [[0, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice"]], "conv2d (c++ function)": [[0, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"]], "conv3d (c++ function)": [[0, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"]], "conv_general (c++ function)": [[0, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"], [0, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"]], "copy (c++ function)": [[0, "_CPPv44copy5array14StreamOrDevice"]], "cos (c++ function)": [[0, "_CPPv43cosRK5array14StreamOrDevice"]], "cosh (c++ function)": [[0, "_CPPv44coshRK5array14StreamOrDevice"]], "cummax (c++ function)": [[0, "_CPPv46cummaxRK5arrayibb14StreamOrDevice"]], "cummin (c++ function)": [[0, "_CPPv46cumminRK5arrayibb14StreamOrDevice"]], "cumprod (c++ function)": [[0, "_CPPv47cumprodRK5arrayibb14StreamOrDevice"]], "cumsum (c++ function)": [[0, "_CPPv46cumsumRK5arrayibb14StreamOrDevice"]], "degrees (c++ function)": [[0, "_CPPv47degreesRK5array14StreamOrDevice"]], "depends (c++ function)": [[0, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE"]], "dequantize (c++ function)": [[0, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice"]], "diag (c++ function)": [[0, "_CPPv44diagRK5arrayi14StreamOrDevice"]], "diagonal (c++ function)": [[0, "_CPPv48diagonalRK5arrayiii14StreamOrDevice"]], "divide (c++ function)": [[0, "_CPPv46divideRK5arrayRK5array14StreamOrDevice"]], "divmod (c++ function)": [[0, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice"]], "equal (c++ function)": [[0, "_CPPv45equalRK5arrayRK5array14StreamOrDevice"]], "erf (c++ function)": [[0, "_CPPv43erfRK5array14StreamOrDevice"]], "erfinv (c++ function)": [[0, "_CPPv46erfinvRK5array14StreamOrDevice"]], "exp (c++ function)": [[0, "_CPPv43expRK5array14StreamOrDevice"]], "expand_dims (c++ function)": [[0, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice"]], "expm1 (c++ function)": [[0, "_CPPv45expm1RK5array14StreamOrDevice"]], "eye (c++ function)": [[0, "_CPPv43eyei14StreamOrDevice"], [0, "_CPPv43eyei5Dtype14StreamOrDevice"], [0, "_CPPv43eyeii14StreamOrDevice"], [0, "_CPPv43eyeiii14StreamOrDevice"], [0, "_CPPv43eyeiii5Dtype14StreamOrDevice"]], "flatten (c++ function)": [[0, "_CPPv47flattenRK5array14StreamOrDevice"], [0, "_CPPv47flattenRK5arrayii14StreamOrDevice"]], "floor (c++ function)": [[0, "_CPPv45floorRK5array14StreamOrDevice"]], "floor_divide (c++ function)": [[0, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice"]], "full (c++ function)": [[0, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice"], [0, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice"], [0, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice"], [0, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice"]], "gather (c++ function)": [[0, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"]], "gather_mm (c++ function)": [[0, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice"]], "gather_qmm (c++ function)": [[0, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice"]], "greater (c++ function)": [[0, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice"]], "greater_equal (c++ function)": [[0, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice"]], "hadamard_transform (c++ function)": [[0, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice"]], "identity (c++ function)": [[0, "_CPPv48identityi14StreamOrDevice"], [0, "_CPPv48identityi5Dtype14StreamOrDevice"]], "inner (c++ function)": [[0, "_CPPv45innerRK5arrayRK5array14StreamOrDevice"]], "isclose (c++ function)": [[0, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice"]], "isinf (c++ function)": [[0, "_CPPv45isinfRK5array14StreamOrDevice"]], "isnan (c++ function)": [[0, "_CPPv45isnanRK5array14StreamOrDevice"]], "isneginf (c++ function)": [[0, "_CPPv48isneginfRK5array14StreamOrDevice"]], "isposinf (c++ function)": [[0, "_CPPv48isposinfRK5array14StreamOrDevice"]], "left_shift (c++ function)": [[0, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice"]], "less (c++ function)": [[0, "_CPPv44lessRK5arrayRK5array14StreamOrDevice"]], "less_equal (c++ function)": [[0, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice"]], "linspace (c++ function)": [[0, "_CPPv48linspaceddi5Dtype14StreamOrDevice"]], "log (c++ function)": [[0, "_CPPv43logRK5array14StreamOrDevice"]], "log10 (c++ function)": [[0, "_CPPv45log10RK5array14StreamOrDevice"]], "log1p (c++ function)": [[0, "_CPPv45log1pRK5array14StreamOrDevice"]], "log2 (c++ function)": [[0, "_CPPv44log2RK5array14StreamOrDevice"]], "logaddexp (c++ function)": [[0, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice"]], "logical_and (c++ function)": [[0, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice"]], "logical_not (c++ function)": [[0, "_CPPv411logical_notRK5array14StreamOrDevice"]], "logical_or (c++ function)": [[0, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice"]], "logsumexp (c++ function)": [[0, "_CPPv49logsumexpRK5array14StreamOrDevice"], [0, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv49logsumexpRK5arrayb14StreamOrDevice"], [0, "_CPPv49logsumexpRK5arrayib14StreamOrDevice"]], "matmul (c++ function)": [[0, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice"]], "max (c++ function)": [[0, "_CPPv43maxRK5array14StreamOrDevice"], [0, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43maxRK5arrayb14StreamOrDevice"], [0, "_CPPv43maxRK5arrayib14StreamOrDevice"]], "maximum (c++ function)": [[0, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice"]], "mean (c++ function)": [[0, "_CPPv44meanRK5array14StreamOrDevice"], [0, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv44meanRK5arrayb14StreamOrDevice"], [0, "_CPPv44meanRK5arrayib14StreamOrDevice"]], "meshgrid (c++ function)": [[0, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice"]], "min (c++ function)": [[0, "_CPPv43minRK5array14StreamOrDevice"], [0, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43minRK5arrayb14StreamOrDevice"], [0, "_CPPv43minRK5arrayib14StreamOrDevice"]], "minimum (c++ function)": [[0, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice"]], "moveaxis (c++ function)": [[0, "_CPPv48moveaxisRK5arrayii14StreamOrDevice"]], "multiply (c++ function)": [[0, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice"]], "nan_to_num (c++ function)": [[0, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice"]], "negative (c++ function)": [[0, "_CPPv48negativeRK5array14StreamOrDevice"]], "not_equal (c++ function)": [[0, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice"]], "number_of_elements (c++ function)": [[0, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice"]], "ones (c++ function)": [[0, "_CPPv44onesRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice"]], "ones_like (c++ function)": [[0, "_CPPv49ones_likeRK5array14StreamOrDevice"]], "operator!= (c++ function)": [[0, "_CPPv4I0Ene5array1TRK5array"], [0, "_CPPv4I0Ene5arrayRK5array1T"], [0, "_CPPv4neRK5arrayRK5array"]], "operator% (c++ function)": [[0, "_CPPv4I0Erm5array1TRK5array"], [0, "_CPPv4I0Erm5arrayRK5array1T"], [0, "_CPPv4rmRK5arrayRK5array"]], "operator& (c++ function)": [[0, "_CPPv4anRK5arrayRK5array"]], "operator&& (c++ function)": [[0, "_CPPv4aaRK5arrayRK5array"]], "operator* (c++ function)": [[0, "_CPPv4I0Eml5array1TRK5array"], [0, "_CPPv4I0Eml5arrayRK5array1T"], [0, "_CPPv4mlRK5arrayRK5array"]], "operator+ (c++ function)": [[0, "_CPPv4I0Epl5array1TRK5array"], [0, "_CPPv4I0Epl5arrayRK5array1T"], [0, "_CPPv4plRK5arrayRK5array"]], "operator- (c++ function)": [[0, "_CPPv4I0Emi5array1TRK5array"], [0, "_CPPv4I0Emi5arrayRK5array1T"], [0, "_CPPv4miRK5array"], [0, "_CPPv4miRK5arrayRK5array"]], "operator/ (c++ function)": [[0, "_CPPv4dvRK5arrayRK5array"], [0, "_CPPv4dvRK5arrayd"], [0, "_CPPv4dvdRK5array"]], "operator< (c++ function)": [[0, "_CPPv4I0Elt5array1TRK5array"], [0, "_CPPv4I0Elt5arrayRK5array1T"], [0, "_CPPv4ltRK5arrayRK5array"]], "operator<< (c++ function)": [[0, "_CPPv4lsRK5arrayRK5array"]], "operator<= (c++ function)": [[0, "_CPPv4I0Ele5array1TRK5array"], [0, "_CPPv4I0Ele5arrayRK5array1T"], [0, "_CPPv4leRK5arrayRK5array"]], "operator== (c++ function)": [[0, "_CPPv4I0Eeq5array1TRK5array"], [0, "_CPPv4I0Eeq5arrayRK5array1T"], [0, "_CPPv4eqRK5arrayRK5array"]], "operator> (c++ function)": [[0, "_CPPv4I0Egt5array1TRK5array"], [0, "_CPPv4I0Egt5arrayRK5array1T"], [0, "_CPPv4gtRK5arrayRK5array"]], "operator>= (c++ function)": [[0, "_CPPv4I0Ege5array1TRK5array"], [0, "_CPPv4I0Ege5arrayRK5array1T"], [0, "_CPPv4geRK5arrayRK5array"]], "operator>> (c++ function)": [[0, "_CPPv4rsRK5arrayRK5array"]], "operator^ (c++ function)": [[0, "_CPPv4eoRK5arrayRK5array"]], "operator| (c++ function)": [[0, "_CPPv4orRK5arrayRK5array"]], "operator|| (c++ function)": [[0, "_CPPv4ooRK5arrayRK5array"]], "outer (c++ function)": [[0, "_CPPv45outerRK5arrayRK5array14StreamOrDevice"]], "pad (c++ function)": [[0, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5array14StreamOrDevice"], [0, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5array14StreamOrDevice"], [0, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5array14StreamOrDevice"], [0, "_CPPv43padRK5arrayiRK5array14StreamOrDevice"]], "partition (c++ function)": [[0, "_CPPv49partitionRK5arrayi14StreamOrDevice"], [0, "_CPPv49partitionRK5arrayii14StreamOrDevice"]], "power (c++ function)": [[0, "_CPPv45powerRK5arrayRK5array14StreamOrDevice"]], "prod (c++ function)": [[0, "_CPPv44prodRK5array14StreamOrDevice"], [0, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv44prodRK5arrayb14StreamOrDevice"], [0, "_CPPv44prodRK5arrayib14StreamOrDevice"]], "quantize (c++ function)": [[0, "_CPPv48quantizeRK5arrayii14StreamOrDevice"]], "quantized_matmul (c++ function)": [[0, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice"]], "radians (c++ function)": [[0, "_CPPv47radiansRK5array14StreamOrDevice"]], "reciprocal (c++ function)": [[0, "_CPPv410reciprocalRK5array14StreamOrDevice"]], "remainder (c++ function)": [[0, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice"]], "repeat (c++ function)": [[0, "_CPPv46repeatRK5arrayi14StreamOrDevice"], [0, "_CPPv46repeatRK5arrayii14StreamOrDevice"]], "reshape (c++ function)": [[0, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice"]], "right_shift (c++ function)": [[0, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice"]], "round (c++ function)": [[0, "_CPPv45roundRK5array14StreamOrDevice"], [0, "_CPPv45roundRK5arrayi14StreamOrDevice"]], "rsqrt (c++ function)": [[0, "_CPPv45rsqrtRK5array14StreamOrDevice"]], "scatter (c++ function)": [[0, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "scatter_add (c++ function)": [[0, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "scatter_max (c++ function)": [[0, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "scatter_min (c++ function)": [[0, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "scatter_prod (c++ function)": [[0, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "sigmoid (c++ function)": [[0, "_CPPv47sigmoidRK5array14StreamOrDevice"]], "sign (c++ function)": [[0, "_CPPv44signRK5array14StreamOrDevice"]], "sin (c++ function)": [[0, "_CPPv43sinRK5array14StreamOrDevice"]], "sinh (c++ function)": [[0, "_CPPv44sinhRK5array14StreamOrDevice"]], "slice (c++ function)": [[0, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"]], "slice_update (c++ function)": [[0, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice"]], "softmax (c++ function)": [[0, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv47softmaxRK5arrayb14StreamOrDevice"], [0, "_CPPv47softmaxRK5arrayib14StreamOrDevice"]], "sort (c++ function)": [[0, "_CPPv44sortRK5array14StreamOrDevice"], [0, "_CPPv44sortRK5arrayi14StreamOrDevice"]], "split (c++ function)": [[0, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice"], [0, "_CPPv45splitRK5arrayi14StreamOrDevice"], [0, "_CPPv45splitRK5arrayii14StreamOrDevice"]], "sqrt (c++ function)": [[0, "_CPPv44sqrtRK5array14StreamOrDevice"]], "square (c++ function)": [[0, "_CPPv46squareRK5array14StreamOrDevice"]], "squeeze (c++ function)": [[0, "_CPPv47squeezeRK5array14StreamOrDevice"], [0, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv47squeezeRK5arrayi14StreamOrDevice"]], "stack (c++ function)": [[0, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice"], [0, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice"]], "std (c++ function)": [[0, "_CPPv4StRK5array14StreamOrDevice"], [0, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"], [0, "_CPPv4StRK5arraybi14StreamOrDevice"], [0, "_CPPv4StRK5arrayibi14StreamOrDevice"]], "stop_gradient (c++ function)": [[0, "_CPPv413stop_gradientRK5array14StreamOrDevice"]], "subtract (c++ function)": [[0, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice"]], "sum (c++ function)": [[0, "_CPPv43sumRK5array14StreamOrDevice"], [0, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43sumRK5arrayb14StreamOrDevice"], [0, "_CPPv43sumRK5arrayib14StreamOrDevice"]], "swapaxes (c++ function)": [[0, "_CPPv48swapaxesRK5arrayii14StreamOrDevice"]], "take (c++ function)": [[0, "_CPPv44takeRK5arrayRK5array14StreamOrDevice"], [0, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice"]], "take_along_axis (c++ function)": [[0, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice"]], "tan (c++ function)": [[0, "_CPPv43tanRK5array14StreamOrDevice"]], "tanh (c++ function)": [[0, "_CPPv44tanhRK5array14StreamOrDevice"]], "tensordot (c++ function)": [[0, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice"], [0, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"]], "tile (c++ function)": [[0, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice"]], "topk (c++ function)": [[0, "_CPPv44topkRK5arrayi14StreamOrDevice"], [0, "_CPPv44topkRK5arrayii14StreamOrDevice"]], "trace (c++ function)": [[0, "_CPPv45traceRK5array14StreamOrDevice"], [0, "_CPPv45traceRK5arrayiii14StreamOrDevice"], [0, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice"]], "transpose (c++ function)": [[0, "_CPPv49transposeRK5array14StreamOrDevice"], [0, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice"], [0, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice"]], "tri (c++ function)": [[0, "_CPPv43trii5Dtype14StreamOrDevice"], [0, "_CPPv43triiii5Dtype14StreamOrDevice"]], "tril (c++ function)": [[0, "_CPPv44tril5arrayi14StreamOrDevice"]], "triu (c++ function)": [[0, "_CPPv44triu5arrayi14StreamOrDevice"]], "var (c++ function)": [[0, "_CPPv43varRK5array14StreamOrDevice"], [0, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"], [0, "_CPPv43varRK5arraybi14StreamOrDevice"], [0, "_CPPv43varRK5arrayibi14StreamOrDevice"]], "view (c++ function)": [[0, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice"]], "where (c++ function)": [[0, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice"]], "zeros (c++ function)": [[0, "_CPPv45zerosRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice"]], "zeros_like (c++ function)": [[0, "_CPPv410zeros_likeRK5array14StreamOrDevice"]], "device (class in mlx.core)": [[8, "mlx.core.Device"]], "__init__() (device method)": [[8, "mlx.core.Device.__init__"]], "dtype (class in mlx.core)": [[9, "mlx.core.Dtype"]], "__init__() (dtype method)": [[9, "mlx.core.Dtype.__init__"]], "dtypecategory (class in mlx.core)": [[10, "mlx.core.DtypeCategory"]], "__init__() (dtypecategory method)": [[10, "mlx.core.DtypeCategory.__init__"]], "abs() (in module mlx.core)": [[11, "mlx.core.abs"]], "add() (in module mlx.core)": [[12, "mlx.core.add"]], "addmm() (in module mlx.core)": [[13, "mlx.core.addmm"]], "all() (in module mlx.core)": [[14, "mlx.core.all"]], "allclose() (in module mlx.core)": [[15, "mlx.core.allclose"]], "any() (in module mlx.core)": [[16, "mlx.core.any"]], "arange() (in module mlx.core)": [[17, "mlx.core.arange"]], "arccos() (in module mlx.core)": [[18, "mlx.core.arccos"]], "arccosh() (in module mlx.core)": [[19, "mlx.core.arccosh"]], "arcsin() (in module mlx.core)": [[20, "mlx.core.arcsin"]], "arcsinh() (in module mlx.core)": [[21, "mlx.core.arcsinh"]], "arctan() (in module mlx.core)": [[22, "mlx.core.arctan"]], "arctan2() (in module mlx.core)": [[23, "mlx.core.arctan2"]], "arctanh() (in module mlx.core)": [[24, "mlx.core.arctanh"]], "argmax() (in module mlx.core)": [[25, "mlx.core.argmax"]], "argmin() (in module mlx.core)": [[26, "mlx.core.argmin"]], "argpartition() (in module mlx.core)": [[27, "mlx.core.argpartition"]], "argsort() (in module mlx.core)": [[28, "mlx.core.argsort"]], "__init__() (array method)": [[29, "mlx.core.array.__init__"]], "array (class in mlx.core)": [[29, "mlx.core.array"]], "t (array property)": [[30, "mlx.core.array.T"]], "abs() (array method)": [[31, "mlx.core.array.abs"]], "all() (array method)": [[32, "mlx.core.array.all"]], "any() (array method)": [[33, "mlx.core.array.any"]], "argmax() (array method)": [[34, "mlx.core.array.argmax"]], "argmin() (array method)": [[35, "mlx.core.array.argmin"]], "astype() (array method)": [[36, "mlx.core.array.astype"]], "at (array property)": [[37, "mlx.core.array.at"]], "conj() (array method)": [[38, "mlx.core.array.conj"]], "cos() (array method)": [[39, "mlx.core.array.cos"]], "cummax() (array method)": [[40, "mlx.core.array.cummax"]], "cummin() (array method)": [[41, "mlx.core.array.cummin"]], "cumprod() (array method)": [[42, "mlx.core.array.cumprod"]], "cumsum() (array method)": [[43, "mlx.core.array.cumsum"]], "diag() (array method)": [[44, "mlx.core.array.diag"]], "diagonal() (array method)": [[45, "mlx.core.array.diagonal"]], "dtype (array property)": [[46, "mlx.core.array.dtype"]], "exp() (array method)": [[47, "mlx.core.array.exp"]], "flatten() (array method)": [[48, "mlx.core.array.flatten"]], "item() (array method)": [[49, "mlx.core.array.item"]], "itemsize (array property)": [[50, "mlx.core.array.itemsize"]], "log() (array method)": [[51, "mlx.core.array.log"]], "log10() (array method)": [[52, "mlx.core.array.log10"]], "log1p() (array method)": [[53, "mlx.core.array.log1p"]], "log2() (array method)": [[54, "mlx.core.array.log2"]], "logsumexp() (array method)": [[55, "mlx.core.array.logsumexp"]], "max() (array method)": [[56, "mlx.core.array.max"]], "mean() (array method)": [[57, "mlx.core.array.mean"]], "min() (array method)": [[58, "mlx.core.array.min"]], "moveaxis() (array method)": [[59, "mlx.core.array.moveaxis"]], "nbytes (array property)": [[60, "mlx.core.array.nbytes"]], "ndim (array property)": [[61, "mlx.core.array.ndim"]], "prod() (array method)": [[62, "mlx.core.array.prod"]], "reciprocal() (array method)": [[63, "mlx.core.array.reciprocal"]], "reshape() (array method)": [[64, "mlx.core.array.reshape"]], "round() (array method)": [[65, "mlx.core.array.round"]], "rsqrt() (array method)": [[66, "mlx.core.array.rsqrt"]], "shape (array property)": [[67, "mlx.core.array.shape"]], "sin() (array method)": [[68, "mlx.core.array.sin"]], "size (array property)": [[69, "mlx.core.array.size"]], "split() (array method)": [[70, "mlx.core.array.split"]], "sqrt() (array method)": [[71, "mlx.core.array.sqrt"]], "square() (array method)": [[72, "mlx.core.array.square"]], "squeeze() (array method)": [[73, "mlx.core.array.squeeze"]], "sum() (array method)": [[74, "mlx.core.array.sum"]], "swapaxes() (array method)": [[75, "mlx.core.array.swapaxes"]], "tolist() (array method)": [[76, "mlx.core.array.tolist"]], "transpose() (array method)": [[77, "mlx.core.array.transpose"]], "var() (array method)": [[78, "mlx.core.array.var"]], "view() (array method)": [[79, "mlx.core.array.view"]], "array_equal() (in module mlx.core)": [[80, "mlx.core.array_equal"]], "as_strided() (in module mlx.core)": [[81, "mlx.core.as_strided"]], "atleast_1d() (in module mlx.core)": [[82, "mlx.core.atleast_1d"]], "atleast_2d() (in module mlx.core)": [[83, "mlx.core.atleast_2d"]], "atleast_3d() (in module mlx.core)": [[84, "mlx.core.atleast_3d"]], "bitwise_and() (in module mlx.core)": [[85, "mlx.core.bitwise_and"]], "bitwise_or() (in module mlx.core)": [[86, "mlx.core.bitwise_or"]], "bitwise_xor() (in module mlx.core)": [[87, "mlx.core.bitwise_xor"]], "block_masked_mm() (in module mlx.core)": [[88, "mlx.core.block_masked_mm"]], "broadcast_to() (in module mlx.core)": [[89, "mlx.core.broadcast_to"]], "ceil() (in module mlx.core)": [[90, "mlx.core.ceil"]], "clip() (in module mlx.core)": [[91, "mlx.core.clip"]], "compile() (in module mlx.core)": [[92, "mlx.core.compile"]], "concatenate() (in module mlx.core)": [[93, "mlx.core.concatenate"]], "conj() (in module mlx.core)": [[94, "mlx.core.conj"]], "conjugate() (in module mlx.core)": [[95, "mlx.core.conjugate"]], "conv1d() (in module mlx.core)": [[96, "mlx.core.conv1d"]], "conv2d() (in module mlx.core)": [[97, "mlx.core.conv2d"]], "conv_general() (in module mlx.core)": [[98, "mlx.core.conv_general"]], "convolve() (in module mlx.core)": [[99, "mlx.core.convolve"]], "cos() (in module mlx.core)": [[100, "mlx.core.cos"]], "cosh() (in module mlx.core)": [[101, "mlx.core.cosh"]], "cummax() (in module mlx.core)": [[102, "mlx.core.cummax"]], "cummin() (in module mlx.core)": [[103, "mlx.core.cummin"]], "cumprod() (in module mlx.core)": [[104, "mlx.core.cumprod"]], "cumsum() (in module mlx.core)": [[105, "mlx.core.cumsum"]], "__init__() (custom_function method)": [[106, "mlx.core.custom_function.__init__"]], "custom_function (class in mlx.core)": [[106, "mlx.core.custom_function"]], "default_device() (in module mlx.core)": [[107, "mlx.core.default_device"]], "default_stream() (in module mlx.core)": [[108, "mlx.core.default_stream"]], "degrees() (in module mlx.core)": [[109, "mlx.core.degrees"]], "dequantize() (in module mlx.core)": [[110, "mlx.core.dequantize"]], "diag() (in module mlx.core)": [[111, "mlx.core.diag"]], "diagonal() (in module mlx.core)": [[112, "mlx.core.diagonal"]], "disable_compile() (in module mlx.core)": [[113, "mlx.core.disable_compile"]], "group (class in mlx.core.distributed)": [[114, "mlx.core.distributed.Group"]], "__init__() (group method)": [[114, "mlx.core.distributed.Group.__init__"]], "all_gather() (in module mlx.core.distributed)": [[115, "mlx.core.distributed.all_gather"]], "all_sum() (in module mlx.core.distributed)": [[116, "mlx.core.distributed.all_sum"]], "init() (in module mlx.core.distributed)": [[117, "mlx.core.distributed.init"]], "is_available() (in module mlx.core.distributed)": [[118, "mlx.core.distributed.is_available"]], "divide() (in module mlx.core)": [[119, "mlx.core.divide"]], "divmod() (in module mlx.core)": [[120, "mlx.core.divmod"]], "einsum() (in module mlx.core)": [[121, "mlx.core.einsum"]], "einsum_path() (in module mlx.core)": [[122, "mlx.core.einsum_path"]], "enable_compile() (in module mlx.core)": [[123, "mlx.core.enable_compile"]], "equal() (in module mlx.core)": [[124, "mlx.core.equal"]], "erf() (in module mlx.core)": [[125, "mlx.core.erf"]], "erfinv() (in module mlx.core)": [[126, "mlx.core.erfinv"]], "eval() (in module mlx.core)": [[127, "mlx.core.eval"]], "exp() (in module mlx.core)": [[128, "mlx.core.exp"]], "expand_dims() (in module mlx.core)": [[129, "mlx.core.expand_dims"]], "expm1() (in module mlx.core)": [[130, "mlx.core.expm1"]], "eye() (in module mlx.core)": [[131, "mlx.core.eye"]], "layer_norm() (in module mlx.core.fast)": [[132, "mlx.core.fast.layer_norm"]], "rms_norm() (in module mlx.core.fast)": [[133, "mlx.core.fast.rms_norm"]], "rope() (in module mlx.core.fast)": [[134, "mlx.core.fast.rope"]], "scaled_dot_product_attention() (in module mlx.core.fast)": [[135, "mlx.core.fast.scaled_dot_product_attention"]], "fft() (in module mlx.core.fft)": [[136, "mlx.core.fft.fft"]], "fft2() (in module mlx.core.fft)": [[137, "mlx.core.fft.fft2"]], "fftn() (in module mlx.core.fft)": [[138, "mlx.core.fft.fftn"]], "ifft() (in module mlx.core.fft)": [[139, "mlx.core.fft.ifft"]], "ifft2() (in module mlx.core.fft)": [[140, "mlx.core.fft.ifft2"]], "ifftn() (in module mlx.core.fft)": [[141, "mlx.core.fft.ifftn"]], "irfft() (in module mlx.core.fft)": [[142, "mlx.core.fft.irfft"]], "irfft2() (in module mlx.core.fft)": [[143, "mlx.core.fft.irfft2"]], "irfftn() (in module mlx.core.fft)": [[144, "mlx.core.fft.irfftn"]], "rfft() (in module mlx.core.fft)": [[145, "mlx.core.fft.rfft"]], "rfft2() (in module mlx.core.fft)": [[146, "mlx.core.fft.rfft2"]], "rfftn() (in module mlx.core.fft)": [[147, "mlx.core.fft.rfftn"]], "flatten() (in module mlx.core)": [[148, "mlx.core.flatten"]], "floor() (in module mlx.core)": [[149, "mlx.core.floor"]], "floor_divide() (in module mlx.core)": [[150, "mlx.core.floor_divide"]], "full() (in module mlx.core)": [[151, "mlx.core.full"]], "gather_mm() (in module mlx.core)": [[152, "mlx.core.gather_mm"]], "gather_qmm() (in module mlx.core)": [[153, "mlx.core.gather_qmm"]], "grad() (in module mlx.core)": [[154, "mlx.core.grad"]], "greater() (in module mlx.core)": [[155, "mlx.core.greater"]], "greater_equal() (in module mlx.core)": [[156, "mlx.core.greater_equal"]], "hadamard_transform() (in module mlx.core)": [[157, "mlx.core.hadamard_transform"]], "identity() (in module mlx.core)": [[158, "mlx.core.identity"]], "inner() (in module mlx.core)": [[159, "mlx.core.inner"]], "isclose() (in module mlx.core)": [[160, "mlx.core.isclose"]], "isinf() (in module mlx.core)": [[161, "mlx.core.isinf"]], "isnan() (in module mlx.core)": [[162, "mlx.core.isnan"]], "isneginf() (in module mlx.core)": [[163, "mlx.core.isneginf"]], "isposinf() (in module mlx.core)": [[164, "mlx.core.isposinf"]], "issubdtype() (in module mlx.core)": [[165, "mlx.core.issubdtype"]], "jvp() (in module mlx.core)": [[166, "mlx.core.jvp"]], "left_shift() (in module mlx.core)": [[167, "mlx.core.left_shift"]], "less() (in module mlx.core)": [[168, "mlx.core.less"]], "less_equal() (in module mlx.core)": [[169, "mlx.core.less_equal"]], "cholesky() (in module mlx.core.linalg)": [[170, "mlx.core.linalg.cholesky"]], "inv() (in module mlx.core.linalg)": [[171, "mlx.core.linalg.inv"]], "norm() (in module mlx.core.linalg)": [[172, "mlx.core.linalg.norm"]], "qr() (in module mlx.core.linalg)": [[173, "mlx.core.linalg.qr"]], "svd() (in module mlx.core.linalg)": [[174, "mlx.core.linalg.svd"]], "linspace() (in module mlx.core)": [[175, "mlx.core.linspace"]], "load() (in module mlx.core)": [[176, "mlx.core.load"]], "log() (in module mlx.core)": [[177, "mlx.core.log"]], "log10() (in module mlx.core)": [[178, "mlx.core.log10"]], "log1p() (in module mlx.core)": [[179, "mlx.core.log1p"]], "log2() (in module mlx.core)": [[180, "mlx.core.log2"]], "logaddexp() (in module mlx.core)": [[181, "mlx.core.logaddexp"]], "logical_and() (in module mlx.core)": [[182, "mlx.core.logical_and"]], "logical_not() (in module mlx.core)": [[183, "mlx.core.logical_not"]], "logical_or() (in module mlx.core)": [[184, "mlx.core.logical_or"]], "logsumexp() (in module mlx.core)": [[185, "mlx.core.logsumexp"]], "matmul() (in module mlx.core)": [[186, "mlx.core.matmul"]], "max() (in module mlx.core)": [[187, "mlx.core.max"]], "maximum() (in module mlx.core)": [[188, "mlx.core.maximum"]], "mean() (in module mlx.core)": [[189, "mlx.core.mean"]], "meshgrid() (in module mlx.core)": [[190, "mlx.core.meshgrid"]], "clear_cache() (in module mlx.core.metal)": [[191, "mlx.core.metal.clear_cache"]], "device_info() (in module mlx.core.metal)": [[192, "mlx.core.metal.device_info"]], "get_active_memory() (in module mlx.core.metal)": [[193, "mlx.core.metal.get_active_memory"]], "get_cache_memory() (in module mlx.core.metal)": [[194, "mlx.core.metal.get_cache_memory"]], "get_peak_memory() (in module mlx.core.metal)": [[195, "mlx.core.metal.get_peak_memory"]], "is_available() (in module mlx.core.metal)": [[196, "mlx.core.metal.is_available"]], "reset_peak_memory() (in module mlx.core.metal)": [[197, "mlx.core.metal.reset_peak_memory"]], "set_cache_limit() (in module mlx.core.metal)": [[198, "mlx.core.metal.set_cache_limit"]], "set_memory_limit() (in module mlx.core.metal)": [[199, "mlx.core.metal.set_memory_limit"]], "start_capture() (in module mlx.core.metal)": [[200, "mlx.core.metal.start_capture"]], "stop_capture() (in module mlx.core.metal)": [[201, "mlx.core.metal.stop_capture"]], "min() (in module mlx.core)": [[202, "mlx.core.min"]], "minimum() (in module mlx.core)": [[203, "mlx.core.minimum"]], "moveaxis() (in module mlx.core)": [[204, "mlx.core.moveaxis"]], "multiply() (in module mlx.core)": [[205, "mlx.core.multiply"]], "nan_to_num() (in module mlx.core)": [[206, "mlx.core.nan_to_num"]], "negative() (in module mlx.core)": [[207, "mlx.core.negative"]], "new_stream() (in module mlx.core)": [[208, "mlx.core.new_stream"]], "not_equal() (in module mlx.core)": [[209, "mlx.core.not_equal"]], "ones() (in module mlx.core)": [[210, "mlx.core.ones"]], "ones_like() (in module mlx.core)": [[211, "mlx.core.ones_like"]], "outer() (in module mlx.core)": [[212, "mlx.core.outer"]], "pad() (in module mlx.core)": [[213, "mlx.core.pad"]], "partition() (in module mlx.core)": [[214, "mlx.core.partition"]], "power() (in module mlx.core)": [[215, "mlx.core.power"]], "prod() (in module mlx.core)": [[216, "mlx.core.prod"]], "quantize() (in module mlx.core)": [[217, "mlx.core.quantize"]], "quantized_matmul() (in module mlx.core)": [[218, "mlx.core.quantized_matmul"]], "radians() (in module mlx.core)": [[219, "mlx.core.radians"]], "bernoulli() (in module mlx.core.random)": [[220, "mlx.core.random.bernoulli"]], "categorical() (in module mlx.core.random)": [[221, "mlx.core.random.categorical"]], "gumbel() (in module mlx.core.random)": [[222, "mlx.core.random.gumbel"]], "key() (in module mlx.core.random)": [[223, "mlx.core.random.key"]], "laplace() (in module mlx.core.random)": [[224, "mlx.core.random.laplace"]], "multivariate_normal() (in module mlx.core.random)": [[225, "mlx.core.random.multivariate_normal"]], "normal() (in module mlx.core.random)": [[226, "mlx.core.random.normal"]], "randint() (in module mlx.core.random)": [[227, "mlx.core.random.randint"]], "seed() (in module mlx.core.random)": [[228, "mlx.core.random.seed"]], "split() (in module mlx.core.random)": [[229, "mlx.core.random.split"]], "truncated_normal() (in module mlx.core.random)": [[230, "mlx.core.random.truncated_normal"]], "uniform() (in module mlx.core.random)": [[231, "mlx.core.random.uniform"]], "reciprocal() (in module mlx.core)": [[232, "mlx.core.reciprocal"]], "remainder() (in module mlx.core)": [[233, "mlx.core.remainder"]], "repeat() (in module mlx.core)": [[234, "mlx.core.repeat"]], "reshape() (in module mlx.core)": [[235, "mlx.core.reshape"]], "right_shift() (in module mlx.core)": [[236, "mlx.core.right_shift"]], "round() (in module mlx.core)": [[237, "mlx.core.round"]], "rsqrt() (in module mlx.core)": [[238, "mlx.core.rsqrt"]], "save() (in module mlx.core)": [[239, "mlx.core.save"]], "save_gguf() (in module mlx.core)": [[240, "mlx.core.save_gguf"]], "save_safetensors() (in module mlx.core)": [[241, "mlx.core.save_safetensors"]], "savez() (in module mlx.core)": [[242, "mlx.core.savez"]], "savez_compressed() (in module mlx.core)": [[243, "mlx.core.savez_compressed"]], "set_default_device() (in module mlx.core)": [[244, "mlx.core.set_default_device"]], "set_default_stream() (in module mlx.core)": [[245, "mlx.core.set_default_stream"]], "sigmoid() (in module mlx.core)": [[246, "mlx.core.sigmoid"]], "sign() (in module mlx.core)": [[247, "mlx.core.sign"]], "sin() (in module mlx.core)": [[248, "mlx.core.sin"]], "sinh() (in module mlx.core)": [[249, "mlx.core.sinh"]], "softmax() (in module mlx.core)": [[250, "mlx.core.softmax"]], "sort() (in module mlx.core)": [[251, "mlx.core.sort"]], "split() (in module mlx.core)": [[252, "mlx.core.split"]], "sqrt() (in module mlx.core)": [[253, "mlx.core.sqrt"]], "square() (in module mlx.core)": [[254, "mlx.core.square"]], "squeeze() (in module mlx.core)": [[255, "mlx.core.squeeze"]], "stack() (in module mlx.core)": [[256, "mlx.core.stack"]], "std() (in module mlx.core)": [[257, "mlx.core.std"]], "stop_gradient() (in module mlx.core)": [[258, "mlx.core.stop_gradient"]], "stream() (in module mlx.core)": [[259, "mlx.core.stream"]], "subtract() (in module mlx.core)": [[260, "mlx.core.subtract"]], "sum() (in module mlx.core)": [[261, "mlx.core.sum"]], "swapaxes() (in module mlx.core)": [[262, "mlx.core.swapaxes"]], "synchronize() (in module mlx.core)": [[263, "mlx.core.synchronize"]], "take() (in module mlx.core)": [[264, "mlx.core.take"]], "take_along_axis() (in module mlx.core)": [[265, "mlx.core.take_along_axis"]], "tan() (in module mlx.core)": [[266, "mlx.core.tan"]], "tanh() (in module mlx.core)": [[267, "mlx.core.tanh"]], "tensordot() (in module mlx.core)": [[268, "mlx.core.tensordot"]], "tile() (in module mlx.core)": [[269, "mlx.core.tile"]], "topk() (in module mlx.core)": [[270, "mlx.core.topk"]], "trace() (in module mlx.core)": [[271, "mlx.core.trace"]], "transpose() (in module mlx.core)": [[272, "mlx.core.transpose"]], "tri() (in module mlx.core)": [[273, "mlx.core.tri"]], "tril() (in module mlx.core)": [[274, "mlx.core.tril"]], "triu() (in module mlx.core)": [[275, "mlx.core.triu"]], "value_and_grad() (in module mlx.core)": [[276, "mlx.core.value_and_grad"]], "var() (in module mlx.core)": [[277, "mlx.core.var"]], "view() (in module mlx.core)": [[278, "mlx.core.view"]], "vjp() (in module mlx.core)": [[279, "mlx.core.vjp"]], "vmap() (in module mlx.core)": [[280, "mlx.core.vmap"]], "where() (in module mlx.core)": [[281, "mlx.core.where"]], "zeros() (in module mlx.core)": [[282, "mlx.core.zeros"]], "zeros_like() (in module mlx.core)": [[283, "mlx.core.zeros_like"]], "quantize() (in module mlx.nn)": [[284, "mlx.nn.quantize"]], "value_and_grad() (in module mlx.nn)": [[285, "mlx.nn.value_and_grad"]], "clip_grad_norm() (in module mlx.optimizers)": [[286, "mlx.optimizers.clip_grad_norm"]], "tree_flatten() (in module mlx.utils)": [[287, "mlx.utils.tree_flatten"]], "tree_map() (in module mlx.utils)": [[288, "mlx.utils.tree_map"]], "tree_map_with_path() (in module mlx.utils)": [[289, "mlx.utils.tree_map_with_path"]], "tree_reduce() (in module mlx.utils)": [[290, "mlx.utils.tree_reduce"]], "tree_unflatten() (in module mlx.utils)": [[291, "mlx.utils.tree_unflatten"]], "stream (class in mlx.core)": [[292, "mlx.core.Stream"]], "__init__() (stream method)": [[292, "mlx.core.Stream.__init__"]], "alibi (class in mlx.nn)": [[302, "mlx.nn.ALiBi"]], "avgpool1d (class in mlx.nn)": [[303, "mlx.nn.AvgPool1d"]], "avgpool2d (class in mlx.nn)": [[304, "mlx.nn.AvgPool2d"]], "batchnorm (class in mlx.nn)": [[305, "mlx.nn.BatchNorm"]], "conv1d (class in mlx.nn)": [[306, "mlx.nn.Conv1d"]], "conv2d (class in mlx.nn)": [[307, "mlx.nn.Conv2d"]], "conv3d (class in mlx.nn)": [[308, "mlx.nn.Conv3d"]], "dropout (class in mlx.nn)": [[309, "mlx.nn.Dropout"]], "dropout2d (class in mlx.nn)": [[310, "mlx.nn.Dropout2d"]], "dropout3d (class in mlx.nn)": [[311, "mlx.nn.Dropout3d"]], "embedding (class in mlx.nn)": [[312, "mlx.nn.Embedding"]], "gelu (class in mlx.nn)": [[313, "mlx.nn.GELU"], [379, "mlx.nn.gelu"]], "glu (class in mlx.nn)": [[314, "mlx.nn.GLU"], [382, "mlx.nn.glu"]], "gru (class in mlx.nn)": [[315, "mlx.nn.GRU"]], "groupnorm (class in mlx.nn)": [[316, "mlx.nn.GroupNorm"]], "hardshrink (class in mlx.nn)": [[317, "mlx.nn.HardShrink"]], "hardtanh (class in mlx.nn)": [[318, "mlx.nn.HardTanh"]], "hardswish (class in mlx.nn)": [[319, "mlx.nn.Hardswish"], [385, "mlx.nn.hardswish"]], "instancenorm (class in mlx.nn)": [[320, "mlx.nn.InstanceNorm"]], "lstm (class in mlx.nn)": [[321, "mlx.nn.LSTM"]], "layernorm (class in mlx.nn)": [[322, "mlx.nn.LayerNorm"]], "leakyrelu (class in mlx.nn)": [[323, "mlx.nn.LeakyReLU"]], "linear (class in mlx.nn)": [[324, "mlx.nn.Linear"]], "maxpool1d (class in mlx.nn)": [[325, "mlx.nn.MaxPool1d"]], "maxpool2d (class in mlx.nn)": [[326, "mlx.nn.MaxPool2d"]], "mish (class in mlx.nn)": [[327, "mlx.nn.Mish"], [403, "mlx.nn.mish"]], "apply() (module method)": [[328, "mlx.nn.Module.apply"]], "apply_to_modules() (module method)": [[329, "mlx.nn.Module.apply_to_modules"]], "children() (module method)": [[330, "mlx.nn.Module.children"]], "eval() (module method)": [[331, "mlx.nn.Module.eval"]], "filter_and_map() (module method)": [[332, "mlx.nn.Module.filter_and_map"]], "freeze() (module method)": [[333, "mlx.nn.Module.freeze"]], "leaf_modules() (module method)": [[334, "mlx.nn.Module.leaf_modules"]], "load_weights() (module method)": [[335, "mlx.nn.Module.load_weights"]], "modules() (module method)": [[336, "mlx.nn.Module.modules"]], "named_modules() (module method)": [[337, "mlx.nn.Module.named_modules"]], "parameters() (module method)": [[338, "mlx.nn.Module.parameters"]], "save_weights() (module method)": [[339, "mlx.nn.Module.save_weights"]], "set_dtype() (module method)": [[340, "mlx.nn.Module.set_dtype"]], "state (module property)": [[341, "mlx.nn.Module.state"]], "train() (module method)": [[342, "mlx.nn.Module.train"]], "trainable_parameters() (module method)": [[343, "mlx.nn.Module.trainable_parameters"]], "training (module property)": [[344, "mlx.nn.Module.training"]], "unfreeze() (module method)": [[345, "mlx.nn.Module.unfreeze"]], "update() (module method)": [[346, "mlx.nn.Module.update"]], "update_modules() (module method)": [[347, "mlx.nn.Module.update_modules"]], "multiheadattention (class in mlx.nn)": [[348, "mlx.nn.MultiHeadAttention"]], "prelu (class in mlx.nn)": [[349, "mlx.nn.PReLU"], [404, "mlx.nn.prelu"]], "quantizedembedding (class in mlx.nn)": [[350, "mlx.nn.QuantizedEmbedding"]], "quantizedlinear (class in mlx.nn)": [[351, "mlx.nn.QuantizedLinear"]], "rmsnorm (class in mlx.nn)": [[352, "mlx.nn.RMSNorm"]], "rnn (class in mlx.nn)": [[353, "mlx.nn.RNN"]], "relu (class in mlx.nn)": [[354, "mlx.nn.ReLU"], [405, "mlx.nn.relu"]], "relu6 (class in mlx.nn)": [[355, "mlx.nn.ReLU6"], [406, "mlx.nn.relu6"]], "rope (class in mlx.nn)": [[356, "mlx.nn.RoPE"]], "selu (class in mlx.nn)": [[357, "mlx.nn.SELU"], [407, "mlx.nn.selu"]], "sequential (class in mlx.nn)": [[358, "mlx.nn.Sequential"]], "silu (class in mlx.nn)": [[359, "mlx.nn.SiLU"], [409, "mlx.nn.silu"]], "sinusoidalpositionalencoding (class in mlx.nn)": [[360, "mlx.nn.SinusoidalPositionalEncoding"]], "softmax (class in mlx.nn)": [[361, "mlx.nn.Softmax"], [410, "mlx.nn.softmax"]], "softmin (class in mlx.nn)": [[362, "mlx.nn.Softmin"], [411, "mlx.nn.softmin"]], "softplus (class in mlx.nn)": [[363, "mlx.nn.Softplus"], [412, "mlx.nn.softplus"]], "softshrink (class in mlx.nn)": [[364, "mlx.nn.Softshrink"], [413, "mlx.nn.softshrink"]], "softsign (class in mlx.nn)": [[365, "mlx.nn.Softsign"]], "step (class in mlx.nn)": [[366, "mlx.nn.Step"], [414, "mlx.nn.step"]], "tanh (class in mlx.nn)": [[367, "mlx.nn.Tanh"], [415, "mlx.nn.tanh"]], "transformer (class in mlx.nn)": [[368, "mlx.nn.Transformer"]], "upsample (class in mlx.nn)": [[369, "mlx.nn.Upsample"]], "constant() (in module mlx.nn.init)": [[370, "mlx.nn.init.constant"]], "glorot_normal() (in module mlx.nn.init)": [[371, "mlx.nn.init.glorot_normal"]], "glorot_uniform() (in module mlx.nn.init)": [[372, "mlx.nn.init.glorot_uniform"]], "he_normal() (in module mlx.nn.init)": [[373, "mlx.nn.init.he_normal"]], "he_uniform() (in module mlx.nn.init)": [[374, "mlx.nn.init.he_uniform"]], "identity() (in module mlx.nn.init)": [[375, "mlx.nn.init.identity"]], "normal() (in module mlx.nn.init)": [[376, "mlx.nn.init.normal"]], "uniform() (in module mlx.nn.init)": [[377, "mlx.nn.init.uniform"]], "elu (class in mlx.nn)": [[378, "mlx.nn.elu"]], "gelu_approx (class in mlx.nn)": [[380, "mlx.nn.gelu_approx"]], "gelu_fast_approx (class in mlx.nn)": [[381, "mlx.nn.gelu_fast_approx"]], "hard_shrink (class in mlx.nn)": [[383, "mlx.nn.hard_shrink"]], "hard_tanh (class in mlx.nn)": [[384, "mlx.nn.hard_tanh"]], "leaky_relu (class in mlx.nn)": [[386, "mlx.nn.leaky_relu"]], "log_sigmoid (class in mlx.nn)": [[387, "mlx.nn.log_sigmoid"]], "log_softmax (class in mlx.nn)": [[388, "mlx.nn.log_softmax"]], "binary_cross_entropy (class in mlx.nn.losses)": [[389, "mlx.nn.losses.binary_cross_entropy"]], "cosine_similarity_loss (class in mlx.nn.losses)": [[390, "mlx.nn.losses.cosine_similarity_loss"]], "cross_entropy (class in mlx.nn.losses)": [[391, "mlx.nn.losses.cross_entropy"]], "gaussian_nll_loss (class in mlx.nn.losses)": [[392, "mlx.nn.losses.gaussian_nll_loss"]], "hinge_loss (class in mlx.nn.losses)": [[393, "mlx.nn.losses.hinge_loss"]], "huber_loss (class in mlx.nn.losses)": [[394, "mlx.nn.losses.huber_loss"]], "kl_div_loss (class in mlx.nn.losses)": [[395, "mlx.nn.losses.kl_div_loss"]], "l1_loss (class in mlx.nn.losses)": [[396, "mlx.nn.losses.l1_loss"]], "log_cosh_loss (class in mlx.nn.losses)": [[397, "mlx.nn.losses.log_cosh_loss"]], "margin_ranking_loss (class in mlx.nn.losses)": [[398, "mlx.nn.losses.margin_ranking_loss"]], "mse_loss (class in mlx.nn.losses)": [[399, "mlx.nn.losses.mse_loss"]], "nll_loss (class in mlx.nn.losses)": [[400, "mlx.nn.losses.nll_loss"]], "smooth_l1_loss (class in mlx.nn.losses)": [[401, "mlx.nn.losses.smooth_l1_loss"]], "triplet_loss (class in mlx.nn.losses)": [[402, "mlx.nn.losses.triplet_loss"]], "sigmoid (class in mlx.nn)": [[408, "mlx.nn.sigmoid"]], "module (class in mlx.nn)": [[420, "mlx.nn.Module"]], "adadelta (class in mlx.optimizers)": [[423, "mlx.optimizers.AdaDelta"]], "adafactor (class in mlx.optimizers)": [[424, "mlx.optimizers.Adafactor"]], "adagrad (class in mlx.optimizers)": [[425, "mlx.optimizers.Adagrad"]], "adam (class in mlx.optimizers)": [[426, "mlx.optimizers.Adam"]], "adamw (class in mlx.optimizers)": [[427, "mlx.optimizers.AdamW"]], "adamax (class in mlx.optimizers)": [[428, "mlx.optimizers.Adamax"]], "lion (class in mlx.optimizers)": [[429, "mlx.optimizers.Lion"]], "apply_gradients() (optimizer method)": [[430, "mlx.optimizers.Optimizer.apply_gradients"]], "init() (optimizer method)": [[431, "mlx.optimizers.Optimizer.init"]], "state (optimizer property)": [[432, "mlx.optimizers.Optimizer.state"]], "update() (optimizer method)": [[433, "mlx.optimizers.Optimizer.update"]], "rmsprop (class in mlx.optimizers)": [[434, "mlx.optimizers.RMSprop"]], "sgd (class in mlx.optimizers)": [[435, "mlx.optimizers.SGD"]], "cosine_decay() (in module mlx.optimizers)": [[436, "mlx.optimizers.cosine_decay"]], "exponential_decay() (in module mlx.optimizers)": [[437, "mlx.optimizers.exponential_decay"]], "join_schedules() (in module mlx.optimizers)": [[438, "mlx.optimizers.join_schedules"]], "linear_schedule() (in module mlx.optimizers)": [[439, "mlx.optimizers.linear_schedule"]], "step_decay() (in module mlx.optimizers)": [[440, "mlx.optimizers.step_decay"]], "optimizer (class in mlx.optimizers)": [[442, "mlx.optimizers.Optimizer"]]}}) \ No newline at end of file +Search.setIndex({"docnames": ["cpp/ops", "dev/extensions", "dev/metal_debugger", "examples/linear_regression", "examples/llama-inference", "examples/mlp", "index", "install", "python/_autosummary/mlx.core.Device", "python/_autosummary/mlx.core.Dtype", "python/_autosummary/mlx.core.DtypeCategory", "python/_autosummary/mlx.core.abs", "python/_autosummary/mlx.core.add", "python/_autosummary/mlx.core.addmm", "python/_autosummary/mlx.core.all", "python/_autosummary/mlx.core.allclose", "python/_autosummary/mlx.core.any", "python/_autosummary/mlx.core.arange", "python/_autosummary/mlx.core.arccos", "python/_autosummary/mlx.core.arccosh", "python/_autosummary/mlx.core.arcsin", "python/_autosummary/mlx.core.arcsinh", "python/_autosummary/mlx.core.arctan", "python/_autosummary/mlx.core.arctan2", "python/_autosummary/mlx.core.arctanh", "python/_autosummary/mlx.core.argmax", "python/_autosummary/mlx.core.argmin", "python/_autosummary/mlx.core.argpartition", "python/_autosummary/mlx.core.argsort", "python/_autosummary/mlx.core.array", "python/_autosummary/mlx.core.array.T", "python/_autosummary/mlx.core.array.abs", "python/_autosummary/mlx.core.array.all", "python/_autosummary/mlx.core.array.any", "python/_autosummary/mlx.core.array.argmax", "python/_autosummary/mlx.core.array.argmin", "python/_autosummary/mlx.core.array.astype", "python/_autosummary/mlx.core.array.at", "python/_autosummary/mlx.core.array.conj", "python/_autosummary/mlx.core.array.cos", "python/_autosummary/mlx.core.array.cummax", "python/_autosummary/mlx.core.array.cummin", "python/_autosummary/mlx.core.array.cumprod", "python/_autosummary/mlx.core.array.cumsum", "python/_autosummary/mlx.core.array.diag", "python/_autosummary/mlx.core.array.diagonal", "python/_autosummary/mlx.core.array.dtype", "python/_autosummary/mlx.core.array.exp", "python/_autosummary/mlx.core.array.flatten", "python/_autosummary/mlx.core.array.item", "python/_autosummary/mlx.core.array.itemsize", "python/_autosummary/mlx.core.array.log", "python/_autosummary/mlx.core.array.log10", "python/_autosummary/mlx.core.array.log1p", "python/_autosummary/mlx.core.array.log2", "python/_autosummary/mlx.core.array.logsumexp", "python/_autosummary/mlx.core.array.max", "python/_autosummary/mlx.core.array.mean", "python/_autosummary/mlx.core.array.min", "python/_autosummary/mlx.core.array.moveaxis", "python/_autosummary/mlx.core.array.nbytes", "python/_autosummary/mlx.core.array.ndim", "python/_autosummary/mlx.core.array.prod", "python/_autosummary/mlx.core.array.reciprocal", "python/_autosummary/mlx.core.array.reshape", "python/_autosummary/mlx.core.array.round", "python/_autosummary/mlx.core.array.rsqrt", "python/_autosummary/mlx.core.array.shape", "python/_autosummary/mlx.core.array.sin", "python/_autosummary/mlx.core.array.size", "python/_autosummary/mlx.core.array.split", "python/_autosummary/mlx.core.array.sqrt", "python/_autosummary/mlx.core.array.square", "python/_autosummary/mlx.core.array.squeeze", "python/_autosummary/mlx.core.array.sum", "python/_autosummary/mlx.core.array.swapaxes", "python/_autosummary/mlx.core.array.tolist", "python/_autosummary/mlx.core.array.transpose", "python/_autosummary/mlx.core.array.var", "python/_autosummary/mlx.core.array.view", "python/_autosummary/mlx.core.array_equal", "python/_autosummary/mlx.core.as_strided", "python/_autosummary/mlx.core.atleast_1d", "python/_autosummary/mlx.core.atleast_2d", "python/_autosummary/mlx.core.atleast_3d", "python/_autosummary/mlx.core.bitwise_and", "python/_autosummary/mlx.core.bitwise_or", "python/_autosummary/mlx.core.bitwise_xor", "python/_autosummary/mlx.core.block_masked_mm", "python/_autosummary/mlx.core.broadcast_to", "python/_autosummary/mlx.core.ceil", "python/_autosummary/mlx.core.clip", "python/_autosummary/mlx.core.compile", "python/_autosummary/mlx.core.concatenate", "python/_autosummary/mlx.core.conj", "python/_autosummary/mlx.core.conjugate", "python/_autosummary/mlx.core.conv1d", "python/_autosummary/mlx.core.conv2d", "python/_autosummary/mlx.core.conv_general", "python/_autosummary/mlx.core.convolve", "python/_autosummary/mlx.core.cos", "python/_autosummary/mlx.core.cosh", "python/_autosummary/mlx.core.cummax", "python/_autosummary/mlx.core.cummin", "python/_autosummary/mlx.core.cumprod", "python/_autosummary/mlx.core.cumsum", "python/_autosummary/mlx.core.custom_function", "python/_autosummary/mlx.core.default_device", "python/_autosummary/mlx.core.default_stream", "python/_autosummary/mlx.core.degrees", "python/_autosummary/mlx.core.dequantize", "python/_autosummary/mlx.core.diag", "python/_autosummary/mlx.core.diagonal", "python/_autosummary/mlx.core.disable_compile", "python/_autosummary/mlx.core.distributed.Group", "python/_autosummary/mlx.core.distributed.all_gather", "python/_autosummary/mlx.core.distributed.all_sum", "python/_autosummary/mlx.core.distributed.init", "python/_autosummary/mlx.core.distributed.is_available", "python/_autosummary/mlx.core.divide", "python/_autosummary/mlx.core.divmod", "python/_autosummary/mlx.core.einsum", "python/_autosummary/mlx.core.einsum_path", "python/_autosummary/mlx.core.enable_compile", "python/_autosummary/mlx.core.equal", "python/_autosummary/mlx.core.erf", "python/_autosummary/mlx.core.erfinv", "python/_autosummary/mlx.core.eval", "python/_autosummary/mlx.core.exp", "python/_autosummary/mlx.core.expand_dims", "python/_autosummary/mlx.core.expm1", "python/_autosummary/mlx.core.eye", "python/_autosummary/mlx.core.fast.layer_norm", "python/_autosummary/mlx.core.fast.rms_norm", "python/_autosummary/mlx.core.fast.rope", "python/_autosummary/mlx.core.fast.scaled_dot_product_attention", "python/_autosummary/mlx.core.fft.fft", "python/_autosummary/mlx.core.fft.fft2", "python/_autosummary/mlx.core.fft.fftn", "python/_autosummary/mlx.core.fft.ifft", "python/_autosummary/mlx.core.fft.ifft2", "python/_autosummary/mlx.core.fft.ifftn", "python/_autosummary/mlx.core.fft.irfft", "python/_autosummary/mlx.core.fft.irfft2", "python/_autosummary/mlx.core.fft.irfftn", "python/_autosummary/mlx.core.fft.rfft", "python/_autosummary/mlx.core.fft.rfft2", "python/_autosummary/mlx.core.fft.rfftn", "python/_autosummary/mlx.core.flatten", "python/_autosummary/mlx.core.floor", "python/_autosummary/mlx.core.floor_divide", "python/_autosummary/mlx.core.full", "python/_autosummary/mlx.core.gather_mm", "python/_autosummary/mlx.core.gather_qmm", "python/_autosummary/mlx.core.grad", "python/_autosummary/mlx.core.greater", "python/_autosummary/mlx.core.greater_equal", "python/_autosummary/mlx.core.hadamard_transform", "python/_autosummary/mlx.core.identity", "python/_autosummary/mlx.core.inner", "python/_autosummary/mlx.core.isclose", "python/_autosummary/mlx.core.isinf", "python/_autosummary/mlx.core.isnan", "python/_autosummary/mlx.core.isneginf", "python/_autosummary/mlx.core.isposinf", "python/_autosummary/mlx.core.issubdtype", "python/_autosummary/mlx.core.jvp", "python/_autosummary/mlx.core.left_shift", "python/_autosummary/mlx.core.less", "python/_autosummary/mlx.core.less_equal", "python/_autosummary/mlx.core.linalg.cholesky", "python/_autosummary/mlx.core.linalg.cholesky_inv", "python/_autosummary/mlx.core.linalg.inv", "python/_autosummary/mlx.core.linalg.norm", "python/_autosummary/mlx.core.linalg.qr", "python/_autosummary/mlx.core.linalg.svd", "python/_autosummary/mlx.core.linalg.tri_inv", "python/_autosummary/mlx.core.linspace", "python/_autosummary/mlx.core.load", "python/_autosummary/mlx.core.log", "python/_autosummary/mlx.core.log10", "python/_autosummary/mlx.core.log1p", "python/_autosummary/mlx.core.log2", "python/_autosummary/mlx.core.logaddexp", "python/_autosummary/mlx.core.logical_and", "python/_autosummary/mlx.core.logical_not", "python/_autosummary/mlx.core.logical_or", "python/_autosummary/mlx.core.logsumexp", "python/_autosummary/mlx.core.matmul", "python/_autosummary/mlx.core.max", "python/_autosummary/mlx.core.maximum", "python/_autosummary/mlx.core.mean", "python/_autosummary/mlx.core.meshgrid", "python/_autosummary/mlx.core.metal.clear_cache", "python/_autosummary/mlx.core.metal.device_info", "python/_autosummary/mlx.core.metal.get_active_memory", "python/_autosummary/mlx.core.metal.get_cache_memory", "python/_autosummary/mlx.core.metal.get_peak_memory", "python/_autosummary/mlx.core.metal.is_available", "python/_autosummary/mlx.core.metal.reset_peak_memory", "python/_autosummary/mlx.core.metal.set_cache_limit", "python/_autosummary/mlx.core.metal.set_memory_limit", "python/_autosummary/mlx.core.metal.start_capture", "python/_autosummary/mlx.core.metal.stop_capture", "python/_autosummary/mlx.core.min", "python/_autosummary/mlx.core.minimum", "python/_autosummary/mlx.core.moveaxis", "python/_autosummary/mlx.core.multiply", "python/_autosummary/mlx.core.nan_to_num", "python/_autosummary/mlx.core.negative", "python/_autosummary/mlx.core.new_stream", "python/_autosummary/mlx.core.not_equal", "python/_autosummary/mlx.core.ones", "python/_autosummary/mlx.core.ones_like", "python/_autosummary/mlx.core.outer", "python/_autosummary/mlx.core.pad", "python/_autosummary/mlx.core.partition", "python/_autosummary/mlx.core.power", "python/_autosummary/mlx.core.prod", "python/_autosummary/mlx.core.quantize", "python/_autosummary/mlx.core.quantized_matmul", "python/_autosummary/mlx.core.radians", "python/_autosummary/mlx.core.random.bernoulli", "python/_autosummary/mlx.core.random.categorical", "python/_autosummary/mlx.core.random.gumbel", "python/_autosummary/mlx.core.random.key", "python/_autosummary/mlx.core.random.laplace", "python/_autosummary/mlx.core.random.multivariate_normal", "python/_autosummary/mlx.core.random.normal", "python/_autosummary/mlx.core.random.randint", "python/_autosummary/mlx.core.random.seed", "python/_autosummary/mlx.core.random.split", "python/_autosummary/mlx.core.random.truncated_normal", "python/_autosummary/mlx.core.random.uniform", "python/_autosummary/mlx.core.reciprocal", "python/_autosummary/mlx.core.remainder", "python/_autosummary/mlx.core.repeat", "python/_autosummary/mlx.core.reshape", "python/_autosummary/mlx.core.right_shift", "python/_autosummary/mlx.core.round", "python/_autosummary/mlx.core.rsqrt", "python/_autosummary/mlx.core.save", "python/_autosummary/mlx.core.save_gguf", "python/_autosummary/mlx.core.save_safetensors", "python/_autosummary/mlx.core.savez", "python/_autosummary/mlx.core.savez_compressed", "python/_autosummary/mlx.core.set_default_device", "python/_autosummary/mlx.core.set_default_stream", "python/_autosummary/mlx.core.sigmoid", "python/_autosummary/mlx.core.sign", "python/_autosummary/mlx.core.sin", "python/_autosummary/mlx.core.sinh", "python/_autosummary/mlx.core.softmax", "python/_autosummary/mlx.core.sort", "python/_autosummary/mlx.core.split", "python/_autosummary/mlx.core.sqrt", "python/_autosummary/mlx.core.square", "python/_autosummary/mlx.core.squeeze", "python/_autosummary/mlx.core.stack", "python/_autosummary/mlx.core.std", "python/_autosummary/mlx.core.stop_gradient", "python/_autosummary/mlx.core.stream", "python/_autosummary/mlx.core.subtract", "python/_autosummary/mlx.core.sum", "python/_autosummary/mlx.core.swapaxes", "python/_autosummary/mlx.core.synchronize", "python/_autosummary/mlx.core.take", "python/_autosummary/mlx.core.take_along_axis", "python/_autosummary/mlx.core.tan", "python/_autosummary/mlx.core.tanh", "python/_autosummary/mlx.core.tensordot", "python/_autosummary/mlx.core.tile", "python/_autosummary/mlx.core.topk", "python/_autosummary/mlx.core.trace", "python/_autosummary/mlx.core.transpose", "python/_autosummary/mlx.core.tri", "python/_autosummary/mlx.core.tril", "python/_autosummary/mlx.core.triu", "python/_autosummary/mlx.core.value_and_grad", "python/_autosummary/mlx.core.var", "python/_autosummary/mlx.core.view", "python/_autosummary/mlx.core.vjp", "python/_autosummary/mlx.core.vmap", "python/_autosummary/mlx.core.where", "python/_autosummary/mlx.core.zeros", "python/_autosummary/mlx.core.zeros_like", "python/_autosummary/mlx.nn.quantize", "python/_autosummary/mlx.nn.value_and_grad", "python/_autosummary/mlx.optimizers.clip_grad_norm", "python/_autosummary/mlx.utils.tree_flatten", "python/_autosummary/mlx.utils.tree_map", "python/_autosummary/mlx.utils.tree_map_with_path", "python/_autosummary/mlx.utils.tree_reduce", "python/_autosummary/mlx.utils.tree_unflatten", "python/_autosummary/stream_class", "python/array", "python/data_types", "python/devices_and_streams", "python/distributed", "python/fast", "python/fft", "python/linalg", "python/metal", "python/nn", "python/nn/_autosummary/mlx.nn.ALiBi", "python/nn/_autosummary/mlx.nn.AvgPool1d", "python/nn/_autosummary/mlx.nn.AvgPool2d", "python/nn/_autosummary/mlx.nn.BatchNorm", "python/nn/_autosummary/mlx.nn.Conv1d", "python/nn/_autosummary/mlx.nn.Conv2d", "python/nn/_autosummary/mlx.nn.Conv3d", "python/nn/_autosummary/mlx.nn.Dropout", "python/nn/_autosummary/mlx.nn.Dropout2d", "python/nn/_autosummary/mlx.nn.Dropout3d", "python/nn/_autosummary/mlx.nn.Embedding", "python/nn/_autosummary/mlx.nn.GELU", "python/nn/_autosummary/mlx.nn.GLU", "python/nn/_autosummary/mlx.nn.GRU", "python/nn/_autosummary/mlx.nn.GroupNorm", "python/nn/_autosummary/mlx.nn.HardShrink", "python/nn/_autosummary/mlx.nn.HardTanh", "python/nn/_autosummary/mlx.nn.Hardswish", "python/nn/_autosummary/mlx.nn.InstanceNorm", "python/nn/_autosummary/mlx.nn.LSTM", "python/nn/_autosummary/mlx.nn.LayerNorm", "python/nn/_autosummary/mlx.nn.LeakyReLU", "python/nn/_autosummary/mlx.nn.Linear", "python/nn/_autosummary/mlx.nn.MaxPool1d", "python/nn/_autosummary/mlx.nn.MaxPool2d", "python/nn/_autosummary/mlx.nn.Mish", "python/nn/_autosummary/mlx.nn.Module.apply", "python/nn/_autosummary/mlx.nn.Module.apply_to_modules", "python/nn/_autosummary/mlx.nn.Module.children", "python/nn/_autosummary/mlx.nn.Module.eval", "python/nn/_autosummary/mlx.nn.Module.filter_and_map", "python/nn/_autosummary/mlx.nn.Module.freeze", "python/nn/_autosummary/mlx.nn.Module.leaf_modules", "python/nn/_autosummary/mlx.nn.Module.load_weights", "python/nn/_autosummary/mlx.nn.Module.modules", "python/nn/_autosummary/mlx.nn.Module.named_modules", "python/nn/_autosummary/mlx.nn.Module.parameters", "python/nn/_autosummary/mlx.nn.Module.save_weights", "python/nn/_autosummary/mlx.nn.Module.set_dtype", "python/nn/_autosummary/mlx.nn.Module.state", "python/nn/_autosummary/mlx.nn.Module.train", "python/nn/_autosummary/mlx.nn.Module.trainable_parameters", "python/nn/_autosummary/mlx.nn.Module.training", "python/nn/_autosummary/mlx.nn.Module.unfreeze", "python/nn/_autosummary/mlx.nn.Module.update", "python/nn/_autosummary/mlx.nn.Module.update_modules", "python/nn/_autosummary/mlx.nn.MultiHeadAttention", "python/nn/_autosummary/mlx.nn.PReLU", "python/nn/_autosummary/mlx.nn.QuantizedEmbedding", "python/nn/_autosummary/mlx.nn.QuantizedLinear", "python/nn/_autosummary/mlx.nn.RMSNorm", "python/nn/_autosummary/mlx.nn.RNN", "python/nn/_autosummary/mlx.nn.ReLU", "python/nn/_autosummary/mlx.nn.ReLU6", "python/nn/_autosummary/mlx.nn.RoPE", "python/nn/_autosummary/mlx.nn.SELU", "python/nn/_autosummary/mlx.nn.Sequential", "python/nn/_autosummary/mlx.nn.SiLU", "python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding", "python/nn/_autosummary/mlx.nn.Softmax", "python/nn/_autosummary/mlx.nn.Softmin", "python/nn/_autosummary/mlx.nn.Softplus", "python/nn/_autosummary/mlx.nn.Softshrink", "python/nn/_autosummary/mlx.nn.Softsign", "python/nn/_autosummary/mlx.nn.Step", "python/nn/_autosummary/mlx.nn.Tanh", "python/nn/_autosummary/mlx.nn.Transformer", "python/nn/_autosummary/mlx.nn.Upsample", "python/nn/_autosummary/mlx.nn.init.constant", "python/nn/_autosummary/mlx.nn.init.glorot_normal", "python/nn/_autosummary/mlx.nn.init.glorot_uniform", "python/nn/_autosummary/mlx.nn.init.he_normal", "python/nn/_autosummary/mlx.nn.init.he_uniform", "python/nn/_autosummary/mlx.nn.init.identity", "python/nn/_autosummary/mlx.nn.init.normal", "python/nn/_autosummary/mlx.nn.init.uniform", "python/nn/_autosummary_functions/mlx.nn.elu", "python/nn/_autosummary_functions/mlx.nn.gelu", "python/nn/_autosummary_functions/mlx.nn.gelu_approx", "python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx", "python/nn/_autosummary_functions/mlx.nn.glu", "python/nn/_autosummary_functions/mlx.nn.hard_shrink", "python/nn/_autosummary_functions/mlx.nn.hard_tanh", "python/nn/_autosummary_functions/mlx.nn.hardswish", "python/nn/_autosummary_functions/mlx.nn.leaky_relu", "python/nn/_autosummary_functions/mlx.nn.log_sigmoid", "python/nn/_autosummary_functions/mlx.nn.log_softmax", "python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy", "python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss", "python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy", "python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss", "python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss", "python/nn/_autosummary_functions/mlx.nn.losses.huber_loss", "python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss", "python/nn/_autosummary_functions/mlx.nn.losses.l1_loss", "python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss", "python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss", "python/nn/_autosummary_functions/mlx.nn.losses.mse_loss", "python/nn/_autosummary_functions/mlx.nn.losses.nll_loss", "python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss", "python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss", "python/nn/_autosummary_functions/mlx.nn.mish", "python/nn/_autosummary_functions/mlx.nn.prelu", "python/nn/_autosummary_functions/mlx.nn.relu", "python/nn/_autosummary_functions/mlx.nn.relu6", "python/nn/_autosummary_functions/mlx.nn.selu", "python/nn/_autosummary_functions/mlx.nn.sigmoid", "python/nn/_autosummary_functions/mlx.nn.silu", "python/nn/_autosummary_functions/mlx.nn.softmax", "python/nn/_autosummary_functions/mlx.nn.softmin", "python/nn/_autosummary_functions/mlx.nn.softplus", "python/nn/_autosummary_functions/mlx.nn.softshrink", "python/nn/_autosummary_functions/mlx.nn.step", "python/nn/_autosummary_functions/mlx.nn.tanh", "python/nn/functions", "python/nn/init", "python/nn/layers", "python/nn/losses", "python/nn/module", "python/ops", "python/optimizers", "python/optimizers/_autosummary/mlx.optimizers.AdaDelta", "python/optimizers/_autosummary/mlx.optimizers.Adafactor", "python/optimizers/_autosummary/mlx.optimizers.Adagrad", "python/optimizers/_autosummary/mlx.optimizers.Adam", "python/optimizers/_autosummary/mlx.optimizers.AdamW", "python/optimizers/_autosummary/mlx.optimizers.Adamax", "python/optimizers/_autosummary/mlx.optimizers.Lion", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.init", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.state", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.update", "python/optimizers/_autosummary/mlx.optimizers.RMSprop", "python/optimizers/_autosummary/mlx.optimizers.SGD", "python/optimizers/_autosummary/mlx.optimizers.cosine_decay", "python/optimizers/_autosummary/mlx.optimizers.exponential_decay", "python/optimizers/_autosummary/mlx.optimizers.join_schedules", "python/optimizers/_autosummary/mlx.optimizers.linear_schedule", "python/optimizers/_autosummary/mlx.optimizers.step_decay", "python/optimizers/common_optimizers", "python/optimizers/optimizer", "python/optimizers/schedulers", "python/random", "python/transforms", "python/tree_utils", "usage/compile", "usage/distributed", "usage/function_transforms", "usage/indexing", "usage/lazy_evaluation", "usage/numpy", "usage/quick_start", "usage/saving_and_loading", "usage/unified_memory", "usage/using_streams"], "filenames": ["cpp/ops.rst", "dev/extensions.rst", "dev/metal_debugger.rst", "examples/linear_regression.rst", "examples/llama-inference.rst", "examples/mlp.rst", "index.rst", "install.rst", "python/_autosummary/mlx.core.Device.rst", "python/_autosummary/mlx.core.Dtype.rst", "python/_autosummary/mlx.core.DtypeCategory.rst", "python/_autosummary/mlx.core.abs.rst", "python/_autosummary/mlx.core.add.rst", "python/_autosummary/mlx.core.addmm.rst", "python/_autosummary/mlx.core.all.rst", "python/_autosummary/mlx.core.allclose.rst", "python/_autosummary/mlx.core.any.rst", "python/_autosummary/mlx.core.arange.rst", "python/_autosummary/mlx.core.arccos.rst", "python/_autosummary/mlx.core.arccosh.rst", "python/_autosummary/mlx.core.arcsin.rst", "python/_autosummary/mlx.core.arcsinh.rst", "python/_autosummary/mlx.core.arctan.rst", "python/_autosummary/mlx.core.arctan2.rst", "python/_autosummary/mlx.core.arctanh.rst", "python/_autosummary/mlx.core.argmax.rst", "python/_autosummary/mlx.core.argmin.rst", "python/_autosummary/mlx.core.argpartition.rst", "python/_autosummary/mlx.core.argsort.rst", "python/_autosummary/mlx.core.array.rst", "python/_autosummary/mlx.core.array.T.rst", "python/_autosummary/mlx.core.array.abs.rst", "python/_autosummary/mlx.core.array.all.rst", "python/_autosummary/mlx.core.array.any.rst", "python/_autosummary/mlx.core.array.argmax.rst", "python/_autosummary/mlx.core.array.argmin.rst", "python/_autosummary/mlx.core.array.astype.rst", "python/_autosummary/mlx.core.array.at.rst", "python/_autosummary/mlx.core.array.conj.rst", "python/_autosummary/mlx.core.array.cos.rst", "python/_autosummary/mlx.core.array.cummax.rst", "python/_autosummary/mlx.core.array.cummin.rst", "python/_autosummary/mlx.core.array.cumprod.rst", "python/_autosummary/mlx.core.array.cumsum.rst", "python/_autosummary/mlx.core.array.diag.rst", "python/_autosummary/mlx.core.array.diagonal.rst", "python/_autosummary/mlx.core.array.dtype.rst", "python/_autosummary/mlx.core.array.exp.rst", "python/_autosummary/mlx.core.array.flatten.rst", "python/_autosummary/mlx.core.array.item.rst", "python/_autosummary/mlx.core.array.itemsize.rst", "python/_autosummary/mlx.core.array.log.rst", "python/_autosummary/mlx.core.array.log10.rst", "python/_autosummary/mlx.core.array.log1p.rst", "python/_autosummary/mlx.core.array.log2.rst", "python/_autosummary/mlx.core.array.logsumexp.rst", "python/_autosummary/mlx.core.array.max.rst", "python/_autosummary/mlx.core.array.mean.rst", "python/_autosummary/mlx.core.array.min.rst", "python/_autosummary/mlx.core.array.moveaxis.rst", "python/_autosummary/mlx.core.array.nbytes.rst", "python/_autosummary/mlx.core.array.ndim.rst", "python/_autosummary/mlx.core.array.prod.rst", "python/_autosummary/mlx.core.array.reciprocal.rst", "python/_autosummary/mlx.core.array.reshape.rst", "python/_autosummary/mlx.core.array.round.rst", "python/_autosummary/mlx.core.array.rsqrt.rst", "python/_autosummary/mlx.core.array.shape.rst", "python/_autosummary/mlx.core.array.sin.rst", "python/_autosummary/mlx.core.array.size.rst", "python/_autosummary/mlx.core.array.split.rst", "python/_autosummary/mlx.core.array.sqrt.rst", "python/_autosummary/mlx.core.array.square.rst", "python/_autosummary/mlx.core.array.squeeze.rst", "python/_autosummary/mlx.core.array.sum.rst", "python/_autosummary/mlx.core.array.swapaxes.rst", "python/_autosummary/mlx.core.array.tolist.rst", "python/_autosummary/mlx.core.array.transpose.rst", "python/_autosummary/mlx.core.array.var.rst", "python/_autosummary/mlx.core.array.view.rst", "python/_autosummary/mlx.core.array_equal.rst", "python/_autosummary/mlx.core.as_strided.rst", "python/_autosummary/mlx.core.atleast_1d.rst", "python/_autosummary/mlx.core.atleast_2d.rst", "python/_autosummary/mlx.core.atleast_3d.rst", "python/_autosummary/mlx.core.bitwise_and.rst", "python/_autosummary/mlx.core.bitwise_or.rst", "python/_autosummary/mlx.core.bitwise_xor.rst", "python/_autosummary/mlx.core.block_masked_mm.rst", "python/_autosummary/mlx.core.broadcast_to.rst", "python/_autosummary/mlx.core.ceil.rst", "python/_autosummary/mlx.core.clip.rst", "python/_autosummary/mlx.core.compile.rst", "python/_autosummary/mlx.core.concatenate.rst", "python/_autosummary/mlx.core.conj.rst", "python/_autosummary/mlx.core.conjugate.rst", "python/_autosummary/mlx.core.conv1d.rst", "python/_autosummary/mlx.core.conv2d.rst", "python/_autosummary/mlx.core.conv_general.rst", "python/_autosummary/mlx.core.convolve.rst", "python/_autosummary/mlx.core.cos.rst", "python/_autosummary/mlx.core.cosh.rst", "python/_autosummary/mlx.core.cummax.rst", "python/_autosummary/mlx.core.cummin.rst", "python/_autosummary/mlx.core.cumprod.rst", "python/_autosummary/mlx.core.cumsum.rst", "python/_autosummary/mlx.core.custom_function.rst", "python/_autosummary/mlx.core.default_device.rst", "python/_autosummary/mlx.core.default_stream.rst", "python/_autosummary/mlx.core.degrees.rst", "python/_autosummary/mlx.core.dequantize.rst", "python/_autosummary/mlx.core.diag.rst", "python/_autosummary/mlx.core.diagonal.rst", "python/_autosummary/mlx.core.disable_compile.rst", "python/_autosummary/mlx.core.distributed.Group.rst", "python/_autosummary/mlx.core.distributed.all_gather.rst", "python/_autosummary/mlx.core.distributed.all_sum.rst", "python/_autosummary/mlx.core.distributed.init.rst", "python/_autosummary/mlx.core.distributed.is_available.rst", "python/_autosummary/mlx.core.divide.rst", "python/_autosummary/mlx.core.divmod.rst", "python/_autosummary/mlx.core.einsum.rst", "python/_autosummary/mlx.core.einsum_path.rst", "python/_autosummary/mlx.core.enable_compile.rst", "python/_autosummary/mlx.core.equal.rst", "python/_autosummary/mlx.core.erf.rst", "python/_autosummary/mlx.core.erfinv.rst", "python/_autosummary/mlx.core.eval.rst", "python/_autosummary/mlx.core.exp.rst", "python/_autosummary/mlx.core.expand_dims.rst", "python/_autosummary/mlx.core.expm1.rst", "python/_autosummary/mlx.core.eye.rst", "python/_autosummary/mlx.core.fast.layer_norm.rst", "python/_autosummary/mlx.core.fast.rms_norm.rst", "python/_autosummary/mlx.core.fast.rope.rst", "python/_autosummary/mlx.core.fast.scaled_dot_product_attention.rst", "python/_autosummary/mlx.core.fft.fft.rst", "python/_autosummary/mlx.core.fft.fft2.rst", "python/_autosummary/mlx.core.fft.fftn.rst", "python/_autosummary/mlx.core.fft.ifft.rst", "python/_autosummary/mlx.core.fft.ifft2.rst", "python/_autosummary/mlx.core.fft.ifftn.rst", "python/_autosummary/mlx.core.fft.irfft.rst", "python/_autosummary/mlx.core.fft.irfft2.rst", "python/_autosummary/mlx.core.fft.irfftn.rst", "python/_autosummary/mlx.core.fft.rfft.rst", "python/_autosummary/mlx.core.fft.rfft2.rst", "python/_autosummary/mlx.core.fft.rfftn.rst", "python/_autosummary/mlx.core.flatten.rst", "python/_autosummary/mlx.core.floor.rst", "python/_autosummary/mlx.core.floor_divide.rst", "python/_autosummary/mlx.core.full.rst", "python/_autosummary/mlx.core.gather_mm.rst", "python/_autosummary/mlx.core.gather_qmm.rst", "python/_autosummary/mlx.core.grad.rst", "python/_autosummary/mlx.core.greater.rst", "python/_autosummary/mlx.core.greater_equal.rst", "python/_autosummary/mlx.core.hadamard_transform.rst", "python/_autosummary/mlx.core.identity.rst", "python/_autosummary/mlx.core.inner.rst", "python/_autosummary/mlx.core.isclose.rst", "python/_autosummary/mlx.core.isinf.rst", "python/_autosummary/mlx.core.isnan.rst", "python/_autosummary/mlx.core.isneginf.rst", "python/_autosummary/mlx.core.isposinf.rst", "python/_autosummary/mlx.core.issubdtype.rst", "python/_autosummary/mlx.core.jvp.rst", "python/_autosummary/mlx.core.left_shift.rst", "python/_autosummary/mlx.core.less.rst", "python/_autosummary/mlx.core.less_equal.rst", "python/_autosummary/mlx.core.linalg.cholesky.rst", "python/_autosummary/mlx.core.linalg.cholesky_inv.rst", "python/_autosummary/mlx.core.linalg.inv.rst", "python/_autosummary/mlx.core.linalg.norm.rst", "python/_autosummary/mlx.core.linalg.qr.rst", "python/_autosummary/mlx.core.linalg.svd.rst", "python/_autosummary/mlx.core.linalg.tri_inv.rst", "python/_autosummary/mlx.core.linspace.rst", "python/_autosummary/mlx.core.load.rst", "python/_autosummary/mlx.core.log.rst", "python/_autosummary/mlx.core.log10.rst", "python/_autosummary/mlx.core.log1p.rst", "python/_autosummary/mlx.core.log2.rst", "python/_autosummary/mlx.core.logaddexp.rst", "python/_autosummary/mlx.core.logical_and.rst", "python/_autosummary/mlx.core.logical_not.rst", "python/_autosummary/mlx.core.logical_or.rst", "python/_autosummary/mlx.core.logsumexp.rst", "python/_autosummary/mlx.core.matmul.rst", "python/_autosummary/mlx.core.max.rst", "python/_autosummary/mlx.core.maximum.rst", "python/_autosummary/mlx.core.mean.rst", "python/_autosummary/mlx.core.meshgrid.rst", "python/_autosummary/mlx.core.metal.clear_cache.rst", "python/_autosummary/mlx.core.metal.device_info.rst", "python/_autosummary/mlx.core.metal.get_active_memory.rst", "python/_autosummary/mlx.core.metal.get_cache_memory.rst", "python/_autosummary/mlx.core.metal.get_peak_memory.rst", "python/_autosummary/mlx.core.metal.is_available.rst", "python/_autosummary/mlx.core.metal.reset_peak_memory.rst", "python/_autosummary/mlx.core.metal.set_cache_limit.rst", "python/_autosummary/mlx.core.metal.set_memory_limit.rst", "python/_autosummary/mlx.core.metal.start_capture.rst", "python/_autosummary/mlx.core.metal.stop_capture.rst", "python/_autosummary/mlx.core.min.rst", "python/_autosummary/mlx.core.minimum.rst", "python/_autosummary/mlx.core.moveaxis.rst", "python/_autosummary/mlx.core.multiply.rst", "python/_autosummary/mlx.core.nan_to_num.rst", "python/_autosummary/mlx.core.negative.rst", "python/_autosummary/mlx.core.new_stream.rst", "python/_autosummary/mlx.core.not_equal.rst", "python/_autosummary/mlx.core.ones.rst", "python/_autosummary/mlx.core.ones_like.rst", "python/_autosummary/mlx.core.outer.rst", "python/_autosummary/mlx.core.pad.rst", "python/_autosummary/mlx.core.partition.rst", "python/_autosummary/mlx.core.power.rst", "python/_autosummary/mlx.core.prod.rst", "python/_autosummary/mlx.core.quantize.rst", "python/_autosummary/mlx.core.quantized_matmul.rst", "python/_autosummary/mlx.core.radians.rst", "python/_autosummary/mlx.core.random.bernoulli.rst", "python/_autosummary/mlx.core.random.categorical.rst", "python/_autosummary/mlx.core.random.gumbel.rst", "python/_autosummary/mlx.core.random.key.rst", "python/_autosummary/mlx.core.random.laplace.rst", "python/_autosummary/mlx.core.random.multivariate_normal.rst", "python/_autosummary/mlx.core.random.normal.rst", "python/_autosummary/mlx.core.random.randint.rst", "python/_autosummary/mlx.core.random.seed.rst", "python/_autosummary/mlx.core.random.split.rst", "python/_autosummary/mlx.core.random.truncated_normal.rst", "python/_autosummary/mlx.core.random.uniform.rst", "python/_autosummary/mlx.core.reciprocal.rst", "python/_autosummary/mlx.core.remainder.rst", "python/_autosummary/mlx.core.repeat.rst", "python/_autosummary/mlx.core.reshape.rst", "python/_autosummary/mlx.core.right_shift.rst", "python/_autosummary/mlx.core.round.rst", "python/_autosummary/mlx.core.rsqrt.rst", "python/_autosummary/mlx.core.save.rst", "python/_autosummary/mlx.core.save_gguf.rst", "python/_autosummary/mlx.core.save_safetensors.rst", "python/_autosummary/mlx.core.savez.rst", "python/_autosummary/mlx.core.savez_compressed.rst", "python/_autosummary/mlx.core.set_default_device.rst", "python/_autosummary/mlx.core.set_default_stream.rst", "python/_autosummary/mlx.core.sigmoid.rst", "python/_autosummary/mlx.core.sign.rst", "python/_autosummary/mlx.core.sin.rst", "python/_autosummary/mlx.core.sinh.rst", "python/_autosummary/mlx.core.softmax.rst", "python/_autosummary/mlx.core.sort.rst", "python/_autosummary/mlx.core.split.rst", "python/_autosummary/mlx.core.sqrt.rst", "python/_autosummary/mlx.core.square.rst", "python/_autosummary/mlx.core.squeeze.rst", "python/_autosummary/mlx.core.stack.rst", "python/_autosummary/mlx.core.std.rst", "python/_autosummary/mlx.core.stop_gradient.rst", "python/_autosummary/mlx.core.stream.rst", "python/_autosummary/mlx.core.subtract.rst", "python/_autosummary/mlx.core.sum.rst", "python/_autosummary/mlx.core.swapaxes.rst", "python/_autosummary/mlx.core.synchronize.rst", "python/_autosummary/mlx.core.take.rst", "python/_autosummary/mlx.core.take_along_axis.rst", "python/_autosummary/mlx.core.tan.rst", "python/_autosummary/mlx.core.tanh.rst", "python/_autosummary/mlx.core.tensordot.rst", "python/_autosummary/mlx.core.tile.rst", "python/_autosummary/mlx.core.topk.rst", "python/_autosummary/mlx.core.trace.rst", "python/_autosummary/mlx.core.transpose.rst", "python/_autosummary/mlx.core.tri.rst", "python/_autosummary/mlx.core.tril.rst", "python/_autosummary/mlx.core.triu.rst", "python/_autosummary/mlx.core.value_and_grad.rst", "python/_autosummary/mlx.core.var.rst", "python/_autosummary/mlx.core.view.rst", "python/_autosummary/mlx.core.vjp.rst", "python/_autosummary/mlx.core.vmap.rst", "python/_autosummary/mlx.core.where.rst", "python/_autosummary/mlx.core.zeros.rst", "python/_autosummary/mlx.core.zeros_like.rst", "python/_autosummary/mlx.nn.quantize.rst", "python/_autosummary/mlx.nn.value_and_grad.rst", "python/_autosummary/mlx.optimizers.clip_grad_norm.rst", "python/_autosummary/mlx.utils.tree_flatten.rst", "python/_autosummary/mlx.utils.tree_map.rst", "python/_autosummary/mlx.utils.tree_map_with_path.rst", "python/_autosummary/mlx.utils.tree_reduce.rst", "python/_autosummary/mlx.utils.tree_unflatten.rst", "python/_autosummary/stream_class.rst", "python/array.rst", "python/data_types.rst", "python/devices_and_streams.rst", "python/distributed.rst", "python/fast.rst", "python/fft.rst", "python/linalg.rst", "python/metal.rst", "python/nn.rst", "python/nn/_autosummary/mlx.nn.ALiBi.rst", "python/nn/_autosummary/mlx.nn.AvgPool1d.rst", "python/nn/_autosummary/mlx.nn.AvgPool2d.rst", "python/nn/_autosummary/mlx.nn.BatchNorm.rst", "python/nn/_autosummary/mlx.nn.Conv1d.rst", "python/nn/_autosummary/mlx.nn.Conv2d.rst", "python/nn/_autosummary/mlx.nn.Conv3d.rst", "python/nn/_autosummary/mlx.nn.Dropout.rst", "python/nn/_autosummary/mlx.nn.Dropout2d.rst", "python/nn/_autosummary/mlx.nn.Dropout3d.rst", "python/nn/_autosummary/mlx.nn.Embedding.rst", "python/nn/_autosummary/mlx.nn.GELU.rst", "python/nn/_autosummary/mlx.nn.GLU.rst", "python/nn/_autosummary/mlx.nn.GRU.rst", "python/nn/_autosummary/mlx.nn.GroupNorm.rst", "python/nn/_autosummary/mlx.nn.HardShrink.rst", "python/nn/_autosummary/mlx.nn.HardTanh.rst", "python/nn/_autosummary/mlx.nn.Hardswish.rst", "python/nn/_autosummary/mlx.nn.InstanceNorm.rst", "python/nn/_autosummary/mlx.nn.LSTM.rst", "python/nn/_autosummary/mlx.nn.LayerNorm.rst", "python/nn/_autosummary/mlx.nn.LeakyReLU.rst", "python/nn/_autosummary/mlx.nn.Linear.rst", "python/nn/_autosummary/mlx.nn.MaxPool1d.rst", "python/nn/_autosummary/mlx.nn.MaxPool2d.rst", "python/nn/_autosummary/mlx.nn.Mish.rst", "python/nn/_autosummary/mlx.nn.Module.apply.rst", "python/nn/_autosummary/mlx.nn.Module.apply_to_modules.rst", "python/nn/_autosummary/mlx.nn.Module.children.rst", "python/nn/_autosummary/mlx.nn.Module.eval.rst", "python/nn/_autosummary/mlx.nn.Module.filter_and_map.rst", "python/nn/_autosummary/mlx.nn.Module.freeze.rst", "python/nn/_autosummary/mlx.nn.Module.leaf_modules.rst", "python/nn/_autosummary/mlx.nn.Module.load_weights.rst", "python/nn/_autosummary/mlx.nn.Module.modules.rst", "python/nn/_autosummary/mlx.nn.Module.named_modules.rst", "python/nn/_autosummary/mlx.nn.Module.parameters.rst", "python/nn/_autosummary/mlx.nn.Module.save_weights.rst", "python/nn/_autosummary/mlx.nn.Module.set_dtype.rst", "python/nn/_autosummary/mlx.nn.Module.state.rst", "python/nn/_autosummary/mlx.nn.Module.train.rst", "python/nn/_autosummary/mlx.nn.Module.trainable_parameters.rst", "python/nn/_autosummary/mlx.nn.Module.training.rst", "python/nn/_autosummary/mlx.nn.Module.unfreeze.rst", "python/nn/_autosummary/mlx.nn.Module.update.rst", "python/nn/_autosummary/mlx.nn.Module.update_modules.rst", "python/nn/_autosummary/mlx.nn.MultiHeadAttention.rst", "python/nn/_autosummary/mlx.nn.PReLU.rst", "python/nn/_autosummary/mlx.nn.QuantizedEmbedding.rst", "python/nn/_autosummary/mlx.nn.QuantizedLinear.rst", "python/nn/_autosummary/mlx.nn.RMSNorm.rst", "python/nn/_autosummary/mlx.nn.RNN.rst", "python/nn/_autosummary/mlx.nn.ReLU.rst", "python/nn/_autosummary/mlx.nn.ReLU6.rst", "python/nn/_autosummary/mlx.nn.RoPE.rst", "python/nn/_autosummary/mlx.nn.SELU.rst", "python/nn/_autosummary/mlx.nn.Sequential.rst", "python/nn/_autosummary/mlx.nn.SiLU.rst", "python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.rst", "python/nn/_autosummary/mlx.nn.Softmax.rst", "python/nn/_autosummary/mlx.nn.Softmin.rst", "python/nn/_autosummary/mlx.nn.Softplus.rst", "python/nn/_autosummary/mlx.nn.Softshrink.rst", "python/nn/_autosummary/mlx.nn.Softsign.rst", "python/nn/_autosummary/mlx.nn.Step.rst", "python/nn/_autosummary/mlx.nn.Tanh.rst", "python/nn/_autosummary/mlx.nn.Transformer.rst", "python/nn/_autosummary/mlx.nn.Upsample.rst", "python/nn/_autosummary/mlx.nn.init.constant.rst", "python/nn/_autosummary/mlx.nn.init.glorot_normal.rst", "python/nn/_autosummary/mlx.nn.init.glorot_uniform.rst", "python/nn/_autosummary/mlx.nn.init.he_normal.rst", "python/nn/_autosummary/mlx.nn.init.he_uniform.rst", "python/nn/_autosummary/mlx.nn.init.identity.rst", "python/nn/_autosummary/mlx.nn.init.normal.rst", "python/nn/_autosummary/mlx.nn.init.uniform.rst", "python/nn/_autosummary_functions/mlx.nn.elu.rst", "python/nn/_autosummary_functions/mlx.nn.gelu.rst", "python/nn/_autosummary_functions/mlx.nn.gelu_approx.rst", "python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.rst", "python/nn/_autosummary_functions/mlx.nn.glu.rst", "python/nn/_autosummary_functions/mlx.nn.hard_shrink.rst", "python/nn/_autosummary_functions/mlx.nn.hard_tanh.rst", "python/nn/_autosummary_functions/mlx.nn.hardswish.rst", "python/nn/_autosummary_functions/mlx.nn.leaky_relu.rst", "python/nn/_autosummary_functions/mlx.nn.log_sigmoid.rst", "python/nn/_autosummary_functions/mlx.nn.log_softmax.rst", "python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.rst", "python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.rst", "python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.rst", "python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.rst", "python/nn/_autosummary_functions/mlx.nn.mish.rst", "python/nn/_autosummary_functions/mlx.nn.prelu.rst", "python/nn/_autosummary_functions/mlx.nn.relu.rst", "python/nn/_autosummary_functions/mlx.nn.relu6.rst", "python/nn/_autosummary_functions/mlx.nn.selu.rst", "python/nn/_autosummary_functions/mlx.nn.sigmoid.rst", "python/nn/_autosummary_functions/mlx.nn.silu.rst", "python/nn/_autosummary_functions/mlx.nn.softmax.rst", "python/nn/_autosummary_functions/mlx.nn.softmin.rst", "python/nn/_autosummary_functions/mlx.nn.softplus.rst", "python/nn/_autosummary_functions/mlx.nn.softshrink.rst", "python/nn/_autosummary_functions/mlx.nn.step.rst", "python/nn/_autosummary_functions/mlx.nn.tanh.rst", "python/nn/functions.rst", "python/nn/init.rst", "python/nn/layers.rst", "python/nn/losses.rst", "python/nn/module.rst", "python/ops.rst", "python/optimizers.rst", "python/optimizers/_autosummary/mlx.optimizers.AdaDelta.rst", "python/optimizers/_autosummary/mlx.optimizers.Adafactor.rst", "python/optimizers/_autosummary/mlx.optimizers.Adagrad.rst", "python/optimizers/_autosummary/mlx.optimizers.Adam.rst", "python/optimizers/_autosummary/mlx.optimizers.AdamW.rst", "python/optimizers/_autosummary/mlx.optimizers.Adamax.rst", "python/optimizers/_autosummary/mlx.optimizers.Lion.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.rst", "python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.rst", "python/optimizers/_autosummary/mlx.optimizers.RMSprop.rst", "python/optimizers/_autosummary/mlx.optimizers.SGD.rst", "python/optimizers/_autosummary/mlx.optimizers.cosine_decay.rst", "python/optimizers/_autosummary/mlx.optimizers.exponential_decay.rst", "python/optimizers/_autosummary/mlx.optimizers.join_schedules.rst", "python/optimizers/_autosummary/mlx.optimizers.linear_schedule.rst", "python/optimizers/_autosummary/mlx.optimizers.step_decay.rst", "python/optimizers/common_optimizers.rst", "python/optimizers/optimizer.rst", "python/optimizers/schedulers.rst", "python/random.rst", "python/transforms.rst", "python/tree_utils.rst", "usage/compile.rst", "usage/distributed.rst", "usage/function_transforms.rst", "usage/indexing.rst", "usage/lazy_evaluation.rst", "usage/numpy.rst", "usage/quick_start.rst", "usage/saving_and_loading.rst", "usage/unified_memory.rst", "usage/using_streams.rst"], "titles": ["Operations", "Custom Extensions in MLX", "Metal Debugger", "Linear Regression", "LLM inference", "Multi-Layer Perceptron", "MLX", "Build and Install", "mlx.core.Device", "mlx.core.Dtype", "mlx.core.DtypeCategory", "mlx.core.abs", "mlx.core.add", "mlx.core.addmm", "mlx.core.all", "mlx.core.allclose", "mlx.core.any", "mlx.core.arange", "mlx.core.arccos", "mlx.core.arccosh", "mlx.core.arcsin", "mlx.core.arcsinh", "mlx.core.arctan", "mlx.core.arctan2", "mlx.core.arctanh", "mlx.core.argmax", "mlx.core.argmin", "mlx.core.argpartition", "mlx.core.argsort", "mlx.core.array", "mlx.core.array.T", "mlx.core.array.abs", "mlx.core.array.all", "mlx.core.array.any", "mlx.core.array.argmax", "mlx.core.array.argmin", "mlx.core.array.astype", "mlx.core.array.at", "mlx.core.array.conj", "mlx.core.array.cos", "mlx.core.array.cummax", "mlx.core.array.cummin", "mlx.core.array.cumprod", "mlx.core.array.cumsum", "mlx.core.array.diag", "mlx.core.array.diagonal", "mlx.core.array.dtype", "mlx.core.array.exp", "mlx.core.array.flatten", "mlx.core.array.item", "mlx.core.array.itemsize", "mlx.core.array.log", "mlx.core.array.log10", "mlx.core.array.log1p", "mlx.core.array.log2", "mlx.core.array.logsumexp", "mlx.core.array.max", "mlx.core.array.mean", "mlx.core.array.min", "mlx.core.array.moveaxis", "mlx.core.array.nbytes", "mlx.core.array.ndim", "mlx.core.array.prod", "mlx.core.array.reciprocal", "mlx.core.array.reshape", "mlx.core.array.round", "mlx.core.array.rsqrt", "mlx.core.array.shape", "mlx.core.array.sin", "mlx.core.array.size", "mlx.core.array.split", "mlx.core.array.sqrt", "mlx.core.array.square", "mlx.core.array.squeeze", "mlx.core.array.sum", "mlx.core.array.swapaxes", "mlx.core.array.tolist", "mlx.core.array.transpose", "mlx.core.array.var", "mlx.core.array.view", "mlx.core.array_equal", "mlx.core.as_strided", "mlx.core.atleast_1d", "mlx.core.atleast_2d", "mlx.core.atleast_3d", "mlx.core.bitwise_and", "mlx.core.bitwise_or", "mlx.core.bitwise_xor", "mlx.core.block_masked_mm", "mlx.core.broadcast_to", "mlx.core.ceil", "mlx.core.clip", "mlx.core.compile", "mlx.core.concatenate", "mlx.core.conj", "mlx.core.conjugate", "mlx.core.conv1d", "mlx.core.conv2d", "mlx.core.conv_general", "mlx.core.convolve", "mlx.core.cos", "mlx.core.cosh", "mlx.core.cummax", "mlx.core.cummin", "mlx.core.cumprod", "mlx.core.cumsum", "mlx.core.custom_function", "mlx.core.default_device", "mlx.core.default_stream", "mlx.core.degrees", "mlx.core.dequantize", "mlx.core.diag", "mlx.core.diagonal", "mlx.core.disable_compile", "mlx.core.distributed.Group", "mlx.core.distributed.all_gather", "mlx.core.distributed.all_sum", "mlx.core.distributed.init", "mlx.core.distributed.is_available", "mlx.core.divide", "mlx.core.divmod", "mlx.core.einsum", "mlx.core.einsum_path", "mlx.core.enable_compile", "mlx.core.equal", "mlx.core.erf", "mlx.core.erfinv", "mlx.core.eval", "mlx.core.exp", "mlx.core.expand_dims", "mlx.core.expm1", "mlx.core.eye", "mlx.core.fast.layer_norm", "mlx.core.fast.rms_norm", "mlx.core.fast.rope", "mlx.core.fast.scaled_dot_product_attention", "mlx.core.fft.fft", "mlx.core.fft.fft2", "mlx.core.fft.fftn", "mlx.core.fft.ifft", "mlx.core.fft.ifft2", "mlx.core.fft.ifftn", "mlx.core.fft.irfft", "mlx.core.fft.irfft2", "mlx.core.fft.irfftn", "mlx.core.fft.rfft", "mlx.core.fft.rfft2", "mlx.core.fft.rfftn", "mlx.core.flatten", "mlx.core.floor", "mlx.core.floor_divide", "mlx.core.full", "mlx.core.gather_mm", "mlx.core.gather_qmm", "mlx.core.grad", "mlx.core.greater", "mlx.core.greater_equal", "mlx.core.hadamard_transform", "mlx.core.identity", "mlx.core.inner", "mlx.core.isclose", "mlx.core.isinf", "mlx.core.isnan", "mlx.core.isneginf", "mlx.core.isposinf", "mlx.core.issubdtype", "mlx.core.jvp", "mlx.core.left_shift", "mlx.core.less", "mlx.core.less_equal", "mlx.core.linalg.cholesky", "mlx.core.linalg.cholesky_inv", "mlx.core.linalg.inv", "mlx.core.linalg.norm", "mlx.core.linalg.qr", "mlx.core.linalg.svd", "mlx.core.linalg.tri_inv", "mlx.core.linspace", "mlx.core.load", "mlx.core.log", "mlx.core.log10", "mlx.core.log1p", "mlx.core.log2", "mlx.core.logaddexp", "mlx.core.logical_and", "mlx.core.logical_not", "mlx.core.logical_or", "mlx.core.logsumexp", "mlx.core.matmul", "mlx.core.max", "mlx.core.maximum", "mlx.core.mean", "mlx.core.meshgrid", "mlx.core.metal.clear_cache", "mlx.core.metal.device_info", "mlx.core.metal.get_active_memory", "mlx.core.metal.get_cache_memory", "mlx.core.metal.get_peak_memory", "mlx.core.metal.is_available", "mlx.core.metal.reset_peak_memory", "mlx.core.metal.set_cache_limit", "mlx.core.metal.set_memory_limit", "mlx.core.metal.start_capture", "mlx.core.metal.stop_capture", "mlx.core.min", "mlx.core.minimum", "mlx.core.moveaxis", "mlx.core.multiply", "mlx.core.nan_to_num", "mlx.core.negative", "mlx.core.new_stream", "mlx.core.not_equal", "mlx.core.ones", "mlx.core.ones_like", "mlx.core.outer", "mlx.core.pad", "mlx.core.partition", "mlx.core.power", "mlx.core.prod", "mlx.core.quantize", "mlx.core.quantized_matmul", "mlx.core.radians", "mlx.core.random.bernoulli", "mlx.core.random.categorical", "mlx.core.random.gumbel", "mlx.core.random.key", "mlx.core.random.laplace", "mlx.core.random.multivariate_normal", "mlx.core.random.normal", "mlx.core.random.randint", "mlx.core.random.seed", "mlx.core.random.split", "mlx.core.random.truncated_normal", "mlx.core.random.uniform", "mlx.core.reciprocal", "mlx.core.remainder", "mlx.core.repeat", "mlx.core.reshape", "mlx.core.right_shift", "mlx.core.round", "mlx.core.rsqrt", "mlx.core.save", "mlx.core.save_gguf", "mlx.core.save_safetensors", "mlx.core.savez", "mlx.core.savez_compressed", "mlx.core.set_default_device", "mlx.core.set_default_stream", "mlx.core.sigmoid", "mlx.core.sign", "mlx.core.sin", "mlx.core.sinh", "mlx.core.softmax", "mlx.core.sort", "mlx.core.split", "mlx.core.sqrt", "mlx.core.square", "mlx.core.squeeze", "mlx.core.stack", "mlx.core.std", "mlx.core.stop_gradient", "mlx.core.stream", "mlx.core.subtract", "mlx.core.sum", "mlx.core.swapaxes", "mlx.core.synchronize", "mlx.core.take", "mlx.core.take_along_axis", "mlx.core.tan", "mlx.core.tanh", "mlx.core.tensordot", "mlx.core.tile", "mlx.core.topk", "mlx.core.trace", "mlx.core.transpose", "mlx.core.tri", "mlx.core.tril", "mlx.core.triu", "mlx.core.value_and_grad", "mlx.core.var", "mlx.core.view", "mlx.core.vjp", "mlx.core.vmap", "mlx.core.where", "mlx.core.zeros", "mlx.core.zeros_like", "mlx.nn.quantize", "mlx.nn.value_and_grad", "mlx.optimizers.clip_grad_norm", "mlx.utils.tree_flatten", "mlx.utils.tree_map", "mlx.utils.tree_map_with_path", "mlx.utils.tree_reduce", "mlx.utils.tree_unflatten", "mlx.core.Stream", "Array", "Data Types", "Devices and Streams", "Distributed Communication", "Fast", "FFT", "Linear Algebra", "Metal", "Neural Networks", "mlx.nn.ALiBi", "mlx.nn.AvgPool1d", "mlx.nn.AvgPool2d", "mlx.nn.BatchNorm", "mlx.nn.Conv1d", "mlx.nn.Conv2d", "mlx.nn.Conv3d", "mlx.nn.Dropout", "mlx.nn.Dropout2d", "mlx.nn.Dropout3d", "mlx.nn.Embedding", "mlx.nn.GELU", "mlx.nn.GLU", "mlx.nn.GRU", "mlx.nn.GroupNorm", "mlx.nn.HardShrink", "mlx.nn.HardTanh", "mlx.nn.Hardswish", "mlx.nn.InstanceNorm", "mlx.nn.LSTM", "mlx.nn.LayerNorm", "mlx.nn.LeakyReLU", "mlx.nn.Linear", "mlx.nn.MaxPool1d", "mlx.nn.MaxPool2d", "mlx.nn.Mish", "mlx.nn.Module.apply", "mlx.nn.Module.apply_to_modules", "mlx.nn.Module.children", "mlx.nn.Module.eval", "mlx.nn.Module.filter_and_map", "mlx.nn.Module.freeze", "mlx.nn.Module.leaf_modules", "mlx.nn.Module.load_weights", "mlx.nn.Module.modules", "mlx.nn.Module.named_modules", "mlx.nn.Module.parameters", "mlx.nn.Module.save_weights", "mlx.nn.Module.set_dtype", "mlx.nn.Module.state", "mlx.nn.Module.train", "mlx.nn.Module.trainable_parameters", "mlx.nn.Module.training", "mlx.nn.Module.unfreeze", "mlx.nn.Module.update", "mlx.nn.Module.update_modules", "mlx.nn.MultiHeadAttention", "mlx.nn.PReLU", "mlx.nn.QuantizedEmbedding", "mlx.nn.QuantizedLinear", "mlx.nn.RMSNorm", "mlx.nn.RNN", "mlx.nn.ReLU", "mlx.nn.ReLU6", "mlx.nn.RoPE", "mlx.nn.SELU", "mlx.nn.Sequential", "mlx.nn.SiLU", "mlx.nn.SinusoidalPositionalEncoding", "mlx.nn.Softmax", "mlx.nn.Softmin", "mlx.nn.Softplus", "mlx.nn.Softshrink", "mlx.nn.Softsign", "mlx.nn.Step", "mlx.nn.Tanh", "mlx.nn.Transformer", "mlx.nn.Upsample", "mlx.nn.init.constant", "mlx.nn.init.glorot_normal", "mlx.nn.init.glorot_uniform", "mlx.nn.init.he_normal", "mlx.nn.init.he_uniform", "mlx.nn.init.identity", "mlx.nn.init.normal", "mlx.nn.init.uniform", "mlx.nn.elu", "mlx.nn.gelu", "mlx.nn.gelu_approx", "mlx.nn.gelu_fast_approx", "mlx.nn.glu", "mlx.nn.hard_shrink", "mlx.nn.hard_tanh", "mlx.nn.hardswish", "mlx.nn.leaky_relu", "mlx.nn.log_sigmoid", "mlx.nn.log_softmax", "mlx.nn.losses.binary_cross_entropy", "mlx.nn.losses.cosine_similarity_loss", "mlx.nn.losses.cross_entropy", "mlx.nn.losses.gaussian_nll_loss", "mlx.nn.losses.hinge_loss", "mlx.nn.losses.huber_loss", "mlx.nn.losses.kl_div_loss", "mlx.nn.losses.l1_loss", "mlx.nn.losses.log_cosh_loss", "mlx.nn.losses.margin_ranking_loss", "mlx.nn.losses.mse_loss", "mlx.nn.losses.nll_loss", "mlx.nn.losses.smooth_l1_loss", "mlx.nn.losses.triplet_loss", "mlx.nn.mish", "mlx.nn.prelu", "mlx.nn.relu", "mlx.nn.relu6", "mlx.nn.selu", "mlx.nn.sigmoid", "mlx.nn.silu", "mlx.nn.softmax", "mlx.nn.softmin", "mlx.nn.softplus", "mlx.nn.softshrink", "mlx.nn.step", "mlx.nn.tanh", "Functions", "Initializers", "Layers", "Loss Functions", "Module", "Operations", "Optimizers", "mlx.optimizers.AdaDelta", "mlx.optimizers.Adafactor", "mlx.optimizers.Adagrad", "mlx.optimizers.Adam", "mlx.optimizers.AdamW", "mlx.optimizers.Adamax", "mlx.optimizers.Lion", "mlx.optimizers.Optimizer.apply_gradients", "mlx.optimizers.Optimizer.init", "mlx.optimizers.Optimizer.state", "mlx.optimizers.Optimizer.update", "mlx.optimizers.RMSprop", "mlx.optimizers.SGD", "mlx.optimizers.cosine_decay", "mlx.optimizers.exponential_decay", "mlx.optimizers.join_schedules", "mlx.optimizers.linear_schedule", "mlx.optimizers.step_decay", "Common Optimizers", "Optimizer", "Schedulers", "Random", "Transforms", "Tree Utils", "Compilation", "Distributed Communication", "Function Transforms", "Indexing Arrays", "Lazy Evaluation", "Conversion to NumPy and Other Frameworks", "Quick Start Guide", "Saving and Loading Arrays", "Unified Memory", "Using Streams"], "terms": {"arrai": [0, 1, 4, 5, 6, 9, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 115, 116, 119, 120, 121, 122, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 204, 205, 206, 207, 208, 209, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 262, 263, 264, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 288, 303, 307, 323, 330, 337, 340, 345, 351, 371, 372, 373, 374, 375, 376, 377, 378, 379, 384, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 416, 419, 422, 425, 426, 427, 428, 429, 430, 431, 436, 437, 438, 439, 440, 441, 442, 449, 450, 451, 453, 454, 455, 457], "arang": [0, 173, 296, 371, 452, 454], "doubl": [0, 4], "start": [0, 1, 3, 4, 6, 7, 17, 134, 177, 202, 254, 292, 449, 452, 457], "stop": [0, 1, 4, 17, 177, 203, 260, 451, 452], "step": [0, 2, 4, 5, 17, 303, 317, 323, 355, 426, 433, 438, 440, 441, 442, 449, 450], "dtype": [0, 1, 4, 10, 17, 29, 36, 37, 76, 79, 131, 148, 151, 158, 165, 173, 174, 177, 212, 224, 226, 227, 228, 229, 232, 233, 273, 275, 280, 284, 296, 342, 371, 372, 373, 374, 375, 376, 377, 378, 379, 391, 393, 400, 438, 439, 440, 441, 442, 449, 450, 451, 452, 454, 455, 456], "streamordevic": [0, 1], "": [0, 1, 3, 4, 5, 46, 50, 61, 92, 108, 110, 137, 138, 140, 141, 143, 144, 146, 147, 154, 171, 173, 175, 178, 191, 214, 219, 223, 239, 242, 243, 259, 261, 278, 279, 280, 282, 287, 303, 306, 317, 323, 328, 334, 335, 337, 341, 342, 343, 347, 355, 424, 433, 434, 446, 449, 451, 453, 454, 455, 456, 457], "A": [0, 1, 4, 6, 7, 8, 67, 80, 92, 132, 133, 135, 154, 166, 171, 173, 174, 175, 178, 187, 188, 189, 194, 204, 219, 222, 223, 224, 226, 227, 228, 229, 232, 233, 254, 258, 261, 278, 281, 282, 286, 287, 288, 289, 290, 291, 292, 293, 294, 303, 307, 312, 317, 318, 322, 324, 334, 338, 339, 342, 348, 349, 354, 360, 362, 370, 373, 374, 376, 383, 404, 405, 422, 424, 428, 430, 432, 433, 435, 440, 449, 450, 451, 453, 454], "1d": [0, 96, 99, 242, 267], "number": [0, 1, 10, 17, 60, 69, 92, 97, 98, 110, 131, 153, 154, 158, 166, 177, 208, 215, 219, 220, 223, 226, 228, 231, 233, 236, 239, 270, 271, 275, 278, 281, 282, 286, 303, 307, 308, 309, 310, 312, 313, 318, 322, 350, 351, 370, 371, 373, 374, 375, 376, 438, 440, 441, 446, 449, 451, 458], "option": [0, 2, 4, 13, 14, 16, 17, 25, 26, 27, 28, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 81, 82, 83, 84, 88, 92, 93, 96, 97, 98, 99, 102, 103, 104, 105, 106, 110, 111, 112, 117, 131, 132, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 151, 152, 153, 154, 157, 158, 163, 164, 170, 171, 172, 173, 174, 175, 176, 177, 178, 187, 189, 191, 192, 201, 204, 208, 212, 215, 216, 218, 219, 220, 222, 223, 224, 226, 227, 228, 229, 231, 232, 233, 236, 237, 252, 253, 254, 257, 258, 259, 263, 265, 266, 270, 272, 273, 274, 275, 276, 277, 278, 279, 282, 284, 286, 289, 290, 291, 292, 305, 306, 307, 308, 309, 310, 317, 323, 326, 327, 328, 330, 334, 335, 337, 342, 347, 350, 352, 353, 355, 358, 362, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 425, 426, 427, 428, 429, 430, 431, 433, 436, 437, 438, 446, 449, 456, 458], "int": [0, 1, 4, 5, 8, 14, 16, 17, 25, 26, 27, 28, 32, 33, 34, 35, 40, 41, 42, 43, 44, 45, 48, 55, 56, 57, 58, 59, 62, 65, 67, 70, 73, 74, 75, 76, 78, 81, 88, 89, 93, 96, 97, 98, 102, 103, 104, 105, 110, 111, 112, 122, 129, 131, 134, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 151, 153, 154, 158, 165, 173, 177, 187, 189, 191, 194, 195, 196, 197, 200, 201, 204, 206, 212, 215, 216, 218, 219, 220, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 236, 237, 239, 252, 253, 254, 257, 258, 259, 263, 264, 266, 267, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 282, 284, 286, 294, 303, 305, 306, 307, 308, 309, 310, 314, 316, 317, 318, 322, 323, 324, 326, 327, 328, 350, 352, 353, 354, 355, 358, 362, 370, 384, 392, 393, 397, 402, 404, 422, 438, 440, 441, 442], "linspac": 0, "num": [0, 4, 177, 231], "50": [0, 177], "float32": [0, 1, 10, 17, 131, 135, 157, 158, 165, 173, 174, 177, 212, 224, 226, 227, 228, 232, 233, 275, 284, 296, 371, 372, 373, 374, 375, 376, 377, 378, 379, 391, 393, 400, 438, 439, 440, 441, 442, 449, 450, 451, 452, 453, 454, 455, 456], "evenli": [0, 177], "space": [0, 1, 177, 391, 402], "rang": [0, 1, 2, 3, 4, 5, 7, 17, 148, 152, 177, 374, 376, 382, 383, 424, 438, 439, 440, 441, 442, 446, 449, 451, 453, 457], "astyp": [0, 1, 4, 330, 454], "convert": [0, 1, 76, 82, 83, 84, 109, 148, 221, 352, 353, 453, 454, 455], "an": [0, 1, 2, 4, 5, 7, 9, 14, 16, 29, 82, 83, 84, 89, 96, 97, 98, 114, 127, 131, 132, 135, 148, 151, 158, 170, 173, 178, 201, 206, 212, 213, 215, 218, 219, 220, 236, 237, 239, 254, 257, 264, 266, 267, 270, 271, 275, 282, 284, 285, 289, 290, 291, 292, 303, 305, 306, 311, 315, 318, 323, 324, 326, 327, 328, 330, 350, 351, 353, 355, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 382, 406, 419, 424, 425, 435, 439, 444, 446, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458], "given": [0, 1, 7, 14, 16, 27, 37, 81, 89, 91, 93, 102, 103, 104, 105, 110, 112, 122, 127, 129, 136, 137, 138, 139, 140, 141, 145, 146, 147, 151, 152, 173, 187, 189, 191, 200, 204, 208, 210, 218, 227, 229, 239, 247, 252, 254, 259, 263, 265, 271, 272, 273, 275, 276, 277, 279, 294, 305, 306, 311, 327, 328, 334, 350, 392, 394, 400], "data": [0, 1, 5, 6, 9, 17, 131, 145, 146, 151, 158, 177, 208, 212, 232, 273, 275, 280, 284, 313, 372, 373, 374, 375, 376, 377, 378, 379, 449, 450, 452, 454], "type": [0, 1, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 36, 67, 76, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 115, 116, 117, 119, 120, 121, 122, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 194, 200, 201, 204, 205, 206, 207, 208, 209, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 262, 263, 264, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 288, 289, 292, 303, 342, 370, 372, 373, 374, 375, 376, 377, 378, 379, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 449, 452], "as_strid": 0, "std": [0, 1, 378], "vector": [0, 1, 3, 6, 159, 166, 173, 266, 281, 282, 314, 393, 455], "shape": [0, 1, 2, 4, 5, 64, 80, 81, 88, 89, 92, 96, 97, 98, 112, 115, 135, 136, 139, 142, 145, 146, 147, 151, 152, 157, 166, 172, 176, 188, 212, 213, 222, 223, 224, 226, 227, 228, 229, 232, 233, 237, 267, 280, 281, 283, 284, 285, 303, 305, 306, 307, 308, 309, 310, 312, 313, 317, 322, 323, 326, 327, 328, 337, 355, 372, 373, 374, 375, 376, 377, 378, 379, 393, 404, 424, 449, 451, 452, 455, 457], "size_t": [0, 1], "stride": [0, 1, 81, 96, 97, 98, 305, 306, 308, 309, 310, 327, 328, 358, 452], "offset": [0, 1, 4, 45, 81, 112, 132, 134, 273], "creat": [0, 1, 4, 7, 81, 117, 131, 158, 261, 303, 422, 424, 440, 449, 452, 454], "view": [0, 2, 81, 454], "copi": [0, 1, 4, 6, 216, 253, 454], "anoth": [0, 91, 165, 188, 262, 283, 296, 303, 330, 449, 451, 452, 457], "full": [0, 1, 5, 64, 77, 99, 252, 348, 349, 394, 449, 450, 453], "val": [0, 29, 151], "fill": [0, 1, 151, 213, 275, 285, 372, 373, 374, 375, 376, 378, 379], "valu": [0, 3, 4, 10, 11, 15, 17, 25, 26, 49, 76, 80, 91, 131, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 151, 160, 173, 175, 177, 194, 208, 215, 222, 223, 224, 226, 227, 228, 229, 232, 233, 242, 266, 267, 278, 282, 287, 289, 290, 291, 292, 296, 306, 311, 312, 313, 319, 322, 326, 328, 334, 350, 351, 366, 368, 370, 372, 391, 392, 393, 394, 395, 396, 398, 399, 400, 401, 402, 403, 416, 422, 426, 429, 438, 439, 441, 442, 451], "templat": [0, 1], "typenam": [0, 1], "t": [0, 1, 4, 7, 125, 135, 153, 170, 171, 220, 278, 303, 305, 317, 323, 327, 355, 425, 426, 427, 428, 429, 430, 431, 436, 437, 449, 451, 457], "zero": [0, 131, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 192, 199, 275, 276, 277, 285, 303, 305, 306, 311, 312, 313, 337, 372, 373, 374, 375, 376, 377, 378, 379, 419, 424, 426, 452], "const": [0, 1, 394], "inlin": 0, "zeros_lik": 0, "ones": [0, 1, 4, 213, 244, 275, 348, 349, 424, 450, 452], "ones_lik": 0, "ey": [0, 4, 172, 176], "n": [0, 1, 4, 29, 88, 96, 97, 98, 131, 136, 138, 139, 141, 142, 145, 147, 157, 158, 227, 259, 275, 279, 305, 306, 307, 308, 309, 310, 312, 313, 317, 323, 327, 328, 355, 371, 399, 404, 450], "m": [0, 1, 4, 7, 88, 131, 152, 157, 173, 275, 305, 306, 327, 328, 425, 449], "k": [0, 4, 44, 88, 111, 131, 135, 152, 157, 272, 275, 276, 277, 305, 326, 327, 335], "specifi": [0, 1, 17, 36, 97, 98, 112, 143, 144, 151, 154, 173, 177, 206, 212, 223, 236, 264, 266, 267, 270, 273, 274, 278, 282, 284, 307, 368, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 416, 450, 451, 457], "diagon": [0, 44, 111, 131, 273, 275, 276, 277], "everywher": 0, "els": [0, 1, 4, 303, 335, 450, 453], "ident": [0, 106, 131, 260, 303, 344], "squar": [0, 3, 4, 133, 158, 172, 176, 240, 255, 278, 290, 303, 354, 401, 403, 425, 426, 428, 429, 430, 451, 454], "matrix": [0, 3, 13, 44, 88, 110, 111, 131, 152, 153, 157, 158, 170, 171, 172, 173, 174, 175, 176, 188, 192, 219, 220, 227, 352, 353, 377, 419], "major": [0, 1], "tri": 0, "tril": 0, "x": [0, 1, 3, 4, 5, 37, 88, 106, 115, 116, 125, 130, 132, 133, 153, 157, 158, 173, 220, 224, 239, 244, 248, 276, 277, 283, 290, 292, 303, 305, 306, 307, 315, 316, 318, 322, 324, 325, 326, 327, 328, 329, 330, 351, 354, 356, 362, 368, 371, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 403, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 422, 424, 431, 449, 450, 451, 452, 453, 454, 455, 457], "0": [0, 1, 3, 4, 5, 7, 8, 13, 17, 37, 44, 45, 48, 65, 70, 78, 81, 93, 96, 97, 98, 111, 112, 131, 135, 148, 152, 154, 172, 173, 174, 176, 193, 200, 208, 215, 222, 226, 228, 233, 236, 239, 254, 258, 259, 273, 275, 276, 277, 278, 279, 282, 288, 289, 291, 292, 303, 305, 306, 307, 308, 309, 310, 311, 312, 313, 315, 318, 319, 322, 324, 325, 327, 328, 351, 356, 358, 362, 366, 368, 370, 372, 373, 374, 375, 376, 377, 378, 379, 380, 382, 383, 385, 386, 387, 388, 391, 393, 395, 396, 400, 403, 404, 406, 407, 408, 409, 415, 416, 419, 422, 425, 426, 428, 429, 430, 431, 433, 436, 437, 438, 439, 440, 441, 442, 446, 449, 450, 451, 452, 453, 454, 455, 456], "triu": 0, "reshap": [0, 4, 173, 371, 452], "flatten": [0, 27, 28, 102, 103, 104, 105, 173, 214, 216, 236, 253, 266, 267, 272, 289], "start_axi": [0, 48, 148], "end_axi": [0, 48, 148], "1": [0, 1, 2, 4, 5, 13, 17, 27, 28, 37, 45, 48, 96, 97, 98, 111, 112, 130, 135, 136, 137, 139, 140, 142, 143, 144, 145, 146, 147, 148, 157, 159, 165, 171, 173, 174, 188, 192, 201, 214, 216, 219, 223, 226, 227, 228, 233, 248, 253, 266, 272, 273, 278, 288, 291, 292, 296, 303, 305, 306, 307, 308, 309, 310, 311, 312, 313, 315, 316, 317, 318, 322, 323, 324, 326, 327, 328, 351, 354, 355, 358, 362, 368, 371, 373, 374, 375, 376, 377, 378, 379, 380, 382, 383, 384, 386, 389, 390, 391, 392, 393, 394, 395, 396, 397, 399, 400, 402, 403, 404, 409, 410, 412, 413, 414, 416, 419, 422, 424, 425, 426, 427, 428, 429, 430, 431, 433, 436, 437, 438, 439, 440, 441, 442, 449, 450, 451, 452, 454, 455, 456, 457], "dimens": [0, 1, 4, 14, 16, 25, 26, 61, 67, 76, 82, 83, 84, 92, 97, 98, 112, 129, 134, 143, 144, 146, 147, 148, 152, 153, 159, 170, 171, 172, 173, 174, 175, 176, 187, 188, 189, 191, 204, 218, 219, 223, 231, 259, 263, 267, 270, 274, 279, 307, 308, 309, 310, 312, 313, 316, 317, 318, 322, 323, 324, 350, 354, 355, 358, 370, 371, 384, 393, 449, 451], "hadamard_transform": 0, "float": [0, 1, 10, 13, 15, 17, 76, 132, 133, 134, 135, 150, 151, 157, 160, 165, 173, 208, 220, 222, 226, 228, 288, 296, 307, 311, 312, 313, 318, 322, 324, 330, 342, 354, 358, 362, 368, 370, 371, 372, 373, 374, 375, 376, 378, 379, 392, 393, 394, 396, 400, 403, 404, 415, 416, 425, 426, 427, 428, 429, 430, 431, 436, 437, 438, 439, 441, 442], "scale": [0, 1, 4, 13, 110, 132, 133, 134, 135, 153, 157, 219, 220, 226, 228, 288, 312, 313, 324, 350, 358, 359, 362, 371, 409, 426], "nullopt": 0, "multipli": [0, 1, 37, 153, 219, 220, 311, 362, 371], "hadamard": [0, 157], "correspond": [0, 1, 14, 16, 76, 91, 110, 112, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 187, 189, 204, 218, 263, 270, 282, 290, 451], "size": [0, 1, 4, 5, 50, 67, 88, 97, 110, 129, 132, 133, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 151, 153, 157, 158, 165, 173, 196, 201, 219, 220, 223, 237, 254, 257, 280, 286, 303, 305, 306, 308, 309, 310, 314, 322, 327, 328, 352, 353, 371, 426, 450, 453, 454], "squeez": [0, 371, 449], "ax": [0, 1, 14, 16, 25, 26, 77, 106, 129, 137, 138, 140, 141, 143, 144, 146, 147, 148, 159, 173, 187, 189, 191, 204, 215, 218, 252, 257, 259, 263, 264, 270, 274, 279, 451], "remov": [0, 112, 188, 223, 257, 393], "singleton": [0, 14, 16, 25, 26, 117, 187, 188, 189, 191, 204, 218, 259, 263, 279], "axi": [0, 1, 4, 5, 14, 16, 25, 26, 27, 28, 32, 33, 34, 35, 40, 41, 42, 43, 55, 56, 57, 58, 62, 70, 73, 74, 78, 93, 102, 103, 104, 105, 112, 115, 129, 132, 133, 136, 139, 142, 143, 144, 145, 146, 147, 148, 157, 173, 187, 189, 191, 204, 206, 215, 216, 218, 223, 236, 252, 253, 254, 257, 258, 259, 263, 264, 266, 267, 271, 272, 273, 274, 279, 280, 282, 305, 306, 316, 327, 328, 355, 384, 390, 392, 393, 397, 402, 404, 412, 413, 452], "all": [0, 1, 2, 5, 7, 15, 27, 37, 82, 83, 84, 92, 97, 98, 106, 115, 116, 117, 131, 138, 141, 144, 147, 152, 153, 175, 188, 215, 216, 257, 286, 303, 330, 331, 335, 338, 339, 340, 345, 347, 350, 362, 370, 371, 419, 422, 444, 446, 449, 452, 453, 455, 458], "expand_dim": 0, "add": [0, 1, 2, 4, 13, 37, 129, 183, 215, 219, 308, 309, 310, 451, 457], "slice": [0, 452], "each": [0, 1, 67, 110, 127, 134, 153, 165, 170, 171, 172, 175, 176, 188, 192, 215, 219, 220, 223, 236, 244, 245, 254, 271, 274, 280, 282, 283, 312, 313, 314, 317, 318, 323, 355, 358, 370, 391, 393, 446, 449, 450, 453], "slice_upd": 0, "src": 0, "updat": [0, 1, 3, 4, 5, 37, 92, 286, 290, 292, 307, 330, 331, 337, 342, 343, 344, 349, 424, 426, 429, 431, 432, 433, 437, 438, 439, 440, 441, 442, 449, 450, 453], "from": [0, 1, 4, 5, 6, 81, 109, 110, 112, 115, 116, 143, 144, 146, 147, 151, 152, 157, 173, 178, 188, 192, 197, 200, 213, 219, 221, 222, 223, 224, 225, 226, 229, 232, 244, 257, 260, 262, 266, 267, 272, 273, 283, 285, 289, 290, 291, 292, 293, 303, 326, 335, 337, 350, 373, 374, 375, 376, 378, 379, 394, 403, 419, 424, 448, 449, 450, 451, 453, 454, 455, 456, 457], "sourc": [0, 1, 2, 59, 206, 274, 450], "split": [0, 316, 318, 384], "num_split": 0, "sub": [0, 5, 112, 231, 273, 286], "along": [0, 1, 25, 26, 92, 93, 102, 103, 104, 105, 115, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 152, 153, 157, 173, 236, 252, 254, 258, 266, 267, 270, 271, 272, 273, 280, 303, 316, 355, 384], "indic": [0, 1, 15, 25, 26, 27, 28, 37, 152, 153, 154, 160, 161, 162, 163, 164, 175, 254, 266, 267, 278, 344, 346, 393, 400, 440, 452], "meshgrid": 0, "bool": [0, 1, 14, 15, 16, 25, 26, 32, 33, 34, 35, 40, 41, 42, 43, 55, 56, 57, 58, 62, 74, 76, 78, 80, 92, 98, 102, 103, 104, 105, 117, 118, 134, 153, 160, 165, 170, 171, 173, 176, 178, 187, 189, 191, 192, 198, 201, 204, 218, 220, 259, 263, 279, 307, 308, 309, 310, 317, 318, 322, 323, 324, 326, 330, 334, 335, 337, 342, 344, 347, 350, 353, 355, 358, 362, 370, 371, 391, 394, 426, 437], "spars": [0, 192], "fals": [0, 1, 4, 14, 15, 16, 25, 26, 32, 33, 34, 35, 40, 41, 42, 43, 55, 56, 57, 58, 62, 74, 78, 80, 92, 98, 102, 103, 104, 105, 117, 160, 165, 170, 171, 173, 176, 178, 187, 189, 191, 192, 201, 204, 218, 259, 263, 279, 283, 286, 289, 290, 291, 292, 296, 318, 322, 324, 326, 335, 337, 347, 350, 353, 358, 362, 370, 371, 391, 394, 426, 437, 454], "string": [0, 1, 122, 194, 215, 454, 456], "index": [0, 1, 6, 8, 27, 37, 129, 131, 154, 192, 216, 266, 267, 278, 294], "xy": [0, 192], "coordin": [0, 192], "clip": [0, 288, 391, 426], "a_min": [0, 91], "a_max": [0, 91], "limit": [0, 1, 91, 200, 201, 452], "concaten": [0, 4, 115], "stack": [0, 449], "new": [0, 1, 5, 89, 112, 206, 210, 237, 258, 274, 280, 290, 291, 342, 350, 422, 424, 435, 440, 449, 452, 453, 454], "repeat": [0, 271], "arr": [0, 241, 452], "tile": [0, 135], "rep": [0, 271], "transpos": [0, 4, 30, 153, 220], "permut": [0, 5], "accord": [0, 224, 283, 286, 350, 373, 374, 375, 376], "initializer_list": 0, "swapax": [0, 106], "axis1": [0, 45, 75, 112, 264, 273], "axis2": [0, 45, 75, 112, 264, 273], "swap": [0, 99, 201, 264, 349], "two": [0, 1, 12, 13, 15, 23, 80, 83, 85, 86, 87, 88, 112, 119, 124, 137, 140, 146, 152, 153, 155, 156, 160, 168, 169, 170, 171, 172, 174, 175, 176, 183, 188, 190, 205, 207, 211, 214, 264, 292, 306, 316, 323, 328, 384, 392, 449, 450, 451, 452, 457], "moveaxi": 0, "destin": [0, 1, 59, 206], "move": [0, 1, 206, 457], "pad": [0, 96, 97, 98, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 305, 306, 308, 309, 310, 327, 328], "low_pad_s": 0, "high_pad_s": 0, "pad_valu": 0, "mode": [0, 1, 99, 215, 333, 344, 346, 371, 375, 376, 450], "constant": [0, 1, 4, 7, 132, 133, 215, 303, 307, 318, 324, 354, 394, 404, 436, 438, 449, 454], "pair": [0, 1, 215, 337, 358], "pad_width": [0, 215], "revers": [0, 1, 40, 41, 42, 43, 81, 102, 103, 104, 105, 274, 362], "order": [0, 27, 81, 98, 122, 173, 216, 219, 272, 303, 318, 348, 360, 433, 449, 451], "broadcast_to": 0, "broadcast": [0, 1, 12, 15, 85, 86, 87, 89, 91, 119, 120, 124, 151, 155, 156, 160, 167, 168, 169, 183, 188, 190, 205, 207, 211, 217, 222, 223, 227, 232, 233, 235, 238, 262, 267, 283, 350], "broadcast_arrai": [0, 1], "input": [0, 1, 3, 4, 11, 12, 13, 14, 15, 16, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 109, 111, 112, 115, 116, 119, 120, 121, 122, 124, 125, 126, 128, 129, 130, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 152, 153, 154, 155, 156, 157, 159, 160, 161, 162, 163, 164, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 204, 205, 206, 207, 208, 209, 211, 213, 214, 215, 216, 217, 218, 219, 220, 221, 231, 234, 235, 236, 237, 238, 239, 240, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 262, 263, 264, 266, 267, 268, 269, 270, 271, 272, 273, 274, 276, 277, 278, 279, 280, 282, 283, 285, 305, 306, 307, 308, 309, 310, 312, 313, 314, 316, 317, 318, 322, 323, 324, 326, 327, 328, 350, 353, 354, 355, 358, 368, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 384, 391, 392, 394, 395, 396, 397, 399, 400, 402, 404, 416, 419, 449, 451, 452, 455, 456], "against": 0, "one": [0, 1, 4, 7, 37, 76, 82, 91, 97, 98, 129, 131, 132, 133, 173, 181, 188, 220, 223, 257, 262, 296, 347, 371, 393, 450, 457], "equal": [0, 15, 27, 80, 131, 156, 160, 169, 211, 216, 229, 254, 322, 326], "b": [0, 1, 2, 4, 12, 13, 15, 23, 80, 85, 86, 87, 88, 119, 120, 124, 150, 152, 155, 156, 159, 160, 167, 168, 169, 173, 183, 184, 186, 188, 190, 205, 207, 211, 214, 217, 219, 226, 235, 238, 262, 270, 278, 291, 292, 316, 326, 355, 371, 384, 451, 452, 453, 454, 455, 456, 457], "return": [0, 1, 3, 4, 5, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 36, 49, 67, 76, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 109, 110, 111, 112, 115, 116, 117, 119, 120, 121, 122, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 196, 200, 201, 204, 205, 206, 207, 208, 209, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 303, 317, 323, 330, 331, 332, 334, 335, 336, 337, 338, 339, 340, 344, 345, 347, 348, 349, 355, 372, 373, 374, 375, 376, 377, 378, 379, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 419, 422, 432, 448, 449, 450, 451, 452, 453, 454, 456, 457], "element": [0, 1, 11, 12, 18, 19, 20, 21, 22, 23, 24, 27, 69, 81, 85, 86, 87, 90, 100, 101, 102, 103, 104, 105, 110, 119, 120, 124, 125, 126, 128, 130, 131, 149, 150, 153, 155, 156, 160, 161, 162, 163, 164, 167, 168, 169, 179, 180, 181, 182, 183, 184, 185, 186, 190, 192, 205, 207, 209, 211, 216, 217, 219, 220, 234, 235, 236, 238, 240, 248, 249, 250, 251, 255, 256, 262, 266, 268, 269, 272, 278, 280, 283, 311, 312, 313, 317, 321, 323, 329, 351, 355, 358, 386, 387, 389, 390, 405, 406, 408, 411, 412, 413, 414, 449, 451], "wise": [0, 1, 11, 12, 18, 19, 20, 21, 22, 23, 24, 85, 86, 87, 90, 100, 101, 119, 120, 124, 125, 126, 128, 130, 149, 150, 155, 156, 160, 167, 168, 169, 179, 180, 181, 182, 183, 184, 185, 186, 190, 205, 207, 209, 211, 217, 234, 235, 238, 240, 248, 249, 250, 251, 255, 256, 262, 268, 269, 312, 313, 321, 329, 351, 386, 387, 389, 390, 405, 406, 408, 411, 412, 413, 414, 449], "not_equ": 0, "greater": [0, 4, 27, 130, 156, 216, 288, 368, 416], "greater_equ": 0, "less": [0, 4, 27, 169, 216, 358, 403], "less_equ": 0, "array_equ": [0, 15, 160], "equal_nan": [0, 15, 80, 160], "true": [0, 1, 3, 4, 15, 40, 41, 42, 43, 80, 92, 102, 103, 104, 105, 134, 153, 160, 165, 170, 171, 173, 178, 192, 201, 220, 252, 283, 286, 289, 290, 291, 292, 296, 303, 307, 308, 309, 310, 317, 318, 322, 323, 324, 326, 334, 335, 337, 344, 347, 353, 355, 358, 362, 370, 371, 391, 399, 426], "have": [0, 1, 4, 7, 15, 80, 82, 83, 84, 88, 115, 143, 144, 146, 147, 153, 160, 188, 202, 223, 280, 289, 323, 350, 360, 431, 433, 448, 449, 450, 452, 453, 457], "same": [0, 1, 4, 7, 15, 37, 80, 89, 92, 97, 98, 99, 115, 132, 133, 142, 145, 146, 147, 153, 154, 160, 166, 215, 223, 239, 280, 281, 283, 291, 303, 306, 307, 311, 318, 322, 328, 352, 372, 373, 374, 375, 376, 377, 378, 379, 393, 404, 422, 432, 446, 449, 450, 452, 457], "isnan": 0, "isinf": 0, "isposinf": 0, "isneginf": 0, "where": [0, 5, 131, 160, 171, 219, 278, 282, 305, 306, 307, 308, 309, 310, 311, 312, 313, 315, 317, 318, 322, 323, 324, 326, 327, 328, 334, 351, 354, 355, 368, 375, 376, 380, 381, 383, 394, 400, 406, 409, 411, 416, 433, 450, 451, 452], "condit": [0, 1, 283, 457], "y": [0, 1, 3, 4, 5, 37, 106, 157, 283, 303, 307, 312, 318, 322, 324, 326, 354, 395, 400, 403, 424, 427, 449, 450, 451, 453, 454], "select": [0, 2, 7, 272, 283, 330, 334, 342], "depend": [0, 1, 2, 3, 7, 76, 173, 317, 323, 355, 450, 452, 456, 457], "nan_to_num": 0, "nan": [0, 15, 80, 160, 162, 208], "0f": 0, "posinf": [0, 208], "neginf": [0, 208], "replac": [0, 4, 208, 348, 349, 370, 403], "infin": [0, 161, 163, 164, 208, 327, 328, 430], "finit": [0, 208], "keepdim": [0, 14, 16, 25, 26, 32, 33, 34, 35, 55, 56, 57, 58, 62, 74, 78, 173, 187, 189, 191, 204, 218, 252, 259, 263, 279], "ar": [0, 1, 3, 4, 5, 6, 7, 15, 17, 80, 88, 89, 91, 92, 98, 99, 106, 112, 127, 131, 137, 138, 140, 141, 143, 144, 146, 147, 148, 153, 154, 160, 161, 162, 163, 164, 165, 166, 173, 174, 178, 188, 201, 214, 215, 216, 219, 220, 222, 223, 224, 229, 232, 233, 244, 245, 257, 258, 266, 278, 281, 282, 286, 289, 290, 296, 307, 308, 309, 310, 311, 312, 313, 318, 322, 324, 326, 337, 350, 353, 371, 391, 393, 394, 418, 422, 424, 431, 433, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457], "non": [0, 1, 7, 192, 345, 355, 405, 422], "allclos": 0, "rtol": [0, 15, 160], "1e": [0, 3, 5, 15, 160, 307, 318, 322, 324, 325, 354, 392, 394, 404, 424, 425, 426, 427, 428, 429, 430, 433, 436, 438, 439, 440, 441, 442], "5": [0, 1, 3, 4, 7, 173, 201, 222, 292, 305, 307, 311, 312, 313, 315, 319, 322, 327, 366, 371, 372, 375, 376, 382, 385, 403, 415, 419, 424, 436, 438, 439, 449, 451, 452], "atol": [0, 15, 160], "8": [0, 1, 4, 7, 173, 219, 296, 306, 322, 328, 370, 392, 425, 426, 427, 428, 429, 430, 436, 449, 452, 455, 457], "within": [0, 2, 27, 160], "toler": [0, 15, 160], "isclos": 0, "boolean": [0, 15, 80, 160, 161, 162, 163, 164, 184, 185, 186, 296, 346, 452], "reduc": [0, 7, 14, 16, 25, 26, 116, 187, 189, 191, 204, 218, 259, 263, 279, 292, 307, 370, 399], "output": [0, 1, 4, 7, 14, 15, 16, 17, 27, 81, 88, 89, 92, 94, 95, 102, 103, 104, 105, 106, 121, 131, 132, 133, 134, 135, 142, 145, 146, 147, 151, 152, 154, 157, 158, 160, 173, 177, 187, 189, 191, 192, 204, 208, 212, 213, 216, 218, 222, 223, 224, 226, 227, 228, 229, 232, 233, 244, 245, 252, 257, 259, 263, 267, 273, 275, 278, 279, 280, 281, 282, 283, 284, 285, 305, 306, 307, 308, 309, 310, 322, 326, 327, 328, 350, 353, 368, 370, 371, 373, 374, 375, 376, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 416, 419, 449, 450, 451, 452, 453, 454, 455, 456, 457], "i": [0, 1, 2, 4, 5, 6, 7, 15, 17, 27, 36, 76, 81, 91, 97, 98, 99, 102, 103, 104, 105, 106, 111, 112, 115, 116, 118, 120, 127, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 150, 151, 152, 153, 157, 160, 165, 166, 170, 171, 172, 173, 174, 175, 176, 178, 183, 187, 188, 192, 198, 201, 215, 216, 219, 220, 227, 228, 237, 241, 242, 243, 248, 252, 254, 259, 260, 265, 266, 267, 270, 273, 274, 278, 279, 280, 281, 282, 283, 286, 288, 289, 290, 291, 292, 296, 298, 303, 305, 306, 307, 308, 309, 310, 311, 312, 313, 315, 317, 318, 322, 323, 324, 326, 327, 328, 334, 335, 341, 343, 344, 346, 347, 349, 350, 351, 353, 354, 355, 358, 362, 368, 370, 371, 375, 376, 381, 383, 391, 392, 394, 399, 400, 403, 404, 406, 411, 416, 422, 424, 426, 429, 431, 432, 433, 438, 440, 441, 446, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458], "ani": [0, 1, 4, 6, 17, 92, 289, 290, 291, 292, 293, 303, 315, 330, 331, 334, 343, 353, 370, 371, 419, 441, 448, 449, 451, 453, 455, 456, 457], "sum": [0, 1, 3, 12, 105, 116, 159, 173, 187, 252, 270, 273, 303, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 450, 452, 454], "mean": [0, 3, 4, 5, 133, 226, 227, 228, 278, 303, 307, 318, 335, 354, 378, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 449, 451, 454], "comput": [0, 1, 3, 4, 5, 6, 7, 102, 103, 104, 105, 106, 110, 122, 130, 134, 154, 166, 170, 171, 172, 173, 176, 183, 191, 214, 219, 235, 252, 259, 260, 270, 278, 279, 281, 287, 303, 307, 317, 318, 322, 323, 324, 335, 348, 353, 354, 358, 370, 373, 374, 375, 376, 382, 383, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 424, 425, 426, 428, 429, 430, 431, 435, 449, 450, 451, 455, 457], "var": [0, 307, 318, 322, 324, 394], "ddof": [0, 78, 259, 279], "varianc": [0, 259, 279, 307, 318, 394], "standard": [0, 5, 49, 76, 188, 224, 228, 259, 370, 373, 375, 378, 450, 455], "deviat": [0, 228, 259, 373, 375, 378], "deviatoin": 0, "prod": 0, "The": [0, 1, 2, 4, 5, 6, 7, 11, 12, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 36, 46, 50, 60, 61, 67, 76, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 115, 116, 117, 119, 120, 121, 122, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 166, 167, 168, 169, 173, 174, 175, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 196, 197, 200, 201, 202, 204, 205, 206, 207, 209, 211, 212, 213, 214, 215, 216, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 242, 243, 248, 249, 250, 251, 252, 253, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 296, 298, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 316, 317, 318, 322, 323, 324, 326, 327, 328, 330, 331, 335, 337, 341, 342, 343, 344, 347, 348, 349, 350, 352, 353, 354, 355, 358, 360, 362, 368, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 384, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 416, 419, 422, 424, 425, 426, 427, 428, 429, 430, 431, 434, 436, 437, 438, 441, 444, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458], "product": [0, 1, 13, 81, 104, 159, 166, 188, 214, 218, 270, 281, 350, 455], "max": [0, 1, 173, 190, 327, 328, 351, 386, 387, 392, 394, 395, 400, 404, 406, 408, 426, 430, 449, 451, 457], "maximum": [0, 5, 25, 37, 91, 102, 197, 201, 288, 303, 325, 356, 362, 382, 383, 388, 407, 422, 453], "min": [0, 1, 173, 205, 351, 386, 387, 406, 408], "minimum": [0, 26, 37, 91, 103, 362, 391, 392], "argmin": 0, "argmax": [0, 5], "sort": [0, 27, 28, 216, 272], "argsort": 0, "partit": [0, 27], "kth": [0, 27, 216], "smaller": [0, 7, 216, 431, 449], "first": [0, 1, 2, 3, 4, 5, 7, 112, 115, 148, 154, 167, 175, 184, 186, 188, 216, 231, 238, 264, 270, 273, 278, 289, 291, 292, 303, 306, 318, 328, 371, 392, 400, 426, 428, 429, 430, 433, 449, 451, 454, 457], "argpartit": 0, "topk": 0, "logsumexp": 0, "ab": [0, 15, 160, 173, 278, 318, 322, 324, 329, 354, 383, 405, 449], "absolut": [0, 11, 15, 160, 382, 383, 403], "neg": [0, 112, 148, 163, 208, 273, 325, 327, 328, 350, 394, 402, 404, 452], "negat": [0, 209], "sign": [0, 15, 160, 296, 431], "logical_not": 0, "logic": [0, 1, 184, 185, 186], "logical_and": 0, "logical_or": 0, "reciproc": [0, 240], "subtract": [0, 37], "divid": [0, 1, 37, 150, 219, 235, 450], "divmod": 0, "quotient": [0, 119, 120, 150], "remaind": [0, 120], "floor_divid": 0, "integ": [0, 10, 150, 152, 153, 165, 173, 194, 215, 219, 220, 222, 229, 254, 270, 282, 296, 314, 342, 440, 452], "divis": [0, 119, 150, 219, 235], "equival": [0, 1, 30, 64, 77, 120, 150, 153, 157, 266, 315, 319, 320, 321, 349, 351, 353, 356, 357, 359, 361, 363, 364, 365, 366, 367, 369], "do": [0, 1, 4, 7, 280, 303, 336, 347, 419, 422, 429, 449, 450, 451, 453], "floor": [0, 150], "between": [0, 1, 6, 91, 148, 370, 392, 395, 396, 399, 440, 450, 453, 454, 457], "ceil": 0, "exp": [0, 130, 183, 187, 224, 252, 380, 397, 409, 410, 414, 449, 457], "exponenti": [0, 128, 130, 359, 380, 409, 439], "sin": [0, 106, 362, 451, 455], "sine": [0, 20, 21, 250, 251, 451], "co": [0, 1, 106, 362, 451], "cosin": [0, 18, 19, 100, 101, 392, 438, 440, 451], "tan": 0, "tangent": [0, 1, 22, 23, 24, 106, 166, 268, 269, 369, 417], "arcsin": 0, "arc": 0, "arcco": 0, "arctan": 0, "arctan2": 0, "invers": [0, 18, 19, 20, 21, 22, 23, 24, 126, 139, 140, 141, 142, 143, 144, 171, 172, 176], "ratio": [0, 23], "sinh": 0, "hyperbol": [0, 19, 21, 24, 101, 251, 269, 369, 417], "cosh": [0, 399], "tanh": [0, 303, 315, 317, 323, 329, 355, 382, 405], "arcsinh": 0, "arccosh": 0, "arctanh": 0, "degre": [0, 221, 404], "radian": [0, 109], "log": [0, 181, 183, 187, 389, 390, 391, 394, 397, 399, 402, 414], "natur": [0, 179, 181, 453], "logarithm": [0, 179, 180, 181, 182], "log2": 0, "base": [0, 1, 134, 173, 180, 182, 217, 358, 370, 422, 424, 430, 444, 446, 449, 452], "2": [0, 1, 3, 4, 5, 37, 97, 111, 112, 125, 137, 140, 142, 143, 144, 145, 146, 147, 148, 157, 165, 170, 171, 172, 173, 174, 175, 176, 182, 188, 219, 227, 231, 270, 273, 275, 276, 277, 288, 292, 296, 303, 305, 306, 309, 315, 325, 327, 328, 354, 362, 371, 372, 373, 374, 375, 376, 377, 378, 379, 382, 393, 394, 396, 403, 404, 419, 422, 424, 425, 427, 428, 429, 433, 436, 449, 450, 451, 452, 453, 454, 455, 456, 457], "log10": 0, "10": [0, 2, 4, 5, 180, 239, 244, 290, 303, 337, 419, 440, 442, 449, 450, 452], "log1p": 0, "plu": [0, 181], "logaddexp": 0, "sigmoid": [0, 4, 303, 361, 383, 389, 391, 411], "logist": [0, 3, 248, 383, 411], "erf": [0, 126, 449], "error": [0, 1, 7, 117, 125, 126, 201, 254, 315, 381, 382, 383, 399, 401, 451, 454], "function": [0, 1, 2, 3, 4, 5, 6, 15, 81, 92, 106, 120, 125, 126, 154, 160, 166, 170, 171, 172, 173, 174, 175, 176, 188, 248, 278, 281, 282, 287, 288, 290, 291, 292, 303, 315, 316, 319, 320, 321, 329, 331, 335, 342, 347, 351, 355, 356, 357, 359, 360, 361, 363, 364, 365, 366, 367, 368, 369, 370, 381, 382, 383, 384, 385, 386, 387, 389, 390, 391, 405, 410, 412, 413, 414, 415, 416, 417, 419, 424, 433, 446, 448, 450, 452, 453, 454, 456], "erfinv": 0, "expm1": 0, "stop_gradi": [0, 451], "flow": [0, 260, 453], "gradient": [0, 3, 5, 106, 154, 260, 278, 287, 288, 303, 335, 348, 353, 370, 399, 422, 424, 425, 426, 428, 429, 430, 431, 432, 435, 437, 449, 450, 451, 452, 453, 454, 455], "round": [0, 219], "decim": [0, 65, 239], "point": [0, 1, 3, 4, 7, 81, 150, 220, 296], "matmul": [0, 152, 457], "multipl": [0, 7, 13, 88, 132, 133, 152, 153, 188, 207, 219, 220, 350, 362, 439, 440, 442, 449, 453, 456], "gather": [0, 115, 152, 153], "slice_s": 0, "entri": [0, 312, 313], "take": [0, 1, 4, 5, 85, 86, 87, 92, 152, 154, 166, 190, 205, 213, 220, 267, 278, 281, 282, 285, 291, 292, 350, 391, 446, 450, 451, 452, 456, 457, 458], "treat": [0, 1, 143, 144, 146, 147, 266, 371, 449], "take_along_axi": [0, 452], "scatter": 0, "paramet": [0, 1, 3, 4, 5, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 36, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 109, 110, 111, 112, 115, 116, 117, 119, 120, 121, 122, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 200, 201, 202, 204, 205, 206, 207, 208, 209, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 322, 323, 324, 325, 326, 327, 328, 330, 331, 334, 335, 337, 342, 343, 344, 347, 348, 349, 350, 351, 352, 353, 354, 355, 358, 360, 362, 366, 368, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 384, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 416, 418, 419, 422, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 435, 436, 437, 438, 439, 440, 441, 442, 444, 449, 450, 451, 453], "determin": [0, 1, 112, 227, 292, 296, 341, 456], "locat": [0, 1, 81, 348, 349, 450, 457], "assum": [0, 1, 4, 88, 174, 290, 303, 305, 306, 318, 327, 328], "d": [0, 1, 4, 111, 112, 159, 173, 188, 192, 214, 266, 273, 275, 276, 277, 293, 310, 313, 317, 323, 355, 425, 428, 430, 457], "simplic": 0, "which": [0, 1, 4, 5, 6, 7, 17, 36, 81, 92, 98, 112, 127, 134, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 154, 161, 162, 163, 164, 166, 170, 171, 172, 173, 174, 175, 176, 178, 192, 202, 219, 223, 224, 236, 237, 241, 242, 243, 244, 245, 257, 258, 266, 273, 278, 281, 282, 286, 306, 312, 313, 315, 328, 330, 334, 358, 391, 393, 396, 400, 403, 419, 432, 433, 446, 449, 450, 451, 452, 453, 457, 458], "appli": [0, 37, 134, 135, 152, 175, 290, 291, 292, 303, 305, 306, 307, 308, 309, 310, 312, 313, 315, 316, 318, 319, 320, 321, 322, 324, 325, 326, 327, 328, 329, 331, 344, 351, 353, 354, 355, 356, 357, 359, 361, 363, 364, 365, 366, 367, 368, 369, 371, 380, 381, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 419, 432, 435, 441, 444, 449, 450], "note": [0, 1, 4, 7, 15, 81, 88, 92, 106, 135, 143, 144, 153, 160, 173, 195, 219, 223, 280, 286, 303, 354, 371, 424, 454, 456], "assign": [0, 1, 37, 422], "henc": [0, 1, 219, 449], "If": [0, 1, 4, 7, 14, 15, 16, 17, 25, 26, 27, 28, 76, 80, 81, 91, 93, 99, 102, 103, 104, 105, 111, 112, 115, 116, 117, 127, 132, 134, 145, 146, 147, 150, 151, 154, 160, 170, 171, 173, 178, 187, 188, 189, 191, 192, 200, 201, 204, 208, 212, 215, 216, 218, 223, 227, 236, 252, 253, 254, 259, 263, 265, 266, 267, 270, 272, 273, 278, 279, 282, 284, 286, 290, 292, 307, 308, 309, 310, 318, 324, 326, 335, 337, 347, 353, 355, 358, 360, 362, 371, 391, 393, 404, 426, 449, 450, 451, 453, 456, 457, 458], "provid": [0, 1, 4, 81, 110, 154, 270, 278, 290, 292, 298, 303, 330, 335, 337, 347, 348, 349, 352, 353, 370, 371, 418, 422, 450, 456, 458], "rank": [0, 400, 450], "must": [0, 1, 2, 7, 88, 91, 151, 153, 173, 222, 223, 227, 229, 232, 233, 283, 371, 454], "In": [0, 1, 4, 5, 37, 188, 219, 290, 303, 312, 318, 422, 425, 427, 428, 430, 431, 432, 448, 449, 450, 451, 453, 456, 457], "other": [0, 1, 4, 6, 165, 173, 303, 336, 422, 431, 449, 450, 452, 453, 455], "word": 0, "ndim": [0, 1, 148, 173, 175, 371], "lead": [0, 17, 81, 449], "remain": [0, 4, 278, 291, 311, 312, 313, 450], "For": [0, 1, 4, 7, 37, 135, 152, 165, 173, 219, 293, 303, 307, 312, 315, 330, 335, 344, 347, 353, 358, 362, 371, 373, 374, 375, 376, 391, 419, 424, 446, 449, 450, 451, 452, 453, 454, 455, 456, 457], "exampl": [0, 2, 3, 4, 5, 7, 17, 37, 106, 148, 173, 174, 261, 266, 288, 291, 292, 303, 305, 306, 307, 322, 327, 328, 335, 337, 344, 347, 371, 372, 373, 374, 375, 376, 377, 378, 379, 391, 393, 400, 419, 424, 433, 438, 439, 440, 441, 442, 446, 451, 452, 453, 454, 455, 456], "auto": [0, 1, 7], "4": [0, 1, 4, 110, 148, 153, 173, 219, 220, 244, 286, 292, 296, 305, 306, 307, 322, 327, 328, 352, 353, 370, 371, 373, 374, 375, 391, 449, 450, 452, 455, 457], "3": [0, 1, 4, 7, 148, 165, 173, 174, 288, 292, 310, 315, 371, 374, 376, 382, 387, 426, 431, 446, 449, 452, 454, 455], "out": [0, 1, 7, 88, 305, 306, 312, 313, 327, 328, 344, 449, 450, 451, 452], "produc": [0, 1, 7, 92, 350, 419], "thi": [0, 1, 4, 5, 7, 14, 15, 16, 17, 25, 26, 27, 28, 81, 106, 123, 152, 153, 157, 160, 166, 170, 171, 172, 173, 174, 175, 176, 183, 187, 188, 189, 191, 193, 195, 204, 216, 218, 223, 247, 252, 253, 254, 259, 263, 266, 272, 279, 288, 291, 292, 303, 311, 312, 313, 316, 317, 323, 331, 332, 334, 335, 338, 339, 340, 345, 347, 348, 349, 350, 353, 355, 368, 373, 374, 375, 376, 382, 383, 384, 391, 399, 416, 422, 433, 448, 449, 450, 451, 453, 454, 456], "row": [0, 1, 81, 131, 158, 219, 275], "posit": [0, 4, 27, 112, 134, 148, 154, 164, 170, 171, 206, 208, 216, 227, 273, 278, 290, 303, 308, 309, 310, 350, 358, 362, 394, 404], "ad": [0, 1, 3, 7, 132, 322, 422, 425, 426, 427, 428, 429, 430, 436, 450, 453, 456], "we": [0, 1, 3, 4, 5, 110, 153, 219, 220, 303, 314, 352, 360, 429, 431, 446, 448, 449, 450, 451, 453, 457], "also": [0, 1, 4, 5, 6, 7, 10, 12, 85, 86, 87, 113, 119, 120, 124, 138, 141, 144, 147, 155, 156, 167, 168, 169, 183, 190, 205, 207, 211, 217, 219, 235, 238, 262, 286, 287, 298, 303, 334, 348, 350, 352, 353, 361, 381, 409, 411, 418, 424, 449, 450, 451, 452, 453, 454, 455, 458], "To": [0, 1, 2, 3, 4, 5, 7, 200, 303, 419, 424, 449, 450, 451, 455], "control": [0, 325, 446, 453], "addit": [0, 1, 4, 7, 12, 13, 132, 133, 135, 178, 307, 318, 324, 350, 354, 422, 451], "item": [0, 1, 3, 4, 5, 290, 424, 453, 454, 455], "togeth": [0, 1, 5, 219, 290, 291, 450], "doe": [0, 1, 2, 4, 7, 195, 280, 288, 303, 449, 452, 453, 454], "perform": [0, 1, 2, 4, 6, 13, 88, 98, 102, 103, 104, 105, 121, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 152, 153, 157, 188, 220, 239, 252, 266, 303, 318, 370, 375, 376, 424, 449, 450, 452, 453, 457], "bound": [0, 229, 232, 233, 315, 379, 449, 452, 457], "check": [0, 1, 7, 80, 118, 165, 198, 337, 451, 452], "access": [0, 4, 49, 303, 422, 433, 450, 453, 457], "undefin": [0, 27, 106, 170, 171, 216, 227, 452], "typic": [0, 135, 314, 424, 449, 453], "result": [0, 4, 13, 17, 37, 76, 81, 92, 132, 133, 153, 173, 188, 220, 227, 236, 258, 290, 291, 292, 362, 391, 449, 451, 454], "unintend": 0, "invalid": [0, 81], "memori": [0, 1, 6, 81, 193, 195, 196, 197, 199, 200, 201, 370, 422, 426, 449, 453, 454], "write": [0, 1, 4, 303, 454], "scatter_add": 0, "scatter_prod": 0, "scatter_max": 0, "linear": [0, 1, 4, 5, 6, 286, 290, 303, 315, 316, 325, 337, 353, 355, 356, 357, 359, 361, 371, 380, 381, 382, 383, 384, 388, 407, 408, 409, 411, 419, 422, 433, 441, 449], "scatter_min": 0, "sqrt": [0, 4, 125, 135, 157, 307, 315, 318, 322, 324, 326, 354, 362, 373, 374, 375, 376, 382, 425, 427, 428, 429, 436, 449], "root": [0, 4, 133, 240, 255, 354], "rsqrt": 0, "softmax": [0, 4, 135, 303, 390, 393], "precis": [0, 1, 4, 130, 135, 303, 315, 354, 391, 432, 449], "power": [0, 451, 454], "rais": [0, 4, 173, 201, 217, 254, 337], "cumsum": 0, "inclus": [0, 40, 41, 42, 43, 102, 103, 104, 105, 148], "cumul": [0, 81, 102, 103, 104, 105], "cumprod": 0, "cummax": 0, "cummin": 0, "conv_gener": 0, "weight": [0, 3, 96, 97, 98, 132, 133, 290, 303, 337, 341, 352, 353, 391, 393, 422, 426, 429, 431, 433, 437, 451, 453], "padding_lo": 0, "padding_hi": 0, "kernel_dil": [0, 98], "input_dil": [0, 98], "group": [0, 96, 97, 98, 110, 115, 116, 117, 135, 153, 219, 220, 280, 286, 318, 352, 353, 450], "flip": [0, 98, 99], "gener": [0, 1, 2, 3, 10, 17, 98, 131, 143, 144, 177, 192, 222, 227, 228, 229, 232, 233, 370, 446, 449, 452, 453, 458], "convolut": [0, 96, 97, 98, 99, 308, 309, 310, 312, 313], "filter": [0, 99, 308, 309, 310, 330, 334], "conv1d": [0, 303], "dilat": [0, 96, 97, 98, 308, 309], "conv2d": [0, 303], "2d": [0, 97, 112, 219, 307, 312], "conv3d": [0, 303], "tupl": [0, 29, 64, 67, 77, 93, 97, 98, 120, 122, 127, 129, 166, 173, 174, 175, 215, 219, 237, 257, 278, 281, 289, 290, 291, 292, 293, 305, 306, 309, 310, 327, 328, 337, 339, 360, 371, 426, 428, 429, 430, 431, 448, 451], "3d": [0, 1, 307, 313, 371], "quantized_matmul": 0, "w": [0, 3, 97, 110, 153, 219, 220, 278, 291, 306, 307, 309, 310, 312, 313, 326, 328, 424, 437, 451], "bias": [0, 110, 153, 219, 220, 317, 323, 335, 347, 350], "group_siz": [0, 110, 153, 219, 220, 286, 352, 353], "64": [0, 88, 110, 153, 219, 220, 286, 296, 352, 353], "bit": [0, 110, 153, 167, 219, 220, 238, 286, 296, 330, 352, 353, 354], "quantiz": [0, 110, 153, 178, 220, 352, 353], "its": [0, 1, 7, 171, 188, 216, 231, 275, 287, 293, 303, 353, 424, 428, 429, 430, 450, 454, 457], "last": [0, 4, 28, 76, 132, 133, 138, 141, 143, 144, 146, 147, 148, 152, 153, 159, 170, 171, 172, 174, 175, 176, 188, 197, 223, 253, 270, 280, 308, 309, 310, 312, 313, 318, 371, 454], "dequant": [0, 219], "gather_qmm": 0, "lhs_indic": [0, 152, 153], "rhs_indic": [0, 152, 153], "level": [0, 152, 153, 375, 376], "tensordot": 0, "contract": [0, 122], "over": [0, 1, 4, 5, 14, 16, 25, 26, 27, 28, 96, 97, 98, 102, 103, 104, 105, 138, 141, 144, 147, 159, 173, 175, 177, 187, 189, 191, 204, 216, 218, 240, 252, 253, 259, 263, 270, 272, 279, 307, 308, 309, 310, 318, 324, 354, 393, 438, 441, 450, 451], "axes_a": 0, "axes_b": 0, "outer": [0, 449, 453], "inner": [0, 449], "addmm": 0, "c": [0, 1, 4, 13, 173, 305, 306, 307, 308, 309, 310, 312, 313, 322, 323, 327, 328, 454, 455, 457], "alpha": [0, 1, 13, 219, 380, 404, 406, 409, 429, 436], "f": [0, 1, 3, 5, 106, 173, 303, 323, 429, 449, 454], "beta": [0, 1, 13, 110, 219, 307, 318, 322, 324, 403, 424, 428, 429, 430, 431], "block_masked_mm": 0, "block_siz": [0, 88], "mask_out": [0, 88], "mask_lh": [0, 88], "mask_rh": [0, 88], "block": [0, 1, 4, 88, 370], "mask": [0, 4, 88, 135, 344, 350, 452], "gather_mm": [0, 153], "extract": [0, 4, 44, 111, 112, 303, 334, 422, 450], "construct": [0, 1, 5, 44, 111, 151, 212, 271, 284], "diag": [0, 175], "trace": [0, 2, 449], "implement": [0, 3, 5, 134, 135, 173, 314, 334, 350, 358, 360, 362, 368, 370, 371, 416, 425, 426, 427, 428, 430, 431, 432, 444, 449, 451], "allow": [0, 1, 165, 288, 303, 349, 422, 444, 450, 452, 455], "inject": 0, "ensur": [0, 1, 7, 288, 399, 450], "been": [0, 1, 4, 196, 453], "when": [0, 1, 4, 6, 7, 92, 98, 106, 170, 171, 172, 173, 175, 176, 178, 308, 309, 310, 371, 375, 376, 391, 397, 403, 422, 424, 440, 446, 449, 450, 457], "atleast_1d": 0, "atleast": 0, "atleast_2d": 0, "atleast_3d": 0, "number_of_el": 0, "invert": 0, "int32": [0, 10, 17, 37, 148, 165, 173, 229, 296, 371, 452, 455], "some": [0, 1, 3, 4, 5, 335, 347, 424, 433, 449, 451, 453], "scalar": [0, 1, 12, 13, 15, 29, 49, 76, 80, 85, 86, 87, 88, 89, 91, 119, 120, 124, 150, 151, 154, 155, 156, 157, 160, 167, 168, 169, 173, 177, 183, 184, 185, 186, 188, 190, 205, 207, 208, 211, 215, 217, 222, 229, 232, 233, 235, 238, 242, 262, 278, 280, 283, 287, 404, 451, 453, 455], "us": [0, 2, 3, 4, 5, 6, 7, 17, 37, 81, 106, 110, 113, 115, 116, 120, 134, 148, 153, 167, 170, 171, 172, 173, 174, 175, 176, 188, 195, 196, 197, 200, 219, 220, 236, 237, 238, 265, 289, 292, 296, 298, 303, 306, 312, 314, 315, 317, 323, 326, 328, 330, 334, 341, 348, 350, 352, 353, 355, 358, 362, 370, 371, 375, 376, 382, 383, 392, 419, 422, 424, 425, 426, 428, 429, 430, 431, 432, 433, 446, 448, 449, 450, 451, 452, 455, 457], "shapeless": [0, 92], "compil": [0, 2, 6, 7, 113, 123, 450, 451, 453], "pun": 0, "intend": [0, 449], "conjug": [0, 94], "bitwise_and": 0, "bitwis": [0, 85, 86, 87, 167, 238], "bitwise_or": 0, "bitwise_xor": 0, "exclus": [0, 81, 87], "left_shift": 0, "shift": [0, 167, 238, 307], "left": [0, 4, 134, 167, 173, 219, 305, 306, 315, 327, 328, 358, 371, 382, 383, 394, 396, 404], "right_shift": 0, "right": [0, 1, 7, 219, 238, 305, 306, 315, 327, 328, 371, 382, 383, 394, 396, 404], "you": [1, 2, 4, 5, 6, 7, 303, 362, 370, 419, 446, 449, 450, 451, 452, 454, 456, 457], "can": [1, 2, 4, 6, 7, 12, 17, 64, 77, 81, 85, 86, 87, 92, 112, 113, 114, 119, 120, 124, 127, 155, 156, 167, 168, 169, 173, 183, 190, 205, 207, 211, 217, 222, 223, 229, 232, 233, 235, 238, 242, 262, 273, 278, 292, 303, 306, 314, 315, 328, 334, 347, 352, 360, 371, 393, 419, 422, 424, 432, 433, 446, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458], "extend": [1, 215], "guid": [1, 6], "explain": 1, "how": [1, 4, 5, 303, 305, 306, 308, 309, 310, 314, 327, 328, 352, 371, 432, 449, 452, 457], "simpl": [1, 4, 5, 303, 314, 418, 424, 449, 450, 451, 453], "let": [1, 3, 4, 171, 449, 451, 453, 454], "sai": [1, 4, 419, 453], "would": [1, 4, 371, 450, 452, 453, 454, 457], "like": [1, 4, 6, 165, 213, 285, 313, 399, 433, 435, 449, 450, 451, 453, 454, 455, 457], "them": [1, 4, 115, 303, 335, 347, 450, 457], "both": [1, 12, 85, 86, 87, 119, 120, 124, 155, 156, 165, 167, 168, 169, 173, 183, 190, 205, 207, 211, 217, 223, 235, 238, 262, 286, 305, 306, 322, 323, 327, 328, 424, 449, 450, 451, 455, 457], "coeffici": [1, 425, 426, 428, 429, 430, 431], "respect": [1, 3, 5, 132, 133, 152, 153, 154, 219, 278, 290, 303, 307, 315, 318, 322, 324, 422, 451, 455], "get": [1, 3, 5, 7, 97, 98, 107, 108, 194, 195, 196, 197, 225, 303, 449, 451, 453, 457], "z": [1, 317, 449, 453], "directli": [1, 4, 81], "import": [1, 2, 3, 4, 5, 7, 106, 157, 173, 244, 278, 289, 290, 291, 292, 293, 303, 305, 306, 307, 322, 327, 328, 337, 371, 391, 393, 400, 419, 422, 424, 449, 450, 451, 452, 453, 454, 455], "core": [1, 2, 3, 4, 5, 286, 303, 305, 306, 307, 322, 327, 328, 337, 340, 342, 345, 371, 372, 373, 374, 375, 376, 377, 378, 379, 391, 393, 400, 419, 422, 424, 449, 450, 454, 455], "mx": [1, 2, 3, 4, 5, 37, 94, 95, 106, 117, 148, 165, 173, 174, 178, 244, 278, 288, 303, 305, 306, 307, 322, 325, 327, 328, 330, 337, 341, 356, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 388, 391, 392, 393, 397, 400, 407, 417, 419, 422, 424, 446, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458], "def": [1, 3, 4, 5, 106, 278, 303, 422, 449, 450, 451, 452, 453, 454, 457], "simple_axpbi": 1, "while": [1, 2, 4, 7, 237, 358, 453, 454], "leav": [1, 127, 290, 291, 292], "howev": [1, 106, 303, 315, 318, 433, 446, 449, 450, 453, 454], "mai": [1, 173, 286, 312, 450, 451, 452], "need": [1, 4, 5, 6, 7, 80, 219, 303, 348, 349, 362, 370, 446, 450, 451, 453, 454, 455, 457], "underli": [1, 280], "perhap": [1, 4], "make": [1, 2, 4, 5, 7, 188, 210, 247, 303, 438, 439, 441, 442, 449, 453, 455, 457], "faster": [1, 7, 120, 381, 391, 449, 451], "differenti": [1, 6], "tutori": 1, "go": [1, 4, 451], "through": [1, 260, 370, 431, 449, 451, 454], "It": [1, 4, 7, 154, 247, 278, 288, 292, 303, 349, 353, 432, 444, 454, 456], "cover": 1, "structur": [1, 432, 451], "librari": [1, 7, 298, 303], "redirect": 1, "acceler": [1, 307], "appropri": [1, 449], "metal": [1, 6], "vjp": [1, 106, 455], "jvp": [1, 106, 455], "graph": [1, 4, 5, 6, 451], "rule": [1, 424], "evalu": [1, 4, 5, 6, 127, 166, 281, 303, 333, 344, 422, 424, 449, 455], "discuss": 1, "more": [1, 2, 5, 9, 76, 112, 152, 170, 171, 172, 175, 176, 188, 200, 201, 242, 243, 296, 303, 307, 312, 358, 362, 370, 371, 373, 374, 375, 376, 391, 446, 449, 450, 451, 452, 455, 457], "detail": [1, 9, 200, 303, 312, 358, 362, 371, 373, 374, 375, 376, 425, 427, 428, 430, 431, 452, 455], "front": 1, "thei": [1, 3, 4, 7, 15, 99, 153, 160, 360, 395, 422, 431, 448, 449, 450, 453, 455, 456, 457], "defin": [1, 3, 4, 5, 7, 106, 153, 173, 220, 286, 289, 454], "api": [1, 315, 450, 451], "axpbi": 1, "follow": [1, 4, 5, 6, 7, 17, 99, 110, 152, 173, 215, 219, 291, 303, 382, 383, 397, 425, 426, 427, 428, 429, 430, 431, 437, 446, 449, 450, 451, 457], "numpi": [1, 4, 5, 6, 12, 15, 17, 85, 86, 87, 89, 119, 120, 124, 155, 156, 160, 167, 168, 169, 183, 188, 190, 205, 207, 211, 217, 235, 238, 262, 453, 455, 456], "style": [1, 12, 15, 85, 86, 87, 119, 120, 124, 155, 156, 160, 167, 168, 169, 183, 188, 190, 205, 207, 211, 217, 235, 238, 262], "upcast": 1, "factor": [1, 13, 157, 170, 171, 174, 371, 393, 439, 442], "stream": [1, 6, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 108, 109, 110, 111, 112, 119, 120, 121, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 226, 227, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 279, 280, 283, 284, 285, 450, 457], "schedul": [1, 201, 424, 438, 439, 440, 441, 442, 444, 457], "simplest": [1, 303, 450], "wai": [1, 4, 7, 303, 371, 449, 450, 451, 452], "term": [1, 394, 425, 426, 427, 428, 429, 430, 436], "exist": [1, 2, 4, 335, 347], "themselv": [1, 449], "contain": [1, 4, 7, 27, 28, 67, 92, 112, 122, 142, 143, 144, 152, 153, 173, 184, 185, 186, 219, 254, 283, 288, 303, 334, 336, 337, 343, 370, 400, 419, 422, 449, 450, 451], "act": [1, 399], "nor": [1, 154, 278], "rather": [1, 451, 457], "easi": [1, 303, 450], "interfac": 1, "part": [1, 451, 452], "further": [1, 7, 451], "ha": [1, 2, 4, 5, 6, 7, 76, 92, 112, 142, 143, 145, 146, 147, 154, 170, 171, 172, 175, 176, 192, 196, 223, 307, 317, 323, 326, 355, 422, 424, 449, 452, 453, 455, 457], "method": [1, 4, 8, 9, 29, 106, 114, 286, 294, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 341, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 422, 425, 426, 427, 428, 429, 430, 431, 433, 436, 437, 444], "run": [1, 2, 4, 5, 6, 7, 8, 214, 294, 307, 330, 425, 426, 428, 429, 430, 449, 450, 453, 457, 458], "our": [1, 4, 5, 360, 425, 426, 427, 428, 430, 431, 450], "concret": [1, 317, 323, 326, 355, 453, 457], "class": [1, 4, 5, 8, 9, 10, 29, 106, 114, 294, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 422, 425, 426, 427, 428, 429, 430, 431, 436, 437, 444], "public": [1, 303], "explicit": [1, 433, 446, 454], "alpha_": 1, "beta_": 1, "know": [1, 4], "itself": [1, 286, 433], "popul": 1, "avoid": [1, 342, 449], "unnecessari": [1, 4], "alloc": [1, 196, 200, 201, 422], "respons": 1, "void": 1, "eval_cpu": 1, "overrid": [1, 123], "eval_gpu": 1, "jacobian": [1, 166, 281, 455], "primal": [1, 106, 166, 281], "argnum": [1, 154, 278, 451], "cotan": [1, 106], "across": [1, 318, 450], "repres": [1, 4, 114, 117, 153, 400, 404, 454], "virtual": 1, "vmap": [1, 106, 451, 453, 455], "print": [1, 3, 4, 5, 7, 288, 289, 290, 291, 293, 303, 446, 449, 450, 451, 452, 453, 454, 455], "ostream": 1, "o": [1, 7, 135, 323], "is_equival": 1, "privat": 1, "fall": [1, 106], "eval": [1, 2, 3, 4, 5, 303, 422, 424, 449, 450, 451, 453, 455], "deriv": [1, 451, 453], "pass": [1, 4, 5, 64, 77, 214, 215, 278, 287, 289, 290, 291, 303, 335, 347, 348, 349, 360, 449, 450, 453], "reimplement": 1, "now": [1, 4, 7, 353, 449, 450, 454], "promot": 1, "promoted_dtyp": 1, "promote_typ": 1, "out_dtyp": 1, "is_floating_point": 1, "cast": [1, 36, 145, 146, 147, 178, 330, 342, 454], "up": [1, 4, 106, 449], "x_cast": 1, "y_cast": 1, "broadcasted_input": 1, "out_shap": 1, "unique_ptr": 1, "make_shar": 1, "to_stream": 1, "handl": [1, 303, 449], "resolv": 1, "No": [1, 4], "happen": [1, 4, 132, 370, 424, 449, 453], "call": [1, 2, 4, 5, 30, 150, 193, 197, 303, 314, 335, 347, 352, 360, 422, 424, 433, 449, 450, 451, 453], "alon": [1, 454], "onli": [1, 4, 6, 7, 80, 88, 97, 98, 173, 219, 227, 280, 303, 334, 335, 337, 342, 344, 347, 348, 349, 422, 449, 450, 451, 456, 457], "execut": [1, 7, 82, 83, 84, 197, 454, 457], "devic": [1, 6, 7, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 107, 108, 109, 110, 111, 112, 119, 120, 121, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 194, 201, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 226, 227, 228, 229, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 279, 280, 283, 284, 285, 294, 457, 458], "user": [1, 4, 303], "naiv": [1, 451], "version": [1, 7, 110, 183, 187, 219, 252, 282, 446, 451, 452], "declar": 1, "member": [1, 303, 340, 345], "earlier": 1, "find": [1, 3, 7, 450], "captur": [1, 2, 92, 202, 203, 303, 449], "axpby_impl": 1, "readi": 1, "malloc_or_wait": 1, "synchron": [1, 449], "avail": [1, 3, 4, 5, 7, 9, 118, 198, 298, 457], "There": [1, 303, 371, 449], "wait": [1, 4, 201], "here": [1, 4, 424, 449, 451, 453, 456, 457], "request": 1, "under": [1, 173], "pressur": 1, "set_data": 1, "nbyte": 1, "collect": [1, 290, 291, 448], "pointer": 1, "x_ptr": 1, "y_ptr": 1, "out_ptr": 1, "relev": 1, "static_cast": 1, "out_idx": 1, "map": [1, 5, 37, 178, 290, 314, 330], "x_offset": 1, "elem_to_loc": 1, "y_offset": 1, "contigu": [1, 81], "regularli": 1, "default": [1, 7, 13, 14, 15, 16, 17, 25, 26, 27, 28, 80, 81, 88, 92, 93, 96, 97, 98, 106, 107, 108, 110, 111, 112, 115, 116, 117, 131, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 152, 153, 154, 157, 158, 160, 170, 171, 172, 173, 174, 175, 176, 177, 178, 187, 189, 191, 192, 200, 201, 204, 208, 212, 215, 216, 218, 219, 220, 222, 223, 224, 226, 227, 228, 229, 231, 232, 233, 236, 237, 239, 246, 247, 253, 254, 257, 258, 259, 261, 263, 265, 270, 272, 273, 274, 275, 276, 277, 278, 279, 282, 284, 286, 296, 305, 306, 307, 308, 309, 310, 316, 317, 319, 322, 323, 325, 326, 327, 328, 330, 335, 337, 342, 344, 347, 350, 351, 352, 353, 355, 358, 362, 366, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 384, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 422, 425, 426, 427, 428, 429, 430, 431, 436, 437, 438, 446, 448, 449, 451, 454, 456, 458], "doesn": [1, 303], "should": [1, 3, 4, 5, 7, 81, 112, 115, 132, 133, 135, 166, 193, 202, 219, 267, 273, 278, 281, 286, 289, 303, 308, 309, 310, 312, 313, 344, 350, 360, 393, 395, 400, 422, 448, 449, 450, 451, 453, 454, 458], "work": [1, 2, 4, 201, 449, 450, 451, 452, 453], "incom": 1, "accordingli": 1, "dispatch": 1, "float16": [1, 10, 157, 178, 296, 330, 453, 454], "bfloat16": [1, 10, 157, 296, 454], "complex64": [1, 10, 296], "throw": [1, 92, 117], "encount": [1, 451], "unexpect": [1, 17], "correct": [1, 7, 428, 429, 430, 452, 453], "float16_t": 1, "bfloat16_t": 1, "complex64_t": 1, "runtime_error": 1, "support": [1, 4, 6, 7, 15, 88, 135, 148, 157, 160, 170, 171, 172, 174, 175, 176, 178, 188, 219, 227, 450, 451, 452, 454, 456], "good": [1, 7, 424, 449, 450, 457], "fallback": 1, "routin": 1, "framework": [1, 6], "certain": [1, 344, 449], "case": [1, 4, 117, 138, 141, 142, 144, 145, 146, 147, 148, 170, 171, 172, 174, 175, 176, 188, 237, 257, 306, 312, 328, 368, 385, 403, 409, 415, 416, 432, 433, 449, 451, 455, 456, 457, 458], "half": [1, 17, 229, 233, 358, 453], "fix": [1, 4, 7, 453], "direct": [1, 4, 332, 431, 457], "column": [1, 131, 158, 219], "place": [1, 4, 37, 239, 286, 450, 453, 454], "expect": [1, 4, 308, 309, 310, 311, 312, 313, 362, 370, 394, 449, 452], "catlas_saxpbi": 1, "axpby_impl_acceler": 1, "special": 1, "copy_inplac": 1, "copytyp": 1, "inplac": [1, 7], "incx": 1, "inci": 1, "fit": [1, 219, 457], "criteria": 1, "With": 1, "mind": [1, 4], "finish": 1, "assert": 1, "singl": [1, 5, 127, 166, 178, 192, 215, 281, 306, 328, 449, 452, 456], "flag": [1, 7, 449, 454], "row_contigu": 1, "col_contigu": 1, "common": [1, 424, 449, 453], "just": [1, 5, 324, 449, 452], "much": [1, 4, 305, 306, 327, 328, 449, 453], "enough": [1, 453], "plan": [1, 449], "enjoi": 1, "speed": 1, "appl": [1, 4, 6, 7, 457], "silicon": [1, 4, 6, 7, 457], "address": 1, "shade": 1, "languag": 1, "kernel": [1, 7, 96, 97, 98, 305, 306, 327, 328, 449, 452], "written": 1, "help": [1, 4, 449, 457], "resourc": 1, "walkthrough": 1, "pipelin": 1, "document": [1, 2, 64, 77, 242, 243, 296, 449, 451, 452], "specif": [1, 7, 450, 451], "cpp": 1, "keep": [1, 14, 16, 25, 26, 187, 189, 191, 204, 218, 259, 263, 279, 303, 334, 451, 453], "launch": [1, 117, 450, 452], "exactli": [1, 4, 337, 451], "mani": [1, 81, 254, 308, 309, 310, 314, 352, 449, 450, 453], "thread": 1, "pick": 1, "axpby_gener": 1, "buffer": [1, 195, 454], "x_stride": 1, "6": [1, 4, 173, 244, 357, 370, 374, 382, 383, 387, 394, 404, 408, 436, 449, 452, 455], "y_stride": 1, "7": [1, 4, 173, 219, 452], "uint": 1, "thread_position_in_grid": 1, "instanti": [1, 5, 453], "give": [1, 4, 5, 27, 449], "uniqu": [1, 446], "host": 1, "name": [1, 153, 178, 219, 220, 242, 243, 244, 245, 303, 318, 334, 337, 339, 450, 452, 456], "so": [1, 4, 7, 154, 157, 278, 311, 371, 424, 449, 450, 453, 457], "identifi": [1, 289, 448], "instantiate_axpbi": 1, "type_nam": 1, "host_nam": 1, "axpby_general_": 1, "set": [1, 4, 5, 7, 92, 106, 113, 115, 116, 117, 123, 132, 134, 194, 200, 201, 246, 247, 261, 315, 324, 326, 333, 335, 342, 343, 344, 347, 348, 353, 358, 368, 392, 404, 416, 422, 424, 426, 433, 446, 451, 453], "grid": [1, 192], "shown": 1, "below": [1, 7, 173, 275, 277, 296, 371, 453], "prepar": [1, 4], "carri": 1, "ostringstream": 1, "kname": 1, "axpby_": 1, "general_": 1, "type_to_nam": 1, "sure": [1, 2, 4, 7, 303, 449], "register_librari": 1, "mlx_ext": 1, "get_kernel": 1, "str": [1, 99, 121, 122, 154, 173, 178, 192, 194, 202, 241, 242, 243, 244, 245, 278, 289, 293, 330, 331, 334, 335, 337, 339, 341, 347, 371, 375, 376, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404], "encod": [1, 134, 358, 362, 370, 393], "compute_encod": 1, "get_command_encod": 1, "setcomputepipelinest": 1, "regist": [1, 5], "those": [1, 4, 303], "nelem": 1, "set_input_arrai": 1, "set_output_arrai": 1, "setbyt": 1, "sizeof": 1, "threadgroup": 1, "higher": [1, 159, 400, 451], "than": [1, 4, 76, 99, 112, 120, 134, 152, 155, 156, 168, 169, 170, 171, 172, 175, 176, 188, 200, 288, 290, 358, 368, 371, 400, 403, 416, 426, 431, 449, 451, 457], "tgp_size": 1, "maxtotalthreadsperthreadgroup": 1, "mtl": 1, "group_dim": 1, "grid_dim": 1, "among": 1, "dispatchthread": 1, "few": [1, 4, 5, 6, 7, 450, 453, 455], "thing": [1, 4, 450], "about": [1, 4, 5, 122, 194, 453, 457], "befor": [1, 4, 7, 27, 216, 334, 370, 433, 450, 452, 453], "track": [1, 303, 307], "activ": [1, 7, 195, 312, 368, 370, 385, 405, 415, 416, 418, 449], "command_buff": 1, "mtlcommandbuff": 1, "associ": [1, 244, 245, 453], "reli": 1, "u": [1, 170, 175, 326, 349, 444, 453], "command": [1, 2, 7, 450], "instead": [1, 7, 106, 303, 349, 362, 450, 451, 453], "end_encod": 1, "until": [1, 453, 455], "hit": 1, "flush": 1, "next": [1, 4, 5, 200], "These": [1, 92, 267, 393, 457], "built": [1, 7, 453], "top": [1, 272, 326, 371], "includ": [1, 102, 103, 104, 105, 195, 196, 201, 324, 331, 343, 353, 394, 424, 449, 451, 452, 455, 456, 458], "forward": [1, 278, 449, 453], "diff": 1, "push": 1, "op": [1, 214, 280, 335, 453], "similarli": [1, 7, 188, 451, 453], "scale_arr": 1, "contribut": 1, "tangent_x": 1, "tangent_i": 1, "cotang": [1, 106, 281], "unus": 1, "arg": [1, 4, 9, 114, 127, 244, 245], "push_back": 1, "fulli": [1, 6, 449, 450, 454, 457], "look": [1, 4, 450], "overal": 1, "directori": [1, 4, 7], "h": [1, 96, 97, 173, 306, 307, 309, 310, 312, 313, 317, 323, 328, 355, 451, 453], "mlx_sample_extens": 1, "__init__": [1, 4, 5, 8, 9, 10, 29, 106, 114, 294, 303, 422], "py": [1, 4, 7, 450], "cmakelist": 1, "txt": 1, "setup": [1, 3, 5, 7, 449], "packag": [1, 3, 5, 7, 298, 419, 450], "hold": [1, 4, 9, 10, 173, 449], "instal": 1, "nanobind": [1, 370], "sinc": [1, 4, 5, 153, 197, 422, 431, 440, 454, 457], "compon": [1, 4], "etc": [1, 219, 303, 371, 450], "alreadi": [1, 2, 4, 450], "nb_modul": 1, "_ext": 1, "doc": [1, 5, 450], "sampl": [1, 3, 4, 177, 222, 223, 224, 226, 227, 229, 232, 233, 373, 374, 375, 376, 378, 379, 394, 400, 404, 446, 449], "_a": 1, "nb": 1, "kw_onli": 1, "none": [1, 4, 8, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 109, 110, 111, 112, 113, 115, 116, 119, 120, 121, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 199, 202, 203, 204, 205, 206, 207, 208, 209, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 243, 244, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 282, 283, 284, 285, 286, 289, 290, 291, 292, 294, 305, 306, 315, 327, 328, 330, 334, 335, 342, 347, 350, 355, 362, 370, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 426, 444, 452], "r": [1, 4, 174, 278, 312, 317], "most": [1, 223, 303, 435, 449, 450, 451, 452, 453], "complex": [1, 94, 95, 143, 144, 145, 146, 147, 289, 296, 303, 349, 449, 451], "abov": [1, 4, 219, 276, 303, 371, 429, 450, 451, 452, 453, 457], "come": [1, 4, 450, 451], "bell": 1, "whistl": 1, "liter": [1, 215, 371, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404], "modul": [1, 4, 5, 286, 287, 360, 370, 419, 435, 448, 449, 453], "caster": 1, "requir": [1, 4, 303, 450, 453, 454], "find_packag": 1, "config": [1, 450], "link": [1, 7], "your": [1, 4, 7, 422, 451, 453], "add_librari": 1, "target_sourc": 1, "cmake_current_list_dir": 1, "header": 1, "target_include_directori": 1, "target_link_librari": 1, "attach": 1, "conveni": [1, 5, 165], "mlx_build_metallib": 1, "metallib": [1, 7], "target": [1, 278, 391, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 449], "automat": [1, 6, 178, 450, 455, 456, 457], "what": [1, 4, 290], "practic": [1, 449], "mlx_build_met": [1, 7], "mlx_ext_metallib": 1, "titl": 1, "include_dir": 1, "project_source_dir": 1, "mlx_include_dir": 1, "output_directori": 1, "cmake_library_output_directori": 1, "add_depend": 1, "endif": 1, "final": [1, 3, 4, 5, 157, 438, 441], "nanobind_add_modul": 1, "nb_static": 1, "stable_abi": 1, "lto": 1, "nomins": 1, "nb_domain": 1, "build_shared_lib": [1, 7], "target_link_opt": 1, "wl": 1, "rpath": 1, "loader_path": 1, "onc": [1, 7, 449], "describ": [1, 453], "util": [1, 4, 6, 7, 244, 303, 424, 450], "__name__": [1, 4], "__main__": [1, 4], "descript": [1, 4, 296], "ext_modul": 1, "cmakeextens": 1, "cmdclass": 1, "build_ext": [1, 7], "cmakebuild": 1, "package_data": 1, "dylib": 1, "extras_requir": 1, "dev": [1, 7], "zip_saf": 1, "python_requir": 1, "even": [1, 4, 92, 449, 453, 454], "though": [1, 4, 449, 453, 454], "pip": [1, 7], "develop": [1, 7], "j8": 1, "libmlx_ext": 1, "cpython": 1, "3x": 1, "darwin": 1, "try": [1, 7], "after": [1, 4, 5, 27, 148, 150, 153, 193, 216, 219, 307, 318, 324, 330, 331, 335, 337, 344, 347, 348, 349, 350, 370, 403, 449, 457], "abl": [1, 219], "simpli": [1, 4, 7, 325, 356, 380, 388, 407, 417, 422, 449, 450, 451], "plai": [1, 4], "quick": [1, 6], "benchmark": [1, 449], "see": [1, 4, 5, 7, 9, 10, 31, 32, 33, 34, 35, 38, 39, 40, 41, 42, 43, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 173, 200, 242, 243, 286, 296, 303, 307, 312, 315, 319, 320, 321, 333, 351, 352, 353, 356, 357, 358, 359, 361, 362, 363, 364, 365, 366, 367, 369, 371, 373, 374, 375, 376, 381, 382, 383, 409, 449, 450, 451, 452, 455, 457], "compar": [1, 80, 449], "time": [1, 4, 7, 201, 271, 303, 305, 306, 317, 323, 327, 328, 355, 449, 451, 453, 457], "set_default_devic": 1, "256": [1, 5], "512": [1, 2, 4, 370, 457], "random": [1, 2, 3, 4, 5, 6, 305, 306, 307, 322, 327, 328, 337, 344, 449, 451, 457, 458], "normal": [1, 3, 4, 132, 133, 227, 232, 303, 305, 306, 307, 318, 322, 324, 327, 328, 354, 370, 373, 375, 454, 457], "bench": 1, "warm": [1, 449], "100": [1, 3, 4, 391, 441, 449, 451, 453, 457], "5000": 1, "e": [1, 5, 7, 106, 125, 152, 153, 166, 248, 307, 308, 309, 310, 312, 313, 318, 322, 324, 335, 354, 389, 390, 412, 413, 418, 424, 427, 449, 453, 458], "simple_tim": 1, "custom_tim": 1, "3f": [1, 5, 449], "114": 1, "109": 1, "modest": 1, "improv": [1, 2, 4, 391, 425, 426, 427, 428, 429, 430, 436, 449, 450], "awai": [1, 4], "nn": [1, 4, 5, 244, 290, 303, 419, 422, 424, 433, 435, 449, 453], "grad": [1, 3, 5, 278, 288, 424, 432, 449, 450, 451, 452, 453, 455], "profil": 2, "kei": [2, 4, 135, 194, 222, 223, 224, 226, 227, 228, 229, 231, 232, 233, 289, 290, 334, 335, 347, 350, 433, 446, 448, 451], "optim": [2, 3, 5, 6, 348, 449, 450, 451, 453], "build": [2, 4, 6, 375, 422, 449], "mlx": [2, 3, 4, 5, 7, 298, 303, 419, 422, 424, 446, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457], "mlx_metal_debug": [2, 7], "debug": [2, 450], "record": [2, 197, 453], "dure": [2, 92, 311, 312, 313, 371, 454], "later": [2, 7, 424], "inspect": [2, 449, 455], "label": [2, 3, 393, 400], "object": [2, 9, 29, 49, 76, 92, 165, 244, 282, 289, 290, 291, 292, 296, 312, 370, 448], "queue": 2, "readabl": 2, "enabl": [2, 4, 7, 92, 123, 437], "python": [2, 4, 49, 67, 76, 127, 289, 290, 291, 292, 293, 422, 432, 433, 435, 448, 450, 451, 454], "prepend": [2, 188], "cmake_arg": 2, "dmlx_metal_debug": 2, "ON": [2, 7], "start_captur": 2, "initi": [2, 3, 4, 117, 292, 303, 307, 318, 322, 324, 326, 351, 354, 372, 373, 374, 375, 376, 377, 378, 379, 422, 433, 438, 439, 441, 442, 449, 450, 453], "gpu": [2, 6, 7, 194, 449, 452, 457], "applic": [2, 7], "mtl_capture_en": 2, "uniform": [2, 303, 326, 337, 374, 376, 419, 446, 449, 451, 457], "trace_fil": 2, "mlx_trace": 2, "gputrac": [2, 202], "path": [2, 7, 122, 202, 244, 245, 286, 291, 337, 450], "_": [2, 3, 4, 291, 303, 438, 439, 440, 441, 442, 446, 449, 453, 457], "stop_captur": 2, "open": [2, 7, 17, 229, 233], "replai": 2, "great": 2, "overview": 2, "oper": [2, 4, 6, 8, 36, 82, 83, 84, 98, 135, 152, 153, 217, 252, 260, 267, 294, 303, 370, 431, 449, 450, 451, 452, 453, 454, 455, 457, 458], "checkout": [2, 449], "inform": [2, 4, 5, 7, 122, 194, 242, 243, 296, 303, 307, 315, 350, 451, 457], "skip": [2, 81], "save": [2, 4, 6, 178, 202, 219, 242, 243, 244, 245, 341, 453], "project": [2, 4, 350], "cmake": [2, 7], "mkdir": [2, 7], "cd": [2, 7], "g": [2, 7, 106, 173, 219, 323, 418, 436, 437, 453, 458], "xcodeproj": 2, "metal_captur": 2, "schema": 2, "basic": [3, 239, 451], "model": [3, 5, 6, 244, 286, 287, 290, 291, 303, 330, 333, 335, 337, 341, 344, 346, 347, 348, 350, 370, 419, 422, 424, 432, 433, 435, 449, 450, 453], "learn": [3, 5, 6, 307, 318, 322, 324, 351, 354, 424, 425, 426, 427, 428, 429, 430, 431, 436, 437], "problem": [3, 5, 303], "metadata": [3, 178, 242, 243], "num_featur": [3, 307], "num_exampl": 3, "1_000": 3, "num_it": 3, "10_000": 3, "iter": [3, 5, 175, 290, 291, 446, 449, 453], "sgd": [3, 5, 424, 431, 433, 438, 439, 442, 449], "lr": [3, 431], "01": [3, 325, 388, 429], "rate": [3, 424, 425, 426, 427, 428, 429, 430, 431, 436, 437], "ll": [3, 5, 396, 449, 451], "synthet": 3, "dataset": [3, 450, 453], "design": [3, 6, 446, 457], "ground": [3, 4, 393, 403], "truth": [3, 393, 403], "w_star": 3, "gaussian": [3, 315, 381, 382, 383, 394], "nois": 3, "noisi": 3, "ep": [3, 132, 133, 307, 318, 322, 324, 354, 392, 394, 404, 424, 425, 426, 427, 428, 429, 430, 436], "loss": [3, 5, 278, 303, 424, 449, 450, 451, 453], "loss_fn": [3, 5, 424, 449, 451], "grad_fn": [3, 449, 451], "randomli": [3, 4, 311, 312, 313], "Then": [3, 7], "repeatedli": 3, "verifi": [3, 7], "close": [3, 6, 7, 15, 160], "error_norm": 3, "5f": 3, "someth": [3, 4, 452], "00005": 3, "00364": 3, "complet": [3, 4, 7, 201, 348, 349, 451, 457], "github": [3, 5, 7, 449], "repo": [3, 5, 7, 449], "effici": [4, 6, 152, 312, 358, 453, 455], "larg": [4, 303, 350, 399, 449, 450, 453], "ish": 4, "transform": [4, 6, 106, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 157, 287, 303, 307, 318, 324, 326, 334, 335, 347, 353, 358, 452], "without": [4, 6, 260, 350, 418, 448, 449, 450, 453, 454, 457], "compromis": 4, "eas": 4, "llama": 4, "famili": 4, "200": [4, 440], "line": [4, 450, 453, 454], "neural": [4, 6, 314, 373, 374, 405, 419, 422, 436], "network": [4, 6, 307, 312, 314, 373, 374, 419, 422, 436, 450], "concis": 4, "architectur": [4, 7, 194, 303, 349, 457], "notabl": [4, 6], "rope": [4, 303], "cach": [4, 7, 193, 195, 196, 200, 449], "llamaattent": 4, "self": [4, 5, 8, 29, 30, 31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 49, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 106, 294, 303, 405, 422], "dim": [4, 134, 135, 314, 318, 322, 324, 350, 352, 354, 358, 362, 370], "num_head": [4, 350, 370], "super": [4, 5, 303, 422], "tradit": [4, 134, 312, 313, 358], "query_proj": 4, "bia": [4, 110, 132, 153, 219, 220, 290, 303, 308, 309, 310, 317, 323, 324, 326, 335, 337, 347, 350, 353, 355, 428, 429, 430, 433, 451], "key_proj": 4, "value_proj": 4, "out_proj": [4, 422], "__call__": [4, 5, 303, 422], "queri": [4, 135, 350], "l": [4, 5, 170, 171, 303, 305, 307, 308, 317, 323, 327, 355, 403], "combin": [4, 175, 292], "key_cach": 4, "value_cach": 4, "math": [4, 404, 449], "score": [4, 135, 400], "values_hat": 4, "possibli": [4, 13, 88, 152, 188, 288], "rm": [4, 7, 133, 426], "swiglu": 4, "rmsnorm": [4, 303], "llamaencoderlay": 4, "mlp_dim": [4, 370], "norm1": 4, "norm2": 4, "linear1": 4, "linear2": 4, "linear3": 4, "instanc": [4, 37, 106, 219, 293, 303, 322, 330, 331, 332, 335, 337, 338, 339, 344, 347, 348, 349, 360, 422, 454], "embed": [4, 286, 303, 352, 358, 362, 392], "emb": [4, 314, 352, 362], "token": [4, 314, 352], "num_lay": [4, 5, 424], "vocab_s": 4, "norm": [4, 133, 288, 318, 404, 430, 431], "multiheadattent": [4, 303], "create_additive_causal_mask": 4, "list": [4, 9, 14, 16, 29, 70, 76, 81, 82, 83, 84, 89, 92, 93, 98, 122, 127, 137, 138, 140, 141, 143, 144, 146, 147, 151, 154, 166, 173, 187, 189, 191, 192, 204, 212, 215, 218, 222, 223, 224, 226, 227, 228, 229, 232, 233, 242, 252, 254, 258, 259, 263, 270, 271, 274, 278, 279, 281, 284, 289, 292, 293, 303, 335, 337, 338, 339, 340, 345, 347, 348, 349, 422, 424, 428, 429, 430, 431, 440, 448, 449, 450, 451, 453], "still": [4, 7, 173, 449, 453], "consid": [4, 15, 80, 160, 289, 290, 291, 318, 448], "train": [4, 5, 303, 307, 311, 312, 313, 333, 335, 347, 373, 374], "process": [4, 98, 99, 114, 115, 116, 117, 290, 291, 313, 314, 370, 448, 450], "ignor": [4, 37, 91, 92, 127, 426], "whatsoev": 4, "rest": [4, 134, 290, 291, 358], "subsect": 4, "prompt": 4, "autoregress": 4, "yield": [4, 5, 446], "temp": 4, "causal": 4, "append": [4, 188, 449, 453], "store": 4, "per": [4, 5, 110, 153, 219, 220, 286, 307, 318, 322, 324, 354, 444, 449, 450, 453], "care": [4, 453], "logit": [4, 223, 391, 393, 449], "categor": 4, "lazili": [4, 303], "noth": [4, 303, 453], "yet": [4, 173, 303, 422, 433, 451, 452, 453, 455], "forc": [4, 5, 303, 450, 455], "choos": [4, 134, 358], "pars": 4, "feed": 4, "back": [4, 106, 198, 454], "loop": [4, 5, 449, 450, 451, 453], "unsqueez": 4, "sequenc": [4, 14, 16, 32, 33, 55, 56, 57, 58, 62, 70, 73, 74, 78, 81, 89, 98, 129, 137, 138, 140, 141, 143, 144, 146, 147, 151, 187, 189, 191, 204, 212, 218, 222, 223, 224, 226, 227, 228, 229, 232, 233, 237, 252, 254, 257, 259, 263, 270, 271, 274, 279, 284, 307, 308, 317, 323, 355, 370, 446, 457], "length": [4, 257, 307, 308, 317, 323, 355, 440], "len": [4, 138, 141, 144, 147, 157, 440], "overwrit": 4, "discard": [4, 289], "old": 4, "moment": [4, 426, 428, 429, 430], "anymor": 4, "everyth": [4, 450], "small": [4, 130, 132, 133, 307, 318, 324, 354, 394, 399, 404, 449, 450, 457], "12": [4, 157, 440], "8192": [4, 157], "1024": 4, "actual": [4, 17, 337, 422, 453], "materi": [4, 6], "could": [4, 303], "20_000": 4, "machin": [4, 6, 7, 436, 450], "8gb": 4, "ram": 4, "32": [4, 5, 88, 219, 220, 296, 306, 328, 354, 449], "44": 4, "bracket": 4, "becaus": [4, 195, 303, 453], "batch": [4, 13, 88, 152, 153, 188, 227, 307, 308, 309, 310, 312, 313, 317, 323, 350, 355, 371, 453], "zip": [4, 5], "haven": 4, "anyth": [4, 278, 453], "veri": [4, 350, 450, 453, 457], "similar": [4, 153, 165, 290, 348, 349, 350, 392, 454, 456], "runtim": [4, 117, 298, 449, 450], "section": [4, 7, 254, 404, 449, 450, 451], "origin": [4, 112, 288, 307, 343, 373, 374, 375, 376, 425, 426, 427, 428, 430, 431, 454], "sentencepiec": 4, "pytorch": [4, 6, 315, 318, 451], "compat": [4, 223, 227, 315, 456], "npz": [4, 178, 244, 245, 337, 341, 456], "file": [4, 7, 178, 241, 242, 243, 244, 245, 337, 341, 450, 451, 456], "argpars": 4, "itertool": [4, 290], "starmap": [4, 290], "np": [4, 5, 450, 454, 455], "torch": [4, 454], "map_torch_to_mlx": 4, "tok_embed": 4, "elif": 4, "attention_norm": 4, "ffn_norm": 4, "wq": 4, "wk": 4, "wv": 4, "wo": 4, "w1": [4, 288], "w2": [4, 288], "w3": 4, "ffn": 4, "separ": [4, 64, 77, 318, 400], "submodul": [4, 5, 303, 331, 335, 336, 347, 349], "feed_forward": 4, "parser": 4, "argumentpars": 4, "add_argu": 4, "torch_weight": 4, "output_fil": 4, "parse_arg": 4, "state": [4, 5, 303, 317, 323, 355, 424, 433, 446, 449], "savez": [4, 341, 456], "v": [4, 99, 135, 303, 335, 454], "disk": 4, "text": [4, 305, 306, 315, 317, 323, 327, 328, 329, 355, 368, 373, 374, 375, 376, 382, 385, 386, 387, 394, 395, 396, 399, 400, 403, 405, 406, 409, 410, 415, 416, 426, 431], "format": [4, 178, 241, 242, 243, 244, 245, 454], "dictionari": [4, 92, 178, 194, 242, 243, 288, 289, 292, 303, 334, 343, 348, 349, 434, 448, 456], "represent": [4, 219, 280, 289, 293], "tree_unflatten": [4, 424], "helper": [4, 449], "weight_fil": 4, "incur": [4, 7], "sever": [4, 7, 96, 97, 98, 244, 245, 449, 456], "futur": [4, 353, 452, 453], "pth": 4, "current": [4, 6, 7, 81, 88, 194, 196, 219, 292, 303, 426, 450, 453], "around": 4, "m1": [4, 449, 451, 457], "ultra": 4, "7b": 4, "me": 4, "ishmael": 4, "year": 4, "ago": 4, "never": [4, 453], "long": 4, "info": [4, 7], "247": 4, "press": [4, 173], "enter": 4, "littl": 4, "monei": 4, "my": [4, 7], "purs": 4, "consequ": 4, "walk": 4, "down": [4, 288], "gower": 4, "street": 4, "afternoon": 4, "heavi": 4, "rain": 4, "saw": [4, 451], "off": [4, 7, 453], "man": 4, "rag": 4, "who": 4, "sat": 4, "upon": [4, 290, 291], "hi": [4, 323], "bundl": 4, "hard": 4, "wet": 4, "he": [4, 375, 376], "were": [4, 457], "cry": 4, "watch": [4, 449], "him": 4, "observ": 4, "numer": [4, 132, 133, 173, 183, 187, 252, 307, 318, 322, 324, 354, 391, 392, 394, 404, 425, 426, 427, 428, 429, 430, 436, 449, 453], "crowd": 4, "wa": [4, 81, 450, 453], "hurri": 4, "437": 4, "330": 4, "second": [4, 7, 112, 167, 184, 186, 188, 238, 264, 273, 278, 306, 328, 392, 400, 426, 428, 429, 430, 451, 457], "spent": 4, "amount": [4, 197, 305, 327], "39": 4, "By": [4, 286, 342, 391, 451, 454], "bigger": [4, 426], "well": [4, 303, 335, 347, 350, 453], "almost": 4, "nobodi": 4, "took": 4, "least": [4, 82, 83, 84, 91, 170, 171, 172, 174, 175, 176, 219], "notic": [4, 451, 456], "distanc": [4, 404], "had": 4, "doubt": 4, "minut": 4, "straight": 4, "slowli": 4, "speak": [4, 173], "resum": 4, "postur": 4, "stood": 4, "feel": 4, "pain": 4, "heart": 4, "said": 4, "smile": 4, "face": 4, "am": 4, "someon": 4, "three": [4, 84, 371], "quarter": 4, "hour": 4, "made": [4, 298], "immedi": [4, 330], "repli": 4, "again": [4, 7, 303, 449], "hand": [4, 451, 453], "did": 4, "accustom": 4, "thu": [4, 303], "question": [4, 453], "reason": [4, 452], "tell": [4, 449, 454], "understand": [4, 373, 374], "579": 4, "690": 4, "500": [4, 457], "628": 4, "went": 4, "nervou": 4, "trembl": 4, "told": 4, "why": 4, "And": [4, 371], "surpris": 4, "matter": [4, 303], "shall": 4, "anyhow": 4, "friend": 4, "ye": 4, "slight": [4, 453], "kind": 4, "want": [4, 450, 451, 457], "longer": [4, 99, 451], "soon": 4, "unless": [4, 15, 160, 173, 422], "unlik": [4, 15, 160, 312, 313, 343], "strang": 4, "amus": 4, "That": 4, "secret": 4, "disappoint": 4, "mine": 4, "cannot": [4, 91, 452, 454], "happi": 4, "ask": [4, 450], "shop": 4, "bui": 4, "food": 4, "633": 4, "21": [4, 442], "475": 4, "su": 4, "j": [4, 7, 173, 312, 427, 428, 430], "lu": 4, "pan": 4, "murtadha": 4, "wen": 4, "liu": 4, "2021": 4, "roform": [4, 358], "enhanc": [4, 358, 453], "rotari": [4, 134, 358], "arxiv": [4, 318, 322, 324, 329, 354, 383, 405, 425, 431], "preprint": [4, 425, 431], "2104": 4, "09864": 4, "zhang": 4, "sennrich": 4, "2019": [4, 429], "advanc": [4, 449], "system": [4, 7, 194, 195, 196], "shazeer": 4, "2020": 4, "glu": [4, 303], "variant": [4, 403, 430], "2002": 4, "05202": 4, "classifi": 5, "mnist": 5, "As": [5, 37, 266, 303, 449], "mlp": [5, 303, 370, 424], "inherit": [5, 448], "idiom": [5, 449], "input_dim": [5, 303, 326, 353], "hidden_dim": [5, 422, 424], "output_dim": [5, 303, 326, 353], "layer_s": 5, "idim": 5, "odim": 5, "cross": [5, 98, 391, 393], "entropi": [5, 391, 393], "commonli": [5, 348, 419, 449], "cross_entropi": [5, 303], "accuraci": 5, "valid": [5, 99, 148, 282, 289, 335, 347, 448], "eval_fn": 5, "load": [5, 6, 298, 337, 450], "loader": 5, "num_class": [5, 424], "batch_siz": [5, 424], "num_epoch": [5, 424], "learning_r": [5, 424, 425, 426, 427, 428, 429, 430, 431, 433, 436, 437, 438, 439, 440, 441, 442, 449], "train_imag": [5, 424], "train_label": [5, 424], "test_imag": 5, "test_label": 5, "re": [5, 7, 419], "shuffl": 5, "minibatch": 5, "batch_iter": [5, 424], "perm": 5, "id": [5, 7], "put": [5, 449, 450], "trainabl": [5, 287, 303, 422], "loss_and_grad_fn": [5, 424, 449, 451], "value_and_grad": [5, 106, 303, 348, 422, 424, 435, 449, 451, 454, 455], "epoch": 5, "test": [5, 7, 450], "confus": 5, "decent": 5, "95": 5, "flexibl": 6, "brought": 6, "research": 6, "except": [6, 131, 142, 143, 145, 146, 147, 318, 337, 452, 454], "featur": [6, 96, 97, 98, 134, 307, 317, 318, 322, 323, 324, 326, 353, 354, 355, 358, 370, 371, 449, 453], "main": [6, 112, 131, 273, 290, 291, 303, 450], "differ": [6, 165, 262, 280, 403, 451], "compos": [6, 303, 449, 451, 455], "lazi": [6, 422, 455], "multi": [6, 135, 308, 309, 310, 452, 454], "cpu": [6, 7, 174, 449, 457], "inspir": 6, "jax": [6, 446], "arrayfir": 6, "unifi": 6, "live": [6, 457], "share": [6, 110, 153, 219, 220, 280, 450], "convers": 6, "distribut": [6, 7, 222, 223, 224, 226, 227, 228, 232, 233, 326, 373, 374, 375, 376, 378, 379, 394, 397, 402, 404, 419], "commun": [6, 114, 117, 118], "regress": [6, 399], "layer": [6, 132, 286, 303, 305, 306, 312, 313, 317, 318, 323, 324, 326, 327, 328, 344, 349, 352, 353, 355, 360, 370, 418, 422], "perceptron": 6, "llm": 6, "infer": [6, 151, 178, 273, 450], "fast": [6, 315, 383, 450, 457], "fft": 6, "algebra": 6, "tree": [6, 92, 127, 154, 278, 282, 289, 290, 291, 292, 293, 432, 433, 435, 444, 451], "custom": [6, 106, 370], "extens": [6, 178, 202, 341, 456], "debugg": 6, "pypi": 7, "own": [7, 454], "meet": 7, "seri": 7, "chip": 7, "nativ": 7, "maco": 7, "13": 7, "highli": 7, "recommend": [7, 201, 431], "14": 7, "sonoma": 7, "conda": [7, 450], "forg": 7, "match": [7, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 195, 337, 371, 393, 450, 452, 454], "probabl": [7, 229, 311, 312, 313, 353, 391, 393, 397, 457], "platform": 7, "processor": 7, "arm": 7, "i386": 7, "switch": 7, "17": 7, "clang": 7, "24": 7, "xcode": 7, "15": [7, 173, 292, 449], "sdk": 7, "environ": [7, 113, 123, 450], "via": [7, 106, 432, 435, 450, 453, 454], "rosetta": 7, "unam": 7, "p": [7, 222, 303, 311, 312, 313, 404, 428, 430], "clone": 7, "git": 7, "com": [7, 450], "ml": 7, "explor": 7, "cmake_build_parallel_level": 7, "edit": [7, 349], "unittest": 7, "discov": 7, "stub": 7, "generate_stub": 7, "either": [7, 12, 64, 76, 77, 85, 86, 87, 91, 119, 120, 124, 150, 155, 156, 167, 168, 169, 173, 183, 188, 190, 205, 207, 211, 217, 235, 238, 262, 278, 306, 328, 360, 371, 375, 376, 454], "static": 7, "libmlx": 7, "preprocessor": 7, "metal_path": 7, "mlx_build_test": 7, "mlx_build_exampl": 7, "mlx_build_benchmark": 7, "mlx_build_cpu": 7, "mlx_build_python_bind": 7, "mlx_build_safetensor": 7, "mlx_build_gguf": 7, "mlx_metal_jit": 7, "wish": 7, "variabl": [7, 92, 113, 123, 154, 166, 278, 281, 282, 450], "export": 7, "developer_dir": 7, "app": 7, "content": [7, 334, 449], "xcrun": 7, "macosx": 7, "show": [7, 296, 449], "cmake_build_typ": 7, "minsizerel": 7, "don": [7, 449, 457], "backend": [7, 117, 118], "safetensor": [7, 178, 243, 337, 341, 424, 453, 456], "gguf": [7, 178, 242, 456], "dcmake_build_typ": 7, "dbuild_shared_lib": 7, "dmlx_build_cpu": 7, "dmlx_build_safetensor": 7, "dmlx_build_gguf": 7, "dmlx_metal_jit": 7, "THE": 7, "pre": [7, 135, 391], "substanti": 7, "cold": 7, "cost": [7, 426, 450, 453], "anwywher": 7, "hundr": 7, "millisecond": [7, 449, 457], "persist": 7, "accross": 7, "reboot": 7, "unabl": 7, "tool": 7, "sudo": 7, "ouptut": 7, "finder": 7, "iterm": 7, "termin": 7, "click": 7, "uncheck": 7, "window": [7, 305, 306, 327, 328], "restart": 7, "grep": 7, "cmake_host_system_processor": 7, "arm64": 7, "x86_64": 7, "wipe": 7, "cahc": 7, "rf": 7, "devicetyp": 8, "attribut": [8, 9, 10, 29, 294, 343, 422, 444], "kwarg": [9, 114, 244, 245, 458], "categori": [10, 296], "bool_": [10, 296], "unsignedinteg": 10, "uint8": [10, 296], "uint16": [10, 296], "uint32": [10, 25, 26, 27, 28, 223, 296], "uint64": [10, 296], "signedinteg": [10, 165], "int8": [10, 296], "int64": [10, 296], "inexact": [10, 165], "complexflo": 10, "issubdtyp": [10, 296], "semant": [12, 85, 86, 87, 89, 119, 120, 124, 155, 156, 167, 168, 169, 183, 188, 190, 205, 207, 211, 217, 235, 238, 262, 457], "reduct": [14, 16, 116, 187, 189, 204, 218, 292, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 450], "unspecifi": [14, 16, 17, 25, 26, 27, 28, 93, 102, 103, 104, 105, 151, 187, 189, 191, 204, 212, 216, 218, 236, 252, 253, 259, 263, 266, 272, 273, 279, 284, 458], "entir": [14, 16, 25, 26, 187, 189, 191, 204, 218, 259, 263, 279, 312, 313], "05": [15, 160, 307, 318, 322, 324, 354], "08": [15, 160, 392, 427, 428, 429, 430, 436], "approxim": [15, 315, 381, 382, 383], "comparison": [15, 124, 155, 156, 168, 169, 211], "infinit": [15, 160], "rel": [15, 160, 426, 449], "interv": [17, 177, 229, 233], "increment": 17, "otherwis": [17, 98, 117, 201, 286, 289, 290, 291, 292, 335, 337, 347, 368, 370, 371, 385, 391, 396, 403, 415, 416, 453, 454], "convent": [17, 99, 121, 122, 371, 429], "fraction": 17, "integr": [17, 266, 453], "dimension": [29, 132, 133, 136, 137, 138, 139, 140, 141, 145, 146, 147, 305, 306, 307, 308, 309, 310, 314, 326, 327, 328, 352, 353, 362, 452, 454], "ndarrai": [29, 452, 453, 455], "properti": [30, 37, 46, 50, 60, 61, 67, 69, 343, 346, 434, 451], "argument": [30, 64, 77, 92, 127, 154, 278, 290, 291, 292, 303, 371, 446, 450, 451, 456, 457, 458], "union": [31, 32, 33, 34, 35, 36, 38, 39, 40, 41, 42, 43, 44, 45, 47, 48, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 63, 64, 65, 66, 68, 70, 71, 72, 73, 74, 75, 77, 78, 79, 82, 83, 84, 88, 121, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 157, 163, 164, 194, 231, 232, 242, 261, 273], "regular": [37, 312, 405, 429, 449, 452], "idx": [37, 452], "correctli": 37, "syntax": [37, 452], "datatyp": 50, "byte": [50, 60, 195, 196, 197, 200, 201, 296], "indices_or_sect": [70, 254], "nest": [76, 92, 292, 303, 422, 448, 451], "alwai": [81, 195, 289, 451], "regardless": [81, 135], "storag": 81, "caution": 81, "chang": [81, 92, 247, 280, 348, 353, 371, 396, 403, 449, 454], "crash": [81, 449], "begin": [81, 171, 197, 219, 306, 317, 323, 328, 368, 385, 396, 403, 409, 415, 416], "ari": [82, 83, 84], "xor": 87, "lhs_mask": 88, "lceil": 88, "rceil": 88, "rhs_mask": 88, "out_mask": 88, "edg": [91, 215, 371, 449], "At": 91, "fun": [92, 154, 166, 278, 281, 282, 449, 452, 453, 457], "callabl": [92, 106, 154, 166, 278, 281, 282, 286, 287, 289, 290, 291, 292, 330, 331, 334, 342, 355, 360, 370, 372, 373, 374, 375, 376, 377, 378, 379, 425, 426, 427, 428, 429, 430, 431, 436, 437, 438, 439, 440, 441, 442], "dict": [92, 127, 178, 194, 242, 243, 244, 288, 340, 345, 348, 349, 422, 424, 432, 433, 435, 448, 451, 456], "arbitrarili": [92, 303, 448, 451, 455], "leaf": [92, 286, 289, 290, 291, 292, 334], "node": [92, 127, 282, 291, 292], "recompil": [92, 449], "Not": [92, 211, 449], "attempt": 92, "elementwis": [94, 95], "alia": [94, 95, 315], "conj": 95, "channel": [96, 97, 98, 307, 308, 309, 310, 312, 313], "c_in": [96, 97, 98], "c_out": [96, 97, 98], "convolv": [96, 97, 98], "spatial": [97, 98, 305, 318, 327, 371], "symmetr": [97, 170, 171], "correl": [98, 312], "discret": [99, 136, 137, 138, 139, 140, 141, 145, 146, 147, 314, 352], "conv": 99, "signal": [99, 371], "th": [102, 103, 104, 105, 111, 131, 440], "definit": [106, 170, 171, 227], "meant": 106, "decor": 106, "behav": 106, "wrap": [106, 303], "behaviour": [106, 170, 171], "usag": [106, 370, 449], "f_vjp": 106, "f_jvp": 106, "dx": 106, "dy": 106, "f_vmap": 106, "ai": 106, "angl": [109, 221, 325], "configur": [110, 450], "formal": [110, 219], "notat": [110, 289, 339], "w_i": [110, 219], "hat": [110, 219], "occupi": [110, 153, 219, 220], "subarrai": [112, 254], "insert": [112, 129, 457], "end": [112, 171, 198, 219, 306, 317, 323, 328, 368, 385, 396, 403, 409, 415, 416, 438, 441], "taken": [112, 266, 273], "global": [113, 115, 116, 117, 123, 230, 288, 446, 449], "disabl": [113, 200, 449], "mlx_disable_compil": [113, 123, 449], "independ": [114, 312, 313], "particip": [115, 116], "strict": [117, 155, 168, 335, 337, 347], "is_avail": 117, "fuction": 120, "subscript": [121, 122], "operand": [121, 122, 152], "einstein": [121, 122], "summat": [121, 122], "equat": [121, 122], "einsum": 122, "chosen": 122, "mathrm": [125, 248, 322], "frac": [125, 219, 248, 305, 306, 307, 311, 312, 313, 318, 322, 324, 326, 327, 328, 354, 373, 374, 375, 376, 392, 394, 396, 399, 410, 412, 413, 425, 427, 428, 429, 430, 436], "pi": [125, 315, 362, 382, 451], "int_0": 125, "dt": 125, "minu": 130, "whose": [131, 286, 287], "translat": [132, 324], "stabil": [132, 133, 307, 318, 322, 324, 354, 391, 392, 394, 425, 426, 427, 428, 429, 430, 436], "rotat": [134, 358], "larger": [134, 358, 431], "unchang": [134, 260, 358], "consecut": [134, 219, 358], "angular": [134, 358], "frequenc": [134, 358, 362], "q": [135, 174], "head": [135, 350, 370], "attent": [135, 335, 350, 362, 370], "One": [136, 139, 145, 215, 240, 449, 451], "fourier": [136, 137, 138, 139, 140, 141, 145, 146, 147], "truncat": [136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 232], "dft": [136, 137, 138, 139, 140, 141, 145, 146, 147], "rfft": 142, "real": [142, 143, 144, 145, 146, 147, 170, 171], "rfft2": 143, "rfftn": 144, "silent": [145, 146, 147], "outsid": 148, "clamp": 148, "explicitli": [152, 303, 446], "flat": [152, 153, 289, 293], "a1": 152, "a2": 152, "AS": 152, "b1": 152, "b2": 152, "pack": [153, 219, 220], "unsign": [153, 219, 220, 296], "whether": [153, 176, 220, 317, 323, 334, 350, 355, 391, 394, 400], "argnam": [154, 278], "neither": [154, 278], "keyword": [154, 244, 245, 278, 290, 303, 446, 456, 458], "walsh": 157, "scipi": 157, "linalg": 157, "20": [157, 173], "28": 157, "16384": 157, "orthonorm": 157, "ordinari": 159, "inifn": 161, "dtypecategori": [165, 296], "subtyp": [165, 296], "subdtyp": 165, "float64": 165, "too": [165, 449, 453], "upper": [170, 171, 176, 219, 229, 232, 233, 379], "decomposit": [170, 171, 175], "semi": [170, 171, 227], "triangular": [170, 171, 176], "lower": [170, 171, 176, 219, 229, 232, 233, 275, 379], "dot": [170, 172, 176, 270, 289, 339, 350], "choleski": 171, "align": [171, 219, 306, 317, 323, 328], "mathbf": 171, "ainv": [172, 176], "ord": 173, "tabl": [173, 296, 314], "frobeniu": 173, "matric": [173, 174, 175], "strictli": 173, "mathemat": 173, "variou": 173, "purpos": 173, "calcul": [173, 391, 394, 400, 426], "fro": 173, "inf": [173, 208, 350], "largest": [173, 208, 272], "sing": 173, "smallest": 173, "singular": [173, 175], "nuclear": 173, "_f": 173, "sum_": [173, 305, 306, 399], "a_": 173, "valueerror": [173, 337, 451], "refer": [173, 322, 329, 343, 373, 374, 375, 376, 383, 405, 452], "golub": 173, "van": 173, "loan": 173, "baltimor": 173, "md": 173, "john": 173, "hopkin": 173, "univers": 173, "1985": 173, "pg": 173, "la": 173, "9": [173, 393, 425, 428, 429, 430, 431, 433, 439, 442, 454], "74597": 173, "84804": 173, "41421": 173, "23607": [173, 174], "74166": 173, "24264": 173, "11": 173, "225": 173, "894427": 174, "447214": 174, "57771": 174, "vt": 175, "return_metadata": 178, "binari": [178, 241, 242, 243, 244, 245, 280, 368, 391, 416, 449], "npy": [178, 241, 456], "matadata": 178, "unsupport": 178, "tensor": [178, 270, 305, 306, 327, 328, 404, 454], "stabl": [183, 187, 252, 399], "multidimension": 192, "dens": [192, 457], "cartesian": 192, "ij": 192, "clear": 193, "get_cache_memori": 193, "max_buffer_s": 194, "max_recommended_working_set_s": 194, "memory_s": 194, "report": [195, 201], "peak": [197, 199], "program": [197, 450], "reset_peak_memori": 197, "reset": 199, "free": 200, "reclaim": 200, "set_memory_limit": 200, "previou": [200, 201], "relax": 201, "task": [201, 399], "exceed": 201, "potenti": 201, "beforehand": 214, "constant_valu": 215, "before_1": 215, "after_1": 215, "before_2": 215, "after_2": 215, "before_n": 215, "after_n": 215, "before_i": 215, "after_i": 215, "side": [215, 305, 306, 327, 328, 449], "everi": [219, 290, 424, 442, 451], "particular": [219, 318], "w_1": 219, "w_g": 219, "max_i": 219, "min_i": 219, "textrm": [219, 315, 316, 381, 384], "1st": 219, "signific": 219, "2nd": 219, "w_q": 219, "prng": [222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 446], "num_sampl": 223, "unnorm": [223, 391, 393], "draw": 223, "cdf": [224, 315, 381], "seed": 225, "loc": [226, 228], "cov": 227, "jointli": 227, "covari": [227, 307], "behavior": [227, 399, 452, 453], "empti": 227, "low": [229, 233, 379, 419], "high": [229, 233, 303, 314, 379, 419], "roadcast": 229, "domain": [232, 450], "uniformli": 233, "repetit": 236, "preserv": [237, 451], "obj": 242, "uncompress": 244, "my_path": 244, "tree_flatten": [244, 290, 293, 303, 424], "transformerencod": 244, "128": [244, 303], "flat_param": 244, "compress": 245, "possibl": [254, 314, 352, 449, 450, 452, 457], "divisor": [259, 279], "being": [260, 303], "prevent": [260, 404, 454], "streamcontext": 261, "context": 261, "manag": [261, 446, 450, 457], "prior": [266, 267], "exclud": 267, "necessarili": 272, "elsewher": [275, 452], "col": 275, "triangl": 275, "mse": 278, "param": [278, 303, 419, 451], "lvalu": 278, "dlvalu": 278, "dparam": 278, "lasso": 278, "l1": [278, 396, 398, 399, 403], "impli": 280, "gaurante": 280, "in_ax": [282, 451], "out_ax": [282, 451], "prefix": [282, 289], "class_pred": 286, "predic": [286, 342], "to_quant": 286, "receiv": [286, 440, 454], "fn": [287, 290, 291, 292, 455], "wrt": 287, "max_norm": 288, "exce": 288, "proportion": 288, "clipped_grad": 288, "total_norm": 288, "rescal": 288, "is_leaf": [289, 290, 291, 292], "arbitrari": [289, 422], "depth": [289, 310, 313, 451], "hello": [289, 293], "charact": 289, "superset": [290, 432], "extra": [290, 291], "closer": 290, "decid": [290, 334], "constitut": 290, "dict_kei": [290, 433], "lambda": [290, 291, 292, 303, 319, 330, 335, 342, 366, 385, 409, 415, 425, 426, 427, 428, 429, 430, 431, 436, 437, 449, 450, 451], "tree_map": [291, 303, 450], "new_tre": 291, "accumul": [292, 354], "acc": 292, "recreat": [293, 424], "world": [293, 450], "42": 293, "16": [296, 305, 322, 327, 330, 422], "int16": 296, "brain": 296, "e8": 296, "m7": 296, "ieee": 296, "e5": 296, "m10": 296, "hierarchi": 296, "mpi": 298, "done": [303, 311, 354, 449, 450, 453, 454], "manual": 303, "solv": 303, "intuit": 303, "freez": [303, 347, 422], "finetun": 303, "in_dim": [303, 422], "out_dim": [303, 422], "enumer": 303, "caus": [303, 449, 453], "local": [303, 312, 450], "scope": 303, "l2_loss": 303, "y_hat": 303, "trainable_paramet": [303, 334, 433], "loss_and_grad": 303, "workhors": 303, "Its": 303, "recurs": [303, 334, 335, 340, 345, 347, 422], "frozen": [303, 335, 345, 347, 353, 422], "individu": [303, 312, 313], "subset": [303, 334], "action": 303, "displai": 303, "count": [303, 440], "num_param": 303, "preclud": 303, "pure": [303, 424], "pattern": [303, 453], "achiev": [303, 450], "other_input": 303, "necessari": 303, "apply_to_modul": [303, 335], "children": 303, "filter_and_map": 303, "leaf_modul": 303, "load_weight": [303, 453], "named_modul": 303, "save_weight": 303, "set_dtyp": 303, "unfreez": [303, 335], "update_modul": 303, "alibi": 303, "avgpool1d": 303, "avgpool2d": 303, "batchnorm": 303, "dropout": [303, 312, 313, 344, 370, 449], "dropout2d": 303, "dropout3d": 303, "gelu": [303, 382, 383, 449], "groupnorm": 303, "gru": 303, "hardshrink": [303, 385], "hardtanh": [303, 386], "hardswish": 303, "instancenorm": 303, "layernorm": 303, "leakyrelu": 303, "lstm": 303, "maxpool1d": 303, "maxpool2d": 303, "mish": 303, "prelu": 303, "quantizedembed": 303, "quantizedlinear": 303, "relu": [303, 351, 370, 406, 419], "relu6": 303, "rnn": [303, 317], "selu": 303, "sequenti": [303, 419], "silu": 303, "sinusoidalpositionalencod": 303, "softmin": 303, "softshrink": 303, "softsign": 303, "softplu": [303, 329, 405], "upsampl": 303, "elu": [303, 409], "gelu_approx": [303, 315, 381], "gelu_fast_approx": [303, 315, 381], "hard_shrink": [303, 319], "hard_tanh": [303, 320], "leaky_relu": 303, "log_sigmoid": 303, "log_softmax": 303, "binary_cross_entropi": [303, 449], "cosine_similarity_loss": 303, "gaussian_nll_loss": 303, "hinge_loss": 303, "huber_loss": 303, "kl_div_loss": 303, "l1_loss": 303, "log_cosh_loss": 303, "margin_ranking_loss": 303, "mse_loss": 303, "nll_loss": 303, "smooth_l1_loss": 303, "triplet_loss": 303, "init": [303, 351, 419, 424, 438, 439, 441, 442, 450], "glorot_norm": 303, "glorot_uniform": 303, "he_norm": 303, "he_uniform": 303, "kernel_s": [305, 306, 308, 309, 310, 327, 328], "averag": [305, 306, 425, 426, 428, 429, 430, 450], "pool": [305, 306, 327, 328, 457], "l_": [305, 327, 396], "n_i": [305, 306, 327, 328], "c_j": [305, 306, 327, 328], "ldot": [305, 306, 327, 328], "lfloor": [305, 306, 327, 328], "_size": [305, 306, 327, 328], "rfloor": [305, 306, 327, 328], "k_h": [306, 328], "k_w": [306, 328], "h_": [306, 317, 323, 328, 355], "w_": [306, 317, 323, 328, 355, 425, 426, 427, 428, 429, 430, 431, 436, 437], "height": [306, 307, 309, 310, 312, 313, 328], "width": [306, 307, 309, 310, 312, 313, 328, 352, 353], "momentum": [307, 431, 433, 437, 449], "affin": [307, 318, 322, 324, 326, 353], "track_running_stat": 307, "epsilon": [307, 318, 322, 324, 354, 392, 394, 425, 427, 428, 429, 430, 436], "gamma": [307, 318, 322, 324, 354, 373, 374, 375, 376], "nc": 307, "nlc": [307, 308], "four": 307, "nhwc": [307, 309], "paper": [307, 362, 425, 426, 427, 428, 430, 431], "deep": [307, 373, 374, 375, 376], "intern": 307, "bn": 307, "in_channel": [308, 309, 310], "out_channel": [308, 309, 310], "learnabl": [308, 309, 310, 360], "imag": [309, 310, 312, 313, 371], "ndhwc": [310, 313], "portion": 311, "nwhc": 312, "whc": 312, "maintain": [312, 313, 431], "benefici": [312, 313, 453], "earli": 312, "adjac": 312, "pixel": 312, "effect": [312, 449, 453], "thompson": 312, "goroshin": 312, "jain": 312, "lecun": 312, "bregler": 312, "2015": [312, 428, 430], "cvpr": 312, "dhwc": 313, "often": 313, "medic": 313, "video": 313, "num_embed": [314, 352], "lookup": 314, "usual": [314, 352, 448, 453], "vocabulari": [314, 352], "approx": 315, "unit": [315, 316, 317, 325, 356, 357, 359, 361, 373, 374, 375, 376, 380, 381, 382, 383, 384, 388, 407, 408, 409, 411], "phi": [315, 381], "geluapprox": 315, "044715": [315, 382], "gelufast": 315, "sigma": [315, 316, 317, 323, 373, 374, 375, 376, 383, 384, 389, 410, 411], "702": [315, 383], "regard": 315, "gate": [316, 317, 384], "halv": [316, 384], "input_s": [317, 323, 355], "hidden_s": [317, 323, 355], "recurr": [317, 323, 355], "nld": [317, 323, 355], "ld": [317, 323, 355], "r_t": 317, "xr": 317, "x_t": [317, 323, 355], "hr": 317, "h_t": [317, 323, 355], "b_": [317, 323], "z_t": 317, "xz": 317, "hz": 317, "n_t": 317, "xn": 317, "odot": [317, 323], "hn": 317, "hidden": [317, 323, 355, 370], "nh": [317, 323, 355], "nlh": [317, 323, 355], "lh": [317, 323, 355], "num_group": 318, "pytorch_compat": 318, "preced": 318, "http": [318, 322, 324, 329, 354, 383, 405], "org": [318, 322, 324, 329, 354, 383, 405], "1803": 318, "08494": 318, "lambd": [319, 366, 385, 415], "denomin": [322, 392, 425, 427, 428, 429, 430, 436], "inorm": 322, "1607": [322, 324], "08022": 322, "i_t": 323, "xi": 323, "f_t": 323, "xf": 323, "hf": 323, "g_t": [323, 425, 427, 428, 429, 430, 431, 436, 437], "xg": 323, "hg": 323, "o_t": 323, "xo": 323, "ho": 323, "c_": [323, 431], "c_t": [323, 431], "cell": 323, "06450": 324, "realli": 324, "negative_slop": [325, 388], "leaki": [325, 388], "rectifi": [325, 356, 357, 375, 376, 388, 407, 408], "slope": 325, "mathcal": 326, "d_i": 326, "max_": [327, 328], "1908": [329, 405], "08681": [329, 405], "map_fn": [330, 334], "filter_fn": [330, 334], "valid_parameter_filt": 330, "apply_fn": 331, "descend": 332, "is_leaf_fn": 334, "found": 334, "drop": 334, "idempot": [335, 347], "endswith": 335, "file_or_weight": 337, "miss": [337, 456], "ok": [337, 451], "save_safetensor": [341, 424, 456], "reflect": [343, 449, 452, 454], "ie": [347, 450], "noop": 347, "unfrozen": 347, "tracer": 348, "partial": [348, 349, 449, 453], "child": 349, "flexibli": 349, "programmat": 349, "query_input_dim": 350, "key_input_dim": 350, "value_input_dim": 350, "value_dim": 350, "value_output_dim": 350, "aggreg": 350, "linearli": 350, "attend": 350, "num_paramet": 351, "25": [351, 371], "parametr": [351, 406], "from_embed": 352, "classmethod": [352, 353], "from_linear": 353, "1910": 354, "07467": 354, "nonlinear": [355, 449], "elman": 355, "ih": 355, "hh": 355, "func": 355, "10000": 358, "slightli": [358, 457], "plain": 360, "known": [361, 411], "swish": [361, 411], "min_freq": 362, "0001": 362, "max_freq": 362, "cos_first": 362, "full_turn": 362, "sinusoid": 362, "threshold": [368, 396, 403, 416], "geq": [368, 416], "num_encoder_lay": 370, "num_decoder_lay": 370, "nb_func": 370, "custom_encod": 370, "custom_decod": 370, "norm_first": 370, "checkpoint": [370, 424], "decod": 370, "interact": 370, "mechan": 370, "chekpoint": 370, "expens": 370, "scale_factor": 371, "nearest": 371, "align_corn": 371, "audio": 371, "4d": 371, "forth": 371, "algorithm": [371, 431], "neighbor": 371, "interpol": 371, "cubic": 371, "bilinear": 371, "trilinear": 371, "bicub": 371, "corner": 371, "bottom": 371, "75": 371, "33333": 371, "66667": 371, "init_fn": [372, 373, 374, 375, 376, 377, 378, 379, 419], "glorot": [373, 374], "fan_in": [373, 374, 375, 376], "fan_out": [373, 374, 375, 376], "fan": [373, 374, 375, 376], "_in": [373, 374], "_out": [373, 374], "difficulti": [373, 374], "feedforward": [373, 374], "191107": 373, "61278": 373, "150594": 373, "363207": 373, "gain": [373, 374, 375, 376], "89613": 373, "53947": 373, "48095": 373, "995016": 373, "223404": 374, "890597": 374, "379159": 374, "776856": 374, "90041": 374, "02264": 374, "912766": 374, "12451": 374, "delv": [375, 376], "surpass": [375, 376], "human": [375, 376], "imagenet": [375, 376], "classif": [375, 376], "25211": 375, "458835": 375, "177208": 375, "0137595": 375, "6967": 375, "02765": 375, "15268": 375, "75787": 375, "kaim": 376, "0300242": 376, "0184009": 376, "793615": 376, "666329": 376, "64331": 376, "16506": 376, "08619": 376, "79854": 376, "982273": 378, "534422": 378, "380709": 378, "0645099": 378, "883935": 379, "863726": 379, "617261": 379, "417497": 379, "exact": [382, 383], "0005": 382, "015": 383, "cdot": [383, 392, 395, 411], "hendryck": 383, "1606": 383, "08415": 383, "min_val": 386, "max_val": 386, "_val": 386, "sum_i": 390, "x_i": [390, 412, 413], "with_logit": 391, "predict": [391, 394, 395, 396, 397, 398, 399, 401, 402, 403], "105361": 391, "223144": 391, "20397": 391, "916291": 391, "539245": 391, "prob": 391, "510826": 391, "x1": 392, "x2": 392, "x_1": [392, 400], "x_2": [392, 400], "label_smooth": 393, "hot": 393, "smooth": [393, 403, 436], "0485873": 393, "348587": 393, "06": [394, 404, 425], "likelihood": [394, 402], "nll": [394, 402], "hing": 395, "y_": [395, 399], "pred": [395, 399], "delta": [396, 425], "huber": 396, "leq": [396, 409], "l2": [396, 399, 437], "kullback": 397, "leibler": 397, "diverg": 397, "logcosh": 399, "sensit": 399, "outlier": 399, "dual": 399, "offer": 399, "balanc": 399, "robust": 399, "approach": [399, 451], "inputs1": 400, "inputs2": 400, "margin": [400, 404], "573409": 400, "765166": 400, "0638": 400, "75596": 400, "225763": 400, "256995": 400, "773433": 400, "formula": 403, "anchor": 404, "triplet": 404, "_p": 404, "pairwis": 404, "instabl": 404, "monoton": 405, "0507": 409, "67326": 409, "sum_j": [412, 413], "x_j": [412, 413], "subclass": 422, "concept": 422, "mymlp": 422, "in_proj": 422, "involv": [424, 449], "subsequ": 424, "far": 424, "serial": 424, "adam": [424, 430, 431, 440, 441], "configu": 424, "thumb": 424, "apply_gradi": 424, "rmsprop": 424, "adagrad": 424, "adafactor": 424, "adadelta": 424, "adamw": [424, 431], "adamax": 424, "lion": 424, "cosine_decai": [424, 440], "exponential_decai": 424, "join_schedul": 424, "linear_schedul": [424, 440], "step_decai": 424, "rho": 425, "zeiler": 425, "2012": [425, 436], "adapt": [425, 426, 427, 450], "1212": 425, "5701": 425, "v_": [425, 427, 428, 429, 430, 436, 437], "v_t": [425, 427, 428, 429, 430, 436, 437], "u_t": 425, "u_": 425, "w_t": [425, 427, 428, 429, 430, 431, 436, 437], "30": 426, "001": 426, "clip_threshold": 426, "decay_r": [426, 439, 442], "beta_1": [426, 428, 429, 430, 431], "weight_decai": [426, 429, 431, 437], "scale_paramet": 426, "relative_step": 426, "warmup_init": 426, "sublinear": 426, "epsilon_1": 426, "epsilon_2": 426, "parameter_scal": 426, "unscal": 426, "decai": [426, 429, 431, 437, 438, 439, 442], "duchi": 427, "hazan": 427, "singer": 427, "2011": 427, "subgradi": 427, "onlin": 427, "stochast": [427, 428, 430, 437, 453], "jmlr": 427, "999": [428, 429, 430], "omit": [428, 430, 450], "estim": [428, 430], "kingma": [428, 430], "ba": [428, 430], "iclr": [428, 429, 430], "m_": [428, 429, 430, 431], "m_t": [428, 429, 430, 431], "beta_2": [428, 429, 430, 431], "contrast": 429, "loshchilov": 429, "hutter": 429, "decoupl": 429, "99": [431, 436], "tend": 431, "10x": 431, "strength": [431, 437], "wd": 431, "chen": 431, "symbol": 431, "discoveri": 431, "2302": 431, "06675": 431, "eta": 431, "opt": [432, 450], "tieleman": 436, "hinton": 436, "lectur": 436, "coursera": 436, "dampen": 437, "nesterov": 437, "descent": [437, 449, 453], "mu": 437, "tau": 437, "penalti": 437, "decay_step": 438, "beyond": [438, 441], "lr_schedul": [438, 439, 440, 442], "1000": [438, 449], "0999961": 438, "06561": 439, "boundari": 440, "join": 440, "transit": 440, "warmup": [440, 441], "0999938": 440, "101": 441, "step_siz": 442, "081": 442, "basi": 444, "implicit": [446, 449, 451], "fine": [446, 453], "grain": 446, "pseudo": 446, "altern": 446, "splittabl": 446, "threefri": 446, "counter": 446, "cycl": 448, "merg": 449, "fuse": 449, "big": 449, "awar": [449, 453], "36788": 449, "compiled_fun": 449, "code": [449, 450, 453], "slow": 449, "rerun": [449, 453], "frequent": [449, 453], "destroi": 449, "anonym": 449, "unari": 449, "overhead": [449, 453, 457], "bandwidth": [449, 450], "fusibl": 449, "consider": 449, "versu": 449, "timeit": [449, 451], "tic": 449, "perf_count": 449, "toc": 449, "tpi": 449, "1e3": 449, "4096": [449, 451, 457], "On": [449, 451, 453], "five": 449, "latest": 449, "won": 449, "placehold": 449, "insid": 449, "disable_compil": 449, "okai": [449, 453], "deal": 449, "pretti": [449, 453], "inconveni": 449, "functool": 449, "particularli": 449, "backward": [449, 451], "compiled_grad_fn": 449, "71828": 449, "opportun": 449, "physic": 450, "lot": [450, 451], "figur": 450, "best": 450, "mac": 450, "abil": 450, "talk": 450, "minim": 450, "all_sum": 450, "mpirun": 450, "mpiexec": 450, "send": 450, "homebrew": 450, "anaconda": 450, "openmpi": 450, "libmpi": 450, "dyld": 450, "dyld_library_path": 450, "lib": 450, "connect": 450, "ssh": 450, "checklist": 450, "issu": [450, 451, 454], "hostnam": 450, "password": 450, "confirm": 450, "foo": 450, "bar": 450, "host1": 450, "host2": 450, "qualifi": 450, "ip": 450, "slot": 450, "hostfil": 450, "parallel": [450, 457], "snippet": 450, "loss_grad_fn": 450, "all_avg": 450, "all_reduce_grad": 450, "latenc": 450, "mca": 450, "btl_tcp_link": 450, "tcp": 450, "idea": [451, 453], "behind": 451, "dfdx": [451, 452], "d2fdx2": 451, "zero_grad": 451, "detach": 451, "requires_grad": 451, "dloss_dw": 451, "dloss_dx": 451, "redund": 451, "continu": 451, "suppos": [451, 457], "nice": [451, 453], "propag": [451, 452], "autom": 451, "contriv": [451, 457], "sake": 451, "clariti": 451, "quit": [451, 454], "difficult": 451, "primit": 451, "priorit": 451, "naive_add": 451, "vmap_add": 451, "total": 451, "390": 451, "wherea": 451, "025": 451, "ten": [451, 453], "Of": 451, "cours": 451, "better": [451, 457], "handi": 451, "ellipsi": 452, "mix": 452, "lack": 452, "extrem": [452, 453], "ineffici": [452, 453], "nonzero": 452, "dynam": 453, "easier": 453, "worri": 453, "fun1": 453, "expensive_fun": 453, "consum": 453, "eager": 453, "thank": 453, "weights_fp16": 453, "trade": 453, "bad": 453, "grow": 453, "computation": 453, "costli": 453, "luckili": 453, "wide": 453, "thousand": 453, "value_and_grad_fn": 453, "implicitli": 453, "anytim": 453, "memoryview": [453, 454], "perfectli": 453, "first_lay": 453, "second_layer_a": 453, "second_layer_b": 453, "protocol": 454, "dlpack": 454, "pep": 454, "3118": 454, "a_view": 454, "owndata": 454, "extern": 454, "demonstr": 454, "x_view": 454, "modifi": 454, "df": 454, "x\u00b2": 454, "2x": 454, "indirectli": 454, "modif": 454, "seen": 454, "occur": 454, "incorpor": 454, "aris": 454, "incorrect": 454, "experiment": 454, "break": 454, "advis": 454, "intermedi": 454, "jnp": 454, "tf": 454, "page": 455, "composit": 455, "archiv": 456, "savez_compress": 456, "save_gguf": 456, "arr_0": 456, "advantag": 457, "race": 457, "interest": 457, "albeit": 457, "d1": 457, "d2": 457, "But": 457, "twice": 457, "measur": 457, "default_stream": 458, "default_devic": 458, "my_devic": 458}, "objects": {"": [[0, 0, 1, "_CPPv43absRK5array14StreamOrDevice", "abs"], [0, 1, 1, "_CPPv43absRK5array14StreamOrDevice", "abs::a"], [0, 1, 1, "_CPPv43absRK5array14StreamOrDevice", "abs::s"], [0, 0, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::a"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::b"], [0, 1, 1, "_CPPv43addRK5arrayRK5array14StreamOrDevice", "add::s"], [0, 0, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::a"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::alpha"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::b"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::beta"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::c"], [0, 1, 1, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", "addmm::s"], [0, 0, 1, "_CPPv43allRK5array14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all"], [0, 0, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all"], [0, 1, 1, "_CPPv43allRK5array14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::a"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::axes"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::axis"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::keepdims"], [0, 1, 1, "_CPPv43allRK5array14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayb14StreamOrDevice", "all::s"], [0, 1, 1, "_CPPv43allRK5arrayib14StreamOrDevice", "all::s"], [0, 0, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::a"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::atol"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::b"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::equal_nan"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::rtol"], [0, 1, 1, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", "allclose::s"], [0, 0, 1, "_CPPv43anyRK5array14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any"], [0, 0, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any"], [0, 1, 1, "_CPPv43anyRK5array14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::a"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::axes"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::axis"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::keepdims"], [0, 1, 1, "_CPPv43anyRK5array14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayb14StreamOrDevice", "any::s"], [0, 1, 1, "_CPPv43anyRK5arrayib14StreamOrDevice", "any::s"], [0, 0, 1, "_CPPv46aranged14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangedd14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeddd14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangei14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeii14StreamOrDevice", "arange"], [0, 0, 1, "_CPPv46arangeiii14StreamOrDevice", "arange"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::dtype"], [0, 1, 1, "_CPPv46aranged14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangei14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::s"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::start"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::step"], [0, 1, 1, "_CPPv46aranged14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46aranged5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangedd14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangedd5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeddd14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeddd5Dtype14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangei14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeii14StreamOrDevice", "arange::stop"], [0, 1, 1, "_CPPv46arangeiii14StreamOrDevice", "arange::stop"], [0, 0, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos"], [0, 1, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos::a"], [0, 1, 1, "_CPPv46arccosRK5array14StreamOrDevice", "arccos::s"], [0, 0, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh"], [0, 1, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh::a"], [0, 1, 1, "_CPPv47arccoshRK5array14StreamOrDevice", "arccosh::s"], [0, 0, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin"], [0, 1, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin::a"], [0, 1, 1, "_CPPv46arcsinRK5array14StreamOrDevice", "arcsin::s"], [0, 0, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh"], [0, 1, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh::a"], [0, 1, 1, "_CPPv47arcsinhRK5array14StreamOrDevice", "arcsinh::s"], [0, 0, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan"], [0, 0, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::a"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::b"], [0, 1, 1, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", "arctan2::s"], [0, 1, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan::a"], [0, 1, 1, "_CPPv46arctanRK5array14StreamOrDevice", "arctan::s"], [0, 0, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh"], [0, 1, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh::a"], [0, 1, 1, "_CPPv47arctanhRK5array14StreamOrDevice", "arctanh::s"], [0, 0, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax"], [0, 0, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax"], [0, 0, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax"], [0, 1, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::a"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::axis"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::keepdims"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::keepdims"], [0, 1, 1, "_CPPv46argmaxRK5array14StreamOrDevice", "argmax::s"], [0, 1, 1, "_CPPv46argmaxRK5arrayb14StreamOrDevice", "argmax::s"], [0, 1, 1, "_CPPv46argmaxRK5arrayib14StreamOrDevice", "argmax::s"], [0, 0, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin"], [0, 0, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin"], [0, 0, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin"], [0, 1, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::a"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::axis"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::keepdims"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::keepdims"], [0, 1, 1, "_CPPv46argminRK5array14StreamOrDevice", "argmin::s"], [0, 1, 1, "_CPPv46argminRK5arrayb14StreamOrDevice", "argmin::s"], [0, 1, 1, "_CPPv46argminRK5arrayib14StreamOrDevice", "argmin::s"], [0, 0, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition"], [0, 0, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::a"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::a"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::axis"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::kth"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::kth"], [0, 1, 1, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", "argpartition::s"], [0, 1, 1, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", "argpartition::s"], [0, 0, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort"], [0, 0, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort"], [0, 1, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort::a"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::a"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::axis"], [0, 1, 1, "_CPPv47argsortRK5array14StreamOrDevice", "argsort::s"], [0, 1, 1, "_CPPv47argsortRK5arrayi14StreamOrDevice", "argsort::s"], [0, 0, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal"], [0, 0, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::a"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::a"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::b"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::b"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::equal_nan"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", "array_equal::s"], [0, 1, 1, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", "array_equal::s"], [0, 0, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::a"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::offset"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::s"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::shape"], [0, 1, 1, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice", "as_strided::strides"], [0, 0, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::a"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::dtype"], [0, 1, 1, "_CPPv46astype5array5Dtype14StreamOrDevice", "astype::s"], [0, 0, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d"], [0, 0, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d"], [0, 1, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d::a"], [0, 1, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d::a"], [0, 1, 1, "_CPPv410atleast_1dRK5array14StreamOrDevice", "atleast_1d::s"], [0, 1, 1, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_1d::s"], [0, 0, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d"], [0, 0, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d"], [0, 1, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d::a"], [0, 1, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d::a"], [0, 1, 1, "_CPPv410atleast_2dRK5array14StreamOrDevice", "atleast_2d::s"], [0, 1, 1, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_2d::s"], [0, 0, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d"], [0, 0, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d"], [0, 1, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d::a"], [0, 1, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d::a"], [0, 1, 1, "_CPPv410atleast_3dRK5array14StreamOrDevice", "atleast_3d::s"], [0, 1, 1, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", "atleast_3d::s"], [0, 0, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::a"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::b"], [0, 1, 1, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", "bitwise_and::s"], [0, 0, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::a"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::b"], [0, 1, 1, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", "bitwise_or::s"], [0, 0, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::a"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::b"], [0, 1, 1, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", "bitwise_xor::s"], [0, 0, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::a"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::b"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::block_size"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_lhs"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_out"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::mask_rhs"], [0, 1, 1, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "block_masked_mm::s"], [0, 0, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays"], [0, 1, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays::inputs"], [0, 1, 1, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", "broadcast_arrays::s"], [0, 0, 1, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "broadcast_to"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "broadcast_to::a"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "broadcast_to::s"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "broadcast_to::shape"], [0, 0, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil"], [0, 1, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil::a"], [0, 1, 1, "_CPPv44ceilRK5array14StreamOrDevice", "ceil::s"], [0, 0, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a_max"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::a_min"], [0, 1, 1, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", "clip::s"], [0, 0, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice", "concatenate"], [0, 0, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::axis"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::s"], [0, 1, 1, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::s"], [0, 0, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::a"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::s"], [0, 0, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::dilation"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::groups"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::input"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::padding"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::s"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::stride"], [0, 1, 1, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv1d::weight"], [0, 0, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::dilation"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::groups"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::input"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::padding"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::s"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::stride"], [0, 1, 1, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv2d::weight"], [0, 0, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::dilation"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::groups"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::input"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::padding"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::s"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::stride"], [0, 1, 1, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv3d::weight"], [0, 0, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general"], [0, 0, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::flip"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::flip"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::groups"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::groups"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::input_dilation"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::kernel_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::kernel_dilation"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding_hi"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::padding_lo"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::s"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::s"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::stride"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::stride"], [0, 1, 1, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::weight"], [0, 1, 1, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", "conv_general::weight"], [0, 0, 1, "_CPPv44copy5array14StreamOrDevice", "copy"], [0, 1, 1, "_CPPv44copy5array14StreamOrDevice", "copy::a"], [0, 1, 1, "_CPPv44copy5array14StreamOrDevice", "copy::s"], [0, 0, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos"], [0, 1, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos::a"], [0, 1, 1, "_CPPv43cosRK5array14StreamOrDevice", "cos::s"], [0, 0, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh"], [0, 1, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh::a"], [0, 1, 1, "_CPPv44coshRK5array14StreamOrDevice", "cosh::s"], [0, 0, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::a"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::axis"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::inclusive"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::reverse"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::s"], [0, 0, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::a"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::axis"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::inclusive"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::reverse"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::s"], [0, 0, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::a"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::axis"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::inclusive"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::reverse"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::s"], [0, 0, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::a"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::axis"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::inclusive"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::reverse"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::s"], [0, 0, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees"], [0, 1, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees::a"], [0, 1, 1, "_CPPv47degreesRK5array14StreamOrDevice", "degrees::s"], [0, 0, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends"], [0, 1, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends::dependencies"], [0, 1, 1, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", "depends::inputs"], [0, 0, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::biases"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::bits"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::group_size"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::s"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::scales"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", "dequantize::w"], [0, 0, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::a"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::k"], [0, 1, 1, "_CPPv44diagRK5arrayi14StreamOrDevice", "diag::s"], [0, 0, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::a"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::axis1"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::axis2"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::offset"], [0, 1, 1, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", "diagonal::s"], [0, 0, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::a"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::b"], [0, 1, 1, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", "divide::s"], [0, 0, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::a"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::b"], [0, 1, 1, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", "divmod::s"], [0, 0, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::a"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::b"], [0, 1, 1, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", "equal::s"], [0, 0, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf"], [0, 1, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf::a"], [0, 1, 1, "_CPPv43erfRK5array14StreamOrDevice", "erf::s"], [0, 0, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv"], [0, 1, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv::a"], [0, 1, 1, "_CPPv46erfinvRK5array14StreamOrDevice", "erfinv::s"], [0, 0, 1, "_CPPv43expRK5array14StreamOrDevice", "exp"], [0, 1, 1, "_CPPv43expRK5array14StreamOrDevice", "exp::a"], [0, 1, 1, "_CPPv43expRK5array14StreamOrDevice", "exp::s"], [0, 0, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims"], [0, 0, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::a"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::a"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::axes"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::axis"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "expand_dims::s"], [0, 1, 1, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", "expand_dims::s"], [0, 0, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1"], [0, 1, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1::a"], [0, 1, 1, "_CPPv45expm1RK5array14StreamOrDevice", "expm1::s"], [0, 0, 1, "_CPPv43eyei14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeii14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeiii14StreamOrDevice", "eye"], [0, 0, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::dtype"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::dtype"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::k"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::k"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::m"], [0, 1, 1, "_CPPv43eyei14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::n"], [0, 1, 1, "_CPPv43eyei14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyei5Dtype14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeii14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeiii14StreamOrDevice", "eye::s"], [0, 1, 1, "_CPPv43eyeiii5Dtype14StreamOrDevice", "eye::s"], [0, 0, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten"], [0, 0, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten"], [0, 1, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten::a"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::a"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::end_axis"], [0, 1, 1, "_CPPv47flattenRK5array14StreamOrDevice", "flatten::s"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::s"], [0, 1, 1, "_CPPv47flattenRK5arrayii14StreamOrDevice", "flatten::start_axis"], [0, 0, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor"], [0, 1, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor::a"], [0, 1, 1, "_CPPv45floorRK5array14StreamOrDevice", "floor::s"], [0, 0, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::a"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::b"], [0, 1, 1, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", "floor_divide::s"], [0, 0, 1, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice", "full"], [0, 0, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full"], [0, 2, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full::T"], [0, 2, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::T"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice", "full::vals"], [0, 1, 1, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice", "full::vals"], [0, 0, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather"], [0, 0, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::axes"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::axis"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "gather::slice_sizes"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "gather::slice_sizes"], [0, 0, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::a"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::b"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::lhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::rhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", "gather_mm::s"], [0, 0, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::biases"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::bits"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::group_size"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::lhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::rhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::s"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::scales"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::transpose"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::w"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", "gather_qmm::x"], [0, 0, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::a"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::b"], [0, 1, 1, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", "greater::s"], [0, 0, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::a"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::b"], [0, 1, 1, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", "greater_equal::s"], [0, 0, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::a"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::s"], [0, 1, 1, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", "hadamard_transform::scale"], [0, 0, 1, "_CPPv48identityi14StreamOrDevice", "identity"], [0, 0, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::dtype"], [0, 1, 1, "_CPPv48identityi14StreamOrDevice", "identity::n"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::n"], [0, 1, 1, "_CPPv48identityi14StreamOrDevice", "identity::s"], [0, 1, 1, "_CPPv48identityi5Dtype14StreamOrDevice", "identity::s"], [0, 0, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::a"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::b"], [0, 1, 1, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", "inner::s"], [0, 0, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::a"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::atol"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::b"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::equal_nan"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::rtol"], [0, 1, 1, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", "isclose::s"], [0, 0, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf"], [0, 1, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf::a"], [0, 1, 1, "_CPPv45isinfRK5array14StreamOrDevice", "isinf::s"], [0, 0, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan"], [0, 1, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan::a"], [0, 1, 1, "_CPPv45isnanRK5array14StreamOrDevice", "isnan::s"], [0, 0, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf"], [0, 1, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf::a"], [0, 1, 1, "_CPPv48isneginfRK5array14StreamOrDevice", "isneginf::s"], [0, 0, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf"], [0, 1, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf::a"], [0, 1, 1, "_CPPv48isposinfRK5array14StreamOrDevice", "isposinf::s"], [0, 0, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::a"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::b"], [0, 1, 1, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", "left_shift::s"], [0, 0, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::a"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::b"], [0, 1, 1, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", "less::s"], [0, 0, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::a"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::b"], [0, 1, 1, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", "less_equal::s"], [0, 0, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::dtype"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::num"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::s"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::start"], [0, 1, 1, "_CPPv48linspaceddi5Dtype14StreamOrDevice", "linspace::stop"], [0, 0, 1, "_CPPv43logRK5array14StreamOrDevice", "log"], [0, 0, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10"], [0, 1, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10::a"], [0, 1, 1, "_CPPv45log10RK5array14StreamOrDevice", "log10::s"], [0, 0, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p"], [0, 1, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p::a"], [0, 1, 1, "_CPPv45log1pRK5array14StreamOrDevice", "log1p::s"], [0, 0, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2"], [0, 1, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2::a"], [0, 1, 1, "_CPPv44log2RK5array14StreamOrDevice", "log2::s"], [0, 1, 1, "_CPPv43logRK5array14StreamOrDevice", "log::a"], [0, 1, 1, "_CPPv43logRK5array14StreamOrDevice", "log::s"], [0, 0, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::a"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::b"], [0, 1, 1, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", "logaddexp::s"], [0, 0, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::a"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::b"], [0, 1, 1, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", "logical_and::s"], [0, 0, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not"], [0, 1, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not::a"], [0, 1, 1, "_CPPv411logical_notRK5array14StreamOrDevice", "logical_not::s"], [0, 0, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::a"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::b"], [0, 1, 1, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", "logical_or::s"], [0, 0, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp"], [0, 0, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp"], [0, 1, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::a"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::axes"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::axis"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::keepdims"], [0, 1, 1, "_CPPv49logsumexpRK5array14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", "logsumexp::s"], [0, 1, 1, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", "logsumexp::s"], [0, 0, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::a"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::b"], [0, 1, 1, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", "matmul::s"], [0, 0, 1, "_CPPv43maxRK5array14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max"], [0, 0, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max"], [0, 1, 1, "_CPPv43maxRK5array14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::a"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::axes"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::axis"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::keepdims"], [0, 1, 1, "_CPPv43maxRK5array14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayb14StreamOrDevice", "max::s"], [0, 1, 1, "_CPPv43maxRK5arrayib14StreamOrDevice", "max::s"], [0, 0, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::a"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::b"], [0, 1, 1, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", "maximum::s"], [0, 0, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean"], [0, 0, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean"], [0, 1, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::a"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::axes"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::axis"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::keepdims"], [0, 1, 1, "_CPPv44meanRK5array14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayb14StreamOrDevice", "mean::s"], [0, 1, 1, "_CPPv44meanRK5arrayib14StreamOrDevice", "mean::s"], [0, 0, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::arrays"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::indexing"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::s"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice", "meshgrid::sparse"], [0, 0, 1, "_CPPv43minRK5array14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min"], [0, 0, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min"], [0, 1, 1, "_CPPv43minRK5array14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::a"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::axes"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::axis"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::keepdims"], [0, 1, 1, "_CPPv43minRK5array14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayb14StreamOrDevice", "min::s"], [0, 1, 1, "_CPPv43minRK5arrayib14StreamOrDevice", "min::s"], [0, 0, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::a"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::b"], [0, 1, 1, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", "minimum::s"], [0, 0, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::a"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::destination"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::s"], [0, 1, 1, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", "moveaxis::source"], [0, 0, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::a"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::b"], [0, 1, 1, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", "multiply::s"], [0, 0, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::a"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::nan"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::neginf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::posinf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::s"], [0, 0, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative"], [0, 1, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative::a"], [0, 1, 1, "_CPPv48negativeRK5array14StreamOrDevice", "negative::s"], [0, 0, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::a"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::b"], [0, 1, 1, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", "not_equal::s"], [0, 0, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::a"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::axes"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::dtype"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::inverted"], [0, 1, 1, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", "number_of_elements::s"], [0, 0, 1, "_CPPv44onesRKNSt6vectorIiEE14StreamOrDevice", "ones"], [0, 0, 1, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice", "ones"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice", "ones::dtype"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE14StreamOrDevice", "ones::shape"], [0, 1, 1, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice", "ones::shape"], [0, 0, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like"], [0, 1, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like::a"], [0, 1, 1, "_CPPv49ones_likeRK5array14StreamOrDevice", "ones_like::s"], [0, 0, 1, "_CPPv4I0Ene5array1TRK5array", "operator!="], [0, 0, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!="], [0, 0, 1, "_CPPv4neRK5arrayRK5array", "operator!="], [0, 2, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::T"], [0, 2, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::T"], [0, 1, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::a"], [0, 1, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::a"], [0, 1, 1, "_CPPv4neRK5arrayRK5array", "operator!=::a"], [0, 1, 1, "_CPPv4I0Ene5array1TRK5array", "operator!=::b"], [0, 1, 1, "_CPPv4I0Ene5arrayRK5array1T", "operator!=::b"], [0, 1, 1, "_CPPv4neRK5arrayRK5array", "operator!=::b"], [0, 0, 1, "_CPPv4I0Erm5array1TRK5array", "operator%"], [0, 0, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%"], [0, 0, 1, "_CPPv4rmRK5arrayRK5array", "operator%"], [0, 2, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::T"], [0, 2, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::T"], [0, 1, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::a"], [0, 1, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::a"], [0, 1, 1, "_CPPv4rmRK5arrayRK5array", "operator%::a"], [0, 1, 1, "_CPPv4I0Erm5array1TRK5array", "operator%::b"], [0, 1, 1, "_CPPv4I0Erm5arrayRK5array1T", "operator%::b"], [0, 1, 1, "_CPPv4rmRK5arrayRK5array", "operator%::b"], [0, 0, 1, "_CPPv4anRK5arrayRK5array", "operator&"], [0, 0, 1, "_CPPv4aaRK5arrayRK5array", "operator&&"], [0, 1, 1, "_CPPv4aaRK5arrayRK5array", "operator&&::a"], [0, 1, 1, "_CPPv4aaRK5arrayRK5array", "operator&&::b"], [0, 1, 1, "_CPPv4anRK5arrayRK5array", "operator&::a"], [0, 1, 1, "_CPPv4anRK5arrayRK5array", "operator&::b"], [0, 0, 1, "_CPPv4I0Eml5array1TRK5array", "operator*"], [0, 0, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*"], [0, 0, 1, "_CPPv4mlRK5arrayRK5array", "operator*"], [0, 2, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::T"], [0, 2, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::T"], [0, 1, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::a"], [0, 1, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::a"], [0, 1, 1, "_CPPv4mlRK5arrayRK5array", "operator*::a"], [0, 1, 1, "_CPPv4I0Eml5array1TRK5array", "operator*::b"], [0, 1, 1, "_CPPv4I0Eml5arrayRK5array1T", "operator*::b"], [0, 1, 1, "_CPPv4mlRK5arrayRK5array", "operator*::b"], [0, 0, 1, "_CPPv4I0Epl5array1TRK5array", "operator+"], [0, 0, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+"], [0, 0, 1, "_CPPv4plRK5arrayRK5array", "operator+"], [0, 2, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::T"], [0, 2, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::T"], [0, 1, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::a"], [0, 1, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::a"], [0, 1, 1, "_CPPv4plRK5arrayRK5array", "operator+::a"], [0, 1, 1, "_CPPv4I0Epl5array1TRK5array", "operator+::b"], [0, 1, 1, "_CPPv4I0Epl5arrayRK5array1T", "operator+::b"], [0, 1, 1, "_CPPv4plRK5arrayRK5array", "operator+::b"], [0, 0, 1, "_CPPv4I0Emi5array1TRK5array", "operator-"], [0, 0, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-"], [0, 0, 1, "_CPPv4miRK5array", "operator-"], [0, 0, 1, "_CPPv4miRK5arrayRK5array", "operator-"], [0, 2, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::T"], [0, 2, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::T"], [0, 1, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::a"], [0, 1, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::a"], [0, 1, 1, "_CPPv4miRK5array", "operator-::a"], [0, 1, 1, "_CPPv4miRK5arrayRK5array", "operator-::a"], [0, 1, 1, "_CPPv4I0Emi5array1TRK5array", "operator-::b"], [0, 1, 1, "_CPPv4I0Emi5arrayRK5array1T", "operator-::b"], [0, 1, 1, "_CPPv4miRK5arrayRK5array", "operator-::b"], [0, 0, 1, "_CPPv4dvRK5arrayRK5array", "operator/"], [0, 0, 1, "_CPPv4dvRK5arrayd", "operator/"], [0, 0, 1, "_CPPv4dvdRK5array", "operator/"], [0, 1, 1, "_CPPv4dvRK5arrayRK5array", "operator/::a"], [0, 1, 1, "_CPPv4dvRK5arrayd", "operator/::a"], [0, 1, 1, "_CPPv4dvdRK5array", "operator/::a"], [0, 1, 1, "_CPPv4dvRK5arrayRK5array", "operator/::b"], [0, 1, 1, "_CPPv4dvRK5arrayd", "operator/::b"], [0, 1, 1, "_CPPv4dvdRK5array", "operator/::b"], [0, 0, 1, "_CPPv4I0Elt5array1TRK5array", "operator<"], [0, 0, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<"], [0, 0, 1, "_CPPv4ltRK5arrayRK5array", "operator<"], [0, 2, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::T"], [0, 2, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::T"], [0, 1, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::a"], [0, 1, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::a"], [0, 1, 1, "_CPPv4ltRK5arrayRK5array", "operator<::a"], [0, 1, 1, "_CPPv4I0Elt5array1TRK5array", "operator<::b"], [0, 1, 1, "_CPPv4I0Elt5arrayRK5array1T", "operator<::b"], [0, 1, 1, "_CPPv4ltRK5arrayRK5array", "operator<::b"], [0, 0, 1, "_CPPv4lsRK5arrayRK5array", "operator<<"], [0, 1, 1, "_CPPv4lsRK5arrayRK5array", "operator<<::a"], [0, 1, 1, "_CPPv4lsRK5arrayRK5array", "operator<<::b"], [0, 0, 1, "_CPPv4I0Ele5array1TRK5array", "operator<="], [0, 0, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<="], [0, 0, 1, "_CPPv4leRK5arrayRK5array", "operator<="], [0, 2, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::T"], [0, 2, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::T"], [0, 1, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::a"], [0, 1, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::a"], [0, 1, 1, "_CPPv4leRK5arrayRK5array", "operator<=::a"], [0, 1, 1, "_CPPv4I0Ele5array1TRK5array", "operator<=::b"], [0, 1, 1, "_CPPv4I0Ele5arrayRK5array1T", "operator<=::b"], [0, 1, 1, "_CPPv4leRK5arrayRK5array", "operator<=::b"], [0, 0, 1, "_CPPv4I0Eeq5array1TRK5array", "operator=="], [0, 0, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator=="], [0, 0, 1, "_CPPv4eqRK5arrayRK5array", "operator=="], [0, 2, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::T"], [0, 2, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::T"], [0, 1, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::a"], [0, 1, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::a"], [0, 1, 1, "_CPPv4eqRK5arrayRK5array", "operator==::a"], [0, 1, 1, "_CPPv4I0Eeq5array1TRK5array", "operator==::b"], [0, 1, 1, "_CPPv4I0Eeq5arrayRK5array1T", "operator==::b"], [0, 1, 1, "_CPPv4eqRK5arrayRK5array", "operator==::b"], [0, 0, 1, "_CPPv4I0Egt5array1TRK5array", "operator>"], [0, 0, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>"], [0, 0, 1, "_CPPv4gtRK5arrayRK5array", "operator>"], [0, 2, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::T"], [0, 2, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::T"], [0, 1, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::a"], [0, 1, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::a"], [0, 1, 1, "_CPPv4gtRK5arrayRK5array", "operator>::a"], [0, 1, 1, "_CPPv4I0Egt5array1TRK5array", "operator>::b"], [0, 1, 1, "_CPPv4I0Egt5arrayRK5array1T", "operator>::b"], [0, 1, 1, "_CPPv4gtRK5arrayRK5array", "operator>::b"], [0, 0, 1, "_CPPv4I0Ege5array1TRK5array", "operator>="], [0, 0, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>="], [0, 0, 1, "_CPPv4geRK5arrayRK5array", "operator>="], [0, 2, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::T"], [0, 2, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::T"], [0, 1, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::a"], [0, 1, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::a"], [0, 1, 1, "_CPPv4geRK5arrayRK5array", "operator>=::a"], [0, 1, 1, "_CPPv4I0Ege5array1TRK5array", "operator>=::b"], [0, 1, 1, "_CPPv4I0Ege5arrayRK5array1T", "operator>=::b"], [0, 1, 1, "_CPPv4geRK5arrayRK5array", "operator>=::b"], [0, 0, 1, "_CPPv4rsRK5arrayRK5array", "operator>>"], [0, 1, 1, "_CPPv4rsRK5arrayRK5array", "operator>>::a"], [0, 1, 1, "_CPPv4rsRK5arrayRK5array", "operator>>::b"], [0, 0, 1, "_CPPv4eoRK5arrayRK5array", "operator^"], [0, 1, 1, "_CPPv4eoRK5arrayRK5array", "operator^::a"], [0, 1, 1, "_CPPv4eoRK5arrayRK5array", "operator^::b"], [0, 0, 1, "_CPPv4orRK5arrayRK5array", "operator|"], [0, 1, 1, "_CPPv4orRK5arrayRK5array", "operator|::a"], [0, 1, 1, "_CPPv4orRK5arrayRK5array", "operator|::b"], [0, 0, 1, "_CPPv4ooRK5arrayRK5array", "operator||"], [0, 1, 1, "_CPPv4ooRK5arrayRK5array", "operator||::a"], [0, 1, 1, "_CPPv4ooRK5arrayRK5array", "operator||::b"], [0, 0, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::a"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::b"], [0, 1, 1, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", "outer::s"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::axes"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::high_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::low_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice", "pad::s"], [0, 0, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition"], [0, 0, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::a"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::a"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::axis"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::kth"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::kth"], [0, 1, 1, "_CPPv49partitionRK5arrayi14StreamOrDevice", "partition::s"], [0, 1, 1, "_CPPv49partitionRK5arrayii14StreamOrDevice", "partition::s"], [0, 0, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::a"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::b"], [0, 1, 1, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", "power::s"], [0, 0, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod"], [0, 0, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod"], [0, 1, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::a"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::axes"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::axis"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::keepdims"], [0, 1, 1, "_CPPv44prodRK5array14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayb14StreamOrDevice", "prod::s"], [0, 1, 1, "_CPPv44prodRK5arrayib14StreamOrDevice", "prod::s"], [0, 0, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::bits"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::group_size"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::s"], [0, 1, 1, "_CPPv48quantizeRK5arrayii14StreamOrDevice", "quantize::w"], [0, 0, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::biases"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::bits"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::group_size"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::s"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::scales"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::transpose"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::w"], [0, 1, 1, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice", "quantized_matmul::x"], [0, 0, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::a"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::s"], [0, 0, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal"], [0, 1, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal::a"], [0, 1, 1, "_CPPv410reciprocalRK5array14StreamOrDevice", "reciprocal::s"], [0, 0, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::a"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::b"], [0, 1, 1, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", "remainder::s"], [0, 0, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat"], [0, 0, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::arr"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::arr"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::axis"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::repeats"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::repeats"], [0, 1, 1, "_CPPv46repeatRK5arrayi14StreamOrDevice", "repeat::s"], [0, 1, 1, "_CPPv46repeatRK5arrayii14StreamOrDevice", "repeat::s"], [0, 0, 1, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice", "reshape"], [0, 1, 1, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice", "reshape::a"], [0, 1, 1, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice", "reshape::s"], [0, 1, 1, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice", "reshape::shape"], [0, 0, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::a"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::b"], [0, 1, 1, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", "right_shift::s"], [0, 0, 1, "_CPPv45roundRK5array14StreamOrDevice", "round"], [0, 0, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round"], [0, 1, 1, "_CPPv45roundRK5array14StreamOrDevice", "round::a"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::a"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::decimals"], [0, 1, 1, "_CPPv45roundRK5array14StreamOrDevice", "round::s"], [0, 1, 1, "_CPPv45roundRK5arrayi14StreamOrDevice", "round::s"], [0, 0, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt"], [0, 1, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt::a"], [0, 1, 1, "_CPPv45rsqrtRK5array14StreamOrDevice", "rsqrt::s"], [0, 0, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter"], [0, 0, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::a"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::a"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::axes"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::axis"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::indices"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::indices"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::s"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::s"], [0, 1, 1, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter::updates"], [0, 1, 1, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter::updates"], [0, 0, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add"], [0, 0, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::a"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::a"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::axes"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::axis"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::indices"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::indices"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::s"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::s"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add::updates"], [0, 1, 1, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_add::updates"], [0, 0, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max"], [0, 0, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::a"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::a"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::axes"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::axis"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::indices"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::indices"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::s"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::s"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_max::updates"], [0, 1, 1, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_max::updates"], [0, 0, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min"], [0, 0, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::a"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::a"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::axes"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::axis"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::indices"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::indices"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::s"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::s"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_min::updates"], [0, 1, 1, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_min::updates"], [0, 0, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod"], [0, 0, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::a"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::a"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::axes"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::axis"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::indices"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::indices"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::s"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::s"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_prod::updates"], [0, 1, 1, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", "scatter_prod::updates"], [0, 0, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid"], [0, 1, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid::a"], [0, 1, 1, "_CPPv47sigmoidRK5array14StreamOrDevice", "sigmoid::s"], [0, 0, 1, "_CPPv44signRK5array14StreamOrDevice", "sign"], [0, 1, 1, "_CPPv44signRK5array14StreamOrDevice", "sign::a"], [0, 1, 1, "_CPPv44signRK5array14StreamOrDevice", "sign::s"], [0, 0, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin"], [0, 1, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin::a"], [0, 1, 1, "_CPPv43sinRK5array14StreamOrDevice", "sin::s"], [0, 0, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh"], [0, 1, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh::a"], [0, 1, 1, "_CPPv44sinhRK5array14StreamOrDevice", "sinh::s"], [0, 0, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice::strides"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::strides"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice", "slice_update::update"], [0, 0, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax"], [0, 0, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax"], [0, 0, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::a"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::axes"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::axis"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::precise"], [0, 1, 1, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "softmax::s"], [0, 1, 1, "_CPPv47softmaxRK5arrayb14StreamOrDevice", "softmax::s"], [0, 1, 1, "_CPPv47softmaxRK5arrayib14StreamOrDevice", "softmax::s"], [0, 0, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort"], [0, 0, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort"], [0, 1, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort::a"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::a"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::axis"], [0, 1, 1, "_CPPv44sortRK5array14StreamOrDevice", "sort::s"], [0, 1, 1, "_CPPv44sortRK5arrayi14StreamOrDevice", "sort::s"], [0, 0, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::s"], [0, 0, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt"], [0, 1, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt::a"], [0, 1, 1, "_CPPv44sqrtRK5array14StreamOrDevice", "sqrt::s"], [0, 0, 1, "_CPPv46squareRK5array14StreamOrDevice", "square"], [0, 1, 1, "_CPPv46squareRK5array14StreamOrDevice", "square::a"], [0, 1, 1, "_CPPv46squareRK5array14StreamOrDevice", "square::s"], [0, 0, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze"], [0, 0, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze"], [0, 0, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze"], [0, 1, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::a"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::axes"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::axis"], [0, 1, 1, "_CPPv47squeezeRK5array14StreamOrDevice", "squeeze::s"], [0, 1, 1, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", "squeeze::s"], [0, 1, 1, "_CPPv47squeezeRK5arrayi14StreamOrDevice", "squeeze::s"], [0, 0, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack"], [0, 0, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack::arrays"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::arrays"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::axis"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", "stack::s"], [0, 1, 1, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", "stack::s"], [0, 0, 1, "_CPPv4StRK5array14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std"], [0, 0, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std"], [0, 1, 1, "_CPPv4StRK5array14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::a"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::axes"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::axis"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::ddof"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::keepdims"], [0, 1, 1, "_CPPv4StRK5array14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arraybi14StreamOrDevice", "std::s"], [0, 1, 1, "_CPPv4StRK5arrayibi14StreamOrDevice", "std::s"], [0, 0, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient"], [0, 1, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient::a"], [0, 1, 1, "_CPPv413stop_gradientRK5array14StreamOrDevice", "stop_gradient::s"], [0, 0, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::a"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::b"], [0, 1, 1, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", "subtract::s"], [0, 0, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum"], [0, 0, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum"], [0, 1, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::a"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::axes"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::axis"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::keepdims"], [0, 1, 1, "_CPPv43sumRK5array14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayb14StreamOrDevice", "sum::s"], [0, 1, 1, "_CPPv43sumRK5arrayib14StreamOrDevice", "sum::s"], [0, 0, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::a"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::axis1"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::axis2"], [0, 1, 1, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", "swapaxes::s"], [0, 0, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::indices"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::indices"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::s"], [0, 0, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::a"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::axis"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::indices"], [0, 1, 1, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", "take_along_axis::s"], [0, 0, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan"], [0, 1, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan::a"], [0, 1, 1, "_CPPv43tanRK5array14StreamOrDevice", "tan::s"], [0, 0, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh"], [0, 1, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh::a"], [0, 1, 1, "_CPPv44tanhRK5array14StreamOrDevice", "tanh::s"], [0, 0, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot"], [0, 0, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::axes_a"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::axes_b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::axis"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::b"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", "tensordot::s"], [0, 1, 1, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", "tensordot::s"], [0, 0, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::arr"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::reps"], [0, 1, 1, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", "tile::s"], [0, 0, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk"], [0, 0, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::a"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::a"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::axis"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::k"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::k"], [0, 1, 1, "_CPPv44topkRK5arrayi14StreamOrDevice", "topk::s"], [0, 1, 1, "_CPPv44topkRK5arrayii14StreamOrDevice", "topk::s"], [0, 0, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace"], [0, 0, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace"], [0, 0, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace"], [0, 1, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::a"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::axis1"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::axis1"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::axis2"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::axis2"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::dtype"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::offset"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::offset"], [0, 1, 1, "_CPPv45traceRK5array14StreamOrDevice", "trace::s"], [0, 1, 1, "_CPPv45traceRK5arrayiii14StreamOrDevice", "trace::s"], [0, 1, 1, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", "trace::s"], [0, 0, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose"], [0, 0, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose"], [0, 0, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose"], [0, 1, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::a"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::axes"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::axes"], [0, 1, 1, "_CPPv49transposeRK5array14StreamOrDevice", "transpose::s"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", "transpose::s"], [0, 1, 1, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", "transpose::s"], [0, 0, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri"], [0, 0, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::k"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::m"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::n"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::n"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::s"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::s"], [0, 1, 1, "_CPPv43trii5Dtype14StreamOrDevice", "tri::type"], [0, 1, 1, "_CPPv43triiii5Dtype14StreamOrDevice", "tri::type"], [0, 0, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::k"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::s"], [0, 1, 1, "_CPPv44tril5arrayi14StreamOrDevice", "tril::x"], [0, 0, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::k"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::s"], [0, 1, 1, "_CPPv44triu5arrayi14StreamOrDevice", "triu::x"], [0, 0, 1, "_CPPv43varRK5array14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var"], [0, 0, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var"], [0, 1, 1, "_CPPv43varRK5array14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::a"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::axes"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::axis"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::ddof"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::keepdims"], [0, 1, 1, "_CPPv43varRK5array14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arraybi14StreamOrDevice", "var::s"], [0, 1, 1, "_CPPv43varRK5arrayibi14StreamOrDevice", "var::s"], [0, 0, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::a"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::dtype"], [0, 1, 1, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", "view::s"], [0, 0, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::condition"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::s"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::x"], [0, 1, 1, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", "where::y"], [0, 0, 1, "_CPPv45zerosRKNSt6vectorIiEE14StreamOrDevice", "zeros"], [0, 0, 1, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice", "zeros"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice", "zeros::dtype"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE14StreamOrDevice", "zeros::shape"], [0, 1, 1, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice", "zeros::shape"], [0, 0, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like"], [0, 1, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like::a"], [0, 1, 1, "_CPPv410zeros_likeRK5array14StreamOrDevice", "zeros_like::s"]], "mlx.core": [[8, 3, 1, "", "Device"], [9, 3, 1, "", "Dtype"], [10, 3, 1, "", "DtypeCategory"], [294, 3, 1, "", "Stream"], [11, 5, 1, "", "abs"], [12, 5, 1, "", "add"], [13, 5, 1, "", "addmm"], [14, 5, 1, "", "all"], [15, 5, 1, "", "allclose"], [16, 5, 1, "", "any"], [17, 5, 1, "", "arange"], [18, 5, 1, "", "arccos"], [19, 5, 1, "", "arccosh"], [20, 5, 1, "", "arcsin"], [21, 5, 1, "", "arcsinh"], [22, 5, 1, "", "arctan"], [23, 5, 1, "", "arctan2"], [24, 5, 1, "", "arctanh"], [25, 5, 1, "", "argmax"], [26, 5, 1, "", "argmin"], [27, 5, 1, "", "argpartition"], [28, 5, 1, "", "argsort"], [29, 3, 1, "", "array"], [80, 5, 1, "", "array_equal"], [81, 5, 1, "", "as_strided"], [82, 5, 1, "", "atleast_1d"], [83, 5, 1, "", "atleast_2d"], [84, 5, 1, "", "atleast_3d"], [85, 5, 1, "", "bitwise_and"], [86, 5, 1, "", "bitwise_or"], [87, 5, 1, "", "bitwise_xor"], [88, 5, 1, "", "block_masked_mm"], [89, 5, 1, "", "broadcast_to"], [90, 5, 1, "", "ceil"], [91, 5, 1, "", "clip"], [92, 5, 1, "", "compile"], [93, 5, 1, "", "concatenate"], [94, 5, 1, "", "conj"], [95, 5, 1, "", "conjugate"], [96, 5, 1, "", "conv1d"], [97, 5, 1, "", "conv2d"], [98, 5, 1, "", "conv_general"], [99, 5, 1, "", "convolve"], [100, 5, 1, "", "cos"], [101, 5, 1, "", "cosh"], [102, 5, 1, "", "cummax"], [103, 5, 1, "", "cummin"], [104, 5, 1, "", "cumprod"], [105, 5, 1, "", "cumsum"], [106, 3, 1, "", "custom_function"], [107, 5, 1, "", "default_device"], [108, 5, 1, "", "default_stream"], [109, 5, 1, "", "degrees"], [110, 5, 1, "", "dequantize"], [111, 5, 1, "", "diag"], [112, 5, 1, "", "diagonal"], [113, 5, 1, "", "disable_compile"], [119, 5, 1, "", "divide"], [120, 5, 1, "", "divmod"], [121, 5, 1, "", "einsum"], [122, 5, 1, "", "einsum_path"], [123, 5, 1, "", "enable_compile"], [124, 5, 1, "", "equal"], [125, 5, 1, "", "erf"], [126, 5, 1, "", "erfinv"], [127, 5, 1, "", "eval"], [128, 5, 1, "", "exp"], [129, 5, 1, "", "expand_dims"], [130, 5, 1, "", "expm1"], [131, 5, 1, "", "eye"], [148, 5, 1, "", "flatten"], [149, 5, 1, "", "floor"], [150, 5, 1, "", "floor_divide"], [151, 5, 1, "", "full"], [152, 5, 1, "", "gather_mm"], [153, 5, 1, "", "gather_qmm"], [154, 5, 1, "", "grad"], [155, 5, 1, "", "greater"], [156, 5, 1, "", "greater_equal"], [157, 5, 1, "", "hadamard_transform"], [158, 5, 1, "", "identity"], [159, 5, 1, "", "inner"], [160, 5, 1, "", "isclose"], [161, 5, 1, "", "isinf"], [162, 5, 1, "", "isnan"], [163, 5, 1, "", "isneginf"], [164, 5, 1, "", "isposinf"], [165, 5, 1, "", "issubdtype"], [166, 5, 1, "", "jvp"], [167, 5, 1, "", "left_shift"], [168, 5, 1, "", "less"], [169, 5, 1, "", "less_equal"], [177, 5, 1, "", "linspace"], [178, 5, 1, "", "load"], [179, 5, 1, "", "log"], [180, 5, 1, "", "log10"], [181, 5, 1, "", "log1p"], [182, 5, 1, "", "log2"], [183, 5, 1, "", "logaddexp"], [184, 5, 1, "", "logical_and"], [185, 5, 1, "", "logical_not"], [186, 5, 1, "", "logical_or"], [187, 5, 1, "", "logsumexp"], [188, 5, 1, "", "matmul"], [189, 5, 1, "", "max"], [190, 5, 1, "", "maximum"], [191, 5, 1, "", "mean"], [192, 5, 1, "", "meshgrid"], [204, 5, 1, "", "min"], [205, 5, 1, "", "minimum"], [206, 5, 1, "", "moveaxis"], [207, 5, 1, "", "multiply"], [208, 5, 1, "", "nan_to_num"], [209, 5, 1, "", "negative"], [210, 5, 1, "", "new_stream"], [211, 5, 1, "", "not_equal"], [212, 5, 1, "", "ones"], [213, 5, 1, "", "ones_like"], [214, 5, 1, "", "outer"], [215, 5, 1, "", "pad"], [216, 5, 1, "", "partition"], [217, 5, 1, "", "power"], [218, 5, 1, "", "prod"], [219, 5, 1, "", "quantize"], [220, 5, 1, "", "quantized_matmul"], [221, 5, 1, "", "radians"], [234, 5, 1, "", "reciprocal"], [235, 5, 1, "", "remainder"], [236, 5, 1, "", "repeat"], [237, 5, 1, "", "reshape"], [238, 5, 1, "", "right_shift"], [239, 5, 1, "", "round"], [240, 5, 1, "", "rsqrt"], [241, 5, 1, "", "save"], [242, 5, 1, "", "save_gguf"], [243, 5, 1, "", "save_safetensors"], [244, 5, 1, "", "savez"], [245, 5, 1, "", "savez_compressed"], [246, 5, 1, "", "set_default_device"], [247, 5, 1, "", "set_default_stream"], [248, 5, 1, "", "sigmoid"], [249, 5, 1, "", "sign"], [250, 5, 1, "", "sin"], [251, 5, 1, "", "sinh"], [252, 5, 1, "", "softmax"], [253, 5, 1, "", "sort"], [254, 5, 1, "", "split"], [255, 5, 1, "", "sqrt"], [256, 5, 1, "", "square"], [257, 5, 1, "", "squeeze"], [258, 5, 1, "", "stack"], [259, 5, 1, "", "std"], [260, 5, 1, "", "stop_gradient"], [261, 5, 1, "", "stream"], [262, 5, 1, "", "subtract"], [263, 5, 1, "", "sum"], [264, 5, 1, "", "swapaxes"], [265, 5, 1, "", "synchronize"], [266, 5, 1, "", "take"], [267, 5, 1, "", "take_along_axis"], [268, 5, 1, "", "tan"], [269, 5, 1, "", "tanh"], [270, 5, 1, "", "tensordot"], [271, 5, 1, "", "tile"], [272, 5, 1, "", "topk"], [273, 5, 1, "", "trace"], [274, 5, 1, "", "transpose"], [275, 5, 1, "", "tri"], [276, 5, 1, "", "tril"], [277, 5, 1, "", "triu"], [278, 5, 1, "", "value_and_grad"], [279, 5, 1, "", "var"], [280, 5, 1, "", "view"], [281, 5, 1, "", "vjp"], [282, 5, 1, "", "vmap"], [283, 5, 1, "", "where"], [284, 5, 1, "", "zeros"], [285, 5, 1, "", "zeros_like"]], "mlx.core.Device": [[8, 4, 1, "", "__init__"]], "mlx.core.Dtype": [[9, 4, 1, "", "__init__"]], "mlx.core.DtypeCategory": [[10, 4, 1, "", "__init__"]], "mlx.core.Stream": [[294, 4, 1, "", "__init__"]], "mlx.core.array": [[30, 6, 1, "", "T"], [29, 4, 1, "", "__init__"], [31, 4, 1, "", "abs"], [32, 4, 1, "", "all"], [33, 4, 1, "", "any"], [34, 4, 1, "", "argmax"], [35, 4, 1, "", "argmin"], [36, 4, 1, "", "astype"], [37, 6, 1, "", "at"], [38, 4, 1, "", "conj"], [39, 4, 1, "", "cos"], [40, 4, 1, "", "cummax"], [41, 4, 1, "", "cummin"], [42, 4, 1, "", "cumprod"], [43, 4, 1, "", "cumsum"], [44, 4, 1, "", "diag"], [45, 4, 1, "", "diagonal"], [46, 6, 1, "", "dtype"], [47, 4, 1, "", "exp"], [48, 4, 1, "", "flatten"], [49, 4, 1, "", "item"], [50, 6, 1, "", "itemsize"], [51, 4, 1, "", "log"], [52, 4, 1, "", "log10"], [53, 4, 1, "", "log1p"], [54, 4, 1, "", "log2"], [55, 4, 1, "", "logsumexp"], [56, 4, 1, "", "max"], [57, 4, 1, "", "mean"], [58, 4, 1, "", "min"], [59, 4, 1, "", "moveaxis"], [60, 6, 1, "", "nbytes"], [61, 6, 1, "", "ndim"], [62, 4, 1, "", "prod"], [63, 4, 1, "", "reciprocal"], [64, 4, 1, "", "reshape"], [65, 4, 1, "", "round"], [66, 4, 1, "", "rsqrt"], [67, 6, 1, "", "shape"], [68, 4, 1, "", "sin"], [69, 6, 1, "", "size"], [70, 4, 1, "", "split"], [71, 4, 1, "", "sqrt"], [72, 4, 1, "", "square"], [73, 4, 1, "", "squeeze"], [74, 4, 1, "", "sum"], [75, 4, 1, "", "swapaxes"], [76, 4, 1, "", "tolist"], [77, 4, 1, "", "transpose"], [78, 4, 1, "", "var"], [79, 4, 1, "", "view"]], "mlx.core.custom_function": [[106, 4, 1, "", "__init__"]], "mlx.core.distributed": [[114, 3, 1, "", "Group"], [115, 5, 1, "", "all_gather"], [116, 5, 1, "", "all_sum"], [117, 5, 1, "", "init"], [118, 5, 1, "", "is_available"]], "mlx.core.distributed.Group": [[114, 4, 1, "", "__init__"]], "mlx.core.fast": [[132, 5, 1, "", "layer_norm"], [133, 5, 1, "", "rms_norm"], [134, 5, 1, "", "rope"], [135, 5, 1, "", "scaled_dot_product_attention"]], "mlx.core.fft": [[136, 5, 1, "", "fft"], [137, 5, 1, "", "fft2"], [138, 5, 1, "", "fftn"], [139, 5, 1, "", "ifft"], [140, 5, 1, "", "ifft2"], [141, 5, 1, "", "ifftn"], [142, 5, 1, "", "irfft"], [143, 5, 1, "", "irfft2"], [144, 5, 1, "", "irfftn"], [145, 5, 1, "", "rfft"], [146, 5, 1, "", "rfft2"], [147, 5, 1, "", "rfftn"]], "mlx.core.linalg": [[170, 5, 1, "", "cholesky"], [171, 5, 1, "", "cholesky_inv"], [172, 5, 1, "", "inv"], [173, 5, 1, "", "norm"], [174, 5, 1, "", "qr"], [175, 5, 1, "", "svd"], [176, 5, 1, "", "tri_inv"]], "mlx.core.metal": [[193, 5, 1, "", "clear_cache"], [194, 5, 1, "", "device_info"], [195, 5, 1, "", "get_active_memory"], [196, 5, 1, "", "get_cache_memory"], [197, 5, 1, "", "get_peak_memory"], [198, 5, 1, "", "is_available"], [199, 5, 1, "", "reset_peak_memory"], [200, 5, 1, "", "set_cache_limit"], [201, 5, 1, "", "set_memory_limit"], [202, 5, 1, "", "start_capture"], [203, 5, 1, "", "stop_capture"]], "mlx.core.random": [[222, 5, 1, "", "bernoulli"], [223, 5, 1, "", "categorical"], [224, 5, 1, "", "gumbel"], [225, 5, 1, "", "key"], [226, 5, 1, "", "laplace"], [227, 5, 1, "", "multivariate_normal"], [228, 5, 1, "", "normal"], [229, 5, 1, "", "randint"], [230, 5, 1, "", "seed"], [231, 5, 1, "", "split"], [232, 5, 1, "", "truncated_normal"], [233, 5, 1, "", "uniform"]], "mlx.nn": [[304, 3, 1, "", "ALiBi"], [305, 3, 1, "", "AvgPool1d"], [306, 3, 1, "", "AvgPool2d"], [307, 3, 1, "", "BatchNorm"], [308, 3, 1, "", "Conv1d"], [309, 3, 1, "", "Conv2d"], [310, 3, 1, "", "Conv3d"], [311, 3, 1, "", "Dropout"], [312, 3, 1, "", "Dropout2d"], [313, 3, 1, "", "Dropout3d"], [314, 3, 1, "", "Embedding"], [315, 3, 1, "", "GELU"], [316, 3, 1, "", "GLU"], [317, 3, 1, "", "GRU"], [318, 3, 1, "", "GroupNorm"], [319, 3, 1, "", "HardShrink"], [320, 3, 1, "", "HardTanh"], [321, 3, 1, "", "Hardswish"], [322, 3, 1, "", "InstanceNorm"], [323, 3, 1, "", "LSTM"], [324, 3, 1, "", "LayerNorm"], [325, 3, 1, "", "LeakyReLU"], [326, 3, 1, "", "Linear"], [327, 3, 1, "", "MaxPool1d"], [328, 3, 1, "", "MaxPool2d"], [329, 3, 1, "", "Mish"], [422, 3, 1, "", "Module"], [350, 3, 1, "", "MultiHeadAttention"], [351, 3, 1, "", "PReLU"], [352, 3, 1, "", "QuantizedEmbedding"], [353, 3, 1, "", "QuantizedLinear"], [354, 3, 1, "", "RMSNorm"], [355, 3, 1, "", "RNN"], [356, 3, 1, "", "ReLU"], [357, 3, 1, "", "ReLU6"], [358, 3, 1, "", "RoPE"], [359, 3, 1, "", "SELU"], [360, 3, 1, "", "Sequential"], [361, 3, 1, "", "SiLU"], [362, 3, 1, "", "SinusoidalPositionalEncoding"], [363, 3, 1, "", "Softmax"], [364, 3, 1, "", "Softmin"], [365, 3, 1, "", "Softplus"], [366, 3, 1, "", "Softshrink"], [367, 3, 1, "", "Softsign"], [368, 3, 1, "", "Step"], [369, 3, 1, "", "Tanh"], [370, 3, 1, "", "Transformer"], [371, 3, 1, "", "Upsample"], [380, 3, 1, "", "elu"], [381, 3, 1, "", "gelu"], [382, 3, 1, "", "gelu_approx"], [383, 3, 1, "", "gelu_fast_approx"], [384, 3, 1, "", "glu"], [385, 3, 1, "", "hard_shrink"], [386, 3, 1, "", "hard_tanh"], [387, 3, 1, "", "hardswish"], [388, 3, 1, "", "leaky_relu"], [389, 3, 1, "", "log_sigmoid"], [390, 3, 1, "", "log_softmax"], [405, 3, 1, "", "mish"], [406, 3, 1, "", "prelu"], [286, 5, 1, "", "quantize"], [407, 3, 1, "", "relu"], [408, 3, 1, "", "relu6"], [409, 3, 1, "", "selu"], [410, 3, 1, "", "sigmoid"], [411, 3, 1, "", "silu"], [412, 3, 1, "", "softmax"], [413, 3, 1, "", "softmin"], [414, 3, 1, "", "softplus"], [415, 3, 1, "", "softshrink"], [416, 3, 1, "", "step"], [417, 3, 1, "", "tanh"], [287, 5, 1, "", "value_and_grad"]], "mlx.nn.Module": [[330, 4, 1, "", "apply"], [331, 4, 1, "", "apply_to_modules"], [332, 4, 1, "", "children"], [333, 4, 1, "", "eval"], [334, 4, 1, "", "filter_and_map"], [335, 4, 1, "", "freeze"], [336, 4, 1, "", "leaf_modules"], [337, 4, 1, "", "load_weights"], [338, 4, 1, "", "modules"], [339, 4, 1, "", "named_modules"], [340, 4, 1, "", "parameters"], [341, 4, 1, "", "save_weights"], [342, 4, 1, "", "set_dtype"], [343, 6, 1, "", "state"], [344, 4, 1, "", "train"], [345, 4, 1, "", "trainable_parameters"], [346, 6, 1, "", "training"], [347, 4, 1, "", "unfreeze"], [348, 4, 1, "", "update"], [349, 4, 1, "", "update_modules"]], "mlx.nn.init": [[372, 5, 1, "", "constant"], [373, 5, 1, "", "glorot_normal"], [374, 5, 1, "", "glorot_uniform"], [375, 5, 1, "", "he_normal"], [376, 5, 1, "", "he_uniform"], [377, 5, 1, "", "identity"], [378, 5, 1, "", "normal"], [379, 5, 1, "", "uniform"]], "mlx.nn.losses": [[391, 3, 1, "", "binary_cross_entropy"], [392, 3, 1, "", "cosine_similarity_loss"], [393, 3, 1, "", "cross_entropy"], [394, 3, 1, "", "gaussian_nll_loss"], [395, 3, 1, "", "hinge_loss"], [396, 3, 1, "", "huber_loss"], [397, 3, 1, "", "kl_div_loss"], [398, 3, 1, "", "l1_loss"], [399, 3, 1, "", "log_cosh_loss"], [400, 3, 1, "", "margin_ranking_loss"], [401, 3, 1, "", "mse_loss"], [402, 3, 1, "", "nll_loss"], [403, 3, 1, "", "smooth_l1_loss"], [404, 3, 1, "", "triplet_loss"]], "mlx.optimizers": [[425, 3, 1, "", "AdaDelta"], [426, 3, 1, "", "Adafactor"], [427, 3, 1, "", "Adagrad"], [428, 3, 1, "", "Adam"], [429, 3, 1, "", "AdamW"], [430, 3, 1, "", "Adamax"], [431, 3, 1, "", "Lion"], [444, 3, 1, "", "Optimizer"], [436, 3, 1, "", "RMSprop"], [437, 3, 1, "", "SGD"], [288, 5, 1, "", "clip_grad_norm"], [438, 5, 1, "", "cosine_decay"], [439, 5, 1, "", "exponential_decay"], [440, 5, 1, "", "join_schedules"], [441, 5, 1, "", "linear_schedule"], [442, 5, 1, "", "step_decay"]], "mlx.optimizers.Optimizer": [[432, 4, 1, "", "apply_gradients"], [433, 4, 1, "", "init"], [434, 6, 1, "", "state"], [435, 4, 1, "", "update"]], "mlx.utils": [[289, 5, 1, "", "tree_flatten"], [290, 5, 1, "", "tree_map"], [291, 5, 1, "", "tree_map_with_path"], [292, 5, 1, "", "tree_reduce"], [293, 5, 1, "", "tree_unflatten"]]}, "objtypes": {"0": "cpp:function", "1": "cpp:functionParam", "2": "cpp:templateParam", "3": "py:class", "4": "py:method", "5": "py:function", "6": "py:property"}, "objnames": {"0": ["cpp", "function", "C++ function"], "1": ["cpp", "functionParam", "C++ function parameter"], "2": ["cpp", "templateParam", "C++ template parameter"], "3": ["py", "class", "Python class"], "4": ["py", "method", "Python method"], "5": ["py", "function", "Python function"], "6": ["py", "property", "Python property"]}, "titleterms": {"oper": [0, 1, 423], "custom": 1, "extens": 1, "mlx": [1, 6, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442], "introduc": 1, "exampl": [1, 6, 449, 450, 457], "primit": 1, "us": [1, 453, 458], "implement": [1, 4], "cpu": 1, "back": 1, "end": 1, "gpu": 1, "transform": [1, 370, 447, 449, 451, 453, 455], "build": [1, 7], "bind": 1, "python": [1, 6, 7], "cmake": 1, "setuptool": 1, "usag": [1, 6], "result": 1, "script": [1, 4], "download": [1, 4], "code": [1, 4], "metal": [2, 7, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 302], "debugg": 2, "xcode": 2, "workflow": 2, "linear": [3, 301, 326], "regress": 3, "llm": 4, "infer": 4, "model": 4, "attent": 4, "layer": [4, 5, 420], "encod": 4, "full": [4, 151], "gener": 4, "put": 4, "all": [4, 14, 32, 450], "togeth": 4, "convert": 4, "weight": 4, "load": [4, 178, 424, 456], "benchmark": 4, "multi": 5, "perceptron": 5, "instal": [6, 7, 450], "api": [6, 7], "refer": 6, "c": [6, 7], "further": 6, "read": 6, "troubleshoot": 7, "from": [7, 452], "sourc": 7, "requir": 7, "option": 7, "binari": 7, "size": [7, 69], "minim": 7, "found": 7, "x86": 7, "shell": 7, "core": [8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 294], "devic": [8, 297], "dtype": [9, 46], "dtypecategori": 10, "ab": [11, 31], "add": 12, "addmm": 13, "allclos": 15, "ani": [16, 33], "arang": 17, "arcco": 18, "arccosh": 19, "arcsin": 20, "arcsinh": 21, "arctan": 22, "arctan2": 23, "arctanh": 24, "argmax": [25, 34], "argmin": [26, 35], "argpartit": 27, "argsort": 28, "arrai": [29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 295, 452, 456], "t": 30, "astyp": 36, "conj": [38, 94], "co": [39, 100], "cummax": [40, 102], "cummin": [41, 103], "cumprod": [42, 104], "cumsum": [43, 105], "diag": [44, 111], "diagon": [45, 112], "exp": [47, 128], "flatten": [48, 148], "item": 49, "items": 50, "log": [51, 179], "log10": [52, 180], "log1p": [53, 181], "log2": [54, 182], "logsumexp": [55, 187], "max": [56, 189], "mean": [57, 191], "min": [58, 204], "moveaxi": [59, 206], "nbyte": 60, "ndim": 61, "prod": [62, 218], "reciproc": [63, 234], "reshap": [64, 237], "round": [65, 239], "rsqrt": [66, 240], "shape": 67, "sin": [68, 250], "split": [70, 231, 254], "sqrt": [71, 255], "squar": [72, 256], "squeez": [73, 257], "sum": [74, 263], "swapax": [75, 264], "tolist": 76, "transpos": [77, 274], "var": [78, 279], "view": [79, 280], "array_equ": 80, "as_strid": 81, "atleast_1d": 82, "atleast_2d": 83, "atleast_3d": 84, "bitwise_and": 85, "bitwise_or": 86, "bitwise_xor": 87, "block_masked_mm": 88, "broadcast_to": 89, "ceil": 90, "clip": 91, "compil": [92, 449], "concaten": 93, "conjug": 95, "conv1d": [96, 308], "conv2d": [97, 309], "conv_gener": 98, "convolv": 99, "cosh": 101, "custom_funct": 106, "default_devic": 107, "default_stream": 108, "degre": 109, "dequant": 110, "disable_compil": 113, "distribut": [114, 115, 116, 117, 118, 298, 450], "group": 114, "all_gath": 115, "all_sum": 116, "init": [117, 372, 373, 374, 375, 376, 377, 378, 379, 433], "is_avail": [118, 198], "divid": 119, "divmod": 120, "einsum": 121, "einsum_path": 122, "enable_compil": 123, "equal": 124, "erf": 125, "erfinv": 126, "eval": [127, 333], "expand_dim": 129, "expm1": 130, "ey": 131, "fast": [132, 133, 134, 135, 299], "layer_norm": 132, "rms_norm": 133, "rope": [134, 358], "scaled_dot_product_attent": 135, "fft": [136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 300], "fft2": 137, "fftn": 138, "ifft": 139, "ifft2": 140, "ifftn": 141, "irfft": 142, "irfft2": 143, "irfftn": 144, "rfft": 145, "rfft2": 146, "rfftn": 147, "floor": 149, "floor_divid": 150, "gather_mm": 152, "gather_qmm": 153, "grad": [154, 303], "greater": 155, "greater_equ": 156, "hadamard_transform": 157, "ident": [158, 377], "inner": 159, "isclos": 160, "isinf": 161, "isnan": 162, "isneginf": 163, "isposinf": 164, "issubdtyp": 165, "jvp": 166, "left_shift": 167, "less": 168, "less_equ": 169, "linalg": [170, 171, 172, 173, 174, 175, 176], "choleski": 170, "cholesky_inv": 171, "inv": 172, "norm": 173, "qr": 174, "svd": 175, "tri_inv": 176, "linspac": 177, "logaddexp": 183, "logical_and": 184, "logical_not": 185, "logical_or": 186, "matmul": 188, "maximum": 190, "meshgrid": 192, "clear_cach": 193, "device_info": 194, "get_active_memori": 195, "get_cache_memori": 196, "get_peak_memori": 197, "reset_peak_memori": 199, "set_cache_limit": 200, "set_memory_limit": 201, "start_captur": 202, "stop_captur": 203, "minimum": 205, "multipli": 207, "nan_to_num": 208, "neg": 209, "new_stream": 210, "not_equ": 211, "ones": 212, "ones_lik": 213, "outer": 214, "pad": 215, "partit": 216, "power": 217, "quantiz": [219, 286], "quantized_matmul": 220, "radian": 221, "random": [222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 446], "bernoulli": 222, "categor": 223, "gumbel": 224, "kei": 225, "laplac": 226, "multivariate_norm": 227, "normal": [228, 378], "randint": 229, "seed": 230, "truncated_norm": 232, "uniform": [233, 379], "remaind": 235, "repeat": 236, "right_shift": 238, "save": [241, 424, 456], "save_gguf": 242, "save_safetensor": 243, "savez": 244, "savez_compress": 245, "set_default_devic": 246, "set_default_stream": 247, "sigmoid": [248, 410], "sign": 249, "sinh": 251, "softmax": [252, 363, 412], "sort": 253, "stack": 258, "std": 259, "stop_gradi": 260, "stream": [261, 294, 297, 458], "subtract": 262, "synchron": 265, "take": 266, "take_along_axi": 267, "tan": 268, "tanh": [269, 369, 417], "tensordot": 270, "tile": 271, "topk": 272, "trace": 273, "tri": 275, "tril": 276, "triu": 277, "value_and_grad": [278, 287], "vjp": 281, "vmap": 282, "where": 283, "zero": 284, "zeros_lik": 285, "nn": [286, 287, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417], "optim": [288, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444], "clip_grad_norm": 288, "util": [289, 290, 291, 292, 293, 448], "tree_flatten": 289, "tree_map": 290, "tree_map_with_path": 291, "tree_reduc": 292, "tree_unflatten": 293, "data": 296, "type": 296, "support": 296, "commun": [298, 450], "algebra": 301, "neural": 303, "network": 303, "quick": [303, 455], "start": [303, 450, 455], "The": 303, "modul": [303, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 422], "class": 303, "paramet": [303, 340], "updat": [303, 348, 435, 452], "inspect": 303, "valu": 303, "alibi": 304, "avgpool1d": 305, "avgpool2d": 306, "batchnorm": 307, "conv3d": 310, "dropout": 311, "dropout2d": 312, "dropout3d": 313, "embed": 314, "gelu": [315, 381], "glu": [316, 384], "gru": 317, "groupnorm": 318, "hardshrink": 319, "hardtanh": 320, "hardswish": [321, 387], "instancenorm": 322, "lstm": 323, "layernorm": 324, "leakyrelu": 325, "maxpool1d": 327, "maxpool2d": 328, "mish": [329, 405], "appli": 330, "apply_to_modul": 331, "children": 332, "filter_and_map": 334, "freez": 335, "leaf_modul": 336, "load_weight": 337, "named_modul": 339, "save_weight": 341, "set_dtyp": 342, "state": [343, 434], "train": [344, 346, 449, 450], "trainable_paramet": 345, "unfreez": 347, "update_modul": 349, "multiheadattent": 350, "prelu": [351, 406], "quantizedembed": 352, "quantizedlinear": 353, "rmsnorm": 354, "rnn": 355, "relu": [356, 407], "relu6": [357, 408], "selu": [359, 409], "sequenti": 360, "silu": [361, 411], "sinusoidalpositionalencod": 362, "softmin": [364, 413], "softplu": [365, 414], "softshrink": [366, 415], "softsign": 367, "step": [368, 416], "upsampl": 371, "constant": 372, "glorot_norm": 373, "glorot_uniform": 374, "he_norm": 375, "he_uniform": 376, "elu": 380, "gelu_approx": 382, "gelu_fast_approx": 383, "hard_shrink": 385, "hard_tanh": 386, "leaky_relu": 388, "log_sigmoid": 389, "log_softmax": 390, "loss": [391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 421], "binary_cross_entropi": 391, "cosine_similarity_loss": 392, "cross_entropi": 393, "gaussian_nll_loss": 394, "hinge_loss": 395, "huber_loss": 396, "kl_div_loss": 397, "l1_loss": 398, "log_cosh_loss": 399, "margin_ranking_loss": 400, "mse_loss": 401, "nll_loss": 402, "smooth_l1_loss": 403, "triplet_loss": 404, "function": [418, 421, 449, 451, 455], "initi": 419, "adadelta": 425, "adafactor": 426, "adagrad": 427, "adam": 428, "adamw": 429, "adamax": 430, "lion": 431, "apply_gradi": 432, "rmsprop": 436, "sgd": 437, "cosine_decai": 438, "exponential_decai": 439, "join_schedul": 440, "linear_schedul": 441, "step_decai": 442, "common": 443, "schedul": 445, "tree": 448, "basic": [449, 455], "speedup": 449, "debug": 449, "pure": 449, "graph": [449, 453, 455], "get": 450, "mpi": 450, "set": 450, "up": 450, "remot": 450, "host": 450, "tune": 450, "reduc": 450, "automat": 451, "differenti": 451, "vector": 451, "index": 452, "differ": 452, "numpi": [452, 454], "In": 452, "place": 452, "lazi": 453, "evalu": 453, "why": 453, "comput": 453, "onli": 453, "what": 453, "you": 453, "when": 453, "convers": 454, "other": 454, "framework": 454, "pytorch": 454, "jax": 454, "tensorflow": 454, "guid": 455, "serial": 456, "format": 456, "unifi": 457, "memori": 457, "A": 457, "simpl": 457, "specifi": 458}, "envversion": {"sphinx.domains.c": 3, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 9, "sphinx.domains.index": 1, "sphinx.domains.javascript": 3, "sphinx.domains.math": 2, "sphinx.domains.python": 4, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.intersphinx": 1, "sphinx": 60}, "alltitles": {"Operations": [[0, "operations"], [1, "operations"], [423, "operations"]], "Custom Extensions in MLX": [[1, "custom-extensions-in-mlx"]], "Introducing the Example": [[1, "introducing-the-example"]], "Operations and Primitives": [[1, "operations-and-primitives"]], "Primitives": [[1, "primitives"]], "Using the Primitive": [[1, "using-the-primitive"]], "Implementing the Primitive": [[1, "implementing-the-primitive"]], "Implementing the CPU Back-end": [[1, "implementing-the-cpu-back-end"]], "Implementing the GPU Back-end": [[1, "implementing-the-gpu-back-end"]], "Primitive Transforms": [[1, "primitive-transforms"]], "Building and Binding": [[1, "building-and-binding"]], "Binding to Python": [[1, "binding-to-python"]], "Building with CMake": [[1, "building-with-cmake"]], "Building with setuptools": [[1, "building-with-setuptools"]], "Usage": [[1, "usage"], [6, null]], "Results": [[1, "results"]], "Scripts": [[1, "scripts"], [4, "scripts"]], "Download the code": [[1, null], [4, null]], "Metal Debugger": [[2, "metal-debugger"]], "Xcode Workflow": [[2, "xcode-workflow"]], "Linear Regression": [[3, "linear-regression"]], "LLM inference": [[4, "llm-inference"]], "Implementing the model": [[4, "implementing-the-model"]], "Attention layer": [[4, "attention-layer"]], "Encoder layer": [[4, "encoder-layer"]], "Full model": [[4, "full-model"]], "Generation": [[4, "generation"]], "Putting it all together": [[4, "putting-it-all-together"]], "Converting the weights": [[4, "converting-the-weights"]], "Weight loading and benchmarking": [[4, "weight-loading-and-benchmarking"]], "Multi-Layer Perceptron": [[5, "multi-layer-perceptron"]], "MLX": [[6, "mlx"]], "Install": [[6, null]], "Examples": [[6, null]], "Python API Reference": [[6, null]], "C++ API Reference": [[6, null]], "Further Reading": [[6, null]], "Build and Install": [[7, "build-and-install"]], "Python Installation": [[7, "python-installation"]], "Troubleshooting": [[7, "troubleshooting"], [7, "id2"]], "Build from source": [[7, "build-from-source"]], "Build Requirements": [[7, "build-requirements"]], "Python API": [[7, "python-api"]], "C++ API": [[7, "c-api"]], "Build Options": [[7, "id3"]], "Binary Size Minimization": [[7, "binary-size-minimization"]], "Metal not found": [[7, "metal-not-found"]], "x86 Shell": [[7, "x86-shell"]], "mlx.core.Device": [[8, "mlx-core-device"]], "mlx.core.Dtype": [[9, "mlx-core-dtype"]], "mlx.core.DtypeCategory": [[10, "mlx-core-dtypecategory"]], "mlx.core.abs": [[11, "mlx-core-abs"]], "mlx.core.add": [[12, "mlx-core-add"]], "mlx.core.addmm": [[13, "mlx-core-addmm"]], "mlx.core.all": [[14, "mlx-core-all"]], "mlx.core.allclose": [[15, "mlx-core-allclose"]], "mlx.core.any": [[16, "mlx-core-any"]], "mlx.core.arange": [[17, "mlx-core-arange"]], "mlx.core.arccos": [[18, "mlx-core-arccos"]], "mlx.core.arccosh": [[19, "mlx-core-arccosh"]], "mlx.core.arcsin": [[20, "mlx-core-arcsin"]], "mlx.core.arcsinh": [[21, "mlx-core-arcsinh"]], "mlx.core.arctan": [[22, "mlx-core-arctan"]], "mlx.core.arctan2": [[23, "mlx-core-arctan2"]], "mlx.core.arctanh": [[24, "mlx-core-arctanh"]], "mlx.core.argmax": [[25, "mlx-core-argmax"]], "mlx.core.argmin": [[26, "mlx-core-argmin"]], "mlx.core.argpartition": [[27, "mlx-core-argpartition"]], "mlx.core.argsort": [[28, "mlx-core-argsort"]], "mlx.core.array": [[29, "mlx-core-array"]], "mlx.core.array.T": [[30, "mlx-core-array-t"]], "mlx.core.array.abs": [[31, "mlx-core-array-abs"]], "mlx.core.array.all": [[32, "mlx-core-array-all"]], "mlx.core.array.any": [[33, "mlx-core-array-any"]], "mlx.core.array.argmax": [[34, "mlx-core-array-argmax"]], "mlx.core.array.argmin": [[35, "mlx-core-array-argmin"]], "mlx.core.array.astype": [[36, "mlx-core-array-astype"]], "mlx.core.array.at": [[37, "mlx-core-array-at"]], "mlx.core.array.conj": [[38, "mlx-core-array-conj"]], "mlx.core.array.cos": [[39, "mlx-core-array-cos"]], "mlx.core.array.cummax": [[40, "mlx-core-array-cummax"]], "mlx.core.array.cummin": [[41, "mlx-core-array-cummin"]], "mlx.core.array.cumprod": [[42, "mlx-core-array-cumprod"]], "mlx.core.array.cumsum": [[43, "mlx-core-array-cumsum"]], "mlx.core.array.diag": [[44, "mlx-core-array-diag"]], "mlx.core.array.diagonal": [[45, "mlx-core-array-diagonal"]], "mlx.core.array.dtype": [[46, "mlx-core-array-dtype"]], "mlx.core.array.exp": [[47, "mlx-core-array-exp"]], "mlx.core.array.flatten": [[48, "mlx-core-array-flatten"]], "mlx.core.array.item": [[49, "mlx-core-array-item"]], "mlx.core.array.itemsize": [[50, "mlx-core-array-itemsize"]], "mlx.core.array.log": [[51, "mlx-core-array-log"]], "mlx.core.array.log10": [[52, "mlx-core-array-log10"]], "mlx.core.array.log1p": [[53, "mlx-core-array-log1p"]], "mlx.core.array.log2": [[54, "mlx-core-array-log2"]], "mlx.core.array.logsumexp": [[55, "mlx-core-array-logsumexp"]], "mlx.core.array.max": [[56, "mlx-core-array-max"]], "mlx.core.array.mean": [[57, "mlx-core-array-mean"]], "mlx.core.array.min": [[58, "mlx-core-array-min"]], "mlx.core.array.moveaxis": [[59, "mlx-core-array-moveaxis"]], "mlx.core.array.nbytes": [[60, "mlx-core-array-nbytes"]], "mlx.core.array.ndim": [[61, "mlx-core-array-ndim"]], "mlx.core.array.prod": [[62, "mlx-core-array-prod"]], "mlx.core.array.reciprocal": [[63, "mlx-core-array-reciprocal"]], "mlx.core.array.reshape": [[64, "mlx-core-array-reshape"]], "mlx.core.array.round": [[65, "mlx-core-array-round"]], "mlx.core.array.rsqrt": [[66, "mlx-core-array-rsqrt"]], "mlx.core.array.shape": [[67, "mlx-core-array-shape"]], "mlx.core.array.sin": [[68, "mlx-core-array-sin"]], "mlx.core.array.size": [[69, "mlx-core-array-size"]], "mlx.core.array.split": [[70, "mlx-core-array-split"]], "mlx.core.array.sqrt": [[71, "mlx-core-array-sqrt"]], "mlx.core.array.square": [[72, "mlx-core-array-square"]], "mlx.core.array.squeeze": [[73, "mlx-core-array-squeeze"]], "mlx.core.array.sum": [[74, "mlx-core-array-sum"]], "mlx.core.array.swapaxes": [[75, "mlx-core-array-swapaxes"]], "mlx.core.array.tolist": [[76, "mlx-core-array-tolist"]], "mlx.core.array.transpose": [[77, "mlx-core-array-transpose"]], "mlx.core.array.var": [[78, "mlx-core-array-var"]], "mlx.core.array.view": [[79, "mlx-core-array-view"]], "mlx.core.array_equal": [[80, "mlx-core-array-equal"]], "mlx.core.as_strided": [[81, "mlx-core-as-strided"]], "mlx.core.atleast_1d": [[82, "mlx-core-atleast-1d"]], "mlx.core.atleast_2d": [[83, "mlx-core-atleast-2d"]], "mlx.core.atleast_3d": [[84, "mlx-core-atleast-3d"]], "mlx.core.bitwise_and": [[85, "mlx-core-bitwise-and"]], "mlx.core.bitwise_or": [[86, "mlx-core-bitwise-or"]], "mlx.core.bitwise_xor": [[87, "mlx-core-bitwise-xor"]], "mlx.core.block_masked_mm": [[88, "mlx-core-block-masked-mm"]], "mlx.core.broadcast_to": [[89, "mlx-core-broadcast-to"]], "mlx.core.ceil": [[90, "mlx-core-ceil"]], "mlx.core.clip": [[91, "mlx-core-clip"]], "mlx.core.compile": [[92, "mlx-core-compile"]], "mlx.core.concatenate": [[93, "mlx-core-concatenate"]], "mlx.core.conj": [[94, "mlx-core-conj"]], "mlx.core.conjugate": [[95, "mlx-core-conjugate"]], "mlx.core.conv1d": [[96, "mlx-core-conv1d"]], "mlx.core.conv2d": [[97, "mlx-core-conv2d"]], "mlx.core.conv_general": [[98, "mlx-core-conv-general"]], "mlx.core.convolve": [[99, "mlx-core-convolve"]], "mlx.core.cos": [[100, "mlx-core-cos"]], "mlx.core.cosh": [[101, "mlx-core-cosh"]], "mlx.core.cummax": [[102, "mlx-core-cummax"]], "mlx.core.cummin": [[103, "mlx-core-cummin"]], "mlx.core.cumprod": [[104, "mlx-core-cumprod"]], "mlx.core.cumsum": [[105, "mlx-core-cumsum"]], "mlx.core.custom_function": [[106, "mlx-core-custom-function"]], "mlx.core.default_device": [[107, "mlx-core-default-device"]], "mlx.core.default_stream": [[108, "mlx-core-default-stream"]], "mlx.core.degrees": [[109, "mlx-core-degrees"]], "mlx.core.dequantize": [[110, "mlx-core-dequantize"]], "mlx.core.diag": [[111, "mlx-core-diag"]], "mlx.core.diagonal": [[112, "mlx-core-diagonal"]], "mlx.core.disable_compile": [[113, "mlx-core-disable-compile"]], "mlx.core.distributed.Group": [[114, "mlx-core-distributed-group"]], "mlx.core.distributed.all_gather": [[115, "mlx-core-distributed-all-gather"]], "mlx.core.distributed.all_sum": [[116, "mlx-core-distributed-all-sum"]], "mlx.core.distributed.init": [[117, "mlx-core-distributed-init"]], "mlx.core.distributed.is_available": [[118, "mlx-core-distributed-is-available"]], "mlx.core.divide": [[119, "mlx-core-divide"]], "mlx.core.divmod": [[120, "mlx-core-divmod"]], "mlx.core.einsum": [[121, "mlx-core-einsum"]], "mlx.core.einsum_path": [[122, "mlx-core-einsum-path"]], "mlx.core.enable_compile": [[123, "mlx-core-enable-compile"]], "mlx.core.equal": [[124, "mlx-core-equal"]], "mlx.core.erf": [[125, "mlx-core-erf"]], "mlx.core.erfinv": [[126, "mlx-core-erfinv"]], "mlx.core.eval": [[127, "mlx-core-eval"]], "mlx.core.exp": [[128, "mlx-core-exp"]], "mlx.core.expand_dims": [[129, "mlx-core-expand-dims"]], "mlx.core.expm1": [[130, "mlx-core-expm1"]], "mlx.core.eye": [[131, "mlx-core-eye"]], "mlx.core.fast.layer_norm": [[132, "mlx-core-fast-layer-norm"]], "mlx.core.fast.rms_norm": [[133, "mlx-core-fast-rms-norm"]], "mlx.core.fast.rope": [[134, "mlx-core-fast-rope"]], "mlx.core.fast.scaled_dot_product_attention": [[135, "mlx-core-fast-scaled-dot-product-attention"]], "mlx.core.fft.fft": [[136, "mlx-core-fft-fft"]], "mlx.core.fft.fft2": [[137, "mlx-core-fft-fft2"]], "mlx.core.fft.fftn": [[138, "mlx-core-fft-fftn"]], "mlx.core.fft.ifft": [[139, "mlx-core-fft-ifft"]], "mlx.core.fft.ifft2": [[140, "mlx-core-fft-ifft2"]], "mlx.core.fft.ifftn": [[141, "mlx-core-fft-ifftn"]], "mlx.core.fft.irfft": [[142, "mlx-core-fft-irfft"]], "mlx.core.fft.irfft2": [[143, "mlx-core-fft-irfft2"]], "mlx.core.fft.irfftn": [[144, "mlx-core-fft-irfftn"]], "mlx.core.fft.rfft": [[145, "mlx-core-fft-rfft"]], "mlx.core.fft.rfft2": [[146, "mlx-core-fft-rfft2"]], "mlx.core.fft.rfftn": [[147, "mlx-core-fft-rfftn"]], "mlx.core.flatten": [[148, "mlx-core-flatten"]], "mlx.core.floor": [[149, "mlx-core-floor"]], "mlx.core.floor_divide": [[150, "mlx-core-floor-divide"]], "mlx.core.full": [[151, "mlx-core-full"]], "mlx.core.gather_mm": [[152, "mlx-core-gather-mm"]], "mlx.core.gather_qmm": [[153, "mlx-core-gather-qmm"]], "mlx.core.grad": [[154, "mlx-core-grad"]], "mlx.core.greater": [[155, "mlx-core-greater"]], "mlx.core.greater_equal": [[156, "mlx-core-greater-equal"]], "mlx.core.hadamard_transform": [[157, "mlx-core-hadamard-transform"]], "mlx.core.identity": [[158, "mlx-core-identity"]], "mlx.core.inner": [[159, "mlx-core-inner"]], "mlx.core.isclose": [[160, "mlx-core-isclose"]], "mlx.core.isinf": [[161, "mlx-core-isinf"]], "mlx.core.isnan": [[162, "mlx-core-isnan"]], "mlx.core.isneginf": [[163, "mlx-core-isneginf"]], "mlx.core.isposinf": [[164, "mlx-core-isposinf"]], "mlx.core.issubdtype": [[165, "mlx-core-issubdtype"]], "mlx.core.jvp": [[166, "mlx-core-jvp"]], "mlx.core.left_shift": [[167, "mlx-core-left-shift"]], "mlx.core.less": [[168, "mlx-core-less"]], "mlx.core.less_equal": [[169, "mlx-core-less-equal"]], "mlx.core.linalg.cholesky": [[170, "mlx-core-linalg-cholesky"]], "mlx.core.linalg.cholesky_inv": [[171, "mlx-core-linalg-cholesky-inv"]], "mlx.core.linalg.inv": [[172, "mlx-core-linalg-inv"]], "mlx.core.linalg.norm": [[173, "mlx-core-linalg-norm"]], "mlx.core.linalg.qr": [[174, "mlx-core-linalg-qr"]], "mlx.core.linalg.svd": [[175, "mlx-core-linalg-svd"]], "mlx.core.linalg.tri_inv": [[176, "mlx-core-linalg-tri-inv"]], "mlx.core.linspace": [[177, "mlx-core-linspace"]], "mlx.core.load": [[178, "mlx-core-load"]], "mlx.core.log": [[179, "mlx-core-log"]], "mlx.core.log10": [[180, "mlx-core-log10"]], "mlx.core.log1p": [[181, "mlx-core-log1p"]], "mlx.core.log2": [[182, "mlx-core-log2"]], "mlx.core.logaddexp": [[183, "mlx-core-logaddexp"]], "mlx.core.logical_and": [[184, "mlx-core-logical-and"]], "mlx.core.logical_not": [[185, "mlx-core-logical-not"]], "mlx.core.logical_or": [[186, "mlx-core-logical-or"]], "mlx.core.logsumexp": [[187, "mlx-core-logsumexp"]], "mlx.core.matmul": [[188, "mlx-core-matmul"]], "mlx.core.max": [[189, "mlx-core-max"]], "mlx.core.maximum": [[190, "mlx-core-maximum"]], "mlx.core.mean": [[191, "mlx-core-mean"]], "mlx.core.meshgrid": [[192, "mlx-core-meshgrid"]], "mlx.core.metal.clear_cache": [[193, "mlx-core-metal-clear-cache"]], "mlx.core.metal.device_info": [[194, "mlx-core-metal-device-info"]], "mlx.core.metal.get_active_memory": [[195, "mlx-core-metal-get-active-memory"]], "mlx.core.metal.get_cache_memory": [[196, "mlx-core-metal-get-cache-memory"]], "mlx.core.metal.get_peak_memory": [[197, "mlx-core-metal-get-peak-memory"]], "mlx.core.metal.is_available": [[198, "mlx-core-metal-is-available"]], "mlx.core.metal.reset_peak_memory": [[199, "mlx-core-metal-reset-peak-memory"]], "mlx.core.metal.set_cache_limit": [[200, "mlx-core-metal-set-cache-limit"]], "mlx.core.metal.set_memory_limit": [[201, "mlx-core-metal-set-memory-limit"]], "mlx.core.metal.start_capture": [[202, "mlx-core-metal-start-capture"]], "mlx.core.metal.stop_capture": [[203, "mlx-core-metal-stop-capture"]], "mlx.core.min": [[204, "mlx-core-min"]], "mlx.core.minimum": [[205, "mlx-core-minimum"]], "mlx.core.moveaxis": [[206, "mlx-core-moveaxis"]], "mlx.core.multiply": [[207, "mlx-core-multiply"]], "mlx.core.nan_to_num": [[208, "mlx-core-nan-to-num"]], "mlx.core.negative": [[209, "mlx-core-negative"]], "mlx.core.new_stream": [[210, "mlx-core-new-stream"]], "mlx.core.not_equal": [[211, "mlx-core-not-equal"]], "mlx.core.ones": [[212, "mlx-core-ones"]], "mlx.core.ones_like": [[213, "mlx-core-ones-like"]], "mlx.core.outer": [[214, "mlx-core-outer"]], "mlx.core.pad": [[215, "mlx-core-pad"]], "mlx.core.partition": [[216, "mlx-core-partition"]], "mlx.core.power": [[217, "mlx-core-power"]], "mlx.core.prod": [[218, "mlx-core-prod"]], "mlx.core.quantize": [[219, "mlx-core-quantize"]], "mlx.core.quantized_matmul": [[220, "mlx-core-quantized-matmul"]], "mlx.core.radians": [[221, "mlx-core-radians"]], "mlx.core.random.bernoulli": [[222, "mlx-core-random-bernoulli"]], "mlx.core.random.categorical": [[223, "mlx-core-random-categorical"]], "mlx.core.random.gumbel": [[224, "mlx-core-random-gumbel"]], "mlx.core.random.key": [[225, "mlx-core-random-key"]], "mlx.core.random.laplace": [[226, "mlx-core-random-laplace"]], "mlx.core.random.multivariate_normal": [[227, "mlx-core-random-multivariate-normal"]], "mlx.core.random.normal": [[228, "mlx-core-random-normal"]], "mlx.core.random.randint": [[229, "mlx-core-random-randint"]], "mlx.core.random.seed": [[230, "mlx-core-random-seed"]], "mlx.core.random.split": [[231, "mlx-core-random-split"]], "mlx.core.random.truncated_normal": [[232, "mlx-core-random-truncated-normal"]], "mlx.core.random.uniform": [[233, "mlx-core-random-uniform"]], "mlx.core.reciprocal": [[234, "mlx-core-reciprocal"]], "mlx.core.remainder": [[235, "mlx-core-remainder"]], "mlx.core.repeat": [[236, "mlx-core-repeat"]], "mlx.core.reshape": [[237, "mlx-core-reshape"]], "mlx.core.right_shift": [[238, "mlx-core-right-shift"]], "mlx.core.round": [[239, "mlx-core-round"]], "mlx.core.rsqrt": [[240, "mlx-core-rsqrt"]], "mlx.core.save": [[241, "mlx-core-save"]], "mlx.core.save_gguf": [[242, "mlx-core-save-gguf"]], "mlx.core.save_safetensors": [[243, "mlx-core-save-safetensors"]], "mlx.core.savez": [[244, "mlx-core-savez"]], "mlx.core.savez_compressed": [[245, "mlx-core-savez-compressed"]], "mlx.core.set_default_device": [[246, "mlx-core-set-default-device"]], "mlx.core.set_default_stream": [[247, "mlx-core-set-default-stream"]], "mlx.core.sigmoid": [[248, "mlx-core-sigmoid"]], "mlx.core.sign": [[249, "mlx-core-sign"]], "mlx.core.sin": [[250, "mlx-core-sin"]], "mlx.core.sinh": [[251, "mlx-core-sinh"]], "mlx.core.softmax": [[252, "mlx-core-softmax"]], "mlx.core.sort": [[253, "mlx-core-sort"]], "mlx.core.split": [[254, "mlx-core-split"]], "mlx.core.sqrt": [[255, "mlx-core-sqrt"]], "mlx.core.square": [[256, "mlx-core-square"]], "mlx.core.squeeze": [[257, "mlx-core-squeeze"]], "mlx.core.stack": [[258, "mlx-core-stack"]], "mlx.core.std": [[259, "mlx-core-std"]], "mlx.core.stop_gradient": [[260, "mlx-core-stop-gradient"]], "mlx.core.stream": [[261, "mlx-core-stream"]], "mlx.core.subtract": [[262, "mlx-core-subtract"]], "mlx.core.sum": [[263, "mlx-core-sum"]], "mlx.core.swapaxes": [[264, "mlx-core-swapaxes"]], "mlx.core.synchronize": [[265, "mlx-core-synchronize"]], "mlx.core.take": [[266, "mlx-core-take"]], "mlx.core.take_along_axis": [[267, "mlx-core-take-along-axis"]], "mlx.core.tan": [[268, "mlx-core-tan"]], "mlx.core.tanh": [[269, "mlx-core-tanh"]], "mlx.core.tensordot": [[270, "mlx-core-tensordot"]], "mlx.core.tile": [[271, "mlx-core-tile"]], "mlx.core.topk": [[272, "mlx-core-topk"]], "mlx.core.trace": [[273, "mlx-core-trace"]], "mlx.core.transpose": [[274, "mlx-core-transpose"]], "mlx.core.tri": [[275, "mlx-core-tri"]], "mlx.core.tril": [[276, "mlx-core-tril"]], "mlx.core.triu": [[277, "mlx-core-triu"]], "mlx.core.value_and_grad": [[278, "mlx-core-value-and-grad"]], "mlx.core.var": [[279, "mlx-core-var"]], "mlx.core.view": [[280, "mlx-core-view"]], "mlx.core.vjp": [[281, "mlx-core-vjp"]], "mlx.core.vmap": [[282, "mlx-core-vmap"]], "mlx.core.where": [[283, "mlx-core-where"]], "mlx.core.zeros": [[284, "mlx-core-zeros"]], "mlx.core.zeros_like": [[285, "mlx-core-zeros-like"]], "mlx.nn.quantize": [[286, "mlx-nn-quantize"]], "mlx.nn.value_and_grad": [[287, "mlx-nn-value-and-grad"]], "mlx.optimizers.clip_grad_norm": [[288, "mlx-optimizers-clip-grad-norm"]], "mlx.utils.tree_flatten": [[289, "mlx-utils-tree-flatten"]], "mlx.utils.tree_map": [[290, "mlx-utils-tree-map"]], "mlx.utils.tree_map_with_path": [[291, "mlx-utils-tree-map-with-path"]], "mlx.utils.tree_reduce": [[292, "mlx-utils-tree-reduce"]], "mlx.utils.tree_unflatten": [[293, "mlx-utils-tree-unflatten"]], "mlx.core.Stream": [[294, "mlx-core-stream"]], "Array": [[295, "array"]], "Data Types": [[296, "data-types"]], "Supported Data Types": [[296, "id2"]], "Devices and Streams": [[297, "devices-and-streams"]], "Distributed Communication": [[298, "distributed-communication"], [450, "distributed-communication"]], "Fast": [[299, "fast"]], "FFT": [[300, "fft"]], "Linear Algebra": [[301, "linear-algebra"]], "Metal": [[302, "metal"]], "Neural Networks": [[303, "neural-networks"]], "Quick Start with Neural Networks": [[303, "quick-start-with-neural-networks"]], "The Module Class": [[303, "the-module-class"]], "Parameters": [[303, "parameters"]], "Updating the Parameters": [[303, "updating-the-parameters"]], "Inspecting Modules": [[303, "inspecting-modules"]], "Value and Grad": [[303, "value-and-grad"]], "mlx.nn.ALiBi": [[304, "mlx-nn-alibi"]], "mlx.nn.AvgPool1d": [[305, "mlx-nn-avgpool1d"]], "mlx.nn.AvgPool2d": [[306, "mlx-nn-avgpool2d"]], "mlx.nn.BatchNorm": [[307, "mlx-nn-batchnorm"]], "mlx.nn.Conv1d": [[308, "mlx-nn-conv1d"]], "mlx.nn.Conv2d": [[309, "mlx-nn-conv2d"]], "mlx.nn.Conv3d": [[310, "mlx-nn-conv3d"]], "mlx.nn.Dropout": [[311, "mlx-nn-dropout"]], "mlx.nn.Dropout2d": [[312, "mlx-nn-dropout2d"]], "mlx.nn.Dropout3d": [[313, "mlx-nn-dropout3d"]], "mlx.nn.Embedding": [[314, "mlx-nn-embedding"]], "mlx.nn.GELU": [[315, "mlx-nn-gelu"]], "mlx.nn.GLU": [[316, "mlx-nn-glu"]], "mlx.nn.GRU": [[317, "mlx-nn-gru"]], "mlx.nn.GroupNorm": [[318, "mlx-nn-groupnorm"]], "mlx.nn.HardShrink": [[319, "mlx-nn-hardshrink"]], "mlx.nn.HardTanh": [[320, "mlx-nn-hardtanh"]], "mlx.nn.Hardswish": [[321, "mlx-nn-hardswish"]], "mlx.nn.InstanceNorm": [[322, "mlx-nn-instancenorm"]], "mlx.nn.LSTM": [[323, "mlx-nn-lstm"]], "mlx.nn.LayerNorm": [[324, "mlx-nn-layernorm"]], "mlx.nn.LeakyReLU": [[325, "mlx-nn-leakyrelu"]], "mlx.nn.Linear": [[326, "mlx-nn-linear"]], "mlx.nn.MaxPool1d": [[327, "mlx-nn-maxpool1d"]], "mlx.nn.MaxPool2d": [[328, "mlx-nn-maxpool2d"]], "mlx.nn.Mish": [[329, "mlx-nn-mish"]], "mlx.nn.Module.apply": [[330, "mlx-nn-module-apply"]], "mlx.nn.Module.apply_to_modules": [[331, "mlx-nn-module-apply-to-modules"]], "mlx.nn.Module.children": [[332, "mlx-nn-module-children"]], "mlx.nn.Module.eval": [[333, "mlx-nn-module-eval"]], "mlx.nn.Module.filter_and_map": [[334, "mlx-nn-module-filter-and-map"]], "mlx.nn.Module.freeze": [[335, "mlx-nn-module-freeze"]], "mlx.nn.Module.leaf_modules": [[336, "mlx-nn-module-leaf-modules"]], "mlx.nn.Module.load_weights": [[337, "mlx-nn-module-load-weights"]], "mlx.nn.Module.modules": [[338, "mlx-nn-module-modules"]], "mlx.nn.Module.named_modules": [[339, "mlx-nn-module-named-modules"]], "mlx.nn.Module.parameters": [[340, "mlx-nn-module-parameters"]], "mlx.nn.Module.save_weights": [[341, "mlx-nn-module-save-weights"]], "mlx.nn.Module.set_dtype": [[342, "mlx-nn-module-set-dtype"]], "mlx.nn.Module.state": [[343, "mlx-nn-module-state"]], "mlx.nn.Module.train": [[344, "mlx-nn-module-train"]], "mlx.nn.Module.trainable_parameters": [[345, "mlx-nn-module-trainable-parameters"]], "mlx.nn.Module.training": [[346, "mlx-nn-module-training"]], "mlx.nn.Module.unfreeze": [[347, "mlx-nn-module-unfreeze"]], "mlx.nn.Module.update": [[348, "mlx-nn-module-update"]], "mlx.nn.Module.update_modules": [[349, "mlx-nn-module-update-modules"]], "mlx.nn.MultiHeadAttention": [[350, "mlx-nn-multiheadattention"]], "mlx.nn.PReLU": [[351, "mlx-nn-prelu"]], "mlx.nn.QuantizedEmbedding": [[352, "mlx-nn-quantizedembedding"]], "mlx.nn.QuantizedLinear": [[353, "mlx-nn-quantizedlinear"]], "mlx.nn.RMSNorm": [[354, "mlx-nn-rmsnorm"]], "mlx.nn.RNN": [[355, "mlx-nn-rnn"]], "mlx.nn.ReLU": [[356, "mlx-nn-relu"]], "mlx.nn.ReLU6": [[357, "mlx-nn-relu6"]], "mlx.nn.RoPE": [[358, "mlx-nn-rope"]], "mlx.nn.SELU": [[359, "mlx-nn-selu"]], "mlx.nn.Sequential": [[360, "mlx-nn-sequential"]], "mlx.nn.SiLU": [[361, "mlx-nn-silu"]], "mlx.nn.SinusoidalPositionalEncoding": [[362, "mlx-nn-sinusoidalpositionalencoding"]], "mlx.nn.Softmax": [[363, "mlx-nn-softmax"]], "mlx.nn.Softmin": [[364, "mlx-nn-softmin"]], "mlx.nn.Softplus": [[365, "mlx-nn-softplus"]], "mlx.nn.Softshrink": [[366, "mlx-nn-softshrink"]], "mlx.nn.Softsign": [[367, "mlx-nn-softsign"]], "mlx.nn.Step": [[368, "mlx-nn-step"]], "mlx.nn.Tanh": [[369, "mlx-nn-tanh"]], "mlx.nn.Transformer": [[370, "mlx-nn-transformer"]], "mlx.nn.Upsample": [[371, "mlx-nn-upsample"]], "mlx.nn.init.constant": [[372, "mlx-nn-init-constant"]], "mlx.nn.init.glorot_normal": [[373, "mlx-nn-init-glorot-normal"]], "mlx.nn.init.glorot_uniform": [[374, "mlx-nn-init-glorot-uniform"]], "mlx.nn.init.he_normal": [[375, "mlx-nn-init-he-normal"]], "mlx.nn.init.he_uniform": [[376, "mlx-nn-init-he-uniform"]], "mlx.nn.init.identity": [[377, "mlx-nn-init-identity"]], "mlx.nn.init.normal": [[378, "mlx-nn-init-normal"]], "mlx.nn.init.uniform": [[379, "mlx-nn-init-uniform"]], "mlx.nn.elu": [[380, "mlx-nn-elu"]], "mlx.nn.gelu": [[381, "mlx-nn-gelu"]], "mlx.nn.gelu_approx": [[382, "mlx-nn-gelu-approx"]], "mlx.nn.gelu_fast_approx": [[383, "mlx-nn-gelu-fast-approx"]], "mlx.nn.glu": [[384, "mlx-nn-glu"]], "mlx.nn.hard_shrink": [[385, "mlx-nn-hard-shrink"]], "mlx.nn.hard_tanh": [[386, "mlx-nn-hard-tanh"]], "mlx.nn.hardswish": [[387, "mlx-nn-hardswish"]], "mlx.nn.leaky_relu": [[388, "mlx-nn-leaky-relu"]], "mlx.nn.log_sigmoid": [[389, "mlx-nn-log-sigmoid"]], "mlx.nn.log_softmax": [[390, "mlx-nn-log-softmax"]], "mlx.nn.losses.binary_cross_entropy": [[391, "mlx-nn-losses-binary-cross-entropy"]], "mlx.nn.losses.cosine_similarity_loss": [[392, "mlx-nn-losses-cosine-similarity-loss"]], "mlx.nn.losses.cross_entropy": [[393, "mlx-nn-losses-cross-entropy"]], "mlx.nn.losses.gaussian_nll_loss": [[394, "mlx-nn-losses-gaussian-nll-loss"]], "mlx.nn.losses.hinge_loss": [[395, "mlx-nn-losses-hinge-loss"]], "mlx.nn.losses.huber_loss": [[396, "mlx-nn-losses-huber-loss"]], "mlx.nn.losses.kl_div_loss": [[397, "mlx-nn-losses-kl-div-loss"]], "mlx.nn.losses.l1_loss": [[398, "mlx-nn-losses-l1-loss"]], "mlx.nn.losses.log_cosh_loss": [[399, "mlx-nn-losses-log-cosh-loss"]], "mlx.nn.losses.margin_ranking_loss": [[400, "mlx-nn-losses-margin-ranking-loss"]], "mlx.nn.losses.mse_loss": [[401, "mlx-nn-losses-mse-loss"]], "mlx.nn.losses.nll_loss": [[402, "mlx-nn-losses-nll-loss"]], "mlx.nn.losses.smooth_l1_loss": [[403, "mlx-nn-losses-smooth-l1-loss"]], "mlx.nn.losses.triplet_loss": [[404, "mlx-nn-losses-triplet-loss"]], "mlx.nn.mish": [[405, "mlx-nn-mish"]], "mlx.nn.prelu": [[406, "mlx-nn-prelu"]], "mlx.nn.relu": [[407, "mlx-nn-relu"]], "mlx.nn.relu6": [[408, "mlx-nn-relu6"]], "mlx.nn.selu": [[409, "mlx-nn-selu"]], "mlx.nn.sigmoid": [[410, "mlx-nn-sigmoid"]], "mlx.nn.silu": [[411, "mlx-nn-silu"]], "mlx.nn.softmax": [[412, "mlx-nn-softmax"]], "mlx.nn.softmin": [[413, "mlx-nn-softmin"]], "mlx.nn.softplus": [[414, "mlx-nn-softplus"]], "mlx.nn.softshrink": [[415, "mlx-nn-softshrink"]], "mlx.nn.step": [[416, "mlx-nn-step"]], "mlx.nn.tanh": [[417, "mlx-nn-tanh"]], "Functions": [[418, "functions"]], "Initializers": [[419, "initializers"]], "Layers": [[420, "layers"]], "Loss Functions": [[421, "loss-functions"]], "Module": [[422, "module"]], "Optimizers": [[424, "optimizers"]], "Saving and Loading": [[424, "saving-and-loading"]], "mlx.optimizers.AdaDelta": [[425, "mlx-optimizers-adadelta"]], "mlx.optimizers.Adafactor": [[426, "mlx-optimizers-adafactor"]], "mlx.optimizers.Adagrad": [[427, "mlx-optimizers-adagrad"]], "mlx.optimizers.Adam": [[428, "mlx-optimizers-adam"]], "mlx.optimizers.AdamW": [[429, "mlx-optimizers-adamw"]], "mlx.optimizers.Adamax": [[430, "mlx-optimizers-adamax"]], "mlx.optimizers.Lion": [[431, "mlx-optimizers-lion"]], "mlx.optimizers.Optimizer.apply_gradients": [[432, "mlx-optimizers-optimizer-apply-gradients"]], "mlx.optimizers.Optimizer.init": [[433, "mlx-optimizers-optimizer-init"]], "mlx.optimizers.Optimizer.state": [[434, "mlx-optimizers-optimizer-state"]], "mlx.optimizers.Optimizer.update": [[435, "mlx-optimizers-optimizer-update"]], "mlx.optimizers.RMSprop": [[436, "mlx-optimizers-rmsprop"]], "mlx.optimizers.SGD": [[437, "mlx-optimizers-sgd"]], "mlx.optimizers.cosine_decay": [[438, "mlx-optimizers-cosine-decay"]], "mlx.optimizers.exponential_decay": [[439, "mlx-optimizers-exponential-decay"]], "mlx.optimizers.join_schedules": [[440, "mlx-optimizers-join-schedules"]], "mlx.optimizers.linear_schedule": [[441, "mlx-optimizers-linear-schedule"]], "mlx.optimizers.step_decay": [[442, "mlx-optimizers-step-decay"]], "Common Optimizers": [[443, "common-optimizers"]], "Optimizer": [[444, "optimizer"]], "Schedulers": [[445, "schedulers"]], "Random": [[446, "random"]], "Transforms": [[447, "transforms"]], "Tree Utils": [[448, "tree-utils"]], "Compilation": [[449, "compilation"]], "Basics of Compile": [[449, "basics-of-compile"]], "Example Speedup": [[449, "example-speedup"]], "Debugging": [[449, "debugging"]], "Pure Functions": [[449, "pure-functions"]], "Compiling Training Graphs": [[449, "compiling-training-graphs"]], "Transformations with Compile": [[449, "transformations-with-compile"]], "Getting Started": [[450, "getting-started"]], "Installing MPI": [[450, "installing-mpi"]], "Setting up Remote Hosts": [[450, "setting-up-remote-hosts"]], "Training Example": [[450, "training-example"]], "Tuning All Reduce": [[450, "tuning-all-reduce"]], "Function Transforms": [[451, "function-transforms"]], "Automatic Differentiation": [[451, "automatic-differentiation"]], "Automatic Vectorization": [[451, "automatic-vectorization"]], "Indexing Arrays": [[452, "indexing-arrays"]], "Differences from NumPy": [[452, "differences-from-numpy"]], "In Place Updates": [[452, "in-place-updates"]], "Lazy Evaluation": [[453, "lazy-evaluation"]], "Why Lazy Evaluation": [[453, "why-lazy-evaluation"]], "Transforming Compute Graphs": [[453, "transforming-compute-graphs"]], "Only Compute What You Use": [[453, "only-compute-what-you-use"]], "When to Evaluate": [[453, "when-to-evaluate"]], "Conversion to NumPy and Other Frameworks": [[454, "conversion-to-numpy-and-other-frameworks"]], "PyTorch": [[454, "pytorch"]], "JAX": [[454, "jax"]], "TensorFlow": [[454, "tensorflow"]], "Quick Start Guide": [[455, "quick-start-guide"]], "Basics": [[455, "basics"]], "Function and Graph Transformations": [[455, "function-and-graph-transformations"]], "Saving and Loading Arrays": [[456, "saving-and-loading-arrays"]], "Serialization Formats": [[456, "id1"]], "Unified Memory": [[457, "unified-memory"]], "A Simple Example": [[457, "a-simple-example"]], "Using Streams": [[458, "using-streams"]], "Specifying the Stream": [[458, "specifying-the-stream"]]}, "indexentries": {"abs (c++ function)": [[0, "_CPPv43absRK5array14StreamOrDevice"]], "add (c++ function)": [[0, "_CPPv43addRK5arrayRK5array14StreamOrDevice"]], "addmm (c++ function)": [[0, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice"]], "all (c++ function)": [[0, "_CPPv43allRK5array14StreamOrDevice"], [0, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43allRK5arrayb14StreamOrDevice"], [0, "_CPPv43allRK5arrayib14StreamOrDevice"]], "allclose (c++ function)": [[0, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice"]], "any (c++ function)": [[0, "_CPPv43anyRK5array14StreamOrDevice"], [0, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43anyRK5arrayb14StreamOrDevice"], [0, "_CPPv43anyRK5arrayib14StreamOrDevice"]], "arange (c++ function)": [[0, "_CPPv46aranged14StreamOrDevice"], [0, "_CPPv46aranged5Dtype14StreamOrDevice"], [0, "_CPPv46arangedd14StreamOrDevice"], [0, "_CPPv46arangedd5Dtype14StreamOrDevice"], [0, "_CPPv46arangeddd14StreamOrDevice"], [0, "_CPPv46arangeddd5Dtype14StreamOrDevice"], [0, "_CPPv46arangei14StreamOrDevice"], [0, "_CPPv46arangeii14StreamOrDevice"], [0, "_CPPv46arangeiii14StreamOrDevice"]], "arccos (c++ function)": [[0, "_CPPv46arccosRK5array14StreamOrDevice"]], "arccosh (c++ function)": [[0, "_CPPv47arccoshRK5array14StreamOrDevice"]], "arcsin (c++ function)": [[0, "_CPPv46arcsinRK5array14StreamOrDevice"]], "arcsinh (c++ function)": [[0, "_CPPv47arcsinhRK5array14StreamOrDevice"]], "arctan (c++ function)": [[0, "_CPPv46arctanRK5array14StreamOrDevice"]], "arctan2 (c++ function)": [[0, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice"]], "arctanh (c++ function)": [[0, "_CPPv47arctanhRK5array14StreamOrDevice"]], "argmax (c++ function)": [[0, "_CPPv46argmaxRK5array14StreamOrDevice"], [0, "_CPPv46argmaxRK5arrayb14StreamOrDevice"], [0, "_CPPv46argmaxRK5arrayib14StreamOrDevice"]], "argmin (c++ function)": [[0, "_CPPv46argminRK5array14StreamOrDevice"], [0, "_CPPv46argminRK5arrayb14StreamOrDevice"], [0, "_CPPv46argminRK5arrayib14StreamOrDevice"]], "argpartition (c++ function)": [[0, "_CPPv412argpartitionRK5arrayi14StreamOrDevice"], [0, "_CPPv412argpartitionRK5arrayii14StreamOrDevice"]], "argsort (c++ function)": [[0, "_CPPv47argsortRK5array14StreamOrDevice"], [0, "_CPPv47argsortRK5arrayi14StreamOrDevice"]], "array_equal (c++ function)": [[0, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice"], [0, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice"]], "as_strided (c++ function)": [[0, "_CPPv410as_strided5arrayNSt6vectorIiEENSt6vectorI6size_tEE6size_t14StreamOrDevice"]], "astype (c++ function)": [[0, "_CPPv46astype5array5Dtype14StreamOrDevice"]], "atleast_1d (c++ function)": [[0, "_CPPv410atleast_1dRK5array14StreamOrDevice"], [0, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice"]], "atleast_2d (c++ function)": [[0, "_CPPv410atleast_2dRK5array14StreamOrDevice"], [0, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice"]], "atleast_3d (c++ function)": [[0, "_CPPv410atleast_3dRK5array14StreamOrDevice"], [0, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice"]], "bitwise_and (c++ function)": [[0, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice"]], "bitwise_or (c++ function)": [[0, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice"]], "bitwise_xor (c++ function)": [[0, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice"]], "block_masked_mm (c++ function)": [[0, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice"]], "broadcast_arrays (c++ function)": [[0, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice"]], "broadcast_to (c++ function)": [[0, "_CPPv412broadcast_toRK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "ceil (c++ function)": [[0, "_CPPv44ceilRK5array14StreamOrDevice"]], "clip (c++ function)": [[0, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice"]], "concatenate (c++ function)": [[0, "_CPPv411concatenateRKNSt6vectorI5arrayEE14StreamOrDevice"], [0, "_CPPv411concatenateRKNSt6vectorI5arrayEEi14StreamOrDevice"]], "conjugate (c++ function)": [[0, "_CPPv49conjugateRK5array14StreamOrDevice"]], "conv1d (c++ function)": [[0, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice"]], "conv2d (c++ function)": [[0, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice"]], "conv3d (c++ function)": [[0, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice"]], "conv_general (c++ function)": [[0, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"], [0, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice"]], "copy (c++ function)": [[0, "_CPPv44copy5array14StreamOrDevice"]], "cos (c++ function)": [[0, "_CPPv43cosRK5array14StreamOrDevice"]], "cosh (c++ function)": [[0, "_CPPv44coshRK5array14StreamOrDevice"]], "cummax (c++ function)": [[0, "_CPPv46cummaxRK5arrayibb14StreamOrDevice"]], "cummin (c++ function)": [[0, "_CPPv46cumminRK5arrayibb14StreamOrDevice"]], "cumprod (c++ function)": [[0, "_CPPv47cumprodRK5arrayibb14StreamOrDevice"]], "cumsum (c++ function)": [[0, "_CPPv46cumsumRK5arrayibb14StreamOrDevice"]], "degrees (c++ function)": [[0, "_CPPv47degreesRK5array14StreamOrDevice"]], "depends (c++ function)": [[0, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE"]], "dequantize (c++ function)": [[0, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice"]], "diag (c++ function)": [[0, "_CPPv44diagRK5arrayi14StreamOrDevice"]], "diagonal (c++ function)": [[0, "_CPPv48diagonalRK5arrayiii14StreamOrDevice"]], "divide (c++ function)": [[0, "_CPPv46divideRK5arrayRK5array14StreamOrDevice"]], "divmod (c++ function)": [[0, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice"]], "equal (c++ function)": [[0, "_CPPv45equalRK5arrayRK5array14StreamOrDevice"]], "erf (c++ function)": [[0, "_CPPv43erfRK5array14StreamOrDevice"]], "erfinv (c++ function)": [[0, "_CPPv46erfinvRK5array14StreamOrDevice"]], "exp (c++ function)": [[0, "_CPPv43expRK5array14StreamOrDevice"]], "expand_dims (c++ function)": [[0, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice"]], "expm1 (c++ function)": [[0, "_CPPv45expm1RK5array14StreamOrDevice"]], "eye (c++ function)": [[0, "_CPPv43eyei14StreamOrDevice"], [0, "_CPPv43eyei5Dtype14StreamOrDevice"], [0, "_CPPv43eyeii14StreamOrDevice"], [0, "_CPPv43eyeiii14StreamOrDevice"], [0, "_CPPv43eyeiii5Dtype14StreamOrDevice"]], "flatten (c++ function)": [[0, "_CPPv47flattenRK5array14StreamOrDevice"], [0, "_CPPv47flattenRK5arrayii14StreamOrDevice"]], "floor (c++ function)": [[0, "_CPPv45floorRK5array14StreamOrDevice"]], "floor_divide (c++ function)": [[0, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice"]], "full (c++ function)": [[0, "_CPPv44fullNSt6vectorIiEE5array14StreamOrDevice"], [0, "_CPPv44fullNSt6vectorIiEE5array5Dtype14StreamOrDevice"], [0, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T14StreamOrDevice"], [0, "_CPPv4I0E4full5arrayNSt6vectorIiEE1T5Dtype14StreamOrDevice"]], "gather (c++ function)": [[0, "_CPPv46gatherRK5arrayRK5arrayiRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"]], "gather_mm (c++ function)": [[0, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice"]], "gather_qmm (c++ function)": [[0, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice"]], "greater (c++ function)": [[0, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice"]], "greater_equal (c++ function)": [[0, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice"]], "hadamard_transform (c++ function)": [[0, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice"]], "identity (c++ function)": [[0, "_CPPv48identityi14StreamOrDevice"], [0, "_CPPv48identityi5Dtype14StreamOrDevice"]], "inner (c++ function)": [[0, "_CPPv45innerRK5arrayRK5array14StreamOrDevice"]], "isclose (c++ function)": [[0, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice"]], "isinf (c++ function)": [[0, "_CPPv45isinfRK5array14StreamOrDevice"]], "isnan (c++ function)": [[0, "_CPPv45isnanRK5array14StreamOrDevice"]], "isneginf (c++ function)": [[0, "_CPPv48isneginfRK5array14StreamOrDevice"]], "isposinf (c++ function)": [[0, "_CPPv48isposinfRK5array14StreamOrDevice"]], "left_shift (c++ function)": [[0, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice"]], "less (c++ function)": [[0, "_CPPv44lessRK5arrayRK5array14StreamOrDevice"]], "less_equal (c++ function)": [[0, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice"]], "linspace (c++ function)": [[0, "_CPPv48linspaceddi5Dtype14StreamOrDevice"]], "log (c++ function)": [[0, "_CPPv43logRK5array14StreamOrDevice"]], "log10 (c++ function)": [[0, "_CPPv45log10RK5array14StreamOrDevice"]], "log1p (c++ function)": [[0, "_CPPv45log1pRK5array14StreamOrDevice"]], "log2 (c++ function)": [[0, "_CPPv44log2RK5array14StreamOrDevice"]], "logaddexp (c++ function)": [[0, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice"]], "logical_and (c++ function)": [[0, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice"]], "logical_not (c++ function)": [[0, "_CPPv411logical_notRK5array14StreamOrDevice"]], "logical_or (c++ function)": [[0, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice"]], "logsumexp (c++ function)": [[0, "_CPPv49logsumexpRK5array14StreamOrDevice"], [0, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv49logsumexpRK5arrayb14StreamOrDevice"], [0, "_CPPv49logsumexpRK5arrayib14StreamOrDevice"]], "matmul (c++ function)": [[0, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice"]], "max (c++ function)": [[0, "_CPPv43maxRK5array14StreamOrDevice"], [0, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43maxRK5arrayb14StreamOrDevice"], [0, "_CPPv43maxRK5arrayib14StreamOrDevice"]], "maximum (c++ function)": [[0, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice"]], "mean (c++ function)": [[0, "_CPPv44meanRK5array14StreamOrDevice"], [0, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv44meanRK5arrayb14StreamOrDevice"], [0, "_CPPv44meanRK5arrayib14StreamOrDevice"]], "meshgrid (c++ function)": [[0, "_CPPv48meshgridRKNSt6vectorI5arrayEEbNSt6stringE14StreamOrDevice"]], "min (c++ function)": [[0, "_CPPv43minRK5array14StreamOrDevice"], [0, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43minRK5arrayb14StreamOrDevice"], [0, "_CPPv43minRK5arrayib14StreamOrDevice"]], "minimum (c++ function)": [[0, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice"]], "moveaxis (c++ function)": [[0, "_CPPv48moveaxisRK5arrayii14StreamOrDevice"]], "multiply (c++ function)": [[0, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice"]], "nan_to_num (c++ function)": [[0, "_CPPv410nan_to_numRK5arrayfRKNSt8optionalIfEERKNSt8optionalIfEE14StreamOrDevice"]], "negative (c++ function)": [[0, "_CPPv48negativeRK5array14StreamOrDevice"]], "not_equal (c++ function)": [[0, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice"]], "number_of_elements (c++ function)": [[0, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice"]], "ones (c++ function)": [[0, "_CPPv44onesRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv44onesRKNSt6vectorIiEE5Dtype14StreamOrDevice"]], "ones_like (c++ function)": [[0, "_CPPv49ones_likeRK5array14StreamOrDevice"]], "operator!= (c++ function)": [[0, "_CPPv4I0Ene5array1TRK5array"], [0, "_CPPv4I0Ene5arrayRK5array1T"], [0, "_CPPv4neRK5arrayRK5array"]], "operator% (c++ function)": [[0, "_CPPv4I0Erm5array1TRK5array"], [0, "_CPPv4I0Erm5arrayRK5array1T"], [0, "_CPPv4rmRK5arrayRK5array"]], "operator& (c++ function)": [[0, "_CPPv4anRK5arrayRK5array"]], "operator&& (c++ function)": [[0, "_CPPv4aaRK5arrayRK5array"]], "operator* (c++ function)": [[0, "_CPPv4I0Eml5array1TRK5array"], [0, "_CPPv4I0Eml5arrayRK5array1T"], [0, "_CPPv4mlRK5arrayRK5array"]], "operator+ (c++ function)": [[0, "_CPPv4I0Epl5array1TRK5array"], [0, "_CPPv4I0Epl5arrayRK5array1T"], [0, "_CPPv4plRK5arrayRK5array"]], "operator- (c++ function)": [[0, "_CPPv4I0Emi5array1TRK5array"], [0, "_CPPv4I0Emi5arrayRK5array1T"], [0, "_CPPv4miRK5array"], [0, "_CPPv4miRK5arrayRK5array"]], "operator/ (c++ function)": [[0, "_CPPv4dvRK5arrayRK5array"], [0, "_CPPv4dvRK5arrayd"], [0, "_CPPv4dvdRK5array"]], "operator< (c++ function)": [[0, "_CPPv4I0Elt5array1TRK5array"], [0, "_CPPv4I0Elt5arrayRK5array1T"], [0, "_CPPv4ltRK5arrayRK5array"]], "operator<< (c++ function)": [[0, "_CPPv4lsRK5arrayRK5array"]], "operator<= (c++ function)": [[0, "_CPPv4I0Ele5array1TRK5array"], [0, "_CPPv4I0Ele5arrayRK5array1T"], [0, "_CPPv4leRK5arrayRK5array"]], "operator== (c++ function)": [[0, "_CPPv4I0Eeq5array1TRK5array"], [0, "_CPPv4I0Eeq5arrayRK5array1T"], [0, "_CPPv4eqRK5arrayRK5array"]], "operator> (c++ function)": [[0, "_CPPv4I0Egt5array1TRK5array"], [0, "_CPPv4I0Egt5arrayRK5array1T"], [0, "_CPPv4gtRK5arrayRK5array"]], "operator>= (c++ function)": [[0, "_CPPv4I0Ege5array1TRK5array"], [0, "_CPPv4I0Ege5arrayRK5array1T"], [0, "_CPPv4geRK5arrayRK5array"]], "operator>> (c++ function)": [[0, "_CPPv4rsRK5arrayRK5array"]], "operator^ (c++ function)": [[0, "_CPPv4eoRK5arrayRK5array"]], "operator| (c++ function)": [[0, "_CPPv4orRK5arrayRK5array"]], "operator|| (c++ function)": [[0, "_CPPv4ooRK5arrayRK5array"]], "outer (c++ function)": [[0, "_CPPv45outerRK5arrayRK5array14StreamOrDevice"]], "pad (c++ function)": [[0, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayKNSt6stringE14StreamOrDevice"], [0, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayKNSt6stringE14StreamOrDevice"], [0, "_CPPv43padRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEERKNSt6vectorIiEERK5arrayKNSt6stringE14StreamOrDevice"], [0, "_CPPv43padRK5arrayiRK5arrayKNSt6stringE14StreamOrDevice"]], "partition (c++ function)": [[0, "_CPPv49partitionRK5arrayi14StreamOrDevice"], [0, "_CPPv49partitionRK5arrayii14StreamOrDevice"]], "power (c++ function)": [[0, "_CPPv45powerRK5arrayRK5array14StreamOrDevice"]], "prod (c++ function)": [[0, "_CPPv44prodRK5array14StreamOrDevice"], [0, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv44prodRK5arrayb14StreamOrDevice"], [0, "_CPPv44prodRK5arrayib14StreamOrDevice"]], "quantize (c++ function)": [[0, "_CPPv48quantizeRK5arrayii14StreamOrDevice"]], "quantized_matmul (c++ function)": [[0, "_CPPv416quantized_matmulRK5arrayRK5arrayRK5arrayRK5arraybii14StreamOrDevice"]], "radians (c++ function)": [[0, "_CPPv47radiansRK5array14StreamOrDevice"]], "reciprocal (c++ function)": [[0, "_CPPv410reciprocalRK5array14StreamOrDevice"]], "remainder (c++ function)": [[0, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice"]], "repeat (c++ function)": [[0, "_CPPv46repeatRK5arrayi14StreamOrDevice"], [0, "_CPPv46repeatRK5arrayii14StreamOrDevice"]], "reshape (c++ function)": [[0, "_CPPv47reshapeRK5arrayNSt6vectorIiEE14StreamOrDevice"]], "right_shift (c++ function)": [[0, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice"]], "round (c++ function)": [[0, "_CPPv45roundRK5array14StreamOrDevice"], [0, "_CPPv45roundRK5arrayi14StreamOrDevice"]], "rsqrt (c++ function)": [[0, "_CPPv45rsqrtRK5array14StreamOrDevice"]], "scatter (c++ function)": [[0, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "scatter_add (c++ function)": [[0, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "scatter_max (c++ function)": [[0, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "scatter_min (c++ function)": [[0, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "scatter_prod (c++ function)": [[0, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice"], [0, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice"]], "sigmoid (c++ function)": [[0, "_CPPv47sigmoidRK5array14StreamOrDevice"]], "sign (c++ function)": [[0, "_CPPv44signRK5array14StreamOrDevice"]], "sin (c++ function)": [[0, "_CPPv43sinRK5array14StreamOrDevice"]], "sinh (c++ function)": [[0, "_CPPv44sinhRK5array14StreamOrDevice"]], "slice (c++ function)": [[0, "_CPPv45sliceRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv45sliceRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"]], "slice_update (c++ function)": [[0, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv412slice_updateRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEE14StreamOrDevice"]], "softmax (c++ function)": [[0, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv47softmaxRK5arrayb14StreamOrDevice"], [0, "_CPPv47softmaxRK5arrayib14StreamOrDevice"]], "sort (c++ function)": [[0, "_CPPv44sortRK5array14StreamOrDevice"], [0, "_CPPv44sortRK5arrayi14StreamOrDevice"]], "split (c++ function)": [[0, "_CPPv45splitRK5arrayRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv45splitRK5arrayRKNSt6vectorIiEEi14StreamOrDevice"], [0, "_CPPv45splitRK5arrayi14StreamOrDevice"], [0, "_CPPv45splitRK5arrayii14StreamOrDevice"]], "sqrt (c++ function)": [[0, "_CPPv44sqrtRK5array14StreamOrDevice"]], "square (c++ function)": [[0, "_CPPv46squareRK5array14StreamOrDevice"]], "squeeze (c++ function)": [[0, "_CPPv47squeezeRK5array14StreamOrDevice"], [0, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv47squeezeRK5arrayi14StreamOrDevice"]], "stack (c++ function)": [[0, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice"], [0, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice"]], "std (c++ function)": [[0, "_CPPv4StRK5array14StreamOrDevice"], [0, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"], [0, "_CPPv4StRK5arraybi14StreamOrDevice"], [0, "_CPPv4StRK5arrayibi14StreamOrDevice"]], "stop_gradient (c++ function)": [[0, "_CPPv413stop_gradientRK5array14StreamOrDevice"]], "subtract (c++ function)": [[0, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice"]], "sum (c++ function)": [[0, "_CPPv43sumRK5array14StreamOrDevice"], [0, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice"], [0, "_CPPv43sumRK5arrayb14StreamOrDevice"], [0, "_CPPv43sumRK5arrayib14StreamOrDevice"]], "swapaxes (c++ function)": [[0, "_CPPv48swapaxesRK5arrayii14StreamOrDevice"]], "take (c++ function)": [[0, "_CPPv44takeRK5arrayRK5array14StreamOrDevice"], [0, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice"]], "take_along_axis (c++ function)": [[0, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice"]], "tan (c++ function)": [[0, "_CPPv43tanRK5array14StreamOrDevice"]], "tanh (c++ function)": [[0, "_CPPv44tanhRK5array14StreamOrDevice"]], "tensordot (c++ function)": [[0, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice"], [0, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice"]], "tile (c++ function)": [[0, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice"]], "topk (c++ function)": [[0, "_CPPv44topkRK5arrayi14StreamOrDevice"], [0, "_CPPv44topkRK5arrayii14StreamOrDevice"]], "trace (c++ function)": [[0, "_CPPv45traceRK5array14StreamOrDevice"], [0, "_CPPv45traceRK5arrayiii14StreamOrDevice"], [0, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice"]], "transpose (c++ function)": [[0, "_CPPv49transposeRK5array14StreamOrDevice"], [0, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice"], [0, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice"]], "tri (c++ function)": [[0, "_CPPv43trii5Dtype14StreamOrDevice"], [0, "_CPPv43triiii5Dtype14StreamOrDevice"]], "tril (c++ function)": [[0, "_CPPv44tril5arrayi14StreamOrDevice"]], "triu (c++ function)": [[0, "_CPPv44triu5arrayi14StreamOrDevice"]], "var (c++ function)": [[0, "_CPPv43varRK5array14StreamOrDevice"], [0, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice"], [0, "_CPPv43varRK5arraybi14StreamOrDevice"], [0, "_CPPv43varRK5arrayibi14StreamOrDevice"]], "view (c++ function)": [[0, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice"]], "where (c++ function)": [[0, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice"]], "zeros (c++ function)": [[0, "_CPPv45zerosRKNSt6vectorIiEE14StreamOrDevice"], [0, "_CPPv45zerosRKNSt6vectorIiEE5Dtype14StreamOrDevice"]], "zeros_like (c++ function)": [[0, "_CPPv410zeros_likeRK5array14StreamOrDevice"]], "device (class in mlx.core)": [[8, "mlx.core.Device"]], "__init__() (device method)": [[8, "mlx.core.Device.__init__"]], "dtype (class in mlx.core)": [[9, "mlx.core.Dtype"]], "__init__() (dtype method)": [[9, "mlx.core.Dtype.__init__"]], "dtypecategory (class in mlx.core)": [[10, "mlx.core.DtypeCategory"]], "__init__() (dtypecategory method)": [[10, "mlx.core.DtypeCategory.__init__"]], "abs() (in module mlx.core)": [[11, "mlx.core.abs"]], "add() (in module mlx.core)": [[12, "mlx.core.add"]], "addmm() (in module mlx.core)": [[13, "mlx.core.addmm"]], "all() (in module mlx.core)": [[14, "mlx.core.all"]], "allclose() (in module mlx.core)": [[15, "mlx.core.allclose"]], "any() (in module mlx.core)": [[16, "mlx.core.any"]], "arange() (in module mlx.core)": [[17, "mlx.core.arange"]], "arccos() (in module mlx.core)": [[18, "mlx.core.arccos"]], "arccosh() (in module mlx.core)": [[19, "mlx.core.arccosh"]], "arcsin() (in module mlx.core)": [[20, "mlx.core.arcsin"]], "arcsinh() (in module mlx.core)": [[21, "mlx.core.arcsinh"]], "arctan() (in module mlx.core)": [[22, "mlx.core.arctan"]], "arctan2() (in module mlx.core)": [[23, "mlx.core.arctan2"]], "arctanh() (in module mlx.core)": [[24, "mlx.core.arctanh"]], "argmax() (in module mlx.core)": [[25, "mlx.core.argmax"]], "argmin() (in module mlx.core)": [[26, "mlx.core.argmin"]], "argpartition() (in module mlx.core)": [[27, "mlx.core.argpartition"]], "argsort() (in module mlx.core)": [[28, "mlx.core.argsort"]], "__init__() (array method)": [[29, "mlx.core.array.__init__"]], "array (class in mlx.core)": [[29, "mlx.core.array"]], "t (array property)": [[30, "mlx.core.array.T"]], "abs() (array method)": [[31, "mlx.core.array.abs"]], "all() (array method)": [[32, "mlx.core.array.all"]], "any() (array method)": [[33, "mlx.core.array.any"]], "argmax() (array method)": [[34, "mlx.core.array.argmax"]], "argmin() (array method)": [[35, "mlx.core.array.argmin"]], "astype() (array method)": [[36, "mlx.core.array.astype"]], "at (array property)": [[37, "mlx.core.array.at"]], "conj() (array method)": [[38, "mlx.core.array.conj"]], "cos() (array method)": [[39, "mlx.core.array.cos"]], "cummax() (array method)": [[40, "mlx.core.array.cummax"]], "cummin() (array method)": [[41, "mlx.core.array.cummin"]], "cumprod() (array method)": [[42, "mlx.core.array.cumprod"]], "cumsum() (array method)": [[43, "mlx.core.array.cumsum"]], "diag() (array method)": [[44, "mlx.core.array.diag"]], "diagonal() (array method)": [[45, "mlx.core.array.diagonal"]], "dtype (array property)": [[46, "mlx.core.array.dtype"]], "exp() (array method)": [[47, "mlx.core.array.exp"]], "flatten() (array method)": [[48, "mlx.core.array.flatten"]], "item() (array method)": [[49, "mlx.core.array.item"]], "itemsize (array property)": [[50, "mlx.core.array.itemsize"]], "log() (array method)": [[51, "mlx.core.array.log"]], "log10() (array method)": [[52, "mlx.core.array.log10"]], "log1p() (array method)": [[53, "mlx.core.array.log1p"]], "log2() (array method)": [[54, "mlx.core.array.log2"]], "logsumexp() (array method)": [[55, "mlx.core.array.logsumexp"]], "max() (array method)": [[56, "mlx.core.array.max"]], "mean() (array method)": [[57, "mlx.core.array.mean"]], "min() (array method)": [[58, "mlx.core.array.min"]], "moveaxis() (array method)": [[59, "mlx.core.array.moveaxis"]], "nbytes (array property)": [[60, "mlx.core.array.nbytes"]], "ndim (array property)": [[61, "mlx.core.array.ndim"]], "prod() (array method)": [[62, "mlx.core.array.prod"]], "reciprocal() (array method)": [[63, "mlx.core.array.reciprocal"]], "reshape() (array method)": [[64, "mlx.core.array.reshape"]], "round() (array method)": [[65, "mlx.core.array.round"]], "rsqrt() (array method)": [[66, "mlx.core.array.rsqrt"]], "shape (array property)": [[67, "mlx.core.array.shape"]], "sin() (array method)": [[68, "mlx.core.array.sin"]], "size (array property)": [[69, "mlx.core.array.size"]], "split() (array method)": [[70, "mlx.core.array.split"]], "sqrt() (array method)": [[71, "mlx.core.array.sqrt"]], "square() (array method)": [[72, "mlx.core.array.square"]], "squeeze() (array method)": [[73, "mlx.core.array.squeeze"]], "sum() (array method)": [[74, "mlx.core.array.sum"]], "swapaxes() (array method)": [[75, "mlx.core.array.swapaxes"]], "tolist() (array method)": [[76, "mlx.core.array.tolist"]], "transpose() (array method)": [[77, "mlx.core.array.transpose"]], "var() (array method)": [[78, "mlx.core.array.var"]], "view() (array method)": [[79, "mlx.core.array.view"]], "array_equal() (in module mlx.core)": [[80, "mlx.core.array_equal"]], "as_strided() (in module mlx.core)": [[81, "mlx.core.as_strided"]], "atleast_1d() (in module mlx.core)": [[82, "mlx.core.atleast_1d"]], "atleast_2d() (in module mlx.core)": [[83, "mlx.core.atleast_2d"]], "atleast_3d() (in module mlx.core)": [[84, "mlx.core.atleast_3d"]], "bitwise_and() (in module mlx.core)": [[85, "mlx.core.bitwise_and"]], "bitwise_or() (in module mlx.core)": [[86, "mlx.core.bitwise_or"]], "bitwise_xor() (in module mlx.core)": [[87, "mlx.core.bitwise_xor"]], "block_masked_mm() (in module mlx.core)": [[88, "mlx.core.block_masked_mm"]], "broadcast_to() (in module mlx.core)": [[89, "mlx.core.broadcast_to"]], "ceil() (in module mlx.core)": [[90, "mlx.core.ceil"]], "clip() (in module mlx.core)": [[91, "mlx.core.clip"]], "compile() (in module mlx.core)": [[92, "mlx.core.compile"]], "concatenate() (in module mlx.core)": [[93, "mlx.core.concatenate"]], "conj() (in module mlx.core)": [[94, "mlx.core.conj"]], "conjugate() (in module mlx.core)": [[95, "mlx.core.conjugate"]], "conv1d() (in module mlx.core)": [[96, "mlx.core.conv1d"]], "conv2d() (in module mlx.core)": [[97, "mlx.core.conv2d"]], "conv_general() (in module mlx.core)": [[98, "mlx.core.conv_general"]], "convolve() (in module mlx.core)": [[99, "mlx.core.convolve"]], "cos() (in module mlx.core)": [[100, "mlx.core.cos"]], "cosh() (in module mlx.core)": [[101, "mlx.core.cosh"]], "cummax() (in module mlx.core)": [[102, "mlx.core.cummax"]], "cummin() (in module mlx.core)": [[103, "mlx.core.cummin"]], "cumprod() (in module mlx.core)": [[104, "mlx.core.cumprod"]], "cumsum() (in module mlx.core)": [[105, "mlx.core.cumsum"]], "__init__() (custom_function method)": [[106, "mlx.core.custom_function.__init__"]], "custom_function (class in mlx.core)": [[106, "mlx.core.custom_function"]], "default_device() (in module mlx.core)": [[107, "mlx.core.default_device"]], "default_stream() (in module mlx.core)": [[108, "mlx.core.default_stream"]], "degrees() (in module mlx.core)": [[109, "mlx.core.degrees"]], "dequantize() (in module mlx.core)": [[110, "mlx.core.dequantize"]], "diag() (in module mlx.core)": [[111, "mlx.core.diag"]], "diagonal() (in module mlx.core)": [[112, "mlx.core.diagonal"]], "disable_compile() (in module mlx.core)": [[113, "mlx.core.disable_compile"]], "group (class in mlx.core.distributed)": [[114, "mlx.core.distributed.Group"]], "__init__() (group method)": [[114, "mlx.core.distributed.Group.__init__"]], "all_gather() (in module mlx.core.distributed)": [[115, "mlx.core.distributed.all_gather"]], "all_sum() (in module mlx.core.distributed)": [[116, "mlx.core.distributed.all_sum"]], "init() (in module mlx.core.distributed)": [[117, "mlx.core.distributed.init"]], "is_available() (in module mlx.core.distributed)": [[118, "mlx.core.distributed.is_available"]], "divide() (in module mlx.core)": [[119, "mlx.core.divide"]], "divmod() (in module mlx.core)": [[120, "mlx.core.divmod"]], "einsum() (in module mlx.core)": [[121, "mlx.core.einsum"]], "einsum_path() (in module mlx.core)": [[122, "mlx.core.einsum_path"]], "enable_compile() (in module mlx.core)": [[123, "mlx.core.enable_compile"]], "equal() (in module mlx.core)": [[124, "mlx.core.equal"]], "erf() (in module mlx.core)": [[125, "mlx.core.erf"]], "erfinv() (in module mlx.core)": [[126, "mlx.core.erfinv"]], "eval() (in module mlx.core)": [[127, "mlx.core.eval"]], "exp() (in module mlx.core)": [[128, "mlx.core.exp"]], "expand_dims() (in module mlx.core)": [[129, "mlx.core.expand_dims"]], "expm1() (in module mlx.core)": [[130, "mlx.core.expm1"]], "eye() (in module mlx.core)": [[131, "mlx.core.eye"]], "layer_norm() (in module mlx.core.fast)": [[132, "mlx.core.fast.layer_norm"]], "rms_norm() (in module mlx.core.fast)": [[133, "mlx.core.fast.rms_norm"]], "rope() (in module mlx.core.fast)": [[134, "mlx.core.fast.rope"]], "scaled_dot_product_attention() (in module mlx.core.fast)": [[135, "mlx.core.fast.scaled_dot_product_attention"]], "fft() (in module mlx.core.fft)": [[136, "mlx.core.fft.fft"]], "fft2() (in module mlx.core.fft)": [[137, "mlx.core.fft.fft2"]], "fftn() (in module mlx.core.fft)": [[138, "mlx.core.fft.fftn"]], "ifft() (in module mlx.core.fft)": [[139, "mlx.core.fft.ifft"]], "ifft2() (in module mlx.core.fft)": [[140, "mlx.core.fft.ifft2"]], "ifftn() (in module mlx.core.fft)": [[141, "mlx.core.fft.ifftn"]], "irfft() (in module mlx.core.fft)": [[142, "mlx.core.fft.irfft"]], "irfft2() (in module mlx.core.fft)": [[143, "mlx.core.fft.irfft2"]], "irfftn() (in module mlx.core.fft)": [[144, "mlx.core.fft.irfftn"]], "rfft() (in module mlx.core.fft)": [[145, "mlx.core.fft.rfft"]], "rfft2() (in module mlx.core.fft)": [[146, "mlx.core.fft.rfft2"]], "rfftn() (in module mlx.core.fft)": [[147, "mlx.core.fft.rfftn"]], "flatten() (in module mlx.core)": [[148, "mlx.core.flatten"]], "floor() (in module mlx.core)": [[149, "mlx.core.floor"]], "floor_divide() (in module mlx.core)": [[150, "mlx.core.floor_divide"]], "full() (in module mlx.core)": [[151, "mlx.core.full"]], "gather_mm() (in module mlx.core)": [[152, "mlx.core.gather_mm"]], "gather_qmm() (in module mlx.core)": [[153, "mlx.core.gather_qmm"]], "grad() (in module mlx.core)": [[154, "mlx.core.grad"]], "greater() (in module mlx.core)": [[155, "mlx.core.greater"]], "greater_equal() (in module mlx.core)": [[156, "mlx.core.greater_equal"]], "hadamard_transform() (in module mlx.core)": [[157, "mlx.core.hadamard_transform"]], "identity() (in module mlx.core)": [[158, "mlx.core.identity"]], "inner() (in module mlx.core)": [[159, "mlx.core.inner"]], "isclose() (in module mlx.core)": [[160, "mlx.core.isclose"]], "isinf() (in module mlx.core)": [[161, "mlx.core.isinf"]], "isnan() (in module mlx.core)": [[162, "mlx.core.isnan"]], "isneginf() (in module mlx.core)": [[163, "mlx.core.isneginf"]], "isposinf() (in module mlx.core)": [[164, "mlx.core.isposinf"]], "issubdtype() (in module mlx.core)": [[165, "mlx.core.issubdtype"]], "jvp() (in module mlx.core)": [[166, "mlx.core.jvp"]], "left_shift() (in module mlx.core)": [[167, "mlx.core.left_shift"]], "less() (in module mlx.core)": [[168, "mlx.core.less"]], "less_equal() (in module mlx.core)": [[169, "mlx.core.less_equal"]], "cholesky() (in module mlx.core.linalg)": [[170, "mlx.core.linalg.cholesky"]], "cholesky_inv() (in module mlx.core.linalg)": [[171, "mlx.core.linalg.cholesky_inv"]], "inv() (in module mlx.core.linalg)": [[172, "mlx.core.linalg.inv"]], "norm() (in module mlx.core.linalg)": [[173, "mlx.core.linalg.norm"]], "qr() (in module mlx.core.linalg)": [[174, "mlx.core.linalg.qr"]], "svd() (in module mlx.core.linalg)": [[175, "mlx.core.linalg.svd"]], "tri_inv() (in module mlx.core.linalg)": [[176, "mlx.core.linalg.tri_inv"]], "linspace() (in module mlx.core)": [[177, "mlx.core.linspace"]], "load() (in module mlx.core)": [[178, "mlx.core.load"]], "log() (in module mlx.core)": [[179, "mlx.core.log"]], "log10() (in module mlx.core)": [[180, "mlx.core.log10"]], "log1p() (in module mlx.core)": [[181, "mlx.core.log1p"]], "log2() (in module mlx.core)": [[182, "mlx.core.log2"]], "logaddexp() (in module mlx.core)": [[183, "mlx.core.logaddexp"]], "logical_and() (in module mlx.core)": [[184, "mlx.core.logical_and"]], "logical_not() (in module mlx.core)": [[185, "mlx.core.logical_not"]], "logical_or() (in module mlx.core)": [[186, "mlx.core.logical_or"]], "logsumexp() (in module mlx.core)": [[187, "mlx.core.logsumexp"]], "matmul() (in module mlx.core)": [[188, "mlx.core.matmul"]], "max() (in module mlx.core)": [[189, "mlx.core.max"]], "maximum() (in module mlx.core)": [[190, "mlx.core.maximum"]], "mean() (in module mlx.core)": [[191, "mlx.core.mean"]], "meshgrid() (in module mlx.core)": [[192, "mlx.core.meshgrid"]], "clear_cache() (in module mlx.core.metal)": [[193, "mlx.core.metal.clear_cache"]], "device_info() (in module mlx.core.metal)": [[194, "mlx.core.metal.device_info"]], "get_active_memory() (in module mlx.core.metal)": [[195, "mlx.core.metal.get_active_memory"]], "get_cache_memory() (in module mlx.core.metal)": [[196, "mlx.core.metal.get_cache_memory"]], "get_peak_memory() (in module mlx.core.metal)": [[197, "mlx.core.metal.get_peak_memory"]], "is_available() (in module mlx.core.metal)": [[198, "mlx.core.metal.is_available"]], "reset_peak_memory() (in module mlx.core.metal)": [[199, "mlx.core.metal.reset_peak_memory"]], "set_cache_limit() (in module mlx.core.metal)": [[200, "mlx.core.metal.set_cache_limit"]], "set_memory_limit() (in module mlx.core.metal)": [[201, "mlx.core.metal.set_memory_limit"]], "start_capture() (in module mlx.core.metal)": [[202, "mlx.core.metal.start_capture"]], "stop_capture() (in module mlx.core.metal)": [[203, "mlx.core.metal.stop_capture"]], "min() (in module mlx.core)": [[204, "mlx.core.min"]], "minimum() (in module mlx.core)": [[205, "mlx.core.minimum"]], "moveaxis() (in module mlx.core)": [[206, "mlx.core.moveaxis"]], "multiply() (in module mlx.core)": [[207, "mlx.core.multiply"]], "nan_to_num() (in module mlx.core)": [[208, "mlx.core.nan_to_num"]], "negative() (in module mlx.core)": [[209, "mlx.core.negative"]], "new_stream() (in module mlx.core)": [[210, "mlx.core.new_stream"]], "not_equal() (in module mlx.core)": [[211, "mlx.core.not_equal"]], "ones() (in module mlx.core)": [[212, "mlx.core.ones"]], "ones_like() (in module mlx.core)": [[213, "mlx.core.ones_like"]], "outer() (in module mlx.core)": [[214, "mlx.core.outer"]], "pad() (in module mlx.core)": [[215, "mlx.core.pad"]], "partition() (in module mlx.core)": [[216, "mlx.core.partition"]], "power() (in module mlx.core)": [[217, "mlx.core.power"]], "prod() (in module mlx.core)": [[218, "mlx.core.prod"]], "quantize() (in module mlx.core)": [[219, "mlx.core.quantize"]], "quantized_matmul() (in module mlx.core)": [[220, "mlx.core.quantized_matmul"]], "radians() (in module mlx.core)": [[221, "mlx.core.radians"]], "bernoulli() (in module mlx.core.random)": [[222, "mlx.core.random.bernoulli"]], "categorical() (in module mlx.core.random)": [[223, "mlx.core.random.categorical"]], "gumbel() (in module mlx.core.random)": [[224, "mlx.core.random.gumbel"]], "key() (in module mlx.core.random)": [[225, "mlx.core.random.key"]], "laplace() (in module mlx.core.random)": [[226, "mlx.core.random.laplace"]], "multivariate_normal() (in module mlx.core.random)": [[227, "mlx.core.random.multivariate_normal"]], "normal() (in module mlx.core.random)": [[228, "mlx.core.random.normal"]], "randint() (in module mlx.core.random)": [[229, "mlx.core.random.randint"]], "seed() (in module mlx.core.random)": [[230, "mlx.core.random.seed"]], "split() (in module mlx.core.random)": [[231, "mlx.core.random.split"]], "truncated_normal() (in module mlx.core.random)": [[232, "mlx.core.random.truncated_normal"]], "uniform() (in module mlx.core.random)": [[233, "mlx.core.random.uniform"]], "reciprocal() (in module mlx.core)": [[234, "mlx.core.reciprocal"]], "remainder() (in module mlx.core)": [[235, "mlx.core.remainder"]], "repeat() (in module mlx.core)": [[236, "mlx.core.repeat"]], "reshape() (in module mlx.core)": [[237, "mlx.core.reshape"]], "right_shift() (in module mlx.core)": [[238, "mlx.core.right_shift"]], "round() (in module mlx.core)": [[239, "mlx.core.round"]], "rsqrt() (in module mlx.core)": [[240, "mlx.core.rsqrt"]], "save() (in module mlx.core)": [[241, "mlx.core.save"]], "save_gguf() (in module mlx.core)": [[242, "mlx.core.save_gguf"]], "save_safetensors() (in module mlx.core)": [[243, "mlx.core.save_safetensors"]], "savez() (in module mlx.core)": [[244, "mlx.core.savez"]], "savez_compressed() (in module mlx.core)": [[245, "mlx.core.savez_compressed"]], "set_default_device() (in module mlx.core)": [[246, "mlx.core.set_default_device"]], "set_default_stream() (in module mlx.core)": [[247, "mlx.core.set_default_stream"]], "sigmoid() (in module mlx.core)": [[248, "mlx.core.sigmoid"]], "sign() (in module mlx.core)": [[249, "mlx.core.sign"]], "sin() (in module mlx.core)": [[250, "mlx.core.sin"]], "sinh() (in module mlx.core)": [[251, "mlx.core.sinh"]], "softmax() (in module mlx.core)": [[252, "mlx.core.softmax"]], "sort() (in module mlx.core)": [[253, "mlx.core.sort"]], "split() (in module mlx.core)": [[254, "mlx.core.split"]], "sqrt() (in module mlx.core)": [[255, "mlx.core.sqrt"]], "square() (in module mlx.core)": [[256, "mlx.core.square"]], "squeeze() (in module mlx.core)": [[257, "mlx.core.squeeze"]], "stack() (in module mlx.core)": [[258, "mlx.core.stack"]], "std() (in module mlx.core)": [[259, "mlx.core.std"]], "stop_gradient() (in module mlx.core)": [[260, "mlx.core.stop_gradient"]], "stream() (in module mlx.core)": [[261, "mlx.core.stream"]], "subtract() (in module mlx.core)": [[262, "mlx.core.subtract"]], "sum() (in module mlx.core)": [[263, "mlx.core.sum"]], "swapaxes() (in module mlx.core)": [[264, "mlx.core.swapaxes"]], "synchronize() (in module mlx.core)": [[265, "mlx.core.synchronize"]], "take() (in module mlx.core)": [[266, "mlx.core.take"]], "take_along_axis() (in module mlx.core)": [[267, "mlx.core.take_along_axis"]], "tan() (in module mlx.core)": [[268, "mlx.core.tan"]], "tanh() (in module mlx.core)": [[269, "mlx.core.tanh"]], "tensordot() (in module mlx.core)": [[270, "mlx.core.tensordot"]], "tile() (in module mlx.core)": [[271, "mlx.core.tile"]], "topk() (in module mlx.core)": [[272, "mlx.core.topk"]], "trace() (in module mlx.core)": [[273, "mlx.core.trace"]], "transpose() (in module mlx.core)": [[274, "mlx.core.transpose"]], "tri() (in module mlx.core)": [[275, "mlx.core.tri"]], "tril() (in module mlx.core)": [[276, "mlx.core.tril"]], "triu() (in module mlx.core)": [[277, "mlx.core.triu"]], "value_and_grad() (in module mlx.core)": [[278, "mlx.core.value_and_grad"]], "var() (in module mlx.core)": [[279, "mlx.core.var"]], "view() (in module mlx.core)": [[280, "mlx.core.view"]], "vjp() (in module mlx.core)": [[281, "mlx.core.vjp"]], "vmap() (in module mlx.core)": [[282, "mlx.core.vmap"]], "where() (in module mlx.core)": [[283, "mlx.core.where"]], "zeros() (in module mlx.core)": [[284, "mlx.core.zeros"]], "zeros_like() (in module mlx.core)": [[285, "mlx.core.zeros_like"]], "quantize() (in module mlx.nn)": [[286, "mlx.nn.quantize"]], "value_and_grad() (in module mlx.nn)": [[287, "mlx.nn.value_and_grad"]], "clip_grad_norm() (in module mlx.optimizers)": [[288, "mlx.optimizers.clip_grad_norm"]], "tree_flatten() (in module mlx.utils)": [[289, "mlx.utils.tree_flatten"]], "tree_map() (in module mlx.utils)": [[290, "mlx.utils.tree_map"]], "tree_map_with_path() (in module mlx.utils)": [[291, "mlx.utils.tree_map_with_path"]], "tree_reduce() (in module mlx.utils)": [[292, "mlx.utils.tree_reduce"]], "tree_unflatten() (in module mlx.utils)": [[293, "mlx.utils.tree_unflatten"]], "stream (class in mlx.core)": [[294, "mlx.core.Stream"]], "__init__() (stream method)": [[294, "mlx.core.Stream.__init__"]], "alibi (class in mlx.nn)": [[304, "mlx.nn.ALiBi"]], "avgpool1d (class in mlx.nn)": [[305, "mlx.nn.AvgPool1d"]], "avgpool2d (class in mlx.nn)": [[306, "mlx.nn.AvgPool2d"]], "batchnorm (class in mlx.nn)": [[307, "mlx.nn.BatchNorm"]], "conv1d (class in mlx.nn)": [[308, "mlx.nn.Conv1d"]], "conv2d (class in mlx.nn)": [[309, "mlx.nn.Conv2d"]], "conv3d (class in mlx.nn)": [[310, "mlx.nn.Conv3d"]], "dropout (class in mlx.nn)": [[311, "mlx.nn.Dropout"]], "dropout2d (class in mlx.nn)": [[312, "mlx.nn.Dropout2d"]], "dropout3d (class in mlx.nn)": [[313, "mlx.nn.Dropout3d"]], "embedding (class in mlx.nn)": [[314, "mlx.nn.Embedding"]], "gelu (class in mlx.nn)": [[315, "mlx.nn.GELU"], [381, "mlx.nn.gelu"]], "glu (class in mlx.nn)": [[316, "mlx.nn.GLU"], [384, "mlx.nn.glu"]], "gru (class in mlx.nn)": [[317, "mlx.nn.GRU"]], "groupnorm (class in mlx.nn)": [[318, "mlx.nn.GroupNorm"]], "hardshrink (class in mlx.nn)": [[319, "mlx.nn.HardShrink"]], "hardtanh (class in mlx.nn)": [[320, "mlx.nn.HardTanh"]], "hardswish (class in mlx.nn)": [[321, "mlx.nn.Hardswish"], [387, "mlx.nn.hardswish"]], "instancenorm (class in mlx.nn)": [[322, "mlx.nn.InstanceNorm"]], "lstm (class in mlx.nn)": [[323, "mlx.nn.LSTM"]], "layernorm (class in mlx.nn)": [[324, "mlx.nn.LayerNorm"]], "leakyrelu (class in mlx.nn)": [[325, "mlx.nn.LeakyReLU"]], "linear (class in mlx.nn)": [[326, "mlx.nn.Linear"]], "maxpool1d (class in mlx.nn)": [[327, "mlx.nn.MaxPool1d"]], "maxpool2d (class in mlx.nn)": [[328, "mlx.nn.MaxPool2d"]], "mish (class in mlx.nn)": [[329, "mlx.nn.Mish"], [405, "mlx.nn.mish"]], "apply() (module method)": [[330, "mlx.nn.Module.apply"]], "apply_to_modules() (module method)": [[331, "mlx.nn.Module.apply_to_modules"]], "children() (module method)": [[332, "mlx.nn.Module.children"]], "eval() (module method)": [[333, "mlx.nn.Module.eval"]], "filter_and_map() (module method)": [[334, "mlx.nn.Module.filter_and_map"]], "freeze() (module method)": [[335, "mlx.nn.Module.freeze"]], "leaf_modules() (module method)": [[336, "mlx.nn.Module.leaf_modules"]], "load_weights() (module method)": [[337, "mlx.nn.Module.load_weights"]], "modules() (module method)": [[338, "mlx.nn.Module.modules"]], "named_modules() (module method)": [[339, "mlx.nn.Module.named_modules"]], "parameters() (module method)": [[340, "mlx.nn.Module.parameters"]], "save_weights() (module method)": [[341, "mlx.nn.Module.save_weights"]], "set_dtype() (module method)": [[342, "mlx.nn.Module.set_dtype"]], "state (module property)": [[343, "mlx.nn.Module.state"]], "train() (module method)": [[344, "mlx.nn.Module.train"]], "trainable_parameters() (module method)": [[345, "mlx.nn.Module.trainable_parameters"]], "training (module property)": [[346, "mlx.nn.Module.training"]], "unfreeze() (module method)": [[347, "mlx.nn.Module.unfreeze"]], "update() (module method)": [[348, "mlx.nn.Module.update"]], "update_modules() (module method)": [[349, "mlx.nn.Module.update_modules"]], "multiheadattention (class in mlx.nn)": [[350, "mlx.nn.MultiHeadAttention"]], "prelu (class in mlx.nn)": [[351, "mlx.nn.PReLU"], [406, "mlx.nn.prelu"]], "quantizedembedding (class in mlx.nn)": [[352, "mlx.nn.QuantizedEmbedding"]], "quantizedlinear (class in mlx.nn)": [[353, "mlx.nn.QuantizedLinear"]], "rmsnorm (class in mlx.nn)": [[354, "mlx.nn.RMSNorm"]], "rnn (class in mlx.nn)": [[355, "mlx.nn.RNN"]], "relu (class in mlx.nn)": [[356, "mlx.nn.ReLU"], [407, "mlx.nn.relu"]], "relu6 (class in mlx.nn)": [[357, "mlx.nn.ReLU6"], [408, "mlx.nn.relu6"]], "rope (class in mlx.nn)": [[358, "mlx.nn.RoPE"]], "selu (class in mlx.nn)": [[359, "mlx.nn.SELU"], [409, "mlx.nn.selu"]], "sequential (class in mlx.nn)": [[360, "mlx.nn.Sequential"]], "silu (class in mlx.nn)": [[361, "mlx.nn.SiLU"], [411, "mlx.nn.silu"]], "sinusoidalpositionalencoding (class in mlx.nn)": [[362, "mlx.nn.SinusoidalPositionalEncoding"]], "softmax (class in mlx.nn)": [[363, "mlx.nn.Softmax"], [412, "mlx.nn.softmax"]], "softmin (class in mlx.nn)": [[364, "mlx.nn.Softmin"], [413, "mlx.nn.softmin"]], "softplus (class in mlx.nn)": [[365, "mlx.nn.Softplus"], [414, "mlx.nn.softplus"]], "softshrink (class in mlx.nn)": [[366, "mlx.nn.Softshrink"], [415, "mlx.nn.softshrink"]], "softsign (class in mlx.nn)": [[367, "mlx.nn.Softsign"]], "step (class in mlx.nn)": [[368, "mlx.nn.Step"], [416, "mlx.nn.step"]], "tanh (class in mlx.nn)": [[369, "mlx.nn.Tanh"], [417, "mlx.nn.tanh"]], "transformer (class in mlx.nn)": [[370, "mlx.nn.Transformer"]], "upsample (class in mlx.nn)": [[371, "mlx.nn.Upsample"]], "constant() (in module mlx.nn.init)": [[372, "mlx.nn.init.constant"]], "glorot_normal() (in module mlx.nn.init)": [[373, "mlx.nn.init.glorot_normal"]], "glorot_uniform() (in module mlx.nn.init)": [[374, "mlx.nn.init.glorot_uniform"]], "he_normal() (in module mlx.nn.init)": [[375, "mlx.nn.init.he_normal"]], "he_uniform() (in module mlx.nn.init)": [[376, "mlx.nn.init.he_uniform"]], "identity() (in module mlx.nn.init)": [[377, "mlx.nn.init.identity"]], "normal() (in module mlx.nn.init)": [[378, "mlx.nn.init.normal"]], "uniform() (in module mlx.nn.init)": [[379, "mlx.nn.init.uniform"]], "elu (class in mlx.nn)": [[380, "mlx.nn.elu"]], "gelu_approx (class in mlx.nn)": [[382, "mlx.nn.gelu_approx"]], "gelu_fast_approx (class in mlx.nn)": [[383, "mlx.nn.gelu_fast_approx"]], "hard_shrink (class in mlx.nn)": [[385, "mlx.nn.hard_shrink"]], "hard_tanh (class in mlx.nn)": [[386, "mlx.nn.hard_tanh"]], "leaky_relu (class in mlx.nn)": [[388, "mlx.nn.leaky_relu"]], "log_sigmoid (class in mlx.nn)": [[389, "mlx.nn.log_sigmoid"]], "log_softmax (class in mlx.nn)": [[390, "mlx.nn.log_softmax"]], "binary_cross_entropy (class in mlx.nn.losses)": [[391, "mlx.nn.losses.binary_cross_entropy"]], "cosine_similarity_loss (class in mlx.nn.losses)": [[392, "mlx.nn.losses.cosine_similarity_loss"]], "cross_entropy (class in mlx.nn.losses)": [[393, "mlx.nn.losses.cross_entropy"]], "gaussian_nll_loss (class in mlx.nn.losses)": [[394, "mlx.nn.losses.gaussian_nll_loss"]], "hinge_loss (class in mlx.nn.losses)": [[395, "mlx.nn.losses.hinge_loss"]], "huber_loss (class in mlx.nn.losses)": [[396, "mlx.nn.losses.huber_loss"]], "kl_div_loss (class in mlx.nn.losses)": [[397, "mlx.nn.losses.kl_div_loss"]], "l1_loss (class in mlx.nn.losses)": [[398, "mlx.nn.losses.l1_loss"]], "log_cosh_loss (class in mlx.nn.losses)": [[399, "mlx.nn.losses.log_cosh_loss"]], "margin_ranking_loss (class in mlx.nn.losses)": [[400, "mlx.nn.losses.margin_ranking_loss"]], "mse_loss (class in mlx.nn.losses)": [[401, "mlx.nn.losses.mse_loss"]], "nll_loss (class in mlx.nn.losses)": [[402, "mlx.nn.losses.nll_loss"]], "smooth_l1_loss (class in mlx.nn.losses)": [[403, "mlx.nn.losses.smooth_l1_loss"]], "triplet_loss (class in mlx.nn.losses)": [[404, "mlx.nn.losses.triplet_loss"]], "sigmoid (class in mlx.nn)": [[410, "mlx.nn.sigmoid"]], "module (class in mlx.nn)": [[422, "mlx.nn.Module"]], "adadelta (class in mlx.optimizers)": [[425, "mlx.optimizers.AdaDelta"]], "adafactor (class in mlx.optimizers)": [[426, "mlx.optimizers.Adafactor"]], "adagrad (class in mlx.optimizers)": [[427, "mlx.optimizers.Adagrad"]], "adam (class in mlx.optimizers)": [[428, "mlx.optimizers.Adam"]], "adamw (class in mlx.optimizers)": [[429, "mlx.optimizers.AdamW"]], "adamax (class in mlx.optimizers)": [[430, "mlx.optimizers.Adamax"]], "lion (class in mlx.optimizers)": [[431, "mlx.optimizers.Lion"]], "apply_gradients() (optimizer method)": [[432, "mlx.optimizers.Optimizer.apply_gradients"]], "init() (optimizer method)": [[433, "mlx.optimizers.Optimizer.init"]], "state (optimizer property)": [[434, "mlx.optimizers.Optimizer.state"]], "update() (optimizer method)": [[435, "mlx.optimizers.Optimizer.update"]], "rmsprop (class in mlx.optimizers)": [[436, "mlx.optimizers.RMSprop"]], "sgd (class in mlx.optimizers)": [[437, "mlx.optimizers.SGD"]], "cosine_decay() (in module mlx.optimizers)": [[438, "mlx.optimizers.cosine_decay"]], "exponential_decay() (in module mlx.optimizers)": [[439, "mlx.optimizers.exponential_decay"]], "join_schedules() (in module mlx.optimizers)": [[440, "mlx.optimizers.join_schedules"]], "linear_schedule() (in module mlx.optimizers)": [[441, "mlx.optimizers.linear_schedule"]], "step_decay() (in module mlx.optimizers)": [[442, "mlx.optimizers.step_decay"]], "optimizer (class in mlx.optimizers)": [[444, "mlx.optimizers.Optimizer"]]}}) \ No newline at end of file diff --git a/docs/build/html/steel__gemm__masked_8h.html b/docs/build/html/steel__gemm__masked_8h.html index 3d6499567f..357b4e7d4e 100644 --- a/docs/build/html/steel__gemm__masked_8h.html +++ b/docs/build/html/steel__gemm__masked_8h.html @@ -115,7 +115,7 @@ Functions
            • METAL_FUNC void qmm_t_impl
            - +
            typedef struct _NoMask nomask_ttypedef struct _NoMask nomask_t
            diff --git a/docs/build/html/steel__gemm__masked_8h_source.html b/docs/build/html/steel__gemm__masked_8h_source.html index 340810d5aa..76241471bc 100644 --- a/docs/build/html/steel__gemm__masked_8h_source.html +++ b/docs/build/html/steel__gemm__masked_8h_source.html @@ -91,9 +91,8 @@ $(function() { codefold.init(0); });
            6
            8// GEMM kernels
            10
            -
            -
            11struct _NoMask {
            -
            12 char x;
            +
            11struct _NoMask {
            +
            12 char x;
            13
            14 constexpr METAL_FUNC operator bool() {
            @@ -116,12 +115,10 @@ $(function() { codefold.init(0); });
            25 }
            26};
            -
            27
            28template <typename OutT, typename InT = OutT>
            -
            -
            29struct ScaleOp {
            -
            30 OutT scale;
            +
            29struct ScaleOp {
            +
            30 OutT scale;
            31
            32 METAL_FUNC OutT apply(InT x) const {
            @@ -129,7 +126,6 @@ $(function() { codefold.init(0); });
            34 }
            35};
            -
            36
            37typedef struct _NoMask nomask_t;
            38
            @@ -821,10 +817,10 @@ $(function() { codefold.init(0); });
            STEEL_PRAGMA_UNROLL
            #define STEEL_PRAGMA_UNROLL
            Definition defines.h:4
            block_masked_gemm
            void block_masked_gemm(const device T *A, const device T *B, device T *D, const constant GEMMParams *params, const constant int *batch_shape, const constant size_t *batch_strides, const device out_mask_t *out_mask, const device op_mask_t *lhs_mask, const device op_mask_t *rhs_mask, const constant int *mask_strides, uint simd_lane_id, uint simd_group_id, uint3 tid, uint3 lid)
            Definition steel_gemm_masked.h:53
            -
            _NoMask
            Definition steel_gemm_masked.h:11
            -
            _NoMask::x
            char x
            Definition steel_gemm_masked.h:12
            -
            ScaleOp
            Definition steel_gemm_masked.h:29
            -
            ScaleOp::scale
            OutT scale
            Definition steel_gemm_masked.h:30
            +
            _NoMask
            Definition gemv_masked.h:10
            +
            _NoMask::x
            char x
            Definition gemv_masked.h:11
            +
            ScaleOp
            Definition gemv_masked.h:30
            +
            ScaleOp::scale
            OutT scale
            Definition gemv_masked.h:31
            ScaleOp::apply
            METAL_FUNC OutT apply(InT x) const
            Definition steel_gemm_masked.h:32
            mlx::steel::GEMMKernel
            Definition gemm.h:37
            mlx::steel::GEMMParams
            Definition params.h:12
            diff --git a/docs/build/html/struct___no_mask-members.html b/docs/build/html/struct___no_mask-members.html index 38748eac71..10abc05ded 100644 --- a/docs/build/html/struct___no_mask-members.html +++ b/docs/build/html/struct___no_mask-members.html @@ -77,6 +77,10 @@ $(function() {

            This is the complete list of members for _NoMask, including all inherited members.

            + + + + diff --git a/docs/build/html/struct___no_mask.html b/docs/build/html/struct___no_mask.html index a77310a386..5b5114dc7e 100644 --- a/docs/build/html/struct___no_mask.html +++ b/docs/build/html/struct___no_mask.html @@ -79,7 +79,7 @@ $(function() {
            -

            #include <steel_gemm_masked.h>

            +

            #include <gemv_masked.h>

            operator bool()_NoMaskinline
            operator bool() const threadgroup_NoMaskinline
            operator bool() const device_NoMaskinline
            operator bool() const constant_NoMaskinline
            operator bool()_NoMaskinline
            operator bool() const threadgroup_NoMaskinline
            operator bool() const device_NoMaskinline
            @@ -91,6 +91,14 @@ Public Member Functions + + + + + + + +

            Public Member Functions

             
            constexpr METAL_FUNC operator bool () const constant
             
            constexpr METAL_FUNC operator bool ()
             
            constexpr METAL_FUNC operator bool () const threadgroup
             
            constexpr METAL_FUNC operator bool () const device
             
            constexpr METAL_FUNC operator bool () const constant
             
            @@ -99,7 +107,32 @@ Public Attributes

            Public Attributes

            Member Function Documentation

            -

            ◆ operator bool() [1/4]

            +

            ◆ operator bool() [1/8]

            + +
            +
            + + + + + +
            + + + + + + + +
            constexpr METAL_FUNC _NoMask::operator bool ()
            +             		+inlineconstexpr
            +
            + +
            +
            + +

            ◆ operator bool() [2/8]

            @@ -124,7 +157,32 @@ Public Attributes
            -

            ◆ operator bool() [2/4]

            +

            ◆ operator bool() [3/8]

            + +
            +
            + + + + + +
            + + + + + + + +
            constexpr METAL_FUNC _NoMask::operator bool () const
            +             		+inlineconstexpr
            +
            + +
            +
            + +

            ◆ operator bool() [4/8]

            @@ -149,7 +207,32 @@ Public Attributes
            -

            ◆ operator bool() [3/4]

            +

            ◆ operator bool() [5/8]

            + +
            +
            + + + + + +
            + + + + + + + +
            constexpr METAL_FUNC _NoMask::operator bool () const
            +             		+inlineconstexpr
            +
            + +
            +
            + +

            ◆ operator bool() [6/8]

            @@ -174,7 +257,32 @@ Public Attributes
            -

            ◆ operator bool() [4/4]

            +

            ◆ operator bool() [7/8]

            + +
            +
            + + + + + +
            + + + + + + + +
            constexpr METAL_FUNC _NoMask::operator bool () const
            +             		+inlineconstexpr
            +
            + +
            +
            + +

            ◆ operator bool() [8/8]

            @@ -213,7 +321,8 @@ Public Attributes
            -
            The documentation for this struct was generated from the following file:
              +
              The documentation for this struct was generated from the following files:
            diff --git a/docs/build/html/struct_g_e_m_v_kernel-members.html b/docs/build/html/struct_g_e_m_v_kernel-members.html new file mode 100644 index 0000000000..c860c0a6d1 --- /dev/null +++ b/docs/build/html/struct_g_e_m_v_kernel-members.html @@ -0,0 +1,99 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + +
            +
            + + + + + + +
            +
            MLX +
            +
            +
            + + + + + + + + +
            +
            + + +
            +
            +
            +
            +
            +
            Loading...
            +
            Searching...
            +
            No Matches
            +
            +
            +
            +
            + +
            +
            +
            GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN > Member List
            +
            +
            + +

            This is the complete list of members for GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >, including all inherited members.

            + + + + + + + + + + + + + + +
            blockMGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            blockNGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            has_mul_operand_maskGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            has_mul_output_maskGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            has_operand_maskGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            has_output_maskGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            load_safe(const device T *src, thread T dst[TN], const int src_offset=0, const int src_size=TN)GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >inlinestatic
            load_unsafe(const device T *src, thread T dst[TN], const int src_offset=0)GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >inlinestatic
            needs_tgp_reductionGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            run(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &matrix_ld, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, threadgroup T *tgp_memory, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >inlinestatic
            tgp_mem_sizeGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            threadsMGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            threadsNGEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            + +
            +Generated by doxygen 1.10.0 +
            + + diff --git a/docs/build/html/struct_g_e_m_v_kernel.html b/docs/build/html/struct_g_e_m_v_kernel.html new file mode 100644 index 0000000000..7cb06ffcfe --- /dev/null +++ b/docs/build/html/struct_g_e_m_v_kernel.html @@ -0,0 +1,546 @@ + + + + + + + +MLX: GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN > Struct Template Reference + + + + + + + + + + + +
            +
            + + + + + + +
            +
            MLX +
            +
            +
            + + + + + + + + +
            +
            + + +
            +
            +
            +
            +
            +
            Loading...
            +
            Searching...
            +
            No Matches
            +
            +
            +
            +
            + +
            +
            +
            +Static Public Member Functions | +Static Public Attributes | +List of all members
            +
            GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN > Struct Template Reference
            +
            +
            + +

            #include <gemv_masked.h>

            + + + + + + + + +

            +Static Public Member Functions

            static METAL_FUNC void load_unsafe (const device T *src, thread T dst[TN], const int src_offset=0)
             
            static METAL_FUNC void load_safe (const device T *src, thread T dst[TN], const int src_offset=0, const int src_size=TN)
             
            static METAL_FUNC void run (const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &matrix_ld, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, threadgroup T *tgp_memory, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)
             
            + + + + + + + + + + + + + + + + + + + + + +

            +Static Public Attributes

            static constant constexpr const int threadsM = BM * SM
             
            static constant constexpr const int threadsN = BN * SN
             
            static constant constexpr const int blockM = threadsM * TM
             
            static constant constexpr const int blockN = threadsN * TN
             
            static constant constexpr const bool has_operand_mask = !metal::is_same_v<op_mask_t, nomask_t>
             
            static constant constexpr const bool has_output_mask = !metal::is_same_v<out_mask_t, nomask_t>
             
            static constant constexpr const bool has_mul_operand_mask
             
            static constant constexpr const bool has_mul_output_mask
             
            static constant constexpr const short tgp_mem_size = BN > 1 ? BN*(blockM + TM) : 0
             
            static constant constexpr const bool needs_tgp_reduction = BN > 1
             
            +

            Member Function Documentation

            + +

            ◆ load_safe()

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + + + + + + + + + + + + + + + + + + +
            static METAL_FUNC void GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::load_safe (const device T * src,
            thread T dst[TN],
            const int src_offset = 0,
            const int src_size = TN )
            +             		+inlinestatic
            +
            + +
            +
            + +

            ◆ load_unsafe()

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + + + + + + + + + + + + + +
            static METAL_FUNC void GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::load_unsafe (const device T * src,
            thread T dst[TN],
            const int src_offset = 0 )
            +             		+inlinestatic
            +
            + +
            +
            + +

            ◆ run()

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
            static METAL_FUNC void GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::run (const device T * mat,
            const device T * in_vec,
            device T * out_vec,
            const constant int & in_vec_size,
            const constant int & out_vec_size,
            const constant int & matrix_ld,
            const device out_mask_t * out_mask,
            const device op_mask_t * mat_mask,
            const device op_mask_t * vec_mask,
            const constant int * mask_strides,
            threadgroup T * tgp_memory,
            uint3 tid,
            uint3 lid,
            uint simd_gid,
            uint simd_lid )
            +             		+inlinestatic
            +
            + +
            +
            +

            Member Data Documentation

            + +

            ◆ blockM

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const int GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::blockM = threadsM * TM
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ blockN

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const int GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::blockN = threadsN * TN
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ has_mul_operand_mask

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::has_mul_operand_mask
            +             		+staticconstexpr
            +
            +Initial value:
            =
            +
            has_operand_mask && !metal::is_same_v<op_mask_t, bool>
            +
            GEMVKernel::has_operand_mask
            static constant constexpr const bool has_operand_mask
            Definition gemv_masked.h:63
            +
            +
            +
            + +

            ◆ has_mul_output_mask

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::has_mul_output_mask
            +             		+staticconstexpr
            +
            +Initial value:
            =
            +
            has_output_mask && !metal::is_same_v<out_mask_t, bool>
            +
            GEMVKernel::has_output_mask
            static constant constexpr const bool has_output_mask
            Definition gemv_masked.h:64
            +
            +
            +
            + +

            ◆ has_operand_mask

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::has_operand_mask = !metal::is_same_v<op_mask_t, nomask_t>
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ has_output_mask

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::has_output_mask = !metal::is_same_v<out_mask_t, nomask_t>
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ needs_tgp_reduction

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::needs_tgp_reduction = BN > 1
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ tgp_mem_size

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const short GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::tgp_mem_size = BN > 1 ? BN*(blockM + TM) : 0
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ threadsM

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const int GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::threadsM = BM * SM
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ threadsN

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const int GEMVKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::threadsN = BN * SN
            +             		+staticconstexpr
            +
            + +
            +
            +
            The documentation for this struct was generated from the following file: +
            + +
            +Generated by doxygen 1.10.0 +
            + + diff --git a/docs/build/html/struct_g_e_m_v_t_kernel-members.html b/docs/build/html/struct_g_e_m_v_t_kernel-members.html new file mode 100644 index 0000000000..2a1e4e9112 --- /dev/null +++ b/docs/build/html/struct_g_e_m_v_t_kernel-members.html @@ -0,0 +1,97 @@ + + + + + + + +MLX: Member List + + + + + + + + + + + +
            +
            + + + + + + +
            +
            MLX +
            +
            +
            + + + + + + + + +
            +
            + + +
            +
            +
            +
            +
            +
            Loading...
            +
            Searching...
            +
            No Matches
            +
            +
            +
            +
            + +
            +
            +
            GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN > Member List
            +
            +
            + +

            This is the complete list of members for GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >, including all inherited members.

            + + + + + + + + + + + + +
            blockMGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            blockNGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            has_mul_operand_maskGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            has_mul_output_maskGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            has_operand_maskGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            has_output_maskGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            needs_tgp_reductionGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            run(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, threadgroup T *tgp_memory, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >inlinestatic
            tgp_mem_sizeGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            threadsMGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            threadsNGEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >static
            + +
            +Generated by doxygen 1.10.0 +
            + + diff --git a/docs/build/html/struct_g_e_m_v_t_kernel.html b/docs/build/html/struct_g_e_m_v_t_kernel.html new file mode 100644 index 0000000000..03bbbcf2b3 --- /dev/null +++ b/docs/build/html/struct_g_e_m_v_t_kernel.html @@ -0,0 +1,471 @@ + + + + + + + +MLX: GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN > Struct Template Reference + + + + + + + + + + + +
            +
            + + + + + + +
            +
            MLX +
            +
            +
            + + + + + + + + +
            +
            + + +
            +
            +
            +
            +
            +
            Loading...
            +
            Searching...
            +
            No Matches
            +
            +
            +
            +
            + +
            +
            +
            +Static Public Member Functions | +Static Public Attributes | +List of all members
            +
            GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN > Struct Template Reference
            +
            +
            + +

            Vector matrix multiplication. + More...

            + +

            #include <gemv_masked.h>

            + + + + +

            +Static Public Member Functions

            static METAL_FUNC void run (const device T *mat, const device T *in_vec, device T *out_vec, const constant int &in_vec_size, const constant int &out_vec_size, const constant int &marix_ld, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, threadgroup T *tgp_memory, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)
             
            + + + + + + + + + + + + + + + + + + + + + +

            +Static Public Attributes

            static constant constexpr const int threadsM = BM * SM
             
            static constant constexpr const int threadsN = BN * SN
             
            static constant constexpr const int blockM = threadsM * TM
             
            static constant constexpr const int blockN = threadsN * TN
             
            static constant constexpr const bool has_operand_mask = !metal::is_same_v<op_mask_t, nomask_t>
             
            static constant constexpr const bool has_output_mask = !metal::is_same_v<out_mask_t, nomask_t>
             
            static constant constexpr const bool has_mul_operand_mask
             
            static constant constexpr const bool has_mul_output_mask
             
            static constant constexpr const short tgp_mem_size = BM > 1 ? BM*(blockN + TN) : 0
             
            static constant constexpr const bool needs_tgp_reduction = BM > 1
             
            +

            Detailed Description

            +
            template<typename T, typename out_mask_t, typename op_mask_t, const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            +struct GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >

            Vector matrix multiplication.

            +

            Member Function Documentation

            + +

            ◆ run()

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
            static METAL_FUNC void GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::run (const device T * mat,
            const device T * in_vec,
            device T * out_vec,
            const constant int & in_vec_size,
            const constant int & out_vec_size,
            const constant int & marix_ld,
            const device out_mask_t * out_mask,
            const device op_mask_t * mat_mask,
            const device op_mask_t * vec_mask,
            const constant int * mask_strides,
            threadgroup T * tgp_memory,
            uint3 tid,
            uint3 lid,
            uint simd_gid,
            uint simd_lid )
            +             		+inlinestatic
            +
            + +
            +
            +

            Member Data Documentation

            + +

            ◆ blockM

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const int GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::blockM = threadsM * TM
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ blockN

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const int GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::blockN = threadsN * TN
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ has_mul_operand_mask

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::has_mul_operand_mask
            +             		+staticconstexpr
            +
            +Initial value:
            =
            +
            has_operand_mask && !metal::is_same_v<op_mask_t, bool>
            +
            GEMVTKernel::has_operand_mask
            static constant constexpr const bool has_operand_mask
            Definition gemv_masked.h:367
            +
            +
            +
            + +

            ◆ has_mul_output_mask

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::has_mul_output_mask
            +             		+staticconstexpr
            +
            +Initial value:
            =
            +
            has_output_mask && !metal::is_same_v<out_mask_t, bool>
            +
            GEMVTKernel::has_output_mask
            static constant constexpr const bool has_output_mask
            Definition gemv_masked.h:368
            +
            +
            +
            + +

            ◆ has_operand_mask

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::has_operand_mask = !metal::is_same_v<op_mask_t, nomask_t>
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ has_output_mask

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::has_output_mask = !metal::is_same_v<out_mask_t, nomask_t>
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ needs_tgp_reduction

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const bool GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::needs_tgp_reduction = BM > 1
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ tgp_mem_size

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const short GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::tgp_mem_size = BM > 1 ? BM*(blockN + TN) : 0
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ threadsM

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const int GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::threadsM = BM * SM
            +             		+staticconstexpr
            +
            + +
            +
            + +

            ◆ threadsN

            + +
            +
            +
            +template<typename T , typename out_mask_t , typename op_mask_t , const int BM, const int BN, const int SM, const int SN, const int TM, const int TN>
            + + + + + +
            + + + + +
            constant constexpr const int GEMVTKernel< T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN >::threadsN = BN * SN
            +             		+staticconstexpr
            +
            + +
            +
            +
            The documentation for this struct was generated from the following file: +
            + +
            +Generated by doxygen 1.10.0 +
            + + diff --git a/docs/build/html/struct_scale_op-members.html b/docs/build/html/struct_scale_op-members.html index 11da1994ab..b1602d9c5b 100644 --- a/docs/build/html/struct_scale_op-members.html +++ b/docs/build/html/struct_scale_op-members.html @@ -78,7 +78,8 @@ $(function() {

            This is the complete list of members for ScaleOp< OutT, InT >, including all inherited members.

            - + +
            apply(InT x) constScaleOp< OutT, InT >inline
            scaleScaleOp< OutT, InT >
            apply(InT x) constScaleOp< OutT, InT >inline
            scaleScaleOp< OutT, InT >
            diff --git a/docs/build/html/struct_scale_op.html b/docs/build/html/struct_scale_op.html index 236b06c54a..3fe2713233 100644 --- a/docs/build/html/struct_scale_op.html +++ b/docs/build/html/struct_scale_op.html @@ -79,12 +79,14 @@ $(function() {
            -

            #include <steel_gemm_masked.h>

            +

            #include <gemv_masked.h>

            + +

            Public Member Functions

            METAL_FUNC OutT apply (InT x) const
             
            METAL_FUNC OutT apply (InT x) const
             
            @@ -93,7 +95,34 @@ Public Attributes

            Public Attributes

            Member Function Documentation

            -

            ◆ apply()

            +

            ◆ apply() [1/2]

            + +
            +
            +
            +template<typename OutT , typename InT = OutT>
            + + + + + +
            + + + + + + + +
            METAL_FUNC OutT ScaleOp< OutT, InT >::apply (InT x) const
            +             		+inline
            +
            + +
            +
            + +

            ◆ apply() [2/2]

            @@ -136,7 +165,8 @@ template<typename OutT , typename InT = OutT>
            -
            The documentation for this struct was generated from the following file:
              +
              The documentation for this struct was generated from the following files:
            diff --git a/docs/build/html/structmlx_1_1core_1_1metal_1_1_command_encoder-members.html b/docs/build/html/structmlx_1_1core_1_1metal_1_1_command_encoder-members.html index 1623eccace..72572a3190 100644 --- a/docs/build/html/structmlx_1_1core_1_1metal_1_1_command_encoder-members.html +++ b/docs/build/html/structmlx_1_1core_1_1metal_1_1_command_encoder-members.html @@ -81,16 +81,16 @@ $(function() {

            This is the complete list of members for mlx::core::metal::CommandEncoder, including all inherited members.

            - + - - + + - +
            CommandEncoder(MTL::CommandBuffer *cbuf)mlx::core::metal::CommandEncoderinline
            CommandEncoder(MTL::CommandBuffer *cbuf)mlx::core::metal::CommandEncoder
            CommandEncoder(const CommandEncoder &)=deletemlx::core::metal::CommandEncoder
            dispatchThreadgroups(MTL::Size grid_dims, MTL::Size group_dims)mlx::core::metal::CommandEncoder
            dispatchThreads(MTL::Size grid_dims, MTL::Size group_dims)mlx::core::metal::CommandEncoder
            operator->()mlx::core::metal::CommandEncoderinline
            operator=(const CommandEncoder &)=deletemlx::core::metal::CommandEncoder
            set_input_array(const array &a, int idx, int64_t offset=0)mlx::core::metal::CommandEncoderinline
            set_output_array(array &a, int idx, int64_t offset=0)mlx::core::metal::CommandEncoderinline
            set_input_array(const array &a, int idx, int64_t offset=0)mlx::core::metal::CommandEncoder
            set_output_array(array &a, int idx, int64_t offset=0)mlx::core::metal::CommandEncoder
            start_concurrent()mlx::core::metal::CommandEncoderinline
            ~CommandEncoder()mlx::core::metal::CommandEncoderinline
            ~CommandEncoder()mlx::core::metal::CommandEncoder
            diff --git a/docs/build/html/structmlx_1_1core_1_1metal_1_1_command_encoder.html b/docs/build/html/structmlx_1_1core_1_1metal_1_1_command_encoder.html index c9d51a5d44..755445fdc1 100644 --- a/docs/build/html/structmlx_1_1core_1_1metal_1_1_command_encoder.html +++ b/docs/build/html/structmlx_1_1core_1_1metal_1_1_command_encoder.html @@ -119,9 +119,6 @@ Public Member Functions
            - - - - - -
            @@ -130,11 +127,6 @@ Public Member Functions
            mlx::core::metal::CommandEncoder::CommandEncoder
            -             		-inline
            @@ -169,9 +161,6 @@ Public Member Functions
            - - - - - -
            @@ -180,11 +169,6 @@ Public Member Functions
            mlx::core::metal::CommandEncoder::~CommandEncoder
            -             		-inline
            @@ -287,9 +271,6 @@ Public Member Functions
            - - - - - -
            @@ -307,11 +288,6 @@ Public Member Functions
            void mlx::core::metal::CommandEncoder::set_input_array
            int64_t offset = 0 )
            -             		-inline
            @@ -321,9 +297,6 @@ Public Member Functions
            - - - - - -
            @@ -341,11 +314,6 @@ Public Member Functions
            void mlx::core::metal::CommandEncoder::set_output_array
            int64_t offset = 0 )
            -             		-inline
            diff --git a/docs/build/html/structmlx_1_1steel_1_1_g_e_m_m_kernel.html b/docs/build/html/structmlx_1_1steel_1_1_g_e_m_m_kernel.html index 6970453616..6b313b930a 100644 --- a/docs/build/html/structmlx_1_1steel_1_1_g_e_m_m_kernel.html +++ b/docs/build/html/structmlx_1_1steel_1_1_g_e_m_m_kernel.html @@ -84,7 +84,7 @@ $(function() {
            -

            #include <gemm.h>

            +

            #include <gemm.h>

            @@ -445,7 +445,7 @@ template<typename T , typename U , int BM, int BN, int BK, int WM, int WN, bo
            The documentation for this struct was generated from the following file:
              -
            • mlx/backend/metal/kernels/steel/gemm/gemm.h
              • +
            • mlx/backend/metal/kernels/steel/gemm/gemm.h
            diff --git a/docs/build/html/structmlx_1_1steel_1_1_loop_alignment.html b/docs/build/html/structmlx_1_1steel_1_1_loop_alignment.html index ecb9d456a4..460ea4e27d 100644 --- a/docs/build/html/structmlx_1_1steel_1_1_loop_alignment.html +++ b/docs/build/html/structmlx_1_1steel_1_1_loop_alignment.html @@ -79,9 +79,9 @@ $(function() {
            -

            #include <gemm.h>

            +

            #include <gemm.h>

            The documentation for this struct was generated from the following file:
              -
            • mlx/backend/metal/kernels/steel/gemm/gemm.h
              • +
            • mlx/backend/metal/kernels/steel/gemm/gemm.h
            diff --git a/docs/build/html/types_2complex_8h_source.html b/docs/build/html/types_2complex_8h_source.html index 605484469f..1ada8fbbbc 100644 --- a/docs/build/html/types_2complex_8h_source.html +++ b/docs/build/html/types_2complex_8h_source.html @@ -219,12 +219,12 @@ $(function() { codefold.init(0); });
            110
            111} // namespace mlx::core
            mlx::core::std
            array std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})
            Computes the standard deviation of the elements of an array.
            -
            mlx::core::operator>=
            array operator>=(const array &a, const array &b)
            Definition ops.h:343
            -
            mlx::core::operator<=
            array operator<=(const array &a, const array &b)
            Definition ops.h:371
            -
            mlx::core::operator>
            array operator>(const array &a, const array &b)
            Definition ops.h:329
            +
            mlx::core::operator>=
            array operator>=(const array &a, const array &b)
            Definition ops.h:347
            +
            mlx::core::operator<=
            array operator<=(const array &a, const array &b)
            Definition ops.h:375
            +
            mlx::core::operator>
            array operator>(const array &a, const array &b)
            Definition ops.h:333
            mlx::core::operator%
            array operator%(const array &a, const array &b)
            mlx::core::operator-
            array operator-(const array &a)
            -
            mlx::core::operator<
            array operator<(const array &a, const array &b)
            Definition ops.h:357
            +
            mlx::core::operator<
            array operator<(const array &a, const array &b)
            Definition ops.h:361
            half_types.h
            mlx::core
            Definition allocator.h:7
            mlx::core::can_convert_to_complex64
            constexpr bool can_convert_to_complex64
            Definition complex.h:31
            diff --git a/docs/build/html/usage/compile.html b/docs/build/html/usage/compile.html index 4b27a2ea64..c34ffb952d 100644 --- a/docs/build/html/usage/compile.html +++ b/docs/build/html/usage/compile.html @@ -8,7 +8,7 @@ - Compilation — MLX 0.16.1 documentation + Compilation — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/distributed.html b/docs/build/html/usage/distributed.html index 791666a087..07e7190585 100644 --- a/docs/build/html/usage/distributed.html +++ b/docs/build/html/usage/distributed.html @@ -8,7 +8,7 @@ - Distributed Communication — MLX 0.16.1 documentation + Distributed Communication — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/function_transforms.html b/docs/build/html/usage/function_transforms.html index a8ceb1a59d..94eb74215b 100644 --- a/docs/build/html/usage/function_transforms.html +++ b/docs/build/html/usage/function_transforms.html @@ -8,7 +8,7 @@ - Function Transforms — MLX 0.16.1 documentation + Function Transforms — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/indexing.html b/docs/build/html/usage/indexing.html index 4a31c2bb55..dd5ed6d1c6 100644 --- a/docs/build/html/usage/indexing.html +++ b/docs/build/html/usage/indexing.html @@ -8,7 +8,7 @@ - Indexing Arrays — MLX 0.16.1 documentation + Indexing Arrays — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/lazy_evaluation.html b/docs/build/html/usage/lazy_evaluation.html index 0fa2d98ad6..8bec68b99a 100644 --- a/docs/build/html/usage/lazy_evaluation.html +++ b/docs/build/html/usage/lazy_evaluation.html @@ -8,7 +8,7 @@ - Lazy Evaluation — MLX 0.16.1 documentation + Lazy Evaluation — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/numpy.html b/docs/build/html/usage/numpy.html index e20516d6e8..e274f12a67 100644 --- a/docs/build/html/usage/numpy.html +++ b/docs/build/html/usage/numpy.html @@ -8,7 +8,7 @@ - Conversion to NumPy and Other Frameworks — MLX 0.16.1 documentation + Conversion to NumPy and Other Frameworks — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/quick_start.html b/docs/build/html/usage/quick_start.html index c1aaad10df..7ef70aca32 100644 --- a/docs/build/html/usage/quick_start.html +++ b/docs/build/html/usage/quick_start.html @@ -8,7 +8,7 @@ - Quick Start Guide — MLX 0.16.1 documentation + Quick Start Guide — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/saving_and_loading.html b/docs/build/html/usage/saving_and_loading.html index 87a9bd8df4..7c44124f6d 100644 --- a/docs/build/html/usage/saving_and_loading.html +++ b/docs/build/html/usage/saving_and_loading.html @@ -8,7 +8,7 @@ - Saving and Loading Arrays — MLX 0.16.1 documentation + Saving and Loading Arrays — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/unified_memory.html b/docs/build/html/usage/unified_memory.html index 145dcfae5f..f7f5672d0b 100644 --- a/docs/build/html/usage/unified_memory.html +++ b/docs/build/html/usage/unified_memory.html @@ -8,7 +8,7 @@ - Unified Memory — MLX 0.16.1 documentation + Unified Memory — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/usage/using_streams.html b/docs/build/html/usage/using_streams.html index cd8f55cd39..c3a2eaf2be 100644 --- a/docs/build/html/usage/using_streams.html +++ b/docs/build/html/usage/using_streams.html @@ -8,7 +8,7 @@ - Using Streams — MLX 0.16.1 documentation + Using Streams — MLX 0.16.2 documentation @@ -36,7 +36,7 @@ - + @@ -131,8 +131,8 @@ - MLX 0.16.1 documentation - Home - + MLX 0.16.2 documentation - Home + @@ -457,8 +457,10 @@
          • Linear Algebra diff --git a/docs/build/html/utils_8h.html b/docs/build/html/utils_8h.html index 3167eb47e2..4540b3a74f 100644 --- a/docs/build/html/utils_8h.html +++ b/docs/build/html/utils_8h.html @@ -130,8 +130,6 @@ Functions
            • - - diff --git a/docs/build/html/utils_8h_source.html b/docs/build/html/utils_8h_source.html index 0a9fd6113f..64c9b877b4 100644 --- a/docs/build/html/utils_8h_source.html +++ b/docs/build/html/utils_8h_source.html @@ -181,60 +181,48 @@ $(function() { codefold.init(0); });
            85}
            86
            -
            -
            87inline bool is_big_endian() {
            -
            88 union ByteOrder {
            -
            89 int32_t i;
            -
            90 uint8_t c[4];
            -
            91 };
            -
            92 ByteOrder b = {0x01234567};
            +
            92int normalize_axis(int axis, int ndim);
            93
            -
            94 return b.c[0] == 0x01;
            -
            95}
            +
            94std::ostream& operator<<(std::ostream& os, const Device& d);
            +
            95std::ostream& operator<<(std::ostream& os, const Stream& s);
            +
            96std::ostream& operator<<(std::ostream& os, const Dtype& d);
            +
            97std::ostream& operator<<(std::ostream& os, const Dtype::Kind& k);
            +
            98std::ostream& operator<<(std::ostream& os, array a);
            +
            99std::ostream& operator<<(std::ostream& os, const std::vector<int>& v);
            +
            100std::ostream& operator<<(std::ostream& os, const std::vector<size_t>& v);
            +
            101std::ostream& operator<<(std::ostream& os, const std::vector<int64_t>& v);
            +
            +
            102inline std::ostream& operator<<(std::ostream& os, const complex64_t& v) {
            +
            103 return os << v.real() << (v.imag() >= 0 ? "+" : "") << v.imag() << "j";
            +
            104}
            -
            96
            -
            102int normalize_axis(int axis, int ndim);
            -
            103
            -
            104std::ostream& operator<<(std::ostream& os, const Device& d);
            -
            105std::ostream& operator<<(std::ostream& os, const Stream& s);
            -
            106std::ostream& operator<<(std::ostream& os, const Dtype& d);
            -
            107std::ostream& operator<<(std::ostream& os, const Dtype::Kind& k);
            -
            108std::ostream& operator<<(std::ostream& os, array a);
            -
            109std::ostream& operator<<(std::ostream& os, const std::vector<int>& v);
            -
            110std::ostream& operator<<(std::ostream& os, const std::vector<size_t>& v);
            -
            111std::ostream& operator<<(std::ostream& os, const std::vector<int64_t>& v);
            +
            +
            105inline std::ostream& operator<<(std::ostream& os, const float16_t& v) {
            +
            106 return os << static_cast<float>(v);
            +
            107}
            +
            +
            +
            108inline std::ostream& operator<<(std::ostream& os, const bfloat16_t& v) {
            +
            109 return os << static_cast<float>(v);
            +
            110}
            +
            +
            111
            -
            112inline std::ostream& operator<<(std::ostream& os, const complex64_t& v) {
            -
            113 return os << v.real() << (v.imag() >= 0 ? "+" : "") << v.imag() << "j";
            +
            112inline bool is_power_of_2(int n) {
            +
            113 return ((n & (n - 1)) == 0) && n != 0;
            114}
            -
            -
            115inline std::ostream& operator<<(std::ostream& os, const float16_t& v) {
            -
            116 return os << static_cast<float>(v);
            -
            117}
            +
            115
            +
            +
            116inline int next_power_of_2(int n) {
            +
            117 if (is_power_of_2(n)) {
            +
            118 return n;
            +
            119 }
            +
            120 return pow(2, std::ceil(std::log2(n)));
            +
            121}
            -
            -
            118inline std::ostream& operator<<(std::ostream& os, const bfloat16_t& v) {
            -
            119 return os << static_cast<float>(v);
            -
            120}
            -
            -
            121
            -
            -
            122inline bool is_power_of_2(int n) {
            -
            123 return ((n & (n - 1)) == 0) && n != 0;
            -
            124}
            -
            -
            125
            -
            -
            126inline int next_power_of_2(int n) {
            -
            127 if (is_power_of_2(n)) {
            -
            128 return n;
            -
            129 }
            -
            130 return pow(2, std::ceil(std::log2(n)));
            -
            131}
            -
            -
            132
            -
            133} // namespace mlx::core
            +
            122
            +
            123} // namespace mlx::core
            mlx::core::array
            Definition array.h:20
            mlx::core::array::dtype
            Dtype dtype() const
            Get the arrays data type.
            Definition array.h:127
            @@ -250,22 +238,21 @@ $(function() { codefold.init(0); });
            mlx::core::set_default_device
            void set_default_device(const Device &d)
            mlx::core::to_stream
            Stream to_stream(StreamOrDevice s)
            mlx::core::promote_types
            Dtype promote_types(const Dtype &t1, const Dtype &t2)
            -
            mlx::core::is_big_endian
            bool is_big_endian()
            Definition utils.h:87
            -
            mlx::core::next_power_of_2
            int next_power_of_2(int n)
            Definition utils.h:126
            +
            mlx::core::next_power_of_2
            int next_power_of_2(int n)
            Definition utils.h:116
            mlx::core::check_shape_dim
            int check_shape_dim(const T dim)
            Returns the shape dimension if it's within allowed range.
            Definition utils.h:73
            mlx::core::result_type
            Dtype result_type(const array &a, const array &b)
            The type from promoting the arrays' types with one another.
            Definition utils.h:57
            mlx::core::StreamOrDevice
            std::variant< std::monostate, Stream, Device > StreamOrDevice
            Definition utils.h:14
            mlx::core::default_stream
            Stream default_stream(Device d)
            Get the default stream for the given device.
            mlx::core::is_same_shape
            bool is_same_shape(const std::vector< array > &arrays)
            -
            mlx::core::is_power_of_2
            bool is_power_of_2(int n)
            Definition utils.h:122
            +
            mlx::core::is_power_of_2
            bool is_power_of_2(int n)
            Definition utils.h:112
            mlx::core::set_default_stream
            void set_default_stream(Stream s)
            Make the stream the default for its device.
            mlx::core::global_formatter
            PrintFormatter global_formatter
            mlx::core::_MLX_BFloat16
            Definition bf16.h:21
            mlx::core::_MLX_Float16
            Definition fp16.h:21
            mlx::core::Device
            Definition device.h:7
            -
            mlx::core::Dtype
            Definition dtype.h:15
            -
            mlx::core::Dtype::Kind
            Kind
            Definition dtype.h:32
            +
            mlx::core::Dtype
            Definition dtype.h:13
            +
            mlx::core::Dtype::Kind
            Kind
            Definition dtype.h:30
            mlx::core::PrintFormatter
            Definition utils.h:38
            mlx::core::PrintFormatter::print
            void print(std::ostream &os, uint32_t val)
            mlx::core::PrintFormatter::print
            void print(std::ostream &os, float val)

            Public Types

            int mlx::core::check_shape_dim (const T dim)
             Returns the shape dimension if it's within allowed range.
             
            bool mlx::core::is_big_endian ()
             
            int mlx::core::normalize_axis (int axis, int ndim)
             Returns the axis normalized to be in the range [0, ndim).