From d890405142ef3f25dac3c9369f0b11d5fa4d2695 Mon Sep 17 00:00:00 2001 From: CircleCI Docs Date: Thu, 3 Apr 2025 20:25:24 +0000 Subject: [PATCH] rebase --- docs/build/html/.buildinfo | 2 +- docs/build/html/_sources/dev/extensions.rst | 14 +- .../_autosummary/mlx.core.async_eval.rst | 6 + .../mlx.core.broadcast_arrays.rst | 6 + .../_autosummary/mlx.core.contiguous.rst | 6 + .../_autosummary/mlx.core.linalg.pinv.rst | 6 + docs/build/html/_sources/python/linalg.rst | 1 + docs/build/html/_sources/python/ops.rst | 2 + .../build/html/_sources/python/transforms.rst | 1 + .../html/_static/documentation_options.js | 2 +- docs/build/html/cpp/ops.html | 12 +- docs/build/html/dev/custom_metal_kernels.html | 12 +- docs/build/html/dev/extensions.html | 26 +- docs/build/html/dev/metal_debugger.html | 12 +- docs/build/html/dev/mlx_in_cpp.html | 12 +- .../html/examples/linear_regression.html | 12 +- docs/build/html/examples/llama-inference.html | 12 +- docs/build/html/examples/mlp.html | 12 +- docs/build/html/genindex.html | 24 +- docs/build/html/index.html | 12 +- docs/build/html/install.html | 12 +- docs/build/html/objects.inv | Bin 27684 -> 27779 bytes .../python/_autosummary/mlx.core.Device.html | 12 +- .../python/_autosummary/mlx.core.Dtype.html | 12 +- .../_autosummary/mlx.core.DtypeCategory.html | 12 +- .../python/_autosummary/mlx.core.abs.html | 12 +- .../python/_autosummary/mlx.core.add.html | 12 +- .../python/_autosummary/mlx.core.addmm.html | 12 +- .../python/_autosummary/mlx.core.all.html | 12 +- .../_autosummary/mlx.core.allclose.html | 12 +- .../python/_autosummary/mlx.core.any.html | 12 +- .../python/_autosummary/mlx.core.arange.html | 12 +- .../python/_autosummary/mlx.core.arccos.html | 12 +- .../python/_autosummary/mlx.core.arccosh.html | 12 +- .../python/_autosummary/mlx.core.arcsin.html | 12 +- .../python/_autosummary/mlx.core.arcsinh.html | 12 +- .../python/_autosummary/mlx.core.arctan.html | 12 +- .../python/_autosummary/mlx.core.arctan2.html | 12 +- .../python/_autosummary/mlx.core.arctanh.html | 12 +- .../python/_autosummary/mlx.core.argmax.html | 12 +- .../python/_autosummary/mlx.core.argmin.html | 12 +- .../_autosummary/mlx.core.argpartition.html | 12 +- .../python/_autosummary/mlx.core.argsort.html | 12 +- .../python/_autosummary/mlx.core.array.T.html | 12 +- .../_autosummary/mlx.core.array.abs.html | 12 +- .../_autosummary/mlx.core.array.all.html | 12 +- .../_autosummary/mlx.core.array.any.html | 12 +- .../_autosummary/mlx.core.array.argmax.html | 12 +- .../_autosummary/mlx.core.array.argmin.html | 12 +- .../_autosummary/mlx.core.array.astype.html | 12 +- .../_autosummary/mlx.core.array.at.html | 12 +- .../_autosummary/mlx.core.array.conj.html | 12 +- .../_autosummary/mlx.core.array.cos.html | 12 +- .../_autosummary/mlx.core.array.cummax.html | 12 +- .../_autosummary/mlx.core.array.cummin.html | 12 +- .../_autosummary/mlx.core.array.cumprod.html | 12 +- .../_autosummary/mlx.core.array.cumsum.html | 12 +- .../_autosummary/mlx.core.array.diag.html | 12 +- .../_autosummary/mlx.core.array.diagonal.html | 12 +- .../_autosummary/mlx.core.array.dtype.html | 12 +- .../_autosummary/mlx.core.array.exp.html | 12 +- .../_autosummary/mlx.core.array.flatten.html | 12 +- .../python/_autosummary/mlx.core.array.html | 12 +- .../_autosummary/mlx.core.array.item.html | 12 +- .../_autosummary/mlx.core.array.itemsize.html | 12 +- .../_autosummary/mlx.core.array.log.html | 12 +- .../_autosummary/mlx.core.array.log10.html | 12 +- .../_autosummary/mlx.core.array.log1p.html | 12 +- .../_autosummary/mlx.core.array.log2.html | 12 +- .../mlx.core.array.logsumexp.html | 12 +- .../_autosummary/mlx.core.array.max.html | 12 +- .../_autosummary/mlx.core.array.mean.html | 12 +- .../_autosummary/mlx.core.array.min.html | 12 +- .../_autosummary/mlx.core.array.moveaxis.html | 12 +- .../_autosummary/mlx.core.array.nbytes.html | 12 +- .../_autosummary/mlx.core.array.ndim.html | 12 +- .../_autosummary/mlx.core.array.prod.html | 12 +- .../mlx.core.array.reciprocal.html | 12 +- .../_autosummary/mlx.core.array.reshape.html | 12 +- .../_autosummary/mlx.core.array.round.html | 12 +- .../_autosummary/mlx.core.array.rsqrt.html | 12 +- .../_autosummary/mlx.core.array.shape.html | 12 +- .../_autosummary/mlx.core.array.sin.html | 12 +- .../_autosummary/mlx.core.array.size.html | 12 +- .../_autosummary/mlx.core.array.split.html | 12 +- .../_autosummary/mlx.core.array.sqrt.html | 12 +- .../_autosummary/mlx.core.array.square.html | 12 +- .../_autosummary/mlx.core.array.squeeze.html | 12 +- .../_autosummary/mlx.core.array.std.html | 12 +- .../_autosummary/mlx.core.array.sum.html | 12 +- .../_autosummary/mlx.core.array.swapaxes.html | 12 +- .../_autosummary/mlx.core.array.tolist.html | 12 +- .../mlx.core.array.transpose.html | 12 +- .../_autosummary/mlx.core.array.var.html | 12 +- .../_autosummary/mlx.core.array.view.html | 12 +- .../_autosummary/mlx.core.array_equal.html | 12 +- .../_autosummary/mlx.core.as_strided.html | 12 +- .../_autosummary/mlx.core.async_eval.html | 1046 +++++++++++++++++ .../_autosummary/mlx.core.atleast_1d.html | 12 +- .../_autosummary/mlx.core.atleast_2d.html | 12 +- .../_autosummary/mlx.core.atleast_3d.html | 12 +- .../_autosummary/mlx.core.bitwise_and.html | 12 +- .../_autosummary/mlx.core.bitwise_invert.html | 12 +- .../_autosummary/mlx.core.bitwise_or.html | 12 +- .../_autosummary/mlx.core.bitwise_xor.html | 12 +- .../mlx.core.block_masked_mm.html | 18 +- .../mlx.core.broadcast_arrays.html | 1033 ++++++++++++++++ .../_autosummary/mlx.core.broadcast_to.html | 18 +- .../python/_autosummary/mlx.core.ceil.html | 12 +- .../_autosummary/mlx.core.clear_cache.html | 12 +- .../python/_autosummary/mlx.core.clip.html | 12 +- .../python/_autosummary/mlx.core.compile.html | 18 +- .../_autosummary/mlx.core.concatenate.html | 18 +- .../python/_autosummary/mlx.core.conj.html | 18 +- .../_autosummary/mlx.core.conjugate.html | 12 +- .../_autosummary/mlx.core.contiguous.html | 1035 ++++++++++++++++ .../python/_autosummary/mlx.core.conv1d.html | 12 +- .../python/_autosummary/mlx.core.conv2d.html | 12 +- .../python/_autosummary/mlx.core.conv3d.html | 12 +- .../_autosummary/mlx.core.conv_general.html | 12 +- .../mlx.core.conv_transpose1d.html | 12 +- .../mlx.core.conv_transpose2d.html | 12 +- .../mlx.core.conv_transpose3d.html | 12 +- .../_autosummary/mlx.core.convolve.html | 12 +- .../python/_autosummary/mlx.core.cos.html | 12 +- .../python/_autosummary/mlx.core.cosh.html | 12 +- .../python/_autosummary/mlx.core.cummax.html | 12 +- .../python/_autosummary/mlx.core.cummin.html | 12 +- .../python/_autosummary/mlx.core.cumprod.html | 12 +- .../python/_autosummary/mlx.core.cumsum.html | 12 +- .../mlx.core.custom_function.html | 12 +- .../_autosummary/mlx.core.default_device.html | 12 +- .../_autosummary/mlx.core.default_stream.html | 12 +- .../python/_autosummary/mlx.core.degrees.html | 12 +- .../_autosummary/mlx.core.dequantize.html | 12 +- .../python/_autosummary/mlx.core.diag.html | 12 +- .../_autosummary/mlx.core.diagonal.html | 12 +- .../mlx.core.disable_compile.html | 12 +- .../mlx.core.distributed.Group.html | 12 +- .../mlx.core.distributed.all_gather.html | 12 +- .../mlx.core.distributed.all_sum.html | 12 +- .../mlx.core.distributed.init.html | 12 +- .../mlx.core.distributed.is_available.html | 12 +- .../mlx.core.distributed.recv.html | 12 +- .../mlx.core.distributed.recv_like.html | 12 +- .../mlx.core.distributed.send.html | 12 +- .../python/_autosummary/mlx.core.divide.html | 12 +- .../python/_autosummary/mlx.core.divmod.html | 12 +- .../python/_autosummary/mlx.core.einsum.html | 12 +- .../_autosummary/mlx.core.einsum_path.html | 12 +- .../_autosummary/mlx.core.enable_compile.html | 12 +- .../python/_autosummary/mlx.core.equal.html | 12 +- .../python/_autosummary/mlx.core.erf.html | 12 +- .../python/_autosummary/mlx.core.erfinv.html | 12 +- .../python/_autosummary/mlx.core.eval.html | 18 +- .../python/_autosummary/mlx.core.exp.html | 12 +- .../_autosummary/mlx.core.expand_dims.html | 12 +- .../python/_autosummary/mlx.core.expm1.html | 12 +- .../mlx.core.export_function.html | 12 +- .../_autosummary/mlx.core.export_to_dot.html | 12 +- .../_autosummary/mlx.core.exporter.html | 12 +- .../python/_autosummary/mlx.core.eye.html | 12 +- .../mlx.core.fast.layer_norm.html | 12 +- .../mlx.core.fast.metal_kernel.html | 12 +- .../_autosummary/mlx.core.fast.rms_norm.html | 12 +- .../_autosummary/mlx.core.fast.rope.html | 12 +- ...ore.fast.scaled_dot_product_attention.html | 12 +- .../python/_autosummary/mlx.core.fft.fft.html | 12 +- .../_autosummary/mlx.core.fft.fft2.html | 12 +- .../_autosummary/mlx.core.fft.fftn.html | 12 +- .../_autosummary/mlx.core.fft.ifft.html | 12 +- .../_autosummary/mlx.core.fft.ifft2.html | 12 +- .../_autosummary/mlx.core.fft.ifftn.html | 12 +- .../_autosummary/mlx.core.fft.irfft.html | 12 +- .../_autosummary/mlx.core.fft.irfft2.html | 12 +- .../_autosummary/mlx.core.fft.irfftn.html | 12 +- .../_autosummary/mlx.core.fft.rfft.html | 12 +- .../_autosummary/mlx.core.fft.rfft2.html | 12 +- .../_autosummary/mlx.core.fft.rfftn.html | 12 +- .../python/_autosummary/mlx.core.finfo.html | 12 +- .../python/_autosummary/mlx.core.flatten.html | 12 +- .../python/_autosummary/mlx.core.floor.html | 12 +- .../_autosummary/mlx.core.floor_divide.html | 12 +- .../python/_autosummary/mlx.core.full.html | 12 +- .../_autosummary/mlx.core.gather_mm.html | 12 +- .../_autosummary/mlx.core.gather_qmm.html | 12 +- .../mlx.core.get_active_memory.html | 12 +- .../mlx.core.get_cache_memory.html | 12 +- .../mlx.core.get_peak_memory.html | 12 +- .../python/_autosummary/mlx.core.grad.html | 12 +- .../python/_autosummary/mlx.core.greater.html | 12 +- .../_autosummary/mlx.core.greater_equal.html | 12 +- .../mlx.core.hadamard_transform.html | 12 +- .../_autosummary/mlx.core.identity.html | 12 +- .../python/_autosummary/mlx.core.imag.html | 12 +- .../mlx.core.import_function.html | 12 +- .../python/_autosummary/mlx.core.inner.html | 12 +- .../python/_autosummary/mlx.core.isclose.html | 12 +- .../_autosummary/mlx.core.isfinite.html | 12 +- .../python/_autosummary/mlx.core.isinf.html | 12 +- .../python/_autosummary/mlx.core.isnan.html | 12 +- .../_autosummary/mlx.core.isneginf.html | 12 +- .../_autosummary/mlx.core.isposinf.html | 12 +- .../_autosummary/mlx.core.issubdtype.html | 12 +- .../python/_autosummary/mlx.core.jvp.html | 12 +- .../python/_autosummary/mlx.core.kron.html | 12 +- .../_autosummary/mlx.core.left_shift.html | 12 +- .../python/_autosummary/mlx.core.less.html | 12 +- .../_autosummary/mlx.core.less_equal.html | 12 +- .../mlx.core.linalg.cholesky.html | 12 +- .../mlx.core.linalg.cholesky_inv.html | 12 +- .../_autosummary/mlx.core.linalg.cross.html | 12 +- .../_autosummary/mlx.core.linalg.eigh.html | 12 +- .../mlx.core.linalg.eigvalsh.html | 12 +- .../_autosummary/mlx.core.linalg.inv.html | 12 +- .../_autosummary/mlx.core.linalg.lu.html | 12 +- .../mlx.core.linalg.lu_factor.html | 18 +- .../_autosummary/mlx.core.linalg.norm.html | 12 +- .../_autosummary/mlx.core.linalg.pinv.html | 1040 ++++++++++++++++ .../_autosummary/mlx.core.linalg.qr.html | 12 +- .../_autosummary/mlx.core.linalg.solve.html | 18 +- .../mlx.core.linalg.solve_triangular.html | 12 +- .../_autosummary/mlx.core.linalg.svd.html | 12 +- .../_autosummary/mlx.core.linalg.tri_inv.html | 12 +- .../_autosummary/mlx.core.linspace.html | 12 +- .../python/_autosummary/mlx.core.load.html | 12 +- .../python/_autosummary/mlx.core.log.html | 12 +- .../python/_autosummary/mlx.core.log10.html | 12 +- .../python/_autosummary/mlx.core.log1p.html | 12 +- .../python/_autosummary/mlx.core.log2.html | 12 +- .../_autosummary/mlx.core.logaddexp.html | 12 +- .../_autosummary/mlx.core.logical_and.html | 12 +- .../_autosummary/mlx.core.logical_not.html | 12 +- .../_autosummary/mlx.core.logical_or.html | 12 +- .../_autosummary/mlx.core.logsumexp.html | 12 +- .../python/_autosummary/mlx.core.matmul.html | 12 +- .../python/_autosummary/mlx.core.max.html | 12 +- .../python/_autosummary/mlx.core.maximum.html | 12 +- .../python/_autosummary/mlx.core.mean.html | 12 +- .../_autosummary/mlx.core.meshgrid.html | 12 +- .../mlx.core.metal.device_info.html | 12 +- .../mlx.core.metal.is_available.html | 12 +- .../mlx.core.metal.start_capture.html | 12 +- .../mlx.core.metal.stop_capture.html | 12 +- .../python/_autosummary/mlx.core.min.html | 12 +- .../python/_autosummary/mlx.core.minimum.html | 12 +- .../_autosummary/mlx.core.moveaxis.html | 12 +- .../_autosummary/mlx.core.multiply.html | 12 +- .../_autosummary/mlx.core.nan_to_num.html | 12 +- .../_autosummary/mlx.core.negative.html | 12 +- .../_autosummary/mlx.core.new_stream.html | 12 +- .../_autosummary/mlx.core.not_equal.html | 12 +- .../python/_autosummary/mlx.core.ones.html | 12 +- .../_autosummary/mlx.core.ones_like.html | 12 +- .../python/_autosummary/mlx.core.outer.html | 12 +- .../python/_autosummary/mlx.core.pad.html | 12 +- .../_autosummary/mlx.core.partition.html | 12 +- .../python/_autosummary/mlx.core.power.html | 12 +- .../python/_autosummary/mlx.core.prod.html | 12 +- .../_autosummary/mlx.core.put_along_axis.html | 12 +- .../_autosummary/mlx.core.quantize.html | 12 +- .../mlx.core.quantized_matmul.html | 12 +- .../python/_autosummary/mlx.core.radians.html | 12 +- .../mlx.core.random.bernoulli.html | 12 +- .../mlx.core.random.categorical.html | 12 +- .../_autosummary/mlx.core.random.gumbel.html | 12 +- .../_autosummary/mlx.core.random.key.html | 12 +- .../_autosummary/mlx.core.random.laplace.html | 12 +- .../mlx.core.random.multivariate_normal.html | 12 +- .../_autosummary/mlx.core.random.normal.html | 12 +- .../mlx.core.random.permutation.html | 12 +- .../_autosummary/mlx.core.random.randint.html | 12 +- .../_autosummary/mlx.core.random.seed.html | 12 +- .../_autosummary/mlx.core.random.split.html | 12 +- .../mlx.core.random.truncated_normal.html | 12 +- .../_autosummary/mlx.core.random.uniform.html | 12 +- .../python/_autosummary/mlx.core.real.html | 12 +- .../_autosummary/mlx.core.reciprocal.html | 12 +- .../_autosummary/mlx.core.remainder.html | 12 +- .../python/_autosummary/mlx.core.repeat.html | 12 +- .../mlx.core.reset_peak_memory.html | 12 +- .../python/_autosummary/mlx.core.reshape.html | 12 +- .../_autosummary/mlx.core.right_shift.html | 12 +- .../python/_autosummary/mlx.core.roll.html | 12 +- .../python/_autosummary/mlx.core.round.html | 12 +- .../python/_autosummary/mlx.core.rsqrt.html | 12 +- .../python/_autosummary/mlx.core.save.html | 12 +- .../_autosummary/mlx.core.save_gguf.html | 12 +- .../mlx.core.save_safetensors.html | 12 +- .../python/_autosummary/mlx.core.savez.html | 12 +- .../mlx.core.savez_compressed.html | 12 +- .../mlx.core.set_cache_limit.html | 12 +- .../mlx.core.set_default_device.html | 12 +- .../mlx.core.set_default_stream.html | 12 +- .../mlx.core.set_memory_limit.html | 12 +- .../mlx.core.set_wired_limit.html | 12 +- .../python/_autosummary/mlx.core.sigmoid.html | 12 +- .../python/_autosummary/mlx.core.sign.html | 12 +- .../python/_autosummary/mlx.core.sin.html | 12 +- .../python/_autosummary/mlx.core.sinh.html | 12 +- .../python/_autosummary/mlx.core.slice.html | 12 +- .../_autosummary/mlx.core.slice_update.html | 12 +- .../python/_autosummary/mlx.core.softmax.html | 12 +- .../python/_autosummary/mlx.core.sort.html | 12 +- .../python/_autosummary/mlx.core.split.html | 12 +- .../python/_autosummary/mlx.core.sqrt.html | 12 +- .../python/_autosummary/mlx.core.square.html | 12 +- .../python/_autosummary/mlx.core.squeeze.html | 12 +- .../python/_autosummary/mlx.core.stack.html | 12 +- .../python/_autosummary/mlx.core.std.html | 12 +- .../_autosummary/mlx.core.stop_gradient.html | 12 +- .../python/_autosummary/mlx.core.stream.html | 12 +- .../_autosummary/mlx.core.subtract.html | 12 +- .../python/_autosummary/mlx.core.sum.html | 12 +- .../_autosummary/mlx.core.swapaxes.html | 12 +- .../_autosummary/mlx.core.synchronize.html | 12 +- .../python/_autosummary/mlx.core.take.html | 12 +- .../mlx.core.take_along_axis.html | 12 +- .../python/_autosummary/mlx.core.tan.html | 12 +- .../python/_autosummary/mlx.core.tanh.html | 12 +- .../_autosummary/mlx.core.tensordot.html | 12 +- .../python/_autosummary/mlx.core.tile.html | 12 +- .../python/_autosummary/mlx.core.topk.html | 12 +- .../python/_autosummary/mlx.core.trace.html | 12 +- .../_autosummary/mlx.core.transpose.html | 12 +- .../python/_autosummary/mlx.core.tri.html | 12 +- .../python/_autosummary/mlx.core.tril.html | 12 +- .../python/_autosummary/mlx.core.triu.html | 12 +- .../_autosummary/mlx.core.unflatten.html | 12 +- .../_autosummary/mlx.core.value_and_grad.html | 12 +- .../python/_autosummary/mlx.core.var.html | 12 +- .../python/_autosummary/mlx.core.view.html | 12 +- .../python/_autosummary/mlx.core.vjp.html | 12 +- .../python/_autosummary/mlx.core.vmap.html | 12 +- .../python/_autosummary/mlx.core.where.html | 12 +- .../python/_autosummary/mlx.core.zeros.html | 12 +- .../_autosummary/mlx.core.zeros_like.html | 12 +- .../mlx.nn.average_gradients.html | 12 +- .../python/_autosummary/mlx.nn.quantize.html | 12 +- .../_autosummary/mlx.nn.value_and_grad.html | 12 +- .../mlx.optimizers.clip_grad_norm.html | 12 +- .../_autosummary/mlx.utils.tree_flatten.html | 12 +- .../_autosummary/mlx.utils.tree_map.html | 12 +- .../mlx.utils.tree_map_with_path.html | 12 +- .../_autosummary/mlx.utils.tree_reduce.html | 12 +- .../mlx.utils.tree_unflatten.html | 12 +- .../python/_autosummary/stream_class.html | 12 +- docs/build/html/python/array.html | 12 +- docs/build/html/python/data_types.html | 12 +- .../html/python/devices_and_streams.html | 12 +- docs/build/html/python/distributed.html | 12 +- docs/build/html/python/export.html | 12 +- docs/build/html/python/fast.html | 12 +- docs/build/html/python/fft.html | 12 +- docs/build/html/python/linalg.html | 19 +- docs/build/html/python/memory_management.html | 12 +- docs/build/html/python/metal.html | 12 +- docs/build/html/python/nn.html | 12 +- .../python/nn/_autosummary/mlx.nn.ALiBi.html | 12 +- .../nn/_autosummary/mlx.nn.AvgPool1d.html | 12 +- .../nn/_autosummary/mlx.nn.AvgPool2d.html | 12 +- .../nn/_autosummary/mlx.nn.AvgPool3d.html | 12 +- .../nn/_autosummary/mlx.nn.BatchNorm.html | 12 +- .../python/nn/_autosummary/mlx.nn.CELU.html | 12 +- .../python/nn/_autosummary/mlx.nn.Conv1d.html | 12 +- .../python/nn/_autosummary/mlx.nn.Conv2d.html | 12 +- .../python/nn/_autosummary/mlx.nn.Conv3d.html | 12 +- .../_autosummary/mlx.nn.ConvTranspose1d.html | 12 +- .../_autosummary/mlx.nn.ConvTranspose2d.html | 12 +- .../_autosummary/mlx.nn.ConvTranspose3d.html | 12 +- .../nn/_autosummary/mlx.nn.Dropout.html | 12 +- .../nn/_autosummary/mlx.nn.Dropout2d.html | 12 +- .../nn/_autosummary/mlx.nn.Dropout3d.html | 12 +- .../python/nn/_autosummary/mlx.nn.ELU.html | 12 +- .../nn/_autosummary/mlx.nn.Embedding.html | 12 +- .../python/nn/_autosummary/mlx.nn.GELU.html | 12 +- .../python/nn/_autosummary/mlx.nn.GLU.html | 12 +- .../python/nn/_autosummary/mlx.nn.GRU.html | 12 +- .../nn/_autosummary/mlx.nn.GroupNorm.html | 12 +- .../nn/_autosummary/mlx.nn.HardShrink.html | 12 +- .../nn/_autosummary/mlx.nn.HardTanh.html | 12 +- .../nn/_autosummary/mlx.nn.Hardswish.html | 12 +- .../nn/_autosummary/mlx.nn.InstanceNorm.html | 12 +- .../python/nn/_autosummary/mlx.nn.LSTM.html | 12 +- .../nn/_autosummary/mlx.nn.LayerNorm.html | 12 +- .../nn/_autosummary/mlx.nn.LeakyReLU.html | 12 +- .../python/nn/_autosummary/mlx.nn.Linear.html | 12 +- .../nn/_autosummary/mlx.nn.LogSigmoid.html | 12 +- .../nn/_autosummary/mlx.nn.LogSoftmax.html | 12 +- .../nn/_autosummary/mlx.nn.MaxPool1d.html | 12 +- .../nn/_autosummary/mlx.nn.MaxPool2d.html | 12 +- .../nn/_autosummary/mlx.nn.MaxPool3d.html | 12 +- .../python/nn/_autosummary/mlx.nn.Mish.html | 12 +- .../nn/_autosummary/mlx.nn.Module.apply.html | 12 +- .../mlx.nn.Module.apply_to_modules.html | 12 +- .../_autosummary/mlx.nn.Module.children.html | 12 +- .../nn/_autosummary/mlx.nn.Module.eval.html | 12 +- .../mlx.nn.Module.filter_and_map.html | 12 +- .../nn/_autosummary/mlx.nn.Module.freeze.html | 12 +- .../mlx.nn.Module.leaf_modules.html | 12 +- .../mlx.nn.Module.load_weights.html | 12 +- .../_autosummary/mlx.nn.Module.modules.html | 12 +- .../mlx.nn.Module.named_modules.html | 12 +- .../mlx.nn.Module.parameters.html | 12 +- .../mlx.nn.Module.save_weights.html | 12 +- .../_autosummary/mlx.nn.Module.set_dtype.html | 12 +- .../nn/_autosummary/mlx.nn.Module.state.html | 12 +- .../nn/_autosummary/mlx.nn.Module.train.html | 12 +- .../mlx.nn.Module.trainable_parameters.html | 12 +- .../_autosummary/mlx.nn.Module.training.html | 12 +- .../_autosummary/mlx.nn.Module.unfreeze.html | 12 +- .../nn/_autosummary/mlx.nn.Module.update.html | 12 +- .../mlx.nn.Module.update_modules.html | 12 +- .../mlx.nn.MultiHeadAttention.html | 12 +- .../python/nn/_autosummary/mlx.nn.PReLU.html | 12 +- .../mlx.nn.QuantizedEmbedding.html | 12 +- .../_autosummary/mlx.nn.QuantizedLinear.html | 12 +- .../nn/_autosummary/mlx.nn.RMSNorm.html | 12 +- .../python/nn/_autosummary/mlx.nn.RNN.html | 12 +- .../python/nn/_autosummary/mlx.nn.ReLU.html | 12 +- .../python/nn/_autosummary/mlx.nn.ReLU6.html | 12 +- .../python/nn/_autosummary/mlx.nn.RoPE.html | 12 +- .../python/nn/_autosummary/mlx.nn.SELU.html | 12 +- .../nn/_autosummary/mlx.nn.Sequential.html | 12 +- .../python/nn/_autosummary/mlx.nn.SiLU.html | 12 +- .../nn/_autosummary/mlx.nn.Sigmoid.html | 12 +- .../mlx.nn.SinusoidalPositionalEncoding.html | 12 +- .../nn/_autosummary/mlx.nn.Softmax.html | 12 +- .../nn/_autosummary/mlx.nn.Softmin.html | 12 +- .../nn/_autosummary/mlx.nn.Softplus.html | 12 +- .../nn/_autosummary/mlx.nn.Softshrink.html | 12 +- .../nn/_autosummary/mlx.nn.Softsign.html | 12 +- .../python/nn/_autosummary/mlx.nn.Step.html | 12 +- .../python/nn/_autosummary/mlx.nn.Tanh.html | 12 +- .../nn/_autosummary/mlx.nn.Transformer.html | 12 +- .../nn/_autosummary/mlx.nn.Upsample.html | 12 +- .../nn/_autosummary/mlx.nn.init.constant.html | 12 +- .../mlx.nn.init.glorot_normal.html | 12 +- .../mlx.nn.init.glorot_uniform.html | 12 +- .../_autosummary/mlx.nn.init.he_normal.html | 12 +- .../_autosummary/mlx.nn.init.he_uniform.html | 12 +- .../nn/_autosummary/mlx.nn.init.identity.html | 12 +- .../nn/_autosummary/mlx.nn.init.normal.html | 12 +- .../nn/_autosummary/mlx.nn.init.uniform.html | 12 +- .../_autosummary_functions/mlx.nn.celu.html | 12 +- .../nn/_autosummary_functions/mlx.nn.elu.html | 12 +- .../_autosummary_functions/mlx.nn.gelu.html | 12 +- .../mlx.nn.gelu_approx.html | 12 +- .../mlx.nn.gelu_fast_approx.html | 12 +- .../nn/_autosummary_functions/mlx.nn.glu.html | 12 +- .../mlx.nn.hard_shrink.html | 12 +- .../mlx.nn.hard_tanh.html | 12 +- .../mlx.nn.hardswish.html | 12 +- .../mlx.nn.leaky_relu.html | 12 +- .../mlx.nn.log_sigmoid.html | 12 +- .../mlx.nn.log_softmax.html | 12 +- .../mlx.nn.losses.binary_cross_entropy.html | 12 +- .../mlx.nn.losses.cosine_similarity_loss.html | 12 +- .../mlx.nn.losses.cross_entropy.html | 12 +- .../mlx.nn.losses.gaussian_nll_loss.html | 12 +- .../mlx.nn.losses.hinge_loss.html | 12 +- .../mlx.nn.losses.huber_loss.html | 12 +- .../mlx.nn.losses.kl_div_loss.html | 12 +- .../mlx.nn.losses.l1_loss.html | 12 +- .../mlx.nn.losses.log_cosh_loss.html | 12 +- .../mlx.nn.losses.margin_ranking_loss.html | 12 +- .../mlx.nn.losses.mse_loss.html | 12 +- .../mlx.nn.losses.nll_loss.html | 12 +- .../mlx.nn.losses.smooth_l1_loss.html | 12 +- .../mlx.nn.losses.triplet_loss.html | 12 +- .../_autosummary_functions/mlx.nn.mish.html | 12 +- .../_autosummary_functions/mlx.nn.prelu.html | 12 +- .../_autosummary_functions/mlx.nn.relu.html | 12 +- .../_autosummary_functions/mlx.nn.relu6.html | 12 +- .../_autosummary_functions/mlx.nn.selu.html | 12 +- .../mlx.nn.sigmoid.html | 12 +- .../_autosummary_functions/mlx.nn.silu.html | 12 +- .../mlx.nn.softmax.html | 12 +- .../mlx.nn.softmin.html | 12 +- .../mlx.nn.softplus.html | 12 +- .../mlx.nn.softshrink.html | 12 +- .../_autosummary_functions/mlx.nn.step.html | 12 +- .../_autosummary_functions/mlx.nn.tanh.html | 12 +- docs/build/html/python/nn/functions.html | 12 +- docs/build/html/python/nn/init.html | 12 +- docs/build/html/python/nn/layers.html | 12 +- docs/build/html/python/nn/losses.html | 12 +- docs/build/html/python/nn/module.html | 12 +- docs/build/html/python/ops.html | 26 +- docs/build/html/python/optimizers.html | 12 +- .../_autosummary/mlx.optimizers.AdaDelta.html | 12 +- .../mlx.optimizers.Adafactor.html | 12 +- .../_autosummary/mlx.optimizers.Adagrad.html | 12 +- .../_autosummary/mlx.optimizers.Adam.html | 12 +- .../_autosummary/mlx.optimizers.AdamW.html | 12 +- .../_autosummary/mlx.optimizers.Adamax.html | 12 +- .../_autosummary/mlx.optimizers.Lion.html | 12 +- ....optimizers.Optimizer.apply_gradients.html | 12 +- .../mlx.optimizers.Optimizer.init.html | 12 +- .../mlx.optimizers.Optimizer.state.html | 12 +- .../mlx.optimizers.Optimizer.update.html | 12 +- .../_autosummary/mlx.optimizers.RMSprop.html | 12 +- .../_autosummary/mlx.optimizers.SGD.html | 12 +- .../mlx.optimizers.cosine_decay.html | 12 +- .../mlx.optimizers.exponential_decay.html | 12 +- .../mlx.optimizers.join_schedules.html | 12 +- .../mlx.optimizers.linear_schedule.html | 12 +- .../mlx.optimizers.step_decay.html | 12 +- .../python/optimizers/common_optimizers.html | 12 +- .../html/python/optimizers/optimizer.html | 12 +- .../html/python/optimizers/schedulers.html | 12 +- docs/build/html/python/random.html | 12 +- docs/build/html/python/transforms.html | 33 +- docs/build/html/python/tree_utils.html | 12 +- docs/build/html/search.html | 12 +- docs/build/html/searchindex.js | 2 +- docs/build/html/usage/compile.html | 12 +- docs/build/html/usage/distributed.html | 12 +- docs/build/html/usage/export.html | 12 +- .../build/html/usage/function_transforms.html | 12 +- docs/build/html/usage/indexing.html | 12 +- .../html/usage/launching_distributed.html | 12 +- docs/build/html/usage/lazy_evaluation.html | 12 +- docs/build/html/usage/numpy.html | 12 +- docs/build/html/usage/quick_start.html | 12 +- docs/build/html/usage/saving_and_loading.html | 12 +- docs/build/html/usage/unified_memory.html | 12 +- docs/build/html/usage/using_streams.html | 12 +- 528 files changed, 8356 insertions(+), 2106 deletions(-) create mode 100644 docs/build/html/_sources/python/_autosummary/mlx.core.async_eval.rst create mode 100644 docs/build/html/_sources/python/_autosummary/mlx.core.broadcast_arrays.rst create mode 100644 docs/build/html/_sources/python/_autosummary/mlx.core.contiguous.rst create mode 100644 docs/build/html/_sources/python/_autosummary/mlx.core.linalg.pinv.rst create mode 100644 docs/build/html/python/_autosummary/mlx.core.async_eval.html create mode 100644 docs/build/html/python/_autosummary/mlx.core.broadcast_arrays.html create mode 100644 docs/build/html/python/_autosummary/mlx.core.contiguous.html create mode 100644 docs/build/html/python/_autosummary/mlx.core.linalg.pinv.html diff --git a/docs/build/html/.buildinfo b/docs/build/html/.buildinfo index 763d8a0c2..1062ede52 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: 2f6ec3ca18b17f6d0ad6bba5887f704c +config: 8e5a9f3fa6ac8cf3fa3d36c262106632 tags: 645f666f9bcd5a90fca523b33c5a78b7 diff --git a/docs/build/html/_sources/dev/extensions.rst b/docs/build/html/_sources/dev/extensions.rst index b8c3a4995..2aef28f99 100644 --- a/docs/build/html/_sources/dev/extensions.rst +++ b/docs/build/html/_sources/dev/extensions.rst @@ -93,9 +93,9 @@ Primitives ^^^^^^^^^^^ A :class:`Primitive` is part of the computation graph of an :class:`array`. It -defines how to create outputs arrays given a input arrays. Further, a +defines how to create output arrays given input arrays. Further, a :class:`Primitive` has methods to run on the CPU or GPU and for function -transformations such as ``vjp`` and ``jvp``. Lets go back to our example to be +transformations such as ``vjp`` and ``jvp``. Let's go back to our example to be more concrete: .. code-block:: C++ @@ -128,7 +128,7 @@ more concrete: /** The vector-Jacobian product. */ std::vector vjp( const std::vector& primals, - const array& cotan, + const std::vector& cotangents, const std::vector& argnums, const std::vector& outputs) override; @@ -469,7 +469,7 @@ one we just defined: const std::vector& tangents, const std::vector& argnums) { // Forward mode diff that pushes along the tangents - // The jvp transform on the primitive can built with ops + // The jvp transform on the primitive can be built with ops // that are scheduled on the same stream as the primitive // If argnums = {0}, we only push along x in which case the @@ -481,7 +481,7 @@ one we just defined: auto scale_arr = array(scale, tangents[0].dtype()); return {multiply(scale_arr, tangents[0], stream())}; } - // If, argnums = {0, 1}, we take contributions from both + // If argnums = {0, 1}, we take contributions from both // which gives us jvp = tangent_x * alpha + tangent_y * beta else { return {axpby(tangents[0], tangents[1], alpha_, beta_, stream())}; @@ -735,7 +735,7 @@ Let's look at a simple script and its results: print(f"c shape: {c.shape}") print(f"c dtype: {c.dtype}") - print(f"c correct: {mx.all(c == 6.0).item()}") + print(f"c is correct: {mx.all(c == 6.0).item()}") Output: @@ -743,7 +743,7 @@ Output: c shape: [3, 4] c dtype: float32 - c correctness: True + c is correct: True Results ^^^^^^^ diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.async_eval.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.async_eval.rst new file mode 100644 index 000000000..f44f1c49e --- /dev/null +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.async_eval.rst @@ -0,0 +1,6 @@ +mlx.core.async\_eval +==================== + +.. currentmodule:: mlx.core + +.. autofunction:: async_eval \ No newline at end of file diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.broadcast_arrays.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.broadcast_arrays.rst new file mode 100644 index 000000000..5482dafe5 --- /dev/null +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.broadcast_arrays.rst @@ -0,0 +1,6 @@ +mlx.core.broadcast\_arrays +========================== + +.. currentmodule:: mlx.core + +.. autofunction:: broadcast_arrays \ No newline at end of file diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.contiguous.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.contiguous.rst new file mode 100644 index 000000000..5b06a91ac --- /dev/null +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.contiguous.rst @@ -0,0 +1,6 @@ +mlx.core.contiguous +=================== + +.. currentmodule:: mlx.core + +.. autofunction:: contiguous \ No newline at end of file diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.pinv.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.pinv.rst new file mode 100644 index 000000000..5b6af868d --- /dev/null +++ b/docs/build/html/_sources/python/_autosummary/mlx.core.linalg.pinv.rst @@ -0,0 +1,6 @@ +mlx.core.linalg.pinv +==================== + +.. currentmodule:: mlx.core.linalg + +.. autofunction:: pinv \ 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 769f4bbb1..b01f74117 100644 --- a/docs/build/html/_sources/python/linalg.rst +++ b/docs/build/html/_sources/python/linalg.rst @@ -20,5 +20,6 @@ Linear Algebra eigh lu lu_factor + pinv solve solve_triangular diff --git a/docs/build/html/_sources/python/ops.rst b/docs/build/html/_sources/python/ops.rst index c0d098b21..66c5764ed 100644 --- a/docs/build/html/_sources/python/ops.rst +++ b/docs/build/html/_sources/python/ops.rst @@ -36,10 +36,12 @@ Operations bitwise_or bitwise_xor block_masked_mm + broadcast_arrays broadcast_to ceil clip concatenate + contiguous conj conjugate convolve diff --git a/docs/build/html/_sources/python/transforms.rst b/docs/build/html/_sources/python/transforms.rst index fbdfd4f08..23f86720b 100644 --- a/docs/build/html/_sources/python/transforms.rst +++ b/docs/build/html/_sources/python/transforms.rst @@ -9,6 +9,7 @@ Transforms :toctree: _autosummary eval + async_eval compile custom_function disable_compile diff --git a/docs/build/html/_static/documentation_options.js b/docs/build/html/_static/documentation_options.js index 77faecbeb..ae734fbf9 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.24.1', + VERSION: '0.24.2', LANGUAGE: 'en', COLLAPSE_INDEX: false, BUILDER: 'html', diff --git a/docs/build/html/cpp/ops.html b/docs/build/html/cpp/ops.html index c95d17758..795da9057 100644 --- a/docs/build/html/cpp/ops.html +++ b/docs/build/html/cpp/ops.html @@ -8,7 +8,7 @@ - Operations — MLX 0.24.1 documentation + Operations — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -453,6 +455,7 @@
  • Transforms
    diff --git a/docs/build/html/dev/custom_metal_kernels.html b/docs/build/html/dev/custom_metal_kernels.html index 29290bbd3..0044d5f97 100644 --- a/docs/build/html/dev/custom_metal_kernels.html +++ b/docs/build/html/dev/custom_metal_kernels.html @@ -8,7 +8,7 @@ - Custom Metal Kernels — MLX 0.24.1 documentation + Custom Metal Kernels — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -453,6 +455,7 @@
  • Transforms
    diff --git a/docs/build/html/dev/extensions.html b/docs/build/html/dev/extensions.html index ee53f152c..0ddbf291c 100644 --- a/docs/build/html/dev/extensions.html +++ b/docs/build/html/dev/extensions.html @@ -8,7 +8,7 @@ - Custom Extensions in MLX — MLX 0.24.1 documentation + Custom Extensions in MLX — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -453,6 +455,7 @@
  • Transforms
    @@ -1008,9 +1012,9 @@ easy to use interface that use

    Primitives#

    A Primitive is part of the computation graph of an array. It -defines how to create outputs arrays given a input arrays. Further, a +defines how to create output arrays given input arrays. Further, a Primitive has methods to run on the CPU or GPU and for function -transformations such as vjp and jvp. Lets go back to our example to be +transformations such as vjp and jvp. Let’s go back to our example to be more concrete:

    class Axpby : public Primitive {
   public:
@@ -1040,7 +1044,7 @@ more concrete:
    
     /** The vector-Jacobian product. */
     std::vector<array> vjp(
         const std::vector<array>& primals,
-        const array& cotan,
+        const std::vector<array>& cotangents,
         const std::vector<int>& argnums,
         const std::vector<array>& outputs) override;
 
@@ -1360,7 +1364,7 @@ one we just defined:
    
         const std::vector<array>& tangents,
         const std::vector<int>& argnums) {
     // Forward mode diff that pushes along the tangents
-    // The jvp transform on the primitive can built with ops
+    // The jvp transform on the primitive can be built with ops
     // that are scheduled on the same stream as the primitive
 
     // If argnums = {0}, we only push along x in which case the
@@ -1372,7 +1376,7 @@ one we just defined:
    
         auto scale_arr = array(scale, tangents[0].dtype());
         return {multiply(scale_arr, tangents[0], stream())};
     }
-    // If, argnums = {0, 1}, we take contributions from both
+    // If argnums = {0, 1}, we take contributions from both
     // which gives us jvp = tangent_x * alpha + tangent_y * beta
     else {
         return {axpby(tangents[0], tangents[1], alpha_, beta_, stream())};
@@ -1608,13 +1612,13 @@ import the Python package and play with it as you would any other MLX operation.
 
 print(f"c shape: {c.shape}")
 print(f"c dtype: {c.dtype}")
-print(f"c correct: {mx.all(c == 6.0).item()}")
+print(f"c is correct: {mx.all(c == 6.0).item()}")

    Output:

    c shape: [3, 4]
 c dtype: float32
-c correctness: True
+c is correct: True
    diff --git a/docs/build/html/dev/metal_debugger.html b/docs/build/html/dev/metal_debugger.html index 9b1f394f9..a37bce53e 100644 --- a/docs/build/html/dev/metal_debugger.html +++ b/docs/build/html/dev/metal_debugger.html @@ -8,7 +8,7 @@ - Metal Debugger — MLX 0.24.1 documentation + Metal Debugger — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -453,6 +455,7 @@
  • Transforms
    diff --git a/docs/build/html/dev/mlx_in_cpp.html b/docs/build/html/dev/mlx_in_cpp.html index b176e2815..9f8943c5f 100644 --- a/docs/build/html/dev/mlx_in_cpp.html +++ b/docs/build/html/dev/mlx_in_cpp.html @@ -8,7 +8,7 @@ - Using MLX in C++ — MLX 0.24.1 documentation + Using MLX in C++ — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -136,8 +136,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -290,10 +290,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -452,6 +454,7 @@
  • Transforms
    diff --git a/docs/build/html/examples/linear_regression.html b/docs/build/html/examples/linear_regression.html index 5fa855e55..84e5edb07 100644 --- a/docs/build/html/examples/linear_regression.html +++ b/docs/build/html/examples/linear_regression.html @@ -8,7 +8,7 @@ - Linear Regression — MLX 0.24.1 documentation + Linear Regression — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -453,6 +455,7 @@
  • Transforms
    diff --git a/docs/build/html/examples/llama-inference.html b/docs/build/html/examples/llama-inference.html index 1ba9d527c..4616d2896 100644 --- a/docs/build/html/examples/llama-inference.html +++ b/docs/build/html/examples/llama-inference.html @@ -8,7 +8,7 @@ - LLM inference — MLX 0.24.1 documentation + LLM inference — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -453,6 +455,7 @@
  • Transforms
    diff --git a/docs/build/html/examples/mlp.html b/docs/build/html/examples/mlp.html index 7d01c75a0..9ab54d496 100644 --- a/docs/build/html/examples/mlp.html +++ b/docs/build/html/examples/mlp.html @@ -8,7 +8,7 @@ - Multi-Layer Perceptron — MLX 0.24.1 documentation + Multi-Layer Perceptron — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -453,6 +455,7 @@
  • Transforms
    diff --git a/docs/build/html/genindex.html b/docs/build/html/genindex.html index 899196f62..a2e694794 100644 --- a/docs/build/html/genindex.html +++ b/docs/build/html/genindex.html @@ -7,7 +7,7 @@ - Index — MLX 0.24.1 documentation + Index — MLX 0.24.2 documentation @@ -35,7 +35,7 @@ - + @@ -136,8 +136,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -290,10 +290,12 @@
  • mlx.core.bitwise_or
  • mlx.core.bitwise_xor
  • mlx.core.block_masked_mm
    • +
  • mlx.core.broadcast_arrays
  • mlx.core.broadcast_to
  • mlx.core.ceil
  • mlx.core.clip
  • mlx.core.concatenate
    • +
  • mlx.core.contiguous
  • mlx.core.conj
  • mlx.core.conjugate
  • mlx.core.convolve
    • @@ -452,6 +454,7 @@
  • Transforms
    @@ -979,10 +983,10 @@ document.write(`
  • arcsin() (in module mlx.core)
    • - - +
    • arcsinh() (in module mlx.core)
    • arctan (C++ function) @@ -1034,6 +1038,8 @@ document.write(`
    • astype (C++ function)
    • astype() (array method) +
      • +
    • async_eval() (in module mlx.core)
    • at (array property)
      • @@ -1092,6 +1098,8 @@ document.write(`
    • block_masked_mm() (in module mlx.core)
    • broadcast_arrays (C++ function) +
      • +
    • broadcast_arrays() (in module mlx.core)
    • broadcast_to (C++ function)
      • @@ -1146,6 +1154,8 @@ document.write(`
    • constant() (in module mlx.nn.init)
    • contiguous (C++ function) +
      • +
    • contiguous() (in module mlx.core)
    • conv1d (C++ function)
      • @@ -1946,6 +1956,8 @@ document.write(`
    • partition() (in module mlx.core)
    • permutation() (in module mlx.core.random) +
      • +
    • pinv() (in module mlx.core.linalg)
    • power (C++ function)
      • diff --git a/docs/build/html/index.html b/docs/build/html/index.html index 19cc170b8..e37a153ac 100644 --- a/docs/build/html/index.html +++ b/docs/build/html/index.html @@ -8,7 +8,7 @@ - MLX — MLX 0.24.1 documentation + MLX — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -139,8 +139,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -293,10 +293,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -455,6 +457,7 @@
    • Transforms
      diff --git a/docs/build/html/install.html b/docs/build/html/install.html index 65f7579e8..9d51a3855 100644 --- a/docs/build/html/install.html +++ b/docs/build/html/install.html @@ -8,7 +8,7 @@ - Build and Install — MLX 0.24.1 documentation + Build and Install — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/objects.inv b/docs/build/html/objects.inv index 9462a408751ac55611e0b8003b93c3272e6a47d7..533c4679449597e0cbedd48691ba7c10a912177e 100644 GIT binary patch delta 27556 zcmZ6yV|XA<6E3{5ZQFJ>8{65~b~f7B6Wg}!Y;4=MZ6_Ol`@H9zKi|)qshaMttGes1 zyLt*&Kmt}lVtGMW(_x9L*zfTMFk! zSfo8pP`n^+CX}Y8z#Bi2s+w`jSpkbz|DyKSxF~N^ik@Ni71!>XAdtz&DI9bcE(yR0J1$>b8e<6UN(f`OW zXa_X20&C=R(--MTmMBY?$yRKuJl)@f4vS10Jiz+6zP9nJwHF5y-*fRf8K#$ z$FpVpS(as%X`ci{^)YP*qH|O_Z7&55AOFyHt6MSMZ?;(p2>*dSspUJPXzy!N(_PAp zSW~nf*V1e$IZ%1xzlwE$Pn-FD%*KA`;*8qG6Yy$purAjdzL-|t%&gIgDRR4N#m*p3 zX3QF@s!vtG7!Z%t6DvMR<2eal->@$GU^**)Aqk6hw2%#;Gp>}%5;>Wden5}X=Jr38 zrd+_DYdLE=GFl9WV@z}Q3NIuj$`}L!mi*U?sFD$-^@aT7)D4;9FR0k#zVHQJ&shtL zjW!F(lNOX`TD9{n92kyLm4}VmDxDk;+G@Rr91|%6TA0@8Rfekd7RG51+#ft<1BIQTxWTmdwIgG`Wv|a-H=wy37+Z14+Bge~froO0GXw zM#u-CY}YebbkbY&TW_;T%L}_FV=joF79mY8quZfJUpy>L@bVQH^9D)^XQz1GclecYXM|U(W3+})D2?zrT zRonHtXetf;2hVNsC{N?joU_3i3$@6c3!g|L{-1$zdmF`7<)?*t2vnZLVs(lgEZ478 zgVVNa&BX&7^YkmtYF7q(aXkb#*t8jzqq0wKKM1fqyeUfe?T$*DvSFIy?`Ms4b#ob| z;|XP8_`ve76$3-nTZ8$l#`@poSE}w6@1879{D=y_EAHv)(8oqrajzDA3%!sF?Lt%sEMoP^8KvtZt@RmEL?< zxt%H>(ZOP>{+B`0Gs7+TE~x1M#oOMdmO*q(k_M<9Ado;m@i1I5tN3ZZujTUYc*~N* zL3D5BY6kW7@mv7T&Lynu!^zo+`WT*6x}A?oc&X~iIX2x>NXvJQp-n4*m!TKtF(|!Y zrwEhtRPK#;Xl|gCl>HdakdzZ8S0mbKL~+JJ*L#0ttXF|RSGm>bsk%k_&_ydZ)=Oxx zYX|fx)l9}?)r?s=3?tZzRo2lknea<=l1gZ5wwB2J_Wu|UKj1TSyy4m@RX_S`@=Ur> zamW6k*<}m~2cRW;a3tH%Y}vU~`qUf@osFS$btJTQoskYh$0vF+Og0^}415k-BG6nv z%KRDw1k#%XPAq-5qsCryGA?C4-jrH+*@14Rpo;Cdpl&_6ig>BKf{UieN!#CCZCHO0 zr>_sX-WZvlLWZ`u;aIzAES0rvFk3KNa7os2W?{VM!aqfLAGL2|DQn7fgbz6@{QMZX zPzz6n!ROa|7bALY?mJZ}6t9j1y|$g<-+2Y4Sw~?HGTtWZ?P>(|yrdw`m_+q3{Q-{h zp07%W(uzA5MrTcbuS}5ZIswXPGwzrLbIYdFF-1-bKItbjah-PQZo=$KTzwBl-lIGP zQ)t2jSRE#Wjf3fsI|B?_o|HY~b_B*E$$J&Hp62zteep!?NaXGg~v(-1Q+oN9N9_w1e$v}d69k2SKp?=NyEG5Rd(nVUL`)vo~=$P4AlWyhs zt>WP{6I~bFben)1-wJ0X%oBahXZGe-cU*WAKa-w@dsCkR-ay8Ida=*b&y!v+8*(@S(l)h?ByyfoMSsXu?(=I$cTbzio z10phL#uH3c$43-j1RTK$z})(sYoXJ}7ofXcyE%84P}5t8MExG@Qgo=H@25+o=;qzd zi9vVftx4BD?=stoZ20t{70EBuNX)?!uOFR?dCoX2LvWsA{E4bY$@AwhG>p$1$H?xV zCoDhT(synNKPjd8kKndM2SYlBGA#b)0`6!8-B+&fsnZ^47)1AgZAHb&^liZTRG0oN z?-j!K&&6S{Jv`diSaUBCKB9*hyVBF^-70}2y6vB{!@Nh~;a39Mm-iB_aZy=s(}mt| z^9GMf2ArO9AGdZQGyM97U}{WIPe#vA;@zM^?;1e_I-S@XhUV^h?&^0AopeN^1shAE z!bLj7vLI7%b$koJONAChr&3LM`Esw`7NLu}o-2jTOledvZqgdIwcjxZr1r!2s9sFZ z-<}s0bCgRlF=ACW)XNR1ziF@<=W<7F{6iJ2F+?8@tCWablJ!9U8Heii5hk5bRQj2_ z!~A-{(d)zY)bx~m)S)p&gBlvnC+i)l10NJa7CaXddM*cIlNW6xaFI&zfQ}yES@#)9 zHB?7&0q~7;xdSMdL%`_L-zr3c(pswp!H=m!3A*Y0y+409nF zM{Tgv9@VP7#Oof3Ij=vwSf!j@I2@|}wLDFeR!p$<7Ib?icKa~u3b#&C@<*M{hb$DF zZY~Zbz6Ger_pHd9zdEAqUydN_-kR1&4CK+3iddg+^G~!-f}dOts8;ve(liZIRG-6* zw=l+K*GFp8BV2pJ!^yiGhArK$$OXM`^gTcHtC%oD#Ty(FqE1-BGVI9{{~QR&gjyrAKGCyS2xjD_s7Td$1`es zx1h%SkyAfHBcanQiFSCCl%b zw`0I!L~6U<%*G?@Q8gB;YiD9>J;N_t)A|D^=b^m8l7x37Ap-S=FUWr4YlnXJ}FK!keYlS|M=paOUW*`W7ej=wlK=KmAsM= zev0!xdhmu+EfOrY>A225?)~gX+(f*&o_Yodjux8{PQF_hG^rzLlAx4UvC^%z1bm5d4N2oMjNLDDEi?!Z+#3JP(TD?!CW0|6Pi0O`QZH2^5^G^(CuemN{FQ)DkRa#C|t&`!yEye-N z5}oO=?jMlJQ*dEKK1j0TsgDd+uY>0;gWH!+OsMf&r>!@eopkrz0?y)v zUf&E=IsPu^4LucYA$xwb!K&(K2FM2B$ z7$15mP00+9sb%n*Ww7lP+g0ZeJQ!Lq)72wI%1<-dM-N#h**^poCa!N1YHDhJMsrwf zE>jcOiBCZ^H8$>7uQCqjAh%I}9m~Qg?X0T}q~uNSi|ipw6cs2!BlyS<{G)+3_UxeK z$3Cqk8LQU%9|Ylr|rYHkS8igFBtPV!ujCeOOjvPua2Dk(IB zb5p}-Qyblv>i^_B@D&fKm1kAU`5`dfDI@UZLFzg@nB*ETZ6<_vRn7V`{kzIKt)H#v zRE7y%jQkk^-5?U3xDqyw=xj5G^t;qH8Q|l~epSo9jX9-`jDu1uz(5mU!vC}|CG-|N zDJ%VFlr?EW9G{hO&fwt?+PN>(gsYeP7O7a(`Wc_cg;O~1nfSw_8Y}Nsovr(}ns(5U zDJ~kpE3qIp*43N^AVjOmA^ux}IR|F`1~|MWe1l&FR$zgS5X&kObqFL?Z)&2;vvuKd zALo8g8NKN=uqq4M#VOQ@Da5$q(hpk^W8b2FXn9hI$MzpO`VSl_B_f=9=XMaBRXz&} z)+`G*^Av^&-s)r;*vz#~LUI4avtKy>y%3VsnQ{eN1L@X%pT;y*%WK3BD}`TparuVK zJUWHMQWe5$>DnihJj@`<{YSWF%(LjJQ@d%YathcFtU!}pt<^1YDKaRRc)2uu^J1Uan~^#N0-~; z3)zzq|F>ViG3l#L)4(H?4*x0O!p(tIDx`F-}5axM4T^Pf&Smv(>a^ay_QZd}{>qGlPVe4~y`2;+QqDHv(FR39dsOLxkyV`?GzMN+^6@40!C$6eRlg{rnLQ zqwqvG_;-Ndo56@^){!5=LevbiHG2lwQ;S)`(y`1^9FwL*mn6ZLYrBiyrFgmc0Q1ra zWWidLYT#1KFdaM6xDJcfeD>Tg#HD@@Ih<@J4WTXrh!;|fcLDlzO1hzMmB=g9pu0A5 zq_?KA)U+7;f?g=OUx3wdh|78$rsM!~6E9xmrb{I$`xl#Dp1a0T`0V?1qMlmUrYNK=^>`o*j1VeryNk{hMmRm(^!?n z`nCREaj^WF(cfMq|DXC@+5wgj+PB{$*o_0c^z~w)fXGnb|Hc@9A2I_;_516Oxd*2e z%B0jQpwVFS(uVCzE+dUbwfh)NCjcioC5~MRuV2pkzB_g^Onbk5@&3?+b4n!&Z~Rr; zyTVB}^lrz__2d7}e&gc*Kl_xjT-EhT2^8v~Y~j6eV$G>*JGkLur!wLrEK{&QstF9J zSUl4ym1O#VqP^mHK3&IytN9}rqF{|G6EH$wnYFOAz;a6Gd29XJW^w-(nmP6mE?dvN zdX4hjQJjxpB>s;G`A-7Cy zv5F;ZC=-~8YKdY&EG@_XQ~?HB3h4ymJjUj4P(XXkMzD=gMAf~`_c6JX|C6w|Z?ac6 z;CBmFCezjorti+K-T%=z%@Q^tOem>RPHUgu{V!&fx*uzo{pw%3Vc)3)m7~|0?yd+) zyxs{Idp}p&Gs{43vy`rQH7c^1Uhc;szt$FB?9L*7)iVhA@Qqzamufa18%OI(NEOO~44R=bB5(Si7!m~`Wqu_=almf-a#Y+M-V?M#PwsNCO4p|$qul~-jr^hw-}K5KdSV&0 zIt8j)Is^a5%pqoQU8{yQ;*}Ej3o`QIo-SPbWDro_xUKVEbUSn*)#~b2j6nU2s9i4C z)#{pkhBR9V{=+EAV|A*o$vK7dJpPH79j}QY;yCIs%fr^5&OzW?B;pQNeB!x9GC9S= z9EJP6@Ln%g-Y03{pcGr(P&-KF%;;xozGF1sj zO&R#u=jm}U{%EE9S4yAVynW?pE1s!OS_ymV`gOnfUpx%8bRg>2zBU@HGrinpgkwB* zRd(k-3iy0KQ?L`$Wl8Iw;e2NQ2SEp-hfx#X(qM8Q<8#N(&V~h3GJNw)%+%~39qdUZ zTIedv&;HfEYN>_ezeZOc_f~!s3)CC36#+@vaM}}xnd3^u%I2syjWB0P&`;s3UQxVf z($Qi~^>x9$YtvmCfXKYsV8~AiLzq=uh~_uDPJa?YQNjQ{k+fJ(0 z>-W>f6_BL}kR^W(7pk{CW5uz(OR|&Tngc|eDQvUi3HpheZ4r8z#RO!o>QYg??y%0V zH-;|yW#J+3j5VJZ55lK-4!Z|iq9d)sDLl%eMU`#U7o0`!NVB^1_+kR08sK){NoFbY zF06rThO%b+N%iORsJB=l+}cOf`4`4w=80WO*%xbJgl4Y?o2?TQ`#k6YLaPj#^2S8p z#zI#&^9L3grR|Jvp!YC;TpwH_;Fxae!9r{vgmM4toqIE$BL`!!vP4d=`?4aH;HQ`S z9K_}4(BoIQ75|T1x5CfySb)(as3W&^rz2JMTEYUe;*@v!NBdCBJcH)Fd$tdU&E2bw z{-9UM+n^Ly!yDf67hUtGWsOt6j1Ri&&@O2|eYD z^7Y_RyBoP2>a)!vrQtJUeW#+FZxM~jBp}Fws6q6zgU->i(PbM0AXqn>Ng`%QTn|_p z_g^C?_{R-_h^&TKL8vo9nJY_lPf-a{bN$isT^#r8TA7x%1GU8&Zn$=v6LG4;fHQKbj%LLC&T zG2=X+P#1-^O4+FZmaS1AU*+;YN#*uz^y5B}PL(3XH1O0oDmbG}q16iA8{#I+v){(X zKHBW8Vcs7#2z~Tb85gzxR!W2tmLC+$tXve=RD)lwAExYi;E6k4XBbdSOEA>*2w#9J z*Yv$KO*q=W4gEu9~lJ`ta%j3m3m;v=S1;PX=EQz*g5a zyczbG1nssr#w+XA%dv~jD3&X*%}!m6om07I4u_83%zF;NxC>RIEE|vdqDm!NB?b`x z=rs+&l{aq#M`|UL>;9N28oZD#nc;Ax2*PnoDAZ4$*lzAg>Vo(IC?5=r#*D?zboe0_U1Ok$l(IcR}KRd&aWvLN!oP z^aa8${OXwDVj2DOWRdD-hcS2NKV*}3o{LDku6HuYk2XaA=Lk+IQZ)cwe9BN`S^S*F zuHNqXc=k38dN!j;Q@BP<@QwuC-#CYLk)B#RVG>jy-FhN0P|;d#5k_}I+K^`4S}y2# z_a9aUjJ}<@;vHif@(<&tk`0xn+<&XUbyAdJwCHo_c3;}2Rtj1ihJD8iQ;g6{D#(n@ z82?$*W4|Le7#L+3udEqyjglFTTFHu(T?40I1J+&7v+A(I0Bh7}AttU1NkQaGLa>$g zf4d+5r}+_xZJ~DwBNUr8w$SU8yv7t0TM5cz9pf=tU0bc1!g1{H@;>s^nazAK8)G_r z+#3@l2}Xc|O)##DwxJEzp_g!g%RAyG?*%P;xQ{z zH*FMcdh1euVgBsORqBELn)Bkks<4GqLWI*$b>-ZVY$WQNR>5LLHs6ON|<~9HyF4Jk-A;f1?ZkwGL|1 zMz12yV+i92Qe!IB>?}1)6dV%%&bq<9vqbAD-Wpd3t01MB1EJMPy;gEKgP8op6Ip~crzaS%;iaE}iu=QHYaiRGbwy-EGaZXLKeKS3-%$iFC`+lF8JzmSZ z7m73TW^bOhhm!34Q$kjW=@dte^P&$^cP_E@)dDU1)ImKWSY)=2QCzc+q&NPE%MzG* zoU0|$Ph1Sb`^Pz*Yz>p_Tz8?FlH6l1x$N0fDt2=AmNdff*NER_luKt+dH+*7K+JX* z42Y^>5Yt5>7FxbHz=FzLNI)IdKIb8F(%GKmY+hrz{)ZuyC+itRSFxmrtFeZ*Yy`aN z#-zPH-%Y<>e$<&dBg}jp9~IGm0=0%A#g$0y1Qq;WyVY(IfAwFvw{1B+P@U^%eSbg@ zesOep7=^CFdoK}%@(?H7W5rgiO$%cjz1w57Epfk>!!)J z#BD?srzVkJ%dn$>%+ov3U@~L=z!1X4B>$fT=5acC&Q320FYN9_Z-;ylglJZMP~TOZ z8CXOCnOl0TD3hA^RB6?Q?+fXc%+}xJN;Hq$7>WW9*73-_kW482@`M#tUo+NOo<8mJ zvm4%j7OL*|VRSzPgcV>ga+2*_oepm9vbWl`jC6XXeM+ z@FR^ZncouF?&9rl5xX$!TGySN!5S5W#tCQxxnVrp2;qp>Z5JvISZQEAYXRHhP1lpB z91|~KaoKl9@zw-c<>(>}w#-x&LDo$Jsv-BEBW9FJdTBXISu55ry%z2d+ZyFq#a~F7bJ8YPSrg< zH#A>=ShV|0meIjewP`TO>8INS&h?o3A|sR%ON>K655(6-Y5GM7GmF5sK%`4Vve?Z#ClC2l(yY&b3=)Fe#D4!^3mDj&2U91we4KCk!Xw)o2< zrd+~fp~KJEY38I0NqA8a4nbtfjcdQAxVx?-7!M6rdIj-hDvj!9Zda@x==olF6v( z3{4X}mwRTAT(f@A9D{yoMaFxSR=KA)^|x+t81(9*N(PU~@Pj40w zh%z)5e+++)@|kT%3Xtr4-DufDKj9D0FaT~0KVYu;zFo`pzajM?t#xd4DXn|c83&Rz zxjKrJR6QWl-|HQ%Rrv)UtunD)pi>*ax^vXX_u!_EwoDYG)lO(eaWBZVu4n$XUdFjT zD2_9y$C7uHOseZ|S{=D3q|y%9kd78yK91s0Igy`({?MUOooCiYA;V_HgeU4osQ`d@ zRw2S=)ym$9U$n~=H0Y&%N$6?Ik#>t_;`R8| zAvVo;;51C-q*`OX20~&F5|Fnrs5UjN@LDdgU0b#4$#%{^(&QY2ScKGRZh2C~$F+J~ z-wcR$EI=qjt;A$nu1%wEYFf!@s{x9MpY8FbImyg|WgWzvxAI2FR`n9g`TLDVlru)v zjvYztLdxcI!ACPHNpS|{6Y&39tD^Xlskx!bc3=s%!%6bL(Dz>{eN2Nk8YDU_45(Px zPNpdOE_K*`YI1NXl7if6IbnCuaJ_n{@iyQ!Z>Xdb8Xo0=`(CUGgSnkKD3B$79h-)Zoalq9(P;!^xAt z6KXbZtCrWaE5)1us#)+Qk$);0+}nX;vBj*z0%luU{u0@C)SonYyue9RX3JmF^`a3} z@tvbuyj*9p+>t}=)Jmm{_HvUPF?lT8^$Rj-{A?)N1!Bk^Gtp^c>Wbyv;E zS|dQp(Inf3Lqcw#lVH28F@@TB4)4qf^P#W;t-+>=n?H?pWWtGBoP+Qn_VK9R%9_)Kc zppa=4LqU>iQ5gsJmkRa!`=XTQ5=j4om=HWmt5o^-bUo#P$AwkA<+W#e{r(a}K?!n1 zUtj7TgxdC-yZ&r?H^6A;yINlPulW4VV9_p#Lvo%DX~v6R2^4!yMI^$oGOawlXx}R3 zuXx7)YDde0+(xF#gWn?~SCwkGP1<0t9Lv!+%eAUgoKbJ8G6!wW@87wPeZ_6=X`6x9Nn5gRsi7Q`T00>~09nfl_qaDM-e$dsS{LG){lSw0CDp`UK4WL?b zsaz$H?3JT31Oi`0`(I2TM_n5zCJI7fWM|_8{FEJksCM(Jw-yRH|8pIrpJ8*6xnp{J z{ctB5A$e-MdM*>6q#{@492(=>(a*`978D{9n&%jMokBVEhA77)UB|yk8)iRH&{h^L zrGGW-CM2Hxqe`~p=OPm#K@dml^+xF}8apM^h!@%B0XBc~o`Oh}tNqALdn_6>lhql= zp88mjT6&AOGp832?7Ppj`dFA7!Axgs2zWtHErUj`A=Ub;4xIKIGku>z)tY~yfbu+o zXdSyrlzx6KuKdl89AU=MMLYVUk(VzX&!fYh>6dR!s%W(EpB}{lXtfcCX9jguMZ%`_ z%wP{X;0HWYo6MDi&^@g8sbTnWe{{w7ldK))8_!2~NZP_+l4k2}3QDu)`2Lwoi-)n; zLC*5cRF?5Q$V9`jReI$fPt{9AgaV$%t_S9zPX#@8ww;)Ns*U+sbe*VpAbbR}CHaV% zasvFe#oGHZI^|DwS$SFRR*^Wb+^e?P9p zPJC$kY&^KG&o{ICcdwCBjj9=BgGts!Z-;iueZ5h{o-gsicN^%^O`#}E&3T=dPM{Lz zv*Xz)o*=!4OA158X)L=SK*TFpc1kKAw#YL!?UBeDp7fyYSfv% zi^LNmV_rQ+^jHI1)>kMlb4_IFn_Xa)chb`zGl^pNB1z~nJT9Vz+oPXON(E^!Y*-cj zZ}CS9OIf8T9l0QwX0ceL;iKoShIqH4GnymdlDt^UsS5phC2{z{jCTG&S!e=Jl`qbT z!yp!j^^FJ9hnOL%iY??Ss$!E_Dj3R8gYxm+_n)9knj_!-jx~imEtMiT>)%xsqO|4j zrrmwc+gL48A7w1)0Tfli1uZZp!xq%kcD|=w5aKZiMar;;2jdzD=-~fSNnXF|!js8O7(q^zfVzk5_2Nn0hI&K$H83s^qa5I3 zU>--flGFXQ3#{&+%z202Gsqn=aZ`9V46u)ucKL_eFS^g=ljsvlR@`G{^NeY ztFazhoz}kzZo?g{1pQ>M*OhrB2L8UJ5MAauo$?3ryDBK3($7IDUwj+`G}8CH5f8n0 z!pyb2Lc5`w6}mWT`7{63(GTu0uf5jFYfYOxZBJLVLT%=Gy+n3BAILi97CBjV z^nDc3kO&BG?K&d=9_po6Nw$ZJd482MbzRzg=Z)D(b_LKji ziK85gS;(q5_8&n{RhthRPdZ)ChT<-53-3SrmYsT~1Z26p+x-f8Ur~@>&7SYwa;P&m z*Sl0Ra&VFg3mAo;MhobNHj)pb+NN##mFcAI_frZ3&ZS%aOV~MvILBcv(ev8K0PZ0h z{VcEIvc>B-pK{`J>j;`NK}k0`yZ8&)CB&!O*OgDEhm8P={;ERf^5gtL1*y^lb|o|^ zsmMY>sQ7oE;@0+@>&?w+<*O^sao~@_eT_NH5KQUj;DWj}4{T>jYFnt>*GbWXLSg2o zt&u_aDd#bePx0~W2&HVo;cVXqlx?d*ejl1&Q7u$h*ibKHQ4OotdQ=XQdRlxRtwemH7BbMoloWz!~H|F-^ZqQaL3>& zTmj>34(3({ak z*?!uCo8jsC83y2;<%He1=4xR+eYuq5a3uI3-PAXx&Eaai6ej|^6Wf5dX-6{u=3$^d z)l~UQ6&Sx4j%d=i@ny3?-UPBGIH0*i86KX0qMvn*H&>2jOlMr)CZMs6PWlEsz^Sg5 zS5II3T>ehSIq33c!QrVPQpki2vQS`+6)ThuroPzEZVtqfq(5ta1T{&xSN{wSZMrHT zmk1OBb&_dVObM~9{vABxdy|QU_jl{rEg?n;cLp--JY;2xUYII-K94D@UME(k1+7i#^@3)RC$(5+zX;XZVE&J(ZnYoe>OzUP zrW1fifSG%dhj90)MH*8+7{QbCl_}=|$1$=}m6{F5FX|q`G4h?qL4c3b2}ukRO?o5d zQUM9^Dqyxhkg&P`YhGmkOVFMrulI`Obrbp7HvX>wk;&(@(Aih!0jFJTCW>wt+t4E3b~ZqWy8|gc&4OSH*;BGgmTRbSS$F1TUb`NB-gcA|6SU~qB-OV0?O~QW~c)L7i`08A|quon}ex7muWA69Ak*!eEgTELqP3< zww7Ezmc5#)1-3I;_d4rGU5SjgbxtPir{dRbRIm@#J$*X{_ritd$8O=+Q3tOp{xQ(F zpiKApfFN+7{Zj3txc-j=u72Ll>;juh_BfL!FkNeS)38uNHO-MX{aRo>vuph9d@Jgo zr<;Do=tBG9z5jrauTlg2%z2;~y%b-EV-C!2m{F#BD8b7W_zKJG5TzTELb70q4Olf1vA4?CU;*#I*X7t^CFpu;sEjA-t$| z&LZyWhCUczfyuoam2Rtu=aSNOVeyLYw zu#Yq%0-w*oSMi&H)aFyqRo5Gc=kquL!Zx2p`ytE@)25xx@O$gqC0^RTyqW*_ zy>kEOcv9#UaR@1vE}J@#vo`qniZsldt2F&O&a$Sihhxy9Vjj6q7zbnlU45IRss0(G zDn+n)6Q#|j(WwIOf3M!>A?%tr@mOx&ZmZ6+P!uk=dR+y|q)WY(8(}XWhnZy|&k3`v zAHCIYYD&J(6&w^lVpCE%Fl>FimC>z}oSHd+%!lkoNuNMbVSEDN0FYn`2EnAC@OrKf zybLjC#|CH32b9pgB(z;m;qjf>Wvk|O!#$m1zCZjjf9*a8eC{isOKlq@Hd>}9euIUS zR7(d<1m=`Y#L+8uU7IS7C?!R@$?@H6`2|Vs{pmq4CedBSD$DVhi$cXLp$#|6fgS=w z`EE!|!1OzdI}Dhcajsv8plHC1hr|@KJuvHfNXiMGNcv4qDuBW(jgJEkfc4TqL| zt3l2SorG6Ue;aDYwktj0-ySXLPaN(~ew2r1PI{n;I}D(p$TD+11xn*@@+xtmO3@a9 z9lGI$$3hN?#{Ol6VViyyR;VRMG)^kkBNsVR?iI~^@qdeh2Zzg3pYYm7=%AOHVL?rS z9@+>M$w1$vk(!^F-jjul4zc=^TAl{41o4Q!oFtF^))%AXXPaaqkIYI%;XhbC(P9sl z^dcRs*aHxt&>o7>i%^YN7|_ZFV&&3^D&`AgdQ`(Y|B!iKF^rP{btFK#j$?s>%?t3b z9PpaPP{b9rXo`FS)k!h6N3%dy+@a|AP5Te}~ zCm~w%`jx8vp+8tf9V9W9vBm_kEojiOR1ABR+eH$8>cwH(f}pz zzcqj!fqe_Xd?>*fY_LNAl(1l&cP)wKO$uY|W`~RIp+qTKI8zpsA^Q6w#J)=^c5KD3 zT$CBpQDtfpJ@eOCF7ZQz`%#aZUM5QK10cEg17<_B_?`ad=U^a}WvNLIdR;*UJYpY2 zF&-?2dl)p>v>zR82e%};V!bTviX<_;B*Zu`hADFamIxCBk?{P4qc0T0j=6{v6(%G| zW)WH+pV5i~D5=I#zBjq>Gx@ZT8?sp$(aVXPZ7Pd6Sij)7v8eOSGNLf)wiKURA;2xB z?_iX|Y3q?qRE7*@Un#8M4kO}zWCut*$8Jim$ij^WPT@1P$ z`KiP(u$O$Uizk8C899T(j~tRw4*)NvM~^>O9Ej7qV7avAqw>ohBk?Ecu?Hp63os4c z-}S~1k)qr=M#e&^2;;`KDm*#0EN%XMfz^x?4to@2)-2SRVy_Hdj$U3k zc!El>4>Trf3Vl2k8k20}0bskt&Z2%y5F~Mx2IYe7L~eyekTsK4c1|@>A zPdBwZF`gM{hn^1v61M z&}8Ef`Y-F>W));dCqO`2DOHC8I*2?0-&VpMAEg=^-{b-W1HP=$-l?P4SadU6cnUWd zJ}%Vo8Pf$PyHmrJJ;eI9ygYSSI02 zEWy?pDdfL2Ip0MU&I@{t|B7XVi|{3229d;=L|O%iiVbv)8l$xGqxK=cR3ye_H;P*C zPH zLH4{Ke&+;1`(_gRz8JM|xN&pKiMEI`C-qGc>d>Pa9brSwpG4Ln=5sYEHTe8C z6hj_NOEMpMJ3``rTGu~m)S~4?SBx()9KWXDR(FkDa7lbP@jEbqM{~~%+A}u%_knR8 z8h7RfduqbniU>8n_zk8#I*g88qxcs{@(;_P21TIw<1vz@wdSE8g?-0``D`#DYIV>4LKvRJXuJ6_VdpD|0yj;x!0oJnr9pNKJ%PcEIgQMN zQkf_lCdkDg#Uq4USkB36D0R?%?+M}Q$gJNkFO^HH0{FgRx>Q)Lz)jYS0{XiarURoB zvgE~yeK1RIJofhr*pF{pm}L6NW#4f0Fj+}cd%jrtCa0Xs(HvUHPR*wgX5EG^F`{@yRyn4SJhp zcnf9$fIf$rcklXNj&AQdg+{SomViCkLn9J0g*FH@o) zJV`^ff&3~6#1q1tZO`Ee)gj$+i#&&qBLPXkY1?YSwxg^0OvrG`JkF~(fD`#KNL?iKC=b{ zhEga2T1qHVDy$KFPFI50qC%-uV#uxbavtZXvu3Q09MN0X17a8FtuA4H*jD0o_Lho2 z3z3OPHKsSd;Zj7`AUovhOGW#{Ku16)|8Ahe5%r6?id|-bXJ8 zD_LIeL)Mk12uEv%8;I7Mll_BDQ$iSxJYpn@3@!Q?MjdkGZ!~}wIVjz;>wqIk)Cwa` zuaH}Driw$VadN(+YN8pe5^`!O{7@vRki)VP1C42py$ve5Mk?;)S3CHOG7AepMQ66r zkv^>@9gt5d5fz{bA0pSx%Am~iWT(tRCay^AVGQ#Q2@($coqsN%cNW_#A{E>a2ah1t z5P*BQ%NusefqjT4H)fpfi-jE`IKb#SwgjSTr3cflb8$k#;5CICG<^@ijl;r~$YWH6Zn3J;?&sYr!^8)8JhC`hIOZdc*czmh#pvBmo) z;+X;wBnL$vh+((P`}}Qaq?kB^I=jspO1LAZ!`su&3%AnH45QD8fTA zw39=p*#8&fDjd}%gPUYsf{wRK?3-|Ch}=C91zr>=Kaifn2#1c8I_T)MXd8)Af4g`U z>78-RcMv73K*t6#NB%>M&vZjo+6#p+A0jFh#&{75X)*F040l8@czPt$$N|t&5qP5? z#faUt`$h{UHrmwn<4m^=J$;rcU+=U)ROO=WAstAGV6o?f34A0{%`i+QMkt*)dAgD@ zF{uDpnnuXZ74rk%MNLB;Jo;`Bf9v@}qGl{{^g{34x5;B2Ef(36=lF>n#gLHLxq#Fu{`>>-ufHGk;_9O7#fFO zv4YFd0iWjvj?S^2EXKGQe-OO}GG8BwYl4_I4U+`%i*z%cVIY%2^h^XxF~lW2(EGln z69fUm4zy$Cq*A!V>6ED3FN%e=|p9sGf9?Vn-pe zT92k#;vst>tLZs10ooM%Wmg*5h6sPpJ=D#D-Z6?!x%#Y0}|%} z9Df|6l8%cI($)KPIr6BG(Hd9g zxJ1N_Ug-D;8W2v>f5L-yy1)w*5*6-Y+*2+p2ZBiN^%BI{P&nB@30EK82@ypiA=a~d z+Ihfj2E?>FL3KRgCQhV~Mx%l$d{7`AeJa9rh#MJ_Sg(VjlOxQ7|A6ao-7FBPj8Nf- zn4@DfN<5y$c&jeFp+ejT!z;W;bccl61DM_v=kP=y%0Sd1e;ptPXaL(iNFoq zNX9-wEZ0TA9eNm9Boz!!$MZ1W3D*@tmN5+RzCJYKz||7>x+8aUg2eHnNO~c%e9>+S zMXbS{Ohhaue}Y9}lsW;!K!-s+(tGxpLaxVPJweRT6B+B(7er~wlPq%sK!ypj69sFE zglIE_hi)KQ1TTif(DyoBeZ4d(Lb7I=4|m}}I*Q>LV|`#I4J7=dBd8mi+cQK8dgxad zav3-(z;FN7+Rtf z*&;{8?S>KqUO)vC+T}+`IuYdQoe1!7A@>yLEF!K}rw;r^$O8{ZTm@;uf+z_QBip+_NhJ{(`IUdHk05uC_s`Qkm@>xiXf>Md5@54f2gx86IL~qiKV6e~y0x@t& z6|%ZCL!V0ZIX2fpeBt2zA}Wf|0%>8K&SE|5tB_pl<7`gkXNooku#9C0cNEtfkwr=S7{Y{D%^<}v z6eS4N^>lQAXahmAi%}|J2yJOzL+?IAWDp=6c9mZELyJfie2Axf9r4k?5L2LYMlwJi zt1pd+kl3Ml=v}%}5U(oWu|5~BDXpwtiq-qPXd}8lcdih|A{dB7+ zKbDhZ$e2>@qCa8u6gow8f6-7vMr;~chB%Rh+7Tkzl;{a~0eCMCvVJF~rPabo-Bd&z z=3xrGR4?B7QtA0y1cD*0oe9_XT|{>Y7-G$42C-L=qf4--82zW?a_5KLy zaoq#9D$_@ErA)%s z-69PLh{>x+5HunMfU`~YMGCqrAPoQ#`c_8ForHkXi8!ydd-3c zrti`~{va~F3n=h2y^IojT3Mt&F%?WtLgyz#kY&TW(ZE!omlTjS zc`QSIlfvU7;dEW3c`h}16%{Xx`NG47e>kgU-*F%H=zsnMM+}L*H^d+<3zV?70r#1RpQN9zGniUg(( zMZ42I4H08=}FF+waLr9Rx_Kdc^ zoQk_qrqACiG~kQ~rc))@U-3w1Xdu&*N)Z=IrCEOeSMet#HAvIHtNbCOA}w|Tfq{vR^$_ro`TN=ar+be zHGG=SfBhs83PEw`Aqzuj<|c89!QCZ-$uJWb03z+Kk>O$Fgm2?94;Y3!+PJBZ4plRJaVI4%6)7YR7|YQ{0}*6CGL~3x56!&9!xQ4ZAQd@WpFQzYo{G@n zdg}mUw2WnutItmn!4iGJ6`BWUY3PQI>mk0?e*qd=SrCOpuuPxgML^&q8`Ik_rAT!y z>*%ZP1hLT~Zbj8FQX(e_+DlV}D}ChELFO=DQo-O1GpTe#ivxY#VagmN!3biQPh1g0 zP~hmn_D5cG~C?rZ!=o?572f>XcNtkiG;XtpiF=CF6$c)P@ ze{ceQ7%O6q$IxPG8HV*R zOPnMof=PIb41pB)(;)ED%+=d_gdfF8e?mP~lsT~9Q~+ zf(hBaJY*31h7E*6Og9n?S-$HK2jR?-9+W{aifkQYxxRiP@$p_+CbNjRZ(H9}gWjXp z`Cxk%1~HsCI#noAC6F~z!4izQv7Ru_kO!j?r%vW8v=BWP=@Y?_HpTFU`e2>Tf7O8q zlLZ3#3cMvPN6+*%V*nHhcOnTN5@sPrm>BsYc!;qE!ffF<`i>=ijE=F$*V`KdUtiYb zM8wsd%y*DSBaFouf-$ANVOWaLQ%t%nA#MOfN`Zk+AO4XpB6fWZ5zL>taqOwg35Z~! zlOoULh+q>W6=-vrr_ZAM7)ia9e|tnQ0K|IZYUE2jS>k*8W*?b3o)<`B!k>k)k4P^H z^rRFIF@mBqq!>XHKXnB$KGUn4vEIcTrH;Nu5W}#eoB;V)lA$dV`=;J?kR(Wi@d_N~ zD1G0I@L~)w7}TjYi!eC(BG%)Q(g(kl15c(T)At53;=BetgC_(6WTsC6f5mRBujUkh zPxS#^;`~__>tm4~;sb_$5XzXlaJOi?jzqLUT$0T6tuj8M3?F?dMGlPk7-=HHw9rox zXc56u*VhZ_Jk{q_^;HK+%n&soxsKzAs#)eKj7WN;2I3(YJ@#cB=pDq7LSvzU>8;Fo zrH-Bwl^8DFfay!5Lo}FFe-ST-%+R#j9axBEBZLVVoNbuOzzy`ygOu248AduLF+BPD zs2|3?M88}QjW0%M! zG4u)I`OQqhfrW_^=o6dBoiGxo7?<=C0WwbFqI{+@y}uMOwm#n*IT$O0*bxHR4wsU+ z^quKjsRF$afPl&YnuPb!gB(koIGCLDz1+Wl{P@Au)%*G@810|>a|4F^|4+|OXn#MI zZ9iF4f75;$()783UZVKJ^~WCnj(_KD+kIK(`a?5kl`fx|F?>{Jt3S(74-s`wPgLzE z9BPRC_W7N9Wm{9V^uvvQcxXRo(TVCGpIV6@wCwli$KP!8f^DCT^xdN~clSYyKW?8D zuh=(}KT$t_UOefty|16_N@@5!#QyWs>A|s?mhA@%yAMU|+CMhk*S*XYKN7bEpSps- zyr>h*pY;#l?DqMl-Yb!3xv0&J1g3h?U%Bsn8U?aX7ug?w`un?|Pk((=6r)Q0mv{eE zhYF4CpQ`y-R`0dI<;%)LFY;WV11XB9i^%1T7Khz`{3bIan+3_AS)-oI>NAb(BxNE!JsT%yytbx9 z**hYC{4xQ5OtY-urM!oiauR;GqY&(NeTv;~g5-8on<{%xfRS!WM8Ksh3P&r@Cu)D5 zU@PpK7|Pyr6jRNqV#PC|DsAz5lJ>_b9wACX%WaGIc=V4pJTF9|7bDZlBoziy3KNR= ziSVNh&kK?0#mF>WrG}BJgyMZ7{Ak1D5s5s1Ba>%C^+Ke2F;Y!h?^%C()}HD4^by;4 zcRx<^kQ$yYZs3`2c+Y!QHFcB8Aiv`?^cA9 zc~cdCWKD{vuZ7Ia^>lQ7EZI<58#AL#_s2hrUx|EL#K@>0*m2fm%6fp&s*4K!GqnMC zLT1|bku5X10e3=!%ytbJCHGY+>C{?kh+99iB*zmvH=Ybpoj1^?VADtjknG0U9lnYD^$R118%L=D{`Q#_d3h+Pc5h|O#QKR9ZT+F5W@W96nN~lsp`BysVKZAB zxhA`=dcbwncCKqih%*@8*oHRsew$MvUui32X4OxuXz2}UupL=9qYnpGkqz7m4e;tm zmdr~1Hj^t=8-MGMY|lQC2i=pHnfmz9hQBGdc};E3osf~cdt}92v7+9SRw8fZ%>1XP zHY&WbibxI#BQ1WnBFxoCDkDV9lz3plxl!oq#Au_FF|l?}CTQ!8-s&P1q3@nW?n%!N zrZ_(rVsqZAWyg6E>2aPOOmH?^#p-6`66$#tx#v7Tuz%z9=$zgVr)SCO(>eViPJe=v z=gr5wRvShL6|ZFOSg)Sg(dK%0Q$uF%#0JQ_hj#pBMdmK60p7B*%H2NS5-Qb|zvt%5$Mh`J<9WwN!nj2e82XIew}7GJQJ(cNf@}=-zN#1{P|EGY6{~~q)&N# zFvY$Ftz^X*|A($A7or**vR4RfLf4TFAh8dStNSn~V5}N|!=3y(q5KAUmPaq?5fCIG-`4 z>T*7S`fl(sL5q6BURl1?ZEl`6R{&#LT@|y}u2SN4mR4GDL>v8>rp0EPzgM$WX{9Ih zY0KCvRV%_g~?j%WB zHS`SWPOeXU{nSc@rrgM~p(@ZxnSbfKXEx-7sO}_Sq@0S})MtXP`rLC+pC`Oh#_HzV z_J)m3QC9Ze|4#0X_U@^L3Om)X?dH9z3bmyvaV=?5~mq%m>VwTiUZC8?+j*Xk};8 zif+?t$f%X2RV%t#t0B8qc7`R}b-=84R;3$t$flMirCW5!pf>i*i_CsILtFZuNdGoL zLOe`nK1ODBfGpdVg#9sX>3<$gIkhLIEwUk|{MSMz%8gC>pB)G%;C119ro+^^_ih4_ zveV}*amRK3-HPx7o$l5=$#t<0^_>CpCN{C>|4}H;Ewn7^90+Du$HC^-?Qwd*3$CbyxERB zxgIZJwj(bJLL+WV-_z;eCP-@9-DAh{_e8bD&0~3;>BSieS0~UT>wcbML$pGmLg^k4 zew!jG2!~+X(mj%5>VN1fil}E=WjcXARr~W4+rA{4t`q37bw5wB)dx~?= z=lu26ehwIA?^y85IQ;vH;4XCGI|BXFIKx(D&8B|I3AxclvlPX&se`Ls-2$}o6;1K=sv6U92nf@yoGxzi?|FPhH z#w*?rzT>^^Oa4GM_J;v0f3P!eP1+lLEo7#xr}qe7mqgBzv*P)%Wz{StuH;AbxhiF+ z+N&gM^C|}vc$HS1P;t(HauhRjA0JNg*Lx;Nji43ibGAQEu$6=<1zq@_rJR~lf1upi z)W5B1@}4Hz4Na!InJjK)qTR=2x{1l+4kps=ONM)wrf*zA2__|FBga{y&(-}tNtg?^ zC7pb|lIfG4p4ibAgsl!&G6S^5)Ha7xLCOey78xL&9F@t}XANDPWqD=Rjy1c3eV)-3 zzoY7YoZ|UJAoHPqN2<_uNL2S^JRtqDEEh_a1lU>xNZ&O<~_0DylL!vPNxgO4zAW z?q_Rr5gkzGQeZ+C!nvycp{mIfRVYB~$zAzAjYUtUUTHUQnVsD=c02v0f1&->ern|; zaL;!9d7Q0ah2G0o5G;8SxnsOMnBY{;Os@jkGad!*`HnwNv3;Z-(7K597(ZGa%tFjS zoR~XH+%ukjA15pe!Z9#i_>M;ZG|fOe({PqBa9NyPC=Zks$+q-8(P45ma!*Ku_Ps1< zKa~X0Ki)9(ealrkACHp`%L?yW=*aNSn2Y7*l|Txqmwdg&^@yve}At&Q<))Rro{s* z&Mn(-DU)8eVrK5+Lo5C^NP1;eju+MLJ*PJU1B6*GLqRJY=czTO34wgCs@;;$9qI>0L#GqZkTM@zaQ_F8Cwc>0pqW67pEv^Oou znO);v-mK4kDJ#_ue??B(zI7{RR^<3_f?rYWp!$_olQpE@uRVCPw(XtTmD(}=b5M5nXKN|8)6oP?+&STb@Glhn zmp;M7wz+ys>Y-ahr`%5WLtSKZ*h|W-sMkU!%HyL6<}zz2>&M6P_k7*2Q&fZ)!SG?+ z>969_vOZI~XOZWTLLEb-V-e`kKKfz&+=tDFKB%Age=vXC!|Jmhb;HDadm>%>o<{#R zMM7|zz3+ugekr6~#?|M-az&U>xsmEoUOk&4r;X0W5!q%GKJcbuf>6Jsz&+RT=P5Qq zm3nKZlb1n}Ke96Lc@?`4tLV>1yi{dDSp&I}x#zq*8K-R+VSP~-z9Z5H@CK}~O4P&2hF2nAj!_%D5ei7=u|#3&aEoh9zMPQOnOQtlyZ`nQiYlXs56 zdD**kUHF~>KTR-Hly#1_^gTy4xvD|`9?{U3h%CQB)KocvD({ixlki&#{lfC{QEfJE88C0@5F3Ke+S2J#qOwYA5QUC1k$R#R z%Tv?eoO=D$ep@Zd-sAr|v6c39#|Zc+d{2O%#u+}oC_>ELmcHX?e;X(HO;vf^mI%G%I=e^;x~koJ5NxC?kIXf ze>a?#PNLj5`3|a8b^DQ;wy^b0a)ZJbO$Lh$YIHz=P{WlgwITTGzSKpB{xchLOUH04 zVI&=M5x2!QAM@tL`i|Kl*&!t^C2Z8Glswo4Nj)K>0^K7UX3`*Sk~8v;57=D_yAfM9 zeQ@}??44Wt`6d%?zt&|y>%OgsVpzFaYFLcF?wxmeu?EB4Zp?{`f=wr`(7(T>NJ2X@8j_d42w=EVstbAx(+glAV6FBlPHm zo{`YAB=qTozLC(MBK#9;{=e3)f2Ox>9EjfYD*|+{DX=z>d(mdo0&$wg33@bCTeY>( zS4FvP_SY{-i4s4C{Spc^p@~m8*6^3U`xvALp<6pgxA839k_hbh@$iq7@Q+sb$7pzQbP?@=HbG1(WsMMi%(}z& z)D`s$o^i$)l(QMZ`2@cve~2I)z9;Q%K>*8s6^uwHAr7CkMhHLEtk3FwPP>dc%AnN~ zBFuN&rwyJ_cpFT@5MvN$1<*1!K};@XjS=SA%j;uKxv3|JNvNzb!eP9FXOsRWl{Ce8 zn&Z~HIw^`V`6e@h)4NC|ig7s2l`0U>09d|~~()OpKJaluj#%LC*;E*b1cj(Xq>3J1GmJoi?7g;R zI2@zzvV#6Vktm2+CJ>nA;fM?BCKSZ0%km@0B+XM;K(Czmf1JhWeXbK9H3|qxP-#dB zzE$lQ7)Wm{YsW-MZca_OUktWE6<@P|%(KJ0z%giOtSQGyj%azHjZYt(vFQV1T>3@r zrjamb=VA?dHyNviI#{-VunN@h`Ej$QO+$tr3?rlVnmgXwbI05G-0?R5aTa$xWpp)TOt#4!<19$I zD(qYp@mx`-nV>l)+yv9MsLzbMW#vP|>_mASXT!=*hP(Z*D?eGZ!hmK? zB(*syl*dHQqfL6GvhK3B?V8`IC@m4MEb+?N7lv?MPT~wm zY_BW`Hza6wnv@>Z!3ls}8NI*L1IIxw{|JB`e-WXDAa0yeEI!zEi8tzuHe}3ZypHC$ zW7y$f5N;}&?KCOaL0Ay(N(m0jpjeFM?CrOr7RmDDvmJlBc1`=5Ce0hplIr!NO5cl< z8*_dQs-ny~-YwXb@Nsa<0z&d(PjUwSP@zXldr>hTvom8wR`2SvTw20TEJ&^6L{)q$ zf9hktRO%cA(m1D*#ht(9-C}W55K!EdN)`7~ic{a0s?(xC^0Z{CzVh#-+QtZwt}%(M zi8sEyJvc2^rpE#G=?+wtJn>l6yDqC=c-3916^I5Y2HeQH+2T7{g#Z<@w7d8Zw1uH2 zFtz1lRnzpmf-JsG%s`Mt(}OCs?~1m}fBU5}^)Qe~JsmBvO6)gRc9*%L^p@Myj&ket zky}#tIF;9-C9__q(q22f*Jxsvm|bRPmr=62VEh4%J*eXjRwQP+urpmyGMzrsfkrvh z5e~z?g7QLLdhg39I0o*ulppH|2)v+zqohxYF!-evi@mf0BB5R=!8nuK|7yuaf2YpD zIMXlI%*zCTgHOrhm*@6p?c7Do?84w-l3n0F@n1z&$}KfCXequR@h}f*;|J{Ww+@sK zL=SKrHb|g&AVZ>u6-sb`A#MCxUH(?UriD1Ub#VUN=3Qnm%^Qh;xM`P>5w8Bb@ZdHk z?+)3AyzH|e2Py)%1&T_ftMZ;Ye@+eI(`6n%j*sfut%erBCraPC1|P9nq^I(UsS=cY zT_)wLSIgG~G2j!Ig!jsspT5X=ckmiT0vgnEr>##!#+urS@e7#HUQ2)$^XN>Ze5OJK|Z)*X+ znWE-tM!&Jz25vAnl>a`wyi-X=`8{+EA2w7o&3~^u@hSV+IV==(P@(^Oy}X%>W32jI znTrrKv#JR7Pku`YqQ1GjKUWX0^2Yt#pLh%ZkNCE6*miJ0fBrZZ`jzbq$OB{lK z5r)gH^u9e8hy23Vmbk$@?Ma9h;D#qMh#2{`)d;(z0dPyE<0p&S4J delta 27448 zcmZs?V{~TG5;hohY}>Xvw$rg~+r}H)R>!t&+w9o3J-PSJnl-cL`?=3KRi}3Gsa|+HGsW}r0IV7g*b>NIbb5$aJYD&v*HvymQr?KzHq`*?AVV)(kflQ_&FKuOH>z_4mfSS9eox9BiHTSnE_6zW1 z3e|>9Z^kCnG(^=UlszzJc(L+gUh`(~QcZm4mWEo>RrCw}>hJHKuk`)z5I?AtzcRI( zJ(aQvRkDPcOEi5efy8Em1v!o+NHRJDllGisAfG)rPY$jR9h+D zYqR2fKQCEH{j)`_B3?~s)`rB@x?81pdGE~vWNrl8qc zj74E3;P0@C@+4VZ%8ISRw9i92-=It*u z834~y$DUfcV<6461XymBTg#rp?QCE(1q-f|`9IZ#hkKUQ`N#cmrRumdu%jMD7sNa| zjy-lecQZWv!N2>k`z-tqF9mwuGIgyb0Ms zspeWL%pT*to3E5OkD=w(|Eh~HVAKR{EI;=i{k@1=!)-vH6Cm{$gJN}8@I;|$=}yJ~ z4uw|NZJkjvHLLUL1WQ{?*Q8}X65g{B{~PQM{#a$9!TEuX3cJ@N;ZRl zqVsMu7p2C(7VTzjTTQ)K9=~^}jD<$hO5XOt&|!mYIWiA??(3V2@JrH;n_*NiXB zn!Uzg(XM6ImM*C`qXBx>m>zW_qL-4XY~3lGe3nH0kx0v)G&5{wjKV%+7qAt#&X1Q_ z-}lY`edU$_e=$;@18kca&SHz~UCh7K0njS|LKr|>#xfd;w)mP2R_5lOtuuo$UiS8w z&%KPo6+(2E1G=%V(tL6KKU#Y_``<#%Ecqxeg=n9l?u%NtapW~++S=%B75@J8?8t>D zBVY^bePu&C%|lz|36yS5gx%L%;J)~TWa-Ag?xsGE)>_nZDfvr4&RF8r$jqW?0k`}W zxM$^^f2DL%axu+%y)I(I3$Uh19yKy-+8vYNG-DIG(veoG=W9aEI)+oTVPZW_l&}Iu z3PCE0{4@7nNJq%NC(`YE)@T71Zv0 zV~QAakT0PD_tBKil`T~+)|?*bfC(oZyeCVY7`8I?RKuAd%JE|tNmVwbmP?t|Hwd6A2i>UwNy$RDNz*5lCkhBMviEddZCY5WMVdR6eQrg*g0tvPA~jo|kkIhXI+^$SzZBRMtS(Wa^zU@m7LV&$(-Prkdcia1Y^ z9nc3W49hJTo$N=$#4uZqpuCWCkzUKJ45-Knx@V0KC-KMx*#LUy4LuG^38o|0NC zma|J%S(n-r8aLCF#O9#yP0NKGg)1em4F;@C2i=}pR9oo^&{}lo*nHXl^_6MY4j4_q zNXY*JjcQE%W@+=~uB~tD;W9o!T`qFxm%(WR|0MpgzeXSG;oX8LWo8SR^>KnZMm??I zdvmgKmd*BysIPiXmHrG{criV@t4@cn>=^6i%@#vc3Hs0i#WtOFYSy@J)k`!Xa;1d0 zn8aRn#8nj_M7AR{6Sf9j+_x*I2N%+%JpsKX$K@B*E;SaH7J_CKNLRx+azSG8 z))Q-9^Cq6gOLk_V^;k6d9yP&V)5{>L6QC1JW(Sh>hb#%Rpj{2gnmJ7Tc35oJa zwia+@=s_Q?EuJ|9&-;Xh&zx{SMlvc|ec=1U?wkr3S9#4Gs;Z!iJg-IFuPvNVC8!=O zuwF=hl^FelUWRa*kjts!x&BtEnEu!5xA|GXzK-p20=N{nKojl`g~j1L-`F10Xb5LyrAq8cFksT-vfZA!@o%7G2J! zu(l{{43&F?(j^dM?71e+G6s> zr4Q!o#enxknUDCTMl_V}c*gKXJ8yrTJRX$OVnvyJDSNH60`$j?xW49arJcgUjP3~C zmu*W+m!u7bzE=L0E5WMNOz`!U4j}#evy6rK5EQbaJl+S((pqmZWc7SSrSxka^$le4 zW$LOUX}aiTHe{;e-H*0uJZeD|gZ$0*V5Z3Ae(0RTZ07!W&N*cLMZ0_%fUDBu{VV)^ zi9a^wIPpfWJ)fZ4`4_)$JoMi6Fm*>vtSB9MUo06Fj0HN*>BRPXwBjGNy!4yl@N_y(;CwgZVH4d<6(2t@>#DF2{b{MxzJ_@j`m!Qj+2Qn+bT&Qvp6!FtcsvgcM-!&D zhGy_D1FhNqj^;y3FY2(~HGpQXrHB4d%QMoWfU^UwCjzwyTa;9Ihv^>slB-FCwaEt? zkL#DbtTA+MmtdQjlXD4K?x8&(Yi-ISR_aZAQ$K-kJqIYh3XuYl?JqnOxoq2{p@x zy|VEQGSo{YJ4CjfX&*S{LPmY@AFat!+!n<6dfv1r0+@3XI-q@tNCqt8cn{1faoUEj z_rCxrRv6FqNw=M*6MN@#`q+1Esb!Vzv#Xr}?>&RfGoFKQRr3_&zaVRJHp?vaf9P@| z|1OPK?-?m@1FEGIo6V0#SVW(PUgRl@-wNphX2Qggw z94(#5MFnLnw6-1BW$B~6WJB@MXhh;pm|x{=qA03kqh0_c_F-A_3)MrP)bcJS=osi! z&|6K5UYh?Mb``a`YpN!tiY^K?vkIvA-CLoy<^(MadRBLF5TRw=W19 zmX%b!iPCyWJ*PopCfLS%yo}bQXG}W4+6JTvzRw_a%v0$Qr!ARVgH2V>S?bMD4Z}`5 zbUYB}V_;FXoc( ztLz6*Z4+|ypzS`$k~1L>&OltNiSEu|6{7pp^Xb+h z7m!pEc?RJ0=L2$FKF6SzSnGh@2ls#KiwozP0>vrC$un3rXM?zKm?E#rq0<<#9^*2N zT)YJUTC9#AZ3`_$V2z@!8P7J=Ma`qVI)(0QC(#OiUs81)>GM{uB<(i*qDiAu@3W4F z{vtq^NlZ2ZQiZfe^}lH*kFPh0qohvW7g!?}_+FYuxNIh&iwiTb_F%*8xcuSvW%hq_ zd(HVja07gG=-h9`szikuRK~##`!*6cn)2%Ad8i*n%w;?#={nWoFI$b@tOcM=qF3rS zOZ>{A756hNH2e`a*qYIGNct`^7s`*rI7-tHKV^-0kjO}=hEpBk+{D+Y0ccWSEdYHS zy;1&$?%Q~-dFb2_D1eqqIxA>}yZVEckO{{CkjuQLm^_AGE&Zf!@|F2FXN!&GGbqg9 zf`AQk-E=VRJNn3UgoYJiLYpZg^Y6SB{PKXO3>KeOZNPs-N-4`#SywKAL^YBlGMOOM zn7+0fDRwR+Ji;&mtxqm+A#z>4{h#*hCS)!C|Jv`-qGxk{U66`@E_e4I~TIr zL2=1|V3TLp1kG8rhF{&!kkRDVs*hPqaG69;3`V8~n8Bzr{zm|2v_@Q*>=AG&>aeyy zY=kq8jr4=Xyt*YG`k*c(Y*$oMM}EB3YlV*uWanxBv3NSyKleM$XkvNQ_ma4J(YPZtpRy6tT*+>@@v7Gf+5*?_qN{K>29QE(uB)WiyZ%yS0HF7~A9a~@k zyemN)+dQuZsaX)q(6J=vR!}qIa;#EA;ZhtsU@L=f&NOqb!-hVGb4wNcIbnZV1X6rx0?Qw zNdf;)DoyeVHT&o1cwJlc?;z9bxaQ60+s=|b8OjA`yqeg1uao$?3jS8O~?e(Hl zpjP}lu)Z@;Pkf=acmpX_UMl^VRuTvTAIzf}kqrg-R?uq;C{5;j_=x{{y{w6NL7x&) zkcOxc00jf-y{#JFYd(r9+2HpQfWH!ymTftw!?D0C55DL7n}z1jXL&kt;|khK=FF(9 zUDU++4OT@B$GY!y>h@x#^7|hjv2(zS_aeb|M7wFeU!KjC1s^(II9@Cnn4t;n9~{SRcVVNz;Qpi6X$V# z(&=J4*xElZOR|72V>UgK*!BdY_)F_bgW9JChUM+qMwMHlMFn-CX0g7@uL@Ph!N5_Y zS*y~H+WgN*f>5!l>Jup28@@3k&Us3yU{x9+CHnrxP=;*1j34yjh;zKtu6Ag{vs%Pz zvg#E?=7oaZge}$71XC*yxPrF5dE^6fDLj_vzxe_;^o4TQ# znpf4pwKL(4`VBJrtsGwOKl-N$u;xcX4Luf#w_ZNv zzclZnrzrZUwW;1^u46C#`!nl7^8#qe&@ZB4_{vz%oA!@Jrw`cXP4Y(G6xP6GQhM;~ zqgR+onJ4nzFj_L@V&-aQ>-OHgBc1U^OWEad&9MQ4`n}lolR>rUS*L-Jy3C_aa*o30 zh&g%%mBqi?TWo$=K#TDdRgQ!gs15XEb4@_I+!YM5&g_+)_BTW6kqDqe`ZJYCXA*ij z)8WTI($NX+Ms+gM_Ds=_il07%Px*0}90IS1FSwuLQFU`aG8bqUB7r$+D%14#M@~~J zUn?n1dH>7U&{g2}RT87|9cSemru_B7Z!i9hLAX5*4G@uV6cFjC+C`A3QCWPieSPsG z)Q?QZ9&byHREz7~5wEh(4T<#T&+OIT7|QoQEC2scMxoInba(um-DPF)OEhZRP<^?@ zYcdG5nqN}Uf6F{jEo${3mP35io2NE@Wv=f;mhFc=30n-esO+;&z#8#K9Yqv5Ta z0%ns5>5|q1%S{Ar5@P=AwHLxo0ZJ2nYm0MgLtv1HVyZBVO}ML(o7}W|B-{yd-F-ek# zy^j$83!BPr^jovs7P~8ztyP<)$XHP-x9e$fJ@hI}f08NJ8MZ!9+Vfo5v><2MOQedp z>3oIq+Le{Ywe6!9c{=<9O%$&U3A_7*CF&53_Dyv>a(L&qy4No(#vveh3l$^Hsy2I) zGleUpx>hID2vDk$gJ7W`NY|mh^z)NZMe5$b7zVY`$K7S{tW;s?ZU$+#A<<{ zLh%*wQY4lx_0!xB?hcnsX%ZiJ@v4YZ#bC9a(1vi4Uwaa=dkSYXjsL9|XIP0{$mj?^ z(9WhQ=aK2t!5P8wy{P%eA2rwR?D zaA-x%g8;DTL;{uXSqE^8K;CmM_u?{zXikGxTF#2nwdF!pd>&~43M5ig321KYNX6(p zRbm{4Z0Q3v{dePj|K4W{k(4!|>lU}C{!=^7^J1;z-UReHmGgJTPEC&Z@WE|Cw13ZOLM z-?bVHT%g8yGSaB<-*};rQd!EcPY3+@Y1p+!_#>BcuUYUT(KcAn4^fs_&YK~J0$oH_ zyI+|bNBMftpL~73XlA|;QDZgk;3iA?=iLJgoIoaWMmpQGQ3796M7w@}|P(53(v~YDrk$vebn6bpQQ8J(Fy$ z$~Yd_tK?Zx#GR|z2X2{smTIX-Ws%I54IQiY4cyzu4OrICfGm_nE-VBj8s>%inQ5FU zhtSMRkp3JepL+{m=Lt?d)Q{IYFv0xB9#VrItt3q_&wN2R|*p6RR(oKyh$7H$a{LVqhR zA$^~}N)}aY%4D>R5eIcUpg2;f;2f6V#4Ez~$PtMo2!+xFUAc)TZOuUb*-A35S78O& zJ@(_ml_fLvCFe*;6qS7fdbg8iSfe$eNKlW_E~i?M1LYZmm0x@=C7YD@Qf1eH7Xam%NHDk+Lv7xr)i2>OTFd_Uz&mg&gVj; z8HrKgBjLLNP(lkHCETzWN`mx2#R;Ormh`>;&!?Z@2#3P|F=0$`!-+X1OP5;1;m3mL`-s@9mzTuH_>jY0!) zm&upP5_GJP9@SizkFjbxb6!c`_i?|hId;&iOAoco(mJsDSLp3H?Q*q}BH zI{(?V!D`6HA`vsJI0E;G;q-m7g7Rd+2BE#uDcqRys;njEgEzQegcmYy&@xU^SS}(K zAY?)6M44w47TA^)o57~~iP#iGCG$!vu5s<8AqlE}?N8wtPAa%Z3by})si3D(+!w9H zHds2^RH)Dx9F1g}njuN9=qt5_l3hyz$*c-5M02?~A^-HihdKDB>c^+)$f6e&TjOI4 zw*rd04pZMFw+%J&U2+z1%0$4Tl#2%n05(IbYd5MVZke%qFe-N_J47{25kvR#=`D@_ zR~ZUZKynt*_OhNh8F2Q{WNr`goG+dqC(pv~^Pow;8rJJoNmv07V@f;7aSGvAVfjom zjqwaNRh~xqX0uS6 z2BhOl-e&N&WL$sMV_q`l6WMvlPi?6(^R%ikB-lRBHC!c6ug=6IsJqreeK=EH*;ZYN zTmd**;xM5)YZ`VI;&5&S77oQEz{M<~k!w=2vlP-*Z&AoEuQJA*I5 zMK++!iA6%88`*)-Ai$Gs`Fu&cDz*@8t+O#yPlJl>@{}BVRdN!x-6`Cjdj(c`X-XzZ z!mkq@58SRix;s4yeiI5FWC6T4F$(GmrAp6-*`+wE?C6~Co%CErV~I4>k%9{>#y-G? z{n|uloDS*MAtIVSvkOKzcU(r<Y(Kp;{W!!;QeWAqrnh-; z=m$0tdbm@%zsAz`0PcTDdGvNc>3_{O!ZUg&5}uB5Wut;#l z;IPrbq6Ek;c~oii!4qbTi+-MNbmypl&fG7W&W{Vga20$)E6I}@GIXyfYZxdf4 zZYUg!ab*H`^BZMUK?0rQ$4~w!Ty{?z>!^laHs|JTjl&PzUr{#)EOZL#d8X~X)=nJY zKFHc*zI1)Cf}$x7#A!8eBcn8EjTxM{0$4pwMh$V+uJY1M96-kz53N(EcDgFvLPHe( zYW`3QeV0&XV%><38JLqj4yCdIcPVx>iuI15|(x|i~ zkef>?Yi9iJY`f~kw>~Hj)Vxwxfc_`%iOvjiMOM({ui6h;bi^?96f^}^Vg(Ngs0CHS zT^zcfi25Bt!-Qfky3Uq6fnxyQ2$hoePCm^pqYetEZIX*gCpmivq0crm6PgvV1PupF zoicGXFdGr&MaB*E&5d^dOqA=|z@k~0-*)Gvf%4&zIoP~9r~UT#1nA`hMb76G3fdE7 zEfqiaglB>WM`r@Q7eHwVtKKjR_)pLd-B;&N`QwRB>XRXrvGyY>fD7n`Q5KtUHx;ps zUkY$9_SnID3Gztz(3u5P-Gy)nd!aG~{BZAGDn3}XqdX~qXA&lve9gf1%Jrvltfn6! z(udQID!?AzPRwoeLke5=kA6ODR4VsO)F86(7>>?GO1p4@;JqYT*ao~5mAU=+%G@He zth25jo^dp*w@uue@?B~>b$A1hnj!dX$ebyP%PVkNx{{V#zSvCyxO#@*k9g!Y+rc)2 z1KegrD1^6=T1I@WA=zS%nC+cV#pJSQjUYeMDN$vx+CHh$2RA!O`Cu$#DUr1d19%b4 zh2o3}aT+LfGa}F(0L?o7zw-pPMY_S31hJEh5VP@jIX7R8g0$;*&(GvW?3g|BIJ|>&PC|kVK*);B)0pguAy7B^8^F z3JY3)Sz=**e7s2*jMYq_r(+Nt>D!g|e`=w__&1b{U?OD? z8pe>5C4pGw5@%q=d^el)i9)D{EWk&k^PunCl7D+LA6~hk`oL}&pnH2@f!<`%GE%uB zA9}wiDu!kl04@sME#e)cn7h85!)H(WJrDVh`=F0`@*cBT?X&Xl&+XdTheN74i;EgN zOq}@!+v5oJi^8d4a=9y?WAhUWRmQI$Egp2dqw-93uek`2znw6v3qvtYEBNy~##}vF zmI2xiIJZPGU0=RsgXcMd_VIW^c4{(a>tC9lae@rTa6ZqLAYY_hVVk&Wu5ScHX#%#M zQrxMbGE)&)wnb<-L)JB;o(Db|@?u!Ab(~8)6l+8m!kjBBS)>s0Gm$PL+5?80p9`;e zkE-1>`EG-sa$|!zHP^rDaTmPmd0%qO;{p3}!=<<1JePQX;~O7MLU+sNPXFp7yy)?N zNJsX^g81YT07(5lUW^F*9?+v01ivxo81wlg9v9reV3V!{KcvT-0Bnc<_$RjE+q+!z z63AvXiV@V}SgED46&lj>l#h<{c_?}G*K??suKdi_um9-IaZO?V$w|oIf;*JCyatF$ zf-y&Fy*+!@KU-st*VPwust-K%S?Z+a zXISd8LwngIEYHvOwI>T&u{8B)UI3I|Xkt0W#dNU6?sLyjYn}mo2g$8i?R967puN#$ zR^6*|&KU+%O4*kx^PQF@j1W)Uu+ED(k}OyE&E*xQ&JACfFVQcQ11GvZFwpxkpGxT% zoE_i{j>=v<*;TKi%Tc_(M9Y}-8aKq1t1cEABZL>VjSj)I!8G&!|L{Y-_`N%F}}HJ`YQKi@%2dC z#{t~bk@glLBcAt+tb|c_O3;<`6`r)PEdwjfw0ZO)*I4-Nd!-*CtpHK-ZsGq0J3x9k zHQ9$^bGG9PZkTjoYP&n;>$vM_YZ&FnG5Hj-I=1^re(#kL|E6OS2QcD^))9RZnes`- zR!U|rHnvOturN>ma8>!lhe=l(h*11qU8mTPEchrdYA%8O*z!mty;uuGx;vex-P(t*8} zz_yeEk&={15o$E-O=;iN(}=hGW!m( z+Bzla0pO{22X~=uus94kVcZN}S{xLuomkr0@e`Q`cV_mlV)O-Yaxh&u9-60J-23+; z90yKLe?~*jUCnwEKbQT^|NdeVaiuwsi)DykuN>W`@$M$Gah{o2*pOiuc({@)P|Vvt~H;DpQi8UU2g+7?CY-CmCu*jHg9bs zw$-ZBBd)f*Q=9F-J87(?(e~;6DDK#R2C@UZmIo~AdV1t`!kZO#e0%zWC1>pH&eG3M zFiyX)l^#14nsIz!M==~&8ZY^>fKU8&IM@GBm32=?02Cve3Do^qY!EjA8REYo*+r%v zOQcB8x@np#=`-XqFK^EgS;nUP0v}*h*2=4AE^?Lo#%0?;8CL#7SVcENFeDa1a2q$cQ!G=y22rBEZ|N*4(EOC)j1|k z1!o>O2sqnqb_>AFSI=dUc_eD8C zJNGKui0amKpC%lCQF|!L^OT^V&HVFo-m=(syl@?~ZbW_~FCzU7jr{Wnf>F3HvVK#M z2gqVoyRwoUBa{XY_%o?^FDMN|_W(a3TH5hh z4oLJS5vLqW3Qg60;v7^ejJ!-*S*KtMbeMkIqvy(;o}H`@PBPSv31`- z?A~Oj6Gr(Y2Ww7%Q#rtXM_hmn7p1@{0{#fejW5zid+n2K3di>aAwL_v0~P`U-}N6D^F}Ujz8_{^_z|_G&1=1)eAx%P zGD*K?Akx2`;@7FW0G+GUFP}p3F^OBFxdqU1Ay2G zB92TN%%&!2Pa zJwJOb5`WwR0x^_*m?K+mJw4$x;-D+V+oUL`XA){ivsH%Cwz*}9>Pawt%1vlfR#LsF zP%P6C8%$27@N}yD#6VZ@rFzca4$$M`>AI`b+vVcF92P^P9__?{~FL2r1=t(2n;jh>J}IH&*ZG z+{+XSzaH0~CcOLOtCkuO{}NGG^Fl@x?d%-vC%qaGX9&)uU)LFJAFK4$75sSjewuve z3?t@x9wvUfihr46yhOfSzFwi-BK4td$y1wX+gJ;>QaBBLzXv+*0dg3fo33_)z1Jfx zGq=(OuL0MDO5$f=Y$UBB~sTw4%0Vrbmc zqZ2tsBJ42dg+&Ti=aqXEz?F<-4;n1RVSmK8X9+$vDqxdjK)}~n~ zX4%Gcfa@32^9RqAxBH()#?$f1iFsPL;M_Jsc6ecqk3|?m43~+y!3|dAtX(^HBazG^ zcrld5Z}r{*E^QHJxIz?@wwz+DI_@@$ouF)nw47GO!F1Talp+JcZTXBdKi(0vbpWZy z!8yaMJRKTx<6ys$cdpdent7;jR@5dff@*>Z_ylPHr7azpb$qon0OLfeXaI&}|lh5%a%&|=>A$lR8LW~*7qC7oCmSr`bOZX7ku;G8( z$w3it8b9cU+}fc1=CP9(A(>K`hYFlH7(?&G0f~`jJ8E)3#skX(QB8TXM;I?M<=AHc zAc?6ZJfwX#P8x{Ph3z8shLmBM0mg6i%YmEwtBwg~pCSwS9A+>96{g=7`4n%znaBgN zE0|W|dr-jMW-yHj2McP0PEsD&;c9Eg-{&cYv)uqtKjXZ|ZiYLjCIz$^|e&asi01&+?4K&u@}=1;*KTSDxt2ulv7 zO?S^xRp4bnt|U@ls)WzsyAv@P&WOo@3E@Z;mLUpRj`&n!5Y$IH-_3=gaYo1{1tJKm z-3cl0F)CUtjwJ40H`3Vgm<8tfCloRSXaLIF9WI3=CiK|>mmD)RM_5~=l6+4e(0YoB z4>E^V59F`F;l_Ao;3w&ASRms#v!B-+DWybZa$?3j%?RXVe#AA{0`)ViBhr+h7xKSd zfkEnGApjibO@^^a8L{u0Ys*qH(--cSAZ-PQ2~Fd%qEf48=h}8l;Ekb>qpU=F8v|k8 z2aTHFrjNu)65>vyG%jhI;_ey({N`Aq3J7-=ElqvT2K>o9P+QG)U>VSS@MI7IEwEAg z+#>|m-I`)WO0(TPB1nONLB$M7Aw)8XV$i}$99io)ps<*8fS_P_*tnEv4T;sG>=$O5 zjGXfqB!gk(4jW0&W6Ovv<7>yDAW5GD4?pYt0%=dJHEE0g(5fn8K zP6|?4QVf-6v>AGHgYYAI7oB_tvD5dF`Z_t_cWV+U=N1qHOl(8^7(xM3oJ``+1*E}a zCIRvijPZS!5)8sY6QB@SUIM>_EvNAC?2MW6ZRLo~;Dkd)haMdPp^zyKu+-H8-APh| z`B_Di+KtH0cev5&ZjKMkdocR`ksQ3wh~?bjt~gZQw7#?V&<2P| zeTEcK^RTk81rf1;Ivr?sQO~_&4QwPMxEv^gGNQne8{$C^;3Sy8VE#D|>B0|`!VfNwr%)x(oTh&Q>wdES#aGA`hjUGTwQTiAehafHDC0;bZ?2;t%j%}P+6Xa+6 zX9@RWXfU1jfPjttAZcRe`2EjFW+cMcgzfzwHjwgyUO_SWkul3#C7e=3;gO_c@oe!2 zmh!cjPC2*)QWn4FqkTd#Z6GX$aYMPw5-DtBY9Y$hL%eBSK`2Pr;f@M5E{IU!N!(#L zBEYKIhiS%cP#iKC1z2QsULe%^6A^gu09wU~gydm?_T1tAi})Toh>X*-yMH&3kWXt7 zT11>;`~_Sw#WYgh%u5xL+L1nf&*&JNgbjx4y9Z*1AUWiM_5nmN*@IJwY{fs~M+PnP z^^S?=i)n{|u1<8p*!xNPJg$^Vpmu+fqgxoZVBMphN=A zBsbA2g@Q|4xV~Re&YK-y$D5toMi{U)o#s=>{J2 z(ZeJ)5+S7(j^OnJ$uER_&n zAiKzTh8Yt-SOx?vq^NyBTej(pb2*Bsz4+{kWYGw&p*osFK5h*C+ z!U9UzYS5z8sYor4@JpbC=*+*!pjR$@q2+)Bzk+_!+EKws`w!g9iNv-M;sduE(%C7& z+|0QwqXRJm*a9-J)`VCp+u1}mDo&@Im-*d*d}Cs z*w*Kw-~lr5h*A19qD3O9LXiA`pwyv7&@6<5Ze?IT2b8>uGbpk8Act~zA}bUkc5cma zUL*(1qZ^2!DnxA*!0BiH!YR1(3$mkafR|wucX6eF%upfUZEa8?CYun+h1&v^geV5_ z)DtPz+z4nA6oWKgDp$>bQn1R2T_hp~acwNDKq;w6B1ed|7E{<0+X(iMUP|}@*qA$B z7;K+Ne^y#2^t_A|c8FGRz(rVueh}**jOC!3XUWJRfjBJyKrk`)k0bV%B0y6S2}+qc zme=`8;95j5jlvzV&0fay9C^-^*zI@B)^*bD>Wb4P2q@EHl;PEUey#*DpGa+X2Qcd& zTs`6gfgWV6H(p{|V)g7iop3^E@2DOqImUXhb`Qe*Aptqi2?P2;5OWO)&U;oTBnFYX zz*1^NA>>aiz#du6r z;3Y_1K`Lt^kJxxwC<$0SCK3mTf)DX{URy5aEtpZ=fqiQ_WKiq|7)i>=oXTT4EE4U5 z%gy*bk9hgO11r&+VqwV)=EaYjJ4M+66WidEvVk~Sb+YyAi0g$ryUl+AADJOF5q$T*el`@*boDYBv$7~c(}_Oan6B$ zNFX(`O$jAJqjic?d&w^O{M}EDb=)0jic3yx3^iv8oPZKV#{4HH8WTBbH zIsq{)QWu%_=3yJW_`h^fCHN7u2O&dalDJL)CynMf;ffvZXs3T2CfyLpCIm*n9YQ`r zF>_8ssGBgL4fsTqnWJurN#_j@xG}H7^lu&{isM0Zi->Q;_#==yte&IMrL7yw&PUp8 zaXjB>le@Pp{7c70*M+&_koyd0pio{U3NsASprhou9PXU|uK*}2*VQFsVp0LHG>wp* zOXdf@i<*CiI(YQmAlCDTM9o;@=!G!606!9LlIf-!2n3EkvH^$KXwk!wO!!WS{{%=c zbSjYs0mgt(WC^jF7{WQ?caF&27dj6}!k9VI^V3u>a1z0Mo?;+Lq=ZXEwyKxs1xJI0 zVmWd4AkHE`aY9ErK1-3@A(G;0;_3xFSS*N%V6lJH8{Csn`$$xb!B;q0=H)Pdnh{5I z?q^6>b2NCS_xvKANZ@RvIM9=sF)?uA%t82>IOqgvjwA(6Bk_Zf=YF2}#JQjtV>en_ zkf#D}EQ(?Cw9T@SB<75`#Ud7ZH;Qmv8S9CW2(di+x8C|2#gUtZL@+cCy@|_aWf!#4P?GP64wMVZ5k#C;uqoOI>SIFh3J_GmSTuYvOw?q7ETZZ2s_Y@ zrIQNo5~pwCJPOh@b`WtQrAW|*CEjW5AVK!wJrW{V7P22$coPrXDgjOPcrPJ5h~ zSj}9DBo^pI9Ak?s60}zY+KeYY22x4vPI-SGrXGs|f%rdnc#i4`2Pt+GBCGXink60r zSWjD9rf~eo`jJ3EU9=7s@-!fVF}Un@ju#<96#9vFO1|{q-aL#kG%z4#Spdf$$Ec*^ zB7}7HKHUs?RLE$JHVINc3K;?x?PB3t^@cB>!E6Y*W12aYiL;?_vVr8TKDrYkibO)JXZN)8fZGg+ zX?23?c-&2#NFt3!1ylH-KsfqTgzFGDG9*WtQ(z*7;S!Vxh?$7qyz zSsvr9y6}b)aUTpX@gC6~5^4`%dQ*R#lO_642BHq(FwPKf2YLcICr;rf3{5{ml8<)f zJE?6c- zT?E{rhml24!QgZ}591woT^{5)!w~Q5Ln97cEpe|qayKVP950H57b45&?WTWF#2T5C z^N7VnuqcdDCtw)pFsMg*&mNP=^%$(ji8*>AV!ir;C{22TKI`EbxGo$skQw5G5gkWPLO)CWZ&_ z6CQGh7|sk4qn`d^9zzdJ^zPFd+to2XfuBc^kLT<;AEBc}aPoVbW&knFg` z<+T(k4GZ(sb3BZ70csYARO%^B>9dd+1*HN_--m_52(J?ZiQc3e!C;{a1Y+QlN@R6u zjy{#@b8N1I_`<>aMN|}_1=3ysbMy`YMEs84B@mWL@*-6$dFUP+T#j|AEScY&tn-x zOearXh9QrrS;T;m@+8U~hG9x12pEx?$W+E|N?fgmBgFWK{62_cy?u?xfzT7IXv7Y} zE23s_&mn?Fq|tv8u15#&S)8LIeVomS{9Mw;0G6{H;f~DoMr3qNILIUnkR@^_K#D>08ks&r7sh&x zLtxBBx73@qMWmO&g&(;@Fntg=LLZILi9%1t5rRn(X)k}Nw+;}&M5y=93yi&b_l$?6 zS}>vandjW)L@+N7(^SWyZmhQ!Vkkj$oP~&VG4v6eMuxzGxv}t(qkG8kg&)8L=~*!D zBf+Ef%?wdqmLucQW=EFeIC|%k%YYo_)H4;CKAxQ*v2^rC%q-DsJHFmfBJ`F4V!o;8 zZMcW5(-D6x=0XI1rdLZtbbmx1#FdH+4#y8uPoIZF4x9>4a=mpxN(2mHLab(xVi<}N zgz9=aIzY66Albzzl`w?1G_Rp|pCK{`5DvRiFZ`iJq!K>FlfI7lXkdsb&^aR!Adl6T zMnp*L&^+`mUBQW06=bnK7wjMm)mxYH*wxb|0^xs47)#WX7KUbssWLq?i?j#91y2IU zarCUF2S>??=L0gmqgF4cAyL;m#e!UFW%W|5-seRd(e=4=i7*zyK*S?2fZO$DAfc0< z?#MGDm|jNlb&BKb?ZRH77XkH^B3Y&irSI z%X|-UDGki?ScJd;{y#++#AJx<0x7W1Gl66yB7$+~pr=962SU_Lx;%ih%w5-$B9}xk zB+qtT$HeI5i^vb)`csSm`c2bf4X!6L68L}D z9Np8^Z8lC_kL#Bb$vEUhFos+NHiEHY_-dVGBm2agh1-m&Jpdzz$EDyHrE+~VSBNBR z-7V69fSA1U1VJNG065!JU!O5H+=+8IohXBLVHUbxqBrdceKitKq+YY2 zf$6(6kU#KT?*a<^TrZ=ms=PfULW(~}VTq?+C|0}q>UeUVOxze(Y7k#M>$(ma=%yz)E`WOpneMC#@0i3#1tbSw#1MC<6nk42v9qi;S-;rb;< z>>0ReA~J?if#w_$kR_Q+^cp{+0YLLepLN$~L%mQCLw7w}iOvHHMuvargb@-0 zy%ClXUx0{ni8wZlB1BfmCF00Wxr^5t!<%Kqxge1^d8|8V!d(v$0{kvZ;BjN$aqw<^ zB3P<7qR14ULZ4_smXRY>N1}ppIL~PQOz7j9h=)0RzutqNAvD%g@JJjHfq1kYAOuff z>X5fP-P4d|PA-tK1JXwbNb`RhdK-hjn*i8}S85#AgTza?zgA z)|XRdZj|fu_Yw^_CxYoz3HFy+q}LVn)qSqVq!2k;F>$Da3n-cjT z0&S+xi^p-0p|57T55Q!ZlMZv2DQ(6@B2f}t-#~&m2yTBYNy0qC8xHjP8YAZD zh|DsP2Tq_5V@1sI7+S6?krP8VN$mj=;R)n3N8dl;;18yE6oxSdVDi~&DKUs6s;S}7U?T+M2JKYu89h!FP1?#5b23Dv^6Ks9UsvwVr_I$Dp(flGkFQZQRJJ5*Y$z8 z9K(8;Cr%O*!32N2MUFr!^V1;k(%jYCd$=FPNkTnUlsmB>qbVSYb#v*5sqYDf9Q~+f^pHlJY*31h7E*6Og9n?S-$HK2jSci9+W{aifkQYxxRiP@$p_+F7k-DZ(H9} zgWjXp`Cxk%1~HsCI#tM1$&ocu!4izQv7Ru_kq4s@r%r$FOSBL@7wHqhkT%8ehWcQg z&eeeklLZ|43cMvPN6+;&V*uny=0pNMB+NsMFfsB)@DO7SgxTD2^c_q37#(AgueUb_ zzP_xQ6j1EO z`f5%N_*5UzCC;Dau|5{*AwFQ}2cd{F7w#5q*Aa*|h)a^WzE#FYl;NWiT*iJxld@RejY#5;H^%NUq~JqH30V5+jn{sDXG0Mvr|F2YLr_B+*!C zV0wQmGhV5qr$hyYOE+Np66p{PCY8j?A#*gXb_W(>*$81m4rd#tB5(t}^B^TQT85F1 z2@FrZKI(^YFVQd8L*t9lW{GYagL$auRYk1l;t=p95wcB3Z%Pr=b*G#o=FIfo+*Duc zfYdOJ;uK?W%(8?LH_js-&fo*%m`6e2gcyJ1QpbrC7RN&G3!=qmIYJ;k8>QWjzS0Pp zN)!b-^kV1}#PgfEgaZo`C(tK0kvm}|PBAX&B?4re#6|gB=6ZiAVr+fBH*zpm2C>6A zvYku_;?j4nZ>0+KLI46P2WSG`M-Or=apGWd()V)z{_*1nS6A=ruVA!)>dy@r?*D&3 zJv*WO<5af&WR3c&{W7HKa|31hr|XYB{saFm*tYw!%Jqk4&MI9#Gh_Iu%vOJvK@Aag zPft|sCma++e*g4NzOt>UTKeHeKRmRbv*<+ik58?{_geP5^W(3!Maj0$M*8m2nY;U- z#h)B`iHM}`(mT^k{5ZwmAR3?R4@7~_q|V}K=$b( z`@>KFc=z+^udm8-RH^^+?nim3(a3(3&4;RfuLUk&Rvvnh=K>u_Sw3AvE^o9v?Dl`3 zOjT(gTs~J+zM_Zd_S97Gemv}c`qXx||JA(UCI78vJJ~7L`#910`%32~b@gQ;>yI*B zWY6DPx_aRgE#apY?Y2P-eVdC{wfwsK6wbA;u_)eg-TgShqrZJ*PoqoUv*_O}NWNu_ ze6H$GG_sSFiS+broSbFKni6I2i12^Q1pG10vVs@l9$tt^_}z{|u-o-1cDo6Z+fi+* z>^%WSx+xJ3mo6zBtw5is{dt0|v~OZ4d(Tl$HK&@D&xES9#qUYlAE$UYuLv!-E#BkN zKicrT5Q$!lOfQpE7)U8hDBdT+k2X9nM4}fX({z;@Mye8u_lfYM4Ub18@{E5>o(h!@k zOHy_EjN7~WU#F-}9|yG~I(;zo-R*A{BwznRr1`}}^z|<`RHP{3Bw?hRju^UPlri@t zVJ4g$f4gEO^{p93$J6gtgw%iDeGxL#TJ`i(Nm75m2n{k%_x-k{BE{3!LS|+)9bF$w zHdNNe%xKg7@z3&CBA*s9GU^9*oQh1T1{kfnDAzwz8*nFNrfnbDGLsu{Cp5@x*ML!S zUsr-ot)+&z^)pLyJfU;r$q+Rj*)WTOJrjAgh>=x4u;r{uww=MN#Rh-5^+PNE8LmnU zdMaV-sN}~zkrWa}O8jm^c=|Frjr97X=ww$W?UC!XuVuRfJ zp$-4GqV^?77+LYVC86ukDP3oX)OBoW_cLhsgS2}~T93}^4Uu}5oOnX-x@Wl|_U@^T z3iP3PXdJd2gm;ororZrFF;hxr{?L$bIo|K8SIIsW)X3^2W~M$qwBawaPmQFq zcS1()?vWLPME$QvN=ZPE{L!)d#j`AIXF6 zNz6=rd}zbpRNF#Pn{y{*G1shAd(uket(=+v^wdU$S5_0rAz`G&?^c9`{6J-d zh?x=(EI2n3U7Z+hbTTH^?#TqL+UTt=LK6D!S>&Gd{9uB!*-ECGjZ5G>i`;XbAJ}nv zbWU%G)3fCC>74!$r$51enH9~4Ldgvygl$(ccdS=W>}U(U)2JaccVYwN-9tP6swQ(+ z^#E^G+i`xJLHco!_2VS%w&L$MJEq=Cu_5emxs$u6zkNEPLR~Szb#rVgd;Tk-d-AJC zkui~U%4C3DYrc``*?>IKK8d>6h1iG_KkMjIvlJ-{$->-e5U#A%k&%|nV5(cjJ_esK6!Z6tt?^C@_ z4Uf8vN8|BKJe~!QPvh}TJiZ0b=2@1qCWKShLI%#$BMatpU`u>=*McSut{zjg@aAga zjcehNwD9L@;g4&7Vbd0>?JKFzwgLuL{qq!?A`qz_hnj?cq2Ryt2`09d5!ddGtM)eQ z^~9BW;~Krq>O6H_-n=4jwH8lYg*UFh+pN6vgKDbX`49nt@J6o3eDic%1tD{G*ftwM z+iial78z__CqQ*13}3&Zi!!x}1+hz8ic@&_Zq4D=W6T z&CS#13SdlstE*!6+SQW3&eBQ?j%cGF)3n%Zi}!N2Dy{T{K5ZF$rIaMh#@!2!sG=WF zY4KVJiY;{U9;5zol4n;@>ZptNd3I9~mMlt&Tj+}4@pM1h@ND*QTST%}#>m({nV>Ca zjZ(K__vEGh7)4utFfBY-EIG(Jx2LG{dxAQjgfGf}R|*4)_et>M1kWqQRIhzYy-pg6 zuxYHnQEYN^7z(gbY(%kB6q=JvpYHTzl6L1(2zPFuYUfUpRCPnoknZIA#Me))RA{P= zs2ZvQos^lrduBsUi0V!PM#`zkO@1Qys!u%!^=ZN@X1z>O-i@ukU?$iS(Lf`)`GV5J(2!xf`oXy$$YTM>PS<*tq8kB z+R{CmYHDXlTVz8_`LBgclpCA$KR*!8Wb4BBOoyqH*xdvoWxvZ=;*RV5yA|PYbjII* zhA4lV_On{L@I4uRnqrWQlBgZa_juZ0t*D5bOmC#jv^E<}UzlL{MRe=-IVF4Pi{7cx zFEFkP{25~S#=b?J1nziGKTojj2y1D7y6`>AZfY&9*v*^mxRdMg5@tJ!vLrO(w)8!n z{%wLp(Qd;!mcJ)b7I$G4D%XoM6s}I7N7ns3#fE5wK!wsh9{e^%QWB13wxxR{<wcbMQ-_Rp_WvT%V>~?==lu26ehwIA?^y7E%Q*b| zn&2*U;X4BT(>TLc=FLXE6olO9B67!ec`(JP6oE~t9*f<6*iQG7K7}#Dx1kAn3F(VQQr5J(0~H^krlHno?ocMTE@()pV-pA zvaOm!-O3Hpw@4 z&)NPw!B!D+dtLaRrJBm^DcAY*FY=kZ!iRQwkLlVTiwk>bSM``K>9M$;hjcNI;YyzA z%YS%Kf(b#{QE`^&b9KK@5*D0oNe3~nWcsA1Cw8C3!W6kbhAMkg@@2I*Tr+7XRNIv!$K2E|HJHjTqWBERf{&j+iauJl_ z3aIJJpLSBcJlBOFuyg`Frtaq{wg%~2cYi)%Lt3KhN%r{fp4%ugYeR5f)ToO6Cg85T zYFMo&_xH`ArYdtIYt-hZgq=FoezrCj(E(*H1txSM9B}HNZkjyWgaWjl+?DUsSoCD- zZD#}boY~!Qw$mS>*>COVI8FlhY{#F+*-BRGO?4%~k{6LX#>;~VPWjCAa-2QmQGei` z@A&f++Xw1Nsf$RD@q^U~DZ~uK8M3p)J>%*3al)!3oX66I?`ZT-(+sr31!oBZm&GB1 z>Off{Vi|w-&Y?9Mr%vo6Y1Y3NGL~24d+Rx z50mZ>&4z8O@_9q@ySHNZtTzwG`G3iW4hN1L+8sDNl@c3blzlHazu0_p6!Ag5$ zAIsly9)F#pDn+&XI+Jx-9k$zozwnPpvUc2;_TN?+y(kjbVLJvl>?gtaQ=;DawYb$5G6z!||aNzmkNT;;v-O%=(EP zE$Q~qYoP(+>H9;EC7b%tUOyyec8z;^vp)BwtW-Oc1!*7Et(aMn@;HAs9*Dcp_$MlCMYkvaeQKH9q{C$dWF9{!?>5BKL^p8_KYUWPLo@9?*QFc>O zud+5@Vr_MObzMBaQ1_i&$=p+3o=nhIgpJU;@I6sAwGolNZr--I zb6YXO^oK5dN1=b3X4pJGA94~+D`DVjf43l1ESHTeRaGk*@0Fm5{(tK6v=+aUsDC%` z{cb_ho+{U0rX~ah_kt$stH%>soHvwjiya|CxRblr;r8jY3Nvp*>Wdl_q1=jeU7k4K z*B2ep#b*9bm+uCJ8Smy=(4fcl<7q8E6g2`Xx)N7%1|6=RPN+~k?<&Fy)v^3NW%uhO zRY~C~`y}NQ1BIZEKYtDXLcxFO6HM%1Q(ICG-5NUOcDf(xBAdfrP_7ES7BW#DA5Ab< zc|%!0K9;}d>wcZ0BE$%WZ>COvFqM||nbJLrJdYIW7$O~uK!^4{2;-L^Y`y_O{rZFX zyAM`hd{7PJ@9l|n>3bUe+Y|}GY4$!PG5HaRb{SV6&?+QhLVx8(sz-VCY>J#VIvYo1 zn^E|XH8tae`W*%CxsE?iu@S1&TRWY+m5KcQje#$3*nM+DeJc4&U6qtIkSm#c&dZZ= z+J+I<7j@w~BK^}e!Bw7-p${4Q(4b?=0nlESYP%!Bs-+UrG$+-wG;=tAgL#0RG@og z!%P~aO>##5@d3L_VK;osrcdx*m%VdqKi_1+?Z@v7Xx*pyTFF1_f1b(<*07B}if>Zk z^01TqEB=1>i~jp8-cHP`w$pby8gcrij)2{RcYpezP*;x&^5w2)6XMlP&>hlM@o%f$ z?obx|W@kxvo2LP)d(feL*A@EP@!F5GGWteDgJ^qEVk`JJcKMd1`6{&YeP@6B+b^e* z-T6R6;p4%NJ2X@8j_d42w= zs95!5>j!@Jb9b-5~RJODDVq38RHcYHr-tCf?Y*tG?#P zV^N<7BpPBFFU}C6Jjwv`jFEuqlvO; z&4dW^!|rK~XB6HRlQ6^>#HRhUjExYROIu@vdG`FW%_%qa2(byZHAXm$ckpb|-=>nL z7*BKB1y?6UF*e_5MsNlfsYEdzr@c}ofL~|7sDGoL5n?fpRvh6Co$v-Eyg?G)(g|-t z!f2;pO=`^6ZacP)H(~4Mcq#MOSqpy%jzHOrMgoCZe9c|S|13>}5g@3?@WDwtoQv|< zzv~lapF7Gi5?5jykKv$H$2(EHtFW za=Y~)sR|KOfV$IOh?KQ3)};{`y!E`5S@rr(Zg$o5h#$o zH~$LPDPpg&L~3SyQC1ZH_k;exse1%CgE&10e}I z4JjeCsvQFZ>8)k$m?+83sSOW{!8WMkYxehfc6>8)9vK>I%kh#US~j%t>BbwIZV=p4&f~a?ku;lk1P<gDGJ^2=j4v)j*ghxB za|myng&QIawD%z1I}pVCk0*UFQ%-nL8z7r+vPWn?w=ApLlF(b#2#Zh~BW&WJpVx;* zzA7QSdbVH|ua$x!Kb1|_^x>Uq&{$uFl}ykQ-lPT(4yti>tC@><&eMwJoPP(4t*fR{wr+3dkn~RcE_@R==7kgTn2Mc_@w8yj3n2vVH!9z<>IE!5v z(5#81E+>WZl*oB~L@vZY!DWG%46Y*eu+aGCM4TVlkGp*I$cTB+Dbn_WbGEHSJ59v|u<#YS51= z{UDA|%=tB_iZbhXw_snw=dvve2+4;%$r<=Vg&r;KMa6u|E{qjfy??LEa%l;Bu^_dc z6IJoKs89J)sZ$V0_*I4@Ra#sT%24pfyq@l@3NE~}q;)m^F;hz2PJ+{n7w;yYP~02Q*dyZ8^Z zg>y|{YRku}rs;VFS$}++*nuF4wg**c-xY0{_e*7(VIYxaI$C0t*l(`fE^|fcEw`y1 z<<{#Xx1{cIDz8UNX1zcq!Ix2R4BTs}FxC?=^MVSFl0GTI;Fne`_J7g}h=h8f1mkS(@T(;s zojM2OO}{uZFCzdBJ|?R#&)v`3xr>?%c2ki2< z36u}S08kAZBv3q%A<@GMB{;y4roL8}zcsLFAx>@`oIkdCmsw2HMj{|?++{?;^?x5e zyN$`4WA-*L`+sbf0~G<>0!1g%RRxd#riKXVvd;m>M-A*&LkkcRWo})Ak610zQ-#D- z2}-^!lkz2~q=;C1DStAOW4qq9UDKwEQ~^M3Dvc;< zcmFnB4hMr$xSc3NP0Jti1GpbkLKJw1GsA(XP;7iu4@)GM%Jf2kT0JkK;QyMU-m}9w z4=sRCiUUfuU5GkMxqfrxQEXfpDFu;dy@jS^U2;l$Mz|suI-t>MyOf6LamX%@4)5;=pX&g=MFlj}-X; zPrSm~yL-C565!PY-ikHMJN@^HWj_Q;e5)$HjDOS%oSpE#*ARQKueg1s(K~69d(k~8 zliVtgQ`uMA&3X7*^MWw0Hsd$y;cSN(Y^q`m=^L*5N*?pYn;s^Q1Je6nJDF2SDn}Zz z(uke5h#Q1khW+rIn&W21sXIIeM(q(xzzXrr<#d#qBgys{LMYoxp*|0P)jo7rR#wc~ z=zr(Y%Eyo#*e1|RjO`LQ0Q^k?O}n{Apr{?w)WqPWr(5>$%^7sEXJ@SPmJo5A-Y;hX zzn!AzX_TL;wlg=FA1ZtwUO(t0tNb=y!-oykMDzP)FFs|zc!!084m$Mjm+N<jl;Hs1M>T&I6UupcO1I7M&c0s zlQ3NGr1$-$IObQOw!{w>Xm4g37wEq6;{x{;d?Q_$Y;yg*IDupphRN1XZNDkqysF~d zYG*%XwH VPttK0aECqAK5+Ev>VL6A6@t{D8q5Fy diff --git a/docs/build/html/python/_autosummary/mlx.core.Device.html b/docs/build/html/python/_autosummary/mlx.core.Device.html index 3646b7d46..db7ea9462 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.24.1 documentation + mlx.core.Device — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.Dtype.html b/docs/build/html/python/_autosummary/mlx.core.Dtype.html index 9a51ec6d7..4e7123efc 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.24.1 documentation + mlx.core.Dtype — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html b/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html index 066d5ddaa..c85da44b0 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.24.1 documentation + mlx.core.DtypeCategory — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.abs.html b/docs/build/html/python/_autosummary/mlx.core.abs.html index 005671c65..9c64d2512 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.24.1 documentation + mlx.core.abs — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.add.html b/docs/build/html/python/_autosummary/mlx.core.add.html index b3de07810..e2626171b 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.24.1 documentation + mlx.core.add — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.addmm.html b/docs/build/html/python/_autosummary/mlx.core.addmm.html index e2958b5f6..7e2056fd0 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.24.1 documentation + mlx.core.addmm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.all.html b/docs/build/html/python/_autosummary/mlx.core.all.html index 93eea68e7..0d1fc24da 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.24.1 documentation + mlx.core.all — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.allclose.html b/docs/build/html/python/_autosummary/mlx.core.allclose.html index 54db77941..290191ea3 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.24.1 documentation + mlx.core.allclose — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.any.html b/docs/build/html/python/_autosummary/mlx.core.any.html index a53fe0021..fbda16506 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.24.1 documentation + mlx.core.any — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arange.html b/docs/build/html/python/_autosummary/mlx.core.arange.html index 70506069c..650fa4383 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.24.1 documentation + mlx.core.arange — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arccos.html b/docs/build/html/python/_autosummary/mlx.core.arccos.html index def7aed40..4364e9dcd 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.24.1 documentation + mlx.core.arccos — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arccosh.html b/docs/build/html/python/_autosummary/mlx.core.arccosh.html index 349a0c180..6930397c8 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.24.1 documentation + mlx.core.arccosh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsin.html b/docs/build/html/python/_autosummary/mlx.core.arcsin.html index 4c1de092b..12e9098c4 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.24.1 documentation + mlx.core.arcsin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsinh.html b/docs/build/html/python/_autosummary/mlx.core.arcsinh.html index ab2b90761..05d641134 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.24.1 documentation + mlx.core.arcsinh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan.html b/docs/build/html/python/_autosummary/mlx.core.arctan.html index a001887b5..9ba896290 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.24.1 documentation + mlx.core.arctan — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan2.html b/docs/build/html/python/_autosummary/mlx.core.arctan2.html index 985c3ab78..f12280329 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.24.1 documentation + mlx.core.arctan2 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arctanh.html b/docs/build/html/python/_autosummary/mlx.core.arctanh.html index 98823439f..75290ac3e 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.24.1 documentation + mlx.core.arctanh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.argmax.html b/docs/build/html/python/_autosummary/mlx.core.argmax.html index 7ddc15d26..dc746b4ac 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.24.1 documentation + mlx.core.argmax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.argmin.html b/docs/build/html/python/_autosummary/mlx.core.argmin.html index 04cb7548f..fa9ebed6d 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.24.1 documentation + mlx.core.argmin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.argpartition.html b/docs/build/html/python/_autosummary/mlx.core.argpartition.html index 509865126..0f2470af1 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.24.1 documentation + mlx.core.argpartition — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.argsort.html b/docs/build/html/python/_autosummary/mlx.core.argsort.html index 23f3edc24..0677105da 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.24.1 documentation + mlx.core.argsort — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 2ee8632db..38fb8278a 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.24.1 documentation + mlx.core.array.T — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 bd551cefa..742211d4b 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.24.1 documentation + mlx.core.array.abs — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f015d10ae..7824a70d6 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.24.1 documentation + mlx.core.array.all — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e4eb2625b..9a4151638 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.24.1 documentation + mlx.core.array.any — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 8803969ee..b8f9726c2 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.24.1 documentation + mlx.core.array.argmax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 db5ea2b79..fcd6fc459 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.24.1 documentation + mlx.core.array.argmin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c00637d10..4bd6e8a29 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.24.1 documentation + mlx.core.array.astype — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 be2138a44..3d7e5a469 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.24.1 documentation + mlx.core.array.at — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 1310ab195..882f6937b 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.24.1 documentation + mlx.core.array.conj — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a801f1b8f..465a6ecfc 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.24.1 documentation + mlx.core.array.cos — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f0cfd54fe..30703c443 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.24.1 documentation + mlx.core.array.cummax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 d38b487c4..339d85056 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.24.1 documentation + mlx.core.array.cummin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 1b1f10b9e..d4929cad0 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.24.1 documentation + mlx.core.array.cumprod — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 d885aec5f..91af8f6e6 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.24.1 documentation + mlx.core.array.cumsum — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 1c091d7fe..ac4cd9ecd 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.24.1 documentation + mlx.core.array.diag — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 537d4414d..d099cbbf4 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.24.1 documentation + mlx.core.array.diagonal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 4d1f07833..27ec4191e 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.24.1 documentation + mlx.core.array.dtype — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 2ca705301..01dc4ce32 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.24.1 documentation + mlx.core.array.exp — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 94cd2484a..78fbb5af0 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.24.1 documentation + mlx.core.array.flatten — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.array.html b/docs/build/html/python/_autosummary/mlx.core.array.html index e2bd3862d..5a71695ed 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.24.1 documentation + mlx.core.array — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 70b50e7c1..73f373b73 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.24.1 documentation + mlx.core.array.item — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 5d923a42d..0733e1b10 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.24.1 documentation + mlx.core.array.itemsize — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 32fdbd684..f4df028bc 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.24.1 documentation + mlx.core.array.log — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 72ccdb11c..80bcdf43f 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.24.1 documentation + mlx.core.array.log10 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 0f8f606d5..cfd3e5cac 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.24.1 documentation + mlx.core.array.log1p — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 8dd1e2420..21b20db31 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.24.1 documentation + mlx.core.array.log2 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 0babde59e..cd4a58d65 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.24.1 documentation + mlx.core.array.logsumexp — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 cba830112..132a0c021 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.24.1 documentation + mlx.core.array.max — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6f8d25050..6c5e8377c 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.24.1 documentation + mlx.core.array.mean — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 28e53131b..dc340f50b 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.24.1 documentation + mlx.core.array.min — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c2ec532b9..841ae54f0 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.24.1 documentation + mlx.core.array.moveaxis — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 093531d9d..b7101cdd0 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.24.1 documentation + mlx.core.array.nbytes — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 005b4ba39..5c61b5505 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.24.1 documentation + mlx.core.array.ndim — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 bdad41d68..847d6e799 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.24.1 documentation + mlx.core.array.prod — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 9bb75fb69..cbf246d3f 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.24.1 documentation + mlx.core.array.reciprocal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 778dea256..14e838acb 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.24.1 documentation + mlx.core.array.reshape — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a03202044..61eebfb6d 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.24.1 documentation + mlx.core.array.round — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 370274127..d73b10c4b 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.24.1 documentation + mlx.core.array.rsqrt — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6b667eb5e..f4f525f49 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.24.1 documentation + mlx.core.array.shape — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 b6c38d4de..526f6d3cf 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.24.1 documentation + mlx.core.array.sin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 0c35cec7d..be85662d0 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.24.1 documentation + mlx.core.array.size — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 37aa1b74c..ddc3498c1 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.24.1 documentation + mlx.core.array.split — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 da2138f89..eec2630a1 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.24.1 documentation + mlx.core.array.sqrt — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 8dd68f5e6..3a60e291f 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.24.1 documentation + mlx.core.array.square — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 550d3c0b4..bd486e6e5 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.24.1 documentation + mlx.core.array.squeeze — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.array.std.html b/docs/build/html/python/_autosummary/mlx.core.array.std.html index cb54881d6..b1244983c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.std.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.std.html @@ -8,7 +8,7 @@ - mlx.core.array.std — MLX 0.24.1 documentation + mlx.core.array.std — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f6f8dab96..4647d1a2b 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.24.1 documentation + mlx.core.array.sum — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 da20c08f3..8b6658677 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.24.1 documentation + mlx.core.array.swapaxes — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 4ca438532..ecf089cdd 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.24.1 documentation + mlx.core.array.tolist — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f05e1f58b..7f7c2fc5a 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.24.1 documentation + mlx.core.array.transpose — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 609672141..a7509f1ff 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.24.1 documentation + mlx.core.array.var — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 29088750f..4a804c914 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.24.1 documentation + mlx.core.array.view — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a78e2645b..c06979816 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.24.1 documentation + mlx.core.array_equal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 dc4afdcfd..1d881c6b6 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.24.1 documentation + mlx.core.as_strided — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.async_eval.html b/docs/build/html/python/_autosummary/mlx.core.async_eval.html new file mode 100644 index 000000000..865693c60 --- /dev/null +++ b/docs/build/html/python/_autosummary/mlx.core.async_eval.html @@ -0,0 +1,1046 @@ + + + + + + + + + + + mlx.core.async_eval — MLX 0.24.2 documentation + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      + + + + + + + + + + +
      +
      +
      +
      + + + Ctrl+K +
      +
      + +
      + +
      + + +
      +
      + + +
      +
      + + + +
      + + + +
      + + + + +
      + +
      +
      + + + + + +
      +
      + +
      +
      +
      + + +
      +
      + + + + +
      + +
      + + + +
      + +
      +
      + +
      +
      + +
      + +
      + +
      + + +
      + +
      + +
      + + + + + + + + + + + + + + + + +
      + + +
      + + + + + + + + + + + + + +
      + +
      + +
      +
      + + + +
      +

      mlx.core.async_eval

      + +
      +
      + +
      +

      Contents

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

      mlx.core.async_eval#

      +
      +
      +async_eval(*args)#
      +

      Asynchronously evaluate an array or tree of array.

      +
      +

      Note

      +

      This is an experimental API and may change in future versions.

      +
      +
      +
      Parameters:
      +

      *args (arrays or trees of arrays) – Each argument can be a single array +or a tree of arrays. If a tree is given the nodes can be a Python +list, tuple or dict. Leaves which are not +arrays are ignored.

      +
      +
      +

      Example

      +
      >>> x = mx.array(1.0)
+>>> y = mx.exp(x)
+>>> mx.async_eval(y)
+>>> print(y)
+>>>
+>>> y = mx.exp(x)
+>>> mx.async_eval(y)
+>>> z = y + 3
+>>> mx.async_eval(z)
+>>> print(z)
+
      +
      +
      + +
      + + +
      + + + + + + + + +
      + + + +
      + + +
      +
      + Contents +
      +
      + +
      + + +
      +
      + +
      + +
      + +

      +By MLX Contributors +

      + +
      + +
      + + +

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

      + +
      + +
      + +
      + +
      + +
      + +
      +
      + + +
      +
      +
      + + + + + +
      +
      + + \ No newline at end of file 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 59703a27e..e58f5e895 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.24.1 documentation + mlx.core.atleast_1d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 2c0bde19b..8f8751d27 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.24.1 documentation + mlx.core.atleast_2d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 ffb3dbba9..bd819cda6 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.24.1 documentation + mlx.core.atleast_3d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a11df94d3..8f779a7b6 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.24.1 documentation + mlx.core.bitwise_and — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_invert.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_invert.html index 4a7ef0457..83c2db5c2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.bitwise_invert.html +++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_invert.html @@ -8,7 +8,7 @@ - mlx.core.bitwise_invert — MLX 0.24.1 documentation + mlx.core.bitwise_invert — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3e75b3c95..98012e7c2 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.24.1 documentation + mlx.core.bitwise_or — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3e85e9b5a..fb3f16421 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.24.1 documentation + mlx.core.bitwise_xor — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 9ed8eda88..0a096007c 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.24.1 documentation + mlx.core.block_masked_mm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -45,7 +45,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      @@ -963,11 +967,11 @@ of size block_size<

      next

      -

      mlx.core.broadcast_to

      +

      mlx.core.broadcast_arrays

       diff --git a/docs/build/html/python/_autosummary/mlx.core.broadcast_arrays.html b/docs/build/html/python/_autosummary/mlx.core.broadcast_arrays.html new file mode 100644 index 000000000..7742a1d56 --- /dev/null +++ b/docs/build/html/python/_autosummary/mlx.core.broadcast_arrays.html @@ -0,0 +1,1033 @@ + + + + + + + + + + + mlx.core.broadcast_arrays — MLX 0.24.2 documentation + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      + + + + + + + + + + +
      +
      +
      +
      + + + Ctrl+K +
      +
      + +
      + +
      + + +
      +
      + + +
      +
      + + + +
      + + + +
      + + + + +
      + +
      +
      + + + + + +
      +
      + +
      +
      +
      + + +
      +
      + + + + +
      + +
      + + + +
      + +
      +
      + +
      +
      + +
      + +
      + +
      + + +
      + +
      + +
      + + + + + + + + + + + + + + + + +
      + + +
      + + + + + + + + + + + + + +
      + +
      + +
      +
      + + + +
      +

      mlx.core.broadcast_arrays

      + +
      +
      + +
      +

      Contents

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

      mlx.core.broadcast_arrays#

      +
      +
      +broadcast_arrays(*arrays: array, stream: None | Stream | Device = None) → Tuple[array, ...]#
      +

      Broadcast arrays against one another.

      +

      The broadcasting semantics are the same as Numpy.

      +
      +
      Parameters:
      +

      *arrays (array) – The input arrays.

      +
      +
      Returns:
      +

      The output arrays with the broadcasted shape.

      +
      +
      Return type:
      +

      tuple(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.broadcast_to.html b/docs/build/html/python/_autosummary/mlx.core.broadcast_to.html index 644ef9ad9..2730f1559 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.24.1 documentation + mlx.core.broadcast_to — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -45,7 +45,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      @@ -941,12 +945,12 @@ document.write(`

      previous

      -

      mlx.core.block_masked_mm

      +

      mlx.core.broadcast_arrays

       - mlx.core.ceil — MLX 0.24.1 documentation + mlx.core.ceil — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home +
      @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.clear_cache.html b/docs/build/html/python/_autosummary/mlx.core.clear_cache.html index c80dd0a6b..c611db032 100644 --- a/docs/build/html/python/_autosummary/mlx.core.clear_cache.html +++ b/docs/build/html/python/_autosummary/mlx.core.clear_cache.html @@ -8,7 +8,7 @@ - mlx.core.clear_cache — MLX 0.24.1 documentation + mlx.core.clear_cache — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.clip.html b/docs/build/html/python/_autosummary/mlx.core.clip.html index f500c4802..975943e5f 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.24.1 documentation + mlx.core.clip — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.compile.html b/docs/build/html/python/_autosummary/mlx.core.compile.html index ba8b035b3..7c5938b3b 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.24.1 documentation + mlx.core.compile — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -45,7 +45,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      @@ -958,12 +962,12 @@ as fun

      previous

      -

      mlx.core.eval

      +

      mlx.core.async_eval

       - mlx.core.concatenate — MLX 0.24.1 documentation + mlx.core.concatenate — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -44,7 +44,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      @@ -950,11 +954,11 @@ unspecified defaults to

      next

      -

      mlx.core.conj

      +

      mlx.core.contiguous

      diff --git a/docs/build/html/python/_autosummary/mlx.core.conj.html b/docs/build/html/python/_autosummary/mlx.core.conj.html index e59931c9a..0a4e159c7 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.24.1 documentation + mlx.core.conj — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -45,7 +45,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      @@ -938,12 +942,12 @@ Alias for mx.conjugate.

      previous

      -

      mlx.core.concatenate

      +

      mlx.core.contiguous

       - mlx.core.conjugate — MLX 0.24.1 documentation + mlx.core.conjugate — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home +
      @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.contiguous.html b/docs/build/html/python/_autosummary/mlx.core.contiguous.html new file mode 100644 index 000000000..22bcf917d --- /dev/null +++ b/docs/build/html/python/_autosummary/mlx.core.contiguous.html @@ -0,0 +1,1035 @@ + + + + + + + + + + + mlx.core.contiguous — MLX 0.24.2 documentation + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      + + + + + + + + + + +
      +
      +
      +
      + + + Ctrl+K +
      +
      + +
      + +
      + + +
      +
      + + +
      +
      + + + +
      + + + +
      + + + + +
      + +
      +
      + + + + + +
      +
      + +
      +
      +
      + + +
      +
      + + + + +
      + +
      + + + +
      + +
      +
      + +
      +
      + +
      + +
      + +
      + + +
      + +
      + +
      + + + + + + + + + + + + + + + + +
      + + +
      + + + + + + + + + + + + + +
      + +
      + +
      +
      + + + +
      +

      mlx.core.contiguous

      + +
      +
      + +
      +

      Contents

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

      mlx.core.contiguous#

      +
      +
      +contiguous(a: array, /, allow_col_major: bool = False, *, stream: None | Stream | Device = None) → array#
      +

      Force an array to be row contiguous. Copy if necessary.

      +
      +
      Parameters:
      +
        +

      • a (array) – The input to make contiguous

        • +

      • allow_col_major (bool) – Consider column major as contiguous and don’t copy

        • +
      +
      +
      Returns:
      +

      The row or col contiguous output.

      +
      +
      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.conv1d.html b/docs/build/html/python/_autosummary/mlx.core.conv1d.html index c156c7f9a..c8042ebec 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.24.1 documentation + mlx.core.conv1d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.conv2d.html b/docs/build/html/python/_autosummary/mlx.core.conv2d.html index 45f8df40d..ade164757 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.24.1 documentation + mlx.core.conv2d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.conv3d.html b/docs/build/html/python/_autosummary/mlx.core.conv3d.html index 995f1d2da..d1591dc86 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv3d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv3d.html @@ -8,7 +8,7 @@ - mlx.core.conv3d — MLX 0.24.1 documentation + mlx.core.conv3d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 dbef07aa5..330b9e434 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.24.1 documentation + mlx.core.conv_general — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_transpose1d.html b/docs/build/html/python/_autosummary/mlx.core.conv_transpose1d.html index bc9c584ee..c0e801913 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv_transpose1d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv_transpose1d.html @@ -8,7 +8,7 @@ - mlx.core.conv_transpose1d — MLX 0.24.1 documentation + mlx.core.conv_transpose1d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_transpose2d.html b/docs/build/html/python/_autosummary/mlx.core.conv_transpose2d.html index 9a6ac3dc4..43be5c861 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv_transpose2d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv_transpose2d.html @@ -8,7 +8,7 @@ - mlx.core.conv_transpose2d — MLX 0.24.1 documentation + mlx.core.conv_transpose2d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_transpose3d.html b/docs/build/html/python/_autosummary/mlx.core.conv_transpose3d.html index 7da98388c..ce9c3ad1b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conv_transpose3d.html +++ b/docs/build/html/python/_autosummary/mlx.core.conv_transpose3d.html @@ -8,7 +8,7 @@ - mlx.core.conv_transpose3d — MLX 0.24.1 documentation + mlx.core.conv_transpose3d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.convolve.html b/docs/build/html/python/_autosummary/mlx.core.convolve.html index 08390e6e4..6c3279899 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.24.1 documentation + mlx.core.convolve — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.cos.html b/docs/build/html/python/_autosummary/mlx.core.cos.html index f651c3716..0e1e49430 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.24.1 documentation + mlx.core.cos — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.cosh.html b/docs/build/html/python/_autosummary/mlx.core.cosh.html index 997fcbbd1..facb6827e 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.24.1 documentation + mlx.core.cosh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.cummax.html b/docs/build/html/python/_autosummary/mlx.core.cummax.html index 07e2982d4..70e1c0297 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.24.1 documentation + mlx.core.cummax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.cummin.html b/docs/build/html/python/_autosummary/mlx.core.cummin.html index d55e8ce02..49a429fde 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.24.1 documentation + mlx.core.cummin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.cumprod.html b/docs/build/html/python/_autosummary/mlx.core.cumprod.html index 218fd3185..504d8f5a5 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.24.1 documentation + mlx.core.cumprod — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.cumsum.html b/docs/build/html/python/_autosummary/mlx.core.cumsum.html index 034572522..06c1e3a46 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.24.1 documentation + mlx.core.cumsum — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a16cef956..eeb0a6267 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.24.1 documentation + mlx.core.custom_function — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 8b5f39bc5..eff98d49c 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.24.1 documentation + mlx.core.default_device — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 d2c6900e2..6e38e8d09 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.24.1 documentation + mlx.core.default_stream — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.degrees.html b/docs/build/html/python/_autosummary/mlx.core.degrees.html index 40692d4ac..308ef963f 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.24.1 documentation + mlx.core.degrees — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.dequantize.html b/docs/build/html/python/_autosummary/mlx.core.dequantize.html index 24491a305..5ef9ad8d5 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.24.1 documentation + mlx.core.dequantize — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.diag.html b/docs/build/html/python/_autosummary/mlx.core.diag.html index 3ee8617a7..9ed862868 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.24.1 documentation + mlx.core.diag — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.diagonal.html b/docs/build/html/python/_autosummary/mlx.core.diagonal.html index 845a230a9..e049eb3d6 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.24.1 documentation + mlx.core.diagonal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 22661f21f..a261e7713 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.24.1 documentation + mlx.core.disable_compile — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 75f56aa59..7037828b9 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.24.1 documentation + mlx.core.distributed.Group — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 413aa1abd..1d55f3115 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.24.1 documentation + mlx.core.distributed.all_gather — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 260a4d5f5..4a851662c 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.24.1 documentation + mlx.core.distributed.all_sum — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 dd201eafc..605b2ee42 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.24.1 documentation + mlx.core.distributed.init — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 062065f0d..2c1215095 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.24.1 documentation + mlx.core.distributed.is_available — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.recv.html b/docs/build/html/python/_autosummary/mlx.core.distributed.recv.html index 3bd37aedd..e319e99e6 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.recv.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.recv.html @@ -8,7 +8,7 @@ - mlx.core.distributed.recv — MLX 0.24.1 documentation + mlx.core.distributed.recv — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.recv_like.html b/docs/build/html/python/_autosummary/mlx.core.distributed.recv_like.html index 0662bed84..b752ece59 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.recv_like.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.recv_like.html @@ -8,7 +8,7 @@ - mlx.core.distributed.recv_like — MLX 0.24.1 documentation + mlx.core.distributed.recv_like — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.send.html b/docs/build/html/python/_autosummary/mlx.core.distributed.send.html index 703ad7818..be89920ea 100644 --- a/docs/build/html/python/_autosummary/mlx.core.distributed.send.html +++ b/docs/build/html/python/_autosummary/mlx.core.distributed.send.html @@ -8,7 +8,7 @@ - mlx.core.distributed.send — MLX 0.24.1 documentation + mlx.core.distributed.send — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.divide.html b/docs/build/html/python/_autosummary/mlx.core.divide.html index 9aa99fffd..eebefc2ab 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.24.1 documentation + mlx.core.divide — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.divmod.html b/docs/build/html/python/_autosummary/mlx.core.divmod.html index c8279f6e1..6a9fb1feb 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.24.1 documentation + mlx.core.divmod — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.einsum.html b/docs/build/html/python/_autosummary/mlx.core.einsum.html index 4037125fc..446454fad 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.24.1 documentation + mlx.core.einsum — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 9db2a7091..efefbc593 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.24.1 documentation + mlx.core.einsum_path — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 dcf73a024..a1cce79db 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.24.1 documentation + mlx.core.enable_compile — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.equal.html b/docs/build/html/python/_autosummary/mlx.core.equal.html index 998862ee9..5f1d22d29 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.24.1 documentation + mlx.core.equal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.erf.html b/docs/build/html/python/_autosummary/mlx.core.erf.html index 990301d0d..4abde814f 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.24.1 documentation + mlx.core.erf — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.erfinv.html b/docs/build/html/python/_autosummary/mlx.core.erfinv.html index 95af064fb..640c26584 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.24.1 documentation + mlx.core.erfinv — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.eval.html b/docs/build/html/python/_autosummary/mlx.core.eval.html index c26335265..e22bf7666 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.24.1 documentation + mlx.core.eval — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -44,7 +44,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      @@ -943,11 +947,11 @@ arrays are ignored.

      next

      -

      mlx.core.compile

      +

      mlx.core.async_eval

       diff --git a/docs/build/html/python/_autosummary/mlx.core.exp.html b/docs/build/html/python/_autosummary/mlx.core.exp.html index 31cc8c498..6603e4574 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.24.1 documentation + mlx.core.exp — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 bda1a2c7e..1cc54fb0f 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.24.1 documentation + mlx.core.expand_dims — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.expm1.html b/docs/build/html/python/_autosummary/mlx.core.expm1.html index fd0fa95cb..d3454ab64 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.24.1 documentation + mlx.core.expm1 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.export_function.html b/docs/build/html/python/_autosummary/mlx.core.export_function.html index a5210a0b7..d377085e8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.export_function.html +++ b/docs/build/html/python/_autosummary/mlx.core.export_function.html @@ -8,7 +8,7 @@ - mlx.core.export_function — MLX 0.24.1 documentation + mlx.core.export_function — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.export_to_dot.html b/docs/build/html/python/_autosummary/mlx.core.export_to_dot.html index 562a766e2..7aa4f769b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.export_to_dot.html +++ b/docs/build/html/python/_autosummary/mlx.core.export_to_dot.html @@ -8,7 +8,7 @@ - mlx.core.export_to_dot — MLX 0.24.1 documentation + mlx.core.export_to_dot — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.exporter.html b/docs/build/html/python/_autosummary/mlx.core.exporter.html index 6e75a423f..89d63a76d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.exporter.html +++ b/docs/build/html/python/_autosummary/mlx.core.exporter.html @@ -8,7 +8,7 @@ - mlx.core.exporter — MLX 0.24.1 documentation + mlx.core.exporter — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.eye.html b/docs/build/html/python/_autosummary/mlx.core.eye.html index 92e564284..cbc49fdd3 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.24.1 documentation + mlx.core.eye — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 5bb0f5923..c18d93085 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.24.1 documentation + mlx.core.fast.layer_norm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.metal_kernel.html b/docs/build/html/python/_autosummary/mlx.core.fast.metal_kernel.html index 6a73faed7..ec8c7addc 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.metal_kernel.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.metal_kernel.html @@ -8,7 +8,7 @@ - mlx.core.fast.metal_kernel — MLX 0.24.1 documentation + mlx.core.fast.metal_kernel — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 034cd407c..59d225e01 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.24.1 documentation + mlx.core.fast.rms_norm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 fa1588536..353282d69 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.24.1 documentation + mlx.core.fast.rope — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 8e04b046b..5e80f765f 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.24.1 documentation + mlx.core.fast.scaled_dot_product_attention — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 1ac39d70a..5ad985a7a 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.24.1 documentation + mlx.core.fft.fft — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 bd8e5496c..e4a3d0bff 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.24.1 documentation + mlx.core.fft.fft2 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f351e6d00..444c139e4 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.24.1 documentation + mlx.core.fft.fftn — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 143545157..cfc253131 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.24.1 documentation + mlx.core.fft.ifft — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e46ae5c7d..2c6c9a00a 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.24.1 documentation + mlx.core.fft.ifft2 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c83b0c4ee..7ecfd1321 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.24.1 documentation + mlx.core.fft.ifftn — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e94ea6c7c..da3dcb248 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.24.1 documentation + mlx.core.fft.irfft — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f3f653135..97776d3f8 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.24.1 documentation + mlx.core.fft.irfft2 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 aee9b2864..b499c5104 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.24.1 documentation + mlx.core.fft.irfftn — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e69c4bf73..9a3140350 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.24.1 documentation + mlx.core.fft.rfft — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 054f964ce..07121aefe 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.24.1 documentation + mlx.core.fft.rfft2 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 dd62c5564..0c167c914 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.24.1 documentation + mlx.core.fft.rfftn — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.finfo.html b/docs/build/html/python/_autosummary/mlx.core.finfo.html index f6614a082..46c84a687 100644 --- a/docs/build/html/python/_autosummary/mlx.core.finfo.html +++ b/docs/build/html/python/_autosummary/mlx.core.finfo.html @@ -8,7 +8,7 @@ - mlx.core.finfo — MLX 0.24.1 documentation + mlx.core.finfo — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.flatten.html b/docs/build/html/python/_autosummary/mlx.core.flatten.html index 912cc4060..c0de396d3 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.24.1 documentation + mlx.core.flatten — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.floor.html b/docs/build/html/python/_autosummary/mlx.core.floor.html index 771e1650e..27e957bde 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.24.1 documentation + mlx.core.floor — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e1ef3b7b9..7cbaae700 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.24.1 documentation + mlx.core.floor_divide — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.full.html b/docs/build/html/python/_autosummary/mlx.core.full.html index 319ec15ed..6d920cbc1 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.24.1 documentation + mlx.core.full — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 001fa4100..2c6c5db9e 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.24.1 documentation + mlx.core.gather_mm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6abe4df7c..41e92d4a0 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.24.1 documentation + mlx.core.gather_qmm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.get_active_memory.html b/docs/build/html/python/_autosummary/mlx.core.get_active_memory.html index d3ea1693c..95ac52324 100644 --- a/docs/build/html/python/_autosummary/mlx.core.get_active_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.get_active_memory.html @@ -8,7 +8,7 @@ - mlx.core.get_active_memory — MLX 0.24.1 documentation + mlx.core.get_active_memory — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.get_cache_memory.html b/docs/build/html/python/_autosummary/mlx.core.get_cache_memory.html index ff2e1c330..4eb255581 100644 --- a/docs/build/html/python/_autosummary/mlx.core.get_cache_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.get_cache_memory.html @@ -8,7 +8,7 @@ - mlx.core.get_cache_memory — MLX 0.24.1 documentation + mlx.core.get_cache_memory — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.get_peak_memory.html b/docs/build/html/python/_autosummary/mlx.core.get_peak_memory.html index 5069279a4..1704f4768 100644 --- a/docs/build/html/python/_autosummary/mlx.core.get_peak_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.get_peak_memory.html @@ -8,7 +8,7 @@ - mlx.core.get_peak_memory — MLX 0.24.1 documentation + mlx.core.get_peak_memory — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.grad.html b/docs/build/html/python/_autosummary/mlx.core.grad.html index 3822ef735..8a40e1ba1 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.24.1 documentation + mlx.core.grad — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.greater.html b/docs/build/html/python/_autosummary/mlx.core.greater.html index bedbf6a5c..680b22050 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.24.1 documentation + mlx.core.greater — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 cb6a45e1e..457b616d3 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.24.1 documentation + mlx.core.greater_equal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e46146c1b..d75c0774f 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.24.1 documentation + mlx.core.hadamard_transform — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.identity.html b/docs/build/html/python/_autosummary/mlx.core.identity.html index f16d40079..716838fa1 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.24.1 documentation + mlx.core.identity — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.imag.html b/docs/build/html/python/_autosummary/mlx.core.imag.html index af041d3c3..2fa6056dd 100644 --- a/docs/build/html/python/_autosummary/mlx.core.imag.html +++ b/docs/build/html/python/_autosummary/mlx.core.imag.html @@ -8,7 +8,7 @@ - mlx.core.imag — MLX 0.24.1 documentation + mlx.core.imag — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.import_function.html b/docs/build/html/python/_autosummary/mlx.core.import_function.html index f798bccb0..01f7614b1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.import_function.html +++ b/docs/build/html/python/_autosummary/mlx.core.import_function.html @@ -8,7 +8,7 @@ - mlx.core.import_function — MLX 0.24.1 documentation + mlx.core.import_function — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.inner.html b/docs/build/html/python/_autosummary/mlx.core.inner.html index 5fdc8b6cd..8c42fcf91 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.24.1 documentation + mlx.core.inner — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.isclose.html b/docs/build/html/python/_autosummary/mlx.core.isclose.html index 0d0c2e2f8..d6a9d3577 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.24.1 documentation + mlx.core.isclose — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.isfinite.html b/docs/build/html/python/_autosummary/mlx.core.isfinite.html index 4d7694501..e270f3aa3 100644 --- a/docs/build/html/python/_autosummary/mlx.core.isfinite.html +++ b/docs/build/html/python/_autosummary/mlx.core.isfinite.html @@ -8,7 +8,7 @@ - mlx.core.isfinite — MLX 0.24.1 documentation + mlx.core.isfinite — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.isinf.html b/docs/build/html/python/_autosummary/mlx.core.isinf.html index f268570ee..b85c78742 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.24.1 documentation + mlx.core.isinf — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.isnan.html b/docs/build/html/python/_autosummary/mlx.core.isnan.html index 0d29958fb..316b79847 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.24.1 documentation + mlx.core.isnan — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.isneginf.html b/docs/build/html/python/_autosummary/mlx.core.isneginf.html index a000d2c33..49f1fc6a1 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.24.1 documentation + mlx.core.isneginf — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.isposinf.html b/docs/build/html/python/_autosummary/mlx.core.isposinf.html index 1d3bc76d5..4934227b2 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.24.1 documentation + mlx.core.isposinf — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.issubdtype.html b/docs/build/html/python/_autosummary/mlx.core.issubdtype.html index a1daf6db8..dbf0045ab 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.24.1 documentation + mlx.core.issubdtype — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.jvp.html b/docs/build/html/python/_autosummary/mlx.core.jvp.html index 3314ce9f1..44d08ed16 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.24.1 documentation + mlx.core.jvp — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.kron.html b/docs/build/html/python/_autosummary/mlx.core.kron.html index 51693dfe9..9adc29731 100644 --- a/docs/build/html/python/_autosummary/mlx.core.kron.html +++ b/docs/build/html/python/_autosummary/mlx.core.kron.html @@ -8,7 +8,7 @@ - mlx.core.kron — MLX 0.24.1 documentation + mlx.core.kron — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 2b7223b27..d4d9ab403 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.24.1 documentation + mlx.core.left_shift — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.less.html b/docs/build/html/python/_autosummary/mlx.core.less.html index c92f68428..a3764dc02 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.24.1 documentation + mlx.core.less — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 2f3dff003..b618f2979 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.24.1 documentation + mlx.core.less_equal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 bd82dbd2d..0b70e7d2e 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.24.1 documentation + mlx.core.linalg.cholesky — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 index 3ab16f4cb..354c5a4b7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky_inv.html @@ -8,7 +8,7 @@ - mlx.core.linalg.cholesky_inv — MLX 0.24.1 documentation + mlx.core.linalg.cholesky_inv — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.cross.html b/docs/build/html/python/_autosummary/mlx.core.linalg.cross.html index ef6b96ce9..0d7da5604 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.cross.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.cross.html @@ -8,7 +8,7 @@ - mlx.core.linalg.cross — MLX 0.24.1 documentation + mlx.core.linalg.cross — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.eigh.html b/docs/build/html/python/_autosummary/mlx.core.linalg.eigh.html index 6c30403ea..ec25f8be8 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.eigh.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.eigh.html @@ -8,7 +8,7 @@ - mlx.core.linalg.eigh — MLX 0.24.1 documentation + mlx.core.linalg.eigh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.eigvalsh.html b/docs/build/html/python/_autosummary/mlx.core.linalg.eigvalsh.html index 1ca88eb11..da9d37118 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.eigvalsh.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.eigvalsh.html @@ -8,7 +8,7 @@ - mlx.core.linalg.eigvalsh — MLX 0.24.1 documentation + mlx.core.linalg.eigvalsh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a97d20d4b..39fb9996a 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.24.1 documentation + mlx.core.linalg.inv — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.lu.html b/docs/build/html/python/_autosummary/mlx.core.linalg.lu.html index bf686e50d..60af16048 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.lu.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.lu.html @@ -8,7 +8,7 @@ - mlx.core.linalg.lu — MLX 0.24.1 documentation + mlx.core.linalg.lu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.lu_factor.html b/docs/build/html/python/_autosummary/mlx.core.linalg.lu_factor.html index 6e4a8e130..90b7b4f77 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.lu_factor.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.lu_factor.html @@ -8,7 +8,7 @@ - mlx.core.linalg.lu_factor — MLX 0.24.1 documentation + mlx.core.linalg.lu_factor — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -44,7 +44,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      @@ -950,11 +954,11 @@ in which case the default stream of the default device is used.

      • next

      -

      mlx.core.linalg.solve

      +

      mlx.core.linalg.pinv

       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 2bf587093..52b70e759 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.24.1 documentation + mlx.core.linalg.norm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.pinv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.pinv.html new file mode 100644 index 000000000..c527899d0 --- /dev/null +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.pinv.html @@ -0,0 +1,1040 @@ + + + + + + + + + + + mlx.core.linalg.pinv — MLX 0.24.2 documentation + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
      + + + + + + + + + + +
      +
      +
      +
      + + + Ctrl+K +
      +
      + +
      + +
      + + +
      +
      + + +
      +
      + + + +
      + + + +
      + + + + +
      + +
      +
      + + + + + +
      +
      + +
      +
      +
      + + +
      +
      + + + + +
      + +
      + + + +
      + +
      +
      + +
      +
      + +
      + +
      + +
      + + +
      + +
      + +
      + + + + + + + + + + + + + + + + +
      + + +
      + + + + + + + + + + + + + +
      + +
      + +
      +
      + + + +
      +

      mlx.core.linalg.pinv

      + +
      +
      + +
      +

      Contents

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

      mlx.core.linalg.pinv#

      +
      +
      +pinv(a: array, *, stream: None | Stream | Device = None) → array#
      +

      Compute the (Moore-Penrose) pseudo-inverse of a matrix.

      +

      This function calculates a generalized inverse of a matrix using its +singular-value decomposition. 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.

        • +

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

        • +
      +
      +
      Returns:
      +

      aplus such that a @ aplus @ a = a

      +
      +
      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.qr.html b/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html index ba5bc69e5..60a8663f4 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html @@ -8,7 +8,7 @@ - mlx.core.linalg.qr — MLX 0.24.1 documentation + mlx.core.linalg.qr — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.solve.html b/docs/build/html/python/_autosummary/mlx.core.linalg.solve.html index 296511c4c..9d2e92738 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.solve.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.solve.html @@ -8,7 +8,7 @@ - mlx.core.linalg.solve — MLX 0.24.1 documentation + mlx.core.linalg.solve — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -45,7 +45,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      @@ -942,12 +946,12 @@ in which case the default stream of the default device is used.

      • previous

      -

      mlx.core.linalg.lu_factor

      +

      mlx.core.linalg.pinv

       - mlx.core.linalg.solve_triangular — MLX 0.24.1 documentation + mlx.core.linalg.solve_triangular — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home +
      @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html b/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html index ff38abbb8..ffc8f3328 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html @@ -8,7 +8,7 @@ - mlx.core.linalg.svd — MLX 0.24.1 documentation + mlx.core.linalg.svd — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 index cdc2168e4..e351511fd 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.tri_inv.html @@ -8,7 +8,7 @@ - mlx.core.linalg.tri_inv — MLX 0.24.1 documentation + mlx.core.linalg.tri_inv — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.linspace.html b/docs/build/html/python/_autosummary/mlx.core.linspace.html index 5e003f827..c7072bc7e 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.24.1 documentation + mlx.core.linspace — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.load.html b/docs/build/html/python/_autosummary/mlx.core.load.html index 314012c3c..4b134e9d9 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.24.1 documentation + mlx.core.load — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.log.html b/docs/build/html/python/_autosummary/mlx.core.log.html index 9ba8478aa..4db728f0b 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.24.1 documentation + mlx.core.log — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.log10.html b/docs/build/html/python/_autosummary/mlx.core.log10.html index a8653ff81..c1cf3c809 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.24.1 documentation + mlx.core.log10 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.log1p.html b/docs/build/html/python/_autosummary/mlx.core.log1p.html index 56ebe4b86..7521b9a69 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.24.1 documentation + mlx.core.log1p — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.log2.html b/docs/build/html/python/_autosummary/mlx.core.log2.html index 22115e074..1a561292f 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.24.1 documentation + mlx.core.log2 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.logaddexp.html b/docs/build/html/python/_autosummary/mlx.core.logaddexp.html index 755206da8..1540497d7 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.24.1 documentation + mlx.core.logaddexp — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 688e5b63e..83df747a4 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.24.1 documentation + mlx.core.logical_and — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f4c40ecd1..5aa4f5787 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.24.1 documentation + mlx.core.logical_not — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 5886c80d7..39408f678 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.24.1 documentation + mlx.core.logical_or — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.logsumexp.html b/docs/build/html/python/_autosummary/mlx.core.logsumexp.html index 9bfbf4ddf..71f5af677 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.24.1 documentation + mlx.core.logsumexp — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.matmul.html b/docs/build/html/python/_autosummary/mlx.core.matmul.html index ed2618bb6..10d809170 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.24.1 documentation + mlx.core.matmul — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.max.html b/docs/build/html/python/_autosummary/mlx.core.max.html index d89bc2c9d..082c953c3 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.24.1 documentation + mlx.core.max — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.maximum.html b/docs/build/html/python/_autosummary/mlx.core.maximum.html index e20eab6a6..4028a74f9 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.24.1 documentation + mlx.core.maximum — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.mean.html b/docs/build/html/python/_autosummary/mlx.core.mean.html index 80512adbb..3c1f4620f 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.24.1 documentation + mlx.core.mean — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.meshgrid.html b/docs/build/html/python/_autosummary/mlx.core.meshgrid.html index dc56e3cde..e202abe89 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.24.1 documentation + mlx.core.meshgrid — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6a1d9e3b1..c93d3c2be 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.24.1 documentation + mlx.core.metal.device_info — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 ca037b2c9..90b2ce652 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.24.1 documentation + mlx.core.metal.is_available — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 71daf6797..90779772b 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.24.1 documentation + mlx.core.metal.start_capture — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 17bd1ec29..06ea909a1 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.24.1 documentation + mlx.core.metal.stop_capture — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.min.html b/docs/build/html/python/_autosummary/mlx.core.min.html index 4be341fca..c4a210f43 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.24.1 documentation + mlx.core.min — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.minimum.html b/docs/build/html/python/_autosummary/mlx.core.minimum.html index f61c21022..f91d8b608 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.24.1 documentation + mlx.core.minimum — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.moveaxis.html b/docs/build/html/python/_autosummary/mlx.core.moveaxis.html index a4e917204..fad75a46b 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.24.1 documentation + mlx.core.moveaxis — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.multiply.html b/docs/build/html/python/_autosummary/mlx.core.multiply.html index 32f37db89..f47286841 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.24.1 documentation + mlx.core.multiply — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f67f312b2..b241f67a5 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.24.1 documentation + mlx.core.nan_to_num — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.negative.html b/docs/build/html/python/_autosummary/mlx.core.negative.html index a9546d6ea..4645701e0 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.24.1 documentation + mlx.core.negative — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 143f58aa0..90336fa8a 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.24.1 documentation + mlx.core.new_stream — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6a280b438..983a7e02a 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.24.1 documentation + mlx.core.not_equal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.ones.html b/docs/build/html/python/_autosummary/mlx.core.ones.html index 16a1c119d..cb7ff1247 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.24.1 documentation + mlx.core.ones — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 854b7f38d..1bbc92f4f 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.24.1 documentation + mlx.core.ones_like — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.outer.html b/docs/build/html/python/_autosummary/mlx.core.outer.html index ad55e1946..877de80fe 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.24.1 documentation + mlx.core.outer — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.pad.html b/docs/build/html/python/_autosummary/mlx.core.pad.html index 04c8a441c..9174ea08a 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.24.1 documentation + mlx.core.pad — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.partition.html b/docs/build/html/python/_autosummary/mlx.core.partition.html index a70aa369c..0eb4b4186 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.24.1 documentation + mlx.core.partition — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.power.html b/docs/build/html/python/_autosummary/mlx.core.power.html index 7150962a0..5933fe7bf 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.24.1 documentation + mlx.core.power — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.prod.html b/docs/build/html/python/_autosummary/mlx.core.prod.html index 226c2310b..0c10f2688 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.24.1 documentation + mlx.core.prod — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.put_along_axis.html b/docs/build/html/python/_autosummary/mlx.core.put_along_axis.html index e4b3089ee..0a8c3932b 100644 --- a/docs/build/html/python/_autosummary/mlx.core.put_along_axis.html +++ b/docs/build/html/python/_autosummary/mlx.core.put_along_axis.html @@ -8,7 +8,7 @@ - mlx.core.put_along_axis — MLX 0.24.1 documentation + mlx.core.put_along_axis — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.quantize.html b/docs/build/html/python/_autosummary/mlx.core.quantize.html index f2d01e4f8..e7c8ae74f 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.24.1 documentation + mlx.core.quantize — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 ef2a4670c..55015d26c 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.24.1 documentation + mlx.core.quantized_matmul — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.radians.html b/docs/build/html/python/_autosummary/mlx.core.radians.html index 25366a7de..590deace0 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.24.1 documentation + mlx.core.radians — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 7dcde74e8..5bb7f9139 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.24.1 documentation + mlx.core.random.bernoulli — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f93aa15fa..b18fedd0a 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.24.1 documentation + mlx.core.random.categorical — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 5ee563742..2269e10e4 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.24.1 documentation + mlx.core.random.gumbel — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e6cf5c1c2..0c7a08e8d 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.24.1 documentation + mlx.core.random.key — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3f5325c41..507fc4930 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.24.1 documentation + mlx.core.random.laplace — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3baee308f..bb70da640 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.24.1 documentation + mlx.core.random.multivariate_normal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 710d653ba..9c54bf156 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.24.1 documentation + mlx.core.random.normal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.random.permutation.html b/docs/build/html/python/_autosummary/mlx.core.random.permutation.html index 97a7bdb47..2eea5976c 100644 --- a/docs/build/html/python/_autosummary/mlx.core.random.permutation.html +++ b/docs/build/html/python/_autosummary/mlx.core.random.permutation.html @@ -8,7 +8,7 @@ - mlx.core.random.permutation — MLX 0.24.1 documentation + mlx.core.random.permutation — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 2694311f8..f098bc999 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.24.1 documentation + mlx.core.random.randint — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 40cc43cf0..e2e540e5a 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.24.1 documentation + mlx.core.random.seed — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 275f049a3..af21d57c3 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.24.1 documentation + mlx.core.random.split — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 01fe50745..88c7f52cd 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.24.1 documentation + mlx.core.random.truncated_normal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 0c04e2e18..4bc0cfabe 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.24.1 documentation + mlx.core.random.uniform — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.real.html b/docs/build/html/python/_autosummary/mlx.core.real.html index 11c2f460f..5967c93d9 100644 --- a/docs/build/html/python/_autosummary/mlx.core.real.html +++ b/docs/build/html/python/_autosummary/mlx.core.real.html @@ -8,7 +8,7 @@ - mlx.core.real — MLX 0.24.1 documentation + mlx.core.real — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.reciprocal.html b/docs/build/html/python/_autosummary/mlx.core.reciprocal.html index 70af160b5..964ffae69 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.24.1 documentation + mlx.core.reciprocal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.remainder.html b/docs/build/html/python/_autosummary/mlx.core.remainder.html index c2ec46759..2c937949e 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.24.1 documentation + mlx.core.remainder — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.repeat.html b/docs/build/html/python/_autosummary/mlx.core.repeat.html index c9b50580e..2c5a79b85 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.24.1 documentation + mlx.core.repeat — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.reset_peak_memory.html b/docs/build/html/python/_autosummary/mlx.core.reset_peak_memory.html index bd64f05b6..ed50d2eb2 100644 --- a/docs/build/html/python/_autosummary/mlx.core.reset_peak_memory.html +++ b/docs/build/html/python/_autosummary/mlx.core.reset_peak_memory.html @@ -8,7 +8,7 @@ - mlx.core.reset_peak_memory — MLX 0.24.1 documentation + mlx.core.reset_peak_memory — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.reshape.html b/docs/build/html/python/_autosummary/mlx.core.reshape.html index 9bbb72a12..d570336d0 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.24.1 documentation + mlx.core.reshape — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 28dc36802..8720753bb 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.24.1 documentation + mlx.core.right_shift — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.roll.html b/docs/build/html/python/_autosummary/mlx.core.roll.html index 5756bba64..6443e7bf0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.roll.html +++ b/docs/build/html/python/_autosummary/mlx.core.roll.html @@ -8,7 +8,7 @@ - mlx.core.roll — MLX 0.24.1 documentation + mlx.core.roll — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.round.html b/docs/build/html/python/_autosummary/mlx.core.round.html index ef0ac7a86..1a0333884 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.24.1 documentation + mlx.core.round — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.rsqrt.html b/docs/build/html/python/_autosummary/mlx.core.rsqrt.html index 63636db81..67258125e 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.24.1 documentation + mlx.core.rsqrt — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.save.html b/docs/build/html/python/_autosummary/mlx.core.save.html index 7b95c1c97..57a041f11 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.24.1 documentation + mlx.core.save — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f72a0fac6..560a6fb05 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.24.1 documentation + mlx.core.save_gguf — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 53b7e594f..435ceed4b 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.24.1 documentation + mlx.core.save_safetensors — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.savez.html b/docs/build/html/python/_autosummary/mlx.core.savez.html index 18d463dea..5b84d0567 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.24.1 documentation + mlx.core.savez — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 1a36101ed..932db2906 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.24.1 documentation + mlx.core.savez_compressed — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.set_cache_limit.html b/docs/build/html/python/_autosummary/mlx.core.set_cache_limit.html index 528935219..c67aab81d 100644 --- a/docs/build/html/python/_autosummary/mlx.core.set_cache_limit.html +++ b/docs/build/html/python/_autosummary/mlx.core.set_cache_limit.html @@ -8,7 +8,7 @@ - mlx.core.set_cache_limit — MLX 0.24.1 documentation + mlx.core.set_cache_limit — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 f8706d2fe..fd2a87713 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.24.1 documentation + mlx.core.set_default_device — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e847298f0..77e44a84b 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.24.1 documentation + mlx.core.set_default_stream — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.set_memory_limit.html b/docs/build/html/python/_autosummary/mlx.core.set_memory_limit.html index 62e5f80db..48a07f90f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.set_memory_limit.html +++ b/docs/build/html/python/_autosummary/mlx.core.set_memory_limit.html @@ -8,7 +8,7 @@ - mlx.core.set_memory_limit — MLX 0.24.1 documentation + mlx.core.set_memory_limit — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.set_wired_limit.html b/docs/build/html/python/_autosummary/mlx.core.set_wired_limit.html index 8b0c708e0..212ad55ad 100644 --- a/docs/build/html/python/_autosummary/mlx.core.set_wired_limit.html +++ b/docs/build/html/python/_autosummary/mlx.core.set_wired_limit.html @@ -8,7 +8,7 @@ - mlx.core.set_wired_limit — MLX 0.24.1 documentation + mlx.core.set_wired_limit — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.sigmoid.html b/docs/build/html/python/_autosummary/mlx.core.sigmoid.html index ffb8b6c99..98755927f 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.24.1 documentation + mlx.core.sigmoid — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.sign.html b/docs/build/html/python/_autosummary/mlx.core.sign.html index 7f14306cc..dae37d641 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.24.1 documentation + mlx.core.sign — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.sin.html b/docs/build/html/python/_autosummary/mlx.core.sin.html index 0dfa14b97..5713a8a13 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.24.1 documentation + mlx.core.sin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.sinh.html b/docs/build/html/python/_autosummary/mlx.core.sinh.html index b0af40c31..3636098b5 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.24.1 documentation + mlx.core.sinh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.slice.html b/docs/build/html/python/_autosummary/mlx.core.slice.html index 8f97a513a..e8ba36e6e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.slice.html +++ b/docs/build/html/python/_autosummary/mlx.core.slice.html @@ -8,7 +8,7 @@ - mlx.core.slice — MLX 0.24.1 documentation + mlx.core.slice — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.slice_update.html b/docs/build/html/python/_autosummary/mlx.core.slice_update.html index 28d22af3f..90cc24f61 100644 --- a/docs/build/html/python/_autosummary/mlx.core.slice_update.html +++ b/docs/build/html/python/_autosummary/mlx.core.slice_update.html @@ -8,7 +8,7 @@ - mlx.core.slice_update — MLX 0.24.1 documentation + mlx.core.slice_update — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.softmax.html b/docs/build/html/python/_autosummary/mlx.core.softmax.html index fc74570a7..b28905f24 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.24.1 documentation + mlx.core.softmax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.sort.html b/docs/build/html/python/_autosummary/mlx.core.sort.html index 7841806cc..21107f0c5 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.24.1 documentation + mlx.core.sort — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.split.html b/docs/build/html/python/_autosummary/mlx.core.split.html index ac4d01bbf..08c7f1f23 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.24.1 documentation + mlx.core.split — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.sqrt.html b/docs/build/html/python/_autosummary/mlx.core.sqrt.html index 277258fcb..ff0b80c35 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.24.1 documentation + mlx.core.sqrt — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.square.html b/docs/build/html/python/_autosummary/mlx.core.square.html index c3500cb69..3904d65a4 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.24.1 documentation + mlx.core.square — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.squeeze.html b/docs/build/html/python/_autosummary/mlx.core.squeeze.html index 53ff5cbc4..987a66882 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.24.1 documentation + mlx.core.squeeze — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.stack.html b/docs/build/html/python/_autosummary/mlx.core.stack.html index 75804e925..8bf66fd24 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.24.1 documentation + mlx.core.stack — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.std.html b/docs/build/html/python/_autosummary/mlx.core.std.html index d43213935..998a18457 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.24.1 documentation + mlx.core.std — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 71d582497..f85e19220 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.24.1 documentation + mlx.core.stop_gradient — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.stream.html b/docs/build/html/python/_autosummary/mlx.core.stream.html index ac2203094..d388864a0 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.24.1 documentation + mlx.core.stream — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.subtract.html b/docs/build/html/python/_autosummary/mlx.core.subtract.html index 57118d029..91df3cb84 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.24.1 documentation + mlx.core.subtract — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.sum.html b/docs/build/html/python/_autosummary/mlx.core.sum.html index 3abc7e321..d3f2ee8f8 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.24.1 documentation + mlx.core.sum — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.swapaxes.html b/docs/build/html/python/_autosummary/mlx.core.swapaxes.html index c6666e864..d9286257e 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.24.1 documentation + mlx.core.swapaxes — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.synchronize.html b/docs/build/html/python/_autosummary/mlx.core.synchronize.html index 6fdbd50e2..3b788037f 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.24.1 documentation + mlx.core.synchronize — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.take.html b/docs/build/html/python/_autosummary/mlx.core.take.html index ab9f984f5..9e776f847 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.24.1 documentation + mlx.core.take — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e1b80abe8..6d57694eb 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.24.1 documentation + mlx.core.take_along_axis — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.tan.html b/docs/build/html/python/_autosummary/mlx.core.tan.html index 20495d7f0..0f86b8603 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.24.1 documentation + mlx.core.tan — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.tanh.html b/docs/build/html/python/_autosummary/mlx.core.tanh.html index 4bf3f4126..abf6932ce 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.24.1 documentation + mlx.core.tanh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.tensordot.html b/docs/build/html/python/_autosummary/mlx.core.tensordot.html index 01774db21..3c618ce6d 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.24.1 documentation + mlx.core.tensordot — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.tile.html b/docs/build/html/python/_autosummary/mlx.core.tile.html index 149076046..1bfdd9ac7 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.24.1 documentation + mlx.core.tile — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.topk.html b/docs/build/html/python/_autosummary/mlx.core.topk.html index 8ccb8031b..1e817683b 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.24.1 documentation + mlx.core.topk — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.trace.html b/docs/build/html/python/_autosummary/mlx.core.trace.html index 9f92261b3..e04e41f7c 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.24.1 documentation + mlx.core.trace — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.transpose.html b/docs/build/html/python/_autosummary/mlx.core.transpose.html index 0be746a6c..089f0be6d 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.24.1 documentation + mlx.core.transpose — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.tri.html b/docs/build/html/python/_autosummary/mlx.core.tri.html index 52a05ccc8..b2f62fde1 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.24.1 documentation + mlx.core.tri — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.tril.html b/docs/build/html/python/_autosummary/mlx.core.tril.html index 2d0c558c6..0c9eff830 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.24.1 documentation + mlx.core.tril — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.triu.html b/docs/build/html/python/_autosummary/mlx.core.triu.html index 4977edce7..a2d5ca336 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.24.1 documentation + mlx.core.triu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.unflatten.html b/docs/build/html/python/_autosummary/mlx.core.unflatten.html index 433ddf79a..97f2b555f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.unflatten.html +++ b/docs/build/html/python/_autosummary/mlx.core.unflatten.html @@ -8,7 +8,7 @@ - mlx.core.unflatten — MLX 0.24.1 documentation + mlx.core.unflatten — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c86106432..d4429a2a2 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.24.1 documentation + mlx.core.value_and_grad — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.var.html b/docs/build/html/python/_autosummary/mlx.core.var.html index f1a6bb5c8..ea608efd7 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.24.1 documentation + mlx.core.var — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.view.html b/docs/build/html/python/_autosummary/mlx.core.view.html index 5daac72d6..42dee18d6 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.24.1 documentation + mlx.core.view — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.vjp.html b/docs/build/html/python/_autosummary/mlx.core.vjp.html index e05fcb969..17df1e350 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.24.1 documentation + mlx.core.vjp — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.vmap.html b/docs/build/html/python/_autosummary/mlx.core.vmap.html index 997949c7d..db1e0a3b8 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.24.1 documentation + mlx.core.vmap — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.where.html b/docs/build/html/python/_autosummary/mlx.core.where.html index 325f03822..f73d96c5b 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.24.1 documentation + mlx.core.where — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.core.zeros.html b/docs/build/html/python/_autosummary/mlx.core.zeros.html index 604e80b65..50bfa3941 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.24.1 documentation + mlx.core.zeros — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c46850235..7c1fc7a88 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.24.1 documentation + mlx.core.zeros_like — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.nn.average_gradients.html b/docs/build/html/python/_autosummary/mlx.nn.average_gradients.html index c8be7ea2f..086aeb04e 100644 --- a/docs/build/html/python/_autosummary/mlx.nn.average_gradients.html +++ b/docs/build/html/python/_autosummary/mlx.nn.average_gradients.html @@ -8,7 +8,7 @@ - mlx.nn.average_gradients — MLX 0.24.1 documentation + mlx.nn.average_gradients — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/mlx.nn.quantize.html b/docs/build/html/python/_autosummary/mlx.nn.quantize.html index f160bcf67..1024caca7 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.24.1 documentation + mlx.nn.quantize — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 60d1a69cb..96e42a74a 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.24.1 documentation + mlx.nn.value_and_grad — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 35f5fee70..4d25bdb75 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.24.1 documentation + mlx.optimizers.clip_grad_norm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 1941affe2..6900e2b45 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.24.1 documentation + mlx.utils.tree_flatten — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 13b403236..1b93b9d67 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.24.1 documentation + mlx.utils.tree_map — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 27b83da39..200a08f98 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.24.1 documentation + mlx.utils.tree_map_with_path — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 ec3cf4f6f..1c36ace74 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.24.1 documentation + mlx.utils.tree_reduce — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 ac6f51176..5bc42febd 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.24.1 documentation + mlx.utils.tree_unflatten — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/_autosummary/stream_class.html b/docs/build/html/python/_autosummary/stream_class.html index 1e0f1eb03..9ab56a89d 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.24.1 documentation + mlx.core.Stream — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/array.html b/docs/build/html/python/array.html index bc3ef539b..d4c051f13 100644 --- a/docs/build/html/python/array.html +++ b/docs/build/html/python/array.html @@ -8,7 +8,7 @@ - Array — MLX 0.24.1 documentation + Array — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/data_types.html b/docs/build/html/python/data_types.html index 9544404c4..b2fca58b4 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.24.1 documentation + Data Types — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/devices_and_streams.html b/docs/build/html/python/devices_and_streams.html index 9e8a6f66d..52bc83d61 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.24.1 documentation + Devices and Streams — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/distributed.html b/docs/build/html/python/distributed.html index 69193bffd..2b54c7d26 100644 --- a/docs/build/html/python/distributed.html +++ b/docs/build/html/python/distributed.html @@ -8,7 +8,7 @@ - Distributed Communication — MLX 0.24.1 documentation + Distributed Communication — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/export.html b/docs/build/html/python/export.html index 71510e265..5faa673e8 100644 --- a/docs/build/html/python/export.html +++ b/docs/build/html/python/export.html @@ -8,7 +8,7 @@ - Export Functions — MLX 0.24.1 documentation + Export Functions — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/fast.html b/docs/build/html/python/fast.html index c7aab2a19..3955eb42a 100644 --- a/docs/build/html/python/fast.html +++ b/docs/build/html/python/fast.html @@ -8,7 +8,7 @@ - Fast — MLX 0.24.1 documentation + Fast — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/fft.html b/docs/build/html/python/fft.html index f96dccdeb..e12df14b3 100644 --- a/docs/build/html/python/fft.html +++ b/docs/build/html/python/fft.html @@ -8,7 +8,7 @@ - FFT — MLX 0.24.1 documentation + FFT — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/linalg.html b/docs/build/html/python/linalg.html index f1e0a1953..70135098f 100644 --- a/docs/build/html/python/linalg.html +++ b/docs/build/html/python/linalg.html @@ -8,7 +8,7 @@ - Linear Algebra — MLX 0.24.1 documentation + Linear Algebra — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      @@ -935,10 +939,13 @@ document.write(`

      lu_factor(a, *[, stream])

      Computes a compact representation of the LU factorization.

      -

      solve(a, b, *[, stream])

      +

      pinv(a, *[, stream])

      +

      Compute the (Moore-Penrose) pseudo-inverse of a matrix.

      + +

      solve(a, b, *[, stream])

      Compute the solution to a system of linear equations AX = B.

      -

      solve_triangular(a, b, *[, upper, stream])

      +

      solve_triangular(a, b, *[, upper, stream])

      Computes the solution of a triangular system of linear equations AX = B.

      diff --git a/docs/build/html/python/memory_management.html b/docs/build/html/python/memory_management.html index 25f21914e..2b6489412 100644 --- a/docs/build/html/python/memory_management.html +++ b/docs/build/html/python/memory_management.html @@ -8,7 +8,7 @@ - Memory Management — MLX 0.24.1 documentation + Memory Management — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/metal.html b/docs/build/html/python/metal.html index accfc24e3..d895bd9b6 100644 --- a/docs/build/html/python/metal.html +++ b/docs/build/html/python/metal.html @@ -8,7 +8,7 @@ - Metal — MLX 0.24.1 documentation + Metal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn.html b/docs/build/html/python/nn.html index d78205a2d..b1e9564bb 100644 --- a/docs/build/html/python/nn.html +++ b/docs/build/html/python/nn.html @@ -8,7 +8,7 @@ - Neural Networks — MLX 0.24.1 documentation + Neural Networks — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 05ca140ff..c5bd11556 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.24.1 documentation + mlx.nn.ALiBi — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 d01ed0948..4c2465b07 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.24.1 documentation + mlx.nn.AvgPool1d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 fa6aec0e7..fc6614745 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.24.1 documentation + mlx.nn.AvgPool2d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool3d.html index 52b9455e2..fc02379b6 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool3d.html @@ -8,7 +8,7 @@ - mlx.nn.AvgPool3d — MLX 0.24.1 documentation + mlx.nn.AvgPool3d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 db4e523a7..1ac44fb46 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.24.1 documentation + mlx.nn.BatchNorm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.CELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.CELU.html index 37c905ce6..c9ae79e21 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.CELU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.CELU.html @@ -8,7 +8,7 @@ - mlx.nn.CELU — MLX 0.24.1 documentation + mlx.nn.CELU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 e66a213b5..ee609bea4 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.24.1 documentation + mlx.nn.Conv1d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 fb74b18d0..c88b8ad61 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.24.1 documentation + mlx.nn.Conv2d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 aeecad5a5..16ffc9041 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.24.1 documentation + mlx.nn.Conv3d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose1d.html index 83bf6b43a..490b17616 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose1d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose1d.html @@ -8,7 +8,7 @@ - mlx.nn.ConvTranspose1d — MLX 0.24.1 documentation + mlx.nn.ConvTranspose1d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose2d.html index 01ac012ae..267c899b5 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose2d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose2d.html @@ -8,7 +8,7 @@ - mlx.nn.ConvTranspose2d — MLX 0.24.1 documentation + mlx.nn.ConvTranspose2d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose3d.html index 527b4b05b..a77aa3c3a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ConvTranspose3d.html @@ -8,7 +8,7 @@ - mlx.nn.ConvTranspose3d — MLX 0.24.1 documentation + mlx.nn.ConvTranspose3d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6e302111d..8f76948dc 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.24.1 documentation + mlx.nn.Dropout — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 a5cde8539..ff931fb2b 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.24.1 documentation + mlx.nn.Dropout2d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 93de9a3ff..0a75622b8 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.24.1 documentation + mlx.nn.Dropout3d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ELU.html index 78f79f642..48fc66a4a 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ELU.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ELU.html @@ -8,7 +8,7 @@ - mlx.nn.ELU — MLX 0.24.1 documentation + mlx.nn.ELU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 cbbb72938..c59fc0dd7 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.24.1 documentation + mlx.nn.Embedding — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a15c580d1..2a1e380af 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.24.1 documentation + mlx.nn.GELU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 598586695..6af518e96 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.24.1 documentation + mlx.nn.GLU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 1fa8c6ea1..72d9b15dc 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.24.1 documentation + mlx.nn.GRU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 d370f17fe..d3573e919 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.24.1 documentation + mlx.nn.GroupNorm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 82fb2e8e4..25d5316c1 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.24.1 documentation + mlx.nn.HardShrink — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 cfac36350..74564e740 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.24.1 documentation + mlx.nn.HardTanh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 09e683eb4..3048d1e01 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.24.1 documentation + mlx.nn.Hardswish — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 0748faac7..0f450b1e5 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.24.1 documentation + mlx.nn.InstanceNorm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 3b2c98860..30e67f9c4 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.24.1 documentation + mlx.nn.LSTM — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 ee0efd4f7..4156b852f 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.24.1 documentation + mlx.nn.LayerNorm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 6a9ca592e..c336d6c5e 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.24.1 documentation + mlx.nn.LeakyReLU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 803718fd3..1f98eda9d 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.24.1 documentation + mlx.nn.Linear — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LogSigmoid.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LogSigmoid.html index e2144e980..5236d19ff 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LogSigmoid.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LogSigmoid.html @@ -8,7 +8,7 @@ - mlx.nn.LogSigmoid — MLX 0.24.1 documentation + mlx.nn.LogSigmoid — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LogSoftmax.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LogSoftmax.html index c6a4e6d73..d93de1e31 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.LogSoftmax.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LogSoftmax.html @@ -8,7 +8,7 @@ - mlx.nn.LogSoftmax — MLX 0.24.1 documentation + mlx.nn.LogSoftmax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c59cccc2e..d0c374a99 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.24.1 documentation + mlx.nn.MaxPool1d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 5a41746a5..24c3ebc76 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.24.1 documentation + mlx.nn.MaxPool2d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool3d.html index 44b60eea9..9862f2026 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool3d.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool3d.html @@ -8,7 +8,7 @@ - mlx.nn.MaxPool3d — MLX 0.24.1 documentation + mlx.nn.MaxPool3d — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6f44f13fa..48ce4007d 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.24.1 documentation + mlx.nn.Mish — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 ea3cb545e..9d7b7c5e8 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.24.1 documentation + mlx.nn.Module.apply — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 305914eac..b8ff72a42 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.24.1 documentation + mlx.nn.Module.apply_to_modules — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 7886ec812..b3279fff6 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.24.1 documentation + mlx.nn.Module.children — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 8a36d98ee..380f1e8be 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.24.1 documentation + mlx.nn.Module.eval — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 ac383d223..f8ee8de23 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.24.1 documentation + mlx.nn.Module.filter_and_map — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 47c5e654b..f9a67bbd6 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.24.1 documentation + mlx.nn.Module.freeze — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 30e2e56d3..19f563248 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.24.1 documentation + mlx.nn.Module.leaf_modules — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 da4c96e4d..2e5b676eb 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.24.1 documentation + mlx.nn.Module.load_weights — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e0895133b..a811971a0 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.24.1 documentation + mlx.nn.Module.modules — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e570774c3..3da3b043b 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.24.1 documentation + mlx.nn.Module.named_modules — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 70dbe63de..2a5d333a4 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.24.1 documentation + mlx.nn.Module.parameters — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 802e42864..62f99bb11 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.24.1 documentation + mlx.nn.Module.save_weights — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 5924893ab..de98330c4 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.24.1 documentation + mlx.nn.Module.set_dtype — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 61ec00470..494c6b0f7 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.24.1 documentation + mlx.nn.Module.state — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 47ebf0646..9ac40feb4 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.24.1 documentation + mlx.nn.Module.train — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a2b43b578..462acb49c 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.24.1 documentation + mlx.nn.Module.trainable_parameters — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 71cf3f09a..6478d488d 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.24.1 documentation + mlx.nn.Module.training — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 186c1460a..506fc548f 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.24.1 documentation + mlx.nn.Module.unfreeze — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e349fc34b..784a7f45a 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.24.1 documentation + mlx.nn.Module.update — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6faa7012d..6f2a8a9e4 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.24.1 documentation + mlx.nn.Module.update_modules — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 545abb85c..53e6d42f4 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.24.1 documentation + mlx.nn.MultiHeadAttention — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3b693b412..35b523636 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.24.1 documentation + mlx.nn.PReLU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 e953f068c..74bf998eb 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.24.1 documentation + mlx.nn.QuantizedEmbedding — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 9053b0b78..e1475261d 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.24.1 documentation + mlx.nn.QuantizedLinear — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 fad04a725..744953aef 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.24.1 documentation + mlx.nn.RMSNorm — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 911758223..52a1cbfed 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.24.1 documentation + mlx.nn.RNN — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 e9eb737a9..dfff811f4 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.24.1 documentation + mlx.nn.ReLU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 cb8ad8467..958c69ec2 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.24.1 documentation + mlx.nn.ReLU6 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a734f8f4c..0bf260448 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.24.1 documentation + mlx.nn.RoPE — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 47f1dd05f..2d4fcbc8d 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.24.1 documentation + mlx.nn.SELU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 65d795dc4..770196a82 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.24.1 documentation + mlx.nn.Sequential — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c8eb9352f..b40dbffe7 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.24.1 documentation + mlx.nn.SiLU — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Sigmoid.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Sigmoid.html index 5fd2e704d..5f490e480 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.Sigmoid.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Sigmoid.html @@ -8,7 +8,7 @@ - mlx.nn.Sigmoid — MLX 0.24.1 documentation + mlx.nn.Sigmoid — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 31b78df75..5ff325a76 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.24.1 documentation + mlx.nn.SinusoidalPositionalEncoding — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 408da5d05..e851be7fd 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.24.1 documentation + mlx.nn.Softmax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 38ea114c7..715613b43 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.24.1 documentation + mlx.nn.Softmin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 557e48153..e4f6e8645 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.24.1 documentation + mlx.nn.Softplus — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3e0cd1a7f..1b0c8c08c 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.24.1 documentation + mlx.nn.Softshrink — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 8cd80d7de..cdcb31d4a 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.24.1 documentation + mlx.nn.Softsign — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 34c5b59f2..f1336c38f 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.24.1 documentation + mlx.nn.Step — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 cd0335e00..226260e49 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.24.1 documentation + mlx.nn.Tanh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3b289765d..08f2110a5 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.24.1 documentation + mlx.nn.Transformer — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 da6677be5..05fe28662 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.24.1 documentation + mlx.nn.Upsample — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a508885c8..86b67456b 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.24.1 documentation + mlx.nn.init.constant — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c4b43b33c..94ea43373 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.24.1 documentation + mlx.nn.init.glorot_normal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 e4c978278..82e0e1f8f 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.24.1 documentation + mlx.nn.init.glorot_uniform — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 6c42e1e69..465e313d9 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.24.1 documentation + mlx.nn.init.he_normal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 e281b87a0..911bfaea9 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.24.1 documentation + mlx.nn.init.he_uniform — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 9c87ed709..59b60d446 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.24.1 documentation + mlx.nn.init.identity — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 e8362b30c..d9b4cc964 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.24.1 documentation + mlx.nn.init.normal — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 5a1264b5a..f0f5f4475 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.24.1 documentation + mlx.nn.init.uniform — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.celu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.celu.html index 5447d245a..c814889df 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.celu.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.celu.html @@ -8,7 +8,7 @@ - mlx.nn.celu — MLX 0.24.1 documentation + mlx.nn.celu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 c6b20552e..9aa98dd60 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.24.1 documentation + mlx.nn.elu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 d3ef54182..4603cdf6b 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.24.1 documentation + mlx.nn.gelu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 2a53ba6f8..cae4ddae8 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.24.1 documentation + mlx.nn.gelu_approx — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 a76fd5563..294367bb6 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.24.1 documentation + mlx.nn.gelu_fast_approx — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 5bc4b5be1..572da04e7 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.24.1 documentation + mlx.nn.glu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 608655870..f62ab0691 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.24.1 documentation + mlx.nn.hard_shrink — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 e1d17f803..b102c433f 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.24.1 documentation + mlx.nn.hard_tanh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 86dd17335..01b79edb2 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.24.1 documentation + mlx.nn.hardswish — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 3bb9d8dc5..c4d079435 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.24.1 documentation + mlx.nn.leaky_relu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 d8c74e32f..adf9394e1 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.24.1 documentation + mlx.nn.log_sigmoid — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 090602161..87e384fe7 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.24.1 documentation + mlx.nn.log_softmax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 bc4881031..c03f6a5c0 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.24.1 documentation + mlx.nn.losses.binary_cross_entropy — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3265dde4d..9e5f85326 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.24.1 documentation + mlx.nn.losses.cosine_similarity_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 f8064c2f9..729b7829f 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.24.1 documentation + mlx.nn.losses.cross_entropy — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 c8e65d9ba..2c3b23e4e 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.24.1 documentation + mlx.nn.losses.gaussian_nll_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 f4a512a0a..7c5dcd2d2 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.24.1 documentation + mlx.nn.losses.hinge_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 b6c5e66dc..453702cb5 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.24.1 documentation + mlx.nn.losses.huber_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 b45c24fa6..6793c44aa 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.24.1 documentation + mlx.nn.losses.kl_div_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 9fe3b491c..478766cf2 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.24.1 documentation + mlx.nn.losses.l1_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 a8cf4917b..84df014c2 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.24.1 documentation + mlx.nn.losses.log_cosh_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 30f98107c..d4757b511 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.24.1 documentation + mlx.nn.losses.margin_ranking_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 c0fac565c..5e9bd2d62 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.24.1 documentation + mlx.nn.losses.mse_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 d62ecea91..5de7f28b4 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.24.1 documentation + mlx.nn.losses.nll_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 ca43dcff4..0ace1e320 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.24.1 documentation + mlx.nn.losses.smooth_l1_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 66385dc9a..5333b8694 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.24.1 documentation + mlx.nn.losses.triplet_loss — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 8b697a7dc..df80dbe9a 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.24.1 documentation + mlx.nn.mish — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 029811ef0..d3a9d1cc7 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.24.1 documentation + mlx.nn.prelu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 c226e922a..d5bc96ca9 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.24.1 documentation + mlx.nn.relu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 2541c6317..af590183e 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.24.1 documentation + mlx.nn.relu6 — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 85bfbe745..6f65ac0f1 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.24.1 documentation + mlx.nn.selu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 bc51a0f92..ea8d0fce1 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.24.1 documentation + mlx.nn.sigmoid — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 7b48f16e8..a1169d09c 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.24.1 documentation + mlx.nn.silu — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 4839cbda5..f3a6b9f5e 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.24.1 documentation + mlx.nn.softmax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 ec45fb110..1cf9cf4c3 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.24.1 documentation + mlx.nn.softmin — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 fd850928b..db91dfefe 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.24.1 documentation + mlx.nn.softplus — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 36c5fa4f8..80b3c9a8a 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.24.1 documentation + mlx.nn.softshrink — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 1ddae31da..38adc1eef 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.24.1 documentation + mlx.nn.step — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 471d4335e..6c2eb6f5a 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.24.1 documentation + mlx.nn.tanh — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/functions.html b/docs/build/html/python/nn/functions.html index 502d2d180..2b53d90b7 100644 --- a/docs/build/html/python/nn/functions.html +++ b/docs/build/html/python/nn/functions.html @@ -8,7 +8,7 @@ - Functions — MLX 0.24.1 documentation + Functions — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/init.html b/docs/build/html/python/nn/init.html index 6bdab565b..794d7d958 100644 --- a/docs/build/html/python/nn/init.html +++ b/docs/build/html/python/nn/init.html @@ -8,7 +8,7 @@ - Initializers — MLX 0.24.1 documentation + Initializers — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/layers.html b/docs/build/html/python/nn/layers.html index 036be7055..cfaf38e76 100644 --- a/docs/build/html/python/nn/layers.html +++ b/docs/build/html/python/nn/layers.html @@ -8,7 +8,7 @@ - Layers — MLX 0.24.1 documentation + Layers — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/losses.html b/docs/build/html/python/nn/losses.html index 4be460d74..4eebeb689 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.24.1 documentation + Loss Functions — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/nn/module.html b/docs/build/html/python/nn/module.html index 96be707c5..aa60c29af 100644 --- a/docs/build/html/python/nn/module.html +++ b/docs/build/html/python/nn/module.html @@ -8,7 +8,7 @@ - Module — MLX 0.24.1 documentation + Module — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/ops.html b/docs/build/html/python/ops.html index b9577689a..b698bd596 100644 --- a/docs/build/html/python/ops.html +++ b/docs/build/html/python/ops.html @@ -8,7 +8,7 @@ - Operations — MLX 0.24.1 documentation + Operations — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      @@ -983,18 +987,24 @@ document.write(`

      block_masked_mm(a, b, /[, block_size, ...])

      Matrix multiplication with block masking.

      -

      broadcast_to(a, /, shape, *[, stream])

      +

      broadcast_arrays(*arrays[, stream])

      +

      Broadcast arrays against one another.

      + +

      broadcast_to(a, /, shape, *[, stream])

      Broadcast an array to the given shape.

      -

      ceil(a, /, *[, stream])

      +

      ceil(a, /, *[, stream])

      Element-wise ceil.

      -

      clip(a, /, a_min, a_max, *[, stream])

      +

      clip(a, /, a_min, a_max, *[, stream])

      Clip the values of the array between the given minimum and maximum.

      -

      concatenate(arrays[, axis, stream])

      +

      concatenate(arrays[, axis, stream])

      Concatenate the arrays along the given axis.

      +

      contiguous(a, /[, allow_col_major, stream])

      +

      Force an array to be row contiguous.

      +

      conj(a, *[, stream])

      Return the elementwise complex conjugate of the input.

      diff --git a/docs/build/html/python/optimizers.html b/docs/build/html/python/optimizers.html index 670a9b16b..528ddb2ad 100644 --- a/docs/build/html/python/optimizers.html +++ b/docs/build/html/python/optimizers.html @@ -8,7 +8,7 @@ - Optimizers — MLX 0.24.1 documentation + Optimizers — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 ab19a3e6d..3597bb8fc 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.24.1 documentation + mlx.optimizers.AdaDelta — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 d535440fd..83cc04738 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.24.1 documentation + mlx.optimizers.Adafactor — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 8b5f42930..7ff56d9ab 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.24.1 documentation + mlx.optimizers.Adagrad — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 74a9d79de..9f2bad714 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.24.1 documentation + mlx.optimizers.Adam — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 c1d511c05..5e389a547 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.24.1 documentation + mlx.optimizers.AdamW — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 951a61963..bcc908267 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.24.1 documentation + mlx.optimizers.Adamax — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 c58186d2d..3cb829a47 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.24.1 documentation + mlx.optimizers.Lion — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 6166de190..7ec24d367 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.24.1 documentation + mlx.optimizers.Optimizer.apply_gradients — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 3ef33e263..fb1c310d0 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.24.1 documentation + mlx.optimizers.Optimizer.init — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 de1e9ece2..f9a3b7aa0 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.24.1 documentation + mlx.optimizers.Optimizer.state — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 fcd2cc75e..baf3613c2 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.24.1 documentation + mlx.optimizers.Optimizer.update — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 ae4fa8361..a4e6cf816 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.24.1 documentation + mlx.optimizers.RMSprop — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 77f138a7e..976e4785f 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.24.1 documentation + mlx.optimizers.SGD — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 189fd346c..4e7ce2724 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.24.1 documentation + mlx.optimizers.cosine_decay — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 6922f80fb..ddaca6293 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.24.1 documentation + mlx.optimizers.exponential_decay — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 9c4b0d51b..ab31ab56f 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.24.1 documentation + mlx.optimizers.join_schedules — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      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 b7b04defd..17d6f879f 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.24.1 documentation + mlx.optimizers.linear_schedule — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      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 66a853c12..06f316860 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.24.1 documentation + mlx.optimizers.step_decay — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/optimizers/common_optimizers.html b/docs/build/html/python/optimizers/common_optimizers.html index a4291b9d8..e0de6ed08 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.24.1 documentation + Common Optimizers — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/optimizers/optimizer.html b/docs/build/html/python/optimizers/optimizer.html index 10d20f6c7..bd783c553 100644 --- a/docs/build/html/python/optimizers/optimizer.html +++ b/docs/build/html/python/optimizers/optimizer.html @@ -8,7 +8,7 @@ - Optimizer — MLX 0.24.1 documentation + Optimizer — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/optimizers/schedulers.html b/docs/build/html/python/optimizers/schedulers.html index b8725d1b9..4e48b2328 100644 --- a/docs/build/html/python/optimizers/schedulers.html +++ b/docs/build/html/python/optimizers/schedulers.html @@ -8,7 +8,7 @@ - Schedulers — MLX 0.24.1 documentation + Schedulers — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/python/random.html b/docs/build/html/python/random.html index 31fbc4b4e..d34bf8341 100644 --- a/docs/build/html/python/random.html +++ b/docs/build/html/python/random.html @@ -8,7 +8,7 @@ - Random — MLX 0.24.1 documentation + Random — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/python/transforms.html b/docs/build/html/python/transforms.html index 89620c5d1..786de5ecd 100644 --- a/docs/build/html/python/transforms.html +++ b/docs/build/html/python/transforms.html @@ -8,7 +8,7 @@ - Transforms — MLX 0.24.1 documentation + Transforms — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      @@ -901,31 +905,34 @@ document.write(`

      eval(*args)

      Evaluate an array or tree of array.

      -

      compile(fun[, inputs, outputs, shapeless])

      +

      async_eval(*args)

      +

      Asynchronously evaluate an array or tree of array.

      + +

      compile(fun[, inputs, outputs, shapeless])

      Returns a compiled function which produces the same output as fun.

      -

      custom_function

      +

      custom_function

      Set up a function for custom gradient and vmap definitions.

      -

      disable_compile()

      +

      disable_compile()

      Globally disable compilation.

      -

      enable_compile()

      +

      enable_compile()

      Globally enable compilation.

      -

      grad(fun[, argnums, argnames])

      +

      grad(fun[, argnums, argnames])

      Returns a function which computes the gradient of fun.

      -

      value_and_grad(fun[, argnums, argnames])

      +

      value_and_grad(fun[, argnums, argnames])

      Returns a function which computes the value and gradient of fun.

      -

      jvp(fun, primals, tangents)

      +

      jvp(fun, primals, tangents)

      Compute the Jacobian-vector product.

      -

      vjp(fun, primals, cotangents)

      +

      vjp(fun, primals, cotangents)

      Compute the vector-Jacobian product.

      -

      vmap(fun[, in_axes, out_axes])

      +

      vmap(fun[, in_axes, out_axes])

      Returns a vectorized version of fun.

      diff --git a/docs/build/html/python/tree_utils.html b/docs/build/html/python/tree_utils.html index 64190b1d8..885b98b12 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.24.1 documentation + Tree Utils — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/search.html b/docs/build/html/search.html index 3a35239ed..87bac05d5 100644 --- a/docs/build/html/search.html +++ b/docs/build/html/search.html @@ -6,7 +6,7 @@ - Search - MLX 0.24.1 documentation + Search - MLX 0.24.2 documentation @@ -34,7 +34,7 @@ - + @@ -138,8 +138,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -292,10 +292,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -454,6 +456,7 @@
    • Transforms
      diff --git a/docs/build/html/searchindex.js b/docs/build/html/searchindex.js index 4cf0f3c73..7c67f491a 100644 --- a/docs/build/html/searchindex.js +++ b/docs/build/html/searchindex.js @@ -1 +1 @@ -Search.setIndex({"alltitles": {"A Simple Example": [[508, "a-simple-example"]], "Array": [[331, null]], "Attention layer": [[6, "attention-layer"]], "Automatic Differentiation": [[501, "automatic-differentiation"]], "Automatic Vectorization": [[501, "automatic-vectorization"]], "Basics": [[506, "basics"]], "Basics of Compile": [[498, "basics-of-compile"]], "Basics of Exporting": [[500, "basics-of-exporting"]], "Binary Size Minimization": [[9, "binary-size-minimization"]], "Binding to Python": [[2, "binding-to-python"]], "Build Options": [[9, "id4"]], "Build Requirements": [[9, "build-requirements"]], "Build and Install": [[9, null]], "Build from source": [[9, "build-from-source"]], "Building and Binding": [[2, "building-and-binding"]], "Building with CMake": [[2, "building-with-cmake"]], "Building with setuptools": [[2, "building-with-setuptools"]], "C++ API": [[9, "c-api"]], "C++ API Reference": [[8, null]], "Common Optimizers": [[492, null]], "Compilation": [[498, null]], "Compiling Training Graphs": [[498, "compiling-training-graphs"]], "Complex Example": [[1, "complex-example"]], "Conversion to NumPy and Other Frameworks": [[505, null]], "Converting the weights": [[6, "converting-the-weights"]], "Custom Extensions in MLX": [[2, null]], "Custom Metal Kernels": [[1, null]], "Data Types": [[332, null]], "Debugging": [[498, "debugging"]], "Defining a Ring": [[499, "defining-a-ring"]], "Devices and Streams": [[333, null]], "Differences from NumPy": [[502, "differences-from-numpy"]], "Distributed Communication": [[334, null], [499, null]], "Download the code": [[2, null], [6, null]], "Encoder layer": [[6, "encoder-layer"]], "Example Speedup": [[498, "example-speedup"]], "Examples": [[8, null]], "Export Functions": [[335, null]], "Exporting Functions": [[500, null]], "Exporting Modules": [[500, "exporting-modules"]], "Exporting Multiple Traces": [[500, "exporting-multiple-traces"]], "FFT": [[337, null]], "Fast": [[336, null]], "Full model": [[6, "full-model"]], "Function Transforms": [[501, null]], "Function and Graph Transformations": [[506, "function-and-graph-transformations"]], "Functions": [[467, null]], "Further Reading": [[8, null]], "Generation": [[6, "generation"]], "Getting Started": [[499, "getting-started"]], "Getting Started with MPI": [[499, "getting-started-with-mpi"]], "Getting Started with Ring": [[499, "getting-started-with-ring"]], "Grid Sample VJP": [[1, "grid-sample-vjp"]], "Implementing the CPU Back-end": [[2, "implementing-the-cpu-back-end"]], "Implementing the GPU Back-end": [[2, "implementing-the-gpu-back-end"]], "Implementing the Primitive": [[2, "implementing-the-primitive"]], "Implementing the model": [[6, "implementing-the-model"]], "Importing Functions in C++": [[500, "importing-functions-in-c"]], "In Place Updates": [[502, "in-place-updates"]], "Indexing Arrays": [[502, null]], "Initializers": [[468, null]], "Inspecting Modules": [[341, "inspecting-modules"]], "Install": [[8, null]], "Installing MPI": [[499, "installing-mpi"]], "Introducing the Example": [[2, "introducing-the-example"]], "JAX": [[505, "jax"]], "LLM inference": [[6, null]], "Launching Distributed Programs": [[503, null]], "Layers": [[469, null]], "Lazy Evaluation": [[504, null]], "Linear Algebra": [[338, null]], "Linear Regression": [[5, null]], "Loss Functions": [[470, null]], "MLX": [[8, null]], "MPI Specifics": [[503, "mpi-specifics"]], "Memory Management": [[339, null]], "Metal": [[340, null]], "Metal Debugger": [[3, null]], "Metal not found": [[9, "metal-not-found"]], "Module": [[471, null]], "More Examples": [[500, "more-examples"]], "Multi-Layer Perceptron": [[7, null]], "Neural Networks": [[341, null]], "Only Compute What You Use": [[504, "only-compute-what-you-use"]], "Operations": [[0, null], [2, "operations"], [472, null]], "Operations and Primitives": [[2, "operations-and-primitives"]], "Optimizer": [[493, null]], "Optimizers": [[473, null]], "Package Variables": [[4, "id1"]], "Parameters": [[341, "parameters"]], "Primitive Transforms": [[2, "primitive-transforms"]], "Primitives": [[2, "primitives"]], "Providing Hosts": [[503, "providing-hosts"]], "Pure Functions": [[498, "pure-functions"]], "Putting it all together": [[6, "putting-it-all-together"]], "PyTorch": [[505, "pytorch"]], "Python API": [[9, "python-api"]], "Python API Reference": [[8, null]], "Python Installation": [[9, "python-installation"]], "Quick Start Guide": [[506, null]], "Quick Start with Neural Networks": [[341, "quick-start-with-neural-networks"]], "Random": [[495, null]], "Results": [[2, "results"]], "Ring Specifics": [[503, "ring-specifics"]], "Running Distributed Programs": [[499, "running-distributed-programs"]], "Saving and Loading": [[473, "saving-and-loading"]], "Saving and Loading Arrays": [[507, null]], "Schedulers": [[494, null]], "Scripts": [[2, "scripts"], [6, "scripts"]], "Selecting Backend": [[499, "selecting-backend"]], "Serialization Formats": [[507, "id1"]], "Setting up Remote Hosts": [[499, "setting-up-remote-hosts"], [503, "setting-up-remote-hosts"]], "Shapeless Compilation": [[498, "shapeless-compilation"]], "Shapeless Exports": [[500, "shapeless-exports"]], "Simple Example": [[1, "simple-example"]], "Specifying the Stream": [[509, "specifying-the-stream"]], "Supported Data Types": [[332, "id2"]], "TensorFlow": [[505, "tensorflow"]], "The Module Class": [[341, "the-module-class"]], "Thunderbolt Ring": [[499, "thunderbolt-ring"]], "Training Example": [[499, "training-example"]], "Transformations with Compile": [[498, "transformations-with-compile"]], "Transformations with Imported Functions": [[500, "transformations-with-imported-functions"]], "Transforming Compute Graphs": [[504, "transforming-compute-graphs"]], "Transforms": [[496, null]], "Tree Utils": [[497, null]], "Troubleshooting": [[9, "troubleshooting"], [9, "id3"]], "Tuning MPI All Reduce": [[499, "tuning-mpi-all-reduce"]], "Unified Memory": [[508, null]], "Updating the Parameters": [[341, "updating-the-parameters"]], "Usage": [[2, "usage"], [8, null], [503, "usage"]], "Using MLX in C++": [[4, null]], "Using Shape/Strides": [[1, "using-shape-strides"]], "Using Streams": [[509, null]], "Using the Primitive": [[2, "using-the-primitive"]], "Utilizing nn.average_gradients": [[499, "utilizing-nn-average-gradients"]], "Value and Grad": [[341, "value-and-grad"]], "Weight loading and benchmarking": [[6, "weight-loading-and-benchmarking"]], "When to Evaluate": [[504, "when-to-evaluate"]], "Why Lazy Evaluation": [[504, "why-lazy-evaluation"]], "Xcode Workflow": [[3, "xcode-workflow"]], "mlx.core.Device": [[10, null]], "mlx.core.Dtype": [[11, null]], "mlx.core.DtypeCategory": [[12, null]], "mlx.core.Stream": [[330, null]], "mlx.core.abs": [[13, null]], "mlx.core.add": [[14, null]], "mlx.core.addmm": [[15, null]], "mlx.core.all": [[16, null]], "mlx.core.allclose": [[17, null]], "mlx.core.any": [[18, null]], "mlx.core.arange": [[19, null]], "mlx.core.arccos": [[20, null]], "mlx.core.arccosh": [[21, null]], "mlx.core.arcsin": [[22, null]], "mlx.core.arcsinh": [[23, null]], "mlx.core.arctan": [[24, null]], "mlx.core.arctan2": [[25, null]], "mlx.core.arctanh": [[26, null]], "mlx.core.argmax": [[27, null]], "mlx.core.argmin": [[28, null]], "mlx.core.argpartition": [[29, null]], "mlx.core.argsort": [[30, null]], "mlx.core.array": [[31, null]], "mlx.core.array.T": [[32, null]], "mlx.core.array.abs": [[33, null]], "mlx.core.array.all": [[34, null]], "mlx.core.array.any": [[35, null]], "mlx.core.array.argmax": [[36, null]], "mlx.core.array.argmin": [[37, null]], "mlx.core.array.astype": [[38, null]], "mlx.core.array.at": [[39, null]], "mlx.core.array.conj": [[40, null]], "mlx.core.array.cos": [[41, null]], "mlx.core.array.cummax": [[42, null]], "mlx.core.array.cummin": [[43, null]], "mlx.core.array.cumprod": [[44, null]], "mlx.core.array.cumsum": [[45, null]], "mlx.core.array.diag": [[46, null]], "mlx.core.array.diagonal": [[47, null]], "mlx.core.array.dtype": [[48, null]], "mlx.core.array.exp": [[49, null]], "mlx.core.array.flatten": [[50, null]], "mlx.core.array.item": [[51, null]], "mlx.core.array.itemsize": [[52, null]], "mlx.core.array.log": [[53, null]], "mlx.core.array.log10": [[54, null]], "mlx.core.array.log1p": [[55, null]], "mlx.core.array.log2": [[56, null]], "mlx.core.array.logsumexp": [[57, null]], "mlx.core.array.max": [[58, null]], "mlx.core.array.mean": [[59, null]], "mlx.core.array.min": [[60, null]], "mlx.core.array.moveaxis": [[61, null]], "mlx.core.array.nbytes": [[62, null]], "mlx.core.array.ndim": [[63, null]], "mlx.core.array.prod": [[64, null]], "mlx.core.array.reciprocal": [[65, null]], "mlx.core.array.reshape": [[66, null]], "mlx.core.array.round": [[67, null]], "mlx.core.array.rsqrt": [[68, null]], "mlx.core.array.shape": [[69, null]], "mlx.core.array.sin": [[70, null]], "mlx.core.array.size": [[71, null]], "mlx.core.array.split": [[72, null]], "mlx.core.array.sqrt": [[73, null]], "mlx.core.array.square": [[74, null]], "mlx.core.array.squeeze": [[75, null]], "mlx.core.array.std": [[76, null]], "mlx.core.array.sum": [[77, null]], "mlx.core.array.swapaxes": [[78, null]], "mlx.core.array.tolist": [[79, null]], "mlx.core.array.transpose": [[80, null]], "mlx.core.array.var": [[81, null]], "mlx.core.array.view": [[82, null]], "mlx.core.array_equal": [[83, null]], "mlx.core.as_strided": [[84, null]], "mlx.core.atleast_1d": [[85, null]], "mlx.core.atleast_2d": [[86, null]], "mlx.core.atleast_3d": [[87, null]], "mlx.core.bitwise_and": [[88, null]], "mlx.core.bitwise_invert": [[89, null]], "mlx.core.bitwise_or": [[90, null]], "mlx.core.bitwise_xor": [[91, null]], "mlx.core.block_masked_mm": [[92, null]], "mlx.core.broadcast_to": [[93, null]], "mlx.core.ceil": [[94, null]], "mlx.core.clear_cache": [[95, null]], "mlx.core.clip": [[96, null]], "mlx.core.compile": [[97, null]], "mlx.core.concatenate": [[98, null]], "mlx.core.conj": [[99, null]], "mlx.core.conjugate": [[100, null]], "mlx.core.conv1d": [[101, null]], "mlx.core.conv2d": [[102, null]], "mlx.core.conv3d": [[103, null]], "mlx.core.conv_general": [[104, null]], "mlx.core.conv_transpose1d": [[105, null]], "mlx.core.conv_transpose2d": [[106, null]], "mlx.core.conv_transpose3d": [[107, null]], "mlx.core.convolve": [[108, null]], "mlx.core.cos": [[109, null]], "mlx.core.cosh": [[110, null]], "mlx.core.cummax": [[111, null]], "mlx.core.cummin": [[112, null]], "mlx.core.cumprod": [[113, null]], "mlx.core.cumsum": [[114, null]], "mlx.core.custom_function": [[115, null]], "mlx.core.default_device": [[116, null]], "mlx.core.default_stream": [[117, null]], "mlx.core.degrees": [[118, null]], "mlx.core.dequantize": [[119, null]], "mlx.core.diag": [[120, null]], "mlx.core.diagonal": [[121, null]], "mlx.core.disable_compile": [[122, null]], "mlx.core.distributed.Group": [[123, null]], "mlx.core.distributed.all_gather": [[124, null]], "mlx.core.distributed.all_sum": [[125, null]], "mlx.core.distributed.init": [[126, null]], "mlx.core.distributed.is_available": [[127, null]], "mlx.core.distributed.recv": [[128, null]], "mlx.core.distributed.recv_like": [[129, null]], "mlx.core.distributed.send": [[130, null]], "mlx.core.divide": [[131, null]], "mlx.core.divmod": [[132, null]], "mlx.core.einsum": [[133, null]], "mlx.core.einsum_path": [[134, null]], "mlx.core.enable_compile": [[135, null]], "mlx.core.equal": [[136, null]], "mlx.core.erf": [[137, null]], "mlx.core.erfinv": [[138, null]], "mlx.core.eval": [[139, null]], "mlx.core.exp": [[140, null]], "mlx.core.expand_dims": [[141, null]], "mlx.core.expm1": [[142, null]], "mlx.core.export_function": [[143, null]], "mlx.core.export_to_dot": [[144, null]], "mlx.core.exporter": [[145, null]], "mlx.core.eye": [[146, null]], "mlx.core.fast.layer_norm": [[147, null]], "mlx.core.fast.metal_kernel": [[148, null]], "mlx.core.fast.rms_norm": [[149, null]], "mlx.core.fast.rope": [[150, null]], "mlx.core.fast.scaled_dot_product_attention": [[151, null]], "mlx.core.fft.fft": [[152, null]], "mlx.core.fft.fft2": [[153, null]], "mlx.core.fft.fftn": [[154, null]], "mlx.core.fft.ifft": [[155, null]], "mlx.core.fft.ifft2": [[156, null]], "mlx.core.fft.ifftn": [[157, null]], "mlx.core.fft.irfft": [[158, null]], "mlx.core.fft.irfft2": [[159, null]], "mlx.core.fft.irfftn": [[160, null]], "mlx.core.fft.rfft": [[161, null]], "mlx.core.fft.rfft2": [[162, null]], "mlx.core.fft.rfftn": [[163, null]], "mlx.core.finfo": [[164, null]], "mlx.core.flatten": [[165, null]], "mlx.core.floor": [[166, null]], "mlx.core.floor_divide": [[167, null]], "mlx.core.full": [[168, null]], "mlx.core.gather_mm": [[169, null]], "mlx.core.gather_qmm": [[170, null]], "mlx.core.get_active_memory": [[171, null]], "mlx.core.get_cache_memory": [[172, null]], "mlx.core.get_peak_memory": [[173, null]], "mlx.core.grad": [[174, null]], "mlx.core.greater": [[175, null]], "mlx.core.greater_equal": [[176, null]], "mlx.core.hadamard_transform": [[177, null]], "mlx.core.identity": [[178, null]], "mlx.core.imag": [[179, null]], "mlx.core.import_function": [[180, null]], "mlx.core.inner": [[181, null]], "mlx.core.isclose": [[182, null]], "mlx.core.isfinite": [[183, null]], "mlx.core.isinf": [[184, null]], "mlx.core.isnan": [[185, null]], "mlx.core.isneginf": [[186, null]], "mlx.core.isposinf": [[187, null]], "mlx.core.issubdtype": [[188, null]], "mlx.core.jvp": [[189, null]], "mlx.core.kron": [[190, null]], "mlx.core.left_shift": [[191, null]], "mlx.core.less": [[192, null]], "mlx.core.less_equal": [[193, null]], "mlx.core.linalg.cholesky": [[194, null]], "mlx.core.linalg.cholesky_inv": [[195, null]], "mlx.core.linalg.cross": [[196, null]], "mlx.core.linalg.eigh": [[197, null]], "mlx.core.linalg.eigvalsh": [[198, null]], "mlx.core.linalg.inv": [[199, null]], "mlx.core.linalg.lu": [[200, null]], "mlx.core.linalg.lu_factor": [[201, null]], "mlx.core.linalg.norm": [[202, null]], "mlx.core.linalg.qr": [[203, null]], "mlx.core.linalg.solve": [[204, null]], "mlx.core.linalg.solve_triangular": [[205, null]], "mlx.core.linalg.svd": [[206, null]], "mlx.core.linalg.tri_inv": [[207, null]], "mlx.core.linspace": [[208, null]], "mlx.core.load": [[209, null]], "mlx.core.log": [[210, null]], "mlx.core.log10": [[211, null]], "mlx.core.log1p": [[212, null]], "mlx.core.log2": [[213, null]], "mlx.core.logaddexp": [[214, null]], "mlx.core.logical_and": [[215, null]], "mlx.core.logical_not": [[216, null]], "mlx.core.logical_or": [[217, null]], "mlx.core.logsumexp": [[218, null]], "mlx.core.matmul": [[219, null]], "mlx.core.max": [[220, null]], "mlx.core.maximum": [[221, null]], "mlx.core.mean": [[222, null]], "mlx.core.meshgrid": [[223, null]], "mlx.core.metal.device_info": [[224, null]], "mlx.core.metal.is_available": [[225, null]], "mlx.core.metal.start_capture": [[226, null]], "mlx.core.metal.stop_capture": [[227, null]], "mlx.core.min": [[228, null]], "mlx.core.minimum": [[229, null]], "mlx.core.moveaxis": [[230, null]], "mlx.core.multiply": [[231, null]], "mlx.core.nan_to_num": [[232, null]], "mlx.core.negative": [[233, null]], "mlx.core.new_stream": [[234, null]], "mlx.core.not_equal": [[235, null]], "mlx.core.ones": [[236, null]], "mlx.core.ones_like": [[237, null]], "mlx.core.outer": [[238, null]], "mlx.core.pad": [[239, null]], "mlx.core.partition": [[240, null]], "mlx.core.power": [[241, null]], "mlx.core.prod": [[242, null]], "mlx.core.put_along_axis": [[243, null]], "mlx.core.quantize": [[244, null]], "mlx.core.quantized_matmul": [[245, null]], "mlx.core.radians": [[246, null]], "mlx.core.random.bernoulli": [[247, null]], "mlx.core.random.categorical": [[248, null]], "mlx.core.random.gumbel": [[249, null]], "mlx.core.random.key": [[250, null]], "mlx.core.random.laplace": [[251, null]], "mlx.core.random.multivariate_normal": [[252, null]], "mlx.core.random.normal": [[253, null]], "mlx.core.random.permutation": [[254, null]], "mlx.core.random.randint": [[255, null]], "mlx.core.random.seed": [[256, null]], "mlx.core.random.split": [[257, null]], "mlx.core.random.truncated_normal": [[258, null]], "mlx.core.random.uniform": [[259, null]], "mlx.core.real": [[260, null]], "mlx.core.reciprocal": [[261, null]], "mlx.core.remainder": [[262, null]], "mlx.core.repeat": [[263, null]], "mlx.core.reset_peak_memory": [[264, null]], "mlx.core.reshape": [[265, null]], "mlx.core.right_shift": [[266, null]], "mlx.core.roll": [[267, null]], "mlx.core.round": [[268, null]], "mlx.core.rsqrt": [[269, null]], "mlx.core.save": [[270, null]], "mlx.core.save_gguf": [[271, null]], "mlx.core.save_safetensors": [[272, null]], "mlx.core.savez": [[273, null]], "mlx.core.savez_compressed": [[274, null]], "mlx.core.set_cache_limit": [[275, null]], "mlx.core.set_default_device": [[276, null]], "mlx.core.set_default_stream": [[277, null]], "mlx.core.set_memory_limit": [[278, null]], "mlx.core.set_wired_limit": [[279, null]], "mlx.core.sigmoid": [[280, null]], "mlx.core.sign": [[281, null]], "mlx.core.sin": [[282, null]], "mlx.core.sinh": [[283, null]], "mlx.core.slice": [[284, null]], "mlx.core.slice_update": [[285, null]], "mlx.core.softmax": [[286, null]], "mlx.core.sort": [[287, null]], "mlx.core.split": [[288, null]], "mlx.core.sqrt": [[289, null]], "mlx.core.square": [[290, null]], "mlx.core.squeeze": [[291, null]], "mlx.core.stack": [[292, null]], "mlx.core.std": [[293, null]], "mlx.core.stop_gradient": [[294, null]], "mlx.core.stream": [[295, null]], "mlx.core.subtract": [[296, null]], "mlx.core.sum": [[297, null]], "mlx.core.swapaxes": [[298, null]], "mlx.core.synchronize": [[299, null]], "mlx.core.take": [[300, null]], "mlx.core.take_along_axis": [[301, null]], "mlx.core.tan": [[302, null]], "mlx.core.tanh": [[303, null]], "mlx.core.tensordot": [[304, null]], "mlx.core.tile": [[305, null]], "mlx.core.topk": [[306, null]], "mlx.core.trace": [[307, null]], "mlx.core.transpose": [[308, null]], "mlx.core.tri": [[309, null]], "mlx.core.tril": [[310, null]], "mlx.core.triu": [[311, null]], "mlx.core.unflatten": [[312, null]], "mlx.core.value_and_grad": [[313, null]], "mlx.core.var": [[314, null]], "mlx.core.view": [[315, null]], "mlx.core.vjp": [[316, null]], "mlx.core.vmap": [[317, null]], "mlx.core.where": [[318, null]], "mlx.core.zeros": [[319, null]], "mlx.core.zeros_like": [[320, null]], "mlx.nn.ALiBi": [[342, null]], "mlx.nn.AvgPool1d": [[343, null]], "mlx.nn.AvgPool2d": [[344, null]], "mlx.nn.AvgPool3d": [[345, null]], "mlx.nn.BatchNorm": [[346, null]], "mlx.nn.CELU": [[347, null]], "mlx.nn.Conv1d": [[348, null]], "mlx.nn.Conv2d": [[349, null]], "mlx.nn.Conv3d": [[350, null]], "mlx.nn.ConvTranspose1d": [[351, null]], "mlx.nn.ConvTranspose2d": [[352, null]], "mlx.nn.ConvTranspose3d": [[353, null]], "mlx.nn.Dropout": [[354, null]], "mlx.nn.Dropout2d": [[355, null]], "mlx.nn.Dropout3d": [[356, null]], "mlx.nn.ELU": [[357, null]], "mlx.nn.Embedding": [[358, null]], "mlx.nn.GELU": [[359, null]], "mlx.nn.GLU": [[360, null]], "mlx.nn.GRU": [[361, null]], "mlx.nn.GroupNorm": [[362, null]], "mlx.nn.HardShrink": [[363, null]], "mlx.nn.HardTanh": [[364, null]], "mlx.nn.Hardswish": [[365, null]], "mlx.nn.InstanceNorm": [[366, null]], "mlx.nn.LSTM": [[367, null]], "mlx.nn.LayerNorm": [[368, null]], "mlx.nn.LeakyReLU": [[369, null]], "mlx.nn.Linear": [[370, null]], "mlx.nn.LogSigmoid": [[371, null]], "mlx.nn.LogSoftmax": [[372, null]], "mlx.nn.MaxPool1d": [[373, null]], "mlx.nn.MaxPool2d": [[374, null]], "mlx.nn.MaxPool3d": [[375, null]], "mlx.nn.Mish": [[376, null]], "mlx.nn.Module.apply": [[377, null]], "mlx.nn.Module.apply_to_modules": [[378, null]], "mlx.nn.Module.children": [[379, null]], "mlx.nn.Module.eval": [[380, null]], "mlx.nn.Module.filter_and_map": [[381, null]], "mlx.nn.Module.freeze": [[382, null]], "mlx.nn.Module.leaf_modules": [[383, null]], "mlx.nn.Module.load_weights": [[384, null]], "mlx.nn.Module.modules": [[385, null]], "mlx.nn.Module.named_modules": [[386, null]], "mlx.nn.Module.parameters": [[387, null]], "mlx.nn.Module.save_weights": [[388, null]], "mlx.nn.Module.set_dtype": [[389, null]], "mlx.nn.Module.state": [[390, null]], "mlx.nn.Module.train": [[391, null]], "mlx.nn.Module.trainable_parameters": [[392, null]], "mlx.nn.Module.training": [[393, null]], "mlx.nn.Module.unfreeze": [[394, null]], "mlx.nn.Module.update": [[395, null]], "mlx.nn.Module.update_modules": [[396, null]], "mlx.nn.MultiHeadAttention": [[397, null]], "mlx.nn.PReLU": [[398, null]], "mlx.nn.QuantizedEmbedding": [[399, null]], "mlx.nn.QuantizedLinear": [[400, null]], "mlx.nn.RMSNorm": [[401, null]], "mlx.nn.RNN": [[402, null]], "mlx.nn.ReLU": [[403, null]], "mlx.nn.ReLU6": [[404, null]], "mlx.nn.RoPE": [[405, null]], "mlx.nn.SELU": [[406, null]], "mlx.nn.Sequential": [[407, null]], "mlx.nn.SiLU": [[408, null]], "mlx.nn.Sigmoid": [[409, null]], "mlx.nn.SinusoidalPositionalEncoding": [[410, null]], "mlx.nn.Softmax": [[411, null]], "mlx.nn.Softmin": [[412, null]], "mlx.nn.Softplus": [[413, null]], "mlx.nn.Softshrink": [[414, null]], "mlx.nn.Softsign": [[415, null]], "mlx.nn.Step": [[416, null]], "mlx.nn.Tanh": [[417, null]], "mlx.nn.Transformer": [[418, null]], "mlx.nn.Upsample": [[419, null]], "mlx.nn.average_gradients": [[321, null]], "mlx.nn.celu": [[428, null]], "mlx.nn.elu": [[429, null]], "mlx.nn.gelu": [[430, null]], "mlx.nn.gelu_approx": [[431, null]], "mlx.nn.gelu_fast_approx": [[432, null]], "mlx.nn.glu": [[433, null]], "mlx.nn.hard_shrink": [[434, null]], "mlx.nn.hard_tanh": [[435, null]], "mlx.nn.hardswish": [[436, null]], "mlx.nn.init.constant": [[420, null]], "mlx.nn.init.glorot_normal": [[421, null]], "mlx.nn.init.glorot_uniform": [[422, null]], "mlx.nn.init.he_normal": [[423, null]], "mlx.nn.init.he_uniform": [[424, null]], "mlx.nn.init.identity": [[425, null]], "mlx.nn.init.normal": [[426, null]], "mlx.nn.init.uniform": [[427, null]], "mlx.nn.leaky_relu": [[437, null]], "mlx.nn.log_sigmoid": [[438, null]], "mlx.nn.log_softmax": [[439, null]], "mlx.nn.losses.binary_cross_entropy": [[440, null]], "mlx.nn.losses.cosine_similarity_loss": [[441, null]], "mlx.nn.losses.cross_entropy": [[442, null]], "mlx.nn.losses.gaussian_nll_loss": [[443, null]], "mlx.nn.losses.hinge_loss": [[444, null]], "mlx.nn.losses.huber_loss": [[445, null]], "mlx.nn.losses.kl_div_loss": [[446, null]], "mlx.nn.losses.l1_loss": [[447, null]], "mlx.nn.losses.log_cosh_loss": [[448, null]], "mlx.nn.losses.margin_ranking_loss": [[449, null]], "mlx.nn.losses.mse_loss": [[450, null]], "mlx.nn.losses.nll_loss": [[451, null]], "mlx.nn.losses.smooth_l1_loss": [[452, null]], "mlx.nn.losses.triplet_loss": [[453, null]], "mlx.nn.mish": [[454, null]], "mlx.nn.prelu": [[455, null]], "mlx.nn.quantize": [[322, null]], "mlx.nn.relu": [[456, null]], "mlx.nn.relu6": [[457, null]], "mlx.nn.selu": [[458, null]], "mlx.nn.sigmoid": [[459, null]], "mlx.nn.silu": [[460, null]], "mlx.nn.softmax": [[461, null]], "mlx.nn.softmin": [[462, null]], "mlx.nn.softplus": [[463, null]], "mlx.nn.softshrink": [[464, null]], "mlx.nn.step": [[465, null]], "mlx.nn.tanh": [[466, null]], "mlx.nn.value_and_grad": [[323, null]], "mlx.optimizers.AdaDelta": [[474, null]], "mlx.optimizers.Adafactor": [[475, null]], "mlx.optimizers.Adagrad": [[476, null]], "mlx.optimizers.Adam": [[477, null]], "mlx.optimizers.AdamW": [[478, null]], "mlx.optimizers.Adamax": [[479, null]], "mlx.optimizers.Lion": [[480, null]], "mlx.optimizers.Optimizer.apply_gradients": [[481, null]], "mlx.optimizers.Optimizer.init": [[482, null]], "mlx.optimizers.Optimizer.state": [[483, null]], "mlx.optimizers.Optimizer.update": [[484, null]], "mlx.optimizers.RMSprop": [[485, null]], "mlx.optimizers.SGD": [[486, null]], "mlx.optimizers.clip_grad_norm": [[324, null]], "mlx.optimizers.cosine_decay": [[487, null]], "mlx.optimizers.exponential_decay": [[488, null]], "mlx.optimizers.join_schedules": [[489, null]], "mlx.optimizers.linear_schedule": [[490, null]], "mlx.optimizers.step_decay": [[491, null]], "mlx.utils.tree_flatten": [[325, null]], "mlx.utils.tree_map": [[326, null]], "mlx.utils.tree_map_with_path": [[327, null]], "mlx.utils.tree_reduce": [[328, null]], "mlx.utils.tree_unflatten": [[329, null]], "x86 Shell": [[9, "x86-shell"]]}, "docnames": ["cpp/ops", "dev/custom_metal_kernels", "dev/extensions", "dev/metal_debugger", "dev/mlx_in_cpp", "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.std", "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_invert", "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.clear_cache", "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.conv3d", "python/_autosummary/mlx.core.conv_general", "python/_autosummary/mlx.core.conv_transpose1d", "python/_autosummary/mlx.core.conv_transpose2d", "python/_autosummary/mlx.core.conv_transpose3d", "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.distributed.recv", "python/_autosummary/mlx.core.distributed.recv_like", "python/_autosummary/mlx.core.distributed.send", "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.export_function", "python/_autosummary/mlx.core.export_to_dot", "python/_autosummary/mlx.core.exporter", "python/_autosummary/mlx.core.eye", "python/_autosummary/mlx.core.fast.layer_norm", "python/_autosummary/mlx.core.fast.metal_kernel", "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.finfo", "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.get_active_memory", "python/_autosummary/mlx.core.get_cache_memory", "python/_autosummary/mlx.core.get_peak_memory", "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.imag", "python/_autosummary/mlx.core.import_function", "python/_autosummary/mlx.core.inner", "python/_autosummary/mlx.core.isclose", "python/_autosummary/mlx.core.isfinite", "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.kron", "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.cross", "python/_autosummary/mlx.core.linalg.eigh", "python/_autosummary/mlx.core.linalg.eigvalsh", "python/_autosummary/mlx.core.linalg.inv", "python/_autosummary/mlx.core.linalg.lu", "python/_autosummary/mlx.core.linalg.lu_factor", "python/_autosummary/mlx.core.linalg.norm", "python/_autosummary/mlx.core.linalg.qr", "python/_autosummary/mlx.core.linalg.solve", "python/_autosummary/mlx.core.linalg.solve_triangular", "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.device_info", "python/_autosummary/mlx.core.metal.is_available", "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.put_along_axis", "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.permutation", "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.real", "python/_autosummary/mlx.core.reciprocal", "python/_autosummary/mlx.core.remainder", "python/_autosummary/mlx.core.repeat", "python/_autosummary/mlx.core.reset_peak_memory", "python/_autosummary/mlx.core.reshape", "python/_autosummary/mlx.core.right_shift", "python/_autosummary/mlx.core.roll", "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_cache_limit", "python/_autosummary/mlx.core.set_default_device", "python/_autosummary/mlx.core.set_default_stream", "python/_autosummary/mlx.core.set_memory_limit", "python/_autosummary/mlx.core.set_wired_limit", "python/_autosummary/mlx.core.sigmoid", "python/_autosummary/mlx.core.sign", "python/_autosummary/mlx.core.sin", "python/_autosummary/mlx.core.sinh", "python/_autosummary/mlx.core.slice", "python/_autosummary/mlx.core.slice_update", "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.unflatten", "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.average_gradients", "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/export", "python/fast", "python/fft", "python/linalg", "python/memory_management", "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.AvgPool3d", "python/nn/_autosummary/mlx.nn.BatchNorm", "python/nn/_autosummary/mlx.nn.CELU", "python/nn/_autosummary/mlx.nn.Conv1d", "python/nn/_autosummary/mlx.nn.Conv2d", "python/nn/_autosummary/mlx.nn.Conv3d", "python/nn/_autosummary/mlx.nn.ConvTranspose1d", "python/nn/_autosummary/mlx.nn.ConvTranspose2d", "python/nn/_autosummary/mlx.nn.ConvTranspose3d", "python/nn/_autosummary/mlx.nn.Dropout", "python/nn/_autosummary/mlx.nn.Dropout2d", "python/nn/_autosummary/mlx.nn.Dropout3d", "python/nn/_autosummary/mlx.nn.ELU", "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.LogSigmoid", "python/nn/_autosummary/mlx.nn.LogSoftmax", "python/nn/_autosummary/mlx.nn.MaxPool1d", "python/nn/_autosummary/mlx.nn.MaxPool2d", "python/nn/_autosummary/mlx.nn.MaxPool3d", "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.Sigmoid", "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.celu", "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/export", "usage/function_transforms", "usage/indexing", "usage/launching_distributed", "usage/lazy_evaluation", "usage/numpy", "usage/quick_start", "usage/saving_and_loading", "usage/unified_memory", "usage/using_streams"], "envversion": {"sphinx": 62, "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}, "filenames": ["cpp/ops.rst", "dev/custom_metal_kernels.rst", "dev/extensions.rst", "dev/metal_debugger.rst", "dev/mlx_in_cpp.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.std.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_invert.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.clear_cache.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.conv3d.rst", "python/_autosummary/mlx.core.conv_general.rst", "python/_autosummary/mlx.core.conv_transpose1d.rst", "python/_autosummary/mlx.core.conv_transpose2d.rst", "python/_autosummary/mlx.core.conv_transpose3d.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.distributed.recv.rst", "python/_autosummary/mlx.core.distributed.recv_like.rst", "python/_autosummary/mlx.core.distributed.send.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.export_function.rst", "python/_autosummary/mlx.core.export_to_dot.rst", "python/_autosummary/mlx.core.exporter.rst", "python/_autosummary/mlx.core.eye.rst", "python/_autosummary/mlx.core.fast.layer_norm.rst", "python/_autosummary/mlx.core.fast.metal_kernel.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.finfo.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.get_active_memory.rst", "python/_autosummary/mlx.core.get_cache_memory.rst", "python/_autosummary/mlx.core.get_peak_memory.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.imag.rst", "python/_autosummary/mlx.core.import_function.rst", "python/_autosummary/mlx.core.inner.rst", "python/_autosummary/mlx.core.isclose.rst", "python/_autosummary/mlx.core.isfinite.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.kron.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.cross.rst", "python/_autosummary/mlx.core.linalg.eigh.rst", "python/_autosummary/mlx.core.linalg.eigvalsh.rst", "python/_autosummary/mlx.core.linalg.inv.rst", "python/_autosummary/mlx.core.linalg.lu.rst", "python/_autosummary/mlx.core.linalg.lu_factor.rst", "python/_autosummary/mlx.core.linalg.norm.rst", "python/_autosummary/mlx.core.linalg.qr.rst", "python/_autosummary/mlx.core.linalg.solve.rst", "python/_autosummary/mlx.core.linalg.solve_triangular.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.device_info.rst", "python/_autosummary/mlx.core.metal.is_available.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.put_along_axis.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.permutation.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.real.rst", "python/_autosummary/mlx.core.reciprocal.rst", "python/_autosummary/mlx.core.remainder.rst", "python/_autosummary/mlx.core.repeat.rst", "python/_autosummary/mlx.core.reset_peak_memory.rst", "python/_autosummary/mlx.core.reshape.rst", "python/_autosummary/mlx.core.right_shift.rst", "python/_autosummary/mlx.core.roll.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_cache_limit.rst", "python/_autosummary/mlx.core.set_default_device.rst", "python/_autosummary/mlx.core.set_default_stream.rst", "python/_autosummary/mlx.core.set_memory_limit.rst", "python/_autosummary/mlx.core.set_wired_limit.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.slice.rst", "python/_autosummary/mlx.core.slice_update.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.unflatten.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.average_gradients.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/export.rst", "python/fast.rst", "python/fft.rst", "python/linalg.rst", "python/memory_management.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.AvgPool3d.rst", "python/nn/_autosummary/mlx.nn.BatchNorm.rst", "python/nn/_autosummary/mlx.nn.CELU.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.ConvTranspose1d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose2d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose3d.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.ELU.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.LogSigmoid.rst", "python/nn/_autosummary/mlx.nn.LogSoftmax.rst", "python/nn/_autosummary/mlx.nn.MaxPool1d.rst", "python/nn/_autosummary/mlx.nn.MaxPool2d.rst", "python/nn/_autosummary/mlx.nn.MaxPool3d.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.Sigmoid.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.celu.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/export.rst", "usage/function_transforms.rst", "usage/indexing.rst", "usage/launching_distributed.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"], "indexentries": {"__init__() (array method)": [[31, "mlx.core.array.__init__", false]], "__init__() (custom_function method)": [[115, "mlx.core.custom_function.__init__", false]], "__init__() (device method)": [[10, "mlx.core.Device.__init__", false]], "__init__() (dtype method)": [[11, "mlx.core.Dtype.__init__", false]], "__init__() (dtypecategory method)": [[12, "mlx.core.DtypeCategory.__init__", false]], "__init__() (finfo method)": [[164, "mlx.core.finfo.__init__", false]], "__init__() (group method)": [[123, "mlx.core.distributed.Group.__init__", false]], "__init__() (stream method)": [[330, "mlx.core.Stream.__init__", false]], "abs (c++ function)": [[0, "_CPPv43absRK5array14StreamOrDevice", false]], "abs() (array method)": [[33, "mlx.core.array.abs", false]], "abs() (in module mlx.core)": [[13, "mlx.core.abs", false]], "adadelta (class in mlx.optimizers)": [[474, "mlx.optimizers.AdaDelta", false]], "adafactor (class in mlx.optimizers)": [[475, "mlx.optimizers.Adafactor", false]], "adagrad (class in mlx.optimizers)": [[476, "mlx.optimizers.Adagrad", false]], "adam (class in mlx.optimizers)": [[477, "mlx.optimizers.Adam", false]], "adamax (class in mlx.optimizers)": [[479, "mlx.optimizers.Adamax", false]], "adamw (class in mlx.optimizers)": [[478, "mlx.optimizers.AdamW", false]], "add (c++ function)": [[0, "_CPPv43addRK5arrayRK5array14StreamOrDevice", false]], "add() (in module mlx.core)": [[14, "mlx.core.add", false]], "addmm (c++ function)": [[0, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", false]], "addmm() (in module mlx.core)": [[15, "mlx.core.addmm", false]], "alibi (class in mlx.nn)": [[342, "mlx.nn.ALiBi", false]], "all (c++ function)": [[0, "_CPPv43allRK5array14StreamOrDevice", false], [0, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43allRK5arrayb14StreamOrDevice", false], [0, "_CPPv43allRK5arrayib14StreamOrDevice", false]], "all() (array method)": [[34, "mlx.core.array.all", false]], "all() (in module mlx.core)": [[16, "mlx.core.all", false]], "all_gather() (in module mlx.core.distributed)": [[124, "mlx.core.distributed.all_gather", false]], "all_sum() (in module mlx.core.distributed)": [[125, "mlx.core.distributed.all_sum", false]], "allclose (c++ function)": [[0, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "allclose() (in module mlx.core)": [[17, "mlx.core.allclose", false]], "any (c++ function)": [[0, "_CPPv43anyRK5array14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayb14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayib14StreamOrDevice", false]], "any() (array method)": [[35, "mlx.core.array.any", false]], "any() (in module mlx.core)": [[18, "mlx.core.any", false]], "apply() (module method)": [[377, "mlx.nn.Module.apply", false]], "apply_gradients() (optimizer method)": [[481, "mlx.optimizers.Optimizer.apply_gradients", false]], "apply_to_modules() (module method)": [[378, "mlx.nn.Module.apply_to_modules", false]], "arange (c++ function)": [[0, "_CPPv46aranged14StreamOrDevice", false], [0, "_CPPv46aranged5Dtype14StreamOrDevice", false], [0, "_CPPv46arangedd14StreamOrDevice", false], [0, "_CPPv46arangedd5Dtype14StreamOrDevice", false], [0, "_CPPv46arangeddd14StreamOrDevice", false], [0, "_CPPv46arangeddd5Dtype14StreamOrDevice", false], [0, "_CPPv46arangei14StreamOrDevice", false], [0, "_CPPv46arangeii14StreamOrDevice", false], [0, "_CPPv46arangeiii14StreamOrDevice", false]], "arange() (in module mlx.core)": [[19, "mlx.core.arange", false]], "arccos (c++ function)": [[0, "_CPPv46arccosRK5array14StreamOrDevice", false]], "arccos() (in module mlx.core)": [[20, "mlx.core.arccos", false]], "arccosh (c++ function)": [[0, "_CPPv47arccoshRK5array14StreamOrDevice", false]], "arccosh() (in module mlx.core)": [[21, "mlx.core.arccosh", false]], "arcsin (c++ function)": [[0, "_CPPv46arcsinRK5array14StreamOrDevice", false]], "arcsin() (in module mlx.core)": [[22, "mlx.core.arcsin", false]], "arcsinh (c++ function)": [[0, "_CPPv47arcsinhRK5array14StreamOrDevice", false]], "arcsinh() (in module mlx.core)": [[23, "mlx.core.arcsinh", false]], "arctan (c++ function)": [[0, "_CPPv46arctanRK5array14StreamOrDevice", false]], "arctan() (in module mlx.core)": [[24, "mlx.core.arctan", false]], "arctan2 (c++ function)": [[0, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", false]], "arctan2() (in module mlx.core)": [[25, "mlx.core.arctan2", false]], "arctanh (c++ function)": [[0, "_CPPv47arctanhRK5array14StreamOrDevice", false]], "arctanh() (in module mlx.core)": [[26, "mlx.core.arctanh", false]], "argmax (c++ function)": [[0, "_CPPv46argmaxRK5array14StreamOrDevice", false], [0, "_CPPv46argmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv46argmaxRK5arrayib14StreamOrDevice", false]], "argmax() (array method)": [[36, "mlx.core.array.argmax", false]], "argmax() (in module mlx.core)": [[27, "mlx.core.argmax", false]], "argmin (c++ function)": [[0, "_CPPv46argminRK5array14StreamOrDevice", false], [0, "_CPPv46argminRK5arrayb14StreamOrDevice", false], [0, "_CPPv46argminRK5arrayib14StreamOrDevice", false]], "argmin() (array method)": [[37, "mlx.core.array.argmin", false]], "argmin() (in module mlx.core)": [[28, "mlx.core.argmin", false]], "argpartition (c++ function)": [[0, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", false]], "argpartition() (in module mlx.core)": [[29, "mlx.core.argpartition", false]], "argsort (c++ function)": [[0, "_CPPv47argsortRK5array14StreamOrDevice", false], [0, "_CPPv47argsortRK5arrayi14StreamOrDevice", false]], "argsort() (in module mlx.core)": [[30, "mlx.core.argsort", false]], "array (class in mlx.core)": [[31, "mlx.core.array", false]], "array_equal (c++ function)": [[0, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", false], [0, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", false]], "array_equal() (in module mlx.core)": [[83, "mlx.core.array_equal", false]], "as_strided (c++ function)": [[0, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", false]], "as_strided() (in module mlx.core)": [[84, "mlx.core.as_strided", false]], "astype (c++ function)": [[0, "_CPPv46astype5array5Dtype14StreamOrDevice", false]], "astype() (array method)": [[38, "mlx.core.array.astype", false]], "at (array property)": [[39, "mlx.core.array.at", false]], "atleast_1d (c++ function)": [[0, "_CPPv410atleast_1dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_1d() (in module mlx.core)": [[85, "mlx.core.atleast_1d", false]], "atleast_2d (c++ function)": [[0, "_CPPv410atleast_2dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_2d() (in module mlx.core)": [[86, "mlx.core.atleast_2d", false]], "atleast_3d (c++ function)": [[0, "_CPPv410atleast_3dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_3d() (in module mlx.core)": [[87, "mlx.core.atleast_3d", false]], "average_gradients() (in module mlx.nn)": [[321, "mlx.nn.average_gradients", false]], "avgpool1d (class in mlx.nn)": [[343, "mlx.nn.AvgPool1d", false]], "avgpool2d (class in mlx.nn)": [[344, "mlx.nn.AvgPool2d", false]], "avgpool3d (class in mlx.nn)": [[345, "mlx.nn.AvgPool3d", false]], "batchnorm (class in mlx.nn)": [[346, "mlx.nn.BatchNorm", false]], "bernoulli() (in module mlx.core.random)": [[247, "mlx.core.random.bernoulli", false]], "binary_cross_entropy (class in mlx.nn.losses)": [[440, "mlx.nn.losses.binary_cross_entropy", false]], "bitwise_and (c++ function)": [[0, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", false]], "bitwise_and() (in module mlx.core)": [[88, "mlx.core.bitwise_and", false]], "bitwise_invert (c++ function)": [[0, "_CPPv414bitwise_invertRK5array14StreamOrDevice", false]], "bitwise_invert() (in module mlx.core)": [[89, "mlx.core.bitwise_invert", false]], "bitwise_or (c++ function)": [[0, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", false]], "bitwise_or() (in module mlx.core)": [[90, "mlx.core.bitwise_or", false]], "bitwise_xor (c++ function)": [[0, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", false]], "bitwise_xor() (in module mlx.core)": [[91, "mlx.core.bitwise_xor", false]], "block_masked_mm (c++ function)": [[0, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "block_masked_mm() (in module mlx.core)": [[92, "mlx.core.block_masked_mm", false]], "broadcast_arrays (c++ function)": [[0, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "broadcast_to (c++ function)": [[0, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", false]], "broadcast_to() (in module mlx.core)": [[93, "mlx.core.broadcast_to", false]], "categorical() (in module mlx.core.random)": [[248, "mlx.core.random.categorical", false]], "ceil (c++ function)": [[0, "_CPPv44ceilRK5array14StreamOrDevice", false]], "ceil() (in module mlx.core)": [[94, "mlx.core.ceil", false]], "celu (class in mlx.nn)": [[347, "mlx.nn.CELU", false], [428, "mlx.nn.celu", false]], "children() (module method)": [[379, "mlx.nn.Module.children", false]], "cholesky() (in module mlx.core.linalg)": [[194, "mlx.core.linalg.cholesky", false]], "cholesky_inv() (in module mlx.core.linalg)": [[195, "mlx.core.linalg.cholesky_inv", false]], "clear_cache() (in module mlx.core)": [[95, "mlx.core.clear_cache", false]], "clip (c++ function)": [[0, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", false]], "clip() (in module mlx.core)": [[96, "mlx.core.clip", false]], "clip_grad_norm() (in module mlx.optimizers)": [[324, "mlx.optimizers.clip_grad_norm", false]], "compile() (in module mlx.core)": [[97, "mlx.core.compile", false]], "concatenate (c++ function)": [[0, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", false]], "concatenate() (in module mlx.core)": [[98, "mlx.core.concatenate", false]], "conj() (array method)": [[40, "mlx.core.array.conj", false]], "conj() (in module mlx.core)": [[99, "mlx.core.conj", false]], "conjugate (c++ function)": [[0, "_CPPv49conjugateRK5array14StreamOrDevice", false]], "conjugate() (in module mlx.core)": [[100, "mlx.core.conjugate", false]], "constant() (in module mlx.nn.init)": [[420, "mlx.nn.init.constant", false]], "contiguous (c++ function)": [[0, "_CPPv410contiguousRK5arrayb14StreamOrDevice", false]], "conv1d (c++ function)": [[0, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv1d (class in mlx.nn)": [[348, "mlx.nn.Conv1d", false]], "conv1d() (in module mlx.core)": [[101, "mlx.core.conv1d", false]], "conv2d (c++ function)": [[0, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv2d (class in mlx.nn)": [[349, "mlx.nn.Conv2d", false]], "conv2d() (in module mlx.core)": [[102, "mlx.core.conv2d", false]], "conv3d (c++ function)": [[0, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv3d (class in mlx.nn)": [[350, "mlx.nn.Conv3d", false]], "conv3d() (in module mlx.core)": [[103, "mlx.core.conv3d", false]], "conv_general (c++ function)": [[0, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false], [0, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false]], "conv_general() (in module mlx.core)": [[104, "mlx.core.conv_general", false]], "conv_transpose1d (c++ function)": [[0, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv_transpose1d() (in module mlx.core)": [[105, "mlx.core.conv_transpose1d", false]], "conv_transpose2d (c++ function)": [[0, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv_transpose2d() (in module mlx.core)": [[106, "mlx.core.conv_transpose2d", false]], "conv_transpose3d (c++ function)": [[0, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv_transpose3d() (in module mlx.core)": [[107, "mlx.core.conv_transpose3d", false]], "convolve() (in module mlx.core)": [[108, "mlx.core.convolve", false]], "convtranspose1d (class in mlx.nn)": [[351, "mlx.nn.ConvTranspose1d", false]], "convtranspose2d (class in mlx.nn)": [[352, "mlx.nn.ConvTranspose2d", false]], "convtranspose3d (class in mlx.nn)": [[353, "mlx.nn.ConvTranspose3d", false]], "copy (c++ function)": [[0, "_CPPv44copy5array14StreamOrDevice", false]], "cos (c++ function)": [[0, "_CPPv43cosRK5array14StreamOrDevice", false]], "cos() (array method)": [[41, "mlx.core.array.cos", false]], "cos() (in module mlx.core)": [[109, "mlx.core.cos", false]], "cosh (c++ function)": [[0, "_CPPv44coshRK5array14StreamOrDevice", false]], "cosh() (in module mlx.core)": [[110, "mlx.core.cosh", false]], "cosine_decay() (in module mlx.optimizers)": [[487, "mlx.optimizers.cosine_decay", false]], "cosine_similarity_loss (class in mlx.nn.losses)": [[441, "mlx.nn.losses.cosine_similarity_loss", false]], "cross() (in module mlx.core.linalg)": [[196, "mlx.core.linalg.cross", false]], "cross_entropy (class in mlx.nn.losses)": [[442, "mlx.nn.losses.cross_entropy", false]], "cummax (c++ function)": [[0, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", false]], "cummax() (array method)": [[42, "mlx.core.array.cummax", false]], "cummax() (in module mlx.core)": [[111, "mlx.core.cummax", false]], "cummin (c++ function)": [[0, "_CPPv46cumminRK5arrayibb14StreamOrDevice", false]], "cummin() (array method)": [[43, "mlx.core.array.cummin", false]], "cummin() (in module mlx.core)": [[112, "mlx.core.cummin", false]], "cumprod (c++ function)": [[0, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", false]], "cumprod() (array method)": [[44, "mlx.core.array.cumprod", false]], "cumprod() (in module mlx.core)": [[113, "mlx.core.cumprod", false]], "cumsum (c++ function)": [[0, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", false]], "cumsum() (array method)": [[45, "mlx.core.array.cumsum", false]], "cumsum() (in module mlx.core)": [[114, "mlx.core.cumsum", false]], "custom_function (class in mlx.core)": [[115, "mlx.core.custom_function", false]], "default_device() (in module mlx.core)": [[116, "mlx.core.default_device", false]], "default_stream() (in module mlx.core)": [[117, "mlx.core.default_stream", false]], "degrees (c++ function)": [[0, "_CPPv47degreesRK5array14StreamOrDevice", false]], "degrees() (in module mlx.core)": [[118, "mlx.core.degrees", false]], "depends (c++ function)": [[0, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", false]], "dequantize (c++ function)": [[0, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", false]], "dequantize() (in module mlx.core)": [[119, "mlx.core.dequantize", false]], "device (class in mlx.core)": [[10, "mlx.core.Device", false]], "device_info() (in module mlx.core.metal)": [[224, "mlx.core.metal.device_info", false]], "diag (c++ function)": [[0, "_CPPv44diagRK5arrayi14StreamOrDevice", false]], "diag() (array method)": [[46, "mlx.core.array.diag", false]], "diag() (in module mlx.core)": [[120, "mlx.core.diag", false]], "diagonal (c++ function)": [[0, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", false]], "diagonal() (array method)": [[47, "mlx.core.array.diagonal", false]], "diagonal() (in module mlx.core)": [[121, "mlx.core.diagonal", false]], "disable_compile() (in module mlx.core)": [[122, "mlx.core.disable_compile", false]], "divide (c++ function)": [[0, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", false]], "divide() (in module mlx.core)": [[131, "mlx.core.divide", false]], "divmod (c++ function)": [[0, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", false]], "divmod() (in module mlx.core)": [[132, "mlx.core.divmod", false]], "dropout (class in mlx.nn)": [[354, "mlx.nn.Dropout", false]], "dropout2d (class in mlx.nn)": [[355, "mlx.nn.Dropout2d", false]], "dropout3d (class in mlx.nn)": [[356, "mlx.nn.Dropout3d", false]], "dtype (array property)": [[48, "mlx.core.array.dtype", false]], "dtype (class in mlx.core)": [[11, "mlx.core.Dtype", false]], "dtypecategory (class in mlx.core)": [[12, "mlx.core.DtypeCategory", false]], "eigh() (in module mlx.core.linalg)": [[197, "mlx.core.linalg.eigh", false]], "eigvalsh() (in module mlx.core.linalg)": [[198, "mlx.core.linalg.eigvalsh", false]], "einsum() (in module mlx.core)": [[133, "mlx.core.einsum", false]], "einsum_path() (in module mlx.core)": [[134, "mlx.core.einsum_path", false]], "elu (class in mlx.nn)": [[357, "mlx.nn.ELU", false], [429, "mlx.nn.elu", false]], "embedding (class in mlx.nn)": [[358, "mlx.nn.Embedding", false]], "enable_compile() (in module mlx.core)": [[135, "mlx.core.enable_compile", false]], "equal (c++ function)": [[0, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", false]], "equal() (in module mlx.core)": [[136, "mlx.core.equal", false]], "erf (c++ function)": [[0, "_CPPv43erfRK5array14StreamOrDevice", false]], "erf() (in module mlx.core)": [[137, "mlx.core.erf", false]], "erfinv (c++ function)": [[0, "_CPPv46erfinvRK5array14StreamOrDevice", false]], "erfinv() (in module mlx.core)": [[138, "mlx.core.erfinv", false]], "eval() (in module mlx.core)": [[139, "mlx.core.eval", false]], "eval() (module method)": [[380, "mlx.nn.Module.eval", false]], "exp (c++ function)": [[0, "_CPPv43expRK5array14StreamOrDevice", false]], "exp() (array method)": [[49, "mlx.core.array.exp", false]], "exp() (in module mlx.core)": [[140, "mlx.core.exp", false]], "expand_dims (c++ function)": [[0, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", false]], "expand_dims() (in module mlx.core)": [[141, "mlx.core.expand_dims", false]], "expm1 (c++ function)": [[0, "_CPPv45expm1RK5array14StreamOrDevice", false]], "expm1() (in module mlx.core)": [[142, "mlx.core.expm1", false]], "exponential_decay() (in module mlx.optimizers)": [[488, "mlx.optimizers.exponential_decay", false]], "export_function() (in module mlx.core)": [[143, "mlx.core.export_function", false]], "export_to_dot() (in module mlx.core)": [[144, "mlx.core.export_to_dot", false]], "exporter() (in module mlx.core)": [[145, "mlx.core.exporter", false]], "eye (c++ function)": [[0, "_CPPv43eyei14StreamOrDevice", false], [0, "_CPPv43eyei5Dtype14StreamOrDevice", false], [0, "_CPPv43eyeii14StreamOrDevice", false], [0, "_CPPv43eyeiii14StreamOrDevice", false], [0, "_CPPv43eyeiii5Dtype14StreamOrDevice", false]], "eye() (in module mlx.core)": [[146, "mlx.core.eye", false]], "fft() (in module mlx.core.fft)": [[152, "mlx.core.fft.fft", false]], "fft2() (in module mlx.core.fft)": [[153, "mlx.core.fft.fft2", false]], "fftn() (in module mlx.core.fft)": [[154, "mlx.core.fft.fftn", false]], "filter_and_map() (module method)": [[381, "mlx.nn.Module.filter_and_map", false]], "finfo (class in mlx.core)": [[164, "mlx.core.finfo", false]], "flatten (c++ function)": [[0, "_CPPv47flattenRK5array14StreamOrDevice", false], [0, "_CPPv47flattenRK5arrayii14StreamOrDevice", false]], "flatten() (array method)": [[50, "mlx.core.array.flatten", false]], "flatten() (in module mlx.core)": [[165, "mlx.core.flatten", false]], "floor (c++ function)": [[0, "_CPPv45floorRK5array14StreamOrDevice", false]], "floor() (in module mlx.core)": [[166, "mlx.core.floor", false]], "floor_divide (c++ function)": [[0, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", false]], "floor_divide() (in module mlx.core)": [[167, "mlx.core.floor_divide", false]], "freeze() (module method)": [[382, "mlx.nn.Module.freeze", false]], "full (c++ function)": [[0, "_CPPv44full5Shape5array14StreamOrDevice", false], [0, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", false], [0, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", false], [0, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", false]], "full() (in module mlx.core)": [[168, "mlx.core.full", false]], "gather (c++ function)": [[0, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", false], [0, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", false]], "gather_mm (c++ function)": [[0, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "gather_mm() (in module mlx.core)": [[169, "mlx.core.gather_mm", false]], "gather_qmm (c++ function)": [[0, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", false]], "gather_qmm() (in module mlx.core)": [[170, "mlx.core.gather_qmm", false]], "gaussian_nll_loss (class in mlx.nn.losses)": [[443, "mlx.nn.losses.gaussian_nll_loss", false]], "gelu (class in mlx.nn)": [[359, "mlx.nn.GELU", false], [430, "mlx.nn.gelu", false]], "gelu_approx (class in mlx.nn)": [[431, "mlx.nn.gelu_approx", false]], "gelu_fast_approx (class in mlx.nn)": [[432, "mlx.nn.gelu_fast_approx", false]], "get_active_memory() (in module mlx.core)": [[171, "mlx.core.get_active_memory", false]], "get_cache_memory() (in module mlx.core)": [[172, "mlx.core.get_cache_memory", false]], "get_peak_memory() (in module mlx.core)": [[173, "mlx.core.get_peak_memory", false]], "glorot_normal() (in module mlx.nn.init)": [[421, "mlx.nn.init.glorot_normal", false]], "glorot_uniform() (in module mlx.nn.init)": [[422, "mlx.nn.init.glorot_uniform", false]], "glu (class in mlx.nn)": [[360, "mlx.nn.GLU", false], [433, "mlx.nn.glu", false]], "grad() (in module mlx.core)": [[174, "mlx.core.grad", false]], "greater (c++ function)": [[0, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", false]], "greater() (in module mlx.core)": [[175, "mlx.core.greater", false]], "greater_equal (c++ function)": [[0, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", false]], "greater_equal() (in module mlx.core)": [[176, "mlx.core.greater_equal", false]], "group (class in mlx.core.distributed)": [[123, "mlx.core.distributed.Group", false]], "groupnorm (class in mlx.nn)": [[362, "mlx.nn.GroupNorm", false]], "gru (class in mlx.nn)": [[361, "mlx.nn.GRU", false]], "gumbel() (in module mlx.core.random)": [[249, "mlx.core.random.gumbel", false]], "hadamard_transform (c++ function)": [[0, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", false]], "hadamard_transform() (in module mlx.core)": [[177, "mlx.core.hadamard_transform", false]], "hard_shrink (class in mlx.nn)": [[434, "mlx.nn.hard_shrink", false]], "hard_tanh (class in mlx.nn)": [[435, "mlx.nn.hard_tanh", false]], "hardshrink (class in mlx.nn)": [[363, "mlx.nn.HardShrink", false]], "hardswish (class in mlx.nn)": [[365, "mlx.nn.Hardswish", false], [436, "mlx.nn.hardswish", false]], "hardtanh (class in mlx.nn)": [[364, "mlx.nn.HardTanh", false]], "he_normal() (in module mlx.nn.init)": [[423, "mlx.nn.init.he_normal", false]], "he_uniform() (in module mlx.nn.init)": [[424, "mlx.nn.init.he_uniform", false]], "hinge_loss (class in mlx.nn.losses)": [[444, "mlx.nn.losses.hinge_loss", false]], "huber_loss (class in mlx.nn.losses)": [[445, "mlx.nn.losses.huber_loss", false]], "identity (c++ function)": [[0, "_CPPv48identityi14StreamOrDevice", false], [0, "_CPPv48identityi5Dtype14StreamOrDevice", false]], "identity() (in module mlx.core)": [[178, "mlx.core.identity", false]], "identity() (in module mlx.nn.init)": [[425, "mlx.nn.init.identity", false]], "ifft() (in module mlx.core.fft)": [[155, "mlx.core.fft.ifft", false]], "ifft2() (in module mlx.core.fft)": [[156, "mlx.core.fft.ifft2", false]], "ifftn() (in module mlx.core.fft)": [[157, "mlx.core.fft.ifftn", false]], "imag (c++ function)": [[0, "_CPPv44imagRK5array14StreamOrDevice", false]], "imag() (in module mlx.core)": [[179, "mlx.core.imag", false]], "import_function() (in module mlx.core)": [[180, "mlx.core.import_function", false]], "init() (in module mlx.core.distributed)": [[126, "mlx.core.distributed.init", false]], "init() (optimizer method)": [[482, "mlx.optimizers.Optimizer.init", false]], "inner (c++ function)": [[0, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", false]], "inner() (in module mlx.core)": [[181, "mlx.core.inner", false]], "instancenorm (class in mlx.nn)": [[366, "mlx.nn.InstanceNorm", false]], "inv() (in module mlx.core.linalg)": [[199, "mlx.core.linalg.inv", false]], "irfft() (in module mlx.core.fft)": [[158, "mlx.core.fft.irfft", false]], "irfft2() (in module mlx.core.fft)": [[159, "mlx.core.fft.irfft2", false]], "irfftn() (in module mlx.core.fft)": [[160, "mlx.core.fft.irfftn", false]], "is_available() (in module mlx.core.distributed)": [[127, "mlx.core.distributed.is_available", false]], "is_available() (in module mlx.core.metal)": [[225, "mlx.core.metal.is_available", false]], "isclose (c++ function)": [[0, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "isclose() (in module mlx.core)": [[182, "mlx.core.isclose", false]], "isfinite (c++ function)": [[0, "_CPPv48isfiniteRK5array14StreamOrDevice", false]], "isfinite() (in module mlx.core)": [[183, "mlx.core.isfinite", false]], "isinf (c++ function)": [[0, "_CPPv45isinfRK5array14StreamOrDevice", false]], "isinf() (in module mlx.core)": [[184, "mlx.core.isinf", false]], "isnan (c++ function)": [[0, "_CPPv45isnanRK5array14StreamOrDevice", false]], "isnan() (in module mlx.core)": [[185, "mlx.core.isnan", false]], "isneginf (c++ function)": [[0, "_CPPv48isneginfRK5array14StreamOrDevice", false]], "isneginf() (in module mlx.core)": [[186, "mlx.core.isneginf", false]], "isposinf (c++ function)": [[0, "_CPPv48isposinfRK5array14StreamOrDevice", false]], "isposinf() (in module mlx.core)": [[187, "mlx.core.isposinf", false]], "issubdtype() (in module mlx.core)": [[188, "mlx.core.issubdtype", false]], "item() (array method)": [[51, "mlx.core.array.item", false]], "itemsize (array property)": [[52, "mlx.core.array.itemsize", false]], "join_schedules() (in module mlx.optimizers)": [[489, "mlx.optimizers.join_schedules", false]], "jvp() (in module mlx.core)": [[189, "mlx.core.jvp", false]], "key() (in module mlx.core.random)": [[250, "mlx.core.random.key", false]], "kl_div_loss (class in mlx.nn.losses)": [[446, "mlx.nn.losses.kl_div_loss", false]], "kron (c++ function)": [[0, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", false]], "kron() (in module mlx.core)": [[190, "mlx.core.kron", false]], "l1_loss (class in mlx.nn.losses)": [[447, "mlx.nn.losses.l1_loss", false]], "laplace() (in module mlx.core.random)": [[251, "mlx.core.random.laplace", false]], "layer_norm() (in module mlx.core.fast)": [[147, "mlx.core.fast.layer_norm", false]], "layernorm (class in mlx.nn)": [[368, "mlx.nn.LayerNorm", false]], "leaf_modules() (module method)": [[383, "mlx.nn.Module.leaf_modules", false]], "leaky_relu (class in mlx.nn)": [[437, "mlx.nn.leaky_relu", false]], "leakyrelu (class in mlx.nn)": [[369, "mlx.nn.LeakyReLU", false]], "left_shift (c++ function)": [[0, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", false]], "left_shift() (in module mlx.core)": [[191, "mlx.core.left_shift", false]], "less (c++ function)": [[0, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", false]], "less() (in module mlx.core)": [[192, "mlx.core.less", false]], "less_equal (c++ function)": [[0, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", false]], "less_equal() (in module mlx.core)": [[193, "mlx.core.less_equal", false]], "linear (class in mlx.nn)": [[370, "mlx.nn.Linear", false]], "linear_schedule() (in module mlx.optimizers)": [[490, "mlx.optimizers.linear_schedule", false]], "linspace (c++ function)": [[0, "_CPPv48linspaceddi5Dtype14StreamOrDevice", false]], "linspace() (in module mlx.core)": [[208, "mlx.core.linspace", false]], "lion (class in mlx.optimizers)": [[480, "mlx.optimizers.Lion", false]], "load() (in module mlx.core)": [[209, "mlx.core.load", false]], "load_weights() (module method)": [[384, "mlx.nn.Module.load_weights", false]], "log (c++ function)": [[0, "_CPPv43logRK5array14StreamOrDevice", false]], "log() (array method)": [[53, "mlx.core.array.log", false]], "log() (in module mlx.core)": [[210, "mlx.core.log", false]], "log10 (c++ function)": [[0, "_CPPv45log10RK5array14StreamOrDevice", false]], "log10() (array method)": [[54, "mlx.core.array.log10", false]], "log10() (in module mlx.core)": [[211, "mlx.core.log10", false]], "log1p (c++ function)": [[0, "_CPPv45log1pRK5array14StreamOrDevice", false]], "log1p() (array method)": [[55, "mlx.core.array.log1p", false]], "log1p() (in module mlx.core)": [[212, "mlx.core.log1p", false]], "log2 (c++ function)": [[0, "_CPPv44log2RK5array14StreamOrDevice", false]], "log2() (array method)": [[56, "mlx.core.array.log2", false]], "log2() (in module mlx.core)": [[213, "mlx.core.log2", false]], "log_cosh_loss (class in mlx.nn.losses)": [[448, "mlx.nn.losses.log_cosh_loss", false]], "log_sigmoid (class in mlx.nn)": [[438, "mlx.nn.log_sigmoid", false]], "log_softmax (class in mlx.nn)": [[439, "mlx.nn.log_softmax", false]], "logaddexp (c++ function)": [[0, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", false]], "logaddexp() (in module mlx.core)": [[214, "mlx.core.logaddexp", false]], "logical_and (c++ function)": [[0, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", false]], "logical_and() (in module mlx.core)": [[215, "mlx.core.logical_and", false]], "logical_not (c++ function)": [[0, "_CPPv411logical_notRK5array14StreamOrDevice", false]], "logical_not() (in module mlx.core)": [[216, "mlx.core.logical_not", false]], "logical_or (c++ function)": [[0, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", false]], "logical_or() (in module mlx.core)": [[217, "mlx.core.logical_or", false]], "logsigmoid (class in mlx.nn)": [[371, "mlx.nn.LogSigmoid", false]], "logsoftmax (class in mlx.nn)": [[372, "mlx.nn.LogSoftmax", false]], "logsumexp (c++ function)": [[0, "_CPPv49logsumexpRK5array14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", false]], "logsumexp() (array method)": [[57, "mlx.core.array.logsumexp", false]], "logsumexp() (in module mlx.core)": [[218, "mlx.core.logsumexp", false]], "lstm (class in mlx.nn)": [[367, "mlx.nn.LSTM", false]], "lu() (in module mlx.core.linalg)": [[200, "mlx.core.linalg.lu", false]], "lu_factor() (in module mlx.core.linalg)": [[201, "mlx.core.linalg.lu_factor", false]], "margin_ranking_loss (class in mlx.nn.losses)": [[449, "mlx.nn.losses.margin_ranking_loss", false]], "matmul (c++ function)": [[0, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", false]], "matmul() (in module mlx.core)": [[219, "mlx.core.matmul", false]], "max (c++ function)": [[0, "_CPPv43maxRK5array14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayib14StreamOrDevice", false]], "max() (array method)": [[58, "mlx.core.array.max", false]], "max() (in module mlx.core)": [[220, "mlx.core.max", false]], "maximum (c++ function)": [[0, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", false]], "maximum() (in module mlx.core)": [[221, "mlx.core.maximum", false]], "maxpool1d (class in mlx.nn)": [[373, "mlx.nn.MaxPool1d", false]], "maxpool2d (class in mlx.nn)": [[374, "mlx.nn.MaxPool2d", false]], "maxpool3d (class in mlx.nn)": [[375, "mlx.nn.MaxPool3d", false]], "mean (c++ function)": [[0, "_CPPv44meanRK5array14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayib14StreamOrDevice", false]], "mean() (array method)": [[59, "mlx.core.array.mean", false]], "mean() (in module mlx.core)": [[222, "mlx.core.mean", false]], "meshgrid (c++ function)": [[0, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", false]], "meshgrid() (in module mlx.core)": [[223, "mlx.core.meshgrid", false]], "metal_kernel() (in module mlx.core.fast)": [[148, "mlx.core.fast.metal_kernel", false]], "min (c++ function)": [[0, "_CPPv43minRK5array14StreamOrDevice", false], [0, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayib14StreamOrDevice", false]], "min() (array method)": [[60, "mlx.core.array.min", false]], "min() (in module mlx.core)": [[228, "mlx.core.min", false]], "minimum (c++ function)": [[0, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", false]], "minimum() (in module mlx.core)": [[229, "mlx.core.minimum", false]], "mish (class in mlx.nn)": [[376, "mlx.nn.Mish", false], [454, "mlx.nn.mish", false]], "module (class in mlx.nn)": [[471, "mlx.nn.Module", false]], "modules() (module method)": [[385, "mlx.nn.Module.modules", false]], "moveaxis (c++ function)": [[0, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", false]], "moveaxis() (array method)": [[61, "mlx.core.array.moveaxis", false]], "moveaxis() (in module mlx.core)": [[230, "mlx.core.moveaxis", false]], "mse_loss (class in mlx.nn.losses)": [[450, "mlx.nn.losses.mse_loss", false]], "multiheadattention (class in mlx.nn)": [[397, "mlx.nn.MultiHeadAttention", false]], "multiply (c++ function)": [[0, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", false]], "multiply() (in module mlx.core)": [[231, "mlx.core.multiply", false]], "multivariate_normal() (in module mlx.core.random)": [[252, "mlx.core.random.multivariate_normal", false]], "named_modules() (module method)": [[386, "mlx.nn.Module.named_modules", false]], "nan_to_num (c++ function)": [[0, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", false]], "nan_to_num() (in module mlx.core)": [[232, "mlx.core.nan_to_num", false]], "nbytes (array property)": [[62, "mlx.core.array.nbytes", false]], "ndim (array property)": [[63, "mlx.core.array.ndim", false]], "negative (c++ function)": [[0, "_CPPv48negativeRK5array14StreamOrDevice", false]], "negative() (in module mlx.core)": [[233, "mlx.core.negative", false]], "new_stream() (in module mlx.core)": [[234, "mlx.core.new_stream", false]], "nll_loss (class in mlx.nn.losses)": [[451, "mlx.nn.losses.nll_loss", false]], "norm() (in module mlx.core.linalg)": [[202, "mlx.core.linalg.norm", false]], "normal() (in module mlx.core.random)": [[253, "mlx.core.random.normal", false]], "normal() (in module mlx.nn.init)": [[426, "mlx.nn.init.normal", false]], "not_equal (c++ function)": [[0, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", false]], "not_equal() (in module mlx.core)": [[235, "mlx.core.not_equal", false]], "number_of_elements (c++ function)": [[0, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", false]], "ones (c++ function)": [[0, "_CPPv44onesRK5Shape14StreamOrDevice", false], [0, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", false]], "ones() (in module mlx.core)": [[236, "mlx.core.ones", false]], "ones_like (c++ function)": [[0, "_CPPv49ones_likeRK5array14StreamOrDevice", false]], "ones_like() (in module mlx.core)": [[237, "mlx.core.ones_like", false]], "operator!= (c++ function)": [[0, "_CPPv4I0Ene5array1TRK5array", false], [0, "_CPPv4I0Ene5arrayRK5array1T", false], [0, "_CPPv4neRK5arrayRK5array", false]], "operator% (c++ function)": [[0, "_CPPv4I0Erm5array1TRK5array", false], [0, "_CPPv4I0Erm5arrayRK5array1T", false], [0, "_CPPv4rmRK5arrayRK5array", false]], "operator& (c++ function)": [[0, "_CPPv4anRK5arrayRK5array", false]], "operator&& (c++ function)": [[0, "_CPPv4aaRK5arrayRK5array", false]], "operator* (c++ function)": [[0, "_CPPv4I0Eml5array1TRK5array", false], [0, "_CPPv4I0Eml5arrayRK5array1T", false], [0, "_CPPv4mlRK5arrayRK5array", false]], "operator+ (c++ function)": [[0, "_CPPv4I0Epl5array1TRK5array", false], [0, "_CPPv4I0Epl5arrayRK5array1T", false], [0, "_CPPv4plRK5arrayRK5array", false]], "operator- (c++ function)": [[0, "_CPPv4I0Emi5array1TRK5array", false], [0, "_CPPv4I0Emi5arrayRK5array1T", false], [0, "_CPPv4miRK5array", false], [0, "_CPPv4miRK5arrayRK5array", false]], "operator/ (c++ function)": [[0, "_CPPv4dvRK5arrayRK5array", false], [0, "_CPPv4dvRK5arrayd", false], [0, "_CPPv4dvdRK5array", false]], "operator< (c++ function)": [[0, "_CPPv4I0Elt5array1TRK5array", false], [0, "_CPPv4I0Elt5arrayRK5array1T", false], [0, "_CPPv4ltRK5arrayRK5array", false]], "operator<< (c++ function)": [[0, "_CPPv4lsRK5arrayRK5array", false]], "operator<= (c++ function)": [[0, "_CPPv4I0Ele5array1TRK5array", false], [0, "_CPPv4I0Ele5arrayRK5array1T", false], [0, "_CPPv4leRK5arrayRK5array", false]], "operator== (c++ function)": [[0, "_CPPv4I0Eeq5array1TRK5array", false], [0, "_CPPv4I0Eeq5arrayRK5array1T", false], [0, "_CPPv4eqRK5arrayRK5array", false]], "operator> (c++ function)": [[0, "_CPPv4I0Egt5array1TRK5array", false], [0, "_CPPv4I0Egt5arrayRK5array1T", false], [0, "_CPPv4gtRK5arrayRK5array", false]], "operator>= (c++ function)": [[0, "_CPPv4I0Ege5array1TRK5array", false], [0, "_CPPv4I0Ege5arrayRK5array1T", false], [0, "_CPPv4geRK5arrayRK5array", false]], "operator>> (c++ function)": [[0, "_CPPv4rsRK5arrayRK5array", false]], "operator^ (c++ function)": [[0, "_CPPv4eoRK5arrayRK5array", false]], "operator| (c++ function)": [[0, "_CPPv4orRK5arrayRK5array", false]], "operator|| (c++ function)": [[0, "_CPPv4ooRK5arrayRK5array", false]], "operator~ (c++ function)": [[0, "_CPPv4coRK5array", false]], "optimizer (class in mlx.optimizers)": [[493, "mlx.optimizers.Optimizer", false]], "outer (c++ function)": [[0, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", false]], "outer() (in module mlx.core)": [[238, "mlx.core.outer", false]], "pad (c++ function)": [[0, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", false]], "pad() (in module mlx.core)": [[239, "mlx.core.pad", false]], "parameters() (module method)": [[387, "mlx.nn.Module.parameters", false]], "partition (c++ function)": [[0, "_CPPv49partitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv49partitionRK5arrayii14StreamOrDevice", false]], "partition() (in module mlx.core)": [[240, "mlx.core.partition", false]], "permutation() (in module mlx.core.random)": [[254, "mlx.core.random.permutation", false]], "power (c++ function)": [[0, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", false]], "power() (in module mlx.core)": [[241, "mlx.core.power", false]], "prelu (class in mlx.nn)": [[398, "mlx.nn.PReLU", false], [455, "mlx.nn.prelu", false]], "prod (c++ function)": [[0, "_CPPv44prodRK5array14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayib14StreamOrDevice", false]], "prod() (array method)": [[64, "mlx.core.array.prod", false]], "prod() (in module mlx.core)": [[242, "mlx.core.prod", false]], "put_along_axis (c++ function)": [[0, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "put_along_axis() (in module mlx.core)": [[243, "mlx.core.put_along_axis", false]], "qr() (in module mlx.core.linalg)": [[203, "mlx.core.linalg.qr", false]], "quantize (c++ function)": [[0, "_CPPv48quantizeRK5arrayii14StreamOrDevice", false]], "quantize() (in module mlx.core)": [[244, "mlx.core.quantize", false]], "quantize() (in module mlx.nn)": [[322, "mlx.nn.quantize", false]], "quantized_matmul (c++ function)": [[0, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", false]], "quantized_matmul() (in module mlx.core)": [[245, "mlx.core.quantized_matmul", false]], "quantizedembedding (class in mlx.nn)": [[399, "mlx.nn.QuantizedEmbedding", false]], "quantizedlinear (class in mlx.nn)": [[400, "mlx.nn.QuantizedLinear", false]], "radians (c++ function)": [[0, "_CPPv47radiansRK5array14StreamOrDevice", false]], "radians() (in module mlx.core)": [[246, "mlx.core.radians", false]], "randint() (in module mlx.core.random)": [[255, "mlx.core.random.randint", false]], "real (c++ function)": [[0, "_CPPv44realRK5array14StreamOrDevice", false]], "real() (in module mlx.core)": [[260, "mlx.core.real", false]], "reciprocal (c++ function)": [[0, "_CPPv410reciprocalRK5array14StreamOrDevice", false]], "reciprocal() (array method)": [[65, "mlx.core.array.reciprocal", false]], "reciprocal() (in module mlx.core)": [[261, "mlx.core.reciprocal", false]], "recv() (in module mlx.core.distributed)": [[128, "mlx.core.distributed.recv", false]], "recv_like() (in module mlx.core.distributed)": [[129, "mlx.core.distributed.recv_like", false]], "relu (class in mlx.nn)": [[403, "mlx.nn.ReLU", false], [456, "mlx.nn.relu", false]], "relu6 (class in mlx.nn)": [[404, "mlx.nn.ReLU6", false], [457, "mlx.nn.relu6", false]], "remainder (c++ function)": [[0, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", false]], "remainder() (in module mlx.core)": [[262, "mlx.core.remainder", false]], "repeat (c++ function)": [[0, "_CPPv46repeatRK5arrayi14StreamOrDevice", false], [0, "_CPPv46repeatRK5arrayii14StreamOrDevice", false]], "repeat() (in module mlx.core)": [[263, "mlx.core.repeat", false]], "reset_peak_memory() (in module mlx.core)": [[264, "mlx.core.reset_peak_memory", false]], "reshape (c++ function)": [[0, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", false]], "reshape() (array method)": [[66, "mlx.core.array.reshape", false]], "reshape() (in module mlx.core)": [[265, "mlx.core.reshape", false]], "rfft() (in module mlx.core.fft)": [[161, "mlx.core.fft.rfft", false]], "rfft2() (in module mlx.core.fft)": [[162, "mlx.core.fft.rfft2", false]], "rfftn() (in module mlx.core.fft)": [[163, "mlx.core.fft.rfftn", false]], "right_shift (c++ function)": [[0, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", false]], "right_shift() (in module mlx.core)": [[266, "mlx.core.right_shift", false]], "rms_norm() (in module mlx.core.fast)": [[149, "mlx.core.fast.rms_norm", false]], "rmsnorm (class in mlx.nn)": [[401, "mlx.nn.RMSNorm", false]], "rmsprop (class in mlx.optimizers)": [[485, "mlx.optimizers.RMSprop", false]], "rnn (class in mlx.nn)": [[402, "mlx.nn.RNN", false]], "roll (c++ function)": [[0, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayi14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayii14StreamOrDevice", false]], "roll() (in module mlx.core)": [[267, "mlx.core.roll", false]], "rope (class in mlx.nn)": [[405, "mlx.nn.RoPE", false]], "rope() (in module mlx.core.fast)": [[150, "mlx.core.fast.rope", false]], "round (c++ function)": [[0, "_CPPv45roundRK5array14StreamOrDevice", false], [0, "_CPPv45roundRK5arrayi14StreamOrDevice", false]], "round() (array method)": [[67, "mlx.core.array.round", false]], "round() (in module mlx.core)": [[268, "mlx.core.round", false]], "rsqrt (c++ function)": [[0, "_CPPv45rsqrtRK5array14StreamOrDevice", false]], "rsqrt() (array method)": [[68, "mlx.core.array.rsqrt", false]], "rsqrt() (in module mlx.core)": [[269, "mlx.core.rsqrt", false]], "save() (in module mlx.core)": [[270, "mlx.core.save", false]], "save_gguf() (in module mlx.core)": [[271, "mlx.core.save_gguf", false]], "save_safetensors() (in module mlx.core)": [[272, "mlx.core.save_safetensors", false]], "save_weights() (module method)": [[388, "mlx.nn.Module.save_weights", false]], "savez() (in module mlx.core)": [[273, "mlx.core.savez", false]], "savez_compressed() (in module mlx.core)": [[274, "mlx.core.savez_compressed", false]], "scaled_dot_product_attention() (in module mlx.core.fast)": [[151, "mlx.core.fast.scaled_dot_product_attention", false]], "scatter (c++ function)": [[0, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_add (c++ function)": [[0, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_add_axis (c++ function)": [[0, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "scatter_max (c++ function)": [[0, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_min (c++ function)": [[0, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_prod (c++ function)": [[0, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "seed() (in module mlx.core.random)": [[256, "mlx.core.random.seed", false]], "selu (class in mlx.nn)": [[406, "mlx.nn.SELU", false], [458, "mlx.nn.selu", false]], "send() (in module mlx.core.distributed)": [[130, "mlx.core.distributed.send", false]], "sequential (class in mlx.nn)": [[407, "mlx.nn.Sequential", false]], "set_cache_limit() (in module mlx.core)": [[275, "mlx.core.set_cache_limit", false]], "set_default_device() (in module mlx.core)": [[276, "mlx.core.set_default_device", false]], "set_default_stream() (in module mlx.core)": [[277, "mlx.core.set_default_stream", false]], "set_dtype() (module method)": [[389, "mlx.nn.Module.set_dtype", false]], "set_memory_limit() (in module mlx.core)": [[278, "mlx.core.set_memory_limit", false]], "set_wired_limit() (in module mlx.core)": [[279, "mlx.core.set_wired_limit", false]], "sgd (class in mlx.optimizers)": [[486, "mlx.optimizers.SGD", false]], "shape (array property)": [[69, "mlx.core.array.shape", false]], "sigmoid (c++ function)": [[0, "_CPPv47sigmoidRK5array14StreamOrDevice", false]], "sigmoid (class in mlx.nn)": [[409, "mlx.nn.Sigmoid", false], [459, "mlx.nn.sigmoid", false]], "sigmoid() (in module mlx.core)": [[280, "mlx.core.sigmoid", false]], "sign (c++ function)": [[0, "_CPPv44signRK5array14StreamOrDevice", false]], "sign() (in module mlx.core)": [[281, "mlx.core.sign", false]], "silu (class in mlx.nn)": [[408, "mlx.nn.SiLU", false], [460, "mlx.nn.silu", false]], "sin (c++ function)": [[0, "_CPPv43sinRK5array14StreamOrDevice", false]], "sin() (array method)": [[70, "mlx.core.array.sin", false]], "sin() (in module mlx.core)": [[282, "mlx.core.sin", false]], "sinh (c++ function)": [[0, "_CPPv44sinhRK5array14StreamOrDevice", false]], "sinh() (in module mlx.core)": [[283, "mlx.core.sinh", false]], "sinusoidalpositionalencoding (class in mlx.nn)": [[410, "mlx.nn.SinusoidalPositionalEncoding", false]], "size (array property)": [[71, "mlx.core.array.size", false]], "slice (c++ function)": [[0, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", false]], "slice() (in module mlx.core)": [[284, "mlx.core.slice", false]], "slice_update (c++ function)": [[0, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", false], [0, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", false], [0, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "slice_update() (in module mlx.core)": [[285, "mlx.core.slice_update", false]], "smooth_l1_loss (class in mlx.nn.losses)": [[452, "mlx.nn.losses.smooth_l1_loss", false]], "softmax (c++ function)": [[0, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayib14StreamOrDevice", false]], "softmax (class in mlx.nn)": [[411, "mlx.nn.Softmax", false], [461, "mlx.nn.softmax", false]], "softmax() (in module mlx.core)": [[286, "mlx.core.softmax", false]], "softmin (class in mlx.nn)": [[412, "mlx.nn.Softmin", false], [462, "mlx.nn.softmin", false]], "softplus (class in mlx.nn)": [[413, "mlx.nn.Softplus", false], [463, "mlx.nn.softplus", false]], "softshrink (class in mlx.nn)": [[414, "mlx.nn.Softshrink", false], [464, "mlx.nn.softshrink", false]], "softsign (class in mlx.nn)": [[415, "mlx.nn.Softsign", false]], "solve() (in module mlx.core.linalg)": [[204, "mlx.core.linalg.solve", false]], "solve_triangular() (in module mlx.core.linalg)": [[205, "mlx.core.linalg.solve_triangular", false]], "sort (c++ function)": [[0, "_CPPv44sortRK5array14StreamOrDevice", false], [0, "_CPPv44sortRK5arrayi14StreamOrDevice", false]], "sort() (in module mlx.core)": [[287, "mlx.core.sort", false]], "split (c++ function)": [[0, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayi14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayii14StreamOrDevice", false]], "split() (array method)": [[72, "mlx.core.array.split", false]], "split() (in module mlx.core)": [[288, "mlx.core.split", false]], "split() (in module mlx.core.random)": [[257, "mlx.core.random.split", false]], "sqrt (c++ function)": [[0, "_CPPv44sqrtRK5array14StreamOrDevice", false]], "sqrt() (array method)": [[73, "mlx.core.array.sqrt", false]], "sqrt() (in module mlx.core)": [[289, "mlx.core.sqrt", false]], "square (c++ function)": [[0, "_CPPv46squareRK5array14StreamOrDevice", false]], "square() (array method)": [[74, "mlx.core.array.square", false]], "square() (in module mlx.core)": [[290, "mlx.core.square", false]], "squeeze (c++ function)": [[0, "_CPPv47squeezeRK5array14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayi14StreamOrDevice", false]], "squeeze() (array method)": [[75, "mlx.core.array.squeeze", false]], "squeeze() (in module mlx.core)": [[291, "mlx.core.squeeze", false]], "stack (c++ function)": [[0, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", false]], "stack() (in module mlx.core)": [[292, "mlx.core.stack", false]], "start_capture() (in module mlx.core.metal)": [[226, "mlx.core.metal.start_capture", false]], "state (module property)": [[390, "mlx.nn.Module.state", false]], "state (optimizer property)": [[483, "mlx.optimizers.Optimizer.state", false]], "std (c++ function)": [[0, "_CPPv4StRK5array14StreamOrDevice", false], [0, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv4StRK5arraybi14StreamOrDevice", false], [0, "_CPPv4StRK5arrayibi14StreamOrDevice", false]], "std() (array method)": [[76, "mlx.core.array.std", false]], "std() (in module mlx.core)": [[293, "mlx.core.std", false]], "step (class in mlx.nn)": [[416, "mlx.nn.Step", false], [465, "mlx.nn.step", false]], "step_decay() (in module mlx.optimizers)": [[491, "mlx.optimizers.step_decay", false]], "stop_capture() (in module mlx.core.metal)": [[227, "mlx.core.metal.stop_capture", false]], "stop_gradient (c++ function)": [[0, "_CPPv413stop_gradientRK5array14StreamOrDevice", false]], "stop_gradient() (in module mlx.core)": [[294, "mlx.core.stop_gradient", false]], "stream (class in mlx.core)": [[330, "mlx.core.Stream", false]], "stream() (in module mlx.core)": [[295, "mlx.core.stream", false]], "subtract (c++ function)": [[0, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", false]], "subtract() (in module mlx.core)": [[296, "mlx.core.subtract", false]], "sum (c++ function)": [[0, "_CPPv43sumRK5array14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayib14StreamOrDevice", false]], "sum() (array method)": [[77, "mlx.core.array.sum", false]], "sum() (in module mlx.core)": [[297, "mlx.core.sum", false]], "svd() (in module mlx.core.linalg)": [[206, "mlx.core.linalg.svd", false]], "swapaxes (c++ function)": [[0, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", false]], "swapaxes() (array method)": [[78, "mlx.core.array.swapaxes", false]], "swapaxes() (in module mlx.core)": [[298, "mlx.core.swapaxes", false]], "synchronize() (in module mlx.core)": [[299, "mlx.core.synchronize", false]], "t (array property)": [[32, "mlx.core.array.T", false]], "take (c++ function)": [[0, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayi14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayii14StreamOrDevice", false]], "take() (in module mlx.core)": [[300, "mlx.core.take", false]], "take_along_axis (c++ function)": [[0, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", false]], "take_along_axis() (in module mlx.core)": [[301, "mlx.core.take_along_axis", false]], "tan (c++ function)": [[0, "_CPPv43tanRK5array14StreamOrDevice", false]], "tan() (in module mlx.core)": [[302, "mlx.core.tan", false]], "tanh (c++ function)": [[0, "_CPPv44tanhRK5array14StreamOrDevice", false]], "tanh (class in mlx.nn)": [[417, "mlx.nn.Tanh", false], [466, "mlx.nn.tanh", false]], "tanh() (in module mlx.core)": [[303, "mlx.core.tanh", false]], "tensordot (c++ function)": [[0, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", false], [0, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", false]], "tensordot() (in module mlx.core)": [[304, "mlx.core.tensordot", false]], "tile (c++ function)": [[0, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "tile() (in module mlx.core)": [[305, "mlx.core.tile", false]], "tolist() (array method)": [[79, "mlx.core.array.tolist", false]], "topk (c++ function)": [[0, "_CPPv44topkRK5arrayi14StreamOrDevice", false], [0, "_CPPv44topkRK5arrayii14StreamOrDevice", false]], "topk() (in module mlx.core)": [[306, "mlx.core.topk", false]], "trace (c++ function)": [[0, "_CPPv45traceRK5array14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", false]], "trace() (in module mlx.core)": [[307, "mlx.core.trace", false]], "train() (module method)": [[391, "mlx.nn.Module.train", false]], "trainable_parameters() (module method)": [[392, "mlx.nn.Module.trainable_parameters", false]], "training (module property)": [[393, "mlx.nn.Module.training", false]], "transformer (class in mlx.nn)": [[418, "mlx.nn.Transformer", false]], "transpose (c++ function)": [[0, "_CPPv49transposeRK5array14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "transpose() (array method)": [[80, "mlx.core.array.transpose", false]], "transpose() (in module mlx.core)": [[308, "mlx.core.transpose", false]], "tree_flatten() (in module mlx.utils)": [[325, "mlx.utils.tree_flatten", false]], "tree_map() (in module mlx.utils)": [[326, "mlx.utils.tree_map", false]], "tree_map_with_path() (in module mlx.utils)": [[327, "mlx.utils.tree_map_with_path", false]], "tree_reduce() (in module mlx.utils)": [[328, "mlx.utils.tree_reduce", false]], "tree_unflatten() (in module mlx.utils)": [[329, "mlx.utils.tree_unflatten", false]], "tri (c++ function)": [[0, "_CPPv43trii5Dtype14StreamOrDevice", false], [0, "_CPPv43triiii5Dtype14StreamOrDevice", false]], "tri() (in module mlx.core)": [[309, "mlx.core.tri", false]], "tri_inv() (in module mlx.core.linalg)": [[207, "mlx.core.linalg.tri_inv", false]], "tril (c++ function)": [[0, "_CPPv44tril5arrayi14StreamOrDevice", false]], "tril() (in module mlx.core)": [[310, "mlx.core.tril", false]], "triplet_loss (class in mlx.nn.losses)": [[453, "mlx.nn.losses.triplet_loss", false]], "triu (c++ function)": [[0, "_CPPv44triu5arrayi14StreamOrDevice", false]], "triu() (in module mlx.core)": [[311, "mlx.core.triu", false]], "truncated_normal() (in module mlx.core.random)": [[258, "mlx.core.random.truncated_normal", false]], "unflatten (c++ function)": [[0, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", false]], "unflatten() (in module mlx.core)": [[312, "mlx.core.unflatten", false]], "unfreeze() (module method)": [[394, "mlx.nn.Module.unfreeze", false]], "uniform() (in module mlx.core.random)": [[259, "mlx.core.random.uniform", false]], "uniform() (in module mlx.nn.init)": [[427, "mlx.nn.init.uniform", false]], "update() (module method)": [[395, "mlx.nn.Module.update", false]], "update() (optimizer method)": [[484, "mlx.optimizers.Optimizer.update", false]], "update_modules() (module method)": [[396, "mlx.nn.Module.update_modules", false]], "upsample (class in mlx.nn)": [[419, "mlx.nn.Upsample", false]], "value_and_grad() (in module mlx.core)": [[313, "mlx.core.value_and_grad", false]], "value_and_grad() (in module mlx.nn)": [[323, "mlx.nn.value_and_grad", false]], "var (c++ function)": [[0, "_CPPv43varRK5array14StreamOrDevice", false], [0, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv43varRK5arraybi14StreamOrDevice", false], [0, "_CPPv43varRK5arrayibi14StreamOrDevice", false]], "var() (array method)": [[81, "mlx.core.array.var", false]], "var() (in module mlx.core)": [[314, "mlx.core.var", false]], "view (c++ function)": [[0, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", false]], "view() (array method)": [[82, "mlx.core.array.view", false]], "view() (in module mlx.core)": [[315, "mlx.core.view", false]], "vjp() (in module mlx.core)": [[316, "mlx.core.vjp", false]], "vmap() (in module mlx.core)": [[317, "mlx.core.vmap", false]], "where (c++ function)": [[0, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", false]], "where() (in module mlx.core)": [[318, "mlx.core.where", false]], "zeros (c++ function)": [[0, "_CPPv45zerosRK5Shape14StreamOrDevice", false], [0, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", false]], "zeros() (in module mlx.core)": [[319, "mlx.core.zeros", false]], "zeros_like (c++ function)": [[0, "_CPPv410zeros_likeRK5array14StreamOrDevice", false]], "zeros_like() (in module mlx.core)": [[320, "mlx.core.zeros_like", false]]}, "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_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::a"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::offset"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::s"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::shape"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_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, "_CPPv414bitwise_invertRK5array14StreamOrDevice", "bitwise_invert"], [0, 1, 1, "_CPPv414bitwise_invertRK5array14StreamOrDevice", "bitwise_invert::a"], [0, 1, 1, "_CPPv414bitwise_invertRK5array14StreamOrDevice", "bitwise_invert::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_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::a"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::s"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "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, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate"], [0, 0, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::axis"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::s"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::s"], [0, 0, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::a"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::s"], [0, 0, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::a"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::allow_col_major"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::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, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::dilation"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::groups"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::input"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::padding"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::s"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::stride"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::weight"], [0, 0, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::dilation"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::groups"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::input"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::padding"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::s"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::stride"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::weight"], [0, 0, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::dilation"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::groups"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::input"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::padding"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::s"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::stride"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::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, "_CPPv44full5Shape5array14StreamOrDevice", "full"], [0, 0, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full"], [0, 2, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::T"], [0, 2, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::T"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::vals"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::vals"], [0, 0, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather"], [0, 0, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::axes"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::axis"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::slice_sizes"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "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, "_CPPv44imagRK5array14StreamOrDevice", "imag"], [0, 1, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag::a"], [0, 1, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag::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, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite"], [0, 1, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite::a"], [0, 1, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite::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, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::a"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::b"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::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, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::arrays"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::indexing"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::s"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "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_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::a"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::nan"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::neginf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::posinf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "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, "_CPPv44onesRK5Shape14StreamOrDevice", "ones"], [0, 0, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::dtype"], [0, 1, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones::shape"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "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, "_CPPv4coRK5array", "operator~"], [0, 1, 1, "_CPPv4coRK5array", "operator~::a"], [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, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::axes"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::high_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::low_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "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, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::a"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::axis"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::indices"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::s"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::values"], [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_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::biases"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::bits"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::group_size"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::s"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::scales"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::transpose"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::w"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::x"], [0, 0, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::a"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::s"], [0, 0, 1, "_CPPv44realRK5array14StreamOrDevice", "real"], [0, 1, 1, "_CPPv44realRK5array14StreamOrDevice", "real::a"], [0, 1, 1, "_CPPv44realRK5array14StreamOrDevice", "real::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, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::a"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::s"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "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, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::axes"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::axes"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::axis"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::axis"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::shift"], [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, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::a"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::axis"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::indices"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::s"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::values"], [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, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::axes"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::slice_size"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::strides"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::strides"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::axes"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::strides"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "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, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "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, 0, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::index"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::index"], [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, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "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, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::a"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::axis"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::s"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::shape"], [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, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros"], [0, 0, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::dtype"], [0, 1, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros::shape"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "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": [[10, 3, 1, "", "Device"], [11, 3, 1, "", "Dtype"], [12, 3, 1, "", "DtypeCategory"], [330, 3, 1, "", "Stream"], [13, 5, 1, "", "abs"], [14, 5, 1, "", "add"], [15, 5, 1, "", "addmm"], [16, 5, 1, "", "all"], [17, 5, 1, "", "allclose"], [18, 5, 1, "", "any"], [19, 5, 1, "", "arange"], [20, 5, 1, "", "arccos"], [21, 5, 1, "", "arccosh"], [22, 5, 1, "", "arcsin"], [23, 5, 1, "", "arcsinh"], [24, 5, 1, "", "arctan"], [25, 5, 1, "", "arctan2"], [26, 5, 1, "", "arctanh"], [27, 5, 1, "", "argmax"], [28, 5, 1, "", "argmin"], [29, 5, 1, "", "argpartition"], [30, 5, 1, "", "argsort"], [31, 3, 1, "", "array"], [83, 5, 1, "", "array_equal"], [84, 5, 1, "", "as_strided"], [85, 5, 1, "", "atleast_1d"], [86, 5, 1, "", "atleast_2d"], [87, 5, 1, "", "atleast_3d"], [88, 5, 1, "", "bitwise_and"], [89, 5, 1, "", "bitwise_invert"], [90, 5, 1, "", "bitwise_or"], [91, 5, 1, "", "bitwise_xor"], [92, 5, 1, "", "block_masked_mm"], [93, 5, 1, "", "broadcast_to"], [94, 5, 1, "", "ceil"], [95, 5, 1, "", "clear_cache"], [96, 5, 1, "", "clip"], [97, 5, 1, "", "compile"], [98, 5, 1, "", "concatenate"], [99, 5, 1, "", "conj"], [100, 5, 1, "", "conjugate"], [101, 5, 1, "", "conv1d"], [102, 5, 1, "", "conv2d"], [103, 5, 1, "", "conv3d"], [104, 5, 1, "", "conv_general"], [105, 5, 1, "", "conv_transpose1d"], [106, 5, 1, "", "conv_transpose2d"], [107, 5, 1, "", "conv_transpose3d"], [108, 5, 1, "", "convolve"], [109, 5, 1, "", "cos"], [110, 5, 1, "", "cosh"], [111, 5, 1, "", "cummax"], [112, 5, 1, "", "cummin"], [113, 5, 1, "", "cumprod"], [114, 5, 1, "", "cumsum"], [115, 3, 1, "", "custom_function"], [116, 5, 1, "", "default_device"], [117, 5, 1, "", "default_stream"], [118, 5, 1, "", "degrees"], [119, 5, 1, "", "dequantize"], [120, 5, 1, "", "diag"], [121, 5, 1, "", "diagonal"], [122, 5, 1, "", "disable_compile"], [131, 5, 1, "", "divide"], [132, 5, 1, "", "divmod"], [133, 5, 1, "", "einsum"], [134, 5, 1, "", "einsum_path"], [135, 5, 1, "", "enable_compile"], [136, 5, 1, "", "equal"], [137, 5, 1, "", "erf"], [138, 5, 1, "", "erfinv"], [139, 5, 1, "", "eval"], [140, 5, 1, "", "exp"], [141, 5, 1, "", "expand_dims"], [142, 5, 1, "", "expm1"], [143, 5, 1, "", "export_function"], [144, 5, 1, "", "export_to_dot"], [145, 5, 1, "", "exporter"], [146, 5, 1, "", "eye"], [164, 3, 1, "", "finfo"], [165, 5, 1, "", "flatten"], [166, 5, 1, "", "floor"], [167, 5, 1, "", "floor_divide"], [168, 5, 1, "", "full"], [169, 5, 1, "", "gather_mm"], [170, 5, 1, "", "gather_qmm"], [171, 5, 1, "", "get_active_memory"], [172, 5, 1, "", "get_cache_memory"], [173, 5, 1, "", "get_peak_memory"], [174, 5, 1, "", "grad"], [175, 5, 1, "", "greater"], [176, 5, 1, "", "greater_equal"], [177, 5, 1, "", "hadamard_transform"], [178, 5, 1, "", "identity"], [179, 5, 1, "", "imag"], [180, 5, 1, "", "import_function"], [181, 5, 1, "", "inner"], [182, 5, 1, "", "isclose"], [183, 5, 1, "", "isfinite"], [184, 5, 1, "", "isinf"], [185, 5, 1, "", "isnan"], [186, 5, 1, "", "isneginf"], [187, 5, 1, "", "isposinf"], [188, 5, 1, "", "issubdtype"], [189, 5, 1, "", "jvp"], [190, 5, 1, "", "kron"], [191, 5, 1, "", "left_shift"], [192, 5, 1, "", "less"], [193, 5, 1, "", "less_equal"], [208, 5, 1, "", "linspace"], [209, 5, 1, "", "load"], [210, 5, 1, "", "log"], [211, 5, 1, "", "log10"], [212, 5, 1, "", "log1p"], [213, 5, 1, "", "log2"], [214, 5, 1, "", "logaddexp"], [215, 5, 1, "", "logical_and"], [216, 5, 1, "", "logical_not"], [217, 5, 1, "", "logical_or"], [218, 5, 1, "", "logsumexp"], [219, 5, 1, "", "matmul"], [220, 5, 1, "", "max"], [221, 5, 1, "", "maximum"], [222, 5, 1, "", "mean"], [223, 5, 1, "", "meshgrid"], [228, 5, 1, "", "min"], [229, 5, 1, "", "minimum"], [230, 5, 1, "", "moveaxis"], [231, 5, 1, "", "multiply"], [232, 5, 1, "", "nan_to_num"], [233, 5, 1, "", "negative"], [234, 5, 1, "", "new_stream"], [235, 5, 1, "", "not_equal"], [236, 5, 1, "", "ones"], [237, 5, 1, "", "ones_like"], [238, 5, 1, "", "outer"], [239, 5, 1, "", "pad"], [240, 5, 1, "", "partition"], [241, 5, 1, "", "power"], [242, 5, 1, "", "prod"], [243, 5, 1, "", "put_along_axis"], [244, 5, 1, "", "quantize"], [245, 5, 1, "", "quantized_matmul"], [246, 5, 1, "", "radians"], [260, 5, 1, "", "real"], [261, 5, 1, "", "reciprocal"], [262, 5, 1, "", "remainder"], [263, 5, 1, "", "repeat"], [264, 5, 1, "", "reset_peak_memory"], [265, 5, 1, "", "reshape"], [266, 5, 1, "", "right_shift"], [267, 5, 1, "", "roll"], [268, 5, 1, "", "round"], [269, 5, 1, "", "rsqrt"], [270, 5, 1, "", "save"], [271, 5, 1, "", "save_gguf"], [272, 5, 1, "", "save_safetensors"], [273, 5, 1, "", "savez"], [274, 5, 1, "", "savez_compressed"], [275, 5, 1, "", "set_cache_limit"], [276, 5, 1, "", "set_default_device"], [277, 5, 1, "", "set_default_stream"], [278, 5, 1, "", "set_memory_limit"], [279, 5, 1, "", "set_wired_limit"], [280, 5, 1, "", "sigmoid"], [281, 5, 1, "", "sign"], [282, 5, 1, "", "sin"], [283, 5, 1, "", "sinh"], [284, 5, 1, "", "slice"], [285, 5, 1, "", "slice_update"], [286, 5, 1, "", "softmax"], [287, 5, 1, "", "sort"], [288, 5, 1, "", "split"], [289, 5, 1, "", "sqrt"], [290, 5, 1, "", "square"], [291, 5, 1, "", "squeeze"], [292, 5, 1, "", "stack"], [293, 5, 1, "", "std"], [294, 5, 1, "", "stop_gradient"], [295, 5, 1, "", "stream"], [296, 5, 1, "", "subtract"], [297, 5, 1, "", "sum"], [298, 5, 1, "", "swapaxes"], [299, 5, 1, "", "synchronize"], [300, 5, 1, "", "take"], [301, 5, 1, "", "take_along_axis"], [302, 5, 1, "", "tan"], [303, 5, 1, "", "tanh"], [304, 5, 1, "", "tensordot"], [305, 5, 1, "", "tile"], [306, 5, 1, "", "topk"], [307, 5, 1, "", "trace"], [308, 5, 1, "", "transpose"], [309, 5, 1, "", "tri"], [310, 5, 1, "", "tril"], [311, 5, 1, "", "triu"], [312, 5, 1, "", "unflatten"], [313, 5, 1, "", "value_and_grad"], [314, 5, 1, "", "var"], [315, 5, 1, "", "view"], [316, 5, 1, "", "vjp"], [317, 5, 1, "", "vmap"], [318, 5, 1, "", "where"], [319, 5, 1, "", "zeros"], [320, 5, 1, "", "zeros_like"]], "mlx.core.Device": [[10, 4, 1, "", "__init__"]], "mlx.core.Dtype": [[11, 4, 1, "", "__init__"]], "mlx.core.DtypeCategory": [[12, 4, 1, "", "__init__"]], "mlx.core.Stream": [[330, 4, 1, "", "__init__"]], "mlx.core.array": [[32, 6, 1, "", "T"], [31, 4, 1, "", "__init__"], [33, 4, 1, "", "abs"], [34, 4, 1, "", "all"], [35, 4, 1, "", "any"], [36, 4, 1, "", "argmax"], [37, 4, 1, "", "argmin"], [38, 4, 1, "", "astype"], [39, 6, 1, "", "at"], [40, 4, 1, "", "conj"], [41, 4, 1, "", "cos"], [42, 4, 1, "", "cummax"], [43, 4, 1, "", "cummin"], [44, 4, 1, "", "cumprod"], [45, 4, 1, "", "cumsum"], [46, 4, 1, "", "diag"], [47, 4, 1, "", "diagonal"], [48, 6, 1, "", "dtype"], [49, 4, 1, "", "exp"], [50, 4, 1, "", "flatten"], [51, 4, 1, "", "item"], [52, 6, 1, "", "itemsize"], [53, 4, 1, "", "log"], [54, 4, 1, "", "log10"], [55, 4, 1, "", "log1p"], [56, 4, 1, "", "log2"], [57, 4, 1, "", "logsumexp"], [58, 4, 1, "", "max"], [59, 4, 1, "", "mean"], [60, 4, 1, "", "min"], [61, 4, 1, "", "moveaxis"], [62, 6, 1, "", "nbytes"], [63, 6, 1, "", "ndim"], [64, 4, 1, "", "prod"], [65, 4, 1, "", "reciprocal"], [66, 4, 1, "", "reshape"], [67, 4, 1, "", "round"], [68, 4, 1, "", "rsqrt"], [69, 6, 1, "", "shape"], [70, 4, 1, "", "sin"], [71, 6, 1, "", "size"], [72, 4, 1, "", "split"], [73, 4, 1, "", "sqrt"], [74, 4, 1, "", "square"], [75, 4, 1, "", "squeeze"], [76, 4, 1, "", "std"], [77, 4, 1, "", "sum"], [78, 4, 1, "", "swapaxes"], [79, 4, 1, "", "tolist"], [80, 4, 1, "", "transpose"], [81, 4, 1, "", "var"], [82, 4, 1, "", "view"]], "mlx.core.custom_function": [[115, 4, 1, "", "__init__"]], "mlx.core.distributed": [[123, 3, 1, "", "Group"], [124, 5, 1, "", "all_gather"], [125, 5, 1, "", "all_sum"], [126, 5, 1, "", "init"], [127, 5, 1, "", "is_available"], [128, 5, 1, "", "recv"], [129, 5, 1, "", "recv_like"], [130, 5, 1, "", "send"]], "mlx.core.distributed.Group": [[123, 4, 1, "", "__init__"]], "mlx.core.fast": [[147, 5, 1, "", "layer_norm"], [148, 5, 1, "", "metal_kernel"], [149, 5, 1, "", "rms_norm"], [150, 5, 1, "", "rope"], [151, 5, 1, "", "scaled_dot_product_attention"]], "mlx.core.fft": [[152, 5, 1, "", "fft"], [153, 5, 1, "", "fft2"], [154, 5, 1, "", "fftn"], [155, 5, 1, "", "ifft"], [156, 5, 1, "", "ifft2"], [157, 5, 1, "", "ifftn"], [158, 5, 1, "", "irfft"], [159, 5, 1, "", "irfft2"], [160, 5, 1, "", "irfftn"], [161, 5, 1, "", "rfft"], [162, 5, 1, "", "rfft2"], [163, 5, 1, "", "rfftn"]], "mlx.core.finfo": [[164, 4, 1, "", "__init__"]], "mlx.core.linalg": [[194, 5, 1, "", "cholesky"], [195, 5, 1, "", "cholesky_inv"], [196, 5, 1, "", "cross"], [197, 5, 1, "", "eigh"], [198, 5, 1, "", "eigvalsh"], [199, 5, 1, "", "inv"], [200, 5, 1, "", "lu"], [201, 5, 1, "", "lu_factor"], [202, 5, 1, "", "norm"], [203, 5, 1, "", "qr"], [204, 5, 1, "", "solve"], [205, 5, 1, "", "solve_triangular"], [206, 5, 1, "", "svd"], [207, 5, 1, "", "tri_inv"]], "mlx.core.metal": [[224, 5, 1, "", "device_info"], [225, 5, 1, "", "is_available"], [226, 5, 1, "", "start_capture"], [227, 5, 1, "", "stop_capture"]], "mlx.core.random": [[247, 5, 1, "", "bernoulli"], [248, 5, 1, "", "categorical"], [249, 5, 1, "", "gumbel"], [250, 5, 1, "", "key"], [251, 5, 1, "", "laplace"], [252, 5, 1, "", "multivariate_normal"], [253, 5, 1, "", "normal"], [254, 5, 1, "", "permutation"], [255, 5, 1, "", "randint"], [256, 5, 1, "", "seed"], [257, 5, 1, "", "split"], [258, 5, 1, "", "truncated_normal"], [259, 5, 1, "", "uniform"]], "mlx.nn": [[342, 3, 1, "", "ALiBi"], [343, 3, 1, "", "AvgPool1d"], [344, 3, 1, "", "AvgPool2d"], [345, 3, 1, "", "AvgPool3d"], [346, 3, 1, "", "BatchNorm"], [347, 3, 1, "", "CELU"], [348, 3, 1, "", "Conv1d"], [349, 3, 1, "", "Conv2d"], [350, 3, 1, "", "Conv3d"], [351, 3, 1, "", "ConvTranspose1d"], [352, 3, 1, "", "ConvTranspose2d"], [353, 3, 1, "", "ConvTranspose3d"], [354, 3, 1, "", "Dropout"], [355, 3, 1, "", "Dropout2d"], [356, 3, 1, "", "Dropout3d"], [357, 3, 1, "", "ELU"], [358, 3, 1, "", "Embedding"], [359, 3, 1, "", "GELU"], [360, 3, 1, "", "GLU"], [361, 3, 1, "", "GRU"], [362, 3, 1, "", "GroupNorm"], [363, 3, 1, "", "HardShrink"], [364, 3, 1, "", "HardTanh"], [365, 3, 1, "", "Hardswish"], [366, 3, 1, "", "InstanceNorm"], [367, 3, 1, "", "LSTM"], [368, 3, 1, "", "LayerNorm"], [369, 3, 1, "", "LeakyReLU"], [370, 3, 1, "", "Linear"], [371, 3, 1, "", "LogSigmoid"], [372, 3, 1, "", "LogSoftmax"], [373, 3, 1, "", "MaxPool1d"], [374, 3, 1, "", "MaxPool2d"], [375, 3, 1, "", "MaxPool3d"], [376, 3, 1, "", "Mish"], [471, 3, 1, "", "Module"], [397, 3, 1, "", "MultiHeadAttention"], [398, 3, 1, "", "PReLU"], [399, 3, 1, "", "QuantizedEmbedding"], [400, 3, 1, "", "QuantizedLinear"], [401, 3, 1, "", "RMSNorm"], [402, 3, 1, "", "RNN"], [403, 3, 1, "", "ReLU"], [404, 3, 1, "", "ReLU6"], [405, 3, 1, "", "RoPE"], [406, 3, 1, "", "SELU"], [407, 3, 1, "", "Sequential"], [408, 3, 1, "", "SiLU"], [409, 3, 1, "", "Sigmoid"], [410, 3, 1, "", "SinusoidalPositionalEncoding"], [411, 3, 1, "", "Softmax"], [412, 3, 1, "", "Softmin"], [413, 3, 1, "", "Softplus"], [414, 3, 1, "", "Softshrink"], [415, 3, 1, "", "Softsign"], [416, 3, 1, "", "Step"], [417, 3, 1, "", "Tanh"], [418, 3, 1, "", "Transformer"], [419, 3, 1, "", "Upsample"], [321, 5, 1, "", "average_gradients"], [428, 3, 1, "", "celu"], [429, 3, 1, "", "elu"], [430, 3, 1, "", "gelu"], [431, 3, 1, "", "gelu_approx"], [432, 3, 1, "", "gelu_fast_approx"], [433, 3, 1, "", "glu"], [434, 3, 1, "", "hard_shrink"], [435, 3, 1, "", "hard_tanh"], [436, 3, 1, "", "hardswish"], [437, 3, 1, "", "leaky_relu"], [438, 3, 1, "", "log_sigmoid"], [439, 3, 1, "", "log_softmax"], [454, 3, 1, "", "mish"], [455, 3, 1, "", "prelu"], [322, 5, 1, "", "quantize"], [456, 3, 1, "", "relu"], [457, 3, 1, "", "relu6"], [458, 3, 1, "", "selu"], [459, 3, 1, "", "sigmoid"], [460, 3, 1, "", "silu"], [461, 3, 1, "", "softmax"], [462, 3, 1, "", "softmin"], [463, 3, 1, "", "softplus"], [464, 3, 1, "", "softshrink"], [465, 3, 1, "", "step"], [466, 3, 1, "", "tanh"], [323, 5, 1, "", "value_and_grad"]], "mlx.nn.Module": [[377, 4, 1, "", "apply"], [378, 4, 1, "", "apply_to_modules"], [379, 4, 1, "", "children"], [380, 4, 1, "", "eval"], [381, 4, 1, "", "filter_and_map"], [382, 4, 1, "", "freeze"], [383, 4, 1, "", "leaf_modules"], [384, 4, 1, "", "load_weights"], [385, 4, 1, "", "modules"], [386, 4, 1, "", "named_modules"], [387, 4, 1, "", "parameters"], [388, 4, 1, "", "save_weights"], [389, 4, 1, "", "set_dtype"], [390, 6, 1, "", "state"], [391, 4, 1, "", "train"], [392, 4, 1, "", "trainable_parameters"], [393, 6, 1, "", "training"], [394, 4, 1, "", "unfreeze"], [395, 4, 1, "", "update"], [396, 4, 1, "", "update_modules"]], "mlx.nn.init": [[420, 5, 1, "", "constant"], [421, 5, 1, "", "glorot_normal"], [422, 5, 1, "", "glorot_uniform"], [423, 5, 1, "", "he_normal"], [424, 5, 1, "", "he_uniform"], [425, 5, 1, "", "identity"], [426, 5, 1, "", "normal"], [427, 5, 1, "", "uniform"]], "mlx.nn.losses": [[440, 3, 1, "", "binary_cross_entropy"], [441, 3, 1, "", "cosine_similarity_loss"], [442, 3, 1, "", "cross_entropy"], [443, 3, 1, "", "gaussian_nll_loss"], [444, 3, 1, "", "hinge_loss"], [445, 3, 1, "", "huber_loss"], [446, 3, 1, "", "kl_div_loss"], [447, 3, 1, "", "l1_loss"], [448, 3, 1, "", "log_cosh_loss"], [449, 3, 1, "", "margin_ranking_loss"], [450, 3, 1, "", "mse_loss"], [451, 3, 1, "", "nll_loss"], [452, 3, 1, "", "smooth_l1_loss"], [453, 3, 1, "", "triplet_loss"]], "mlx.optimizers": [[474, 3, 1, "", "AdaDelta"], [475, 3, 1, "", "Adafactor"], [476, 3, 1, "", "Adagrad"], [477, 3, 1, "", "Adam"], [478, 3, 1, "", "AdamW"], [479, 3, 1, "", "Adamax"], [480, 3, 1, "", "Lion"], [493, 3, 1, "", "Optimizer"], [485, 3, 1, "", "RMSprop"], [486, 3, 1, "", "SGD"], [324, 5, 1, "", "clip_grad_norm"], [487, 5, 1, "", "cosine_decay"], [488, 5, 1, "", "exponential_decay"], [489, 5, 1, "", "join_schedules"], [490, 5, 1, "", "linear_schedule"], [491, 5, 1, "", "step_decay"]], "mlx.optimizers.Optimizer": [[481, 4, 1, "", "apply_gradients"], [482, 4, 1, "", "init"], [483, 6, 1, "", "state"], [484, 4, 1, "", "update"]], "mlx.utils": [[325, 5, 1, "", "tree_flatten"], [326, 5, 1, "", "tree_map"], [327, 5, 1, "", "tree_map_with_path"], [328, 5, 1, "", "tree_reduce"], [329, 5, 1, "", "tree_unflatten"]]}, "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"]}, "objtypes": {"0": "cpp:function", "1": "cpp:functionParam", "2": "cpp:templateParam", "3": "py:class", "4": "py:method", "5": "py:function", "6": "py:property"}, "terms": {"": [0, 1, 2, 5, 6, 7, 48, 52, 63, 97, 117, 119, 153, 154, 156, 157, 159, 160, 162, 163, 174, 195, 202, 206, 209, 222, 238, 244, 248, 268, 271, 272, 293, 295, 313, 314, 315, 317, 323, 341, 344, 345, 361, 367, 374, 375, 381, 382, 384, 388, 389, 390, 394, 402, 473, 482, 483, 495, 498, 500, 501, 505, 506, 507, 508], "0": [0, 1, 2, 4, 5, 6, 7, 9, 10, 15, 19, 39, 46, 47, 50, 67, 72, 76, 81, 84, 95, 98, 101, 102, 103, 104, 105, 106, 107, 115, 120, 121, 146, 148, 151, 165, 169, 174, 180, 190, 197, 199, 200, 202, 203, 207, 232, 239, 247, 251, 253, 254, 259, 263, 268, 275, 279, 284, 285, 288, 292, 293, 307, 309, 310, 311, 312, 313, 314, 317, 321, 324, 325, 327, 328, 341, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 359, 362, 363, 366, 368, 369, 373, 374, 375, 398, 403, 405, 410, 414, 416, 418, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 431, 432, 434, 435, 436, 437, 440, 442, 444, 445, 449, 452, 453, 455, 456, 457, 458, 464, 465, 468, 471, 474, 475, 477, 478, 479, 480, 482, 485, 486, 487, 488, 489, 490, 491, 495, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507], "00005": 5, "0001": 410, "0005": 431, "001": 475, "00364": 5, "01": [5, 369, 437, 478], "0137595": 423, "015": 432, "0184009": 424, "02264": 422, "024": 501, "02765": 423, "0300242": 424, "044715": [359, 431], "0485873": 442, "05": [17, 182, 346, 362, 366, 368, 401], "0507": 458, "05202": 6, "06": [443, 453, 474], "0638": 449, "06450": 368, "0645099": 426, "06561": 488, "06675": 480, "07467": 401, "08": [17, 182, 441, 476, 477, 478, 479, 485], "08022": 366, "081": 491, "08415": 432, "08494": 362, "08619": 424, "08681": [376, 454], "09864": 6, "0999938": 489, "0999961": 487, "0f": 0, "1": [0, 1, 2, 3, 4, 6, 7, 15, 19, 29, 30, 39, 47, 50, 101, 102, 103, 104, 105, 106, 107, 115, 120, 121, 142, 143, 144, 145, 148, 151, 152, 153, 155, 156, 158, 159, 160, 161, 162, 163, 165, 177, 181, 188, 190, 195, 196, 197, 198, 200, 202, 203, 219, 223, 238, 240, 244, 248, 251, 252, 253, 259, 278, 280, 284, 285, 287, 300, 306, 307, 312, 313, 324, 327, 328, 332, 341, 343, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 359, 360, 361, 362, 366, 367, 368, 370, 373, 398, 401, 402, 405, 409, 410, 416, 419, 421, 422, 423, 424, 425, 426, 427, 428, 429, 431, 432, 433, 435, 438, 439, 440, 441, 442, 443, 444, 445, 446, 448, 449, 451, 452, 453, 458, 459, 461, 462, 463, 465, 468, 471, 473, 474, 475, 476, 477, 478, 479, 480, 482, 485, 486, 487, 488, 489, 490, 491, 498, 499, 500, 501, 502, 503, 505, 506, 507, 508], "10": [0, 3, 6, 7, 190, 211, 268, 273, 326, 341, 384, 468, 489, 491, 498, 499, 502], "100": [2, 5, 6, 440, 490, 498, 501, 504, 508], "1000": [2, 487, 498], "10000": 405, "101": 490, "1024": [1, 6], "105361": 440, "10_000": 5, "10x": 480, "11": 202, "12": [6, 177, 190, 489], "1212": 474, "123": [499, 503], "12451": 422, "128": [273, 341], "13": 9, "14": [9, 190], "15": [1, 9, 190, 202, 279, 328, 498], "150594": 421, "15268": 423, "16": [1, 148, 332, 343, 345, 366, 373, 375, 377, 471], "1606": 432, "1607": [366, 368], "16384": 177, "16506": 424, "168": 499, "17": [4, 9], "177208": 423, "18": 190, "1803": 362, "1908": [376, 454], "1910": 401, "191107": 421, "192": 499, "1985": 202, "1_000": 5, "1d": [0, 101, 105, 108, 271, 301], "1e": [0, 5, 7, 17, 182, 346, 362, 366, 368, 369, 401, 441, 443, 453, 473, 474, 475, 476, 477, 478, 479, 482, 485, 487, 488, 489, 490, 491], "1e3": 498, "1st": 244, "2": [0, 1, 2, 4, 5, 6, 7, 39, 102, 106, 115, 120, 121, 137, 143, 144, 145, 153, 156, 158, 159, 160, 161, 162, 163, 165, 177, 188, 190, 194, 195, 196, 197, 198, 199, 200, 202, 203, 206, 207, 213, 219, 244, 252, 257, 284, 285, 304, 307, 309, 310, 311, 312, 324, 328, 332, 341, 343, 344, 345, 349, 352, 359, 369, 373, 374, 375, 401, 410, 419, 420, 421, 422, 423, 424, 425, 426, 427, 431, 442, 443, 445, 452, 453, 468, 471, 473, 474, 476, 477, 478, 482, 485, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508], "20": [177, 190, 202], "200": [6, 489, 501], "2002": 6, "2011": 476, "2012": [474, 485], "2015": [355, 477, 479], "2019": [6, 478], "2020": 6, "2021": 6, "20397": 440, "20_000": 6, "21": [6, 190, 491], "2104": 6, "223144": 440, "223404": 422, "225": 202, "225763": 449, "2302": 480, "23607": [202, 203], "24": 190, "24264": 202, "247": 6, "25": [9, 398, 419], "25211": 423, "256": [1, 7, 148], "256995": 449, "27": 4, "28": [177, 190], "2d": [0, 102, 106, 121, 244, 346, 355], "2nd": 244, "2x": 505, "3": [0, 1, 2, 4, 6, 9, 103, 107, 115, 143, 145, 165, 188, 190, 196, 197, 198, 202, 203, 284, 285, 312, 324, 328, 345, 350, 353, 359, 375, 419, 422, 424, 431, 436, 475, 480, 495, 498, 499, 500, 502, 505, 506], "30": 475, "3118": 505, "32": [1, 6, 7, 92, 244, 245, 332, 344, 345, 374, 375, 401, 498], "32mib": 321, "330": 6, "33333": 419, "33554432": 321, "348587": 442, "363207": 421, "36788": 498, "379159": 422, "380709": 426, "39": 6, "3d": [0, 2, 103, 107, 346, 356, 419], "3f": [2, 7, 498], "3x": 2, "4": [0, 1, 2, 6, 119, 148, 151, 165, 170, 190, 202, 244, 245, 273, 284, 312, 322, 328, 332, 343, 344, 345, 346, 366, 373, 374, 375, 399, 400, 418, 419, 421, 422, 423, 440, 498, 499, 500, 502, 506, 508], "4096": [2, 498, 501, 508], "40x": 1, "41421": 202, "417497": 427, "42": 329, "437": 6, "44": 6, "447214": 203, "458835": 423, "475": 6, "48095": 421, "4d": [1, 419], "4m": 1, "5": [0, 1, 2, 5, 6, 9, 190, 202, 247, 278, 284, 328, 343, 346, 354, 355, 356, 359, 363, 366, 373, 414, 419, 420, 423, 424, 431, 434, 452, 464, 468, 473, 485, 487, 488, 498, 501, 502], "50": [0, 208], "500": [6, 508], "510826": 440, "512": [3, 6, 418, 508], "534422": 426, "539245": 440, "53947": 421, "55": 1, "559": 2, "5701": 474, "573409": 449, "57771": 203, "579": 6, "5f": 5, "6": [1, 2, 6, 115, 190, 202, 273, 284, 404, 418, 422, 431, 432, 436, 443, 453, 457, 485, 498, 502, 506], "61278": 421, "617261": 427, "628": 6, "633": 6, "639": 501, "64": [0, 1, 92, 119, 170, 244, 245, 322, 332, 399, 400], "64331": 424, "666329": 424, "66667": 419, "67326": 458, "676": 1, "690": 6, "6967": 423, "7": [2, 6, 190, 202, 244, 502], "702": [359, 432], "707107": 197, "71828": 498, "74166": 202, "74597": 202, "75": 419, "75596": 449, "75787": 423, "765166": 449, "773433": 449, "774": 2, "776856": 422, "793615": 424, "79854": 424, "7b": 6, "7m": 1, "8": [0, 1, 2, 6, 9, 202, 244, 332, 344, 345, 366, 374, 375, 418, 441, 474, 475, 476, 477, 478, 479, 485, 498, 502, 506, 508], "8192": [6, 177], "84804": 202, "863726": 427, "883935": 427, "890597": 422, "894427": 203, "89613": 421, "8gb": 6, "8x": 1, "9": [4, 9, 202, 442, 474, 477, 478, 479, 480, 482, 488, 491, 505], "90041": 422, "912766": 422, "916291": 440, "95": 7, "982273": 426, "99": [480, 485], "995016": 421, "999": [477, 478, 479], "A": [0, 2, 6, 8, 9, 10, 69, 83, 97, 143, 144, 147, 148, 149, 151, 174, 188, 189, 195, 197, 198, 200, 202, 203, 206, 209, 218, 219, 220, 224, 228, 244, 247, 248, 249, 251, 252, 253, 254, 255, 258, 259, 288, 292, 295, 313, 316, 317, 322, 323, 324, 325, 326, 327, 328, 329, 330, 341, 346, 355, 361, 362, 366, 368, 381, 385, 386, 389, 395, 396, 401, 407, 410, 418, 421, 422, 424, 432, 453, 454, 471, 473, 477, 479, 481, 482, 484, 489, 498, 499, 500, 501, 503, 504, 505], "AS": 169, "And": [4, 6, 419], "As": [7, 39, 300, 341, 499], "At": [96, 312, 499], "But": [500, 508], "By": [6, 322, 389, 440, 499, 501, 505], "For": [0, 1, 2, 4, 6, 9, 39, 115, 151, 169, 188, 202, 244, 329, 341, 346, 355, 359, 377, 382, 391, 394, 400, 405, 410, 419, 421, 422, 423, 424, 440, 468, 473, 495, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508], "If": [0, 1, 2, 4, 6, 9, 16, 17, 18, 19, 27, 28, 29, 30, 79, 83, 84, 96, 98, 108, 111, 112, 113, 114, 120, 121, 124, 125, 126, 128, 129, 130, 139, 147, 149, 150, 151, 161, 162, 163, 167, 168, 174, 182, 194, 195, 196, 202, 206, 209, 218, 219, 220, 222, 223, 228, 232, 236, 239, 240, 242, 243, 248, 252, 254, 263, 267, 275, 278, 286, 287, 288, 293, 297, 299, 300, 301, 304, 306, 307, 313, 314, 317, 319, 321, 322, 326, 328, 346, 348, 349, 350, 351, 352, 353, 362, 368, 370, 382, 384, 394, 400, 402, 405, 407, 410, 419, 440, 442, 453, 475, 477, 478, 498, 499, 500, 501, 503, 504, 507, 508, 509], "In": [0, 1, 2, 6, 7, 39, 151, 219, 244, 326, 341, 355, 362, 471, 474, 476, 477, 479, 480, 481, 497, 498, 499, 500, 501, 503, 504, 507, 508], "It": [2, 6, 9, 129, 174, 277, 313, 324, 328, 341, 396, 400, 481, 493, 499, 503, 505, 507], "Its": [341, 500], "No": [2, 6, 197, 198, 499], "Not": [97, 235, 498], "ON": [3, 4, 9], "Of": 501, "On": [1, 498, 501, 504], "One": [152, 155, 161, 239, 269, 498, 500, 501, 503], "THE": 9, "That": 6, "The": [0, 1, 2, 3, 4, 6, 7, 8, 9, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 48, 52, 62, 63, 69, 79, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 118, 119, 120, 121, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 136, 137, 138, 140, 141, 142, 143, 144, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 168, 169, 170, 172, 173, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 189, 190, 191, 192, 193, 196, 197, 198, 200, 201, 202, 203, 204, 205, 206, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 226, 228, 229, 230, 231, 233, 235, 236, 237, 238, 239, 240, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 257, 258, 259, 260, 261, 262, 263, 265, 266, 267, 271, 272, 275, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 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, 328, 329, 332, 334, 343, 344, 345, 346, 348, 349, 350, 351, 352, 353, 354, 355, 356, 358, 360, 361, 362, 366, 367, 368, 370, 373, 374, 375, 377, 378, 382, 384, 388, 389, 390, 391, 394, 395, 396, 397, 399, 400, 401, 402, 405, 407, 410, 416, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 433, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 465, 468, 471, 473, 474, 475, 476, 477, 478, 479, 480, 483, 485, 486, 487, 490, 493, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509], "Then": [5, 9], "There": [1, 341, 419, 498], "These": [1, 2, 97, 243, 301, 442, 508], "To": [0, 2, 3, 5, 6, 7, 9, 200, 275, 341, 468, 473, 498, 499, 500, 501, 506], "With": 500, "_": [1, 3, 5, 6, 327, 341, 487, 488, 489, 490, 491, 495, 498, 504, 508], "__call__": [1, 6, 7, 341, 471, 500], "__init__": [2, 6, 7, 10, 11, 12, 31, 115, 123, 164, 330, 341, 471], "__main__": [2, 6], "__name__": [2, 6], "_a": 2, "_ext": 2, "_f": 202, "_in": [421, 422], "_out": [421, 422], "_p": 453, "_val": 435, "a1": 169, "a2": 169, "a_": 202, "a_max": [0, 96], "a_min": [0, 96], "a_ndim": 1, "a_shap": 1, "a_strid": 1, "a_view": 505, "ab": [0, 17, 182, 202, 313, 362, 366, 368, 376, 401, 432, 454, 498, 500], "abil": 499, "abl": [2, 4, 244, 503], "abort": 115, "about": [1, 2, 6, 7, 134, 224, 504, 508], "abov": [1, 2, 6, 244, 310, 341, 419, 499, 500, 501, 502, 503, 504, 508], "absolut": [0, 13, 17, 182, 431, 432, 452, 499], "acc": 328, "acceler": [4, 346], "accept": [499, 503], "access": [0, 6, 51, 341, 471, 482, 499, 504, 508], "accord": [0, 249, 318, 322, 397, 421, 422, 423, 424], "accordingli": 2, "accumul": [328, 401], "accuraci": 7, "accustom": 6, "achiev": [341, 499], "across": [1, 2, 9, 321, 362, 499], "act": [2, 448], "action": 341, "activ": [2, 9, 171, 355, 416, 418, 434, 454, 464, 465, 467, 498], "actual": [6, 19, 384, 471, 504], "ad": [0, 1, 2, 5, 9, 147, 327, 366, 471, 474, 475, 476, 477, 478, 479, 485, 499, 504, 507], "adadelta": 473, "adafactor": 473, "adagrad": 473, "adam": [473, 479, 480, 489, 490], "adamax": 473, "adamw": [473, 480], "adapt": [474, 475, 476, 499], "add": [0, 1, 2, 3, 4, 6, 15, 39, 141, 214, 239, 244, 348, 349, 350, 351, 352, 353, 500, 501, 503, 508], "add_argu": 6, "add_depend": 2, "add_execut": 4, "add_fun": 500, "add_librari": 2, "addit": [0, 2, 4, 6, 9, 14, 15, 143, 147, 149, 151, 209, 346, 362, 368, 397, 401, 471, 501], "addmm": 0, "address": 2, "adjac": 355, "advanc": [6, 498], "advantag": 508, "advis": 505, "affin": [346, 362, 366, 368, 370, 400], "after": [2, 6, 7, 29, 95, 165, 167, 170, 240, 244, 346, 362, 368, 377, 378, 382, 384, 391, 394, 395, 396, 397, 418, 452, 498, 499, 508], "after_1": 239, "after_2": 239, "after_i": 239, "after_n": 239, "afternoon": 6, "again": [6, 9, 341, 498], "against": [0, 4], "aggreg": [397, 499], "ago": 6, "ai": 115, "aim": 499, "ainv": [199, 207], "albeit": 508, "algebra": 8, "algorithm": [419, 480], "alia": [99, 100, 359], "alibi": 341, "align": [195, 244, 361, 367], "align_corn": 419, "all": [0, 1, 2, 3, 7, 9, 17, 29, 39, 85, 86, 87, 97, 102, 103, 104, 106, 107, 115, 124, 125, 126, 144, 146, 154, 157, 160, 163, 169, 170, 206, 219, 239, 240, 267, 291, 321, 322, 341, 377, 378, 382, 385, 386, 387, 392, 394, 397, 410, 418, 419, 468, 471, 493, 495, 498, 502, 503, 504, 506, 509], "all_avg": 499, "all_reduce_grad": 499, "all_reduce_s": 321, "all_sum": 499, "allclos": [0, 1, 148], "alloc": [2, 172, 275, 278, 471], "allow": [0, 1, 2, 143, 145, 188, 324, 341, 396, 471, 493, 499, 502, 503, 506], "allow_col_major": 0, "almost": [6, 499], "alon": [2, 505], "along": [0, 2, 27, 28, 97, 98, 111, 112, 113, 114, 124, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 169, 170, 177, 196, 202, 243, 254, 263, 267, 286, 288, 292, 300, 301, 304, 305, 306, 307, 315, 341, 360, 402, 433], "alpha": [0, 2, 15, 244, 347, 357, 428, 429, 453, 455, 458, 478, 485], "alpha_": 2, "alreadi": [2, 3, 6, 499], "also": [0, 1, 2, 4, 6, 7, 8, 9, 12, 14, 88, 90, 91, 122, 131, 132, 136, 154, 157, 160, 163, 175, 176, 191, 192, 193, 214, 221, 229, 231, 235, 241, 244, 262, 266, 296, 322, 323, 334, 341, 381, 395, 397, 399, 400, 408, 430, 458, 460, 467, 473, 498, 499, 500, 501, 502, 503, 504, 505, 506, 509], "altern": 495, "although": 499, "alwai": [1, 84, 171, 180, 325, 499, 500, 501], "am": 6, "among": 2, "amount": [6, 173, 278, 343, 373, 500], "amus": 6, "an": [0, 1, 2, 3, 4, 6, 7, 9, 11, 16, 18, 31, 85, 86, 87, 93, 101, 102, 103, 104, 105, 106, 107, 123, 128, 129, 130, 139, 143, 145, 146, 147, 151, 165, 168, 178, 180, 183, 194, 202, 209, 230, 236, 237, 239, 242, 243, 244, 245, 254, 263, 265, 267, 268, 278, 279, 288, 291, 298, 300, 301, 304, 305, 309, 312, 317, 319, 320, 325, 326, 327, 328, 332, 341, 354, 359, 362, 367, 368, 370, 377, 397, 398, 400, 402, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 431, 455, 468, 473, 474, 484, 488, 493, 495, 497, 498, 499, 500, 501, 502, 504, 505, 506, 507, 508, 509], "anaconda": 499, "anchor": 453, "angl": [118, 246, 369], "angular": [150, 405], "ani": [0, 1, 2, 6, 8, 19, 97, 115, 126, 321, 325, 326, 327, 328, 329, 341, 359, 377, 378, 381, 390, 400, 418, 419, 468, 490, 497, 498, 499, 501, 504, 506, 507, 508], "anonym": 498, "anoth": [0, 96, 188, 219, 296, 318, 332, 341, 377, 498, 500, 501, 502, 508], "anwywher": 9, "anyhow": 6, "anymor": 6, "anyth": [6, 313, 499, 504], "anytim": 504, "api": [1, 2, 143, 145, 180, 359, 499, 500, 501], "app": 9, "append": [6, 219, 498, 504], "appl": [2, 6, 8, 9, 508], "appli": [0, 39, 150, 151, 169, 206, 326, 327, 328, 341, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 355, 356, 357, 359, 360, 362, 363, 364, 365, 366, 368, 369, 370, 371, 372, 373, 374, 375, 376, 378, 391, 398, 400, 401, 402, 403, 404, 406, 408, 409, 411, 412, 413, 414, 415, 416, 417, 419, 428, 429, 430, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 468, 477, 478, 481, 484, 490, 493, 498, 499], "applic": [3, 9], "apply_fn": 378, "apply_gradi": 473, "apply_to_modul": [341, 382], "approach": [448, 501], "appropri": 498, "approx": 359, "approxim": [17, 359, 430, 431, 432], "ar": [0, 1, 2, 5, 6, 7, 8, 9, 17, 19, 83, 92, 93, 96, 97, 104, 108, 115, 121, 126, 128, 129, 139, 146, 148, 151, 153, 154, 156, 157, 159, 160, 162, 163, 165, 170, 174, 182, 183, 184, 185, 186, 187, 188, 189, 197, 198, 200, 202, 203, 209, 219, 238, 239, 240, 244, 245, 247, 248, 249, 254, 255, 258, 259, 267, 273, 274, 291, 292, 300, 313, 316, 317, 322, 325, 326, 332, 346, 348, 349, 350, 351, 352, 353, 354, 355, 356, 362, 366, 368, 370, 384, 397, 400, 419, 440, 442, 443, 467, 471, 473, 480, 482, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508], "arang": [0, 1, 202, 254, 332, 419, 502, 505], "arbitrari": [325, 471, 499], "arbitrarili": [1, 97, 341, 497, 501, 506], "arc": 0, "arcco": 0, "arccosh": 0, "architectur": [6, 9, 224, 341, 396, 508], "archiv": 507, "arcsin": 0, "arcsinh": 0, "arctan": 0, "arctan2": 0, "arctanh": 0, "arg": [2, 6, 11, 19, 123, 139, 143, 144, 145, 164, 180, 273, 274, 330, 503], "arg1": 188, "arg2": 188, "argmax": [0, 7], "argmin": 0, "argnam": [174, 313], "argnum": [2, 115, 174, 313, 501], "argpars": 6, "argpartit": 0, "argsort": 0, "argument": [1, 32, 66, 80, 97, 139, 174, 313, 326, 327, 328, 341, 419, 495, 499, 500, 501, 503, 507, 508, 509], "argumentpars": 6, "ari": [85, 86, 87], "aris": 505, "arm": 9, "arm64": 9, "around": 6, "arr": [0, 270, 502], "arr_0": 507, "arrai": [0, 1, 2, 4, 6, 7, 8, 11, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 118, 119, 120, 121, 124, 125, 128, 129, 130, 131, 132, 133, 134, 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, 165, 166, 167, 168, 169, 170, 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, 228, 229, 230, 231, 232, 233, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 257, 258, 259, 260, 261, 262, 263, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 296, 297, 298, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 324, 332, 341, 346, 367, 377, 384, 387, 392, 398, 419, 420, 421, 422, 423, 424, 425, 426, 427, 433, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 465, 468, 471, 474, 475, 476, 477, 478, 479, 480, 485, 486, 487, 488, 489, 490, 491, 498, 499, 500, 501, 504, 505, 506, 508], "array_equ": [0, 17, 182], "arrayfir": 8, "arxiv": [6, 362, 366, 368, 376, 401, 432, 454, 474, 480], "as_strid": 0, "ascend": [197, 198], "ask": [6, 499, 503], "assert": [1, 2, 148], "assign": [0, 2, 39, 471, 499], "associ": [2, 273, 274, 504], "assum": [0, 6, 92, 196, 197, 198, 203, 326, 341, 362, 499], "astyp": [0, 1, 2, 6, 148, 377, 505], "atleast": 0, "atleast_1d": 0, "atleast_2d": 0, "atleast_3d": 0, "atol": [0, 17, 182], "atom": [1, 148], "atomic_fetch_add_explicit": 1, "atomic_output": [1, 148], "attach": 2, "attempt": [97, 499], "attend": 397, "attent": [151, 382, 397, 410, 418], "attention_norm": 6, "attribut": [1, 10, 11, 12, 31, 164, 330, 390, 471, 493], "audio": 419, "auto": [0, 2, 4, 9, 499, 500], "autom": 501, "automat": [1, 2, 8, 148, 209, 499, 506, 507, 508], "autoregress": 6, "avail": [2, 5, 6, 7, 9, 11, 126, 127, 225, 278, 334, 499, 503, 508], "averag": [321, 343, 344, 345, 474, 475, 477, 478, 479, 499], "avgpool1d": 341, "avgpool2d": 341, "avgpool3d": 341, "avoid": [1, 2, 389, 498, 499], "awai": [2, 6], "awar": [498, 504], "ax": [0, 2, 16, 18, 27, 28, 80, 115, 141, 153, 154, 156, 157, 159, 160, 162, 163, 165, 181, 202, 204, 205, 218, 220, 222, 228, 239, 242, 267, 284, 285, 286, 291, 293, 297, 298, 304, 308, 314, 501], "axes_a": 0, "axes_b": 0, "axi": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 72, 75, 76, 77, 81, 98, 111, 112, 113, 114, 121, 124, 141, 147, 149, 152, 155, 158, 159, 160, 161, 162, 163, 165, 177, 196, 200, 202, 218, 220, 222, 228, 230, 239, 240, 242, 243, 248, 254, 263, 267, 286, 287, 288, 291, 292, 293, 297, 298, 300, 301, 305, 306, 307, 308, 312, 314, 315, 317, 343, 344, 345, 360, 373, 374, 375, 402, 433, 439, 441, 442, 446, 451, 453, 461, 462, 502], "axis1": [0, 47, 78, 121, 298, 307], "axis2": [0, 47, 78, 121, 298, 307], "axpbi": 2, "axpby_": 2, "axpby_gener": 2, "axpby_general_bfloat16": 2, "axpby_general_complex64": 2, "axpby_general_float16": 2, "axpby_general_float32": 2, "axpby_impl": 2, "b": [0, 1, 2, 3, 4, 6, 14, 15, 17, 25, 83, 88, 90, 91, 92, 131, 132, 136, 148, 151, 167, 169, 175, 176, 180, 181, 182, 190, 191, 192, 193, 196, 202, 204, 205, 214, 215, 217, 219, 221, 229, 231, 235, 238, 241, 244, 251, 262, 266, 296, 304, 313, 327, 328, 360, 370, 402, 419, 433, 501, 502, 504, 505, 506, 507, 508], "b1": 169, "b2": 169, "b_": [361, 367], "b_stride": 1, "ba": [477, 479], "back": [6, 115, 225, 505], "backend": [1, 9, 126, 127, 503], "backward": [1, 498, 501], "bad": 504, "balanc": 448, "baltimor": 202, "bandwidth": [498, 499], "base": [0, 2, 4, 150, 211, 213, 241, 405, 418, 471, 473, 479, 493, 495, 498, 502], "base_idx": 1, "basi": 493, "basic": [5, 268, 501], "batch": [6, 15, 92, 151, 169, 170, 219, 252, 346, 348, 349, 350, 351, 352, 353, 355, 356, 361, 367, 397, 402, 419, 504], "batch_idx": 1, "batch_iter": [7, 473], "batch_siz": [7, 473], "batchnorm": 341, "becaus": [6, 171, 341, 498, 499, 500, 504], "becom": 126, "been": [0, 2, 6, 172, 504], "befor": [1, 2, 6, 9, 29, 148, 240, 321, 381, 418, 482, 499, 502, 504], "before_1": 239, "before_2": 239, "before_i": 239, "before_n": 239, "beforehand": 238, "beggin": 267, "begin": [84, 173, 195, 244, 361, 367, 416, 434, 445, 452, 458, 464, 465, 499], "behav": 115, "behavior": [200, 252, 448, 502, 504], "behaviour": [115, 194, 195], "behind": 501, "being": [294, 341], "bell": 2, "below": [2, 9, 202, 309, 311, 332, 419, 499, 504], "bench": 2, "benchmark": [2, 498], "benefici": [355, 356, 504], "benefit": 499, "best": 499, "beta": [0, 2, 15, 119, 244, 346, 362, 366, 368, 452, 473, 477, 478, 479, 480], "beta_": 2, "beta_1": [475, 477, 478, 479, 480], "beta_2": [477, 478, 479, 480], "better": [321, 501, 508], "between": [0, 2, 8, 96, 165, 418, 441, 444, 445, 448, 489, 499, 503, 504, 505, 508], "beyond": [267, 487, 490], "bfloat16": [2, 12, 177, 332, 505], "bfloat16_t": 2, "bia": [6, 119, 147, 170, 244, 245, 326, 341, 348, 349, 350, 351, 352, 353, 361, 367, 368, 370, 382, 384, 394, 397, 400, 402, 477, 478, 479, 482, 501], "bias": [0, 119, 170, 244, 245, 361, 367, 382, 394, 397], "bias_correct": [477, 478], "bicub": 419, "big": [1, 321, 498], "bigger": [6, 475], "bilinear": [1, 419], "binari": [209, 270, 271, 272, 273, 274, 315, 416, 440, 465, 498, 503], "binary_cross_entropi": [341, 498], "bind": 503, "bit": [0, 119, 170, 191, 244, 245, 266, 322, 332, 377, 399, 400, 401], "bitwis": [0, 88, 89, 90, 91, 191, 266], "bitwise_and": 0, "bitwise_invert": 0, "bitwise_or": 0, "bitwise_xor": 0, "block": [0, 2, 6, 92, 418], "block_masked_mm": 0, "block_siz": [0, 92], "bn": 346, "bodi": [1, 148], "bool": [0, 1, 2, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 76, 77, 79, 81, 83, 97, 104, 111, 112, 113, 114, 126, 127, 143, 145, 148, 150, 170, 182, 188, 194, 195, 202, 205, 206, 207, 209, 218, 220, 222, 223, 225, 228, 242, 245, 293, 297, 314, 322, 346, 348, 349, 350, 351, 352, 353, 361, 362, 366, 367, 368, 370, 377, 381, 382, 384, 389, 391, 394, 397, 400, 402, 405, 410, 418, 419, 440, 443, 475, 477, 478, 486], "bool_": [12, 332], "boolean": [0, 17, 83, 151, 182, 183, 184, 185, 186, 187, 188, 215, 216, 217, 332, 393, 502], "both": [1, 2, 14, 88, 90, 91, 131, 132, 136, 175, 176, 188, 191, 192, 193, 202, 214, 221, 229, 231, 235, 241, 248, 262, 266, 296, 322, 343, 344, 345, 366, 367, 373, 374, 375, 473, 498, 499, 500, 501, 506, 508], "bottom": 419, "bound": [0, 255, 258, 259, 359, 427, 498, 502, 508], "boundari": 489, "bracket": 6, "brain": 332, "break": 505, "bregler": 355, "bridg": 499, "broadcast": [0, 2, 14, 17, 88, 90, 91, 93, 96, 131, 132, 136, 151, 168, 175, 176, 182, 191, 192, 193, 214, 219, 221, 229, 231, 235, 241, 243, 247, 248, 252, 255, 258, 259, 262, 266, 296, 301, 318, 397], "broadcast_arrai": [0, 2], "broadcast_to": 0, "broadcasted_input": 2, "brought": 8, "btl_tcp_if_includ": [499, 503], "btl_tcp_link": [499, 503], "buffer": [1, 2, 171, 505], "bui": 6, "build": [3, 4, 6, 8, 423, 471, 498, 500], "build_ext": [2, 9], "build_shared_lib": [2, 9], "built": [1, 2, 4, 9, 504], "bundl": 6, "byte": [52, 62, 171, 172, 173, 275, 278, 279, 321, 332, 503], "c": [0, 1, 2, 6, 15, 202, 346, 348, 349, 350, 351, 352, 353, 355, 356, 366, 367, 505, 506, 508], "c_": [367, 480], "c_in": [101, 102, 103, 104, 105, 106, 107], "c_out": [101, 102, 103, 104, 105, 106, 107], "c_pad": 1, "c_t": [367, 480], "cabl": 499, "cach": [6, 9, 95, 171, 172, 275, 498], "calcul": [202, 440, 443, 449, 475], "call": [2, 3, 6, 7, 32, 95, 126, 129, 167, 173, 180, 321, 341, 358, 382, 394, 399, 407, 471, 473, 482, 498, 499, 500, 501, 503, 504], "callabl": [97, 115, 143, 145, 148, 174, 180, 189, 313, 316, 317, 322, 323, 325, 326, 327, 328, 377, 378, 381, 389, 402, 407, 418, 420, 421, 422, 423, 424, 425, 426, 427, 474, 475, 476, 477, 478, 479, 480, 485, 486, 487, 488, 489, 490, 491], "can": [1, 2, 3, 4, 6, 8, 9, 14, 19, 66, 80, 84, 88, 90, 91, 97, 121, 122, 123, 131, 132, 136, 139, 143, 144, 151, 175, 176, 180, 191, 192, 193, 202, 214, 221, 229, 231, 235, 241, 247, 248, 255, 258, 259, 262, 266, 271, 279, 296, 307, 312, 313, 328, 341, 344, 345, 358, 359, 374, 375, 381, 394, 399, 407, 419, 442, 468, 471, 473, 481, 482, 495, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509], "cannot": [6, 96, 502, 505], "captur": [2, 3, 97, 115, 226, 227, 341, 498], "care": [6, 499, 500, 503, 504], "carefulli": [498, 500], "carri": 2, "cartesian": 223, "case": [2, 6, 124, 125, 126, 128, 129, 130, 154, 157, 158, 160, 161, 162, 163, 165, 194, 195, 196, 197, 198, 199, 200, 201, 203, 204, 205, 206, 207, 219, 265, 291, 312, 344, 345, 355, 374, 375, 416, 434, 452, 458, 464, 465, 481, 482, 498, 499, 500, 501, 503, 506, 507, 508, 509], "cast": [2, 38, 161, 162, 163, 209, 321, 377, 389, 505], "caster": 2, "categor": 6, "categori": [12, 188, 332], "caus": [341, 498, 504], "causal": [6, 151], "caution": 84, "cd": [3, 9], "cdf": [249, 359, 430], "cdot": [432, 441, 444, 460], "ceil": 0, "ceildiv": 1, "cell": 367, "celu": 341, "certain": [391, 498], "chang": [84, 97, 143, 145, 180, 277, 315, 395, 400, 419, 445, 452, 498, 505], "channel": [1, 101, 102, 103, 104, 105, 106, 107, 346, 348, 349, 350, 351, 352, 353, 355, 356], "channel_idx": 1, "charact": 325, "check": [0, 2, 9, 83, 127, 188, 197, 198, 225, 384, 499, 500, 501, 502], "checklist": [499, 503], "checkout": [3, 498], "checkpoint": [418, 473], "chen": 480, "child": 396, "children": 341, "chip": 9, "choleski": 195, "choos": [6, 150, 405, 503], "chosen": 134, "clamp": 165, "clang": 9, "clarifi": 499, "clariti": 501, "class": [2, 6, 7, 10, 11, 12, 31, 115, 123, 164, 330, 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, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 471, 474, 475, 476, 477, 478, 479, 480, 485, 486, 493], "class_pred": 322, "classif": [423, 424], "classifi": 7, "classmethod": [399, 400], "clear": 95, "click": 9, "clip": [0, 324, 440, 475], "clip_threshold": 475, "clipped_grad": 324, "clone": 9, "close": [5, 8, 9, 17, 182], "closer": 326, "cmake": [3, 4, 9], "cmake_arg": 3, "cmake_build_parallel_level": 9, "cmake_build_typ": 9, "cmake_current_list_dir": 2, "cmake_cxx_standard": 4, "cmake_cxx_standard_requir": 4, "cmake_host_system_processor": 9, "cmake_library_output_directori": 2, "cmake_minimum_requir": 4, "cmakebuild": 2, "cmakeextens": 2, "cmakelist": [2, 4], "cmdclass": 2, "co": [0, 2, 115, 410, 501], "code": [1, 148, 498, 499, 500, 504], "coeffici": [2, 474, 475, 477, 478, 479, 480], "col": 309, "cold": 9, "collect": [326, 327, 497], "column": [146, 178, 197, 244], "com": 9, "combin": [6, 206, 328], "come": [2, 6, 499, 501], "command": [2, 3, 4, 9, 499, 503], "command_buff": 2, "common": [473, 498, 504], "commonli": [7, 395, 468, 498], "commun": [8, 123, 126, 127, 321, 503], "communication_typ": 321, "compact": 201, "compar": [2, 83, 498], "comparison": [17, 136, 175, 176, 192, 193, 235], "compat": [6, 143, 145, 151, 180, 248, 252, 359, 507], "compil": [0, 3, 4, 8, 9, 122, 135, 148, 499, 500, 501, 504], "compiled_fun": [498, 500], "compiled_grad_fn": 498, "complement": 89, "complet": [5, 6, 9, 395, 396, 500, 501, 508], "complex": [2, 99, 100, 159, 160, 161, 162, 163, 179, 197, 198, 260, 325, 332, 341, 396, 498, 500, 501], "complex64": [2, 12, 332], "complex64_t": 2, "complexflo": 12, "compon": [2, 4, 6, 206], "compos": [8, 341, 498, 501, 506], "composit": 506, "compress": 274, "compromis": 6, "comput": [0, 1, 2, 5, 6, 7, 8, 9, 111, 112, 113, 114, 115, 119, 134, 142, 150, 174, 189, 190, 194, 195, 196, 197, 198, 199, 200, 201, 202, 204, 205, 207, 214, 222, 238, 244, 262, 286, 293, 294, 304, 313, 314, 316, 323, 341, 346, 361, 362, 366, 367, 368, 382, 395, 400, 401, 405, 418, 421, 422, 423, 424, 431, 432, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 473, 474, 475, 477, 478, 479, 480, 484, 498, 499, 500, 501, 506, 508], "computation": 504, "compute_encod": 2, "compute_uv": 206, "concaten": [0, 6, 124, 321], "concept": 471, "concis": 6, "concret": [2, 361, 367, 370, 402, 504, 508], "conda": [9, 499], "condit": [0, 318, 498, 508], "config": [2, 4, 499], "configu": 473, "configur": [119, 499], "confirm": [499, 503], "confus": 7, "conj": 100, "conjug": [0, 99], "connect": [499, 503], "consecut": [150, 244, 405], "consequ": 6, "consid": [6, 17, 83, 182, 325, 326, 327, 362, 497, 499], "consider": 498, "const": [0, 1, 2, 443], "constant": [0, 2, 6, 9, 115, 147, 149, 239, 341, 346, 362, 368, 401, 443, 453, 485, 487, 498, 500, 505], "constant_valu": 239, "constitut": 326, "construct": [0, 2, 7, 46, 120, 168, 200, 236, 305, 319], "consult": 499, "consum": 504, "contain": [2, 6, 9, 29, 30, 69, 97, 121, 134, 158, 159, 160, 169, 170, 197, 202, 215, 216, 217, 244, 288, 318, 321, 324, 341, 381, 383, 384, 390, 418, 449, 468, 471, 498, 501], "content": [9, 381, 498], "context": [295, 500], "contigu": [0, 1, 2, 84, 148], "continu": [347, 428, 499, 501], "contract": [0, 134], "contribut": 2, "contriv": [501, 508], "control": [0, 369, 495, 504], "conv": 108, "conv1d": [0, 341], "conv2d": [0, 341], "conv3d": [0, 341], "conv_gener": 0, "conv_transpose1d": 0, "conv_transpose2d": 0, "conv_transpose3d": 0, "conveni": [1, 2, 7, 188], "convent": [19, 108, 133, 134, 419], "convers": 8, "convert": [0, 1, 2, 79, 85, 86, 87, 118, 165, 246, 399, 400, 504, 505, 506], "convolut": [0, 101, 102, 103, 104, 105, 106, 107, 108, 348, 349, 350, 351, 352, 353, 355, 356], "convolv": [101, 102, 103, 104, 105, 106, 107], "convtranspose1d": 341, "convtranspose2d": 341, "convtranspose3d": 341, "coordin": [0, 223], "copi": [0, 1, 2, 6, 8, 240, 287, 505], "core": [1, 2, 3, 4, 5, 6, 7, 322, 341, 343, 344, 345, 346, 366, 373, 374, 375, 384, 387, 389, 392, 419, 420, 421, 422, 423, 424, 425, 426, 427, 440, 442, 449, 468, 471, 473, 498, 499, 505, 506], "corner": 419, "correct": [2, 9, 477, 478, 479, 502, 504], "correctli": [39, 499], "correl": [104, 355], "correspond": [0, 1, 2, 16, 18, 79, 96, 119, 121, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 197, 218, 220, 228, 242, 284, 285, 297, 304, 312, 317, 326, 499, 501, 503], "cos_first": 410, "cosh": [0, 448], "cosin": [0, 20, 21, 109, 110, 441, 487, 489, 501], "cosine_decai": [473, 489], "cosine_similarity_loss": 341, "cost": [9, 475, 499, 504], "costli": 504, "cot": 1, "cot_index": 1, "cotan": [2, 115], "cotang": [1, 2, 115, 316], "could": [6, 341], "count": [341, 489], "counter": 495, "cours": 501, "coursera": 485, "cout": [4, 500], "cov": 252, "covari": [252, 346], "cover": 2, "cpp": [2, 4], "cpu": [8, 9, 197, 198, 203, 332, 508], "cpython": 2, "crash": [84, 498], "creat": [0, 2, 6, 9, 84, 126, 146, 178, 295, 341, 471, 473, 489, 498, 499, 500, 502, 503, 505], "create_additive_causal_mask": 6, "cross": [7, 104, 440, 442], "cross_entropi": [7, 341], "crowd": 6, "cry": 6, "cubic": 419, "cummax": 0, "cummin": 0, "cumprod": 0, "cumsum": 0, "cumul": [0, 84, 111, 112, 113, 114], "current": [6, 8, 9, 84, 92, 103, 106, 107, 130, 172, 224, 244, 328, 341, 475, 499, 504], "custom": [8, 115, 148, 418], "custom_decod": 418, "custom_encod": 418, "custom_funct": 1, "custom_kernel_myexp_float": 1, "custom_tim": 2, "cvpr": 355, "cxx": 4, "cycl": 497, "d": [0, 1, 2, 6, 103, 107, 120, 121, 151, 181, 202, 219, 223, 238, 300, 307, 309, 310, 311, 329, 350, 353, 356, 361, 367, 402, 474, 477, 479, 508], "d1": 508, "d2": 508, "d2fdx2": 501, "d_i": 370, "dampen": 486, "darwin": 2, "data": [0, 2, 7, 8, 11, 19, 128, 146, 161, 162, 168, 178, 208, 232, 236, 249, 258, 307, 309, 315, 319, 356, 420, 421, 422, 423, 424, 425, 426, 427, 498, 499, 500, 502, 505], "dataset": [5, 499, 504], "datatyp": 52, "dbuild_shared_lib": 9, "dcmake_build_typ": [4, 9], "ddof": [0, 76, 81, 293, 314], "deal": 498, "debug": [1, 3, 499, 503], "debugg": 8, "decai": [475, 478, 480, 486, 487, 488, 491], "decay_r": [475, 488, 491], "decay_step": 487, "decent": 7, "decid": [326, 381], "decim": [0, 67, 268], "declar": 2, "decltyp": 1, "decod": 418, "decomposit": [194, 195, 206], "decor": [1, 115], "decoupl": 478, "dedic": 499, "deep": [346, 421, 422, 423, 424], "def": [1, 2, 5, 6, 7, 115, 143, 145, 148, 313, 341, 471, 498, 499, 500, 501, 502, 504, 505, 508], "default": [1, 2, 9, 15, 16, 17, 18, 19, 27, 28, 29, 30, 83, 84, 92, 97, 98, 101, 102, 103, 104, 105, 106, 107, 115, 116, 117, 119, 120, 121, 124, 125, 126, 128, 129, 130, 143, 145, 146, 148, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 169, 170, 174, 177, 178, 182, 190, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 218, 220, 222, 223, 228, 232, 236, 239, 240, 242, 244, 245, 247, 248, 249, 251, 252, 253, 254, 255, 257, 258, 259, 263, 265, 268, 275, 276, 277, 278, 279, 287, 288, 291, 292, 293, 295, 297, 299, 304, 306, 307, 308, 309, 310, 311, 312, 313, 314, 317, 319, 321, 322, 332, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 357, 360, 361, 363, 366, 367, 369, 370, 373, 374, 375, 377, 382, 384, 389, 391, 394, 397, 398, 399, 400, 402, 405, 410, 414, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 433, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 471, 474, 475, 476, 477, 478, 479, 480, 485, 486, 487, 495, 497, 498, 499, 500, 501, 503, 505, 507, 509], "default_devic": 509, "default_stream": 509, "defin": [1, 2, 5, 6, 7, 9, 115, 129, 148, 170, 196, 202, 245, 322, 325, 503, 505], "definit": [115, 194, 195, 252], "degre": [0, 246, 453], "delta": [445, 474], "delv": [423, 424], "demonstr": 505, "denomin": [366, 441, 474, 476, 477, 478, 479, 485], "dens": [223, 508], "depend": [0, 2, 3, 4, 5, 9, 79, 202, 361, 367, 402, 498, 499, 502, 507, 508], "depth": [325, 345, 350, 353, 356, 375, 501], "dequant": [0, 244], "deriv": [2, 500, 501, 504], "descend": 379, "descent": [486, 498, 504], "describ": [2, 504], "descript": [2, 4, 6, 332], "design": [1, 5, 8, 495, 508], "destin": [0, 2, 61, 130, 230, 243], "destroi": 498, "detach": 501, "detail": [1, 2, 11, 275, 341, 355, 405, 410, 419, 421, 422, 423, 424, 474, 476, 477, 479, 480, 499, 502, 506], "detect": 498, "determin": [0, 2, 121, 252, 328, 332, 388, 507], "dev": [2, 9], "develop": [2, 4, 9], "developer_dir": 9, "deviat": [0, 253, 293, 421, 423, 426], "deviatoin": 0, "devic": [1, 2, 8, 9, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 116, 117, 118, 119, 120, 121, 124, 125, 128, 129, 130, 131, 132, 133, 136, 137, 138, 140, 141, 142, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 168, 169, 170, 175, 176, 177, 178, 179, 181, 182, 183, 184, 185, 186, 187, 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, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 251, 252, 253, 254, 255, 257, 258, 259, 260, 261, 262, 263, 265, 266, 267, 268, 269, 276, 277, 278, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 314, 315, 318, 319, 320, 330, 508, 509], "device_info": 279, "devicetyp": 10, "df": 505, "dfdx": [500, 501, 502], "dft": [152, 153, 154, 155, 156, 157, 161, 162, 163], "dhwc": 356, "diag": [0, 206], "diagon": [0, 46, 120, 146, 307, 309, 310, 311], "dict": [97, 139, 144, 209, 224, 271, 272, 273, 322, 324, 387, 392, 395, 396, 471, 473, 481, 482, 484, 497, 500, 501, 507], "dict_kei": [326, 482], "dictionari": [6, 97, 143, 180, 209, 224, 271, 272, 324, 325, 328, 341, 381, 390, 395, 396, 483, 497, 507], "did": 6, "diff": 2, "differ": [8, 188, 296, 315, 452, 498, 499, 500, 501, 503], "differenti": [1, 8, 347, 428], "difficult": 501, "difficulti": [421, 422], "dilat": [0, 101, 102, 103, 104, 105, 106, 107, 348, 349, 350, 351, 352, 353], "dim": [1, 6, 150, 151, 358, 362, 366, 368, 397, 399, 401, 405, 410, 418], "dimens": [0, 1, 2, 6, 16, 18, 27, 28, 63, 69, 79, 85, 86, 87, 97, 102, 103, 104, 106, 107, 121, 141, 150, 151, 159, 160, 162, 163, 165, 169, 170, 181, 194, 195, 197, 198, 199, 200, 202, 203, 206, 207, 218, 219, 220, 222, 228, 242, 243, 244, 248, 257, 293, 297, 301, 304, 308, 314, 346, 348, 349, 350, 351, 352, 353, 355, 356, 360, 361, 362, 366, 367, 368, 397, 401, 402, 405, 418, 419, 433, 442, 498, 501], "dimension": [31, 147, 149, 152, 153, 154, 155, 156, 157, 161, 162, 163, 343, 344, 345, 346, 348, 349, 350, 351, 352, 353, 358, 370, 373, 374, 375, 399, 400, 410, 502, 505], "dir": 4, "direct": [6, 379, 480, 508], "directli": [2, 6, 84], "directori": [2, 4, 6, 9], "disabl": [122, 275, 321, 498, 499], "disable_compil": 498, "disappoint": 6, "discard": [6, 325], "discov": [9, 499], "discoveri": 480, "discret": [108, 152, 153, 154, 155, 156, 157, 161, 162, 163, 358, 399], "discuss": 2, "disk": 6, "dispatch": 2, "dispatch_thread": 2, "dispatchthread": 1, "displai": 341, "distanc": [6, 453], "distribut": [8, 9, 247, 248, 249, 251, 252, 253, 258, 259, 321, 370, 421, 422, 423, 424, 426, 427, 443, 446, 451, 453, 468], "distributed_config": [499, 503], "diverg": 446, "divid": [0, 2, 39, 167, 244, 262, 499], "divis": [0, 131, 167, 244, 262], "divisor": [293, 314], "divmod": 0, "dloss_dw": 501, "dloss_dx": 501, "dlpack": 505, "dlvalu": 313, "dmlx_build_cpu": 9, "dmlx_build_gguf": 9, "dmlx_build_safetensor": 9, "dmlx_metal_debug": 3, "dmlx_metal_jit": 9, "do": [0, 2, 6, 9, 200, 315, 341, 383, 394, 468, 471, 498, 499, 500, 501, 504], "doc": [2, 7, 499, 503], "document": [2, 3, 4, 66, 80, 148, 271, 272, 332, 498, 499, 500, 501, 502], "doe": [0, 2, 3, 6, 9, 171, 315, 324, 341, 498, 499, 502, 503, 504, 505], "doesn": [2, 341, 500], "domain": 258, "don": [1, 9, 498, 508], "done": [341, 354, 401, 498, 499, 504, 505], "dot": [144, 199, 207, 304, 325, 386, 397, 499], "doubl": [0, 6, 332], "doubt": 6, "down": [6, 324], "downsampl": [343, 344, 345, 373, 374, 375], "dparam": 313, "draw": 248, "drop": 381, "dropout": [341, 355, 356, 391, 418, 498], "dropout2d": 341, "dropout3d": 341, "dst": 130, "dt": 137, "dtype": [0, 1, 2, 6, 12, 19, 31, 38, 39, 79, 82, 128, 129, 146, 148, 164, 165, 168, 178, 188, 190, 197, 198, 202, 203, 208, 236, 249, 251, 252, 253, 255, 258, 259, 284, 285, 307, 309, 312, 315, 319, 321, 332, 389, 419, 420, 421, 422, 423, 424, 425, 426, 427, 440, 442, 449, 487, 488, 489, 490, 491, 498, 499, 500, 501, 502, 505, 506, 507], "dtypecategori": [188, 332], "dual": 448, "duchi": 476, "duplic": 500, "dure": [3, 97, 278, 354, 355, 356, 419, 505], "dx": 115, "dy": 115, "dyld": 499, "dyld_library_path": 499, "dylib": 2, "dynam": [0, 500, 504], "e": [2, 7, 9, 115, 137, 148, 169, 170, 189, 280, 346, 348, 349, 350, 351, 352, 353, 355, 356, 362, 366, 368, 382, 401, 438, 439, 461, 462, 467, 473, 476, 498, 500, 504, 509], "e5": 332, "e8": 332, "each": [0, 1, 2, 69, 119, 139, 150, 170, 188, 194, 195, 197, 198, 199, 206, 207, 219, 223, 239, 244, 245, 248, 263, 273, 274, 288, 305, 308, 315, 317, 318, 355, 356, 358, 361, 362, 367, 402, 405, 418, 440, 442, 495, 498, 499, 500, 503, 504], "eager": 504, "earli": 355, "eas": 6, "easi": [2, 341], "easier": [1, 144, 504], "easiest": 499, "edg": [96, 239, 419, 498], "edit": [9, 396], "effect": [355, 498, 504], "effici": [6, 8, 169, 355, 405, 499, 504, 506], "eigenvalu": [197, 198], "eigenvector": 197, "einstein": [133, 134], "einsum": 134, "either": [9, 14, 66, 79, 80, 88, 90, 91, 96, 131, 132, 136, 167, 175, 176, 180, 191, 192, 193, 202, 214, 219, 221, 229, 231, 235, 241, 262, 266, 296, 313, 344, 345, 374, 375, 407, 419, 423, 424, 499, 503, 505], "elem": [1, 148], "elem_to_loc": [1, 2], "element": [0, 1, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 29, 71, 84, 88, 89, 90, 91, 94, 109, 110, 111, 112, 113, 114, 119, 131, 132, 136, 137, 138, 140, 142, 146, 166, 167, 170, 175, 176, 182, 183, 184, 185, 186, 187, 191, 192, 193, 210, 211, 212, 213, 214, 215, 216, 217, 221, 223, 229, 231, 233, 235, 240, 241, 244, 245, 261, 262, 263, 266, 267, 269, 280, 281, 282, 283, 289, 290, 296, 300, 302, 303, 306, 313, 315, 318, 347, 354, 355, 356, 361, 365, 367, 376, 398, 402, 405, 409, 428, 435, 436, 438, 439, 454, 455, 457, 460, 461, 462, 463, 498, 501], "elementwis": [1, 99, 100], "elif": 6, "ellipsi": 502, "elman": 402, "els": [0, 2, 6, 341, 382, 499, 504], "elsewher": [309, 502], "elu": [341, 458], "emb": [6, 358, 399, 410], "embed": [6, 322, 341, 399, 405, 410, 441], "empti": 252, "en0": 503, "en2": 499, "enabl": [3, 6, 9, 97, 135, 321, 486], "enclos": 500, "encod": [2, 150, 405, 410, 418, 442], "encount": [2, 501], "end": [121, 195, 225, 244, 267, 361, 367, 416, 434, 445, 452, 458, 464, 465, 487, 490, 500], "end_axi": [0, 50, 165], "end_encod": 2, "endif": 2, "endl": [4, 500], "endswith": 382, "enhanc": [6, 405, 504], "enough": [2, 504], "ensur": [0, 1, 2, 9, 148, 324, 448, 499, 500], "ensure_row_contigu": [1, 148], "enter": 6, "entir": [16, 18, 27, 28, 218, 220, 222, 228, 242, 293, 297, 314, 355, 356], "entri": [0, 254, 312, 355, 356], "entropi": [7, 440, 442], "enumer": 341, "environ": [9, 122, 135, 499], "ep": [5, 147, 149, 346, 362, 366, 368, 401, 441, 443, 453, 473, 474, 475, 476, 477, 478, 479, 485], "epoch": 7, "epsilon": [346, 362, 366, 368, 401, 441, 443, 474, 476, 477, 478, 479, 485], "epsilon_1": 475, "epsilon_2": 475, "equal": [0, 1, 17, 29, 83, 146, 176, 182, 193, 235, 240, 255, 288, 321, 366, 370], "equal_nan": [0, 17, 83, 182], "equat": [133, 134, 204, 205], "equival": [0, 2, 32, 66, 80, 129, 132, 167, 170, 177, 300, 347, 357, 359, 363, 364, 365, 371, 372, 396, 398, 400, 403, 404, 406, 408, 411, 412, 413, 414, 415, 417, 499], "erf": [0, 138, 498], "erfinv": 0, "error": [0, 2, 9, 126, 137, 138, 279, 288, 359, 430, 431, 432, 448, 450, 498, 501, 503, 505], "error_norm": 5, "estim": 479, "eta": 480, "etc": [2, 244, 341, 419, 499], "ethernet": [499, 503], "eval": [2, 3, 5, 6, 7, 341, 471, 473, 498, 499, 500, 501, 504, 506], "eval_cpu": 2, "eval_fn": 7, "eval_gpu": 2, "evalu": [2, 6, 7, 8, 130, 139, 189, 278, 316, 341, 380, 391, 471, 473, 498, 500, 506], "even": [1, 2, 6, 97, 498, 499, 500, 504, 505], "evenli": [0, 208], "everi": [244, 326, 473, 491, 501, 503], "everyth": [6, 499], "everywher": 0, "exact": [431, 432], "exactli": [2, 6, 150, 384, 501], "exampl": [0, 3, 4, 5, 6, 7, 9, 19, 39, 115, 126, 143, 144, 145, 148, 151, 165, 180, 188, 190, 197, 198, 202, 203, 284, 285, 295, 300, 312, 324, 327, 328, 341, 343, 344, 345, 346, 366, 373, 374, 375, 382, 384, 391, 394, 419, 420, 421, 422, 423, 424, 425, 426, 427, 440, 442, 449, 468, 473, 482, 487, 488, 489, 490, 491, 495, 501, 502, 503, 504, 505, 506, 507], "exce": [321, 324], "exceed": 278, "except": [8, 115, 146, 158, 159, 161, 162, 163, 278, 332, 362, 384, 500, 502, 505], "exclud": [243, 301], "exclus": [0, 84, 91], "execut": [2, 4, 9, 85, 86, 87, 173, 190, 499, 505, 508], "execute_process": 4, "exist": [2, 3, 6, 382, 394, 499], "exp": [0, 1, 142, 148, 214, 218, 249, 286, 347, 357, 409, 428, 429, 446, 458, 459, 463, 498, 500, 508], "exp_elementwis": [1, 148], "expand_dim": 0, "expect": [6, 348, 349, 350, 351, 352, 353, 354, 355, 356, 410, 418, 443, 498, 499, 502], "expens": 418, "expensive_fun": 504, "experiment": [143, 145, 180, 505], "explain": 2, "explicit": [2, 482, 495, 505], "explicitli": [169, 341, 495, 503], "explor": 9, "expm1": 0, "exponenti": [0, 140, 142, 347, 357, 406, 428, 429, 458, 488], "exponential_decai": 473, "export": [8, 9, 143, 144, 180], "export_funct": 500, "ext_modul": 2, "extend": [2, 239], "extens": [8, 209, 226, 388, 507], "extern": 505, "extra": [1, 326, 327, 500], "extract": [0, 6, 46, 120, 121, 284, 341, 381, 471], "extras_requir": 2, "extrem": [502, 504], "ey": [0, 6, 199, 207], "f": [0, 2, 5, 7, 115, 202, 341, 367, 478, 498, 505], "f_jvp": 115, "f_t": 367, "f_vjp": 115, "f_vmap": 115, "face": 6, "factor": [2, 15, 177, 194, 195, 200, 201, 203, 419, 442, 488, 491], "fail": [498, 499, 503], "fall": [2, 115], "fals": [0, 1, 2, 6, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 76, 77, 81, 83, 97, 104, 111, 112, 113, 114, 126, 143, 145, 148, 182, 188, 194, 195, 202, 205, 206, 207, 209, 218, 220, 222, 223, 228, 242, 293, 297, 314, 318, 322, 325, 326, 327, 328, 332, 362, 366, 368, 370, 382, 384, 394, 397, 400, 405, 410, 418, 419, 440, 443, 475, 477, 478, 486, 500, 505], "famili": 6, "fan": [421, 422, 423, 424], "fan_in": [421, 422, 423, 424], "fan_out": [421, 422, 423, 424], "far": 473, "fast": [1, 8, 359, 432, 499, 508], "faster": [1, 2, 9, 132, 430, 440, 498, 499, 501], "featur": [1, 8, 101, 102, 103, 104, 105, 106, 107, 150, 346, 361, 362, 366, 367, 368, 370, 400, 401, 402, 405, 418, 419, 498, 499, 504], "feed": 6, "feed_forward": 6, "feedforward": [421, 422], "feel": 6, "fetch": 1, "few": [1, 2, 6, 7, 8, 9, 500, 504, 506], "fewer": 499, "ffn": 6, "ffn_norm": 6, "fft": 8, "fi": 499, "figur": 499, "file": [4, 6, 9, 143, 144, 145, 180, 209, 270, 271, 272, 273, 274, 384, 388, 499, 500, 501, 507], "file_or_weight": 384, "fill": [0, 168, 237, 309, 320, 420, 421, 422, 423, 424, 426, 427], "filter": [0, 108, 348, 349, 350, 351, 352, 353, 377, 381], "filter_and_map": 341, "filter_fn": [377, 381], "final": [2, 4, 5, 6, 7, 177, 487, 490, 499, 503], "find": [2, 4, 5, 9, 499], "find_packag": [2, 4], "finder": 9, "fine": [495, 500, 504], "finetun": 341, "finit": [0, 183, 232], "first": [0, 1, 2, 3, 4, 5, 6, 7, 9, 121, 124, 126, 165, 174, 188, 190, 191, 206, 215, 217, 219, 240, 257, 266, 298, 304, 307, 313, 325, 327, 328, 341, 344, 345, 362, 374, 375, 419, 441, 449, 475, 479, 482, 498, 499, 500, 501, 503, 505, 508], "first_lay": 504, "firt": 498, "fit": [244, 508], "five": 498, "fix": [2, 6, 9, 498, 504], "flag": [4, 9, 498, 505], "flat": [169, 170, 325, 329], "flat_param": 273, "flatten": [0, 29, 30, 111, 112, 113, 114, 202, 238, 240, 243, 263, 267, 287, 300, 301, 306, 325, 498], "flexibl": 8, "flexibli": 396, "flip": [0, 104, 108], "float": [0, 1, 2, 12, 15, 17, 19, 79, 147, 148, 149, 150, 151, 164, 167, 168, 177, 182, 188, 202, 232, 245, 247, 251, 253, 321, 324, 332, 346, 354, 355, 356, 362, 366, 368, 377, 389, 401, 405, 410, 416, 418, 419, 420, 421, 422, 423, 424, 426, 427, 441, 442, 443, 445, 449, 452, 453, 464, 465, 474, 475, 476, 477, 478, 479, 480, 485, 486, 487, 488, 490, 491], "float16": [1, 2, 12, 148, 177, 209, 332, 377, 504, 505], "float16_t": [1, 2], "float32": [0, 1, 2, 12, 19, 146, 148, 151, 177, 178, 188, 197, 198, 202, 203, 208, 236, 249, 251, 252, 253, 258, 259, 285, 309, 319, 332, 419, 420, 421, 422, 423, 424, 425, 426, 427, 440, 442, 449, 487, 488, 489, 490, 491, 498, 499, 500, 501, 502, 504, 505, 506, 507], "float64": [12, 188, 332, 505], "floor": [0, 1, 167], "floor_divid": 0, "flow": [0, 294, 504], "flush": 2, "fn": [180, 323, 326, 327, 328, 506], "follow": [1, 2, 4, 6, 7, 8, 9, 19, 108, 119, 151, 169, 202, 239, 244, 327, 341, 431, 432, 446, 474, 475, 476, 479, 480, 486, 495, 498, 499, 500, 501, 503, 508], "food": 6, "forc": [6, 7, 341, 499, 506], "forg": [9, 499], "formal": [119, 244], "format": [6, 144, 209, 270, 271, 272, 273, 274, 499, 505], "formul": [347, 357], "formula": 452, "forth": [419, 499], "forward": [1, 2, 313, 498, 503, 504], "found": [4, 381], "four": 346, "fourier": [152, 153, 154, 155, 156, 157, 161, 162, 163], "fourth": 500, "frac": [137, 244, 280, 346, 354, 355, 356, 362, 366, 368, 370, 401, 409, 421, 422, 423, 424, 441, 443, 445, 448, 459, 461, 462, 474, 476, 477, 478, 479, 485], "fraction": 19, "framework": 8, "free": 275, "freez": [341, 394, 471], "freq": 150, "frequenc": [150, 405, 410], "frequent": [498, 504], "friend": 6, "fro": 202, "frobeniu": 202, "from": [0, 1, 2, 4, 6, 7, 8, 84, 118, 119, 121, 124, 125, 128, 129, 130, 148, 159, 160, 162, 163, 168, 169, 173, 177, 180, 202, 209, 219, 223, 237, 244, 246, 247, 248, 249, 250, 251, 255, 258, 273, 275, 284, 291, 294, 296, 300, 301, 306, 307, 318, 320, 325, 326, 327, 328, 329, 341, 370, 382, 384, 397, 421, 422, 423, 424, 426, 427, 443, 452, 468, 473, 497, 498, 499, 500, 501, 504, 505, 506, 507, 508], "from_embed": 399, "from_linear": 400, "front": [2, 500], "frozen": [341, 382, 392, 394, 400, 471], "fuction": 132, "full": [0, 1, 2, 7, 66, 80, 108, 148, 200, 286, 395, 396, 443, 498, 499, 500, 504], "full_turn": 410, "fulli": [2, 8, 503, 505, 508], "fun": [97, 143, 145, 174, 189, 313, 316, 317, 498, 500, 502, 504, 508], "fun1": 504, "func": 402, "function": [0, 1, 2, 3, 5, 6, 7, 8, 17, 19, 84, 97, 115, 132, 137, 138, 143, 145, 148, 174, 180, 182, 189, 194, 195, 197, 198, 199, 202, 203, 206, 207, 219, 279, 280, 313, 316, 317, 323, 324, 326, 327, 328, 341, 347, 357, 359, 360, 363, 364, 365, 371, 372, 376, 378, 382, 389, 394, 398, 402, 403, 404, 406, 407, 408, 409, 411, 412, 413, 414, 415, 416, 417, 418, 430, 431, 432, 433, 434, 435, 436, 438, 439, 440, 454, 459, 461, 462, 463, 464, 465, 466, 468, 473, 482, 495, 497, 499, 502, 504, 505, 507], "functionexport": 145, "functool": 498, "further": [2, 9, 501], "fuse": [1, 498], "fusibl": 498, "futur": [6, 143, 145, 180, 400, 502, 504], "fx": 115, "g": [3, 9, 115, 148, 202, 244, 367, 467, 485, 486, 500, 504, 509], "g_t": [367, 474, 476, 477, 478, 479, 480, 485, 486], "gain": [421, 422, 423, 424], "gamma": [346, 362, 366, 368, 401, 421, 422, 423, 424], "gap": 1, "gate": [360, 361, 433], "gather": [0, 124, 169, 170], "gather_mm": [0, 170], "gather_qmm": 0, "gaurante": 315, "gaussian": [5, 359, 430, 431, 432, 443], "gaussian_nll_loss": 341, "gc_func": 418, "gelu": [341, 431, 432, 498], "gelu_approx": [341, 359, 430], "gelu_fast_approx": [341, 359, 430], "geluapprox": 359, "gelufast": 359, "gener": [0, 1, 3, 5, 12, 19, 104, 146, 148, 159, 160, 208, 223, 247, 252, 253, 254, 255, 258, 259, 418, 495, 498, 502, 504, 509], "general_": 2, "generate_stub": 9, "geq": [416, 465], "get": [2, 5, 7, 9, 102, 103, 104, 106, 107, 116, 117, 164, 171, 172, 173, 224, 250, 341, 498, 500, 501, 504, 508], "get_cache_memori": 95, "get_command_encod": 2, "get_kernel": 2, "gguf": [9, 209, 271, 507], "gh": 1, "gii": 1, "git": 9, "github": [5, 7, 9, 498], "give": [2, 6, 7, 29, 498], "given": [0, 2, 9, 16, 18, 29, 39, 84, 93, 96, 98, 111, 112, 113, 114, 119, 121, 134, 139, 141, 151, 152, 153, 154, 155, 156, 157, 161, 162, 163, 168, 169, 200, 202, 218, 220, 222, 228, 232, 234, 242, 252, 254, 255, 267, 268, 275, 277, 286, 288, 293, 297, 299, 305, 306, 307, 309, 310, 311, 314, 330, 354, 381, 397, 441, 443, 449], "gix": 1, "gix_mult": 1, "giy_mult": 1, "global": [122, 124, 125, 126, 128, 129, 130, 135, 256, 321, 324, 495, 498], "glorot": [421, 422], "glorot_norm": 341, "glorot_uniform": 341, "glu": [6, 341], "gm": 1, "gn": 1, "go": [2, 6, 499, 501], "golub": 202, "good": [2, 9, 473, 498, 499, 503, 508], "goroshin": 355, "gower": 6, "gpu": [1, 3, 8, 9, 224, 332, 502, 508], "gputrac": [3, 226], "grad": [2, 5, 7, 115, 313, 324, 473, 481, 498, 499, 500, 501, 502, 504, 506], "grad_fn": [5, 498, 501], "gradient": [0, 5, 7, 115, 174, 294, 313, 321, 323, 324, 341, 382, 395, 400, 418, 448, 471, 473, 474, 475, 477, 478, 479, 480, 481, 484, 486, 498, 499, 501, 502, 504, 505, 506], "grain": 495, "graph": [2, 6, 7, 8, 144, 278, 500, 501], "great": 3, "greater": [0, 6, 29, 142, 176, 240, 324, 416, 465], "greater_equ": 0, "grep": 9, "grid": [2, 148, 223], "grid_dim": 2, "grid_grad": 1, "grid_idx": 1, "grid_sampl": 1, "grid_sample_grad": 1, "grid_sample_ref": 1, "grid_sample_vjp": 1, "grid_shap": 1, "grid_siz": 1, "ground": [5, 6, 442, 452], "group": [0, 1, 101, 102, 103, 104, 105, 106, 107, 119, 124, 125, 126, 128, 129, 130, 151, 170, 244, 245, 315, 321, 322, 348, 349, 362, 399, 400, 499], "group_dim": 2, "group_siz": [0, 119, 170, 244, 245, 322, 399, 400], "groupnorm": 341, "grow": 504, "gru": 341, "guid": [2, 4, 8, 499, 500], "guidelin": 278, "gw": 1, "h": [1, 2, 4, 101, 102, 103, 105, 106, 107, 202, 346, 349, 350, 352, 353, 355, 356, 361, 367, 402, 501, 504], "h_": [361, 367, 402], "h_in": 1, "h_stride": 1, "h_t": [361, 367, 402], "ha": [2, 3, 6, 7, 8, 9, 79, 97, 121, 130, 158, 159, 161, 162, 163, 172, 174, 194, 195, 197, 198, 199, 206, 207, 223, 248, 346, 361, 367, 370, 402, 471, 473, 498, 499, 500, 502, 503, 504, 506, 508], "had": 6, "hadamard": [0, 177], "hadamard_transform": 0, "half": [19, 255, 259, 405, 504], "halv": [360, 433], "hand": [6, 501, 504], "handi": 501, "handl": [2, 341, 498], "happen": [2, 6, 147, 149, 418, 473, 498, 504], "happi": 6, "hard": 6, "hard_shrink": [341, 363], "hard_tanh": [341, 364], "hardcod": 498, "hardshrink": [341, 434], "hardswish": 341, "hardtanh": [341, 435], "hat": [119, 244], "have": [0, 1, 2, 6, 9, 17, 83, 85, 86, 87, 92, 115, 124, 151, 159, 160, 162, 163, 170, 182, 219, 226, 248, 315, 321, 325, 367, 397, 407, 480, 482, 497, 498, 499, 500, 502, 503, 504, 508], "haven": 6, "hazan": 476, "he": [6, 423, 424], "he_norm": 341, "he_uniform": 341, "head": [151, 397, 418], "header": [2, 148], "heart": 6, "heavi": 6, "height": [344, 345, 346, 349, 350, 352, 353, 355, 356, 374, 375], "hello": [325, 329], "help": [2, 6, 498, 508], "helper": [6, 148, 321, 498, 499, 503], "henc": [0, 2, 244, 498], "hendryck": 432, "here": [2, 6, 473, 498, 500, 501, 504, 507, 508], "hermitian": [197, 198], "hf": 367, "hg": 367, "hh": 402, "hi": [6, 367], "hidden": [361, 367, 402, 418], "hidden_dim": [7, 471, 473], "hidden_s": [361, 367, 402], "hierarchi": 332, "high": [255, 259, 341, 358, 427, 468], "high_pad_s": 0, "higher": [2, 181, 279, 449, 499, 501], "highli": 9, "him": 6, "hing": 444, "hinge_loss": 341, "hinton": 485, "hit": 2, "hn": 361, "ho": 367, "hold": [2, 6, 11, 12, 202, 498], "homebrew": 499, "hopkin": 202, "host": 2, "host1": 499, "host2": 499, "host3": 499, "host4": 499, "host_nam": 1, "hostfil": [499, 503], "hostnam": [499, 503], "hostname1": [499, 503], "hostname2": [499, 503], "hostname3": 499, "hostname4": 499, "hot": 442, "hour": 6, "how": [2, 4, 6, 7, 341, 343, 344, 345, 348, 349, 350, 351, 352, 353, 358, 373, 374, 375, 399, 419, 481, 498, 502, 508], "howev": [2, 115, 341, 359, 362, 482, 495, 498, 499, 504, 505], "hr": 361, "http": [362, 366, 368, 376, 401, 432, 454], "huber": 445, "huber_loss": 341, "human": [423, 424], "hundr": 9, "hurri": 6, "hutter": 478, "hyperbol": [0, 21, 23, 26, 110, 283, 303, 417, 466], "hz": 361, "i": [0, 1, 2, 3, 4, 6, 7, 8, 9, 17, 19, 29, 38, 79, 84, 96, 102, 103, 104, 106, 107, 108, 111, 112, 113, 114, 115, 120, 121, 124, 125, 127, 128, 129, 130, 132, 139, 143, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 167, 168, 169, 170, 177, 180, 182, 183, 188, 189, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 209, 214, 218, 219, 223, 225, 239, 240, 243, 244, 245, 252, 253, 254, 265, 267, 270, 271, 272, 278, 279, 280, 286, 288, 293, 294, 299, 300, 301, 304, 307, 308, 312, 313, 314, 315, 316, 317, 318, 321, 322, 324, 325, 326, 327, 328, 332, 334, 341, 343, 344, 345, 346, 348, 349, 350, 351, 352, 353, 354, 355, 356, 359, 361, 362, 366, 367, 368, 370, 373, 374, 375, 381, 382, 388, 390, 391, 393, 394, 396, 397, 398, 400, 401, 402, 405, 410, 416, 418, 419, 423, 424, 430, 432, 440, 441, 443, 448, 449, 452, 453, 455, 460, 465, 471, 473, 475, 477, 478, 480, 481, 482, 487, 489, 490, 495, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509], "i386": 9, "i_n": 1, "i_nw": 1, "i_s": 1, "i_sw": 1, "i_t": 367, "iclr": [477, 478, 479], "id": [7, 9], "idea": [501, 504], "idempot": [382, 394], "ident": [0, 115, 130, 146, 294, 341, 391, 499], "identifi": [2, 325, 497], "idim": 7, "idiom": [7, 498], "idx": [39, 502], "ie": 394, "ieee": 332, "ifac": 499, "ignor": [6, 39, 96, 97, 139, 475, 503], "ih": 402, "ii": 1, "ij": 223, "imag": [0, 349, 350, 352, 353, 355, 356, 419], "imagenet": [423, 424], "imaginari": 179, "immedi": [6, 377], "implement": [0, 1, 5, 7, 150, 151, 358, 381, 397, 405, 407, 410, 416, 418, 419, 465, 474, 475, 476, 479, 480, 481, 493, 498, 501], "impli": 315, "implicit": [495, 498, 501], "implicitli": 504, "import": [2, 3, 5, 6, 7, 9, 115, 126, 177, 180, 202, 273, 313, 325, 326, 327, 328, 329, 341, 343, 344, 345, 346, 366, 373, 374, 375, 384, 419, 440, 442, 449, 468, 471, 473, 498, 499, 501, 502, 504, 505, 506], "import_funct": 500, "imported_ab": 500, "imported_fun": 500, "imported_funct": 500, "improv": [1, 2, 3, 6, 440, 474, 475, 476, 477, 478, 479, 485, 498, 499], "in_ax": [317, 501], "in_channel": [348, 349, 350, 351, 352, 353], "in_dim": [341, 471], "in_proj": 471, "includ": [1, 2, 4, 111, 112, 113, 114, 144, 148, 171, 172, 278, 368, 378, 390, 400, 443, 473, 498, 500, 501, 502, 506, 507, 509], "include_dir": 2, "inclus": [0, 42, 43, 44, 45, 111, 112, 113, 114, 165], "incom": 2, "inconveni": 498, "incorpor": 505, "incorrect": 505, "increas": [279, 503], "increment": 19, "incur": [6, 9], "independ": [123, 355, 356], "index": [0, 1, 2, 8, 10, 29, 39, 141, 146, 174, 223, 240, 284, 285, 300, 301, 313], "indic": [0, 2, 17, 27, 28, 29, 30, 39, 169, 170, 174, 182, 183, 184, 185, 186, 187, 188, 200, 206, 243, 284, 285, 288, 300, 301, 313, 391, 393, 442, 449, 489, 502], "indices_or_sect": [72, 288], "indirectli": 505, "individu": [341, 355, 356], "ineffici": [502, 504], "inexact": [12, 188], "inf": [202, 232, 397], "infer": [8, 168, 209, 307, 312, 499, 500], "infin": [0, 184, 186, 187, 232, 373, 374, 375, 479], "infinit": [17, 182, 183], "info": [6, 9], "inform": [3, 4, 6, 7, 9, 134, 164, 224, 271, 272, 332, 341, 346, 359, 397, 499, 500, 501, 508], "inherit": [7, 497], "inifn": 184, "init": [341, 398, 468, 473, 487, 488, 490, 491, 499], "init_fn": [420, 421, 422, 423, 424, 425, 426, 427, 468], "init_valu": 1, "initi": [1, 3, 5, 6, 126, 328, 341, 346, 362, 366, 368, 370, 398, 401, 420, 421, 422, 423, 424, 425, 426, 427, 471, 482, 487, 488, 490, 491, 498, 499, 500, 504], "initializer_list": 0, "inject": 0, "inlin": 0, "inner": [0, 498], "inorm": 366, "inp": [1, 148], "inp_ndim": 1, "inp_shap": 1, "inp_strid": 1, "inplac": [2, 9], "input": [0, 1, 2, 5, 6, 13, 14, 15, 16, 17, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 118, 120, 121, 124, 125, 130, 131, 132, 133, 134, 136, 137, 138, 140, 141, 142, 143, 144, 145, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 169, 170, 174, 175, 176, 177, 179, 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, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 228, 229, 230, 231, 232, 233, 235, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 254, 257, 260, 261, 262, 263, 265, 266, 267, 268, 269, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 296, 297, 298, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 313, 314, 315, 317, 318, 320, 343, 344, 345, 346, 348, 349, 350, 351, 352, 353, 355, 356, 358, 360, 361, 362, 366, 367, 368, 370, 373, 374, 375, 397, 400, 401, 402, 405, 416, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 433, 440, 441, 443, 444, 445, 446, 448, 449, 451, 453, 465, 468, 498, 500, 501, 502, 503, 506, 507], "input_dil": [0, 104], "input_dim": [7, 341, 370, 400], "input_nam": [1, 148], "input_s": [361, 367, 402], "inputs1": 449, "inputs2": 449, "insert": [121, 141, 508], "insid": [498, 500], "inspect": [3, 498, 506], "inspir": 8, "instabl": 453, "instal": [2, 4, 503], "instanc": [6, 39, 115, 244, 329, 341, 366, 377, 378, 379, 382, 384, 385, 386, 391, 394, 395, 396, 407, 471, 499, 503, 505], "instancenorm": 341, "instanti": [1, 2, 7, 504], "instantiate_kernel": 2, "instead": [2, 9, 115, 341, 396, 410, 501, 504], "instruct": [4, 500], "int": [0, 1, 2, 4, 6, 7, 10, 16, 18, 19, 27, 28, 29, 30, 34, 35, 36, 37, 42, 43, 44, 45, 46, 47, 50, 57, 58, 59, 60, 61, 64, 67, 69, 72, 75, 76, 77, 78, 79, 81, 84, 92, 93, 98, 101, 102, 103, 104, 105, 106, 107, 111, 112, 113, 114, 119, 120, 121, 128, 129, 130, 134, 141, 146, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 168, 170, 171, 172, 173, 174, 178, 188, 196, 202, 208, 218, 220, 222, 224, 228, 230, 236, 239, 240, 242, 243, 244, 245, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 263, 265, 267, 268, 275, 278, 279, 284, 285, 286, 287, 288, 291, 292, 293, 297, 298, 300, 301, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 317, 319, 321, 322, 341, 343, 344, 345, 346, 348, 349, 350, 351, 352, 353, 358, 360, 361, 362, 366, 367, 368, 370, 373, 374, 375, 397, 399, 400, 401, 402, 405, 410, 418, 433, 441, 442, 446, 451, 453, 471, 487, 489, 490, 491], "int16": 332, "int32": [0, 1, 12, 19, 39, 165, 188, 190, 202, 255, 284, 312, 332, 419, 502, 506], "int64": [12, 332], "int64_t": 2, "int8": [12, 332], "int_0": 137, "integ": [0, 12, 167, 169, 170, 188, 202, 224, 239, 244, 245, 247, 254, 255, 288, 300, 304, 317, 332, 358, 389, 489, 502], "integr": [19, 300, 504], "intend": [0, 498], "interact": 418, "interest": 508, "interfac": [2, 499, 503], "intermedi": 505, "intern": 346, "interpol": 419, "interpret": 4, "interv": [19, 208, 255, 259], "introduc": [0, 267], "intuit": 341, "invalid": [0, 84], "invers": [0, 20, 21, 22, 23, 24, 25, 26, 89, 138, 155, 156, 157, 158, 159, 160, 195, 199, 207], "invert": 0, "involv": [473, 498], "iogpu": 279, "iostream": 4, "ip": [499, 503], "ip1": [499, 503], "ip2": [499, 503], "ip3": 499, "ip4": 499, "is_avail": 126, "is_equival": 2, "is_leaf": [325, 326, 327, 328], "is_leaf_fn": 381, "isclos": 0, "isfinit": 0, "ish": 6, "ishmael": 6, "isinf": 0, "isnan": 0, "isneginf": 0, "isposinf": 0, "issu": [499, 501, 505], "issubdtyp": [2, 12, 332], "item": [0, 2, 5, 6, 7, 326, 473, 500, 504, 505, 506], "iter": [5, 7, 206, 326, 327, 495, 498, 504], "iterm": 9, "itertool": [6, 326], "its": [0, 1, 2, 9, 151, 195, 219, 240, 257, 309, 323, 329, 341, 400, 473, 477, 478, 479, 499, 504, 505, 508], "itself": [2, 322, 482], "ix": 1, "ix_n": 1, "ix_nw": 1, "ix_s": 1, "ix_sw": 1, "iy_n": 1, "iy_nw": 1, "iy_s": 1, "iy_sw": 1, "j": [6, 9, 202, 355, 476, 477, 479], "j8": 2, "jacobian": [2, 189, 316, 506], "jain": 355, "jax": [8, 495], "jit": 148, "jmlr": 476, "jnp": 505, "john": 202, "join": 489, "join_schedul": 473, "jointli": 252, "json": [499, 503], "just": [2, 4, 7, 368, 498, 500, 502], "jvp": [2, 115, 506], "k": [0, 6, 46, 92, 120, 146, 151, 169, 177, 306, 309, 310, 311, 370, 382], "kaim": 424, "keep": [2, 16, 18, 27, 28, 218, 220, 222, 228, 242, 293, 297, 314, 341, 381, 501, 504], "keepdim": [0, 16, 18, 27, 28, 34, 35, 36, 37, 57, 58, 59, 60, 64, 76, 77, 81, 202, 218, 220, 222, 228, 242, 286, 293, 297, 314], "kei": [1, 3, 6, 143, 151, 180, 224, 247, 248, 249, 251, 252, 253, 254, 255, 257, 258, 259, 325, 326, 381, 382, 394, 397, 482, 495, 497, 500, 501], "kept": 279, "kernel": [2, 8, 9, 101, 102, 103, 104, 105, 106, 107, 148, 343, 373, 498, 502], "kernel_dil": [0, 104], "kernel_s": [343, 344, 345, 348, 349, 350, 351, 352, 353, 373, 374, 375], "key_cach": 6, "key_input_dim": 397, "key_proj": 6, "keyword": [143, 174, 273, 274, 313, 326, 341, 495, 500, 507, 509], "kind": 6, "kingma": [477, 479], "kl_div_loss": 341, "kname": 2, "know": [2, 6], "known": [408, 460], "kron": 0, "kroneck": [0, 190], "kth": [0, 29, 240], "kullback": 446, "kw_onli": 2, "kwarg": [11, 123, 143, 144, 180, 273, 274, 330, 500, 509], "l": [6, 7, 194, 195, 197, 198, 200, 341, 346, 348, 351, 361, 367, 402, 452], "l1": [313, 445, 447, 448, 452], "l1_loss": 341, "l2": [445, 448, 486], "l2_loss": 341, "l_": 445, "la": 202, "label": [3, 5, 442, 449], "label_smooth": 442, "lack": 502, "lambd": [363, 414, 434, 464], "lambda": [326, 327, 328, 341, 363, 377, 382, 389, 414, 434, 458, 464, 474, 475, 476, 477, 478, 479, 480, 485, 486, 498, 499, 500, 501], "languag": [1, 2, 4], "larg": [6, 341, 397, 448, 498, 500, 504], "larger": [1, 150, 279, 405, 480], "largest": [202, 232, 306], "lasso": 313, "last": [0, 1, 6, 30, 79, 147, 149, 154, 157, 159, 160, 162, 163, 165, 169, 170, 173, 181, 194, 195, 197, 198, 199, 203, 206, 207, 219, 248, 287, 304, 315, 348, 349, 350, 351, 352, 353, 355, 356, 362, 419, 505], "later": [3, 9, 473], "launch": [1, 2, 126, 499, 502], "layer": [8, 147, 322, 341, 343, 344, 345, 355, 356, 361, 362, 367, 368, 370, 373, 374, 375, 391, 396, 399, 400, 402, 407, 418, 467, 471, 500, 503], "layer_s": 7, "layernorm": 341, "layout": 1, "lazi": [8, 471, 506], "lazili": [6, 341], "lceil": 92, "ld": [361, 367, 402], "lead": [0, 19, 84, 498], "leaf": [97, 322, 325, 326, 327, 328, 381], "leaf_modul": 341, "leaki": [369, 437], "leaky_relu": 341, "leakyrelu": 341, "learn": [5, 7, 8, 346, 362, 366, 368, 398, 401, 473, 474, 475, 476, 477, 478, 479, 480, 485, 486], "learnabl": [348, 349, 350, 351, 352, 353, 407], "learning_r": [7, 473, 474, 475, 476, 477, 478, 479, 480, 482, 485, 486, 487, 488, 489, 490, 491, 498], "least": [6, 85, 86, 87, 96, 194, 195, 197, 198, 199, 203, 206, 207, 244], "leav": [2, 139, 326, 327, 328], "lectur": 485, "lecun": 355, "left": [0, 6, 150, 191, 202, 244, 267, 359, 405, 419, 431, 432, 443, 445, 453], "left_shift": 0, "leibler": 446, "len": [6, 154, 157, 160, 163, 177, 489], "length": [6, 291, 346, 348, 351, 361, 367, 402, 489], "leq": [445, 458], "less": [0, 1, 6, 29, 193, 240, 279, 321, 405, 452, 499], "less_equ": 0, "let": [1, 2, 5, 6, 195, 498, 500, 501, 504, 505], "level": [0, 169, 170, 423, 424], "lh": [361, 367, 402], "lhs_indic": [0, 169, 170], "lhs_mask": 92, "lib": 499, "libmlx": 9, "libmlx_ext": 2, "libmpi": 499, "librari": [2, 4, 9, 334, 341, 499, 500], "like": [2, 6, 8, 129, 143, 145, 180, 188, 237, 320, 356, 448, 482, 484, 498, 499, 500, 501, 503, 504, 505, 506, 508], "likelihood": [443, 451], "limit": [0, 2, 96, 275, 278, 279, 502], "linalg": 177, "line": [6, 499, 500, 503, 504, 505], "linear": [0, 2, 6, 7, 8, 204, 205, 322, 326, 341, 347, 357, 359, 360, 369, 384, 400, 402, 403, 404, 406, 408, 419, 428, 429, 430, 431, 432, 433, 437, 456, 457, 458, 460, 468, 471, 482, 489, 490, 498, 500], "linear1": 6, "linear2": 6, "linear3": 6, "linear_schedul": [473, 489], "linearli": 397, "link": [2, 4, 9], "linspac": 0, "lion": 473, "list": [1, 6, 11, 16, 18, 31, 72, 79, 84, 85, 86, 87, 93, 97, 98, 104, 134, 139, 148, 153, 154, 156, 157, 159, 160, 162, 163, 168, 174, 189, 202, 218, 220, 222, 223, 228, 236, 239, 242, 247, 248, 249, 251, 252, 253, 255, 258, 259, 271, 286, 288, 292, 293, 297, 304, 305, 308, 313, 314, 316, 319, 325, 328, 329, 341, 382, 384, 385, 386, 387, 392, 394, 395, 396, 471, 473, 477, 478, 479, 480, 489, 497, 498, 499, 500, 501, 503, 504], "listen": 499, "liter": [2, 239, 419, 423, 424, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453], "littl": 6, "liu": 6, "live": [8, 148, 508], "ll": [1, 5, 7, 445, 498, 501], "llama": 6, "llamaattent": 6, "llamaencoderlay": 6, "llm": 8, "load": [7, 8, 334, 384, 499], "load_weight": [341, 504], "loader": 7, "loader_path": 2, "loan": 202, "loc": [1, 251, 253], "local": [341, 355, 499], "localhost": [499, 503], "locat": [0, 2, 4, 84, 284, 285, 395, 396, 499, 508], "log": [0, 212, 214, 218, 371, 372, 438, 439, 440, 443, 446, 448, 451, 463], "log10": 0, "log1p": 0, "log2": 0, "log_cosh_loss": 341, "log_sigmoid": [341, 371], "log_softmax": [341, 372], "logaddexp": 0, "logarithm": [0, 210, 211, 212, 213], "logcosh": 448, "logic": [0, 2, 215, 216, 217, 499], "logical_and": 0, "logical_not": 0, "logical_or": 0, "logist": [0, 5, 280, 432, 460], "logit": [6, 248, 440, 442, 498], "logsigmoid": 341, "logsoftmax": 341, "logsumexp": 0, "long": 6, "longer": [6, 108, 501], "look": [2, 6, 499], "lookup": 358, "loop": [6, 7, 498, 499, 501, 504], "loshchilov": 478, "loss": [5, 7, 313, 341, 473, 498, 499, 501, 504], "loss_and_grad": 341, "loss_and_grad_fn": [7, 473, 498, 501], "loss_fn": [5, 7, 473, 498, 501], "loss_grad_fn": 499, "lot": 501, "low": [255, 259, 427, 468], "low_pad_s": 0, "lower": [194, 195, 197, 198, 205, 207, 244, 255, 258, 259, 309, 427], "lr": [5, 480], "lr_schedul": [487, 488, 489, 490, 491], "lstm": 341, "lto": 2, "lu": [6, 201], "luckili": 504, "lvalu": 313, "m": [0, 2, 4, 6, 9, 92, 146, 169, 177, 202, 309, 474, 498], "m1": [1, 6, 498, 501, 508], "m10": 332, "m7": 332, "m_": [477, 478, 479, 480], "m_t": [477, 478, 479, 480], "mac": 499, "machin": [6, 8, 9, 485, 499], "maco": [9, 279], "macosx": 9, "made": [6, 334], "mai": [2, 4, 143, 145, 180, 202, 322, 355, 499, 501, 502], "main": [4, 8, 121, 146, 148, 307, 326, 327, 341, 499], "maintain": [355, 356, 480], "major": [0, 2], "make": [1, 2, 3, 4, 6, 7, 9, 144, 145, 219, 234, 277, 341, 487, 488, 490, 491, 498, 504, 506, 508], "make_shar": 2, "malloc": 2, "man": 6, "manag": [8, 295, 495, 499, 500, 508], "mani": [2, 84, 288, 348, 349, 350, 351, 352, 353, 358, 399, 498, 499, 500, 504], "manual": [341, 499], "map": [2, 7, 39, 209, 326, 358, 377, 500], "map_fn": [377, 381], "map_torch_to_mlx": 6, "margin": [449, 453], "margin_ranking_loss": 341, "mask": [0, 6, 92, 151, 391, 397, 502], "mask_lh": [0, 92], "mask_n": 1, "mask_nw": 1, "mask_out": [0, 92], "mask_rh": [0, 92], "mask_s": 1, "mask_sw": 1, "matadata": 209, "match": [9, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 171, 384, 419, 442, 502, 505], "materi": [6, 8], "math": [6, 453, 498], "mathbf": 195, "mathcal": 370, "mathemat": 202, "mathrm": [137, 280, 366], "matmul": [0, 169, 508], "matric": [202, 203, 206], "matrix": [0, 5, 15, 46, 92, 119, 120, 146, 169, 170, 177, 178, 194, 195, 197, 198, 199, 200, 201, 202, 203, 206, 207, 219, 223, 244, 245, 252, 399, 400, 425, 468], "matter": [6, 341, 500], "matur": 499, "max": [0, 1, 2, 202, 221, 347, 373, 374, 375, 398, 428, 435, 436, 441, 443, 444, 449, 453, 455, 457, 475, 479, 498, 501, 508], "max_buffer_s": 224, "max_freq": 410, "max_i": 244, "max_norm": 324, "max_recommended_working_set_s": [224, 279], "max_val": 435, "maximum": [0, 7, 27, 39, 96, 111, 173, 278, 324, 341, 369, 373, 374, 375, 403, 410, 431, 432, 437, 456, 471, 504], "maxpool1d": 341, "maxpool2d": 341, "maxpool3d": 341, "maxtotalthreadsperthreadgroup": 2, "mca": [499, 503], "md": 202, "me": 6, "mean": [0, 1, 5, 6, 7, 149, 251, 252, 253, 313, 341, 346, 362, 382, 401, 426, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 498, 499, 501, 505], "meant": 115, "measur": 508, "mechan": 418, "medic": 356, "meet": 9, "member": [341, 387, 392], "memori": [0, 1, 2, 8, 84, 95, 171, 172, 173, 264, 275, 278, 279, 418, 471, 475, 498, 504, 505], "memory_order_relax": 1, "memory_s": [224, 279], "memoryview": [504, 505], "merg": 498, "meshgrid": 0, "metadata": [5, 209, 271, 272], "metal": [2, 4, 8, 148, 278], "metal_captur": 3, "metal_kernel": 1, "metal_path": 9, "metallib": [2, 9], "method": [2, 6, 10, 11, 31, 115, 123, 164, 322, 330, 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, 388, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 471, 474, 475, 476, 477, 478, 479, 480, 482, 485, 486, 493], "millisecond": [9, 498, 508], "min": [0, 2, 202, 229, 347, 398, 428, 435, 436, 455, 457], "min_freq": 410, "min_i": 244, "min_val": 435, "mind": 6, "mine": 6, "minibatch": 7, "minim": [499, 503], "minimum": [0, 28, 39, 96, 112, 410, 440, 441], "minsizerel": 9, "minu": 142, "minut": 6, "mish": 341, "mismatch": 500, "miss": [384, 500, 507], "mix": 502, "mkdir": [3, 9], "ml": 9, "mlp": [7, 341, 418, 473], "mlp_dim": [6, 418], "mlx": [1, 3, 5, 6, 7, 9, 334, 341, 468, 471, 473, 495, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508], "mlx_build_acceler": 4, "mlx_build_benchmark": 9, "mlx_build_cpu": 9, "mlx_build_exampl": 9, "mlx_build_gguf": 9, "mlx_build_met": [2, 4, 9], "mlx_build_metallib": 2, "mlx_build_python_bind": 9, "mlx_build_safetensor": 9, "mlx_build_test": 9, "mlx_cxx_flag": 4, "mlx_disable_compil": [122, 135, 498], "mlx_ext": 2, "mlx_ext_metallib": 2, "mlx_found": 4, "mlx_include_dir": [2, 4], "mlx_librari": 4, "mlx_metal_debug": [3, 9], "mlx_metal_jit": 9, "mlx_root": 4, "mlx_sample_extens": 2, "mlx_trace": 3, "mlxfn": [143, 145, 180, 500], "mnist": 7, "mode": [0, 1, 2, 108, 239, 380, 391, 393, 419, 423, 424], "model": [5, 7, 8, 273, 322, 323, 326, 327, 341, 377, 380, 382, 384, 388, 391, 393, 394, 395, 397, 418, 468, 471, 473, 481, 482, 484, 498, 499, 500, 504], "modest": 2, "modif": 505, "modifi": 505, "modul": [2, 4, 6, 7, 322, 323, 407, 418, 468, 484, 497, 498, 504], "moment": [6, 475, 479, 499], "momentum": [346, 480, 482, 486, 498], "monei": 6, "monitor": 503, "monoton": 454, "more": [1, 2, 3, 4, 7, 11, 79, 121, 143, 169, 194, 195, 197, 198, 199, 200, 206, 207, 219, 271, 272, 275, 278, 332, 341, 346, 355, 405, 410, 418, 419, 421, 422, 423, 424, 440, 495, 498, 499, 501, 502, 506, 508], "moreov": 503, "most": [2, 151, 248, 312, 341, 484, 498, 499, 501, 502, 504], "move": [0, 2, 230, 508], "moveaxi": 0, "mpi": [126, 334], "mpirun": [499, 503], "mse": 313, "mse_loss": 341, "mtl": 2, "mtl_capture_en": 3, "mtlcommandbuff": 2, "mu": 486, "much": [1, 2, 6, 343, 344, 345, 373, 374, 375, 498, 504], "multi": [8, 151, 348, 349, 350, 351, 352, 353, 500, 502, 505], "multidimension": 223, "multiheadattent": [6, 341], "multipl": [0, 1, 9, 15, 92, 145, 147, 149, 169, 170, 219, 231, 244, 245, 397, 410, 488, 489, 491, 498, 504, 507], "multipli": [0, 2, 39, 170, 244, 245, 354, 410, 419], "murtadha": 6, "must": [0, 1, 2, 3, 9, 92, 96, 143, 150, 151, 168, 170, 197, 198, 202, 247, 248, 252, 255, 258, 259, 318, 419, 505], "mx": [1, 2, 3, 4, 5, 6, 7, 39, 99, 100, 115, 126, 129, 143, 144, 145, 148, 165, 180, 188, 190, 197, 198, 200, 202, 203, 209, 254, 273, 284, 285, 312, 313, 324, 341, 343, 344, 345, 346, 357, 366, 369, 373, 374, 375, 377, 384, 388, 403, 419, 420, 421, 422, 423, 424, 425, 426, 427, 429, 437, 440, 441, 442, 446, 449, 456, 466, 468, 471, 473, 495, 498, 499, 500, 501, 502, 504, 505, 506, 507, 508, 509], "my": [6, 9], "my_devic": 509, "my_path": 273, "my_script": [499, 503], "myexp": [1, 148], "myexp_strid": 1, "mymlp": 471, "n": [0, 1, 2, 6, 31, 92, 101, 102, 103, 104, 105, 106, 107, 146, 151, 152, 154, 155, 157, 158, 161, 163, 177, 178, 252, 293, 309, 314, 346, 348, 349, 350, 351, 352, 353, 355, 356, 361, 367, 402, 419, 448, 453, 499, 503], "n_kv": 151, "n_q": 151, "n_t": 361, "naiv": [2, 501], "naive_add": 501, "name": [1, 2, 115, 144, 148, 170, 209, 244, 245, 271, 272, 273, 274, 341, 362, 381, 384, 386, 499, 502, 507], "named_modul": 341, "namespac": 4, "nan": [0, 17, 83, 182, 183, 185, 232], "nan_to_num": 0, "nanobind": 2, "nanobind_add_modul": 2, "nativ": [9, 499], "natur": [0, 210, 212, 504], "nb": 2, "nb_domain": 2, "nb_modul": 2, "nb_static": 2, "nbyte": 2, "nc": 346, "ndarrai": [31, 502, 504, 506], "ndhwc": [350, 353, 356], "ndim": [0, 1, 2, 165, 202, 206, 419], "ne": 1, "nearest": [1, 419], "necessari": 341, "necessarili": 306, "need": [1, 2, 4, 6, 7, 8, 9, 83, 244, 341, 395, 396, 410, 418, 495, 499, 501, 503, 504, 505, 506, 508], "neg": [0, 121, 165, 186, 232, 267, 307, 369, 373, 374, 375, 397, 443, 451, 453, 502], "negat": [0, 233], "negative_slop": [369, 437], "neginf": [0, 232], "neighbor": [419, 503], "neither": [174, 313], "nelem": 2, "nervou": 6, "nest": [79, 97, 328, 341, 471, 497, 501], "nesterov": 486, "network": [6, 8, 321, 346, 355, 358, 421, 422, 468, 471, 485, 499], "neural": [6, 8, 358, 421, 422, 454, 468, 471, 485], "never": [6, 504], "new": [0, 2, 7, 93, 121, 230, 234, 265, 292, 308, 315, 326, 327, 389, 397, 471, 473, 484, 489, 498, 500, 502, 504, 505], "new_tre": 327, "next": [2, 4, 6, 7, 275, 500], "nh": [361, 367, 402], "nhwc": [346, 349, 352], "nice": [501, 504], "nlc": [346, 348, 351], "nld": [361, 367, 402], "nlh": [361, 367, 402], "nll": [443, 451], "nll_loss": 341, "nn": [2, 6, 7, 273, 326, 341, 468, 471, 473, 482, 484, 498, 500, 504], "nobodi": 6, "node": [97, 139, 317, 327, 328, 499, 503], "nois": 5, "noisi": 5, "nomins": 2, "non": [0, 1, 2, 4, 9, 223, 392, 402, 454, 471], "none": [1, 2, 6, 10, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 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, 118, 119, 120, 121, 122, 124, 125, 128, 129, 130, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 146, 147, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 174, 175, 176, 177, 178, 179, 181, 182, 183, 184, 185, 186, 187, 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, 226, 227, 228, 229, 230, 231, 232, 233, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 272, 273, 276, 277, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 317, 318, 319, 320, 321, 322, 325, 326, 327, 328, 343, 344, 345, 359, 373, 374, 375, 377, 381, 382, 389, 394, 397, 402, 410, 418, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 475, 493, 500, 502], "nonlinear": [402, 498], "nonzero": 502, "noop": [394, 499], "nor": [2, 174, 313], "norm": [6, 149, 324, 362, 453, 479, 480], "norm1": 6, "norm2": 6, "norm_first": 418, "normal": [1, 2, 5, 6, 147, 148, 149, 197, 252, 258, 341, 343, 344, 345, 346, 362, 366, 368, 373, 374, 375, 401, 418, 421, 423, 505, 508], "not_equ": 0, "notabl": [6, 8], "notat": [119, 325, 386], "note": [0, 1, 2, 4, 6, 9, 17, 19, 84, 92, 97, 103, 106, 107, 115, 151, 159, 160, 170, 171, 182, 200, 202, 244, 248, 315, 322, 341, 401, 419, 473, 505, 507], "noth": [6, 115, 341, 504], "notic": [6, 500, 501, 507], "now": [1, 2, 6, 9, 400, 498, 505], "np": [1, 6, 7, 499, 505, 506], "npy": [209, 270, 507], "npz": [6, 209, 273, 274, 384, 388, 507], "nuc": 202, "nuclear": 202, "nuisanc": 499, "nullopt": 0, "num": [0, 6, 208, 257], "num_class": [7, 473], "num_decoder_lay": 418, "num_embed": [358, 399], "num_encoder_lay": 418, "num_epoch": [7, 473], "num_exampl": 5, "num_featur": [5, 346], "num_group": 362, "num_head": [6, 397, 418], "num_it": 5, "num_lay": [6, 7, 473], "num_param": 341, "num_paramet": 398, "num_sampl": 248, "num_split": 0, "number": [0, 2, 12, 19, 62, 71, 97, 102, 103, 104, 106, 107, 119, 144, 146, 151, 170, 174, 178, 189, 208, 232, 239, 244, 245, 248, 251, 253, 257, 259, 263, 267, 268, 304, 305, 309, 313, 316, 317, 321, 322, 341, 346, 348, 349, 350, 351, 352, 353, 355, 356, 362, 366, 397, 398, 418, 419, 421, 422, 423, 424, 487, 489, 490, 495, 498, 501, 503, 509], "number_of_el": 0, "numer": [6, 147, 149, 202, 214, 218, 286, 346, 362, 366, 368, 401, 440, 441, 443, 453, 474, 475, 476, 477, 478, 479, 485, 498, 504], "numpi": [2, 6, 7, 8, 14, 17, 19, 88, 90, 91, 93, 131, 132, 136, 175, 176, 182, 191, 192, 193, 214, 219, 221, 229, 231, 235, 241, 262, 266, 296, 504, 506, 507], "nw": 1, "nwhc": 355, "o": [2, 9, 151, 367], "o_t": 367, "obj": 271, "object": [3, 11, 31, 51, 79, 97, 144, 145, 148, 188, 273, 317, 325, 326, 327, 328, 332, 355, 418, 497, 503], "observ": 6, "occupi": [119, 170, 244, 245], "occur": 505, "odim": 7, "odot": [361, 367], "off": [6, 9, 504], "offer": 448, "offset": [0, 1, 2, 6, 47, 84, 121, 147, 150, 307], "often": 356, "ok": [384, 498, 500, 501], "okai": [498, 504], "old": 6, "older": [143, 145, 180], "omit": [479, 499], "onc": [2, 9, 498, 500], "one": [0, 2, 4, 6, 9, 39, 79, 85, 96, 102, 103, 104, 106, 107, 126, 141, 143, 146, 147, 149, 150, 202, 212, 219, 245, 248, 291, 296, 312, 321, 332, 394, 419, 442, 499, 500, 503, 508], "ones": [0, 2, 6, 237, 273, 285, 309, 395, 396, 473, 499, 502], "ones_lik": 0, "onli": [1, 2, 6, 8, 9, 83, 92, 102, 103, 104, 106, 107, 115, 197, 198, 202, 206, 244, 252, 279, 315, 332, 341, 381, 382, 384, 389, 391, 394, 395, 396, 471, 498, 499, 500, 501, 503, 507, 508], "onlin": 476, "op": [1, 2, 238, 315, 382, 504], "open": [3, 9, 19, 255, 259, 499], "openmpi": 499, "oper": [3, 6, 8, 10, 38, 85, 86, 87, 104, 151, 169, 170, 241, 243, 286, 294, 301, 330, 332, 341, 418, 480, 498, 499, 501, 502, 504, 505, 506, 508, 509], "operand": [133, 134, 169], "opportun": 498, "opt": [481, 499], "optim": [1, 3, 5, 7, 8, 395, 498, 499, 501, 504], "option": [0, 3, 6, 15, 16, 18, 19, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 84, 85, 86, 87, 92, 97, 98, 101, 102, 103, 104, 105, 106, 107, 108, 111, 112, 113, 114, 115, 119, 120, 121, 124, 125, 126, 128, 129, 130, 143, 145, 146, 147, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 168, 169, 170, 174, 178, 186, 187, 190, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 218, 220, 222, 223, 228, 232, 236, 239, 240, 242, 244, 245, 247, 248, 249, 251, 252, 253, 254, 255, 257, 258, 259, 263, 265, 267, 286, 287, 288, 291, 292, 293, 297, 299, 300, 304, 306, 307, 308, 309, 310, 311, 312, 313, 314, 317, 319, 321, 322, 325, 326, 327, 328, 343, 344, 345, 346, 348, 349, 350, 351, 352, 353, 361, 367, 370, 373, 374, 375, 377, 381, 382, 384, 389, 394, 397, 399, 400, 402, 405, 410, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 474, 475, 476, 477, 478, 479, 480, 482, 485, 486, 487, 495, 498, 500, 507, 509], "ord": 202, "order": [0, 1, 29, 84, 104, 134, 197, 198, 202, 240, 244, 306, 341, 362, 395, 407, 482, 498, 501, 503], "ordinari": 181, "org": [362, 366, 368, 376, 401, 432, 454], "origin": [6, 121, 324, 346, 390, 421, 422, 423, 424, 474, 475, 476, 479, 480, 500, 505], "orthonorm": 177, "ostream": 2, "ostringstream": 2, "other": [0, 2, 6, 8, 188, 202, 341, 383, 471, 480, 498, 499, 500, 502, 503, 504, 506], "other_input": 341, "otherwis": [19, 104, 126, 254, 322, 325, 326, 327, 328, 382, 384, 394, 416, 418, 419, 434, 440, 445, 452, 464, 465, 504, 505], "our": [1, 2, 6, 7, 407, 474, 475, 476, 479, 480, 499], "out": [0, 1, 2, 9, 92, 148, 180, 355, 356, 391, 498, 499, 500, 501, 502], "out_ax": [317, 501], "out_channel": [348, 349, 350, 351, 352, 353], "out_dim": [341, 471], "out_dtyp": 2, "out_idx": 2, "out_mask": 92, "out_proj": [6, 471], "out_ptr": 2, "out_shap": [1, 2], "outer": [0, 498, 504], "outlier": 448, "output": [0, 1, 2, 6, 9, 16, 17, 18, 19, 29, 84, 92, 93, 97, 99, 100, 111, 112, 113, 114, 115, 133, 144, 146, 147, 148, 149, 150, 151, 158, 161, 162, 163, 168, 169, 174, 177, 178, 182, 202, 208, 218, 220, 222, 223, 228, 232, 236, 237, 240, 242, 243, 247, 248, 249, 251, 252, 253, 255, 258, 259, 273, 274, 284, 285, 286, 291, 293, 297, 301, 307, 309, 313, 314, 315, 316, 317, 318, 319, 320, 346, 348, 349, 350, 351, 352, 353, 366, 370, 397, 400, 416, 418, 419, 421, 422, 423, 424, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 465, 468, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508], "output_dim": [7, 341, 370, 400], "output_directori": 2, "output_dtyp": [1, 148], "output_fil": 6, "output_nam": [1, 148], "output_shap": [1, 148], "output_strip_trailing_whitespac": 4, "output_vari": 4, "outsid": [148, 165], "over": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 101, 102, 103, 104, 105, 106, 107, 111, 112, 113, 114, 154, 157, 160, 163, 181, 202, 206, 208, 218, 220, 222, 228, 240, 242, 269, 286, 287, 293, 297, 304, 306, 314, 346, 348, 349, 350, 351, 352, 353, 362, 368, 401, 442, 487, 490, 499, 501, 503], "overal": 2, "overhead": [498, 504, 508], "overlap": 1, "overload": 19, "overrid": [2, 135], "overview": 3, "overwrit": 6, "own": [9, 499, 505], "owndata": 505, "p": [9, 200, 247, 341, 354, 355, 356, 453, 477, 479], "pack": [170, 244, 245], "packag": [2, 5, 7, 9, 334, 468, 499, 503], "package_data": 2, "pad": [0, 1, 101, 102, 103, 104, 105, 106, 107, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 343, 344, 345, 348, 349, 350, 351, 352, 353, 373, 374, 375], "pad_valu": 0, "pad_width": [0, 239], "padding_hi": 0, "padding_lo": 0, "page": [499, 506], "pain": 6, "pair": [0, 2, 239, 384, 405], "pairwis": 453, "pan": 6, "paper": [346, 410, 474, 475, 476, 479, 480], "parallel": [499, 508], "param": [313, 322, 341, 468, 500, 501], "paramet": [0, 1, 2, 5, 6, 7, 13, 14, 15, 16, 17, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 118, 119, 120, 121, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 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, 165, 166, 167, 168, 169, 170, 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, 226, 228, 229, 230, 231, 232, 233, 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, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 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, 328, 329, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 366, 367, 368, 369, 370, 373, 374, 375, 377, 378, 381, 382, 384, 389, 390, 391, 394, 395, 396, 397, 398, 399, 400, 401, 402, 405, 407, 410, 414, 416, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 433, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 465, 467, 468, 471, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 484, 485, 486, 487, 488, 489, 490, 491, 493, 498, 499, 500, 501, 504], "parameter_scal": 475, "parametr": [398, 455], "pars": [6, 144], "parse_arg": 6, "parser": 6, "part": [1, 2, 143, 145, 179, 180, 260, 501, 502], "parti": 499, "partial": [395, 396, 498, 504], "particip": [124, 125, 128, 129, 130], "particular": [244, 362], "particularli": 498, "partit": [0, 29], "pass": [1, 2, 6, 7, 66, 80, 238, 239, 313, 321, 323, 325, 326, 327, 341, 382, 394, 395, 396, 407, 498, 499, 500, 503, 504], "password": [499, 503], "path": [3, 4, 9, 134, 143, 144, 145, 180, 226, 273, 274, 322, 327, 384, 499, 503], "pattern": [341, 504], "peak": [173, 264], "penalti": 486, "pep": 505, "per": [6, 7, 119, 151, 170, 244, 245, 321, 322, 346, 362, 366, 368, 401, 493, 498, 499, 503, 504], "perceptron": [8, 500], "perf_count": 498, "perfectli": 504, "perform": [0, 1, 2, 3, 6, 8, 15, 92, 104, 111, 112, 113, 114, 130, 133, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 169, 170, 177, 197, 198, 219, 245, 268, 286, 300, 321, 341, 362, 418, 423, 424, 473, 498, 499, 502, 504, 508], "perhap": [2, 6], "perm": 7, "permtuat": 254, "permuat": 200, "permut": [0, 7], "persist": 9, "pg": 202, "phi": [359, 430], "physic": 499, "pi": [137, 359, 410, 431, 501], "pick": 2, "pip": [2, 4, 9], "pipelin": 2, "pivot": [200, 201], "pixel": 355, "place": [6, 39, 267, 268, 322, 499, 504, 505], "placehold": 498, "plai": [2, 6], "plain": 407, "plan": [2, 498], "platform": 9, "plot": 499, "plu": [0, 212], "png": 499, "point": [0, 2, 5, 6, 9, 84, 164, 167, 245, 332], "pool": [343, 344, 345, 373, 374, 375, 508], "popul": 2, "port": 503, "portion": 354, "posinf": [0, 232], "posit": [0, 6, 29, 121, 150, 165, 174, 187, 194, 195, 230, 232, 240, 252, 267, 307, 313, 326, 341, 348, 349, 350, 351, 352, 353, 397, 405, 410, 443, 453, 500], "possibl": [126, 288, 358, 399, 498, 499, 502, 508], "possibli": [6, 15, 92, 169, 219, 324], "postur": 6, "power": [0, 501, 505], "practic": [2, 498], "pre": [9, 151, 440], "preced": 362, "precis": [0, 6, 142, 151, 341, 359, 401, 440, 481, 498], "preclud": 341, "pred": [444, 448], "predic": [322, 389], "predict": [440, 443, 444, 445, 446, 447, 448, 450, 451, 452], "prefix": [317, 325, 327], "prelu": 341, "prepar": [2, 6, 499], "prepend": [3, 219], "preprint": [6, 474, 480], "preprocessor": 9, "present": 1, "preserv": [265, 501], "press": [6, 202], "pretti": [498, 504], "prevent": [294, 453, 505], "previou": [275, 278, 279], "primal": [1, 2, 115, 189, 316], "primit": 501, "print": [1, 2, 5, 6, 7, 9, 115, 190, 324, 325, 326, 327, 329, 341, 495, 498, 499, 500, 501, 502, 503, 504, 505, 506], "prior": [243, 300, 301], "priorit": 501, "privat": [2, 4], "prng": [247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 495], "prob": 440, "probabl": [9, 255, 354, 355, 356, 400, 440, 442, 446, 508], "problem": [5, 7, 341], "process": [6, 104, 108, 123, 124, 125, 126, 128, 129, 130, 321, 326, 327, 356, 358, 418, 497, 499, 503], "processor": 9, "prod": [0, 1], "produc": [0, 2, 9, 97, 397, 468, 500], "product": [0, 2, 15, 84, 113, 181, 189, 190, 196, 219, 238, 242, 304, 316, 397, 506], "profil": 3, "program": [4, 173], "programmat": 396, "project": [3, 4, 6, 397, 500], "project_source_dir": 2, "promot": [2, 151], "promote_typ": 2, "promoted_dtyp": 2, "prompt": 6, "propag": [501, 502], "properti": [32, 39, 48, 52, 62, 63, 69, 71, 390, 393, 483, 499, 501], "proportion": 324, "protocol": 505, "provid": [0, 2, 6, 84, 119, 143, 144, 174, 254, 267, 304, 313, 321, 326, 328, 334, 341, 377, 382, 384, 394, 395, 396, 399, 400, 418, 419, 467, 471, 499, 500, 507, 509], "pseudo": 495, "pth": 6, "public": [2, 341], "pun": 0, "pure": [1, 115, 341, 473], "purpos": [1, 202, 499], "purs": 6, "push": 2, "push_back": 2, "put": [0, 1, 7, 243, 498, 499], "put_along_axi": [0, 200], "py": [2, 6, 9, 499, 503], "pypi": 9, "python": [1, 3, 4, 6, 51, 69, 79, 139, 321, 325, 326, 327, 328, 329, 471, 481, 482, 484, 497, 499, 500, 501, 503, 505], "python_execut": 4, "python_requir": 2, "pytorch": [6, 8, 359, 362, 501], "pytorch_compat": 362, "q": [151, 203], "quantiz": [0, 119, 170, 209, 245, 399, 400], "quantized_matmul": 0, "quantizedembed": 341, "quantizedlinear": 341, "quarter": 6, "queri": [6, 151, 279, 397], "query_input_dim": 397, "query_proj": 6, "question": [6, 504], "queue": 3, "quick": [2, 8], "quit": [501, 505], "quotient": [0, 131, 132, 167], "r": [2, 6, 203, 313, 355, 361], "r_t": 361, "race": 508, "radian": [0, 118], "rag": 6, "rain": 6, "rais": [0, 6, 115, 202, 241, 288, 384, 500], "ram": [6, 278], "random": [1, 2, 3, 5, 6, 7, 8, 148, 343, 344, 345, 346, 366, 373, 374, 375, 384, 391, 498, 500, 501, 508, 509], "randomli": [5, 6, 254, 354, 355, 356], "rang": [0, 2, 3, 5, 6, 7, 9, 19, 165, 169, 208, 422, 424, 431, 432, 473, 487, 488, 489, 490, 491, 495, 498, 501, 504, 508], "rank": [0, 128, 129, 130, 449, 499, 503], "rate": [5, 473, 474, 475, 476, 477, 478, 479, 480, 485, 486], "rather": [2, 501, 508], "ratio": [0, 25], "rceil": 92, "re": [7, 9, 468], "reachabl": 499, "readabl": 3, "real": [0, 158, 159, 160, 161, 162, 163, 194, 195, 197, 198], "realli": 368, "reason": [1, 6, 502], "reboot": 9, "receiv": [128, 129, 322, 489, 499, 505], "reciproc": [0, 269], "reclaim": 275, "recommend": [9, 278, 480], "recompil": [97, 498], "reconstruct": 200, "record": [3, 173, 504], "recreat": [329, 473], "rectifi": [369, 403, 404, 423, 424, 437, 456, 457], "recurr": [361, 367, 402], "recurs": [144, 341, 381, 382, 387, 392, 394, 471], "recv": [129, 499], "reduc": [0, 1, 9, 16, 18, 27, 28, 125, 218, 220, 222, 228, 242, 293, 297, 314, 321, 328, 346, 418, 448], "reduct": [16, 18, 125, 218, 220, 228, 242, 328, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453], "redund": 501, "refer": [202, 366, 376, 390, 421, 422, 423, 424, 432, 454, 502], "reflect": [390, 498, 502, 505], "regard": 359, "regardless": [84, 151, 499], "regist": [2, 7], "register_librari": 2, "regress": [8, 448], "regular": [39, 355, 454, 478, 498, 500, 502], "regularli": 2, "reimplement": 2, "rel": [17, 182, 475, 498, 499], "relative_step": 475, "releas": 4, "relev": 2, "reli": [1, 2], "relu": [341, 398, 418, 455, 468], "relu6": 341, "remain": [0, 6, 279, 313, 327, 354, 355, 356, 499], "remaind": [0, 132], "remov": [0, 121, 219, 248, 291, 442], "rep": [0, 305], "repeat": [0, 305], "repeatedli": 5, "repetit": 263, "replac": [0, 6, 232, 395, 396, 418, 452], "replai": 3, "repli": 6, "repo": [5, 7, 9, 498], "report": [171, 278], "repres": [2, 6, 123, 126, 170, 449, 453, 505], "represent": [6, 201, 244, 315, 325, 329], "requir": [1, 2, 4, 6, 341, 499, 503, 504, 505], "requires_grad": 501, "rerun": [498, 504], "rescal": 324, "research": 8, "reset": 264, "reset_peak_memori": 173, "reshap": [0, 6, 202, 419, 498, 502], "resid": 279, "resolv": 2, "resourc": 2, "resource_limit": 224, "respect": [2, 5, 7, 115, 147, 149, 169, 170, 174, 244, 313, 326, 341, 346, 359, 362, 366, 368, 471, 499, 501, 503, 506], "respons": 2, "rest": [6, 150, 326, 327, 405, 503], "restart": 9, "restor": 267, "result": [0, 6, 15, 19, 39, 79, 84, 97, 144, 147, 149, 170, 190, 202, 219, 245, 252, 263, 278, 292, 326, 327, 328, 332, 410, 440, 498, 499, 501, 505], "resum": 6, "return": [0, 1, 2, 4, 5, 6, 7, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 51, 69, 79, 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, 118, 119, 120, 121, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 136, 137, 138, 140, 141, 142, 143, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 168, 169, 170, 172, 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, 228, 229, 230, 231, 232, 233, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 257, 258, 259, 260, 261, 262, 263, 265, 266, 268, 269, 275, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 322, 323, 324, 325, 326, 327, 328, 329, 341, 361, 367, 377, 378, 379, 381, 382, 383, 384, 385, 386, 387, 391, 392, 394, 395, 396, 402, 420, 421, 422, 423, 424, 425, 426, 427, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 468, 471, 481, 497, 498, 499, 500, 501, 502, 504, 505, 507, 508], "return_metadata": 209, "revers": [0, 2, 42, 43, 44, 45, 84, 111, 112, 113, 114, 308, 410], "rf": 9, "rfft": 158, "rfft2": 159, "rfftn": 160, "rho": 474, "rhs_indic": [0, 169, 170], "rhs_mask": 92, "right": [0, 1, 2, 9, 244, 266, 267, 359, 419, 431, 432, 443, 445, 453], "right_shift": 0, "ring": 126, "rm": [6, 9, 149, 475], "rmsnorm": [6, 341], "rmsprop": 473, "rnn": [341, 361], "robust": 448, "roform": [6, 405], "roll": 0, "root": [0, 6, 149, 269, 289, 401], "rope": [6, 341], "rosetta": 9, "rotari": [6, 150, 405], "rotat": [150, 405], "round": [0, 244], "row": [0, 1, 2, 84, 146, 148, 178, 244, 309], "rpath": 2, "rsqrt": 0, "rtol": [0, 17, 182], "rule": [2, 473], "run": [1, 2, 3, 4, 6, 7, 8, 9, 10, 148, 238, 330, 346, 377, 474, 475, 477, 478, 479, 498, 500, 503, 504, 508, 509], "runtim": [6, 126, 334, 498, 499], "runtime_error": 2, "safetensor": [9, 209, 272, 384, 388, 473, 504, 507], "sai": [2, 6, 468, 504], "said": 6, "sake": 501, "same": [0, 2, 6, 9, 17, 39, 83, 93, 97, 102, 103, 104, 106, 107, 108, 124, 147, 149, 158, 161, 162, 163, 170, 174, 182, 189, 239, 248, 267, 268, 285, 315, 316, 318, 321, 327, 341, 344, 345, 346, 354, 362, 366, 374, 375, 399, 420, 421, 422, 423, 424, 425, 426, 427, 442, 453, 471, 481, 495, 498, 499, 500, 502, 503, 508], "sampl": [2, 5, 6, 208, 247, 248, 249, 251, 252, 255, 258, 259, 421, 422, 423, 424, 426, 427, 443, 449, 453, 495, 498, 500], "sat": 6, "save": [3, 6, 8, 209, 226, 244, 271, 272, 273, 274, 388, 500, 504], "save_gguf": 507, "save_safetensor": [388, 473, 507], "save_weight": 341, "savez": [6, 388, 507], "savez_compress": 507, "saw": [6, 501], "scalar": [0, 2, 14, 15, 17, 31, 51, 79, 83, 88, 89, 90, 91, 92, 93, 96, 131, 132, 136, 167, 168, 174, 175, 176, 177, 182, 191, 192, 193, 208, 214, 215, 216, 217, 219, 221, 229, 231, 232, 235, 239, 241, 247, 255, 258, 259, 262, 266, 271, 296, 313, 315, 318, 323, 453, 500, 501, 504, 506], "scale": [0, 2, 6, 15, 119, 147, 149, 150, 151, 170, 177, 244, 245, 251, 253, 324, 355, 356, 368, 397, 405, 406, 410, 419, 458, 475], "scale_arr": 2, "scale_factor": 419, "scale_paramet": 475, "scatter": 0, "scatter_add": 0, "scatter_add_axi": 0, "scatter_max": 0, "scatter_min": 0, "scatter_prod": 0, "schedul": [2, 473, 487, 488, 489, 490, 491, 493, 508], "schema": [3, 503], "scipi": [177, 200], "scope": 341, "score": [6, 151, 449], "script": [499, 503], "sdk": 9, "se": 1, "second": [6, 9, 121, 188, 190, 191, 215, 217, 219, 266, 298, 307, 313, 344, 345, 374, 375, 441, 449, 475, 479, 498, 500, 501, 508], "second_layer_a": 504, "second_layer_b": 504, "secret": 6, "section": [1, 6, 9, 288, 453, 498, 499, 501], "see": [1, 2, 4, 6, 7, 9, 11, 12, 33, 34, 35, 36, 37, 40, 41, 42, 43, 44, 45, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 202, 271, 272, 275, 322, 332, 341, 346, 347, 355, 357, 359, 363, 364, 365, 371, 372, 380, 398, 399, 400, 403, 404, 405, 406, 408, 410, 411, 412, 413, 414, 415, 417, 419, 421, 422, 423, 424, 430, 431, 432, 458, 498, 499, 500, 501, 502, 503, 506, 508], "seed": 250, "seen": [499, 505], "select": [0, 3, 9, 197, 198, 306, 318, 377, 381, 389, 503], "self": [6, 7, 10, 31, 32, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 51, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 115, 164, 341, 454, 471], "selu": 341, "semant": [14, 88, 90, 91, 93, 131, 132, 136, 175, 176, 191, 192, 193, 214, 219, 221, 229, 231, 235, 241, 262, 266, 296, 508], "semi": [194, 195, 252], "send": 499, "sender": 499, "sennrich": 6, "sensit": 448, "sentencepiec": 6, "separ": [6, 66, 80, 362, 449], "sequenc": [6, 16, 18, 34, 35, 57, 58, 59, 60, 64, 72, 75, 76, 77, 81, 84, 93, 104, 128, 141, 148, 153, 154, 156, 157, 159, 160, 162, 163, 168, 174, 218, 220, 222, 228, 236, 242, 247, 248, 249, 251, 252, 253, 255, 258, 259, 265, 284, 285, 286, 288, 291, 293, 297, 304, 305, 308, 312, 313, 314, 319, 346, 348, 351, 361, 367, 402, 418, 495, 508], "sequenti": [341, 468], "seri": 9, "serial": 473, "set": [2, 4, 6, 7, 9, 97, 115, 122, 124, 125, 126, 128, 129, 130, 135, 147, 149, 150, 224, 275, 276, 277, 278, 279, 295, 321, 359, 368, 370, 380, 382, 389, 390, 391, 394, 395, 400, 405, 416, 441, 453, 465, 471, 473, 475, 477, 478, 482, 495, 500, 501, 504], "set_byt": 2, "set_compute_pipeline_st": 2, "set_data": 2, "set_dtyp": 341, "set_input_arrai": 2, "set_memory_limit": 275, "set_output_arrai": 2, "set_vector_byt": 2, "setup": [2, 4, 5, 7, 9, 498, 499, 500], "sever": [6, 9, 101, 102, 103, 104, 105, 106, 107, 273, 274, 321, 498, 499, 503, 507], "sgd": [5, 7, 473, 480, 482, 487, 488, 491, 498], "shade": [1, 2], "shall": 6, "shape": [0, 2, 3, 6, 7, 66, 83, 84, 92, 93, 97, 101, 102, 103, 104, 105, 106, 107, 121, 124, 128, 129, 143, 145, 148, 151, 152, 155, 158, 161, 162, 163, 168, 169, 177, 189, 199, 207, 219, 236, 237, 247, 248, 249, 251, 252, 253, 255, 258, 259, 265, 267, 285, 312, 315, 316, 318, 319, 320, 341, 343, 344, 345, 346, 348, 349, 350, 351, 352, 353, 355, 356, 361, 366, 367, 370, 373, 374, 375, 384, 402, 420, 421, 422, 423, 424, 425, 426, 427, 442, 453, 473, 498, 500, 501, 502, 506, 508], "shapeless": [0, 97, 143, 145], "share": [8, 119, 170, 244, 245, 315, 499], "shazeer": 6, "shift": [0, 191, 266, 267, 346], "shop": 6, "should": [1, 2, 4, 5, 6, 7, 9, 84, 95, 121, 124, 147, 148, 149, 151, 189, 226, 243, 244, 279, 301, 307, 313, 316, 321, 322, 325, 341, 348, 349, 350, 351, 352, 353, 355, 356, 391, 397, 407, 442, 444, 449, 471, 497, 498, 499, 500, 501, 504, 505, 509], "show": [9, 332, 498], "shown": 2, "shuffl": 7, "side": [0, 239, 343, 344, 345, 373, 374, 375, 498], "sigma": [359, 360, 361, 367, 409, 421, 422, 423, 424, 432, 433, 438, 459, 460], "sigmoid": [0, 6, 341, 371, 408, 432, 438, 440, 460], "sign": [0, 17, 182, 332, 480], "signal": [108, 419], "signatur": [1, 148], "signedinteg": [12, 188], "signific": 244, "significantli": 499, "silent": [161, 162, 163], "silicon": [2, 6, 8, 9, 508], "silu": 341, "simd": 1, "simd_sum": 1, "simdgroup": 1, "simdgroup_s": 1, "similar": [6, 170, 188, 326, 395, 396, 397, 441, 499, 505, 507], "similarli": [2, 9, 219, 501, 504], "simpl": [2, 6, 7, 341, 358, 467, 473, 498, 499, 500, 501, 503, 504], "simple_axpbi": 2, "simple_tim": 2, "simplest": [2, 341, 499], "simpli": [2, 6, 9, 357, 369, 403, 429, 437, 456, 466, 471, 498, 499, 501, 503], "simplic": 0, "simplifi": 499, "simultan": 1, "sin": [0, 115, 410, 500, 501, 506], "sinc": [1, 2, 6, 7, 170, 173, 471, 480, 489, 498, 500, 505, 508], "sine": [0, 22, 23, 282, 283, 500, 501], "sing": 202, "singer": 476, "singl": [7, 139, 189, 209, 223, 239, 316, 344, 345, 374, 375, 498, 500, 502, 507], "singleton": [0, 16, 18, 27, 28, 126, 218, 219, 220, 222, 228, 242, 293, 297, 314, 499], "singular": [202, 206], "sinh": 0, "sinusoid": 410, "sinusoidalpositionalencod": 341, "size": [0, 1, 2, 6, 7, 52, 69, 92, 102, 103, 106, 107, 119, 141, 147, 148, 149, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 168, 170, 172, 177, 178, 188, 196, 202, 244, 245, 248, 265, 278, 279, 284, 288, 291, 312, 315, 321, 322, 341, 343, 344, 345, 348, 349, 350, 351, 352, 353, 358, 366, 373, 374, 375, 399, 400, 419, 475, 499, 504, 505], "size_in_megabyt": 279, "size_t": [0, 2], "skip": [3, 84], "slice": [0, 285, 502], "slice_s": [0, 284], "slice_upd": 0, "slide": [343, 344, 345, 373, 374, 375], "slight": [6, 504], "slightli": [405, 508], "slope": 369, "slow": 498, "slowli": 6, "small": [6, 142, 147, 149, 321, 346, 362, 368, 401, 443, 448, 453, 498, 508], "smaller": [0, 9, 240, 321, 480, 498], "smallest": 202, "smile": 6, "smooth": [442, 452, 485], "smooth_l1_loss": 341, "sned": 130, "snippet": 499, "so": [1, 2, 6, 9, 174, 177, 313, 354, 419, 473, 498, 499, 504, 508], "socket": 499, "softmax": [0, 6, 151, 341, 372, 439, 442], "softmin": 341, "softplu": [341, 376, 454], "softshrink": 341, "softsign": 341, "solut": [204, 205], "solv": 341, "some": [0, 2, 5, 6, 7, 144, 382, 394, 473, 482, 498, 499, 500, 501, 503, 504], "someon": 6, "someth": [5, 6, 502], "sometim": 498, "sonoma": 9, "soon": 6, "sort": [0, 29, 30, 240, 306], "sourc": [0, 1, 2, 3, 4, 61, 128, 129, 148, 230, 308, 499], "space": [0, 2, 208, 440, 451], "spars": [0, 223], "spatial": [102, 103, 104, 106, 107, 343, 344, 345, 362, 373, 374, 375, 419], "speak": [6, 202], "specif": [1, 2, 9, 499, 501], "specifi": [0, 2, 19, 38, 102, 103, 104, 106, 107, 121, 159, 160, 168, 174, 196, 202, 208, 230, 236, 243, 248, 263, 298, 300, 301, 304, 307, 308, 313, 317, 319, 346, 416, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 465, 498, 499, 500, 501, 508], "speed": 1, "spent": 6, "split": [0, 360, 362, 433], "splittabl": 495, "sqrt": [0, 6, 137, 151, 177, 346, 359, 362, 366, 368, 370, 401, 410, 421, 422, 423, 424, 431, 474, 476, 477, 478, 485, 498], "squar": [0, 5, 6, 149, 178, 199, 207, 269, 289, 313, 326, 341, 401, 450, 452, 474, 475, 477, 478, 479, 501, 505], "squeez": [0, 419, 498], "src": [0, 128, 129], "ssh": [499, 503], "stabil": [147, 149, 346, 362, 366, 368, 401, 440, 441, 443, 474, 475, 476, 477, 478, 479, 485], "stabl": [214, 218, 286, 448], "stable_abi": 2, "stack": [0, 498], "standard": [0, 1, 4, 7, 51, 79, 219, 249, 253, 293, 418, 421, 423, 426, 499, 506], "starmap": [6, 326], "start": [0, 1, 2, 5, 6, 8, 9, 19, 150, 208, 226, 284, 285, 288, 328, 498, 500, 502, 503, 508], "start_axi": [0, 50, 165], "start_captur": 3, "start_indic": [284, 285], "state": [6, 7, 341, 361, 367, 402, 473, 482, 495, 498], "static": [9, 498], "static_cast": 2, "std": [0, 2, 4, 426, 500], "stderr": 503, "stdout": 503, "step": [0, 3, 4, 6, 7, 19, 321, 341, 361, 367, 402, 475, 482, 487, 489, 490, 491, 498, 499], "step_decai": 473, "step_siz": 491, "still": [6, 9, 202, 498, 504], "stochast": [476, 477, 479, 486, 504], "stood": 6, "stop": [0, 2, 6, 19, 208, 227, 294, 501, 502], "stop_captur": 3, "stop_gradi": [0, 501], "storag": 84, "store": 6, "str": [2, 108, 126, 133, 134, 143, 144, 145, 148, 151, 174, 180, 197, 198, 202, 209, 223, 224, 226, 270, 271, 272, 273, 274, 313, 322, 325, 329, 377, 378, 381, 382, 384, 386, 388, 394, 419, 423, 424, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453], "straight": 6, "strang": 6, "stream": [2, 8, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 117, 118, 119, 120, 121, 124, 125, 128, 129, 130, 131, 132, 133, 136, 137, 138, 140, 141, 142, 146, 147, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 166, 167, 168, 169, 170, 175, 176, 177, 178, 179, 181, 182, 183, 184, 185, 186, 187, 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, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 251, 252, 253, 254, 255, 257, 258, 259, 260, 261, 262, 263, 265, 266, 267, 268, 269, 277, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 314, 315, 318, 319, 320, 499, 508], "streamcontext": 295, "streamordevic": [0, 2], "street": 6, "strength": [480, 486], "strict": [126, 175, 192, 382, 384, 394], "strictli": [202, 279], "stride": [0, 2, 84, 101, 102, 103, 104, 105, 106, 107, 343, 344, 345, 348, 349, 350, 351, 352, 353, 373, 374, 375, 405, 502], "string": [0, 2, 134, 143, 148, 180, 224, 239, 500, 505, 507], "stronger": 503, "structur": [2, 321, 481, 501], "stub": 9, "style": [2, 14, 17, 88, 90, 91, 131, 132, 136, 175, 176, 182, 191, 192, 193, 214, 219, 221, 229, 231, 235, 241, 262, 266, 296], "su": 6, "sub": [0, 7, 121, 257, 284, 285, 307, 322], "subarrai": [121, 288], "subclass": 471, "subdivid": 1, "subdtyp": 188, "subgradi": 476, "sublinear": 475, "submodul": [6, 7, 341, 378, 382, 383, 394, 396], "subnetwork": 499, "suboptim": 500, "subscript": [133, 134], "subsect": 6, "subsequ": [126, 473, 499, 503], "subset": [341, 381], "substanti": 9, "subtl": 498, "subtract": [0, 39], "subtyp": [188, 332], "succe": 126, "successfulli": 499, "sudo": [9, 279, 499], "suggest": 499, "sum": [0, 2, 5, 14, 114, 125, 145, 181, 202, 218, 286, 304, 307, 341, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 499, 502, 505], "sum_": [202, 448], "sum_i": 439, "sum_j": [461, 462], "summat": [133, 134], "super": [6, 7, 341, 471], "superset": [326, 481], "support": [1, 2, 6, 8, 9, 17, 92, 103, 106, 107, 151, 165, 177, 182, 194, 195, 197, 198, 199, 203, 206, 207, 209, 219, 244, 252, 499, 501, 502, 505, 507], "suppos": [501, 508], "sure": [2, 3, 6, 9, 341, 498], "surpass": [423, 424], "surpris": 6, "sw": 1, "swap": [0, 108, 278, 298, 396], "swapax": [0, 115], "swiglu": 6, "swish": [408, 460], "switch": 9, "symbol": 480, "symmetr": [102, 103, 106, 107, 194, 195, 197, 198], "symmetri": [197, 198], "synchron": [2, 498], "syntax": [39, 502], "synthet": 5, "sysctl": 279, "system": [4, 6, 9, 171, 172, 204, 205, 224, 279], "t": [0, 1, 2, 4, 6, 9, 137, 148, 151, 170, 194, 195, 245, 313, 341, 361, 367, 402, 474, 475, 476, 477, 478, 479, 480, 485, 486, 498, 500, 501, 508], "t_kv": 151, "t_q": 151, "tabl": [1, 202, 332, 358], "take": [0, 2, 6, 7, 88, 89, 90, 91, 97, 143, 169, 174, 189, 221, 229, 237, 245, 301, 313, 316, 317, 320, 327, 328, 343, 344, 345, 373, 374, 375, 397, 440, 495, 499, 500, 501, 502, 503, 507, 508, 509], "take_along_axi": [0, 200, 502], "taken": [121, 300, 307], "talk": 499, "tan": 0, "tangent": [0, 2, 24, 25, 26, 115, 189, 302, 303, 417, 466], "tangent_i": 2, "tangent_x": 2, "tanh": [0, 341, 359, 361, 367, 376, 402, 431, 454], "target": [2, 313, 440, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 498], "target_include_directori": 2, "target_link_librari": [2, 4], "target_link_opt": 2, "target_sourc": 2, "task": 448, "tau": 486, "tcp": 499, "tediou": 499, "tell": [4, 6, 498, 505], "temp": 6, "templat": [0, 1, 2, 148], "ten": 504, "tend": 480, "tensor": [209, 304, 453, 505], "tensordot": 0, "term": [2, 443, 474, 475, 476, 477, 478, 479, 485], "termin": [9, 503], "test": [7, 9, 499, 503], "test_imag": 7, "test_label": 7, "text": [6, 359, 361, 367, 376, 402, 409, 416, 421, 422, 423, 424, 431, 434, 435, 436, 443, 444, 445, 448, 449, 452, 454, 455, 458, 459, 464, 465, 475, 480], "textrm": [244, 359, 360, 430, 433], "tf": 505, "tgp_size": 2, "th": [111, 112, 113, 114, 120, 146, 197, 489], "than": [1, 2, 6, 79, 108, 121, 132, 150, 169, 175, 176, 192, 193, 194, 195, 197, 198, 199, 200, 206, 207, 219, 275, 279, 324, 326, 405, 416, 419, 449, 452, 465, 475, 480, 498, 500, 501, 508], "thank": 504, "thei": [1, 2, 5, 6, 9, 17, 108, 170, 182, 407, 444, 471, 480, 497, 498, 499, 500, 504, 506, 507, 508], "them": [0, 2, 6, 124, 341, 382, 394, 499, 500, 503, 508], "themselv": [2, 498], "thi": [0, 1, 2, 4, 6, 7, 9, 16, 17, 18, 19, 27, 28, 29, 30, 84, 95, 115, 135, 143, 145, 148, 169, 170, 171, 177, 180, 182, 189, 194, 195, 197, 198, 199, 202, 203, 206, 207, 214, 218, 219, 220, 222, 228, 240, 242, 248, 277, 279, 286, 287, 288, 293, 297, 300, 306, 314, 321, 324, 327, 328, 341, 354, 355, 356, 360, 361, 367, 378, 379, 381, 382, 385, 386, 387, 392, 394, 395, 396, 397, 400, 402, 416, 421, 422, 423, 424, 431, 432, 433, 440, 448, 465, 471, 482, 497, 498, 499, 500, 501, 503, 504, 505, 507], "thin": 503, "thing": [2, 6], "third": [196, 345, 375, 499, 500], "thompson": 355, "those": [2, 6, 341], "though": [2, 6, 498, 500, 504, 505], "thousand": 504, "thread": [1, 2], "thread_index_in_simdgroup": 1, "thread_position_in_grid": [1, 2, 148], "threadgroup": [1, 2, 148], "threads_per_simdgroup": 1, "three": [6, 87, 345, 375, 419], "threefri": 495, "threshold": [416, 445, 452, 465], "through": [1, 2, 294, 418, 480, 498, 499, 500, 501, 505], "throw": [2, 97, 126], "thu": [6, 341], "thumb": 473, "tic": 498, "tieleman": 485, "tile": [0, 151], "time": [2, 6, 9, 278, 305, 341, 361, 367, 402, 498, 499, 501, 504, 508], "timeit": [498, 501], "titl": 2, "tmp": [1, 148], "to_quant": 322, "to_stream": 2, "toc": 498, "togeth": [0, 1, 2, 7, 244, 326, 327, 499], "tok_embed": 6, "token": [6, 358, 399], "told": 6, "toler": [0, 17, 182], "too": [188, 498, 504], "took": 6, "tool": 9, "top": [2, 306, 370, 419], "topk": 0, "torch": [6, 505], "torch_weight": 6, "total": [279, 501], "total_norm": 324, "tpi": 498, "tpng": 499, "trace": [0, 3, 145, 498], "trace_fil": 3, "tracer": 395, "track": [2, 341, 346], "track_running_stat": 346, "trade": 504, "tradit": [6, 150, 355, 356, 405], "train": [6, 7, 341, 346, 354, 355, 356, 380, 382, 394, 421, 422, 500], "train_imag": [7, 473], "train_label": [7, 473], "trainabl": [7, 323, 341, 471], "trainable_paramet": [341, 381, 482], "transfer": 503, "transform": [1, 6, 8, 115, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 177, 323, 341, 346, 362, 368, 370, 381, 382, 394, 400, 405, 502], "transformerencod": 273, "transit": 489, "translat": [147, 368], "transpos": [0, 6, 32, 105, 106, 107, 170, 245, 351, 352, 353], "treat": [0, 2, 115, 159, 160, 162, 163, 300, 419, 498], "tree": [8, 97, 139, 174, 313, 317, 321, 325, 326, 327, 328, 329, 481, 482, 484, 493, 501], "tree_flatten": [273, 326, 329, 341, 473, 500], "tree_map": [327, 341, 499], "tree_unflatten": [6, 473, 500], "trembl": 6, "tri": [0, 126], "triangl": [197, 198, 309], "triangular": [194, 195, 205, 207], "trigger": 498, "tril": 0, "trilinear": 419, "triplet": 453, "triplet_loss": 341, "triu": 0, "true": [0, 1, 2, 4, 5, 6, 17, 42, 43, 44, 45, 83, 97, 111, 112, 113, 114, 148, 150, 170, 182, 188, 194, 195, 202, 206, 209, 223, 245, 286, 318, 322, 325, 326, 327, 328, 332, 341, 346, 348, 349, 350, 351, 352, 353, 361, 362, 366, 367, 368, 370, 381, 382, 384, 391, 394, 400, 402, 405, 410, 418, 419, 440, 448, 475, 477, 478, 498, 500], "truncat": [152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 258], "truth": [5, 442, 452], "try": [2, 9, 499], "tupl": [0, 31, 66, 69, 80, 98, 102, 103, 104, 106, 107, 128, 132, 134, 139, 141, 143, 180, 189, 197, 200, 201, 202, 203, 206, 239, 244, 265, 267, 284, 285, 291, 312, 313, 316, 325, 326, 327, 328, 329, 343, 344, 345, 349, 350, 352, 353, 373, 374, 375, 384, 386, 407, 419, 475, 477, 478, 479, 480, 497, 500, 501], "tutori": 2, "twice": 508, "two": [0, 2, 14, 15, 17, 25, 83, 86, 88, 90, 91, 92, 121, 131, 136, 153, 156, 162, 169, 170, 175, 176, 182, 190, 192, 193, 194, 195, 196, 197, 198, 199, 203, 206, 207, 214, 219, 221, 229, 231, 235, 238, 298, 328, 344, 360, 367, 374, 433, 441, 498, 499, 500, 501, 502, 508], "txt": [2, 4], "type": [0, 1, 2, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 69, 79, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 118, 119, 120, 121, 124, 125, 126, 128, 129, 130, 131, 132, 133, 134, 136, 137, 138, 140, 141, 142, 146, 147, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 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, 228, 229, 230, 231, 232, 233, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 257, 258, 259, 260, 261, 262, 263, 265, 266, 268, 269, 275, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 296, 297, 298, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 324, 325, 328, 341, 389, 418, 420, 421, 422, 423, 424, 425, 426, 427, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 498, 500, 502, 505], "type_to_nam": 2, "typenam": [0, 1, 2], "typic": [0, 151, 321, 358, 473, 498, 504], "u": [1, 2, 4, 194, 197, 198, 200, 206, 370, 396, 493, 499, 503, 504], "u_": 474, "u_t": 474, "uint": [1, 2, 148], "uint16": [12, 332], "uint3": 1, "uint32": [12, 27, 28, 29, 30, 248, 332], "uint64": [12, 332], "uint8": [12, 332], "ultra": 6, "unabl": 9, "unam": 9, "unari": 498, "unchang": [150, 294, 405], "uncheck": 9, "uncompress": 273, "undefin": [0, 29, 115, 194, 195, 240, 252, 502], "under": 202, "underli": [2, 315], "understand": [6, 421, 422], "unevalu": 144, "unexpect": [2, 19], "unexpectedli": 503, "unflatten": 0, "unfreez": [341, 382], "unfrozen": 394, "unifi": 8, "uniform": [3, 341, 370, 384, 422, 424, 468, 495, 498, 501, 508], "uniformli": 259, "unintend": 0, "union": [19, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 85, 86, 87, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 186, 187, 188, 190, 197, 198, 206, 224, 271, 295], "uniqu": [2, 204, 205, 495, 499], "unique_ptr": 2, "unit": [347, 357, 359, 360, 361, 369, 403, 404, 406, 408, 421, 422, 423, 424, 428, 429, 430, 431, 432, 433, 437, 456, 457, 458, 460], "unittest": 9, "univers": 202, "unless": [6, 17, 182, 202, 471], "unlik": [6, 17, 182, 200, 355, 356, 390], "unnecessari": [2, 6], "unnorm": [248, 440, 442], "unscal": 475, "unsign": [170, 244, 245, 332], "unsignedinteg": 12, "unspecifi": [16, 18, 19, 27, 28, 29, 30, 98, 111, 112, 113, 114, 168, 218, 220, 222, 228, 236, 240, 242, 263, 286, 287, 293, 297, 300, 306, 307, 314, 319, 509], "unsqueez": 6, "unsupport": 209, "until": [2, 321, 504, 506], "unus": 2, "up": [1, 2, 6, 115, 498], "upcast": 2, "updat": [0, 1, 2, 5, 6, 7, 39, 97, 285, 322, 326, 328, 346, 377, 378, 384, 389, 390, 391, 396, 473, 475, 478, 480, 481, 482, 486, 487, 488, 489, 490, 491, 498, 499, 500, 504], "update_modul": 341, "uplo": [197, 198], "upon": [6, 326, 327], "upper": [194, 195, 197, 198, 205, 207, 244, 255, 258, 259, 427], "upsampl": 341, "us": [0, 3, 5, 6, 7, 8, 9, 19, 39, 84, 115, 119, 122, 124, 125, 128, 129, 130, 132, 148, 150, 165, 170, 171, 172, 173, 191, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 219, 244, 245, 263, 265, 266, 267, 275, 278, 279, 299, 312, 321, 325, 328, 332, 334, 341, 344, 345, 355, 358, 359, 361, 367, 370, 374, 375, 377, 381, 388, 395, 397, 399, 400, 402, 405, 410, 418, 419, 423, 424, 431, 432, 441, 468, 471, 473, 474, 475, 477, 478, 479, 480, 481, 482, 495, 497, 498, 499, 500, 501, 502, 503, 506, 508], "usag": [418, 498, 499], "user": [2, 6, 341], "usual": [358, 399, 497, 504], "util": [1, 2, 6, 8, 9, 273, 341, 473, 503], "v": [6, 108, 151, 197, 341, 382, 505], "v_": [474, 476, 477, 478, 479, 485, 486], "v_t": [474, 476, 477, 478, 479, 485, 486], "val": [0, 31, 168], "valid": [7, 108, 165, 317, 325, 382, 394, 497, 499], "valid_parameter_filt": 377, "valu": [0, 1, 5, 6, 12, 13, 17, 19, 27, 28, 51, 79, 83, 96, 126, 143, 146, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 168, 180, 182, 196, 202, 206, 208, 224, 232, 239, 243, 247, 248, 249, 251, 252, 253, 255, 258, 259, 267, 271, 279, 300, 301, 313, 317, 323, 325, 326, 327, 328, 332, 344, 345, 347, 354, 355, 356, 357, 363, 366, 370, 374, 375, 381, 397, 398, 414, 416, 418, 420, 440, 441, 442, 443, 444, 445, 447, 448, 449, 450, 451, 452, 465, 471, 475, 478, 487, 488, 490, 491, 501], "value_and_grad": [7, 115, 341, 395, 471, 473, 484, 498, 501, 505, 506], "value_and_grad_fn": 504, "value_cach": 6, "value_dim": 397, "value_input_dim": 397, "value_output_dim": 397, "value_proj": 6, "valueerror": [115, 202, 384, 501], "values_hat": 6, "van": 202, "var": [0, 346, 362, 366, 368, 443], "variabl": [9, 97, 115, 122, 135, 143, 144, 145, 174, 189, 313, 316, 317, 498, 499, 500], "varianc": [0, 293, 314, 346, 362, 443], "variant": [6, 452, 479], "variou": 202, "vector": [0, 2, 5, 8, 181, 189, 202, 300, 316, 317, 358, 442, 500, 506], "verbos": [1, 148, 499], "veri": [6, 397, 503, 504, 508], "verifi": [5, 9], "versa": 267, "version": [2, 4, 9, 119, 143, 145, 180, 214, 218, 244, 286, 317, 495, 501, 502], "versu": 498, "via": [9, 115, 481, 484, 499, 503, 504, 505], "vice": 267, "video": 356, "view": [0, 3, 84, 505], "virtual": 2, "visual": 144, "vjp": [2, 115, 506], "vmap": [2, 115, 500, 501, 504, 506], "vmap_add": 501, "vocab_s": 6, "vocabulari": [358, 399], "void": [1, 2], "vt": 206, "w": [0, 1, 5, 102, 103, 106, 107, 119, 170, 197, 244, 245, 313, 327, 346, 349, 350, 352, 353, 355, 356, 370, 473, 486, 501], "w1": [6, 324], "w2": [6, 324], "w3": 6, "w_": [361, 367, 402, 474, 475, 476, 477, 478, 479, 480, 485, 486], "w_1": 244, "w_g": 244, "w_i": [119, 244], "w_in": 1, "w_q": 244, "w_star": 5, "w_stride": 1, "w_t": [474, 476, 477, 478, 479, 480, 485, 486], "wa": [4, 6, 84, 128, 129, 499, 500, 504], "wai": [2, 6, 9, 341, 419, 498, 499, 500, 501, 502, 503], "wait": 6, "walk": [6, 500], "walkthrough": 2, "walsh": 177, "want": [1, 2, 6, 499, 500, 501, 503, 508], "warm": [2, 498], "warmup_init": 475, "watch": [6, 498], "wd": 480, "we": [0, 1, 2, 5, 6, 7, 115, 119, 128, 129, 170, 244, 245, 341, 358, 399, 407, 478, 480, 495, 497, 498, 499, 500, 501, 503, 504, 508], "weight": [0, 5, 101, 102, 103, 104, 105, 106, 107, 147, 149, 326, 341, 384, 388, 399, 400, 440, 442, 471, 475, 478, 480, 482, 486, 501, 504], "weight_decai": [475, 478, 480, 486], "weight_fil": 6, "weights_fp16": 504, "well": [6, 341, 382, 394, 397, 499, 504], "wen": 6, "went": 6, "were": [6, 508], "wet": 6, "what": [2, 6, 326, 503], "whatsoev": 6, "whc": 355, "when": [0, 1, 2, 6, 8, 9, 97, 104, 115, 130, 194, 195, 197, 198, 199, 202, 206, 207, 209, 278, 348, 349, 350, 351, 352, 353, 419, 423, 424, 440, 446, 452, 471, 473, 489, 495, 498, 499, 500, 508], "where": [0, 4, 7, 146, 182, 195, 244, 313, 317, 346, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 359, 361, 362, 366, 367, 368, 370, 381, 398, 401, 402, 416, 423, 424, 429, 430, 432, 443, 449, 455, 458, 460, 465, 482, 499, 501, 502], "wherea": 501, "whether": [143, 145, 148, 170, 197, 198, 205, 207, 245, 361, 367, 381, 397, 402, 440, 443, 449], "which": [0, 1, 2, 6, 7, 8, 9, 19, 38, 84, 97, 104, 121, 124, 125, 126, 128, 129, 130, 139, 143, 145, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 165, 174, 180, 183, 184, 185, 186, 187, 189, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 209, 223, 226, 244, 248, 249, 263, 265, 267, 270, 271, 272, 273, 274, 291, 292, 300, 307, 312, 313, 316, 317, 322, 344, 345, 355, 356, 359, 374, 375, 377, 381, 405, 440, 442, 445, 449, 452, 468, 481, 482, 495, 498, 499, 500, 501, 502, 503, 504, 508, 509], "while": [2, 3, 6, 9, 265, 405, 504, 505], "whistl": 2, "who": 6, "whose": [146, 322, 323], "why": 6, "wi": 499, "wide": 504, "width": [344, 345, 346, 349, 350, 352, 353, 355, 356, 374, 375, 399, 400], "window": [9, 343, 344, 345, 373, 374, 375], "wipe": 9, "wire": 279, "wired_limit_mb": 279, "wise": [0, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 88, 89, 90, 91, 94, 109, 110, 131, 132, 136, 137, 138, 140, 142, 166, 167, 175, 176, 182, 191, 192, 193, 210, 211, 212, 213, 214, 215, 216, 217, 221, 229, 231, 233, 235, 241, 261, 262, 266, 269, 280, 281, 282, 283, 289, 290, 296, 302, 303, 347, 355, 356, 365, 376, 398, 409, 428, 435, 436, 438, 439, 454, 455, 457, 460, 461, 462, 463, 498], "wish": 9, "with_logit": 440, "within": [0, 3, 29, 182], "without": [1, 6, 8, 294, 397, 467, 497, 498, 499, 500, 503, 504, 505, 508], "wk": 6, "wl": 2, "wo": 6, "word": 0, "work": [2, 3, 6, 278, 332, 498, 499, 500, 501, 502, 503, 504], "workhors": 341, "world": [329, 499], "world2": 499, "world_ani": 499, "world_mpi": 499, "world_r": 499, "worri": [1, 504], "would": [2, 6, 419, 499, 500, 502, 504, 505, 508], "wq": 6, "wrap": [115, 341], "wrapper": [500, 503], "write": [0, 1, 6, 341, 505], "written": [2, 500], "wrong": 500, "wrt": 323, "wv": 6, "x": [0, 1, 2, 4, 5, 6, 7, 39, 92, 115, 124, 125, 129, 130, 137, 142, 143, 144, 147, 148, 149, 170, 177, 178, 180, 202, 245, 249, 254, 268, 273, 280, 310, 311, 318, 326, 328, 341, 343, 344, 345, 346, 347, 357, 359, 360, 362, 366, 368, 369, 370, 373, 374, 375, 376, 377, 398, 401, 403, 409, 410, 416, 419, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 452, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 471, 473, 480, 498, 499, 500, 501, 502, 504, 505, 506, 508], "x1": 441, "x2": 441, "x86_64": 9, "x_1": [441, 449], "x_2": [441, 449], "x_cast": 2, "x_grad": 1, "x_i": [439, 461, 462], "x_j": [461, 462], "x_offset": 2, "x_ptr": 2, "x_shape": 1, "x_stride": 2, "x_t": [361, 367, 402], "x_view": 505, "xcode": 9, "xcodeproj": 3, "xcrun": 9, "xf": 367, "xg": 367, "xi": 367, "xn": 361, "xo": 367, "xor": 91, "xr": 361, "xy": [0, 223], "xz": 361, "x\u00b2": 505, "y": [0, 2, 4, 5, 6, 7, 39, 115, 143, 144, 177, 180, 318, 341, 346, 355, 362, 366, 368, 370, 401, 444, 449, 452, 473, 476, 498, 499, 500, 501, 504, 505], "y_": [444, 448], "y_cast": 2, "y_hat": 341, "y_offset": 2, "y_ptr": 2, "y_stride": 2, "ye": 6, "year": 6, "yet": [6, 341, 471, 482, 501, 502, 504, 506], "yield": [6, 7, 495], "you": [2, 3, 4, 6, 7, 8, 9, 279, 341, 410, 418, 468, 495, 498, 499, 500, 501, 502, 503, 505, 507, 508], "your": [2, 6, 9, 471, 499, 501, 504], "z": [2, 361, 498, 500, 504], "z_t": 361, "zeiler": 474, "zero": [0, 143, 146, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 196, 223, 264, 285, 309, 310, 311, 320, 341, 343, 344, 345, 354, 355, 356, 384, 420, 421, 422, 423, 424, 425, 426, 427, 468, 473, 475, 500, 502], "zero_grad": 501, "zeros_lik": [0, 200], "zhang": 6, "zip": [6, 7], "zip_saf": 2}, "titles": ["Operations", "Custom Metal Kernels", "Custom Extensions in MLX", "Metal Debugger", "Using MLX in C++", "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.std", "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_invert", "mlx.core.bitwise_or", "mlx.core.bitwise_xor", "mlx.core.block_masked_mm", "mlx.core.broadcast_to", "mlx.core.ceil", "mlx.core.clear_cache", "mlx.core.clip", "mlx.core.compile", "mlx.core.concatenate", "mlx.core.conj", "mlx.core.conjugate", "mlx.core.conv1d", "mlx.core.conv2d", "mlx.core.conv3d", "mlx.core.conv_general", "mlx.core.conv_transpose1d", "mlx.core.conv_transpose2d", "mlx.core.conv_transpose3d", "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.distributed.recv", "mlx.core.distributed.recv_like", "mlx.core.distributed.send", "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.export_function", "mlx.core.export_to_dot", "mlx.core.exporter", "mlx.core.eye", "mlx.core.fast.layer_norm", "mlx.core.fast.metal_kernel", "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.finfo", "mlx.core.flatten", "mlx.core.floor", "mlx.core.floor_divide", "mlx.core.full", "mlx.core.gather_mm", "mlx.core.gather_qmm", "mlx.core.get_active_memory", "mlx.core.get_cache_memory", "mlx.core.get_peak_memory", "mlx.core.grad", "mlx.core.greater", "mlx.core.greater_equal", "mlx.core.hadamard_transform", "mlx.core.identity", "mlx.core.imag", "mlx.core.import_function", "mlx.core.inner", "mlx.core.isclose", "mlx.core.isfinite", "mlx.core.isinf", "mlx.core.isnan", "mlx.core.isneginf", "mlx.core.isposinf", "mlx.core.issubdtype", "mlx.core.jvp", "mlx.core.kron", "mlx.core.left_shift", "mlx.core.less", "mlx.core.less_equal", "mlx.core.linalg.cholesky", "mlx.core.linalg.cholesky_inv", "mlx.core.linalg.cross", "mlx.core.linalg.eigh", "mlx.core.linalg.eigvalsh", "mlx.core.linalg.inv", "mlx.core.linalg.lu", "mlx.core.linalg.lu_factor", "mlx.core.linalg.norm", "mlx.core.linalg.qr", "mlx.core.linalg.solve", "mlx.core.linalg.solve_triangular", "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.device_info", "mlx.core.metal.is_available", "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.put_along_axis", "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.permutation", "mlx.core.random.randint", "mlx.core.random.seed", "mlx.core.random.split", "mlx.core.random.truncated_normal", "mlx.core.random.uniform", "mlx.core.real", "mlx.core.reciprocal", "mlx.core.remainder", "mlx.core.repeat", "mlx.core.reset_peak_memory", "mlx.core.reshape", "mlx.core.right_shift", "mlx.core.roll", "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_cache_limit", "mlx.core.set_default_device", "mlx.core.set_default_stream", "mlx.core.set_memory_limit", "mlx.core.set_wired_limit", "mlx.core.sigmoid", "mlx.core.sign", "mlx.core.sin", "mlx.core.sinh", "mlx.core.slice", "mlx.core.slice_update", "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.unflatten", "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.average_gradients", "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", "Export Functions", "Fast", "FFT", "Linear Algebra", "Memory Management", "Metal", "Neural Networks", "mlx.nn.ALiBi", "mlx.nn.AvgPool1d", "mlx.nn.AvgPool2d", "mlx.nn.AvgPool3d", "mlx.nn.BatchNorm", "mlx.nn.CELU", "mlx.nn.Conv1d", "mlx.nn.Conv2d", "mlx.nn.Conv3d", "mlx.nn.ConvTranspose1d", "mlx.nn.ConvTranspose2d", "mlx.nn.ConvTranspose3d", "mlx.nn.Dropout", "mlx.nn.Dropout2d", "mlx.nn.Dropout3d", "mlx.nn.ELU", "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.LogSigmoid", "mlx.nn.LogSoftmax", "mlx.nn.MaxPool1d", "mlx.nn.MaxPool2d", "mlx.nn.MaxPool3d", "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.Sigmoid", "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.celu", "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", "Exporting Functions", "Function Transforms", "Indexing Arrays", "Launching Distributed Programs", "Lazy Evaluation", "Conversion to NumPy and Other Frameworks", "Quick Start Guide", "Saving and Loading Arrays", "Unified Memory", "Using Streams"], "titleterms": {"A": 508, "In": 502, "The": 341, "ab": [13, 33], "adadelta": 474, "adafactor": 475, "adagrad": 476, "adam": 477, "adamax": 479, "adamw": 478, "add": 14, "addmm": 15, "algebra": 338, "alibi": 342, "all": [6, 16, 34, 499], "all_gath": 124, "all_sum": 125, "allclos": 17, "ani": [18, 35], "api": [8, 9], "appli": 377, "apply_gradi": 481, "apply_to_modul": 378, "arang": 19, "arcco": 20, "arccosh": 21, "arcsin": 22, "arcsinh": 23, "arctan": 24, "arctan2": 25, "arctanh": 26, "argmax": [27, 36], "argmin": [28, 37], "argpartit": 29, "argsort": 30, "arrai": [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, 331, 502, 507], "array_equ": 83, "as_strid": 84, "astyp": 38, "atleast_1d": 85, "atleast_2d": 86, "atleast_3d": 87, "attent": 6, "automat": 501, "average_gradi": [321, 499], "avgpool1d": 343, "avgpool2d": 344, "avgpool3d": 345, "back": 2, "backend": 499, "basic": [498, 500, 506], "batchnorm": 346, "benchmark": 6, "bernoulli": 247, "binari": 9, "binary_cross_entropi": 440, "bind": 2, "bitwise_and": 88, "bitwise_invert": 89, "bitwise_or": 90, "bitwise_xor": 91, "block_masked_mm": 92, "broadcast_to": 93, "build": [2, 9], "c": [4, 8, 9, 500], "categor": 248, "ceil": 94, "celu": [347, 428], "children": 379, "choleski": 194, "cholesky_inv": 195, "class": 341, "clear_cach": 95, "clip": 96, "clip_grad_norm": 324, "cmake": 2, "co": [41, 109], "code": [2, 6], "common": 492, "commun": [334, 499], "compil": [97, 498], "complex": 1, "comput": 504, "concaten": 98, "conj": [40, 99], "conjug": 100, "constant": 420, "conv1d": [101, 348], "conv2d": [102, 349], "conv3d": [103, 350], "conv_gener": 104, "conv_transpose1d": 105, "conv_transpose2d": 106, "conv_transpose3d": 107, "convers": 505, "convert": 6, "convolv": 108, "convtranspose1d": 351, "convtranspose2d": 352, "convtranspose3d": 353, "core": [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, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 330], "cosh": 110, "cosine_decai": 487, "cosine_similarity_loss": 441, "cpu": 2, "cross": 196, "cross_entropi": 442, "cummax": [42, 111], "cummin": [43, 112], "cumprod": [44, 113], "cumsum": [45, 114], "custom": [1, 2], "custom_funct": 115, "data": 332, "debug": 498, "debugg": 3, "default_devic": 116, "default_stream": 117, "defin": 499, "degre": 118, "dequant": 119, "devic": [10, 333], "device_info": 224, "diag": [46, 120], "diagon": [47, 121], "differ": 502, "differenti": 501, "disable_compil": 122, "distribut": [123, 124, 125, 126, 127, 128, 129, 130, 334, 499, 503], "divid": 131, "divmod": 132, "download": [2, 6], "dropout": 354, "dropout2d": 355, "dropout3d": 356, "dtype": [11, 48], "dtypecategori": 12, "eigh": 197, "eigvalsh": 198, "einsum": 133, "einsum_path": 134, "elu": [357, 429], "embed": 358, "enable_compil": 135, "encod": 6, "end": 2, "equal": 136, "erf": 137, "erfinv": 138, "eval": [139, 380], "evalu": 504, "exampl": [1, 2, 8, 498, 499, 500, 508], "exp": [49, 140], "expand_dim": 141, "expm1": 142, "exponential_decai": 488, "export": [145, 335, 500], "export_funct": 143, "export_to_dot": 144, "extens": 2, "ey": 146, "fast": [147, 148, 149, 150, 151, 336], "fft": [152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 337], "fft2": 153, "fftn": 154, "filter_and_map": 381, "finfo": 164, "flatten": [50, 165], "floor": 166, "floor_divid": 167, "format": 507, "found": 9, "framework": 505, "freez": 382, "from": [9, 502], "full": [6, 168], "function": [335, 467, 470, 498, 500, 501, 506], "further": 8, "gather_mm": 169, "gather_qmm": 170, "gaussian_nll_loss": 443, "gelu": [359, 430], "gelu_approx": 431, "gelu_fast_approx": 432, "gener": 6, "get": 499, "get_active_memori": 171, "get_cache_memori": 172, "get_peak_memori": 173, "glorot_norm": 421, "glorot_uniform": 422, "glu": [360, 433], "gpu": 2, "grad": [174, 341], "graph": [498, 504, 506], "greater": 175, "greater_equ": 176, "grid": 1, "group": 123, "groupnorm": 362, "gru": 361, "guid": 506, "gumbel": 249, "hadamard_transform": 177, "hard_shrink": 434, "hard_tanh": 435, "hardshrink": 363, "hardswish": [365, 436], "hardtanh": 364, "he_norm": 423, "he_uniform": 424, "hinge_loss": 444, "host": [499, 503], "huber_loss": 445, "ident": [178, 425], "ifft": 155, "ifft2": 156, "ifftn": 157, "imag": 179, "implement": [2, 6], "import": 500, "import_funct": 180, "index": 502, "infer": 6, "init": [126, 420, 421, 422, 423, 424, 425, 426, 427, 482], "initi": 468, "inner": 181, "inspect": 341, "instal": [8, 9, 499], "instancenorm": 366, "introduc": 2, "inv": 199, "irfft": 158, "irfft2": 159, "irfftn": 160, "is_avail": [127, 225], "isclos": 182, "isfinit": 183, "isinf": 184, "isnan": 185, "isneginf": 186, "isposinf": 187, "issubdtyp": 188, "item": 51, "items": 52, "jax": 505, "join_schedul": 489, "jvp": 189, "kei": 250, "kernel": 1, "kl_div_loss": 446, "kron": 190, "l1_loss": 447, "laplac": 251, "launch": 503, "layer": [6, 7, 469], "layer_norm": 147, "layernorm": 368, "lazi": 504, "leaf_modul": 383, "leaky_relu": 437, "leakyrelu": 369, "left_shift": 191, "less": 192, "less_equ": 193, "linalg": [194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207], "linear": [5, 338, 370], "linear_schedul": 490, "linspac": 208, "lion": 480, "llm": 6, "load": [6, 209, 473, 507], "load_weight": 384, "log": [53, 210], "log10": [54, 211], "log1p": [55, 212], "log2": [56, 213], "log_cosh_loss": 448, "log_sigmoid": 438, "log_softmax": 439, "logaddexp": 214, "logical_and": 215, "logical_not": 216, "logical_or": 217, "logsigmoid": 371, "logsoftmax": 372, "logsumexp": [57, 218], "loss": [440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 470], "lstm": 367, "lu": 200, "lu_factor": 201, "manag": 339, "margin_ranking_loss": 449, "matmul": 219, "max": [58, 220], "maximum": 221, "maxpool1d": 373, "maxpool2d": 374, "maxpool3d": 375, "mean": [59, 222], "memori": [339, 508], "meshgrid": 223, "metal": [1, 3, 9, 224, 225, 226, 227, 340], "metal_kernel": 148, "min": [60, 228], "minim": 9, "minimum": 229, "mish": [376, 454], "mlx": [2, 4, 8, 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, 295, 296, 297, 298, 299, 300, 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, 328, 329, 330, 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, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491], "model": 6, "modul": [341, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 471, 500], "more": 500, "moveaxi": [61, 230], "mpi": [499, 503], "mse_loss": 450, "multi": 7, "multiheadattent": 397, "multipl": 500, "multipli": 231, "multivariate_norm": 252, "named_modul": 386, "nan_to_num": 232, "nbyte": 62, "ndim": 63, "neg": 233, "network": 341, "neural": 341, "new_stream": 234, "nll_loss": 451, "nn": [321, 322, 323, 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, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 499], "norm": 202, "normal": [253, 426], "not_equ": 235, "numpi": [502, 505], "ones": 236, "ones_lik": 237, "onli": 504, "oper": [0, 2, 472], "optim": [324, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493], "option": 9, "other": 505, "outer": 238, "packag": 4, "pad": 239, "paramet": [341, 387], "partit": 240, "perceptron": 7, "permut": 254, "place": 502, "power": 241, "prelu": [398, 455], "primit": 2, "prod": [64, 242], "program": [499, 503], "provid": 503, "pure": 498, "put": 6, "put_along_axi": 243, "python": [2, 8, 9], "pytorch": 505, "qr": 203, "quantiz": [244, 322], "quantized_matmul": 245, "quantizedembed": 399, "quantizedlinear": 400, "quick": [341, 506], "radian": 246, "randint": 255, "random": [247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 495], "read": 8, "real": 260, "reciproc": [65, 261], "recv": 128, "recv_lik": 129, "reduc": 499, "refer": 8, "regress": 5, "relu": [403, 456], "relu6": [404, 457], "remaind": 262, "remot": [499, 503], "repeat": 263, "requir": 9, "reset_peak_memori": 264, "reshap": [66, 265], "result": 2, "rfft": 161, "rfft2": 162, "rfftn": 163, "right_shift": 266, "ring": [499, 503], "rms_norm": 149, "rmsnorm": 401, "rmsprop": 485, "rnn": 402, "roll": 267, "rope": [150, 405], "round": [67, 268], "rsqrt": [68, 269], "run": 499, "sampl": 1, "save": [270, 473, 507], "save_gguf": 271, "save_safetensor": 272, "save_weight": 388, "savez": 273, "savez_compress": 274, "scaled_dot_product_attent": 151, "schedul": 494, "script": [2, 6], "seed": 256, "select": 499, "selu": [406, 458], "send": 130, "sequenti": 407, "serial": 507, "set": [499, 503], "set_cache_limit": 275, "set_default_devic": 276, "set_default_stream": 277, "set_dtyp": 389, "set_memory_limit": 278, "set_wired_limit": 279, "setuptool": 2, "sgd": 486, "shape": [1, 69], "shapeless": [498, 500], "shell": 9, "sigmoid": [280, 409, 459], "sign": 281, "silu": [408, 460], "simpl": [1, 508], "sin": [70, 282], "sinh": 283, "sinusoidalpositionalencod": 410, "size": [9, 71], "slice": 284, "slice_upd": 285, "smooth_l1_loss": 452, "softmax": [286, 411, 461], "softmin": [412, 462], "softplu": [413, 463], "softshrink": [414, 464], "softsign": 415, "solv": 204, "solve_triangular": 205, "sort": 287, "sourc": 9, "specif": 503, "specifi": 509, "speedup": 498, "split": [72, 257, 288], "sqrt": [73, 289], "squar": [74, 290], "squeez": [75, 291], "stack": 292, "start": [341, 499, 506], "start_captur": 226, "state": [390, 483], "std": [76, 293], "step": [416, 465], "step_decai": 491, "stop_captur": 227, "stop_gradi": 294, "stream": [295, 330, 333, 509], "stride": 1, "subtract": 296, "sum": [77, 297], "support": 332, "svd": 206, "swapax": [78, 298], "synchron": 299, "t": 32, "take": 300, "take_along_axi": 301, "tan": 302, "tanh": [303, 417, 466], "tensordot": 304, "tensorflow": 505, "thunderbolt": 499, "tile": 305, "togeth": 6, "tolist": 79, "topk": 306, "trace": [307, 500], "train": [391, 393, 498, 499], "trainable_paramet": 392, "transform": [2, 418, 496, 498, 500, 501, 504, 506], "transpos": [80, 308], "tree": 497, "tree_flatten": 325, "tree_map": 326, "tree_map_with_path": 327, "tree_reduc": 328, "tree_unflatten": 329, "tri": 309, "tri_inv": 207, "tril": 310, "triplet_loss": 453, "triu": 311, "troubleshoot": 9, "truncated_norm": 258, "tune": 499, "type": 332, "unflatten": 312, "unfreez": 394, "unifi": 508, "uniform": [259, 427], "up": [499, 503], "updat": [341, 395, 484, 502], "update_modul": 396, "upsampl": 419, "us": [1, 2, 4, 504, 509], "usag": [2, 8, 503], "util": [325, 326, 327, 328, 329, 497, 499], "valu": 341, "value_and_grad": [313, 323], "var": [81, 314], "variabl": 4, "vector": 501, "view": [82, 315], "vjp": [1, 316], "vmap": 317, "weight": 6, "what": 504, "when": 504, "where": 318, "why": 504, "workflow": 3, "x86": 9, "xcode": 3, "you": 504, "zero": 319, "zeros_lik": 320}}) \ No newline at end of file +Search.setIndex({"alltitles": {"A Simple Example": [[512, "a-simple-example"]], "Array": [[335, null]], "Attention layer": [[6, "attention-layer"]], "Automatic Differentiation": [[505, "automatic-differentiation"]], "Automatic Vectorization": [[505, "automatic-vectorization"]], "Basics": [[510, "basics"]], "Basics of Compile": [[502, "basics-of-compile"]], "Basics of Exporting": [[504, "basics-of-exporting"]], "Binary Size Minimization": [[9, "binary-size-minimization"]], "Binding to Python": [[2, "binding-to-python"]], "Build Options": [[9, "id4"]], "Build Requirements": [[9, "build-requirements"]], "Build and Install": [[9, null]], "Build from source": [[9, "build-from-source"]], "Building and Binding": [[2, "building-and-binding"]], "Building with CMake": [[2, "building-with-cmake"]], "Building with setuptools": [[2, "building-with-setuptools"]], "C++ API": [[9, "c-api"]], "C++ API Reference": [[8, null]], "Common Optimizers": [[496, null]], "Compilation": [[502, null]], "Compiling Training Graphs": [[502, "compiling-training-graphs"]], "Complex Example": [[1, "complex-example"]], "Conversion to NumPy and Other Frameworks": [[509, null]], "Converting the weights": [[6, "converting-the-weights"]], "Custom Extensions in MLX": [[2, null]], "Custom Metal Kernels": [[1, null]], "Data Types": [[336, null]], "Debugging": [[502, "debugging"]], "Defining a Ring": [[503, "defining-a-ring"]], "Devices and Streams": [[337, null]], "Differences from NumPy": [[506, "differences-from-numpy"]], "Distributed Communication": [[338, null], [503, null]], "Download the code": [[2, null], [6, null]], "Encoder layer": [[6, "encoder-layer"]], "Example Speedup": [[502, "example-speedup"]], "Examples": [[8, null]], "Export Functions": [[339, null]], "Exporting Functions": [[504, null]], "Exporting Modules": [[504, "exporting-modules"]], "Exporting Multiple Traces": [[504, "exporting-multiple-traces"]], "FFT": [[341, null]], "Fast": [[340, null]], "Full model": [[6, "full-model"]], "Function Transforms": [[505, null]], "Function and Graph Transformations": [[510, "function-and-graph-transformations"]], "Functions": [[471, null]], "Further Reading": [[8, null]], "Generation": [[6, "generation"]], "Getting Started": [[503, "getting-started"]], "Getting Started with MPI": [[503, "getting-started-with-mpi"]], "Getting Started with Ring": [[503, "getting-started-with-ring"]], "Grid Sample VJP": [[1, "grid-sample-vjp"]], "Implementing the CPU Back-end": [[2, "implementing-the-cpu-back-end"]], "Implementing the GPU Back-end": [[2, "implementing-the-gpu-back-end"]], "Implementing the Primitive": [[2, "implementing-the-primitive"]], "Implementing the model": [[6, "implementing-the-model"]], "Importing Functions in C++": [[504, "importing-functions-in-c"]], "In Place Updates": [[506, "in-place-updates"]], "Indexing Arrays": [[506, null]], "Initializers": [[472, null]], "Inspecting Modules": [[345, "inspecting-modules"]], "Install": [[8, null]], "Installing MPI": [[503, "installing-mpi"]], "Introducing the Example": [[2, "introducing-the-example"]], "JAX": [[509, "jax"]], "LLM inference": [[6, null]], "Launching Distributed Programs": [[507, null]], "Layers": [[473, null]], "Lazy Evaluation": [[508, null]], "Linear Algebra": [[342, null]], "Linear Regression": [[5, null]], "Loss Functions": [[474, null]], "MLX": [[8, null]], "MPI Specifics": [[507, "mpi-specifics"]], "Memory Management": [[343, null]], "Metal": [[344, null]], "Metal Debugger": [[3, null]], "Metal not found": [[9, "metal-not-found"]], "Module": [[475, null]], "More Examples": [[504, "more-examples"]], "Multi-Layer Perceptron": [[7, null]], "Neural Networks": [[345, null]], "Only Compute What You Use": [[508, "only-compute-what-you-use"]], "Operations": [[0, null], [2, "operations"], [476, null]], "Operations and Primitives": [[2, "operations-and-primitives"]], "Optimizer": [[497, null]], "Optimizers": [[477, null]], "Package Variables": [[4, "id1"]], "Parameters": [[345, "parameters"]], "Primitive Transforms": [[2, "primitive-transforms"]], "Primitives": [[2, "primitives"]], "Providing Hosts": [[507, "providing-hosts"]], "Pure Functions": [[502, "pure-functions"]], "Putting it all together": [[6, "putting-it-all-together"]], "PyTorch": [[509, "pytorch"]], "Python API": [[9, "python-api"]], "Python API Reference": [[8, null]], "Python Installation": [[9, "python-installation"]], "Quick Start Guide": [[510, null]], "Quick Start with Neural Networks": [[345, "quick-start-with-neural-networks"]], "Random": [[499, null]], "Results": [[2, "results"]], "Ring Specifics": [[507, "ring-specifics"]], "Running Distributed Programs": [[503, "running-distributed-programs"]], "Saving and Loading": [[477, "saving-and-loading"]], "Saving and Loading Arrays": [[511, null]], "Schedulers": [[498, null]], "Scripts": [[2, "scripts"], [6, "scripts"]], "Selecting Backend": [[503, "selecting-backend"]], "Serialization Formats": [[511, "id1"]], "Setting up Remote Hosts": [[503, "setting-up-remote-hosts"], [507, "setting-up-remote-hosts"]], "Shapeless Compilation": [[502, "shapeless-compilation"]], "Shapeless Exports": [[504, "shapeless-exports"]], "Simple Example": [[1, "simple-example"]], "Specifying the Stream": [[513, "specifying-the-stream"]], "Supported Data Types": [[336, "id2"]], "TensorFlow": [[509, "tensorflow"]], "The Module Class": [[345, "the-module-class"]], "Thunderbolt Ring": [[503, "thunderbolt-ring"]], "Training Example": [[503, "training-example"]], "Transformations with Compile": [[502, "transformations-with-compile"]], "Transformations with Imported Functions": [[504, "transformations-with-imported-functions"]], "Transforming Compute Graphs": [[508, "transforming-compute-graphs"]], "Transforms": [[500, null]], "Tree Utils": [[501, null]], "Troubleshooting": [[9, "troubleshooting"], [9, "id3"]], "Tuning MPI All Reduce": [[503, "tuning-mpi-all-reduce"]], "Unified Memory": [[512, null]], "Updating the Parameters": [[345, "updating-the-parameters"]], "Usage": [[2, "usage"], [8, null], [507, "usage"]], "Using MLX in C++": [[4, null]], "Using Shape/Strides": [[1, "using-shape-strides"]], "Using Streams": [[513, null]], "Using the Primitive": [[2, "using-the-primitive"]], "Utilizing nn.average_gradients": [[503, "utilizing-nn-average-gradients"]], "Value and Grad": [[345, "value-and-grad"]], "Weight loading and benchmarking": [[6, "weight-loading-and-benchmarking"]], "When to Evaluate": [[508, "when-to-evaluate"]], "Why Lazy Evaluation": [[508, "why-lazy-evaluation"]], "Xcode Workflow": [[3, "xcode-workflow"]], "mlx.core.Device": [[10, null]], "mlx.core.Dtype": [[11, null]], "mlx.core.DtypeCategory": [[12, null]], "mlx.core.Stream": [[334, null]], "mlx.core.abs": [[13, null]], "mlx.core.add": [[14, null]], "mlx.core.addmm": [[15, null]], "mlx.core.all": [[16, null]], "mlx.core.allclose": [[17, null]], "mlx.core.any": [[18, null]], "mlx.core.arange": [[19, null]], "mlx.core.arccos": [[20, null]], "mlx.core.arccosh": [[21, null]], "mlx.core.arcsin": [[22, null]], "mlx.core.arcsinh": [[23, null]], "mlx.core.arctan": [[24, null]], "mlx.core.arctan2": [[25, null]], "mlx.core.arctanh": [[26, null]], "mlx.core.argmax": [[27, null]], "mlx.core.argmin": [[28, null]], "mlx.core.argpartition": [[29, null]], "mlx.core.argsort": [[30, null]], "mlx.core.array": [[31, null]], "mlx.core.array.T": [[32, null]], "mlx.core.array.abs": [[33, null]], "mlx.core.array.all": [[34, null]], "mlx.core.array.any": [[35, null]], "mlx.core.array.argmax": [[36, null]], "mlx.core.array.argmin": [[37, null]], "mlx.core.array.astype": [[38, null]], "mlx.core.array.at": [[39, null]], "mlx.core.array.conj": [[40, null]], "mlx.core.array.cos": [[41, null]], "mlx.core.array.cummax": [[42, null]], "mlx.core.array.cummin": [[43, null]], "mlx.core.array.cumprod": [[44, null]], "mlx.core.array.cumsum": [[45, null]], "mlx.core.array.diag": [[46, null]], "mlx.core.array.diagonal": [[47, null]], "mlx.core.array.dtype": [[48, null]], "mlx.core.array.exp": [[49, null]], "mlx.core.array.flatten": [[50, null]], "mlx.core.array.item": [[51, null]], "mlx.core.array.itemsize": [[52, null]], "mlx.core.array.log": [[53, null]], "mlx.core.array.log10": [[54, null]], "mlx.core.array.log1p": [[55, null]], "mlx.core.array.log2": [[56, null]], "mlx.core.array.logsumexp": [[57, null]], "mlx.core.array.max": [[58, null]], "mlx.core.array.mean": [[59, null]], "mlx.core.array.min": [[60, null]], "mlx.core.array.moveaxis": [[61, null]], "mlx.core.array.nbytes": [[62, null]], "mlx.core.array.ndim": [[63, null]], "mlx.core.array.prod": [[64, null]], "mlx.core.array.reciprocal": [[65, null]], "mlx.core.array.reshape": [[66, null]], "mlx.core.array.round": [[67, null]], "mlx.core.array.rsqrt": [[68, null]], "mlx.core.array.shape": [[69, null]], "mlx.core.array.sin": [[70, null]], "mlx.core.array.size": [[71, null]], "mlx.core.array.split": [[72, null]], "mlx.core.array.sqrt": [[73, null]], "mlx.core.array.square": [[74, null]], "mlx.core.array.squeeze": [[75, null]], "mlx.core.array.std": [[76, null]], "mlx.core.array.sum": [[77, null]], "mlx.core.array.swapaxes": [[78, null]], "mlx.core.array.tolist": [[79, null]], "mlx.core.array.transpose": [[80, null]], "mlx.core.array.var": [[81, null]], "mlx.core.array.view": [[82, null]], "mlx.core.array_equal": [[83, null]], "mlx.core.as_strided": [[84, null]], "mlx.core.async_eval": [[85, null]], "mlx.core.atleast_1d": [[86, null]], "mlx.core.atleast_2d": [[87, null]], "mlx.core.atleast_3d": [[88, null]], "mlx.core.bitwise_and": [[89, null]], "mlx.core.bitwise_invert": [[90, null]], "mlx.core.bitwise_or": [[91, null]], "mlx.core.bitwise_xor": [[92, null]], "mlx.core.block_masked_mm": [[93, null]], "mlx.core.broadcast_arrays": [[94, null]], "mlx.core.broadcast_to": [[95, null]], "mlx.core.ceil": [[96, null]], "mlx.core.clear_cache": [[97, null]], "mlx.core.clip": [[98, null]], "mlx.core.compile": [[99, null]], "mlx.core.concatenate": [[100, null]], "mlx.core.conj": [[101, null]], "mlx.core.conjugate": [[102, null]], "mlx.core.contiguous": [[103, null]], "mlx.core.conv1d": [[104, null]], "mlx.core.conv2d": [[105, null]], "mlx.core.conv3d": [[106, null]], "mlx.core.conv_general": [[107, null]], "mlx.core.conv_transpose1d": [[108, null]], "mlx.core.conv_transpose2d": [[109, null]], "mlx.core.conv_transpose3d": [[110, null]], "mlx.core.convolve": [[111, null]], "mlx.core.cos": [[112, null]], "mlx.core.cosh": [[113, null]], "mlx.core.cummax": [[114, null]], "mlx.core.cummin": [[115, null]], "mlx.core.cumprod": [[116, null]], "mlx.core.cumsum": [[117, null]], "mlx.core.custom_function": [[118, null]], "mlx.core.default_device": [[119, null]], "mlx.core.default_stream": [[120, null]], "mlx.core.degrees": [[121, null]], "mlx.core.dequantize": [[122, null]], "mlx.core.diag": [[123, null]], "mlx.core.diagonal": [[124, null]], "mlx.core.disable_compile": [[125, null]], "mlx.core.distributed.Group": [[126, null]], "mlx.core.distributed.all_gather": [[127, null]], "mlx.core.distributed.all_sum": [[128, null]], "mlx.core.distributed.init": [[129, null]], "mlx.core.distributed.is_available": [[130, null]], "mlx.core.distributed.recv": [[131, null]], "mlx.core.distributed.recv_like": [[132, null]], "mlx.core.distributed.send": [[133, null]], "mlx.core.divide": [[134, null]], "mlx.core.divmod": [[135, null]], "mlx.core.einsum": [[136, null]], "mlx.core.einsum_path": [[137, null]], "mlx.core.enable_compile": [[138, null]], "mlx.core.equal": [[139, null]], "mlx.core.erf": [[140, null]], "mlx.core.erfinv": [[141, null]], "mlx.core.eval": [[142, null]], "mlx.core.exp": [[143, null]], "mlx.core.expand_dims": [[144, null]], "mlx.core.expm1": [[145, null]], "mlx.core.export_function": [[146, null]], "mlx.core.export_to_dot": [[147, null]], "mlx.core.exporter": [[148, null]], "mlx.core.eye": [[149, null]], "mlx.core.fast.layer_norm": [[150, null]], "mlx.core.fast.metal_kernel": [[151, null]], "mlx.core.fast.rms_norm": [[152, null]], "mlx.core.fast.rope": [[153, null]], "mlx.core.fast.scaled_dot_product_attention": [[154, null]], "mlx.core.fft.fft": [[155, null]], "mlx.core.fft.fft2": [[156, null]], "mlx.core.fft.fftn": [[157, null]], "mlx.core.fft.ifft": [[158, null]], "mlx.core.fft.ifft2": [[159, null]], "mlx.core.fft.ifftn": [[160, null]], "mlx.core.fft.irfft": [[161, null]], "mlx.core.fft.irfft2": [[162, null]], "mlx.core.fft.irfftn": [[163, null]], "mlx.core.fft.rfft": [[164, null]], "mlx.core.fft.rfft2": [[165, null]], "mlx.core.fft.rfftn": [[166, null]], "mlx.core.finfo": [[167, null]], "mlx.core.flatten": [[168, null]], "mlx.core.floor": [[169, null]], "mlx.core.floor_divide": [[170, null]], "mlx.core.full": [[171, null]], "mlx.core.gather_mm": [[172, null]], "mlx.core.gather_qmm": [[173, null]], "mlx.core.get_active_memory": [[174, null]], "mlx.core.get_cache_memory": [[175, null]], "mlx.core.get_peak_memory": [[176, null]], "mlx.core.grad": [[177, null]], "mlx.core.greater": [[178, null]], "mlx.core.greater_equal": [[179, null]], "mlx.core.hadamard_transform": [[180, null]], "mlx.core.identity": [[181, null]], "mlx.core.imag": [[182, null]], "mlx.core.import_function": [[183, null]], "mlx.core.inner": [[184, null]], "mlx.core.isclose": [[185, null]], "mlx.core.isfinite": [[186, null]], "mlx.core.isinf": [[187, null]], "mlx.core.isnan": [[188, null]], "mlx.core.isneginf": [[189, null]], "mlx.core.isposinf": [[190, null]], "mlx.core.issubdtype": [[191, null]], "mlx.core.jvp": [[192, null]], "mlx.core.kron": [[193, null]], "mlx.core.left_shift": [[194, null]], "mlx.core.less": [[195, null]], "mlx.core.less_equal": [[196, null]], "mlx.core.linalg.cholesky": [[197, null]], "mlx.core.linalg.cholesky_inv": [[198, null]], "mlx.core.linalg.cross": [[199, null]], "mlx.core.linalg.eigh": [[200, null]], "mlx.core.linalg.eigvalsh": [[201, null]], "mlx.core.linalg.inv": [[202, null]], "mlx.core.linalg.lu": [[203, null]], "mlx.core.linalg.lu_factor": [[204, null]], "mlx.core.linalg.norm": [[205, null]], "mlx.core.linalg.pinv": [[206, null]], "mlx.core.linalg.qr": [[207, null]], "mlx.core.linalg.solve": [[208, null]], "mlx.core.linalg.solve_triangular": [[209, null]], "mlx.core.linalg.svd": [[210, null]], "mlx.core.linalg.tri_inv": [[211, null]], "mlx.core.linspace": [[212, null]], "mlx.core.load": [[213, null]], "mlx.core.log": [[214, null]], "mlx.core.log10": [[215, null]], "mlx.core.log1p": [[216, null]], "mlx.core.log2": [[217, null]], "mlx.core.logaddexp": [[218, null]], "mlx.core.logical_and": [[219, null]], "mlx.core.logical_not": [[220, null]], "mlx.core.logical_or": [[221, null]], "mlx.core.logsumexp": [[222, null]], "mlx.core.matmul": [[223, null]], "mlx.core.max": [[224, null]], "mlx.core.maximum": [[225, null]], "mlx.core.mean": [[226, null]], "mlx.core.meshgrid": [[227, null]], "mlx.core.metal.device_info": [[228, null]], "mlx.core.metal.is_available": [[229, null]], "mlx.core.metal.start_capture": [[230, null]], "mlx.core.metal.stop_capture": [[231, null]], "mlx.core.min": [[232, null]], "mlx.core.minimum": [[233, null]], "mlx.core.moveaxis": [[234, null]], "mlx.core.multiply": [[235, null]], "mlx.core.nan_to_num": [[236, null]], "mlx.core.negative": [[237, null]], "mlx.core.new_stream": [[238, null]], "mlx.core.not_equal": [[239, null]], "mlx.core.ones": [[240, null]], "mlx.core.ones_like": [[241, null]], "mlx.core.outer": [[242, null]], "mlx.core.pad": [[243, null]], "mlx.core.partition": [[244, null]], "mlx.core.power": [[245, null]], "mlx.core.prod": [[246, null]], "mlx.core.put_along_axis": [[247, null]], "mlx.core.quantize": [[248, null]], "mlx.core.quantized_matmul": [[249, null]], "mlx.core.radians": [[250, null]], "mlx.core.random.bernoulli": [[251, null]], "mlx.core.random.categorical": [[252, null]], "mlx.core.random.gumbel": [[253, null]], "mlx.core.random.key": [[254, null]], "mlx.core.random.laplace": [[255, null]], "mlx.core.random.multivariate_normal": [[256, null]], "mlx.core.random.normal": [[257, null]], "mlx.core.random.permutation": [[258, null]], "mlx.core.random.randint": [[259, null]], "mlx.core.random.seed": [[260, null]], "mlx.core.random.split": [[261, null]], "mlx.core.random.truncated_normal": [[262, null]], "mlx.core.random.uniform": [[263, null]], "mlx.core.real": [[264, null]], "mlx.core.reciprocal": [[265, null]], "mlx.core.remainder": [[266, null]], "mlx.core.repeat": [[267, null]], "mlx.core.reset_peak_memory": [[268, null]], "mlx.core.reshape": [[269, null]], "mlx.core.right_shift": [[270, null]], "mlx.core.roll": [[271, null]], "mlx.core.round": [[272, null]], "mlx.core.rsqrt": [[273, null]], "mlx.core.save": [[274, null]], "mlx.core.save_gguf": [[275, null]], "mlx.core.save_safetensors": [[276, null]], "mlx.core.savez": [[277, null]], "mlx.core.savez_compressed": [[278, null]], "mlx.core.set_cache_limit": [[279, null]], "mlx.core.set_default_device": [[280, null]], "mlx.core.set_default_stream": [[281, null]], "mlx.core.set_memory_limit": [[282, null]], "mlx.core.set_wired_limit": [[283, null]], "mlx.core.sigmoid": [[284, null]], "mlx.core.sign": [[285, null]], "mlx.core.sin": [[286, null]], "mlx.core.sinh": [[287, null]], "mlx.core.slice": [[288, null]], "mlx.core.slice_update": [[289, null]], "mlx.core.softmax": [[290, null]], "mlx.core.sort": [[291, null]], "mlx.core.split": [[292, null]], "mlx.core.sqrt": [[293, null]], "mlx.core.square": [[294, null]], "mlx.core.squeeze": [[295, null]], "mlx.core.stack": [[296, null]], "mlx.core.std": [[297, null]], "mlx.core.stop_gradient": [[298, null]], "mlx.core.stream": [[299, null]], "mlx.core.subtract": [[300, null]], "mlx.core.sum": [[301, null]], "mlx.core.swapaxes": [[302, null]], "mlx.core.synchronize": [[303, null]], "mlx.core.take": [[304, null]], "mlx.core.take_along_axis": [[305, null]], "mlx.core.tan": [[306, null]], "mlx.core.tanh": [[307, null]], "mlx.core.tensordot": [[308, null]], "mlx.core.tile": [[309, null]], "mlx.core.topk": [[310, null]], "mlx.core.trace": [[311, null]], "mlx.core.transpose": [[312, null]], "mlx.core.tri": [[313, null]], "mlx.core.tril": [[314, null]], "mlx.core.triu": [[315, null]], "mlx.core.unflatten": [[316, null]], "mlx.core.value_and_grad": [[317, null]], "mlx.core.var": [[318, null]], "mlx.core.view": [[319, null]], "mlx.core.vjp": [[320, null]], "mlx.core.vmap": [[321, null]], "mlx.core.where": [[322, null]], "mlx.core.zeros": [[323, null]], "mlx.core.zeros_like": [[324, null]], "mlx.nn.ALiBi": [[346, null]], "mlx.nn.AvgPool1d": [[347, null]], "mlx.nn.AvgPool2d": [[348, null]], "mlx.nn.AvgPool3d": [[349, null]], "mlx.nn.BatchNorm": [[350, null]], "mlx.nn.CELU": [[351, null]], "mlx.nn.Conv1d": [[352, null]], "mlx.nn.Conv2d": [[353, null]], "mlx.nn.Conv3d": [[354, null]], "mlx.nn.ConvTranspose1d": [[355, null]], "mlx.nn.ConvTranspose2d": [[356, null]], "mlx.nn.ConvTranspose3d": [[357, null]], "mlx.nn.Dropout": [[358, null]], "mlx.nn.Dropout2d": [[359, null]], "mlx.nn.Dropout3d": [[360, null]], "mlx.nn.ELU": [[361, null]], "mlx.nn.Embedding": [[362, null]], "mlx.nn.GELU": [[363, null]], "mlx.nn.GLU": [[364, null]], "mlx.nn.GRU": [[365, null]], "mlx.nn.GroupNorm": [[366, null]], "mlx.nn.HardShrink": [[367, null]], "mlx.nn.HardTanh": [[368, null]], "mlx.nn.Hardswish": [[369, null]], "mlx.nn.InstanceNorm": [[370, null]], "mlx.nn.LSTM": [[371, null]], "mlx.nn.LayerNorm": [[372, null]], "mlx.nn.LeakyReLU": [[373, null]], "mlx.nn.Linear": [[374, null]], "mlx.nn.LogSigmoid": [[375, null]], "mlx.nn.LogSoftmax": [[376, null]], "mlx.nn.MaxPool1d": [[377, null]], "mlx.nn.MaxPool2d": [[378, null]], "mlx.nn.MaxPool3d": [[379, null]], "mlx.nn.Mish": [[380, null]], "mlx.nn.Module.apply": [[381, null]], "mlx.nn.Module.apply_to_modules": [[382, null]], "mlx.nn.Module.children": [[383, null]], "mlx.nn.Module.eval": [[384, null]], "mlx.nn.Module.filter_and_map": [[385, null]], "mlx.nn.Module.freeze": [[386, null]], "mlx.nn.Module.leaf_modules": [[387, null]], "mlx.nn.Module.load_weights": [[388, null]], "mlx.nn.Module.modules": [[389, null]], "mlx.nn.Module.named_modules": [[390, null]], "mlx.nn.Module.parameters": [[391, null]], "mlx.nn.Module.save_weights": [[392, null]], "mlx.nn.Module.set_dtype": [[393, null]], "mlx.nn.Module.state": [[394, null]], "mlx.nn.Module.train": [[395, null]], "mlx.nn.Module.trainable_parameters": [[396, null]], "mlx.nn.Module.training": [[397, null]], "mlx.nn.Module.unfreeze": [[398, null]], "mlx.nn.Module.update": [[399, null]], "mlx.nn.Module.update_modules": [[400, null]], "mlx.nn.MultiHeadAttention": [[401, null]], "mlx.nn.PReLU": [[402, null]], "mlx.nn.QuantizedEmbedding": [[403, null]], "mlx.nn.QuantizedLinear": [[404, null]], "mlx.nn.RMSNorm": [[405, null]], "mlx.nn.RNN": [[406, null]], "mlx.nn.ReLU": [[407, null]], "mlx.nn.ReLU6": [[408, null]], "mlx.nn.RoPE": [[409, null]], "mlx.nn.SELU": [[410, null]], "mlx.nn.Sequential": [[411, null]], "mlx.nn.SiLU": [[412, null]], "mlx.nn.Sigmoid": [[413, null]], "mlx.nn.SinusoidalPositionalEncoding": [[414, null]], "mlx.nn.Softmax": [[415, null]], "mlx.nn.Softmin": [[416, null]], "mlx.nn.Softplus": [[417, null]], "mlx.nn.Softshrink": [[418, null]], "mlx.nn.Softsign": [[419, null]], "mlx.nn.Step": [[420, null]], "mlx.nn.Tanh": [[421, null]], "mlx.nn.Transformer": [[422, null]], "mlx.nn.Upsample": [[423, null]], "mlx.nn.average_gradients": [[325, null]], "mlx.nn.celu": [[432, null]], "mlx.nn.elu": [[433, null]], "mlx.nn.gelu": [[434, null]], "mlx.nn.gelu_approx": [[435, null]], "mlx.nn.gelu_fast_approx": [[436, null]], "mlx.nn.glu": [[437, null]], "mlx.nn.hard_shrink": [[438, null]], "mlx.nn.hard_tanh": [[439, null]], "mlx.nn.hardswish": [[440, null]], "mlx.nn.init.constant": [[424, null]], "mlx.nn.init.glorot_normal": [[425, null]], "mlx.nn.init.glorot_uniform": [[426, null]], "mlx.nn.init.he_normal": [[427, null]], "mlx.nn.init.he_uniform": [[428, null]], "mlx.nn.init.identity": [[429, null]], "mlx.nn.init.normal": [[430, null]], "mlx.nn.init.uniform": [[431, null]], "mlx.nn.leaky_relu": [[441, null]], "mlx.nn.log_sigmoid": [[442, null]], "mlx.nn.log_softmax": [[443, null]], "mlx.nn.losses.binary_cross_entropy": [[444, null]], "mlx.nn.losses.cosine_similarity_loss": [[445, null]], "mlx.nn.losses.cross_entropy": [[446, null]], "mlx.nn.losses.gaussian_nll_loss": [[447, null]], "mlx.nn.losses.hinge_loss": [[448, null]], "mlx.nn.losses.huber_loss": [[449, null]], "mlx.nn.losses.kl_div_loss": [[450, null]], "mlx.nn.losses.l1_loss": [[451, null]], "mlx.nn.losses.log_cosh_loss": [[452, null]], "mlx.nn.losses.margin_ranking_loss": [[453, null]], "mlx.nn.losses.mse_loss": [[454, null]], "mlx.nn.losses.nll_loss": [[455, null]], "mlx.nn.losses.smooth_l1_loss": [[456, null]], "mlx.nn.losses.triplet_loss": [[457, null]], "mlx.nn.mish": [[458, null]], "mlx.nn.prelu": [[459, null]], "mlx.nn.quantize": [[326, null]], "mlx.nn.relu": [[460, null]], "mlx.nn.relu6": [[461, null]], "mlx.nn.selu": [[462, null]], "mlx.nn.sigmoid": [[463, null]], "mlx.nn.silu": [[464, null]], "mlx.nn.softmax": [[465, null]], "mlx.nn.softmin": [[466, null]], "mlx.nn.softplus": [[467, null]], "mlx.nn.softshrink": [[468, null]], "mlx.nn.step": [[469, null]], "mlx.nn.tanh": [[470, null]], "mlx.nn.value_and_grad": [[327, null]], "mlx.optimizers.AdaDelta": [[478, null]], "mlx.optimizers.Adafactor": [[479, null]], "mlx.optimizers.Adagrad": [[480, null]], "mlx.optimizers.Adam": [[481, null]], "mlx.optimizers.AdamW": [[482, null]], "mlx.optimizers.Adamax": [[483, null]], "mlx.optimizers.Lion": [[484, null]], "mlx.optimizers.Optimizer.apply_gradients": [[485, null]], "mlx.optimizers.Optimizer.init": [[486, null]], "mlx.optimizers.Optimizer.state": [[487, null]], "mlx.optimizers.Optimizer.update": [[488, null]], "mlx.optimizers.RMSprop": [[489, null]], "mlx.optimizers.SGD": [[490, null]], "mlx.optimizers.clip_grad_norm": [[328, null]], "mlx.optimizers.cosine_decay": [[491, null]], "mlx.optimizers.exponential_decay": [[492, null]], "mlx.optimizers.join_schedules": [[493, null]], "mlx.optimizers.linear_schedule": [[494, null]], "mlx.optimizers.step_decay": [[495, null]], "mlx.utils.tree_flatten": [[329, null]], "mlx.utils.tree_map": [[330, null]], "mlx.utils.tree_map_with_path": [[331, null]], "mlx.utils.tree_reduce": [[332, null]], "mlx.utils.tree_unflatten": [[333, null]], "x86 Shell": [[9, "x86-shell"]]}, "docnames": ["cpp/ops", "dev/custom_metal_kernels", "dev/extensions", "dev/metal_debugger", "dev/mlx_in_cpp", "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.std", "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.async_eval", "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_invert", "python/_autosummary/mlx.core.bitwise_or", "python/_autosummary/mlx.core.bitwise_xor", "python/_autosummary/mlx.core.block_masked_mm", "python/_autosummary/mlx.core.broadcast_arrays", "python/_autosummary/mlx.core.broadcast_to", "python/_autosummary/mlx.core.ceil", "python/_autosummary/mlx.core.clear_cache", "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.contiguous", "python/_autosummary/mlx.core.conv1d", "python/_autosummary/mlx.core.conv2d", "python/_autosummary/mlx.core.conv3d", "python/_autosummary/mlx.core.conv_general", "python/_autosummary/mlx.core.conv_transpose1d", "python/_autosummary/mlx.core.conv_transpose2d", "python/_autosummary/mlx.core.conv_transpose3d", "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.distributed.recv", "python/_autosummary/mlx.core.distributed.recv_like", "python/_autosummary/mlx.core.distributed.send", "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.export_function", "python/_autosummary/mlx.core.export_to_dot", "python/_autosummary/mlx.core.exporter", "python/_autosummary/mlx.core.eye", "python/_autosummary/mlx.core.fast.layer_norm", "python/_autosummary/mlx.core.fast.metal_kernel", "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.finfo", "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.get_active_memory", "python/_autosummary/mlx.core.get_cache_memory", "python/_autosummary/mlx.core.get_peak_memory", "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.imag", "python/_autosummary/mlx.core.import_function", "python/_autosummary/mlx.core.inner", "python/_autosummary/mlx.core.isclose", "python/_autosummary/mlx.core.isfinite", "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.kron", "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.cross", "python/_autosummary/mlx.core.linalg.eigh", "python/_autosummary/mlx.core.linalg.eigvalsh", "python/_autosummary/mlx.core.linalg.inv", "python/_autosummary/mlx.core.linalg.lu", "python/_autosummary/mlx.core.linalg.lu_factor", "python/_autosummary/mlx.core.linalg.norm", "python/_autosummary/mlx.core.linalg.pinv", "python/_autosummary/mlx.core.linalg.qr", "python/_autosummary/mlx.core.linalg.solve", "python/_autosummary/mlx.core.linalg.solve_triangular", "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.device_info", "python/_autosummary/mlx.core.metal.is_available", "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.put_along_axis", "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.permutation", "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.real", "python/_autosummary/mlx.core.reciprocal", "python/_autosummary/mlx.core.remainder", "python/_autosummary/mlx.core.repeat", "python/_autosummary/mlx.core.reset_peak_memory", "python/_autosummary/mlx.core.reshape", "python/_autosummary/mlx.core.right_shift", "python/_autosummary/mlx.core.roll", "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_cache_limit", "python/_autosummary/mlx.core.set_default_device", "python/_autosummary/mlx.core.set_default_stream", "python/_autosummary/mlx.core.set_memory_limit", "python/_autosummary/mlx.core.set_wired_limit", "python/_autosummary/mlx.core.sigmoid", "python/_autosummary/mlx.core.sign", "python/_autosummary/mlx.core.sin", "python/_autosummary/mlx.core.sinh", "python/_autosummary/mlx.core.slice", "python/_autosummary/mlx.core.slice_update", "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.unflatten", "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.average_gradients", "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/export", "python/fast", "python/fft", "python/linalg", "python/memory_management", "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.AvgPool3d", "python/nn/_autosummary/mlx.nn.BatchNorm", "python/nn/_autosummary/mlx.nn.CELU", "python/nn/_autosummary/mlx.nn.Conv1d", "python/nn/_autosummary/mlx.nn.Conv2d", "python/nn/_autosummary/mlx.nn.Conv3d", "python/nn/_autosummary/mlx.nn.ConvTranspose1d", "python/nn/_autosummary/mlx.nn.ConvTranspose2d", "python/nn/_autosummary/mlx.nn.ConvTranspose3d", "python/nn/_autosummary/mlx.nn.Dropout", "python/nn/_autosummary/mlx.nn.Dropout2d", "python/nn/_autosummary/mlx.nn.Dropout3d", "python/nn/_autosummary/mlx.nn.ELU", "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.LogSigmoid", "python/nn/_autosummary/mlx.nn.LogSoftmax", "python/nn/_autosummary/mlx.nn.MaxPool1d", "python/nn/_autosummary/mlx.nn.MaxPool2d", "python/nn/_autosummary/mlx.nn.MaxPool3d", "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.Sigmoid", "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.celu", "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/export", "usage/function_transforms", "usage/indexing", "usage/launching_distributed", "usage/lazy_evaluation", "usage/numpy", "usage/quick_start", "usage/saving_and_loading", "usage/unified_memory", "usage/using_streams"], "envversion": {"sphinx": 62, "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}, "filenames": ["cpp/ops.rst", "dev/custom_metal_kernels.rst", "dev/extensions.rst", "dev/metal_debugger.rst", "dev/mlx_in_cpp.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.std.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.async_eval.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_invert.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_arrays.rst", "python/_autosummary/mlx.core.broadcast_to.rst", "python/_autosummary/mlx.core.ceil.rst", "python/_autosummary/mlx.core.clear_cache.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.contiguous.rst", "python/_autosummary/mlx.core.conv1d.rst", "python/_autosummary/mlx.core.conv2d.rst", "python/_autosummary/mlx.core.conv3d.rst", "python/_autosummary/mlx.core.conv_general.rst", "python/_autosummary/mlx.core.conv_transpose1d.rst", "python/_autosummary/mlx.core.conv_transpose2d.rst", "python/_autosummary/mlx.core.conv_transpose3d.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.distributed.recv.rst", "python/_autosummary/mlx.core.distributed.recv_like.rst", "python/_autosummary/mlx.core.distributed.send.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.export_function.rst", "python/_autosummary/mlx.core.export_to_dot.rst", "python/_autosummary/mlx.core.exporter.rst", "python/_autosummary/mlx.core.eye.rst", "python/_autosummary/mlx.core.fast.layer_norm.rst", "python/_autosummary/mlx.core.fast.metal_kernel.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.finfo.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.get_active_memory.rst", "python/_autosummary/mlx.core.get_cache_memory.rst", "python/_autosummary/mlx.core.get_peak_memory.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.imag.rst", "python/_autosummary/mlx.core.import_function.rst", "python/_autosummary/mlx.core.inner.rst", "python/_autosummary/mlx.core.isclose.rst", "python/_autosummary/mlx.core.isfinite.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.kron.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.cross.rst", "python/_autosummary/mlx.core.linalg.eigh.rst", "python/_autosummary/mlx.core.linalg.eigvalsh.rst", "python/_autosummary/mlx.core.linalg.inv.rst", "python/_autosummary/mlx.core.linalg.lu.rst", "python/_autosummary/mlx.core.linalg.lu_factor.rst", "python/_autosummary/mlx.core.linalg.norm.rst", "python/_autosummary/mlx.core.linalg.pinv.rst", "python/_autosummary/mlx.core.linalg.qr.rst", "python/_autosummary/mlx.core.linalg.solve.rst", "python/_autosummary/mlx.core.linalg.solve_triangular.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.device_info.rst", "python/_autosummary/mlx.core.metal.is_available.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.put_along_axis.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.permutation.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.real.rst", "python/_autosummary/mlx.core.reciprocal.rst", "python/_autosummary/mlx.core.remainder.rst", "python/_autosummary/mlx.core.repeat.rst", "python/_autosummary/mlx.core.reset_peak_memory.rst", "python/_autosummary/mlx.core.reshape.rst", "python/_autosummary/mlx.core.right_shift.rst", "python/_autosummary/mlx.core.roll.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_cache_limit.rst", "python/_autosummary/mlx.core.set_default_device.rst", "python/_autosummary/mlx.core.set_default_stream.rst", "python/_autosummary/mlx.core.set_memory_limit.rst", "python/_autosummary/mlx.core.set_wired_limit.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.slice.rst", "python/_autosummary/mlx.core.slice_update.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.unflatten.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.average_gradients.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/export.rst", "python/fast.rst", "python/fft.rst", "python/linalg.rst", "python/memory_management.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.AvgPool3d.rst", "python/nn/_autosummary/mlx.nn.BatchNorm.rst", "python/nn/_autosummary/mlx.nn.CELU.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.ConvTranspose1d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose2d.rst", "python/nn/_autosummary/mlx.nn.ConvTranspose3d.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.ELU.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.LogSigmoid.rst", "python/nn/_autosummary/mlx.nn.LogSoftmax.rst", "python/nn/_autosummary/mlx.nn.MaxPool1d.rst", "python/nn/_autosummary/mlx.nn.MaxPool2d.rst", "python/nn/_autosummary/mlx.nn.MaxPool3d.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.Sigmoid.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.celu.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/export.rst", "usage/function_transforms.rst", "usage/indexing.rst", "usage/launching_distributed.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"], "indexentries": {"__init__() (array method)": [[31, "mlx.core.array.__init__", false]], "__init__() (custom_function method)": [[118, "mlx.core.custom_function.__init__", false]], "__init__() (device method)": [[10, "mlx.core.Device.__init__", false]], "__init__() (dtype method)": [[11, "mlx.core.Dtype.__init__", false]], "__init__() (dtypecategory method)": [[12, "mlx.core.DtypeCategory.__init__", false]], "__init__() (finfo method)": [[167, "mlx.core.finfo.__init__", false]], "__init__() (group method)": [[126, "mlx.core.distributed.Group.__init__", false]], "__init__() (stream method)": [[334, "mlx.core.Stream.__init__", false]], "abs (c++ function)": [[0, "_CPPv43absRK5array14StreamOrDevice", false]], "abs() (array method)": [[33, "mlx.core.array.abs", false]], "abs() (in module mlx.core)": [[13, "mlx.core.abs", false]], "adadelta (class in mlx.optimizers)": [[478, "mlx.optimizers.AdaDelta", false]], "adafactor (class in mlx.optimizers)": [[479, "mlx.optimizers.Adafactor", false]], "adagrad (class in mlx.optimizers)": [[480, "mlx.optimizers.Adagrad", false]], "adam (class in mlx.optimizers)": [[481, "mlx.optimizers.Adam", false]], "adamax (class in mlx.optimizers)": [[483, "mlx.optimizers.Adamax", false]], "adamw (class in mlx.optimizers)": [[482, "mlx.optimizers.AdamW", false]], "add (c++ function)": [[0, "_CPPv43addRK5arrayRK5array14StreamOrDevice", false]], "add() (in module mlx.core)": [[14, "mlx.core.add", false]], "addmm (c++ function)": [[0, "_CPPv45addmm5array5array5arrayRKfRKf14StreamOrDevice", false]], "addmm() (in module mlx.core)": [[15, "mlx.core.addmm", false]], "alibi (class in mlx.nn)": [[346, "mlx.nn.ALiBi", false]], "all (c++ function)": [[0, "_CPPv43allRK5array14StreamOrDevice", false], [0, "_CPPv43allRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43allRK5arrayb14StreamOrDevice", false], [0, "_CPPv43allRK5arrayib14StreamOrDevice", false]], "all() (array method)": [[34, "mlx.core.array.all", false]], "all() (in module mlx.core)": [[16, "mlx.core.all", false]], "all_gather() (in module mlx.core.distributed)": [[127, "mlx.core.distributed.all_gather", false]], "all_sum() (in module mlx.core.distributed)": [[128, "mlx.core.distributed.all_sum", false]], "allclose (c++ function)": [[0, "_CPPv48allcloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "allclose() (in module mlx.core)": [[17, "mlx.core.allclose", false]], "any (c++ function)": [[0, "_CPPv43anyRK5array14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayb14StreamOrDevice", false], [0, "_CPPv43anyRK5arrayib14StreamOrDevice", false]], "any() (array method)": [[35, "mlx.core.array.any", false]], "any() (in module mlx.core)": [[18, "mlx.core.any", false]], "apply() (module method)": [[381, "mlx.nn.Module.apply", false]], "apply_gradients() (optimizer method)": [[485, "mlx.optimizers.Optimizer.apply_gradients", false]], "apply_to_modules() (module method)": [[382, "mlx.nn.Module.apply_to_modules", false]], "arange (c++ function)": [[0, "_CPPv46aranged14StreamOrDevice", false], [0, "_CPPv46aranged5Dtype14StreamOrDevice", false], [0, "_CPPv46arangedd14StreamOrDevice", false], [0, "_CPPv46arangedd5Dtype14StreamOrDevice", false], [0, "_CPPv46arangeddd14StreamOrDevice", false], [0, "_CPPv46arangeddd5Dtype14StreamOrDevice", false], [0, "_CPPv46arangei14StreamOrDevice", false], [0, "_CPPv46arangeii14StreamOrDevice", false], [0, "_CPPv46arangeiii14StreamOrDevice", false]], "arange() (in module mlx.core)": [[19, "mlx.core.arange", false]], "arccos (c++ function)": [[0, "_CPPv46arccosRK5array14StreamOrDevice", false]], "arccos() (in module mlx.core)": [[20, "mlx.core.arccos", false]], "arccosh (c++ function)": [[0, "_CPPv47arccoshRK5array14StreamOrDevice", false]], "arccosh() (in module mlx.core)": [[21, "mlx.core.arccosh", false]], "arcsin (c++ function)": [[0, "_CPPv46arcsinRK5array14StreamOrDevice", false]], "arcsin() (in module mlx.core)": [[22, "mlx.core.arcsin", false]], "arcsinh (c++ function)": [[0, "_CPPv47arcsinhRK5array14StreamOrDevice", false]], "arcsinh() (in module mlx.core)": [[23, "mlx.core.arcsinh", false]], "arctan (c++ function)": [[0, "_CPPv46arctanRK5array14StreamOrDevice", false]], "arctan() (in module mlx.core)": [[24, "mlx.core.arctan", false]], "arctan2 (c++ function)": [[0, "_CPPv47arctan2RK5arrayRK5array14StreamOrDevice", false]], "arctan2() (in module mlx.core)": [[25, "mlx.core.arctan2", false]], "arctanh (c++ function)": [[0, "_CPPv47arctanhRK5array14StreamOrDevice", false]], "arctanh() (in module mlx.core)": [[26, "mlx.core.arctanh", false]], "argmax (c++ function)": [[0, "_CPPv46argmaxRK5array14StreamOrDevice", false], [0, "_CPPv46argmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv46argmaxRK5arrayib14StreamOrDevice", false]], "argmax() (array method)": [[36, "mlx.core.array.argmax", false]], "argmax() (in module mlx.core)": [[27, "mlx.core.argmax", false]], "argmin (c++ function)": [[0, "_CPPv46argminRK5array14StreamOrDevice", false], [0, "_CPPv46argminRK5arrayb14StreamOrDevice", false], [0, "_CPPv46argminRK5arrayib14StreamOrDevice", false]], "argmin() (array method)": [[37, "mlx.core.array.argmin", false]], "argmin() (in module mlx.core)": [[28, "mlx.core.argmin", false]], "argpartition (c++ function)": [[0, "_CPPv412argpartitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv412argpartitionRK5arrayii14StreamOrDevice", false]], "argpartition() (in module mlx.core)": [[29, "mlx.core.argpartition", false]], "argsort (c++ function)": [[0, "_CPPv47argsortRK5array14StreamOrDevice", false], [0, "_CPPv47argsortRK5arrayi14StreamOrDevice", false]], "argsort() (in module mlx.core)": [[30, "mlx.core.argsort", false]], "array (class in mlx.core)": [[31, "mlx.core.array", false]], "array_equal (c++ function)": [[0, "_CPPv411array_equalRK5arrayRK5array14StreamOrDevice", false], [0, "_CPPv411array_equalRK5arrayRK5arrayb14StreamOrDevice", false]], "array_equal() (in module mlx.core)": [[83, "mlx.core.array_equal", false]], "as_strided (c++ function)": [[0, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", false]], "as_strided() (in module mlx.core)": [[84, "mlx.core.as_strided", false]], "astype (c++ function)": [[0, "_CPPv46astype5array5Dtype14StreamOrDevice", false]], "astype() (array method)": [[38, "mlx.core.array.astype", false]], "async_eval() (in module mlx.core)": [[85, "mlx.core.async_eval", false]], "at (array property)": [[39, "mlx.core.array.at", false]], "atleast_1d (c++ function)": [[0, "_CPPv410atleast_1dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_1dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_1d() (in module mlx.core)": [[86, "mlx.core.atleast_1d", false]], "atleast_2d (c++ function)": [[0, "_CPPv410atleast_2dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_2d() (in module mlx.core)": [[87, "mlx.core.atleast_2d", false]], "atleast_3d (c++ function)": [[0, "_CPPv410atleast_3dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_3d() (in module mlx.core)": [[88, "mlx.core.atleast_3d", false]], "average_gradients() (in module mlx.nn)": [[325, "mlx.nn.average_gradients", false]], "avgpool1d (class in mlx.nn)": [[347, "mlx.nn.AvgPool1d", false]], "avgpool2d (class in mlx.nn)": [[348, "mlx.nn.AvgPool2d", false]], "avgpool3d (class in mlx.nn)": [[349, "mlx.nn.AvgPool3d", false]], "batchnorm (class in mlx.nn)": [[350, "mlx.nn.BatchNorm", false]], "bernoulli() (in module mlx.core.random)": [[251, "mlx.core.random.bernoulli", false]], "binary_cross_entropy (class in mlx.nn.losses)": [[444, "mlx.nn.losses.binary_cross_entropy", false]], "bitwise_and (c++ function)": [[0, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", false]], "bitwise_and() (in module mlx.core)": [[89, "mlx.core.bitwise_and", false]], "bitwise_invert (c++ function)": [[0, "_CPPv414bitwise_invertRK5array14StreamOrDevice", false]], "bitwise_invert() (in module mlx.core)": [[90, "mlx.core.bitwise_invert", false]], "bitwise_or (c++ function)": [[0, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", false]], "bitwise_or() (in module mlx.core)": [[91, "mlx.core.bitwise_or", false]], "bitwise_xor (c++ function)": [[0, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", false]], "bitwise_xor() (in module mlx.core)": [[92, "mlx.core.bitwise_xor", false]], "block_masked_mm (c++ function)": [[0, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "block_masked_mm() (in module mlx.core)": [[93, "mlx.core.block_masked_mm", false]], "broadcast_arrays (c++ function)": [[0, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "broadcast_arrays() (in module mlx.core)": [[94, "mlx.core.broadcast_arrays", false]], "broadcast_to (c++ function)": [[0, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", false]], "broadcast_to() (in module mlx.core)": [[95, "mlx.core.broadcast_to", false]], "categorical() (in module mlx.core.random)": [[252, "mlx.core.random.categorical", false]], "ceil (c++ function)": [[0, "_CPPv44ceilRK5array14StreamOrDevice", false]], "ceil() (in module mlx.core)": [[96, "mlx.core.ceil", false]], "celu (class in mlx.nn)": [[351, "mlx.nn.CELU", false], [432, "mlx.nn.celu", false]], "children() (module method)": [[383, "mlx.nn.Module.children", false]], "cholesky() (in module mlx.core.linalg)": [[197, "mlx.core.linalg.cholesky", false]], "cholesky_inv() (in module mlx.core.linalg)": [[198, "mlx.core.linalg.cholesky_inv", false]], "clear_cache() (in module mlx.core)": [[97, "mlx.core.clear_cache", false]], "clip (c++ function)": [[0, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", false]], "clip() (in module mlx.core)": [[98, "mlx.core.clip", false]], "clip_grad_norm() (in module mlx.optimizers)": [[328, "mlx.optimizers.clip_grad_norm", false]], "compile() (in module mlx.core)": [[99, "mlx.core.compile", false]], "concatenate (c++ function)": [[0, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", false]], "concatenate() (in module mlx.core)": [[100, "mlx.core.concatenate", false]], "conj() (array method)": [[40, "mlx.core.array.conj", false]], "conj() (in module mlx.core)": [[101, "mlx.core.conj", false]], "conjugate (c++ function)": [[0, "_CPPv49conjugateRK5array14StreamOrDevice", false]], "conjugate() (in module mlx.core)": [[102, "mlx.core.conjugate", false]], "constant() (in module mlx.nn.init)": [[424, "mlx.nn.init.constant", false]], "contiguous (c++ function)": [[0, "_CPPv410contiguousRK5arrayb14StreamOrDevice", false]], "contiguous() (in module mlx.core)": [[103, "mlx.core.contiguous", false]], "conv1d (c++ function)": [[0, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv1d (class in mlx.nn)": [[352, "mlx.nn.Conv1d", false]], "conv1d() (in module mlx.core)": [[104, "mlx.core.conv1d", false]], "conv2d (c++ function)": [[0, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv2d (class in mlx.nn)": [[353, "mlx.nn.Conv2d", false]], "conv2d() (in module mlx.core)": [[105, "mlx.core.conv2d", false]], "conv3d (c++ function)": [[0, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv3d (class in mlx.nn)": [[354, "mlx.nn.Conv3d", false]], "conv3d() (in module mlx.core)": [[106, "mlx.core.conv3d", false]], "conv_general (c++ function)": [[0, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false], [0, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false]], "conv_general() (in module mlx.core)": [[107, "mlx.core.conv_general", false]], "conv_transpose1d (c++ function)": [[0, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv_transpose1d() (in module mlx.core)": [[108, "mlx.core.conv_transpose1d", false]], "conv_transpose2d (c++ function)": [[0, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv_transpose2d() (in module mlx.core)": [[109, "mlx.core.conv_transpose2d", false]], "conv_transpose3d (c++ function)": [[0, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv_transpose3d() (in module mlx.core)": [[110, "mlx.core.conv_transpose3d", false]], "convolve() (in module mlx.core)": [[111, "mlx.core.convolve", false]], "convtranspose1d (class in mlx.nn)": [[355, "mlx.nn.ConvTranspose1d", false]], "convtranspose2d (class in mlx.nn)": [[356, "mlx.nn.ConvTranspose2d", false]], "convtranspose3d (class in mlx.nn)": [[357, "mlx.nn.ConvTranspose3d", false]], "copy (c++ function)": [[0, "_CPPv44copy5array14StreamOrDevice", false]], "cos (c++ function)": [[0, "_CPPv43cosRK5array14StreamOrDevice", false]], "cos() (array method)": [[41, "mlx.core.array.cos", false]], "cos() (in module mlx.core)": [[112, "mlx.core.cos", false]], "cosh (c++ function)": [[0, "_CPPv44coshRK5array14StreamOrDevice", false]], "cosh() (in module mlx.core)": [[113, "mlx.core.cosh", false]], "cosine_decay() (in module mlx.optimizers)": [[491, "mlx.optimizers.cosine_decay", false]], "cosine_similarity_loss (class in mlx.nn.losses)": [[445, "mlx.nn.losses.cosine_similarity_loss", false]], "cross() (in module mlx.core.linalg)": [[199, "mlx.core.linalg.cross", false]], "cross_entropy (class in mlx.nn.losses)": [[446, "mlx.nn.losses.cross_entropy", false]], "cummax (c++ function)": [[0, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", false]], "cummax() (array method)": [[42, "mlx.core.array.cummax", false]], "cummax() (in module mlx.core)": [[114, "mlx.core.cummax", false]], "cummin (c++ function)": [[0, "_CPPv46cumminRK5arrayibb14StreamOrDevice", false]], "cummin() (array method)": [[43, "mlx.core.array.cummin", false]], "cummin() (in module mlx.core)": [[115, "mlx.core.cummin", false]], "cumprod (c++ function)": [[0, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", false]], "cumprod() (array method)": [[44, "mlx.core.array.cumprod", false]], "cumprod() (in module mlx.core)": [[116, "mlx.core.cumprod", false]], "cumsum (c++ function)": [[0, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", false]], "cumsum() (array method)": [[45, "mlx.core.array.cumsum", false]], "cumsum() (in module mlx.core)": [[117, "mlx.core.cumsum", false]], "custom_function (class in mlx.core)": [[118, "mlx.core.custom_function", false]], "default_device() (in module mlx.core)": [[119, "mlx.core.default_device", false]], "default_stream() (in module mlx.core)": [[120, "mlx.core.default_stream", false]], "degrees (c++ function)": [[0, "_CPPv47degreesRK5array14StreamOrDevice", false]], "degrees() (in module mlx.core)": [[121, "mlx.core.degrees", false]], "depends (c++ function)": [[0, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", false]], "dequantize (c++ function)": [[0, "_CPPv410dequantizeRK5arrayRK5arrayRK5arrayii14StreamOrDevice", false]], "dequantize() (in module mlx.core)": [[122, "mlx.core.dequantize", false]], "device (class in mlx.core)": [[10, "mlx.core.Device", false]], "device_info() (in module mlx.core.metal)": [[228, "mlx.core.metal.device_info", false]], "diag (c++ function)": [[0, "_CPPv44diagRK5arrayi14StreamOrDevice", false]], "diag() (array method)": [[46, "mlx.core.array.diag", false]], "diag() (in module mlx.core)": [[123, "mlx.core.diag", false]], "diagonal (c++ function)": [[0, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", false]], "diagonal() (array method)": [[47, "mlx.core.array.diagonal", false]], "diagonal() (in module mlx.core)": [[124, "mlx.core.diagonal", false]], "disable_compile() (in module mlx.core)": [[125, "mlx.core.disable_compile", false]], "divide (c++ function)": [[0, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", false]], "divide() (in module mlx.core)": [[134, "mlx.core.divide", false]], "divmod (c++ function)": [[0, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", false]], "divmod() (in module mlx.core)": [[135, "mlx.core.divmod", false]], "dropout (class in mlx.nn)": [[358, "mlx.nn.Dropout", false]], "dropout2d (class in mlx.nn)": [[359, "mlx.nn.Dropout2d", false]], "dropout3d (class in mlx.nn)": [[360, "mlx.nn.Dropout3d", false]], "dtype (array property)": [[48, "mlx.core.array.dtype", false]], "dtype (class in mlx.core)": [[11, "mlx.core.Dtype", false]], "dtypecategory (class in mlx.core)": [[12, "mlx.core.DtypeCategory", false]], "eigh() (in module mlx.core.linalg)": [[200, "mlx.core.linalg.eigh", false]], "eigvalsh() (in module mlx.core.linalg)": [[201, "mlx.core.linalg.eigvalsh", false]], "einsum() (in module mlx.core)": [[136, "mlx.core.einsum", false]], "einsum_path() (in module mlx.core)": [[137, "mlx.core.einsum_path", false]], "elu (class in mlx.nn)": [[361, "mlx.nn.ELU", false], [433, "mlx.nn.elu", false]], "embedding (class in mlx.nn)": [[362, "mlx.nn.Embedding", false]], "enable_compile() (in module mlx.core)": [[138, "mlx.core.enable_compile", false]], "equal (c++ function)": [[0, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", false]], "equal() (in module mlx.core)": [[139, "mlx.core.equal", false]], "erf (c++ function)": [[0, "_CPPv43erfRK5array14StreamOrDevice", false]], "erf() (in module mlx.core)": [[140, "mlx.core.erf", false]], "erfinv (c++ function)": [[0, "_CPPv46erfinvRK5array14StreamOrDevice", false]], "erfinv() (in module mlx.core)": [[141, "mlx.core.erfinv", false]], "eval() (in module mlx.core)": [[142, "mlx.core.eval", false]], "eval() (module method)": [[384, "mlx.nn.Module.eval", false]], "exp (c++ function)": [[0, "_CPPv43expRK5array14StreamOrDevice", false]], "exp() (array method)": [[49, "mlx.core.array.exp", false]], "exp() (in module mlx.core)": [[143, "mlx.core.exp", false]], "expand_dims (c++ function)": [[0, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", false]], "expand_dims() (in module mlx.core)": [[144, "mlx.core.expand_dims", false]], "expm1 (c++ function)": [[0, "_CPPv45expm1RK5array14StreamOrDevice", false]], "expm1() (in module mlx.core)": [[145, "mlx.core.expm1", false]], "exponential_decay() (in module mlx.optimizers)": [[492, "mlx.optimizers.exponential_decay", false]], "export_function() (in module mlx.core)": [[146, "mlx.core.export_function", false]], "export_to_dot() (in module mlx.core)": [[147, "mlx.core.export_to_dot", false]], "exporter() (in module mlx.core)": [[148, "mlx.core.exporter", false]], "eye (c++ function)": [[0, "_CPPv43eyei14StreamOrDevice", false], [0, "_CPPv43eyei5Dtype14StreamOrDevice", false], [0, "_CPPv43eyeii14StreamOrDevice", false], [0, "_CPPv43eyeiii14StreamOrDevice", false], [0, "_CPPv43eyeiii5Dtype14StreamOrDevice", false]], "eye() (in module mlx.core)": [[149, "mlx.core.eye", false]], "fft() (in module mlx.core.fft)": [[155, "mlx.core.fft.fft", false]], "fft2() (in module mlx.core.fft)": [[156, "mlx.core.fft.fft2", false]], "fftn() (in module mlx.core.fft)": [[157, "mlx.core.fft.fftn", false]], "filter_and_map() (module method)": [[385, "mlx.nn.Module.filter_and_map", false]], "finfo (class in mlx.core)": [[167, "mlx.core.finfo", false]], "flatten (c++ function)": [[0, "_CPPv47flattenRK5array14StreamOrDevice", false], [0, "_CPPv47flattenRK5arrayii14StreamOrDevice", false]], "flatten() (array method)": [[50, "mlx.core.array.flatten", false]], "flatten() (in module mlx.core)": [[168, "mlx.core.flatten", false]], "floor (c++ function)": [[0, "_CPPv45floorRK5array14StreamOrDevice", false]], "floor() (in module mlx.core)": [[169, "mlx.core.floor", false]], "floor_divide (c++ function)": [[0, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", false]], "floor_divide() (in module mlx.core)": [[170, "mlx.core.floor_divide", false]], "freeze() (module method)": [[386, "mlx.nn.Module.freeze", false]], "full (c++ function)": [[0, "_CPPv44full5Shape5array14StreamOrDevice", false], [0, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", false], [0, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", false], [0, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", false]], "full() (in module mlx.core)": [[171, "mlx.core.full", false]], "gather (c++ function)": [[0, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", false], [0, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", false]], "gather_mm (c++ function)": [[0, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "gather_mm() (in module mlx.core)": [[172, "mlx.core.gather_mm", false]], "gather_qmm (c++ function)": [[0, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRK5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEbii14StreamOrDevice", false]], "gather_qmm() (in module mlx.core)": [[173, "mlx.core.gather_qmm", false]], "gaussian_nll_loss (class in mlx.nn.losses)": [[447, "mlx.nn.losses.gaussian_nll_loss", false]], "gelu (class in mlx.nn)": [[363, "mlx.nn.GELU", false], [434, "mlx.nn.gelu", false]], "gelu_approx (class in mlx.nn)": [[435, "mlx.nn.gelu_approx", false]], "gelu_fast_approx (class in mlx.nn)": [[436, "mlx.nn.gelu_fast_approx", false]], "get_active_memory() (in module mlx.core)": [[174, "mlx.core.get_active_memory", false]], "get_cache_memory() (in module mlx.core)": [[175, "mlx.core.get_cache_memory", false]], "get_peak_memory() (in module mlx.core)": [[176, "mlx.core.get_peak_memory", false]], "glorot_normal() (in module mlx.nn.init)": [[425, "mlx.nn.init.glorot_normal", false]], "glorot_uniform() (in module mlx.nn.init)": [[426, "mlx.nn.init.glorot_uniform", false]], "glu (class in mlx.nn)": [[364, "mlx.nn.GLU", false], [437, "mlx.nn.glu", false]], "grad() (in module mlx.core)": [[177, "mlx.core.grad", false]], "greater (c++ function)": [[0, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", false]], "greater() (in module mlx.core)": [[178, "mlx.core.greater", false]], "greater_equal (c++ function)": [[0, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", false]], "greater_equal() (in module mlx.core)": [[179, "mlx.core.greater_equal", false]], "group (class in mlx.core.distributed)": [[126, "mlx.core.distributed.Group", false]], "groupnorm (class in mlx.nn)": [[366, "mlx.nn.GroupNorm", false]], "gru (class in mlx.nn)": [[365, "mlx.nn.GRU", false]], "gumbel() (in module mlx.core.random)": [[253, "mlx.core.random.gumbel", false]], "hadamard_transform (c++ function)": [[0, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", false]], "hadamard_transform() (in module mlx.core)": [[180, "mlx.core.hadamard_transform", false]], "hard_shrink (class in mlx.nn)": [[438, "mlx.nn.hard_shrink", false]], "hard_tanh (class in mlx.nn)": [[439, "mlx.nn.hard_tanh", false]], "hardshrink (class in mlx.nn)": [[367, "mlx.nn.HardShrink", false]], "hardswish (class in mlx.nn)": [[369, "mlx.nn.Hardswish", false], [440, "mlx.nn.hardswish", false]], "hardtanh (class in mlx.nn)": [[368, "mlx.nn.HardTanh", false]], "he_normal() (in module mlx.nn.init)": [[427, "mlx.nn.init.he_normal", false]], "he_uniform() (in module mlx.nn.init)": [[428, "mlx.nn.init.he_uniform", false]], "hinge_loss (class in mlx.nn.losses)": [[448, "mlx.nn.losses.hinge_loss", false]], "huber_loss (class in mlx.nn.losses)": [[449, "mlx.nn.losses.huber_loss", false]], "identity (c++ function)": [[0, "_CPPv48identityi14StreamOrDevice", false], [0, "_CPPv48identityi5Dtype14StreamOrDevice", false]], "identity() (in module mlx.core)": [[181, "mlx.core.identity", false]], "identity() (in module mlx.nn.init)": [[429, "mlx.nn.init.identity", false]], "ifft() (in module mlx.core.fft)": [[158, "mlx.core.fft.ifft", false]], "ifft2() (in module mlx.core.fft)": [[159, "mlx.core.fft.ifft2", false]], "ifftn() (in module mlx.core.fft)": [[160, "mlx.core.fft.ifftn", false]], "imag (c++ function)": [[0, "_CPPv44imagRK5array14StreamOrDevice", false]], "imag() (in module mlx.core)": [[182, "mlx.core.imag", false]], "import_function() (in module mlx.core)": [[183, "mlx.core.import_function", false]], "init() (in module mlx.core.distributed)": [[129, "mlx.core.distributed.init", false]], "init() (optimizer method)": [[486, "mlx.optimizers.Optimizer.init", false]], "inner (c++ function)": [[0, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", false]], "inner() (in module mlx.core)": [[184, "mlx.core.inner", false]], "instancenorm (class in mlx.nn)": [[370, "mlx.nn.InstanceNorm", false]], "inv() (in module mlx.core.linalg)": [[202, "mlx.core.linalg.inv", false]], "irfft() (in module mlx.core.fft)": [[161, "mlx.core.fft.irfft", false]], "irfft2() (in module mlx.core.fft)": [[162, "mlx.core.fft.irfft2", false]], "irfftn() (in module mlx.core.fft)": [[163, "mlx.core.fft.irfftn", false]], "is_available() (in module mlx.core.distributed)": [[130, "mlx.core.distributed.is_available", false]], "is_available() (in module mlx.core.metal)": [[229, "mlx.core.metal.is_available", false]], "isclose (c++ function)": [[0, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "isclose() (in module mlx.core)": [[185, "mlx.core.isclose", false]], "isfinite (c++ function)": [[0, "_CPPv48isfiniteRK5array14StreamOrDevice", false]], "isfinite() (in module mlx.core)": [[186, "mlx.core.isfinite", false]], "isinf (c++ function)": [[0, "_CPPv45isinfRK5array14StreamOrDevice", false]], "isinf() (in module mlx.core)": [[187, "mlx.core.isinf", false]], "isnan (c++ function)": [[0, "_CPPv45isnanRK5array14StreamOrDevice", false]], "isnan() (in module mlx.core)": [[188, "mlx.core.isnan", false]], "isneginf (c++ function)": [[0, "_CPPv48isneginfRK5array14StreamOrDevice", false]], "isneginf() (in module mlx.core)": [[189, "mlx.core.isneginf", false]], "isposinf (c++ function)": [[0, "_CPPv48isposinfRK5array14StreamOrDevice", false]], "isposinf() (in module mlx.core)": [[190, "mlx.core.isposinf", false]], "issubdtype() (in module mlx.core)": [[191, "mlx.core.issubdtype", false]], "item() (array method)": [[51, "mlx.core.array.item", false]], "itemsize (array property)": [[52, "mlx.core.array.itemsize", false]], "join_schedules() (in module mlx.optimizers)": [[493, "mlx.optimizers.join_schedules", false]], "jvp() (in module mlx.core)": [[192, "mlx.core.jvp", false]], "key() (in module mlx.core.random)": [[254, "mlx.core.random.key", false]], "kl_div_loss (class in mlx.nn.losses)": [[450, "mlx.nn.losses.kl_div_loss", false]], "kron (c++ function)": [[0, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", false]], "kron() (in module mlx.core)": [[193, "mlx.core.kron", false]], "l1_loss (class in mlx.nn.losses)": [[451, "mlx.nn.losses.l1_loss", false]], "laplace() (in module mlx.core.random)": [[255, "mlx.core.random.laplace", false]], "layer_norm() (in module mlx.core.fast)": [[150, "mlx.core.fast.layer_norm", false]], "layernorm (class in mlx.nn)": [[372, "mlx.nn.LayerNorm", false]], "leaf_modules() (module method)": [[387, "mlx.nn.Module.leaf_modules", false]], "leaky_relu (class in mlx.nn)": [[441, "mlx.nn.leaky_relu", false]], "leakyrelu (class in mlx.nn)": [[373, "mlx.nn.LeakyReLU", false]], "left_shift (c++ function)": [[0, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", false]], "left_shift() (in module mlx.core)": [[194, "mlx.core.left_shift", false]], "less (c++ function)": [[0, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", false]], "less() (in module mlx.core)": [[195, "mlx.core.less", false]], "less_equal (c++ function)": [[0, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", false]], "less_equal() (in module mlx.core)": [[196, "mlx.core.less_equal", false]], "linear (class in mlx.nn)": [[374, "mlx.nn.Linear", false]], "linear_schedule() (in module mlx.optimizers)": [[494, "mlx.optimizers.linear_schedule", false]], "linspace (c++ function)": [[0, "_CPPv48linspaceddi5Dtype14StreamOrDevice", false]], "linspace() (in module mlx.core)": [[212, "mlx.core.linspace", false]], "lion (class in mlx.optimizers)": [[484, "mlx.optimizers.Lion", false]], "load() (in module mlx.core)": [[213, "mlx.core.load", false]], "load_weights() (module method)": [[388, "mlx.nn.Module.load_weights", false]], "log (c++ function)": [[0, "_CPPv43logRK5array14StreamOrDevice", false]], "log() (array method)": [[53, "mlx.core.array.log", false]], "log() (in module mlx.core)": [[214, "mlx.core.log", false]], "log10 (c++ function)": [[0, "_CPPv45log10RK5array14StreamOrDevice", false]], "log10() (array method)": [[54, "mlx.core.array.log10", false]], "log10() (in module mlx.core)": [[215, "mlx.core.log10", false]], "log1p (c++ function)": [[0, "_CPPv45log1pRK5array14StreamOrDevice", false]], "log1p() (array method)": [[55, "mlx.core.array.log1p", false]], "log1p() (in module mlx.core)": [[216, "mlx.core.log1p", false]], "log2 (c++ function)": [[0, "_CPPv44log2RK5array14StreamOrDevice", false]], "log2() (array method)": [[56, "mlx.core.array.log2", false]], "log2() (in module mlx.core)": [[217, "mlx.core.log2", false]], "log_cosh_loss (class in mlx.nn.losses)": [[452, "mlx.nn.losses.log_cosh_loss", false]], "log_sigmoid (class in mlx.nn)": [[442, "mlx.nn.log_sigmoid", false]], "log_softmax (class in mlx.nn)": [[443, "mlx.nn.log_softmax", false]], "logaddexp (c++ function)": [[0, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", false]], "logaddexp() (in module mlx.core)": [[218, "mlx.core.logaddexp", false]], "logical_and (c++ function)": [[0, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", false]], "logical_and() (in module mlx.core)": [[219, "mlx.core.logical_and", false]], "logical_not (c++ function)": [[0, "_CPPv411logical_notRK5array14StreamOrDevice", false]], "logical_not() (in module mlx.core)": [[220, "mlx.core.logical_not", false]], "logical_or (c++ function)": [[0, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", false]], "logical_or() (in module mlx.core)": [[221, "mlx.core.logical_or", false]], "logsigmoid (class in mlx.nn)": [[375, "mlx.nn.LogSigmoid", false]], "logsoftmax (class in mlx.nn)": [[376, "mlx.nn.LogSoftmax", false]], "logsumexp (c++ function)": [[0, "_CPPv49logsumexpRK5array14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", false]], "logsumexp() (array method)": [[57, "mlx.core.array.logsumexp", false]], "logsumexp() (in module mlx.core)": [[222, "mlx.core.logsumexp", false]], "lstm (class in mlx.nn)": [[371, "mlx.nn.LSTM", false]], "lu() (in module mlx.core.linalg)": [[203, "mlx.core.linalg.lu", false]], "lu_factor() (in module mlx.core.linalg)": [[204, "mlx.core.linalg.lu_factor", false]], "margin_ranking_loss (class in mlx.nn.losses)": [[453, "mlx.nn.losses.margin_ranking_loss", false]], "matmul (c++ function)": [[0, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", false]], "matmul() (in module mlx.core)": [[223, "mlx.core.matmul", false]], "max (c++ function)": [[0, "_CPPv43maxRK5array14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayib14StreamOrDevice", false]], "max() (array method)": [[58, "mlx.core.array.max", false]], "max() (in module mlx.core)": [[224, "mlx.core.max", false]], "maximum (c++ function)": [[0, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", false]], "maximum() (in module mlx.core)": [[225, "mlx.core.maximum", false]], "maxpool1d (class in mlx.nn)": [[377, "mlx.nn.MaxPool1d", false]], "maxpool2d (class in mlx.nn)": [[378, "mlx.nn.MaxPool2d", false]], "maxpool3d (class in mlx.nn)": [[379, "mlx.nn.MaxPool3d", false]], "mean (c++ function)": [[0, "_CPPv44meanRK5array14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayib14StreamOrDevice", false]], "mean() (array method)": [[59, "mlx.core.array.mean", false]], "mean() (in module mlx.core)": [[226, "mlx.core.mean", false]], "meshgrid (c++ function)": [[0, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", false]], "meshgrid() (in module mlx.core)": [[227, "mlx.core.meshgrid", false]], "metal_kernel() (in module mlx.core.fast)": [[151, "mlx.core.fast.metal_kernel", false]], "min (c++ function)": [[0, "_CPPv43minRK5array14StreamOrDevice", false], [0, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayib14StreamOrDevice", false]], "min() (array method)": [[60, "mlx.core.array.min", false]], "min() (in module mlx.core)": [[232, "mlx.core.min", false]], "minimum (c++ function)": [[0, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", false]], "minimum() (in module mlx.core)": [[233, "mlx.core.minimum", false]], "mish (class in mlx.nn)": [[380, "mlx.nn.Mish", false], [458, "mlx.nn.mish", false]], "module (class in mlx.nn)": [[475, "mlx.nn.Module", false]], "modules() (module method)": [[389, "mlx.nn.Module.modules", false]], "moveaxis (c++ function)": [[0, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", false]], "moveaxis() (array method)": [[61, "mlx.core.array.moveaxis", false]], "moveaxis() (in module mlx.core)": [[234, "mlx.core.moveaxis", false]], "mse_loss (class in mlx.nn.losses)": [[454, "mlx.nn.losses.mse_loss", false]], "multiheadattention (class in mlx.nn)": [[401, "mlx.nn.MultiHeadAttention", false]], "multiply (c++ function)": [[0, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", false]], "multiply() (in module mlx.core)": [[235, "mlx.core.multiply", false]], "multivariate_normal() (in module mlx.core.random)": [[256, "mlx.core.random.multivariate_normal", false]], "named_modules() (module method)": [[390, "mlx.nn.Module.named_modules", false]], "nan_to_num (c++ function)": [[0, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", false]], "nan_to_num() (in module mlx.core)": [[236, "mlx.core.nan_to_num", false]], "nbytes (array property)": [[62, "mlx.core.array.nbytes", false]], "ndim (array property)": [[63, "mlx.core.array.ndim", false]], "negative (c++ function)": [[0, "_CPPv48negativeRK5array14StreamOrDevice", false]], "negative() (in module mlx.core)": [[237, "mlx.core.negative", false]], "new_stream() (in module mlx.core)": [[238, "mlx.core.new_stream", false]], "nll_loss (class in mlx.nn.losses)": [[455, "mlx.nn.losses.nll_loss", false]], "norm() (in module mlx.core.linalg)": [[205, "mlx.core.linalg.norm", false]], "normal() (in module mlx.core.random)": [[257, "mlx.core.random.normal", false]], "normal() (in module mlx.nn.init)": [[430, "mlx.nn.init.normal", false]], "not_equal (c++ function)": [[0, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", false]], "not_equal() (in module mlx.core)": [[239, "mlx.core.not_equal", false]], "number_of_elements (c++ function)": [[0, "_CPPv418number_of_elementsRK5arrayNSt6vectorIiEEb5Dtype14StreamOrDevice", false]], "ones (c++ function)": [[0, "_CPPv44onesRK5Shape14StreamOrDevice", false], [0, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", false]], "ones() (in module mlx.core)": [[240, "mlx.core.ones", false]], "ones_like (c++ function)": [[0, "_CPPv49ones_likeRK5array14StreamOrDevice", false]], "ones_like() (in module mlx.core)": [[241, "mlx.core.ones_like", false]], "operator!= (c++ function)": [[0, "_CPPv4I0Ene5array1TRK5array", false], [0, "_CPPv4I0Ene5arrayRK5array1T", false], [0, "_CPPv4neRK5arrayRK5array", false]], "operator% (c++ function)": [[0, "_CPPv4I0Erm5array1TRK5array", false], [0, "_CPPv4I0Erm5arrayRK5array1T", false], [0, "_CPPv4rmRK5arrayRK5array", false]], "operator& (c++ function)": [[0, "_CPPv4anRK5arrayRK5array", false]], "operator&& (c++ function)": [[0, "_CPPv4aaRK5arrayRK5array", false]], "operator* (c++ function)": [[0, "_CPPv4I0Eml5array1TRK5array", false], [0, "_CPPv4I0Eml5arrayRK5array1T", false], [0, "_CPPv4mlRK5arrayRK5array", false]], "operator+ (c++ function)": [[0, "_CPPv4I0Epl5array1TRK5array", false], [0, "_CPPv4I0Epl5arrayRK5array1T", false], [0, "_CPPv4plRK5arrayRK5array", false]], "operator- (c++ function)": [[0, "_CPPv4I0Emi5array1TRK5array", false], [0, "_CPPv4I0Emi5arrayRK5array1T", false], [0, "_CPPv4miRK5array", false], [0, "_CPPv4miRK5arrayRK5array", false]], "operator/ (c++ function)": [[0, "_CPPv4dvRK5arrayRK5array", false], [0, "_CPPv4dvRK5arrayd", false], [0, "_CPPv4dvdRK5array", false]], "operator< (c++ function)": [[0, "_CPPv4I0Elt5array1TRK5array", false], [0, "_CPPv4I0Elt5arrayRK5array1T", false], [0, "_CPPv4ltRK5arrayRK5array", false]], "operator<< (c++ function)": [[0, "_CPPv4lsRK5arrayRK5array", false]], "operator<= (c++ function)": [[0, "_CPPv4I0Ele5array1TRK5array", false], [0, "_CPPv4I0Ele5arrayRK5array1T", false], [0, "_CPPv4leRK5arrayRK5array", false]], "operator== (c++ function)": [[0, "_CPPv4I0Eeq5array1TRK5array", false], [0, "_CPPv4I0Eeq5arrayRK5array1T", false], [0, "_CPPv4eqRK5arrayRK5array", false]], "operator> (c++ function)": [[0, "_CPPv4I0Egt5array1TRK5array", false], [0, "_CPPv4I0Egt5arrayRK5array1T", false], [0, "_CPPv4gtRK5arrayRK5array", false]], "operator>= (c++ function)": [[0, "_CPPv4I0Ege5array1TRK5array", false], [0, "_CPPv4I0Ege5arrayRK5array1T", false], [0, "_CPPv4geRK5arrayRK5array", false]], "operator>> (c++ function)": [[0, "_CPPv4rsRK5arrayRK5array", false]], "operator^ (c++ function)": [[0, "_CPPv4eoRK5arrayRK5array", false]], "operator| (c++ function)": [[0, "_CPPv4orRK5arrayRK5array", false]], "operator|| (c++ function)": [[0, "_CPPv4ooRK5arrayRK5array", false]], "operator~ (c++ function)": [[0, "_CPPv4coRK5array", false]], "optimizer (class in mlx.optimizers)": [[497, "mlx.optimizers.Optimizer", false]], "outer (c++ function)": [[0, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", false]], "outer() (in module mlx.core)": [[242, "mlx.core.outer", false]], "pad (c++ function)": [[0, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", false]], "pad() (in module mlx.core)": [[243, "mlx.core.pad", false]], "parameters() (module method)": [[391, "mlx.nn.Module.parameters", false]], "partition (c++ function)": [[0, "_CPPv49partitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv49partitionRK5arrayii14StreamOrDevice", false]], "partition() (in module mlx.core)": [[244, "mlx.core.partition", false]], "permutation() (in module mlx.core.random)": [[258, "mlx.core.random.permutation", false]], "pinv() (in module mlx.core.linalg)": [[206, "mlx.core.linalg.pinv", false]], "power (c++ function)": [[0, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", false]], "power() (in module mlx.core)": [[245, "mlx.core.power", false]], "prelu (class in mlx.nn)": [[402, "mlx.nn.PReLU", false], [459, "mlx.nn.prelu", false]], "prod (c++ function)": [[0, "_CPPv44prodRK5array14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayib14StreamOrDevice", false]], "prod() (array method)": [[64, "mlx.core.array.prod", false]], "prod() (in module mlx.core)": [[246, "mlx.core.prod", false]], "put_along_axis (c++ function)": [[0, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "put_along_axis() (in module mlx.core)": [[247, "mlx.core.put_along_axis", false]], "qr() (in module mlx.core.linalg)": [[207, "mlx.core.linalg.qr", false]], "quantize (c++ function)": [[0, "_CPPv48quantizeRK5arrayii14StreamOrDevice", false]], "quantize() (in module mlx.core)": [[248, "mlx.core.quantize", false]], "quantize() (in module mlx.nn)": [[326, "mlx.nn.quantize", false]], "quantized_matmul (c++ function)": [[0, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", false]], "quantized_matmul() (in module mlx.core)": [[249, "mlx.core.quantized_matmul", false]], "quantizedembedding (class in mlx.nn)": [[403, "mlx.nn.QuantizedEmbedding", false]], "quantizedlinear (class in mlx.nn)": [[404, "mlx.nn.QuantizedLinear", false]], "radians (c++ function)": [[0, "_CPPv47radiansRK5array14StreamOrDevice", false]], "radians() (in module mlx.core)": [[250, "mlx.core.radians", false]], "randint() (in module mlx.core.random)": [[259, "mlx.core.random.randint", false]], "real (c++ function)": [[0, "_CPPv44realRK5array14StreamOrDevice", false]], "real() (in module mlx.core)": [[264, "mlx.core.real", false]], "reciprocal (c++ function)": [[0, "_CPPv410reciprocalRK5array14StreamOrDevice", false]], "reciprocal() (array method)": [[65, "mlx.core.array.reciprocal", false]], "reciprocal() (in module mlx.core)": [[265, "mlx.core.reciprocal", false]], "recv() (in module mlx.core.distributed)": [[131, "mlx.core.distributed.recv", false]], "recv_like() (in module mlx.core.distributed)": [[132, "mlx.core.distributed.recv_like", false]], "relu (class in mlx.nn)": [[407, "mlx.nn.ReLU", false], [460, "mlx.nn.relu", false]], "relu6 (class in mlx.nn)": [[408, "mlx.nn.ReLU6", false], [461, "mlx.nn.relu6", false]], "remainder (c++ function)": [[0, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", false]], "remainder() (in module mlx.core)": [[266, "mlx.core.remainder", false]], "repeat (c++ function)": [[0, "_CPPv46repeatRK5arrayi14StreamOrDevice", false], [0, "_CPPv46repeatRK5arrayii14StreamOrDevice", false]], "repeat() (in module mlx.core)": [[267, "mlx.core.repeat", false]], "reset_peak_memory() (in module mlx.core)": [[268, "mlx.core.reset_peak_memory", false]], "reshape (c++ function)": [[0, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", false]], "reshape() (array method)": [[66, "mlx.core.array.reshape", false]], "reshape() (in module mlx.core)": [[269, "mlx.core.reshape", false]], "rfft() (in module mlx.core.fft)": [[164, "mlx.core.fft.rfft", false]], "rfft2() (in module mlx.core.fft)": [[165, "mlx.core.fft.rfft2", false]], "rfftn() (in module mlx.core.fft)": [[166, "mlx.core.fft.rfftn", false]], "right_shift (c++ function)": [[0, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", false]], "right_shift() (in module mlx.core)": [[270, "mlx.core.right_shift", false]], "rms_norm() (in module mlx.core.fast)": [[152, "mlx.core.fast.rms_norm", false]], "rmsnorm (class in mlx.nn)": [[405, "mlx.nn.RMSNorm", false]], "rmsprop (class in mlx.optimizers)": [[489, "mlx.optimizers.RMSprop", false]], "rnn (class in mlx.nn)": [[406, "mlx.nn.RNN", false]], "roll (c++ function)": [[0, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayi14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv44rollRK5arrayii14StreamOrDevice", false]], "roll() (in module mlx.core)": [[271, "mlx.core.roll", false]], "rope (class in mlx.nn)": [[409, "mlx.nn.RoPE", false]], "rope() (in module mlx.core.fast)": [[153, "mlx.core.fast.rope", false]], "round (c++ function)": [[0, "_CPPv45roundRK5array14StreamOrDevice", false], [0, "_CPPv45roundRK5arrayi14StreamOrDevice", false]], "round() (array method)": [[67, "mlx.core.array.round", false]], "round() (in module mlx.core)": [[272, "mlx.core.round", false]], "rsqrt (c++ function)": [[0, "_CPPv45rsqrtRK5array14StreamOrDevice", false]], "rsqrt() (array method)": [[68, "mlx.core.array.rsqrt", false]], "rsqrt() (in module mlx.core)": [[273, "mlx.core.rsqrt", false]], "save() (in module mlx.core)": [[274, "mlx.core.save", false]], "save_gguf() (in module mlx.core)": [[275, "mlx.core.save_gguf", false]], "save_safetensors() (in module mlx.core)": [[276, "mlx.core.save_safetensors", false]], "save_weights() (module method)": [[392, "mlx.nn.Module.save_weights", false]], "savez() (in module mlx.core)": [[277, "mlx.core.savez", false]], "savez_compressed() (in module mlx.core)": [[278, "mlx.core.savez_compressed", false]], "scaled_dot_product_attention() (in module mlx.core.fast)": [[154, "mlx.core.fast.scaled_dot_product_attention", false]], "scatter (c++ function)": [[0, "_CPPv47scatterRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv47scatterRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_add (c++ function)": [[0, "_CPPv411scatter_addRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_addRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_add_axis (c++ function)": [[0, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "scatter_max (c++ function)": [[0, "_CPPv411scatter_maxRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_maxRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_min (c++ function)": [[0, "_CPPv411scatter_minRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv411scatter_minRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "scatter_prod (c++ function)": [[0, "_CPPv412scatter_prodRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv412scatter_prodRK5arrayRKNSt6vectorI5arrayEERK5arrayRKNSt6vectorIiEE14StreamOrDevice", false]], "seed() (in module mlx.core.random)": [[260, "mlx.core.random.seed", false]], "selu (class in mlx.nn)": [[410, "mlx.nn.SELU", false], [462, "mlx.nn.selu", false]], "send() (in module mlx.core.distributed)": [[133, "mlx.core.distributed.send", false]], "sequential (class in mlx.nn)": [[411, "mlx.nn.Sequential", false]], "set_cache_limit() (in module mlx.core)": [[279, "mlx.core.set_cache_limit", false]], "set_default_device() (in module mlx.core)": [[280, "mlx.core.set_default_device", false]], "set_default_stream() (in module mlx.core)": [[281, "mlx.core.set_default_stream", false]], "set_dtype() (module method)": [[393, "mlx.nn.Module.set_dtype", false]], "set_memory_limit() (in module mlx.core)": [[282, "mlx.core.set_memory_limit", false]], "set_wired_limit() (in module mlx.core)": [[283, "mlx.core.set_wired_limit", false]], "sgd (class in mlx.optimizers)": [[490, "mlx.optimizers.SGD", false]], "shape (array property)": [[69, "mlx.core.array.shape", false]], "sigmoid (c++ function)": [[0, "_CPPv47sigmoidRK5array14StreamOrDevice", false]], "sigmoid (class in mlx.nn)": [[413, "mlx.nn.Sigmoid", false], [463, "mlx.nn.sigmoid", false]], "sigmoid() (in module mlx.core)": [[284, "mlx.core.sigmoid", false]], "sign (c++ function)": [[0, "_CPPv44signRK5array14StreamOrDevice", false]], "sign() (in module mlx.core)": [[285, "mlx.core.sign", false]], "silu (class in mlx.nn)": [[412, "mlx.nn.SiLU", false], [464, "mlx.nn.silu", false]], "sin (c++ function)": [[0, "_CPPv43sinRK5array14StreamOrDevice", false]], "sin() (array method)": [[70, "mlx.core.array.sin", false]], "sin() (in module mlx.core)": [[286, "mlx.core.sin", false]], "sinh (c++ function)": [[0, "_CPPv44sinhRK5array14StreamOrDevice", false]], "sinh() (in module mlx.core)": [[287, "mlx.core.sinh", false]], "sinusoidalpositionalencoding (class in mlx.nn)": [[414, "mlx.nn.SinusoidalPositionalEncoding", false]], "size (array property)": [[71, "mlx.core.array.size", false]], "slice (c++ function)": [[0, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", false], [0, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", false]], "slice() (in module mlx.core)": [[288, "mlx.core.slice", false]], "slice_update (c++ function)": [[0, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", false], [0, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", false], [0, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "slice_update() (in module mlx.core)": [[289, "mlx.core.slice_update", false]], "smooth_l1_loss (class in mlx.nn.losses)": [[456, "mlx.nn.losses.smooth_l1_loss", false]], "softmax (c++ function)": [[0, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayib14StreamOrDevice", false]], "softmax (class in mlx.nn)": [[415, "mlx.nn.Softmax", false], [465, "mlx.nn.softmax", false]], "softmax() (in module mlx.core)": [[290, "mlx.core.softmax", false]], "softmin (class in mlx.nn)": [[416, "mlx.nn.Softmin", false], [466, "mlx.nn.softmin", false]], "softplus (class in mlx.nn)": [[417, "mlx.nn.Softplus", false], [467, "mlx.nn.softplus", false]], "softshrink (class in mlx.nn)": [[418, "mlx.nn.Softshrink", false], [468, "mlx.nn.softshrink", false]], "softsign (class in mlx.nn)": [[419, "mlx.nn.Softsign", false]], "solve() (in module mlx.core.linalg)": [[208, "mlx.core.linalg.solve", false]], "solve_triangular() (in module mlx.core.linalg)": [[209, "mlx.core.linalg.solve_triangular", false]], "sort (c++ function)": [[0, "_CPPv44sortRK5array14StreamOrDevice", false], [0, "_CPPv44sortRK5arrayi14StreamOrDevice", false]], "sort() (in module mlx.core)": [[291, "mlx.core.sort", false]], "split (c++ function)": [[0, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayi14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayii14StreamOrDevice", false]], "split() (array method)": [[72, "mlx.core.array.split", false]], "split() (in module mlx.core)": [[292, "mlx.core.split", false]], "split() (in module mlx.core.random)": [[261, "mlx.core.random.split", false]], "sqrt (c++ function)": [[0, "_CPPv44sqrtRK5array14StreamOrDevice", false]], "sqrt() (array method)": [[73, "mlx.core.array.sqrt", false]], "sqrt() (in module mlx.core)": [[293, "mlx.core.sqrt", false]], "square (c++ function)": [[0, "_CPPv46squareRK5array14StreamOrDevice", false]], "square() (array method)": [[74, "mlx.core.array.square", false]], "square() (in module mlx.core)": [[294, "mlx.core.square", false]], "squeeze (c++ function)": [[0, "_CPPv47squeezeRK5array14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayi14StreamOrDevice", false]], "squeeze() (array method)": [[75, "mlx.core.array.squeeze", false]], "squeeze() (in module mlx.core)": [[295, "mlx.core.squeeze", false]], "stack (c++ function)": [[0, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", false]], "stack() (in module mlx.core)": [[296, "mlx.core.stack", false]], "start_capture() (in module mlx.core.metal)": [[230, "mlx.core.metal.start_capture", false]], "state (module property)": [[394, "mlx.nn.Module.state", false]], "state (optimizer property)": [[487, "mlx.optimizers.Optimizer.state", false]], "std (c++ function)": [[0, "_CPPv4StRK5array14StreamOrDevice", false], [0, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv4StRK5arraybi14StreamOrDevice", false], [0, "_CPPv4StRK5arrayibi14StreamOrDevice", false]], "std() (array method)": [[76, "mlx.core.array.std", false]], "std() (in module mlx.core)": [[297, "mlx.core.std", false]], "step (class in mlx.nn)": [[420, "mlx.nn.Step", false], [469, "mlx.nn.step", false]], "step_decay() (in module mlx.optimizers)": [[495, "mlx.optimizers.step_decay", false]], "stop_capture() (in module mlx.core.metal)": [[231, "mlx.core.metal.stop_capture", false]], "stop_gradient (c++ function)": [[0, "_CPPv413stop_gradientRK5array14StreamOrDevice", false]], "stop_gradient() (in module mlx.core)": [[298, "mlx.core.stop_gradient", false]], "stream (class in mlx.core)": [[334, "mlx.core.Stream", false]], "stream() (in module mlx.core)": [[299, "mlx.core.stream", false]], "subtract (c++ function)": [[0, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", false]], "subtract() (in module mlx.core)": [[300, "mlx.core.subtract", false]], "sum (c++ function)": [[0, "_CPPv43sumRK5array14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayib14StreamOrDevice", false]], "sum() (array method)": [[77, "mlx.core.array.sum", false]], "sum() (in module mlx.core)": [[301, "mlx.core.sum", false]], "svd() (in module mlx.core.linalg)": [[210, "mlx.core.linalg.svd", false]], "swapaxes (c++ function)": [[0, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", false]], "swapaxes() (array method)": [[78, "mlx.core.array.swapaxes", false]], "swapaxes() (in module mlx.core)": [[302, "mlx.core.swapaxes", false]], "synchronize() (in module mlx.core)": [[303, "mlx.core.synchronize", false]], "t (array property)": [[32, "mlx.core.array.T", false]], "take (c++ function)": [[0, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayi14StreamOrDevice", false], [0, "_CPPv44takeRK5arrayii14StreamOrDevice", false]], "take() (in module mlx.core)": [[304, "mlx.core.take", false]], "take_along_axis (c++ function)": [[0, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", false]], "take_along_axis() (in module mlx.core)": [[305, "mlx.core.take_along_axis", false]], "tan (c++ function)": [[0, "_CPPv43tanRK5array14StreamOrDevice", false]], "tan() (in module mlx.core)": [[306, "mlx.core.tan", false]], "tanh (c++ function)": [[0, "_CPPv44tanhRK5array14StreamOrDevice", false]], "tanh (class in mlx.nn)": [[421, "mlx.nn.Tanh", false], [470, "mlx.nn.tanh", false]], "tanh() (in module mlx.core)": [[307, "mlx.core.tanh", false]], "tensordot (c++ function)": [[0, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", false], [0, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", false]], "tensordot() (in module mlx.core)": [[308, "mlx.core.tensordot", false]], "tile (c++ function)": [[0, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "tile() (in module mlx.core)": [[309, "mlx.core.tile", false]], "tolist() (array method)": [[79, "mlx.core.array.tolist", false]], "topk (c++ function)": [[0, "_CPPv44topkRK5arrayi14StreamOrDevice", false], [0, "_CPPv44topkRK5arrayii14StreamOrDevice", false]], "topk() (in module mlx.core)": [[310, "mlx.core.topk", false]], "trace (c++ function)": [[0, "_CPPv45traceRK5array14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", false]], "trace() (in module mlx.core)": [[311, "mlx.core.trace", false]], "train() (module method)": [[395, "mlx.nn.Module.train", false]], "trainable_parameters() (module method)": [[396, "mlx.nn.Module.trainable_parameters", false]], "training (module property)": [[397, "mlx.nn.Module.training", false]], "transformer (class in mlx.nn)": [[422, "mlx.nn.Transformer", false]], "transpose (c++ function)": [[0, "_CPPv49transposeRK5array14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "transpose() (array method)": [[80, "mlx.core.array.transpose", false]], "transpose() (in module mlx.core)": [[312, "mlx.core.transpose", false]], "tree_flatten() (in module mlx.utils)": [[329, "mlx.utils.tree_flatten", false]], "tree_map() (in module mlx.utils)": [[330, "mlx.utils.tree_map", false]], "tree_map_with_path() (in module mlx.utils)": [[331, "mlx.utils.tree_map_with_path", false]], "tree_reduce() (in module mlx.utils)": [[332, "mlx.utils.tree_reduce", false]], "tree_unflatten() (in module mlx.utils)": [[333, "mlx.utils.tree_unflatten", false]], "tri (c++ function)": [[0, "_CPPv43trii5Dtype14StreamOrDevice", false], [0, "_CPPv43triiii5Dtype14StreamOrDevice", false]], "tri() (in module mlx.core)": [[313, "mlx.core.tri", false]], "tri_inv() (in module mlx.core.linalg)": [[211, "mlx.core.linalg.tri_inv", false]], "tril (c++ function)": [[0, "_CPPv44tril5arrayi14StreamOrDevice", false]], "tril() (in module mlx.core)": [[314, "mlx.core.tril", false]], "triplet_loss (class in mlx.nn.losses)": [[457, "mlx.nn.losses.triplet_loss", false]], "triu (c++ function)": [[0, "_CPPv44triu5arrayi14StreamOrDevice", false]], "triu() (in module mlx.core)": [[315, "mlx.core.triu", false]], "truncated_normal() (in module mlx.core.random)": [[262, "mlx.core.random.truncated_normal", false]], "unflatten (c++ function)": [[0, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", false]], "unflatten() (in module mlx.core)": [[316, "mlx.core.unflatten", false]], "unfreeze() (module method)": [[398, "mlx.nn.Module.unfreeze", false]], "uniform() (in module mlx.core.random)": [[263, "mlx.core.random.uniform", false]], "uniform() (in module mlx.nn.init)": [[431, "mlx.nn.init.uniform", false]], "update() (module method)": [[399, "mlx.nn.Module.update", false]], "update() (optimizer method)": [[488, "mlx.optimizers.Optimizer.update", false]], "update_modules() (module method)": [[400, "mlx.nn.Module.update_modules", false]], "upsample (class in mlx.nn)": [[423, "mlx.nn.Upsample", false]], "value_and_grad() (in module mlx.core)": [[317, "mlx.core.value_and_grad", false]], "value_and_grad() (in module mlx.nn)": [[327, "mlx.nn.value_and_grad", false]], "var (c++ function)": [[0, "_CPPv43varRK5array14StreamOrDevice", false], [0, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv43varRK5arraybi14StreamOrDevice", false], [0, "_CPPv43varRK5arrayibi14StreamOrDevice", false]], "var() (array method)": [[81, "mlx.core.array.var", false]], "var() (in module mlx.core)": [[318, "mlx.core.var", false]], "view (c++ function)": [[0, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", false]], "view() (array method)": [[82, "mlx.core.array.view", false]], "view() (in module mlx.core)": [[319, "mlx.core.view", false]], "vjp() (in module mlx.core)": [[320, "mlx.core.vjp", false]], "vmap() (in module mlx.core)": [[321, "mlx.core.vmap", false]], "where (c++ function)": [[0, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", false]], "where() (in module mlx.core)": [[322, "mlx.core.where", false]], "zeros (c++ function)": [[0, "_CPPv45zerosRK5Shape14StreamOrDevice", false], [0, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", false]], "zeros() (in module mlx.core)": [[323, "mlx.core.zeros", false]], "zeros_like (c++ function)": [[0, "_CPPv410zeros_likeRK5array14StreamOrDevice", false]], "zeros_like() (in module mlx.core)": [[324, "mlx.core.zeros_like", false]]}, "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_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::a"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::offset"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::s"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", "as_strided::shape"], [0, 1, 1, "_CPPv410as_strided5array5Shape7Strides6size_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, "_CPPv414bitwise_invertRK5array14StreamOrDevice", "bitwise_invert"], [0, 1, 1, "_CPPv414bitwise_invertRK5array14StreamOrDevice", "bitwise_invert::a"], [0, 1, 1, "_CPPv414bitwise_invertRK5array14StreamOrDevice", "bitwise_invert::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_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::a"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "broadcast_to::s"], [0, 1, 1, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", "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, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate"], [0, 0, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::arrays"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::axis"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", "concatenate::s"], [0, 1, 1, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", "concatenate::s"], [0, 0, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::a"], [0, 1, 1, "_CPPv49conjugateRK5array14StreamOrDevice", "conjugate::s"], [0, 0, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::a"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::allow_col_major"], [0, 1, 1, "_CPPv410contiguousRK5arrayb14StreamOrDevice", "contiguous::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, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::dilation"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::groups"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::input"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::padding"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::s"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::stride"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiii14StreamOrDevice", "conv_transpose1d::weight"], [0, 0, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::dilation"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::groups"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::input"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::padding"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::s"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::stride"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::weight"], [0, 0, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::dilation"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::groups"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::input"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::padding"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::s"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::stride"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::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, "_CPPv44full5Shape5array14StreamOrDevice", "full"], [0, 0, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full"], [0, 0, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full"], [0, 2, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::T"], [0, 2, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::T"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::dtype"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::s"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::shape"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv4I0E4full5array5Shape1T5Dtype14StreamOrDevice", "full::val"], [0, 1, 1, "_CPPv44full5Shape5array14StreamOrDevice", "full::vals"], [0, 1, 1, "_CPPv44full5Shape5array5Dtype14StreamOrDevice", "full::vals"], [0, 0, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather"], [0, 0, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::a"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::axes"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::axis"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::indices"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "gather::s"], [0, 1, 1, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", "gather::slice_sizes"], [0, 1, 1, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", "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, "_CPPv44imagRK5array14StreamOrDevice", "imag"], [0, 1, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag::a"], [0, 1, 1, "_CPPv44imagRK5array14StreamOrDevice", "imag::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, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite"], [0, 1, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite::a"], [0, 1, 1, "_CPPv48isfiniteRK5array14StreamOrDevice", "isfinite::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, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::a"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::b"], [0, 1, 1, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", "kron::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, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::arrays"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::indexing"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "meshgrid::s"], [0, 1, 1, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", "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_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::a"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::nan"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::neginf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "nan_to_num::posinf"], [0, 1, 1, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", "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, "_CPPv44onesRK5Shape14StreamOrDevice", "ones"], [0, 0, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::dtype"], [0, 1, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "ones::s"], [0, 1, 1, "_CPPv44onesRK5Shape14StreamOrDevice", "ones::shape"], [0, 1, 1, "_CPPv44onesRK5Shape5Dtype14StreamOrDevice", "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, "_CPPv4coRK5array", "operator~"], [0, 1, 1, "_CPPv4coRK5array", "operator~::a"], [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, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 0, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::a"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::axes"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::high_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::low_pad_size"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::mode"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_value"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "pad::pad_width"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", "pad::s"], [0, 1, 1, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", "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, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::a"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::axis"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::indices"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::s"], [0, 1, 1, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "put_along_axis::values"], [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_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::biases"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::bits"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::group_size"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::s"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::scales"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::transpose"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::w"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5array5arraybii14StreamOrDevice", "quantized_matmul::x"], [0, 0, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::a"], [0, 1, 1, "_CPPv47radiansRK5array14StreamOrDevice", "radians::s"], [0, 0, 1, "_CPPv44realRK5array14StreamOrDevice", "real"], [0, 1, 1, "_CPPv44realRK5array14StreamOrDevice", "real::a"], [0, 1, 1, "_CPPv44realRK5array14StreamOrDevice", "real::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, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::a"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "reshape::s"], [0, 1, 1, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", "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, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll"], [0, 0, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::a"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::axes"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::axes"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::axis"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::axis"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::s"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shape14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayRK5ShapeRKNSt6vectorIiEE14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayRK5Shapei14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayi14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayiRKNSt6vectorIiEE14StreamOrDevice", "roll::shift"], [0, 1, 1, "_CPPv44rollRK5arrayii14StreamOrDevice", "roll::shift"], [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, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::a"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::axis"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::indices"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::s"], [0, 1, 1, "_CPPv416scatter_add_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", "scatter_add_axis::values"], [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, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice"], [0, 0, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::a"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::axes"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::s"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::slice_size"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5arrayRK5arrayNSt6vectorIiEE5Shape14StreamOrDevice", "slice::start"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::stop"], [0, 1, 1, "_CPPv45sliceRK5array5Shape5Shape5Shape14StreamOrDevice", "slice::strides"], [0, 1, 1, "_CPPv45sliceRK5arrayNSt16initializer_listIiEE5Shape5Shape14StreamOrDevice", "slice::strides"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update"], [0, 0, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::axes"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::s"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::src"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "slice_update::start"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::stop"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::strides"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5array5Shape5Shape5Shape14StreamOrDevice", "slice_update::update"], [0, 1, 1, "_CPPv412slice_updateRK5arrayRK5arrayRK5arrayNSt6vectorIiEE14StreamOrDevice", "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, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split"], [0, 0, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::a"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::axis"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "split::indices"], [0, 1, 1, "_CPPv45splitRK5arrayi14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayii14StreamOrDevice", "split::num_splits"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", "split::s"], [0, 1, 1, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", "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, 0, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take"], [0, 0, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take"], [0, 1, 1, "_CPPv44takeRK5arrayRK5array14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::a"], [0, 1, 1, "_CPPv44takeRK5arrayRK5arrayi14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::axis"], [0, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::index"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "take::index"], [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, 1, 1, "_CPPv44takeRK5arrayi14StreamOrDevice", "take::s"], [0, 1, 1, "_CPPv44takeRK5arrayii14StreamOrDevice", "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, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::a"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::axis"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::s"], [0, 1, 1, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", "unflatten::shape"], [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, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros"], [0, 0, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::dtype"], [0, 1, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "zeros::s"], [0, 1, 1, "_CPPv45zerosRK5Shape14StreamOrDevice", "zeros::shape"], [0, 1, 1, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", "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": [[10, 3, 1, "", "Device"], [11, 3, 1, "", "Dtype"], [12, 3, 1, "", "DtypeCategory"], [334, 3, 1, "", "Stream"], [13, 5, 1, "", "abs"], [14, 5, 1, "", "add"], [15, 5, 1, "", "addmm"], [16, 5, 1, "", "all"], [17, 5, 1, "", "allclose"], [18, 5, 1, "", "any"], [19, 5, 1, "", "arange"], [20, 5, 1, "", "arccos"], [21, 5, 1, "", "arccosh"], [22, 5, 1, "", "arcsin"], [23, 5, 1, "", "arcsinh"], [24, 5, 1, "", "arctan"], [25, 5, 1, "", "arctan2"], [26, 5, 1, "", "arctanh"], [27, 5, 1, "", "argmax"], [28, 5, 1, "", "argmin"], [29, 5, 1, "", "argpartition"], [30, 5, 1, "", "argsort"], [31, 3, 1, "", "array"], [83, 5, 1, "", "array_equal"], [84, 5, 1, "", "as_strided"], [85, 5, 1, "", "async_eval"], [86, 5, 1, "", "atleast_1d"], [87, 5, 1, "", "atleast_2d"], [88, 5, 1, "", "atleast_3d"], [89, 5, 1, "", "bitwise_and"], [90, 5, 1, "", "bitwise_invert"], [91, 5, 1, "", "bitwise_or"], [92, 5, 1, "", "bitwise_xor"], [93, 5, 1, "", "block_masked_mm"], [94, 5, 1, "", "broadcast_arrays"], [95, 5, 1, "", "broadcast_to"], [96, 5, 1, "", "ceil"], [97, 5, 1, "", "clear_cache"], [98, 5, 1, "", "clip"], [99, 5, 1, "", "compile"], [100, 5, 1, "", "concatenate"], [101, 5, 1, "", "conj"], [102, 5, 1, "", "conjugate"], [103, 5, 1, "", "contiguous"], [104, 5, 1, "", "conv1d"], [105, 5, 1, "", "conv2d"], [106, 5, 1, "", "conv3d"], [107, 5, 1, "", "conv_general"], [108, 5, 1, "", "conv_transpose1d"], [109, 5, 1, "", "conv_transpose2d"], [110, 5, 1, "", "conv_transpose3d"], [111, 5, 1, "", "convolve"], [112, 5, 1, "", "cos"], [113, 5, 1, "", "cosh"], [114, 5, 1, "", "cummax"], [115, 5, 1, "", "cummin"], [116, 5, 1, "", "cumprod"], [117, 5, 1, "", "cumsum"], [118, 3, 1, "", "custom_function"], [119, 5, 1, "", "default_device"], [120, 5, 1, "", "default_stream"], [121, 5, 1, "", "degrees"], [122, 5, 1, "", "dequantize"], [123, 5, 1, "", "diag"], [124, 5, 1, "", "diagonal"], [125, 5, 1, "", "disable_compile"], [134, 5, 1, "", "divide"], [135, 5, 1, "", "divmod"], [136, 5, 1, "", "einsum"], [137, 5, 1, "", "einsum_path"], [138, 5, 1, "", "enable_compile"], [139, 5, 1, "", "equal"], [140, 5, 1, "", "erf"], [141, 5, 1, "", "erfinv"], [142, 5, 1, "", "eval"], [143, 5, 1, "", "exp"], [144, 5, 1, "", "expand_dims"], [145, 5, 1, "", "expm1"], [146, 5, 1, "", "export_function"], [147, 5, 1, "", "export_to_dot"], [148, 5, 1, "", "exporter"], [149, 5, 1, "", "eye"], [167, 3, 1, "", "finfo"], [168, 5, 1, "", "flatten"], [169, 5, 1, "", "floor"], [170, 5, 1, "", "floor_divide"], [171, 5, 1, "", "full"], [172, 5, 1, "", "gather_mm"], [173, 5, 1, "", "gather_qmm"], [174, 5, 1, "", "get_active_memory"], [175, 5, 1, "", "get_cache_memory"], [176, 5, 1, "", "get_peak_memory"], [177, 5, 1, "", "grad"], [178, 5, 1, "", "greater"], [179, 5, 1, "", "greater_equal"], [180, 5, 1, "", "hadamard_transform"], [181, 5, 1, "", "identity"], [182, 5, 1, "", "imag"], [183, 5, 1, "", "import_function"], [184, 5, 1, "", "inner"], [185, 5, 1, "", "isclose"], [186, 5, 1, "", "isfinite"], [187, 5, 1, "", "isinf"], [188, 5, 1, "", "isnan"], [189, 5, 1, "", "isneginf"], [190, 5, 1, "", "isposinf"], [191, 5, 1, "", "issubdtype"], [192, 5, 1, "", "jvp"], [193, 5, 1, "", "kron"], [194, 5, 1, "", "left_shift"], [195, 5, 1, "", "less"], [196, 5, 1, "", "less_equal"], [212, 5, 1, "", "linspace"], [213, 5, 1, "", "load"], [214, 5, 1, "", "log"], [215, 5, 1, "", "log10"], [216, 5, 1, "", "log1p"], [217, 5, 1, "", "log2"], [218, 5, 1, "", "logaddexp"], [219, 5, 1, "", "logical_and"], [220, 5, 1, "", "logical_not"], [221, 5, 1, "", "logical_or"], [222, 5, 1, "", "logsumexp"], [223, 5, 1, "", "matmul"], [224, 5, 1, "", "max"], [225, 5, 1, "", "maximum"], [226, 5, 1, "", "mean"], [227, 5, 1, "", "meshgrid"], [232, 5, 1, "", "min"], [233, 5, 1, "", "minimum"], [234, 5, 1, "", "moveaxis"], [235, 5, 1, "", "multiply"], [236, 5, 1, "", "nan_to_num"], [237, 5, 1, "", "negative"], [238, 5, 1, "", "new_stream"], [239, 5, 1, "", "not_equal"], [240, 5, 1, "", "ones"], [241, 5, 1, "", "ones_like"], [242, 5, 1, "", "outer"], [243, 5, 1, "", "pad"], [244, 5, 1, "", "partition"], [245, 5, 1, "", "power"], [246, 5, 1, "", "prod"], [247, 5, 1, "", "put_along_axis"], [248, 5, 1, "", "quantize"], [249, 5, 1, "", "quantized_matmul"], [250, 5, 1, "", "radians"], [264, 5, 1, "", "real"], [265, 5, 1, "", "reciprocal"], [266, 5, 1, "", "remainder"], [267, 5, 1, "", "repeat"], [268, 5, 1, "", "reset_peak_memory"], [269, 5, 1, "", "reshape"], [270, 5, 1, "", "right_shift"], [271, 5, 1, "", "roll"], [272, 5, 1, "", "round"], [273, 5, 1, "", "rsqrt"], [274, 5, 1, "", "save"], [275, 5, 1, "", "save_gguf"], [276, 5, 1, "", "save_safetensors"], [277, 5, 1, "", "savez"], [278, 5, 1, "", "savez_compressed"], [279, 5, 1, "", "set_cache_limit"], [280, 5, 1, "", "set_default_device"], [281, 5, 1, "", "set_default_stream"], [282, 5, 1, "", "set_memory_limit"], [283, 5, 1, "", "set_wired_limit"], [284, 5, 1, "", "sigmoid"], [285, 5, 1, "", "sign"], [286, 5, 1, "", "sin"], [287, 5, 1, "", "sinh"], [288, 5, 1, "", "slice"], [289, 5, 1, "", "slice_update"], [290, 5, 1, "", "softmax"], [291, 5, 1, "", "sort"], [292, 5, 1, "", "split"], [293, 5, 1, "", "sqrt"], [294, 5, 1, "", "square"], [295, 5, 1, "", "squeeze"], [296, 5, 1, "", "stack"], [297, 5, 1, "", "std"], [298, 5, 1, "", "stop_gradient"], [299, 5, 1, "", "stream"], [300, 5, 1, "", "subtract"], [301, 5, 1, "", "sum"], [302, 5, 1, "", "swapaxes"], [303, 5, 1, "", "synchronize"], [304, 5, 1, "", "take"], [305, 5, 1, "", "take_along_axis"], [306, 5, 1, "", "tan"], [307, 5, 1, "", "tanh"], [308, 5, 1, "", "tensordot"], [309, 5, 1, "", "tile"], [310, 5, 1, "", "topk"], [311, 5, 1, "", "trace"], [312, 5, 1, "", "transpose"], [313, 5, 1, "", "tri"], [314, 5, 1, "", "tril"], [315, 5, 1, "", "triu"], [316, 5, 1, "", "unflatten"], [317, 5, 1, "", "value_and_grad"], [318, 5, 1, "", "var"], [319, 5, 1, "", "view"], [320, 5, 1, "", "vjp"], [321, 5, 1, "", "vmap"], [322, 5, 1, "", "where"], [323, 5, 1, "", "zeros"], [324, 5, 1, "", "zeros_like"]], "mlx.core.Device": [[10, 4, 1, "", "__init__"]], "mlx.core.Dtype": [[11, 4, 1, "", "__init__"]], "mlx.core.DtypeCategory": [[12, 4, 1, "", "__init__"]], "mlx.core.Stream": [[334, 4, 1, "", "__init__"]], "mlx.core.array": [[32, 6, 1, "", "T"], [31, 4, 1, "", "__init__"], [33, 4, 1, "", "abs"], [34, 4, 1, "", "all"], [35, 4, 1, "", "any"], [36, 4, 1, "", "argmax"], [37, 4, 1, "", "argmin"], [38, 4, 1, "", "astype"], [39, 6, 1, "", "at"], [40, 4, 1, "", "conj"], [41, 4, 1, "", "cos"], [42, 4, 1, "", "cummax"], [43, 4, 1, "", "cummin"], [44, 4, 1, "", "cumprod"], [45, 4, 1, "", "cumsum"], [46, 4, 1, "", "diag"], [47, 4, 1, "", "diagonal"], [48, 6, 1, "", "dtype"], [49, 4, 1, "", "exp"], [50, 4, 1, "", "flatten"], [51, 4, 1, "", "item"], [52, 6, 1, "", "itemsize"], [53, 4, 1, "", "log"], [54, 4, 1, "", "log10"], [55, 4, 1, "", "log1p"], [56, 4, 1, "", "log2"], [57, 4, 1, "", "logsumexp"], [58, 4, 1, "", "max"], [59, 4, 1, "", "mean"], [60, 4, 1, "", "min"], [61, 4, 1, "", "moveaxis"], [62, 6, 1, "", "nbytes"], [63, 6, 1, "", "ndim"], [64, 4, 1, "", "prod"], [65, 4, 1, "", "reciprocal"], [66, 4, 1, "", "reshape"], [67, 4, 1, "", "round"], [68, 4, 1, "", "rsqrt"], [69, 6, 1, "", "shape"], [70, 4, 1, "", "sin"], [71, 6, 1, "", "size"], [72, 4, 1, "", "split"], [73, 4, 1, "", "sqrt"], [74, 4, 1, "", "square"], [75, 4, 1, "", "squeeze"], [76, 4, 1, "", "std"], [77, 4, 1, "", "sum"], [78, 4, 1, "", "swapaxes"], [79, 4, 1, "", "tolist"], [80, 4, 1, "", "transpose"], [81, 4, 1, "", "var"], [82, 4, 1, "", "view"]], "mlx.core.custom_function": [[118, 4, 1, "", "__init__"]], "mlx.core.distributed": [[126, 3, 1, "", "Group"], [127, 5, 1, "", "all_gather"], [128, 5, 1, "", "all_sum"], [129, 5, 1, "", "init"], [130, 5, 1, "", "is_available"], [131, 5, 1, "", "recv"], [132, 5, 1, "", "recv_like"], [133, 5, 1, "", "send"]], "mlx.core.distributed.Group": [[126, 4, 1, "", "__init__"]], "mlx.core.fast": [[150, 5, 1, "", "layer_norm"], [151, 5, 1, "", "metal_kernel"], [152, 5, 1, "", "rms_norm"], [153, 5, 1, "", "rope"], [154, 5, 1, "", "scaled_dot_product_attention"]], "mlx.core.fft": [[155, 5, 1, "", "fft"], [156, 5, 1, "", "fft2"], [157, 5, 1, "", "fftn"], [158, 5, 1, "", "ifft"], [159, 5, 1, "", "ifft2"], [160, 5, 1, "", "ifftn"], [161, 5, 1, "", "irfft"], [162, 5, 1, "", "irfft2"], [163, 5, 1, "", "irfftn"], [164, 5, 1, "", "rfft"], [165, 5, 1, "", "rfft2"], [166, 5, 1, "", "rfftn"]], "mlx.core.finfo": [[167, 4, 1, "", "__init__"]], "mlx.core.linalg": [[197, 5, 1, "", "cholesky"], [198, 5, 1, "", "cholesky_inv"], [199, 5, 1, "", "cross"], [200, 5, 1, "", "eigh"], [201, 5, 1, "", "eigvalsh"], [202, 5, 1, "", "inv"], [203, 5, 1, "", "lu"], [204, 5, 1, "", "lu_factor"], [205, 5, 1, "", "norm"], [206, 5, 1, "", "pinv"], [207, 5, 1, "", "qr"], [208, 5, 1, "", "solve"], [209, 5, 1, "", "solve_triangular"], [210, 5, 1, "", "svd"], [211, 5, 1, "", "tri_inv"]], "mlx.core.metal": [[228, 5, 1, "", "device_info"], [229, 5, 1, "", "is_available"], [230, 5, 1, "", "start_capture"], [231, 5, 1, "", "stop_capture"]], "mlx.core.random": [[251, 5, 1, "", "bernoulli"], [252, 5, 1, "", "categorical"], [253, 5, 1, "", "gumbel"], [254, 5, 1, "", "key"], [255, 5, 1, "", "laplace"], [256, 5, 1, "", "multivariate_normal"], [257, 5, 1, "", "normal"], [258, 5, 1, "", "permutation"], [259, 5, 1, "", "randint"], [260, 5, 1, "", "seed"], [261, 5, 1, "", "split"], [262, 5, 1, "", "truncated_normal"], [263, 5, 1, "", "uniform"]], "mlx.nn": [[346, 3, 1, "", "ALiBi"], [347, 3, 1, "", "AvgPool1d"], [348, 3, 1, "", "AvgPool2d"], [349, 3, 1, "", "AvgPool3d"], [350, 3, 1, "", "BatchNorm"], [351, 3, 1, "", "CELU"], [352, 3, 1, "", "Conv1d"], [353, 3, 1, "", "Conv2d"], [354, 3, 1, "", "Conv3d"], [355, 3, 1, "", "ConvTranspose1d"], [356, 3, 1, "", "ConvTranspose2d"], [357, 3, 1, "", "ConvTranspose3d"], [358, 3, 1, "", "Dropout"], [359, 3, 1, "", "Dropout2d"], [360, 3, 1, "", "Dropout3d"], [361, 3, 1, "", "ELU"], [362, 3, 1, "", "Embedding"], [363, 3, 1, "", "GELU"], [364, 3, 1, "", "GLU"], [365, 3, 1, "", "GRU"], [366, 3, 1, "", "GroupNorm"], [367, 3, 1, "", "HardShrink"], [368, 3, 1, "", "HardTanh"], [369, 3, 1, "", "Hardswish"], [370, 3, 1, "", "InstanceNorm"], [371, 3, 1, "", "LSTM"], [372, 3, 1, "", "LayerNorm"], [373, 3, 1, "", "LeakyReLU"], [374, 3, 1, "", "Linear"], [375, 3, 1, "", "LogSigmoid"], [376, 3, 1, "", "LogSoftmax"], [377, 3, 1, "", "MaxPool1d"], [378, 3, 1, "", "MaxPool2d"], [379, 3, 1, "", "MaxPool3d"], [380, 3, 1, "", "Mish"], [475, 3, 1, "", "Module"], [401, 3, 1, "", "MultiHeadAttention"], [402, 3, 1, "", "PReLU"], [403, 3, 1, "", "QuantizedEmbedding"], [404, 3, 1, "", "QuantizedLinear"], [405, 3, 1, "", "RMSNorm"], [406, 3, 1, "", "RNN"], [407, 3, 1, "", "ReLU"], [408, 3, 1, "", "ReLU6"], [409, 3, 1, "", "RoPE"], [410, 3, 1, "", "SELU"], [411, 3, 1, "", "Sequential"], [412, 3, 1, "", "SiLU"], [413, 3, 1, "", "Sigmoid"], [414, 3, 1, "", "SinusoidalPositionalEncoding"], [415, 3, 1, "", "Softmax"], [416, 3, 1, "", "Softmin"], [417, 3, 1, "", "Softplus"], [418, 3, 1, "", "Softshrink"], [419, 3, 1, "", "Softsign"], [420, 3, 1, "", "Step"], [421, 3, 1, "", "Tanh"], [422, 3, 1, "", "Transformer"], [423, 3, 1, "", "Upsample"], [325, 5, 1, "", "average_gradients"], [432, 3, 1, "", "celu"], [433, 3, 1, "", "elu"], [434, 3, 1, "", "gelu"], [435, 3, 1, "", "gelu_approx"], [436, 3, 1, "", "gelu_fast_approx"], [437, 3, 1, "", "glu"], [438, 3, 1, "", "hard_shrink"], [439, 3, 1, "", "hard_tanh"], [440, 3, 1, "", "hardswish"], [441, 3, 1, "", "leaky_relu"], [442, 3, 1, "", "log_sigmoid"], [443, 3, 1, "", "log_softmax"], [458, 3, 1, "", "mish"], [459, 3, 1, "", "prelu"], [326, 5, 1, "", "quantize"], [460, 3, 1, "", "relu"], [461, 3, 1, "", "relu6"], [462, 3, 1, "", "selu"], [463, 3, 1, "", "sigmoid"], [464, 3, 1, "", "silu"], [465, 3, 1, "", "softmax"], [466, 3, 1, "", "softmin"], [467, 3, 1, "", "softplus"], [468, 3, 1, "", "softshrink"], [469, 3, 1, "", "step"], [470, 3, 1, "", "tanh"], [327, 5, 1, "", "value_and_grad"]], "mlx.nn.Module": [[381, 4, 1, "", "apply"], [382, 4, 1, "", "apply_to_modules"], [383, 4, 1, "", "children"], [384, 4, 1, "", "eval"], [385, 4, 1, "", "filter_and_map"], [386, 4, 1, "", "freeze"], [387, 4, 1, "", "leaf_modules"], [388, 4, 1, "", "load_weights"], [389, 4, 1, "", "modules"], [390, 4, 1, "", "named_modules"], [391, 4, 1, "", "parameters"], [392, 4, 1, "", "save_weights"], [393, 4, 1, "", "set_dtype"], [394, 6, 1, "", "state"], [395, 4, 1, "", "train"], [396, 4, 1, "", "trainable_parameters"], [397, 6, 1, "", "training"], [398, 4, 1, "", "unfreeze"], [399, 4, 1, "", "update"], [400, 4, 1, "", "update_modules"]], "mlx.nn.init": [[424, 5, 1, "", "constant"], [425, 5, 1, "", "glorot_normal"], [426, 5, 1, "", "glorot_uniform"], [427, 5, 1, "", "he_normal"], [428, 5, 1, "", "he_uniform"], [429, 5, 1, "", "identity"], [430, 5, 1, "", "normal"], [431, 5, 1, "", "uniform"]], "mlx.nn.losses": [[444, 3, 1, "", "binary_cross_entropy"], [445, 3, 1, "", "cosine_similarity_loss"], [446, 3, 1, "", "cross_entropy"], [447, 3, 1, "", "gaussian_nll_loss"], [448, 3, 1, "", "hinge_loss"], [449, 3, 1, "", "huber_loss"], [450, 3, 1, "", "kl_div_loss"], [451, 3, 1, "", "l1_loss"], [452, 3, 1, "", "log_cosh_loss"], [453, 3, 1, "", "margin_ranking_loss"], [454, 3, 1, "", "mse_loss"], [455, 3, 1, "", "nll_loss"], [456, 3, 1, "", "smooth_l1_loss"], [457, 3, 1, "", "triplet_loss"]], "mlx.optimizers": [[478, 3, 1, "", "AdaDelta"], [479, 3, 1, "", "Adafactor"], [480, 3, 1, "", "Adagrad"], [481, 3, 1, "", "Adam"], [482, 3, 1, "", "AdamW"], [483, 3, 1, "", "Adamax"], [484, 3, 1, "", "Lion"], [497, 3, 1, "", "Optimizer"], [489, 3, 1, "", "RMSprop"], [490, 3, 1, "", "SGD"], [328, 5, 1, "", "clip_grad_norm"], [491, 5, 1, "", "cosine_decay"], [492, 5, 1, "", "exponential_decay"], [493, 5, 1, "", "join_schedules"], [494, 5, 1, "", "linear_schedule"], [495, 5, 1, "", "step_decay"]], "mlx.optimizers.Optimizer": [[485, 4, 1, "", "apply_gradients"], [486, 4, 1, "", "init"], [487, 6, 1, "", "state"], [488, 4, 1, "", "update"]], "mlx.utils": [[329, 5, 1, "", "tree_flatten"], [330, 5, 1, "", "tree_map"], [331, 5, 1, "", "tree_map_with_path"], [332, 5, 1, "", "tree_reduce"], [333, 5, 1, "", "tree_unflatten"]]}, "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"]}, "objtypes": {"0": "cpp:function", "1": "cpp:functionParam", "2": "cpp:templateParam", "3": "py:class", "4": "py:method", "5": "py:function", "6": "py:property"}, "terms": {"": [0, 1, 2, 5, 6, 7, 48, 52, 63, 99, 120, 122, 156, 157, 159, 160, 162, 163, 165, 166, 177, 198, 205, 210, 213, 226, 242, 248, 252, 272, 275, 276, 297, 299, 317, 318, 319, 321, 327, 345, 348, 349, 365, 371, 378, 379, 385, 386, 388, 392, 393, 394, 398, 406, 477, 486, 487, 499, 502, 504, 505, 509, 510, 511, 512], "0": [0, 1, 2, 4, 5, 6, 7, 9, 10, 15, 19, 39, 46, 47, 50, 67, 72, 76, 81, 84, 85, 97, 100, 104, 105, 106, 107, 108, 109, 110, 118, 123, 124, 149, 151, 154, 168, 172, 177, 183, 193, 200, 202, 203, 205, 207, 211, 236, 243, 251, 255, 257, 258, 263, 267, 272, 279, 283, 288, 289, 292, 296, 297, 311, 313, 314, 315, 316, 317, 318, 321, 325, 328, 329, 331, 332, 345, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 363, 366, 367, 370, 372, 373, 377, 378, 379, 402, 407, 409, 414, 418, 420, 422, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 435, 436, 438, 439, 440, 441, 444, 446, 448, 449, 453, 456, 457, 459, 460, 461, 462, 468, 469, 472, 475, 478, 479, 481, 482, 483, 484, 486, 489, 490, 491, 492, 493, 494, 495, 499, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511], "00005": 5, "0001": 414, "0005": 435, "001": 479, "00364": 5, "01": [5, 373, 441, 482], "0137595": 427, "015": 436, "0184009": 428, "02264": 426, "024": 505, "02765": 427, "0300242": 428, "044715": [363, 435], "0485873": 446, "05": [17, 185, 350, 366, 370, 372, 405], "0507": 462, "05202": 6, "06": [447, 457, 478], "0638": 453, "06450": 372, "0645099": 430, "06561": 492, "06675": 484, "07467": 405, "08": [17, 185, 445, 480, 481, 482, 483, 489], "08022": 370, "081": 495, "08415": 436, "08494": 366, "08619": 428, "08681": [380, 458], "09864": 6, "0999938": 493, "0999961": 491, "0f": 0, "1": [0, 1, 2, 3, 4, 6, 7, 15, 19, 29, 30, 39, 47, 50, 85, 104, 105, 106, 107, 108, 109, 110, 118, 123, 124, 145, 146, 147, 148, 151, 154, 155, 156, 158, 159, 161, 162, 163, 164, 165, 166, 168, 180, 184, 191, 193, 198, 199, 200, 201, 203, 205, 207, 223, 227, 242, 244, 248, 252, 255, 256, 257, 263, 282, 284, 288, 289, 291, 304, 310, 311, 316, 317, 328, 331, 332, 336, 345, 347, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 363, 364, 365, 366, 370, 371, 372, 374, 377, 402, 405, 406, 409, 413, 414, 420, 423, 425, 426, 427, 428, 429, 430, 431, 432, 433, 435, 436, 437, 439, 442, 443, 444, 445, 446, 447, 448, 449, 450, 452, 453, 455, 456, 457, 462, 463, 465, 466, 467, 469, 472, 475, 477, 478, 479, 480, 481, 482, 483, 484, 486, 489, 490, 491, 492, 493, 494, 495, 502, 503, 504, 505, 506, 507, 509, 510, 511, 512], "10": [0, 3, 6, 7, 193, 215, 272, 277, 330, 345, 388, 472, 493, 495, 502, 503, 506], "100": [2, 5, 6, 444, 494, 502, 505, 508, 512], "1000": [2, 491, 502], "10000": 409, "101": 494, "1024": [1, 6], "105361": 444, "10_000": 5, "10x": 484, "11": 205, "12": [6, 180, 193, 493], "1212": 478, "123": [503, 507], "12451": 426, "128": [277, 345], "13": 9, "14": [9, 193], "15": [1, 9, 193, 205, 283, 332, 502], "150594": 425, "15268": 427, "16": [1, 151, 336, 347, 349, 370, 377, 379, 381, 475], "1606": 436, "1607": [370, 372], "16384": 180, "16506": 428, "168": 503, "17": [4, 9], "177208": 427, "18": 193, "1803": 366, "1908": [380, 458], "1910": 405, "191107": 425, "192": 503, "1985": 205, "1_000": 5, "1d": [0, 104, 108, 111, 275, 305], "1e": [0, 5, 7, 17, 185, 350, 366, 370, 372, 373, 405, 445, 447, 457, 477, 478, 479, 480, 481, 482, 483, 486, 489, 491, 492, 493, 494, 495], "1e3": 502, "1st": 248, "2": [0, 1, 2, 4, 5, 6, 7, 39, 105, 109, 118, 123, 124, 140, 146, 147, 148, 156, 159, 161, 162, 163, 164, 165, 166, 168, 180, 191, 193, 197, 198, 199, 200, 201, 202, 203, 205, 206, 207, 210, 211, 217, 223, 248, 256, 261, 288, 289, 308, 311, 313, 314, 315, 316, 328, 332, 336, 345, 347, 348, 349, 353, 356, 363, 373, 377, 378, 379, 405, 414, 423, 424, 425, 426, 427, 428, 429, 430, 431, 435, 446, 447, 449, 456, 457, 472, 475, 477, 478, 480, 481, 482, 486, 489, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512], "20": [180, 193, 205], "200": [6, 493, 505], "2002": 6, "2011": 480, "2012": [478, 489], "2015": [359, 481, 483], "2019": [6, 482], "2020": 6, "2021": 6, "20397": 444, "20_000": 6, "21": [6, 193, 495], "2104": 6, "223144": 444, "223404": 426, "225": 205, "225763": 453, "2302": 484, "23607": [205, 207], "24": 193, "24264": 205, "247": 6, "25": [9, 402, 423], "25211": 427, "256": [1, 7, 151], "256995": 453, "27": 4, "28": [180, 193], "2d": [0, 105, 109, 124, 248, 350, 359], "2nd": 248, "2x": 509, "3": [0, 1, 2, 4, 6, 9, 85, 106, 110, 118, 146, 148, 168, 191, 193, 199, 200, 201, 205, 207, 288, 289, 316, 328, 332, 349, 354, 357, 363, 379, 423, 426, 428, 435, 440, 479, 484, 499, 502, 503, 504, 506, 509, 510], "30": 479, "3118": 509, "32": [1, 6, 7, 93, 248, 249, 336, 348, 349, 378, 379, 405, 502], "32mib": 325, "330": 6, "33333": 423, "33554432": 325, "348587": 446, "363207": 425, "36788": 502, "379159": 426, "380709": 430, "39": 6, "3d": [0, 2, 106, 110, 350, 360, 423], "3f": [2, 7, 502], "3x": 2, "4": [0, 1, 2, 6, 122, 151, 154, 168, 173, 193, 205, 248, 249, 277, 288, 316, 326, 332, 336, 347, 348, 349, 350, 370, 377, 378, 379, 403, 404, 422, 423, 425, 426, 427, 444, 502, 503, 504, 506, 510, 512], "4096": [2, 502, 505, 512], "40x": 1, "41421": 205, "417497": 431, "42": 333, "437": 6, "44": 6, "447214": 207, "458835": 427, "475": 6, "48095": 425, "4d": [1, 423], "4m": 1, "5": [0, 1, 2, 5, 6, 9, 193, 205, 251, 282, 288, 332, 347, 350, 358, 359, 360, 363, 367, 370, 377, 418, 423, 424, 427, 428, 435, 438, 456, 468, 472, 477, 489, 491, 492, 502, 505, 506], "50": [0, 212], "500": [6, 512], "510826": 444, "512": [3, 6, 422, 512], "534422": 430, "539245": 444, "53947": 425, "55": 1, "559": 2, "5701": 478, "573409": 453, "57771": 207, "579": 6, "5f": 5, "6": [1, 2, 6, 118, 193, 205, 277, 288, 408, 422, 426, 435, 436, 440, 447, 457, 461, 489, 502, 506, 510], "61278": 425, "617261": 431, "628": 6, "633": 6, "639": 505, "64": [0, 1, 93, 122, 173, 248, 249, 326, 336, 403, 404], "64331": 428, "666329": 428, "66667": 423, "67326": 462, "676": 1, "690": 6, "6967": 427, "7": [2, 6, 193, 205, 248, 506], "702": [363, 436], "707107": 200, "71828": 502, "74166": 205, "74597": 205, "75": 423, "75596": 453, "75787": 427, "765166": 453, "773433": 453, "774": 2, "776856": 426, "793615": 428, "79854": 428, "7b": 6, "7m": 1, "8": [0, 1, 2, 6, 9, 205, 248, 336, 348, 349, 370, 378, 379, 422, 445, 478, 479, 480, 481, 482, 483, 489, 502, 506, 510, 512], "8192": [6, 180], "84804": 205, "863726": 431, "883935": 431, "890597": 426, "894427": 207, "89613": 425, "8gb": 6, "8x": 1, "9": [4, 9, 205, 446, 478, 481, 482, 483, 484, 486, 492, 495, 509], "90041": 426, "912766": 426, "916291": 444, "95": 7, "982273": 430, "99": [484, 489], "995016": 425, "999": [481, 482, 483], "A": [0, 2, 6, 8, 9, 10, 69, 83, 99, 146, 147, 150, 151, 152, 154, 177, 191, 192, 198, 200, 201, 203, 205, 207, 210, 213, 222, 223, 224, 228, 232, 248, 251, 252, 253, 255, 256, 257, 258, 259, 262, 263, 292, 296, 299, 317, 320, 321, 326, 327, 328, 329, 330, 331, 332, 333, 334, 345, 350, 359, 365, 366, 370, 372, 385, 389, 390, 393, 399, 400, 405, 411, 414, 422, 425, 426, 428, 436, 457, 458, 475, 477, 481, 483, 485, 486, 488, 493, 502, 503, 504, 505, 507, 508, 509], "AS": 172, "And": [4, 6, 423], "As": [7, 39, 304, 345, 503], "At": [98, 316, 503], "But": [504, 512], "By": [6, 326, 393, 444, 503, 505, 509], "For": [0, 1, 2, 4, 6, 9, 39, 118, 154, 172, 191, 205, 248, 333, 345, 350, 359, 363, 381, 386, 395, 398, 404, 409, 414, 423, 425, 426, 427, 428, 444, 472, 477, 499, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512], "If": [0, 1, 2, 4, 6, 9, 16, 17, 18, 19, 27, 28, 29, 30, 79, 83, 84, 85, 98, 100, 111, 114, 115, 116, 117, 123, 124, 127, 128, 129, 131, 132, 133, 142, 150, 152, 153, 154, 164, 165, 166, 170, 171, 177, 185, 197, 198, 199, 205, 210, 213, 222, 223, 224, 226, 227, 232, 236, 240, 243, 244, 246, 247, 252, 256, 258, 267, 271, 279, 282, 290, 291, 292, 297, 301, 303, 304, 305, 308, 310, 311, 317, 318, 321, 323, 325, 326, 330, 332, 350, 352, 353, 354, 355, 356, 357, 366, 372, 374, 386, 388, 398, 404, 406, 409, 411, 414, 423, 444, 446, 457, 479, 481, 482, 502, 503, 504, 505, 507, 508, 511, 512, 513], "In": [0, 1, 2, 6, 7, 39, 154, 223, 248, 330, 345, 359, 366, 475, 478, 480, 481, 483, 484, 485, 501, 502, 503, 504, 505, 507, 508, 511, 512], "It": [2, 6, 9, 132, 177, 281, 317, 328, 332, 345, 400, 404, 485, 497, 503, 507, 509, 511], "Its": [345, 504], "No": [2, 6, 200, 201, 503], "Not": [99, 239, 502], "ON": [3, 4, 9], "Of": 505, "On": [1, 502, 505, 508], "One": [155, 158, 164, 243, 273, 502, 504, 505, 507], "THE": 9, "That": 6, "The": [0, 1, 2, 3, 4, 6, 7, 8, 9, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 48, 52, 62, 63, 69, 79, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 121, 122, 123, 124, 127, 128, 129, 131, 132, 133, 134, 135, 136, 137, 139, 140, 141, 143, 144, 145, 146, 147, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 171, 172, 173, 175, 176, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 192, 193, 194, 195, 196, 199, 200, 201, 203, 204, 205, 207, 208, 209, 210, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 230, 232, 233, 234, 235, 237, 239, 240, 241, 242, 243, 244, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 261, 262, 263, 264, 265, 266, 267, 269, 270, 271, 275, 276, 279, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 293, 294, 295, 296, 297, 298, 299, 300, 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, 328, 329, 330, 331, 332, 333, 336, 338, 347, 348, 349, 350, 352, 353, 354, 355, 356, 357, 358, 359, 360, 362, 364, 365, 366, 370, 371, 372, 374, 377, 378, 379, 381, 382, 386, 388, 392, 393, 394, 395, 398, 399, 400, 401, 403, 404, 405, 406, 409, 411, 414, 420, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 437, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 469, 472, 475, 477, 478, 479, 480, 481, 482, 483, 484, 487, 489, 490, 491, 494, 497, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513], "Then": [5, 9], "There": [1, 345, 423, 502], "These": [1, 2, 99, 247, 305, 446, 512], "To": [0, 2, 3, 5, 6, 7, 9, 203, 279, 345, 472, 477, 502, 503, 504, 505, 510], "With": 504, "_": [1, 3, 5, 6, 331, 345, 491, 492, 493, 494, 495, 499, 502, 508, 512], "__call__": [1, 6, 7, 345, 475, 504], "__init__": [2, 6, 7, 10, 11, 12, 31, 118, 126, 167, 334, 345, 475], "__main__": [2, 6], "__name__": [2, 6], "_a": 2, "_ext": 2, "_f": 205, "_in": [425, 426], "_out": [425, 426], "_p": 457, "_val": 439, "a1": 172, "a2": 172, "a_": 205, "a_max": [0, 98], "a_min": [0, 98], "a_ndim": 1, "a_shap": 1, "a_strid": 1, "a_view": 509, "ab": [0, 17, 185, 205, 317, 366, 370, 372, 380, 405, 436, 458, 502, 504], "abil": 503, "abl": [2, 4, 248, 507], "abort": 118, "about": [1, 2, 6, 7, 137, 228, 508, 512], "abov": [1, 2, 6, 248, 314, 345, 423, 503, 504, 505, 506, 507, 508, 512], "absolut": [0, 13, 17, 185, 435, 436, 456, 503], "acc": 332, "acceler": [4, 350], "accept": [503, 507], "access": [0, 6, 51, 345, 475, 486, 503, 508, 512], "accord": [0, 253, 322, 326, 401, 425, 426, 427, 428], "accordingli": 2, "accumul": [332, 405], "accuraci": 7, "accustom": 6, "achiev": [345, 503], "across": [1, 2, 9, 325, 366, 503], "act": [2, 452], "action": 345, "activ": [2, 9, 174, 359, 420, 422, 438, 458, 468, 469, 471, 502], "actual": [6, 19, 388, 475, 508], "ad": [0, 1, 2, 5, 9, 150, 331, 370, 475, 478, 479, 480, 481, 482, 483, 489, 503, 508, 511], "adadelta": 477, "adafactor": 477, "adagrad": 477, "adam": [477, 483, 484, 493, 494], "adamax": 477, "adamw": [477, 484], "adapt": [478, 479, 480, 503], "add": [0, 1, 2, 3, 4, 6, 15, 39, 144, 218, 243, 248, 352, 353, 354, 355, 356, 357, 504, 505, 507, 512], "add_argu": 6, "add_depend": 2, "add_execut": 4, "add_fun": 504, "add_librari": 2, "addit": [0, 2, 4, 6, 9, 14, 15, 146, 150, 152, 154, 213, 350, 366, 372, 401, 405, 475, 505], "addmm": 0, "address": 2, "adjac": 359, "advanc": [6, 502], "advantag": 512, "advis": 509, "affin": [350, 366, 370, 372, 374, 404], "after": [2, 6, 7, 29, 97, 168, 170, 173, 244, 248, 350, 366, 372, 381, 382, 386, 388, 395, 398, 399, 400, 401, 422, 456, 502, 503, 512], "after_1": 243, "after_2": 243, "after_i": 243, "after_n": 243, "afternoon": 6, "again": [6, 9, 345, 502], "against": [0, 4, 94], "aggreg": [401, 503], "ago": 6, "ai": 118, "aim": 503, "ainv": [202, 211], "albeit": 512, "algebra": 8, "algorithm": [423, 484], "alia": [101, 102, 363], "alibi": 345, "align": [198, 248, 365, 371], "align_corn": 423, "all": [0, 1, 2, 3, 7, 9, 17, 29, 39, 86, 87, 88, 99, 105, 106, 107, 109, 110, 118, 127, 128, 129, 147, 149, 157, 160, 163, 166, 172, 173, 210, 223, 243, 244, 271, 295, 325, 326, 345, 381, 382, 386, 389, 390, 391, 396, 398, 401, 414, 422, 423, 472, 475, 497, 499, 502, 506, 507, 508, 510, 513], "all_avg": 503, "all_reduce_grad": 503, "all_reduce_s": 325, "all_sum": 503, "allclos": [0, 1, 151], "alloc": [2, 175, 279, 282, 475], "allow": [0, 1, 2, 146, 148, 191, 328, 345, 400, 475, 497, 503, 506, 507, 510], "allow_col_major": [0, 103], "almost": [6, 503], "alon": [2, 509], "along": [0, 2, 27, 28, 99, 100, 114, 115, 116, 117, 127, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 172, 173, 180, 199, 205, 247, 258, 267, 271, 290, 292, 296, 304, 305, 308, 309, 310, 311, 319, 345, 364, 406, 437], "alpha": [0, 2, 15, 248, 351, 361, 432, 433, 457, 459, 462, 482, 489], "alpha_": 2, "alreadi": [2, 3, 6, 503], "also": [0, 1, 2, 4, 6, 7, 8, 9, 12, 14, 89, 91, 92, 125, 134, 135, 139, 157, 160, 163, 166, 178, 179, 194, 195, 196, 218, 225, 233, 235, 239, 245, 248, 266, 270, 300, 326, 327, 338, 345, 385, 399, 401, 403, 404, 412, 434, 462, 464, 471, 477, 502, 503, 504, 505, 506, 507, 508, 509, 510, 513], "altern": 499, "although": 503, "alwai": [1, 84, 174, 183, 329, 503, 504, 505], "am": 6, "among": 2, "amount": [6, 176, 282, 347, 377, 504], "amus": 6, "an": [0, 1, 2, 3, 4, 6, 7, 9, 11, 16, 18, 31, 85, 86, 87, 88, 95, 103, 104, 105, 106, 107, 108, 109, 110, 126, 131, 132, 133, 142, 146, 148, 149, 150, 154, 168, 171, 181, 183, 186, 197, 205, 213, 234, 240, 241, 243, 246, 247, 248, 249, 258, 267, 269, 271, 272, 282, 283, 292, 295, 302, 304, 305, 308, 309, 313, 316, 321, 323, 324, 329, 330, 331, 332, 336, 345, 358, 363, 366, 371, 372, 374, 381, 401, 402, 404, 406, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 435, 459, 472, 477, 478, 488, 492, 497, 499, 501, 502, 503, 504, 505, 506, 508, 509, 510, 511, 512, 513], "anaconda": 503, "anchor": 457, "angl": [121, 250, 373], "angular": [153, 409], "ani": [0, 1, 2, 6, 8, 19, 99, 118, 129, 325, 329, 330, 331, 332, 333, 345, 363, 381, 382, 385, 394, 404, 422, 423, 472, 494, 501, 502, 503, 505, 508, 510, 511, 512], "anonym": 502, "anoth": [0, 94, 98, 191, 223, 300, 322, 336, 345, 381, 502, 504, 505, 506, 512], "anwywher": 9, "anyhow": 6, "anymor": 6, "anyth": [6, 317, 503, 508], "anytim": 508, "api": [1, 2, 85, 146, 148, 183, 363, 503, 504, 505], "aplu": 206, "app": 9, "append": [6, 223, 502, 508], "appl": [2, 6, 8, 9, 512], "appli": [0, 39, 153, 154, 172, 210, 330, 331, 332, 345, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 359, 360, 361, 363, 364, 366, 367, 368, 369, 370, 372, 373, 374, 375, 376, 377, 378, 379, 380, 382, 395, 402, 404, 405, 406, 407, 408, 410, 412, 413, 415, 416, 417, 418, 419, 420, 421, 423, 432, 433, 434, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 472, 481, 482, 485, 488, 494, 497, 502, 503], "applic": [3, 9], "apply_fn": 382, "apply_gradi": 477, "apply_to_modul": [345, 386], "approach": [452, 505], "appropri": 502, "approx": 363, "approxim": [17, 363, 434, 435, 436], "ar": [0, 1, 2, 5, 6, 7, 8, 9, 17, 19, 83, 85, 93, 94, 95, 98, 99, 107, 111, 118, 124, 129, 131, 132, 142, 149, 151, 154, 156, 157, 159, 160, 162, 163, 165, 166, 168, 173, 177, 185, 186, 187, 188, 189, 190, 191, 192, 200, 201, 203, 205, 207, 213, 223, 242, 243, 244, 248, 249, 251, 252, 253, 258, 259, 262, 263, 271, 277, 278, 295, 296, 304, 317, 320, 321, 326, 329, 330, 336, 350, 352, 353, 354, 355, 356, 357, 358, 359, 360, 366, 370, 372, 374, 388, 401, 404, 423, 444, 446, 447, 471, 475, 477, 484, 486, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512], "arang": [0, 1, 205, 258, 336, 423, 506, 509], "arbitrari": [329, 475, 503], "arbitrarili": [1, 99, 345, 501, 505, 510], "arc": 0, "arcco": 0, "arccosh": 0, "architectur": [6, 9, 228, 345, 400, 512], "archiv": 511, "arcsin": 0, "arcsinh": 0, "arctan": 0, "arctan2": 0, "arctanh": 0, "arg": [2, 6, 11, 19, 85, 126, 142, 146, 147, 148, 167, 183, 277, 278, 334, 507], "arg1": 191, "arg2": 191, "argmax": [0, 7], "argmin": 0, "argnam": [177, 317], "argnum": [2, 118, 177, 317, 505], "argpars": 6, "argpartit": 0, "argsort": 0, "argument": [1, 32, 66, 80, 85, 99, 142, 177, 317, 330, 331, 332, 345, 423, 499, 503, 504, 505, 507, 511, 512, 513], "argumentpars": 6, "ari": [86, 87, 88], "aris": 509, "arm": 9, "arm64": 9, "around": 6, "arr": [0, 274, 506], "arr_0": 511, "arrai": [0, 1, 2, 4, 6, 7, 8, 11, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 121, 122, 123, 124, 127, 128, 131, 132, 133, 134, 135, 136, 137, 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, 168, 169, 170, 171, 172, 173, 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, 232, 233, 234, 235, 236, 237, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 261, 262, 263, 264, 265, 266, 267, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 300, 301, 302, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 328, 336, 345, 350, 371, 381, 388, 391, 396, 402, 423, 424, 425, 426, 427, 428, 429, 430, 431, 437, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 469, 472, 475, 478, 479, 480, 481, 482, 483, 484, 489, 490, 491, 492, 493, 494, 495, 502, 503, 504, 505, 508, 509, 510, 512], "array_equ": [0, 17, 185], "arrayfir": 8, "arxiv": [6, 366, 370, 372, 380, 405, 436, 458, 478, 484], "as_strid": 0, "ascend": [200, 201], "ask": [6, 503, 507], "assert": [1, 2, 151], "assign": [0, 2, 39, 475, 503], "associ": [2, 277, 278, 508], "assum": [0, 6, 93, 199, 200, 201, 207, 330, 345, 366, 503], "astyp": [0, 1, 2, 6, 151, 381, 509], "asynchron": 85, "atleast": 0, "atleast_1d": 0, "atleast_2d": 0, "atleast_3d": 0, "atol": [0, 17, 185], "atom": [1, 151], "atomic_fetch_add_explicit": 1, "atomic_output": [1, 151], "attach": 2, "attempt": [99, 503], "attend": 401, "attent": [154, 386, 401, 414, 422], "attention_norm": 6, "attribut": [1, 10, 11, 12, 31, 167, 334, 394, 475, 497], "audio": 423, "auto": [0, 2, 4, 9, 503, 504], "autom": 505, "automat": [1, 2, 8, 151, 213, 503, 510, 511, 512], "autoregress": 6, "avail": [2, 5, 6, 7, 9, 11, 129, 130, 229, 282, 338, 503, 507, 512], "averag": [325, 347, 348, 349, 478, 479, 481, 482, 483, 503], "avgpool1d": 345, "avgpool2d": 345, "avgpool3d": 345, "avoid": [1, 2, 393, 502, 503], "awai": [2, 6], "awar": [502, 508], "ax": [0, 2, 16, 18, 27, 28, 80, 118, 144, 156, 157, 159, 160, 162, 163, 165, 166, 168, 184, 205, 208, 209, 222, 224, 226, 232, 243, 246, 271, 288, 289, 290, 295, 297, 301, 302, 308, 312, 318, 505], "axes_a": 0, "axes_b": 0, "axi": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 72, 75, 76, 77, 81, 100, 114, 115, 116, 117, 124, 127, 144, 150, 152, 155, 158, 161, 162, 163, 164, 165, 166, 168, 180, 199, 203, 205, 222, 224, 226, 232, 234, 243, 244, 246, 247, 252, 258, 267, 271, 290, 291, 292, 295, 296, 297, 301, 302, 304, 305, 309, 310, 311, 312, 316, 318, 319, 321, 347, 348, 349, 364, 377, 378, 379, 406, 437, 443, 445, 446, 450, 455, 457, 465, 466, 506], "axis1": [0, 47, 78, 124, 302, 311], "axis2": [0, 47, 78, 124, 302, 311], "axpbi": 2, "axpby_": 2, "axpby_gener": 2, "axpby_general_bfloat16": 2, "axpby_general_complex64": 2, "axpby_general_float16": 2, "axpby_general_float32": 2, "axpby_impl": 2, "b": [0, 1, 2, 3, 4, 6, 14, 15, 17, 25, 83, 89, 91, 92, 93, 134, 135, 139, 151, 154, 170, 172, 178, 179, 183, 184, 185, 193, 194, 195, 196, 199, 205, 208, 209, 218, 219, 221, 223, 225, 233, 235, 239, 242, 245, 248, 255, 266, 270, 300, 308, 317, 331, 332, 364, 374, 406, 423, 437, 505, 506, 508, 509, 510, 511, 512], "b1": 172, "b2": 172, "b_": [365, 371], "b_stride": 1, "ba": [481, 483], "back": [6, 118, 229, 509], "backend": [1, 9, 129, 130, 507], "backward": [1, 502, 505], "bad": 508, "balanc": 452, "baltimor": 205, "bandwidth": [502, 503], "base": [0, 2, 4, 153, 215, 217, 245, 409, 422, 475, 477, 483, 497, 499, 502, 506], "base_idx": 1, "basi": 497, "basic": [5, 272, 505], "batch": [6, 15, 93, 154, 172, 173, 223, 256, 350, 352, 353, 354, 355, 356, 357, 359, 360, 365, 371, 401, 406, 423, 508], "batch_idx": 1, "batch_iter": [7, 477], "batch_siz": [7, 477], "batchnorm": 345, "becaus": [6, 174, 345, 502, 503, 504, 508], "becom": 129, "been": [0, 2, 6, 175, 508], "befor": [1, 2, 6, 9, 29, 151, 244, 325, 385, 422, 486, 503, 506, 508], "before_1": 243, "before_2": 243, "before_i": 243, "before_n": 243, "beforehand": 242, "beggin": 271, "begin": [84, 176, 198, 248, 365, 371, 420, 438, 449, 456, 462, 468, 469, 503], "behav": 118, "behavior": [203, 256, 452, 506, 508], "behaviour": [118, 197, 198], "behind": 505, "being": [298, 345], "bell": 2, "below": [2, 9, 205, 313, 315, 336, 423, 503, 508], "bench": 2, "benchmark": [2, 502], "benefici": [359, 360, 508], "benefit": 503, "best": 503, "beta": [0, 2, 15, 122, 248, 350, 366, 370, 372, 456, 477, 481, 482, 483, 484], "beta_": 2, "beta_1": [479, 481, 482, 483, 484], "beta_2": [481, 482, 483, 484], "better": [325, 505, 512], "between": [0, 2, 8, 98, 168, 422, 445, 448, 449, 452, 493, 503, 507, 508, 509, 512], "beyond": [271, 491, 494], "bfloat16": [2, 12, 180, 336, 509], "bfloat16_t": 2, "bia": [6, 122, 150, 173, 248, 249, 330, 345, 352, 353, 354, 355, 356, 357, 365, 371, 372, 374, 386, 388, 398, 401, 404, 406, 481, 482, 483, 486, 505], "bias": [0, 122, 173, 248, 249, 365, 371, 386, 398, 401], "bias_correct": [481, 482], "bicub": 423, "big": [1, 325, 502], "bigger": [6, 479], "bilinear": [1, 423], "binari": [213, 274, 275, 276, 277, 278, 319, 420, 444, 469, 502, 507], "binary_cross_entropi": [345, 502], "bind": 507, "bit": [0, 122, 173, 194, 248, 249, 270, 326, 336, 381, 403, 404, 405], "bitwis": [0, 89, 90, 91, 92, 194, 270], "bitwise_and": 0, "bitwise_invert": 0, "bitwise_or": 0, "bitwise_xor": 0, "block": [0, 2, 6, 93, 422], "block_masked_mm": 0, "block_siz": [0, 93], "bn": 350, "bodi": [1, 151], "bool": [0, 1, 2, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 76, 77, 79, 81, 83, 99, 103, 107, 114, 115, 116, 117, 129, 130, 146, 148, 151, 153, 173, 185, 191, 197, 198, 205, 209, 210, 211, 213, 222, 224, 226, 227, 229, 232, 246, 249, 297, 301, 318, 326, 350, 352, 353, 354, 355, 356, 357, 365, 366, 370, 371, 372, 374, 381, 385, 386, 388, 393, 395, 398, 401, 404, 406, 409, 414, 422, 423, 444, 447, 479, 481, 482, 490], "bool_": [12, 336], "boolean": [0, 17, 83, 154, 185, 186, 187, 188, 189, 190, 191, 219, 220, 221, 336, 397, 506], "both": [1, 2, 14, 89, 91, 92, 134, 135, 139, 178, 179, 191, 194, 195, 196, 205, 218, 225, 233, 235, 239, 245, 252, 266, 270, 300, 326, 347, 348, 349, 370, 371, 377, 378, 379, 477, 502, 503, 504, 505, 510, 512], "bottom": 423, "bound": [0, 259, 262, 263, 363, 431, 502, 506, 512], "boundari": 493, "bracket": 6, "brain": 336, "break": 509, "bregler": 359, "bridg": 503, "broadcast": [0, 2, 14, 17, 89, 91, 92, 94, 95, 98, 134, 135, 139, 154, 171, 178, 179, 185, 194, 195, 196, 218, 223, 225, 233, 235, 239, 245, 247, 251, 252, 256, 259, 262, 263, 266, 270, 300, 305, 322, 401], "broadcast_arrai": [0, 2], "broadcast_to": 0, "broadcasted_input": 2, "brought": 8, "btl_tcp_if_includ": [503, 507], "btl_tcp_link": [503, 507], "buffer": [1, 2, 174, 509], "bui": 6, "build": [3, 4, 6, 8, 427, 475, 502, 504], "build_ext": [2, 9], "build_shared_lib": [2, 9], "built": [1, 2, 4, 9, 508], "bundl": 6, "byte": [52, 62, 174, 175, 176, 279, 282, 283, 325, 336, 507], "c": [0, 1, 2, 6, 15, 205, 350, 352, 353, 354, 355, 356, 357, 359, 360, 370, 371, 509, 510, 512], "c_": [371, 484], "c_in": [104, 105, 106, 107, 108, 109, 110], "c_out": [104, 105, 106, 107, 108, 109, 110], "c_pad": 1, "c_t": [371, 484], "cabl": 503, "cach": [6, 9, 97, 174, 175, 279, 502], "calcul": [205, 206, 444, 447, 453, 479], "call": [2, 3, 6, 7, 32, 97, 129, 132, 170, 176, 183, 325, 345, 362, 386, 398, 403, 411, 475, 477, 486, 502, 503, 504, 505, 507, 508], "callabl": [99, 118, 146, 148, 151, 177, 183, 192, 317, 320, 321, 326, 327, 329, 330, 331, 332, 381, 382, 385, 393, 406, 411, 422, 424, 425, 426, 427, 428, 429, 430, 431, 478, 479, 480, 481, 482, 483, 484, 489, 490, 491, 492, 493, 494, 495], "can": [1, 2, 3, 4, 6, 8, 9, 14, 19, 66, 80, 84, 85, 89, 91, 92, 99, 124, 125, 126, 134, 135, 139, 142, 146, 147, 154, 178, 179, 183, 194, 195, 196, 205, 218, 225, 233, 235, 239, 245, 251, 252, 259, 262, 263, 266, 270, 275, 283, 300, 311, 316, 317, 332, 345, 348, 349, 362, 363, 378, 379, 385, 398, 403, 411, 423, 446, 472, 475, 477, 485, 486, 499, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513], "cannot": [6, 98, 506, 509], "captur": [2, 3, 99, 118, 230, 231, 345, 502], "care": [6, 503, 504, 507, 508], "carefulli": [502, 504], "carri": 2, "cartesian": 227, "case": [2, 6, 127, 128, 129, 131, 132, 133, 157, 160, 161, 163, 164, 165, 166, 168, 197, 198, 199, 200, 201, 202, 203, 204, 206, 207, 208, 209, 210, 211, 223, 269, 295, 316, 348, 349, 359, 378, 379, 420, 438, 456, 462, 468, 469, 485, 486, 502, 503, 504, 505, 507, 510, 511, 512, 513], "cast": [2, 38, 164, 165, 166, 213, 325, 381, 393, 509], "caster": 2, "categor": 6, "categori": [12, 191, 336], "caus": [345, 502, 508], "causal": [6, 154], "caution": 84, "cd": [3, 9], "cdf": [253, 363, 434], "cdot": [436, 445, 448, 464], "ceil": 0, "ceildiv": 1, "cell": 371, "celu": 345, "certain": [395, 502], "chang": [84, 85, 99, 146, 148, 183, 281, 319, 399, 404, 423, 449, 456, 502, 509], "channel": [1, 104, 105, 106, 107, 108, 109, 110, 350, 352, 353, 354, 355, 356, 357, 359, 360], "channel_idx": 1, "charact": 329, "check": [0, 2, 9, 83, 130, 191, 200, 201, 229, 388, 503, 504, 505, 506], "checklist": [503, 507], "checkout": [3, 502], "checkpoint": [422, 477], "chen": 484, "child": 400, "children": 345, "chip": 9, "choleski": 198, "choos": [6, 153, 409, 507], "chosen": 137, "clamp": 168, "clang": 9, "clarifi": 503, "clariti": 505, "class": [2, 6, 7, 10, 11, 12, 31, 118, 126, 167, 334, 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, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 475, 478, 479, 480, 481, 482, 483, 484, 489, 490, 497], "class_pred": 326, "classif": [427, 428], "classifi": 7, "classmethod": [403, 404], "clear": 97, "click": 9, "clip": [0, 328, 444, 479], "clip_threshold": 479, "clipped_grad": 328, "clone": 9, "close": [5, 8, 9, 17, 185], "closer": 330, "cmake": [3, 4, 9], "cmake_arg": 3, "cmake_build_parallel_level": 9, "cmake_build_typ": 9, "cmake_current_list_dir": 2, "cmake_cxx_standard": 4, "cmake_cxx_standard_requir": 4, "cmake_host_system_processor": 9, "cmake_library_output_directori": 2, "cmake_minimum_requir": 4, "cmakebuild": 2, "cmakeextens": 2, "cmakelist": [2, 4], "cmdclass": 2, "co": [0, 2, 118, 414, 505], "code": [1, 151, 502, 503, 504, 508], "coeffici": [2, 478, 479, 481, 482, 483, 484], "col": [103, 313], "cold": 9, "collect": [330, 331, 501], "column": [103, 149, 181, 200, 248], "com": 9, "combin": [6, 210, 332], "come": [2, 6, 503, 505], "command": [2, 3, 4, 9, 503, 507], "command_buff": 2, "common": [477, 502, 508], "commonli": [7, 399, 472, 502], "commun": [8, 126, 129, 130, 325, 507], "communication_typ": 325, "compact": 204, "compar": [2, 83, 502], "comparison": [17, 139, 178, 179, 195, 196, 239], "compat": [6, 146, 148, 154, 183, 252, 256, 363, 511], "compil": [0, 3, 4, 8, 9, 125, 138, 151, 503, 504, 505, 508], "compiled_fun": [502, 504], "compiled_grad_fn": 502, "complement": 90, "complet": [5, 6, 9, 399, 400, 504, 505, 512], "complex": [2, 101, 102, 162, 163, 164, 165, 166, 182, 200, 201, 264, 329, 336, 345, 400, 502, 504, 505], "complex64": [2, 12, 336], "complex64_t": 2, "complexflo": 12, "compon": [2, 4, 6, 210], "compos": [8, 345, 502, 505, 510], "composit": 510, "compress": 278, "compromis": 6, "comput": [0, 1, 2, 5, 6, 7, 8, 9, 114, 115, 116, 117, 118, 122, 137, 145, 153, 177, 192, 193, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 208, 209, 211, 218, 226, 242, 248, 266, 290, 297, 298, 308, 317, 318, 320, 327, 345, 350, 365, 366, 370, 371, 372, 386, 399, 404, 405, 409, 422, 425, 426, 427, 428, 435, 436, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 477, 478, 479, 481, 482, 483, 484, 488, 502, 503, 504, 505, 510, 512], "computation": 508, "compute_encod": 2, "compute_uv": 210, "concaten": [0, 6, 127, 325], "concept": 475, "concis": 6, "concret": [2, 365, 371, 374, 406, 508, 512], "conda": [9, 503], "condit": [0, 322, 502, 512], "config": [2, 4, 503], "configu": 477, "configur": [122, 503], "confirm": [503, 507], "confus": 7, "conj": 102, "conjug": [0, 101], "connect": [503, 507], "consecut": [153, 248, 409], "consequ": 6, "consid": [6, 17, 83, 103, 185, 329, 330, 331, 366, 501, 503], "consider": 502, "const": [0, 1, 2, 447], "constant": [0, 2, 6, 9, 118, 150, 152, 243, 345, 350, 366, 372, 405, 447, 457, 489, 491, 502, 504, 509], "constant_valu": 243, "constitut": 330, "construct": [0, 2, 7, 46, 123, 171, 203, 240, 309, 323], "consult": 503, "consum": 508, "contain": [2, 6, 9, 29, 30, 69, 99, 124, 137, 161, 162, 163, 172, 173, 200, 205, 219, 220, 221, 248, 292, 322, 325, 328, 345, 385, 387, 388, 394, 422, 453, 472, 475, 502, 505], "content": [9, 385, 502], "context": [299, 504], "contigu": [0, 1, 2, 84, 151], "continu": [351, 432, 503, 505], "contract": [0, 137], "contribut": 2, "contriv": [505, 512], "control": [0, 373, 499, 508], "conv": 111, "conv1d": [0, 345], "conv2d": [0, 345], "conv3d": [0, 345], "conv_gener": 0, "conv_transpose1d": 0, "conv_transpose2d": 0, "conv_transpose3d": 0, "conveni": [1, 2, 7, 191], "convent": [19, 111, 136, 137, 423], "convers": 8, "convert": [0, 1, 2, 79, 86, 87, 88, 121, 168, 250, 403, 404, 508, 509, 510], "convolut": [0, 104, 105, 106, 107, 108, 109, 110, 111, 352, 353, 354, 355, 356, 357, 359, 360], "convolv": [104, 105, 106, 107, 108, 109, 110], "convtranspose1d": 345, "convtranspose2d": 345, "convtranspose3d": 345, "coordin": [0, 227], "copi": [0, 1, 2, 6, 8, 103, 244, 291, 509], "core": [1, 2, 3, 4, 5, 6, 7, 326, 345, 347, 348, 349, 350, 370, 377, 378, 379, 388, 391, 393, 396, 423, 424, 425, 426, 427, 428, 429, 430, 431, 444, 446, 453, 472, 475, 477, 502, 503, 509, 510], "corner": 423, "correct": [2, 9, 481, 482, 483, 506, 508], "correctli": [39, 503], "correl": [107, 359], "correspond": [0, 1, 2, 16, 18, 79, 98, 122, 124, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 200, 222, 224, 232, 246, 288, 289, 301, 308, 316, 321, 330, 503, 505, 507], "cos_first": 414, "cosh": [0, 452], "cosin": [0, 20, 21, 112, 113, 445, 491, 493, 505], "cosine_decai": [477, 493], "cosine_similarity_loss": 345, "cost": [9, 479, 503, 508], "costli": 508, "cot": 1, "cot_index": 1, "cotan": 118, "cotang": [1, 2, 118, 320], "could": [6, 345], "count": [345, 493], "counter": 499, "cours": 505, "coursera": 489, "cout": [4, 504], "cov": 256, "covari": [256, 350], "cover": 2, "cpp": [2, 4], "cpu": [8, 9, 200, 201, 207, 336, 512], "cpython": 2, "crash": [84, 502], "creat": [0, 2, 6, 9, 84, 129, 149, 181, 299, 345, 475, 477, 493, 502, 503, 504, 506, 507, 509], "create_additive_causal_mask": 6, "cross": [7, 107, 444, 446], "cross_entropi": [7, 345], "crowd": 6, "cry": 6, "cubic": 423, "cummax": 0, "cummin": 0, "cumprod": 0, "cumsum": 0, "cumul": [0, 84, 114, 115, 116, 117], "current": [6, 8, 9, 84, 93, 106, 109, 110, 133, 175, 228, 248, 332, 345, 479, 503, 508], "custom": [8, 118, 151, 422], "custom_decod": 422, "custom_encod": 422, "custom_funct": 1, "custom_kernel_myexp_float": 1, "custom_tim": 2, "cvpr": 359, "cxx": 4, "cycl": 501, "d": [0, 1, 2, 6, 106, 110, 123, 124, 154, 184, 205, 223, 227, 242, 304, 311, 313, 314, 315, 333, 354, 357, 360, 365, 371, 406, 478, 481, 483, 512], "d1": 512, "d2": 512, "d2fdx2": 505, "d_i": 374, "dampen": 490, "darwin": 2, "data": [0, 2, 7, 8, 11, 19, 131, 149, 164, 165, 171, 181, 212, 236, 240, 253, 262, 311, 313, 319, 323, 360, 424, 425, 426, 427, 428, 429, 430, 431, 502, 503, 504, 506, 509], "dataset": [5, 503, 508], "datatyp": 52, "dbuild_shared_lib": 9, "dcmake_build_typ": [4, 9], "ddof": [0, 76, 81, 297, 318], "deal": 502, "debug": [1, 3, 503, 507], "debugg": 8, "decai": [479, 482, 484, 490, 491, 492, 495], "decay_r": [479, 492, 495], "decay_step": 491, "decent": 7, "decid": [330, 385], "decim": [0, 67, 272], "declar": 2, "decltyp": 1, "decod": 422, "decomposit": [197, 198, 206, 210], "decor": [1, 118], "decoupl": 482, "dedic": 503, "deep": [350, 425, 426, 427, 428], "def": [1, 2, 5, 6, 7, 118, 146, 148, 151, 317, 345, 475, 502, 503, 504, 505, 506, 508, 509, 512], "default": [1, 2, 9, 15, 16, 17, 18, 19, 27, 28, 29, 30, 83, 84, 93, 99, 100, 104, 105, 106, 107, 108, 109, 110, 118, 119, 120, 122, 123, 124, 127, 128, 129, 131, 132, 133, 146, 148, 149, 151, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 172, 173, 177, 180, 181, 185, 193, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 222, 224, 226, 227, 232, 236, 240, 243, 244, 246, 248, 249, 251, 252, 253, 255, 256, 257, 258, 259, 261, 262, 263, 267, 269, 272, 279, 280, 281, 282, 283, 291, 292, 295, 296, 297, 299, 301, 303, 308, 310, 311, 312, 313, 314, 315, 316, 317, 318, 321, 323, 325, 326, 336, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 361, 364, 365, 367, 370, 371, 373, 374, 377, 378, 379, 381, 386, 388, 393, 395, 398, 401, 402, 403, 404, 406, 409, 414, 418, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 437, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 475, 478, 479, 480, 481, 482, 483, 484, 489, 490, 491, 499, 501, 502, 503, 504, 505, 507, 509, 511, 513], "default_devic": 513, "default_stream": 513, "defin": [1, 2, 5, 6, 7, 9, 118, 132, 151, 173, 199, 205, 249, 326, 329, 507, 509], "definit": [118, 197, 198, 256], "degre": [0, 250, 457], "delta": [449, 478], "delv": [427, 428], "demonstr": 509, "denomin": [370, 445, 478, 480, 481, 482, 483, 489], "dens": [227, 512], "depend": [0, 2, 3, 4, 5, 9, 79, 205, 365, 371, 406, 502, 503, 506, 511, 512], "depth": [329, 349, 354, 357, 360, 379, 505], "dequant": [0, 248], "deriv": [2, 504, 505, 508], "descend": 383, "descent": [490, 502, 508], "describ": [2, 508], "descript": [2, 4, 6, 336], "design": [1, 5, 8, 499, 512], "destin": [0, 2, 61, 133, 234, 247], "destroi": 502, "detach": 505, "detail": [1, 2, 11, 279, 345, 359, 409, 414, 423, 425, 426, 427, 428, 478, 480, 481, 483, 484, 503, 506, 510], "detect": 502, "determin": [0, 2, 124, 256, 332, 336, 392, 511], "dev": [2, 9], "develop": [2, 4, 9], "developer_dir": 9, "deviat": [0, 257, 297, 425, 427, 430], "deviatoin": 0, "devic": [1, 2, 8, 9, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 98, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 119, 120, 121, 122, 123, 124, 127, 128, 131, 132, 133, 134, 135, 136, 139, 140, 141, 143, 144, 145, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 171, 172, 173, 178, 179, 180, 181, 182, 184, 185, 186, 187, 188, 189, 190, 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, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 255, 256, 257, 258, 259, 261, 262, 263, 264, 265, 266, 267, 269, 270, 271, 272, 273, 280, 281, 282, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 318, 319, 322, 323, 324, 334, 512, 513], "device_info": 283, "devicetyp": 10, "df": 509, "dfdx": [504, 505, 506], "dft": [155, 156, 157, 158, 159, 160, 164, 165, 166], "dhwc": 360, "diag": [0, 210], "diagon": [0, 46, 123, 149, 311, 313, 314, 315], "dict": [85, 99, 142, 147, 213, 228, 275, 276, 277, 326, 328, 391, 396, 399, 400, 475, 477, 485, 486, 488, 501, 504, 505, 511], "dict_kei": [330, 486], "dictionari": [6, 99, 146, 183, 213, 228, 275, 276, 328, 329, 332, 345, 385, 394, 399, 400, 487, 501, 511], "did": 6, "diff": 2, "differ": [8, 191, 300, 319, 456, 502, 503, 504, 505, 507], "differenti": [1, 8, 351, 432], "difficult": 505, "difficulti": [425, 426], "dilat": [0, 104, 105, 106, 107, 108, 109, 110, 352, 353, 354, 355, 356, 357], "dim": [1, 6, 153, 154, 362, 366, 370, 372, 401, 403, 405, 409, 414, 422], "dimens": [0, 1, 2, 6, 16, 18, 27, 28, 63, 69, 79, 86, 87, 88, 99, 105, 106, 107, 109, 110, 124, 144, 153, 154, 162, 163, 165, 166, 168, 172, 173, 184, 197, 198, 200, 201, 202, 203, 205, 206, 207, 210, 211, 222, 223, 224, 226, 232, 246, 247, 248, 252, 261, 297, 301, 305, 308, 312, 318, 350, 352, 353, 354, 355, 356, 357, 359, 360, 364, 365, 366, 370, 371, 372, 401, 405, 406, 409, 422, 423, 437, 446, 502, 505], "dimension": [31, 150, 152, 155, 156, 157, 158, 159, 160, 164, 165, 166, 347, 348, 349, 350, 352, 353, 354, 355, 356, 357, 362, 374, 377, 378, 379, 403, 404, 414, 506, 509], "dir": 4, "direct": [6, 383, 484, 512], "directli": [2, 6, 84], "directori": [2, 4, 6, 9], "disabl": [125, 279, 325, 502, 503], "disable_compil": 502, "disappoint": 6, "discard": [6, 329], "discov": [9, 503], "discoveri": 484, "discret": [111, 155, 156, 157, 158, 159, 160, 164, 165, 166, 362, 403], "discuss": 2, "disk": 6, "dispatch": 2, "dispatch_thread": 2, "dispatchthread": 1, "displai": 345, "distanc": [6, 457], "distribut": [8, 9, 251, 252, 253, 255, 256, 257, 262, 263, 325, 374, 425, 426, 427, 428, 430, 431, 447, 450, 455, 457, 472], "distributed_config": [503, 507], "diverg": 450, "divid": [0, 2, 39, 170, 248, 266, 503], "divis": [0, 134, 170, 248, 266], "divisor": [297, 318], "divmod": 0, "dloss_dw": 505, "dloss_dx": 505, "dlpack": 509, "dlvalu": 317, "dmlx_build_cpu": 9, "dmlx_build_gguf": 9, "dmlx_build_safetensor": 9, "dmlx_metal_debug": 3, "dmlx_metal_jit": 9, "do": [0, 2, 6, 9, 203, 319, 345, 387, 398, 472, 475, 502, 503, 504, 505, 508], "doc": [2, 7, 503, 507], "document": [2, 3, 4, 66, 80, 151, 275, 276, 336, 502, 503, 504, 505, 506], "doe": [0, 2, 3, 6, 9, 174, 319, 328, 345, 502, 503, 506, 507, 508, 509], "doesn": [2, 345, 504], "domain": 262, "don": [1, 9, 103, 502, 512], "done": [345, 358, 405, 502, 503, 508, 509], "dot": [147, 202, 211, 308, 329, 390, 401, 503], "doubl": [0, 6, 336], "doubt": 6, "down": [6, 328], "downsampl": [347, 348, 349, 377, 378, 379], "dparam": 317, "draw": 252, "drop": 385, "dropout": [345, 359, 360, 395, 422, 502], "dropout2d": 345, "dropout3d": 345, "dst": 133, "dt": 140, "dtype": [0, 1, 2, 6, 12, 19, 31, 38, 39, 79, 82, 131, 132, 149, 151, 167, 168, 171, 181, 191, 193, 200, 201, 205, 207, 212, 240, 253, 255, 256, 257, 259, 262, 263, 288, 289, 311, 313, 316, 319, 323, 325, 336, 393, 423, 424, 425, 426, 427, 428, 429, 430, 431, 444, 446, 453, 491, 492, 493, 494, 495, 502, 503, 504, 505, 506, 509, 510, 511], "dtypecategori": [191, 336], "dual": 452, "duchi": 480, "duplic": 504, "dure": [3, 99, 282, 358, 359, 360, 423, 509], "dx": 118, "dy": 118, "dyld": 503, "dyld_library_path": 503, "dylib": 2, "dynam": [0, 504, 508], "e": [2, 7, 9, 118, 140, 151, 172, 173, 192, 284, 350, 352, 353, 354, 355, 356, 357, 359, 360, 366, 370, 372, 386, 405, 442, 443, 465, 466, 471, 477, 480, 502, 504, 508, 513], "e5": 336, "e8": 336, "each": [0, 1, 2, 69, 85, 122, 142, 153, 173, 191, 197, 198, 200, 201, 202, 206, 210, 211, 223, 227, 243, 248, 249, 252, 267, 277, 278, 292, 309, 312, 319, 321, 322, 359, 360, 362, 365, 366, 371, 406, 409, 422, 444, 446, 499, 502, 503, 504, 507, 508], "eager": 508, "earli": 359, "eas": 6, "easi": [2, 345], "easier": [1, 147, 508], "easiest": 503, "edg": [98, 243, 423, 502], "edit": [9, 400], "effect": [359, 502, 508], "effici": [6, 8, 172, 359, 409, 503, 508, 510], "eigenvalu": [200, 201], "eigenvector": 200, "einstein": [136, 137], "einsum": 137, "either": [9, 14, 66, 79, 80, 89, 91, 92, 98, 134, 135, 139, 170, 178, 179, 183, 194, 195, 196, 205, 218, 223, 225, 233, 235, 239, 245, 266, 270, 300, 317, 348, 349, 378, 379, 411, 423, 427, 428, 503, 507, 509], "elem": [1, 151], "elem_to_loc": [1, 2], "element": [0, 1, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 29, 71, 84, 89, 90, 91, 92, 96, 112, 113, 114, 115, 116, 117, 122, 134, 135, 139, 140, 141, 143, 145, 149, 169, 170, 173, 178, 179, 185, 186, 187, 188, 189, 190, 194, 195, 196, 214, 215, 216, 217, 218, 219, 220, 221, 225, 227, 233, 235, 237, 239, 244, 245, 248, 249, 265, 266, 267, 270, 271, 273, 284, 285, 286, 287, 293, 294, 300, 304, 306, 307, 310, 317, 319, 322, 351, 358, 359, 360, 365, 369, 371, 380, 402, 406, 409, 413, 432, 439, 440, 442, 443, 458, 459, 461, 464, 465, 466, 467, 502, 505], "elementwis": [1, 101, 102], "elif": 6, "ellipsi": 506, "elman": 406, "els": [0, 2, 6, 345, 386, 503, 508], "elsewher": [313, 506], "elu": [345, 462], "emb": [6, 362, 403, 414], "embed": [6, 326, 345, 403, 409, 414, 445], "empti": 256, "en0": 507, "en2": 503, "enabl": [3, 6, 9, 99, 138, 325, 490], "enclos": 504, "encod": [2, 153, 409, 414, 422, 446], "encount": [2, 505], "end": [124, 198, 229, 248, 271, 365, 371, 420, 438, 449, 456, 462, 468, 469, 491, 494, 504], "end_axi": [0, 50, 168], "end_encod": 2, "endif": 2, "endl": [4, 504], "endswith": 386, "enhanc": [6, 409, 508], "enough": [2, 508], "ensur": [0, 1, 2, 9, 151, 328, 452, 503, 504], "ensure_row_contigu": [1, 151], "enter": 6, "entir": [16, 18, 27, 28, 222, 224, 226, 232, 246, 297, 301, 318, 359, 360], "entri": [0, 258, 316, 359, 360], "entropi": [7, 444, 446], "enumer": 345, "environ": [9, 125, 138, 503], "ep": [5, 150, 152, 350, 366, 370, 372, 405, 445, 447, 457, 477, 478, 479, 480, 481, 482, 483, 489], "epoch": 7, "epsilon": [350, 366, 370, 372, 405, 445, 447, 478, 480, 481, 482, 483, 489], "epsilon_1": 479, "epsilon_2": 479, "equal": [0, 1, 17, 29, 83, 149, 179, 185, 196, 239, 244, 259, 292, 325, 370, 374], "equal_nan": [0, 17, 83, 185], "equat": [136, 137, 208, 209], "equival": [0, 2, 32, 66, 80, 132, 135, 170, 173, 180, 304, 351, 361, 363, 367, 368, 369, 375, 376, 400, 402, 404, 407, 408, 410, 412, 415, 416, 417, 418, 419, 421, 503], "erf": [0, 141, 502], "erfinv": 0, "error": [0, 2, 9, 129, 140, 141, 283, 292, 363, 434, 435, 436, 452, 454, 502, 505, 507, 509], "error_norm": 5, "estim": 483, "eta": 484, "etc": [2, 248, 345, 423, 503], "ethernet": [503, 507], "eval": [2, 3, 5, 6, 7, 345, 475, 477, 502, 503, 504, 505, 508, 510], "eval_cpu": 2, "eval_fn": 7, "eval_gpu": 2, "evalu": [2, 6, 7, 8, 85, 133, 142, 192, 282, 320, 345, 384, 395, 475, 477, 502, 504, 510], "even": [1, 2, 6, 99, 502, 503, 504, 508, 509], "evenli": [0, 212], "everi": [248, 330, 477, 495, 505, 507], "everyth": [6, 503], "everywher": 0, "exact": [435, 436], "exactli": [2, 6, 153, 388, 505], "exampl": [0, 3, 4, 5, 6, 7, 9, 19, 39, 85, 118, 129, 146, 147, 148, 151, 154, 168, 183, 191, 193, 200, 201, 205, 207, 288, 289, 299, 304, 316, 328, 331, 332, 345, 347, 348, 349, 350, 370, 377, 378, 379, 386, 388, 395, 398, 423, 424, 425, 426, 427, 428, 429, 430, 431, 444, 446, 453, 472, 477, 486, 491, 492, 493, 494, 495, 499, 505, 506, 507, 508, 509, 510, 511], "exce": [325, 328], "exceed": 282, "except": [8, 118, 149, 161, 162, 164, 165, 166, 282, 336, 366, 388, 504, 506, 509], "exclud": [247, 305], "exclus": [0, 84, 92], "execut": [2, 4, 9, 86, 87, 88, 176, 193, 503, 509, 512], "execute_process": 4, "exist": [2, 3, 6, 386, 398, 503], "exp": [0, 1, 85, 145, 151, 218, 222, 253, 290, 351, 361, 413, 432, 433, 450, 462, 463, 467, 502, 504, 512], "exp_elementwis": [1, 151], "expand_dim": 0, "expect": [6, 352, 353, 354, 355, 356, 357, 358, 359, 360, 414, 422, 447, 502, 503, 506], "expens": 422, "expensive_fun": 508, "experiment": [85, 146, 148, 183, 509], "explain": 2, "explicit": [2, 486, 499, 509], "explicitli": [172, 345, 499, 507], "explor": 9, "expm1": 0, "exponenti": [0, 143, 145, 351, 361, 410, 432, 433, 462, 492], "exponential_decai": 477, "export": [8, 9, 146, 147, 183], "export_funct": 504, "ext_modul": 2, "extend": [2, 243], "extens": [8, 213, 230, 392, 511], "extern": 509, "extra": [1, 330, 331, 504], "extract": [0, 6, 46, 123, 124, 288, 345, 385, 475], "extras_requir": 2, "extrem": [506, 508], "ey": [0, 6, 202, 211], "f": [0, 2, 5, 7, 118, 205, 345, 371, 482, 502, 509], "f_jvp": 118, "f_t": 371, "f_vjp": 118, "f_vmap": 118, "face": 6, "factor": [2, 15, 180, 197, 198, 203, 204, 207, 423, 446, 492, 495], "fail": [502, 503, 507], "fall": [2, 118], "fals": [0, 1, 2, 6, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 57, 58, 59, 60, 64, 76, 77, 81, 83, 99, 103, 107, 114, 115, 116, 117, 129, 146, 148, 151, 185, 191, 197, 198, 205, 209, 210, 211, 213, 222, 224, 226, 227, 232, 246, 297, 301, 318, 322, 326, 329, 330, 331, 332, 336, 366, 370, 372, 374, 386, 388, 398, 401, 404, 409, 414, 422, 423, 444, 447, 479, 481, 482, 490, 504, 509], "famili": 6, "fan": [425, 426, 427, 428], "fan_in": [425, 426, 427, 428], "fan_out": [425, 426, 427, 428], "far": 477, "fast": [1, 8, 363, 436, 503, 512], "faster": [1, 2, 9, 135, 434, 444, 502, 503, 505], "featur": [1, 8, 104, 105, 106, 107, 108, 109, 110, 153, 350, 365, 366, 370, 371, 372, 374, 404, 405, 406, 409, 422, 423, 502, 503, 508], "feed": 6, "feed_forward": 6, "feedforward": [425, 426], "feel": 6, "fetch": 1, "few": [1, 2, 6, 7, 8, 9, 504, 508, 510], "fewer": 503, "ffn": 6, "ffn_norm": 6, "fft": 8, "fi": 503, "figur": 503, "file": [4, 6, 9, 146, 147, 148, 183, 213, 274, 275, 276, 277, 278, 388, 392, 503, 504, 505, 511], "file_or_weight": 388, "fill": [0, 171, 241, 313, 324, 424, 425, 426, 427, 428, 430, 431], "filter": [0, 111, 352, 353, 354, 355, 356, 357, 381, 385], "filter_and_map": 345, "filter_fn": [381, 385], "final": [2, 4, 5, 6, 7, 180, 491, 494, 503, 507], "find": [2, 4, 5, 9, 503], "find_packag": [2, 4], "finder": 9, "fine": [499, 504, 508], "finetun": 345, "finit": [0, 186, 236], "first": [0, 1, 2, 3, 4, 5, 6, 7, 9, 124, 127, 129, 168, 177, 191, 193, 194, 210, 219, 221, 223, 244, 261, 270, 302, 308, 311, 317, 329, 331, 332, 345, 348, 349, 366, 378, 379, 423, 445, 453, 479, 483, 486, 502, 503, 504, 505, 507, 509, 512], "first_lay": 508, "firt": 502, "fit": [248, 512], "five": 502, "fix": [2, 6, 9, 502, 508], "flag": [4, 9, 502, 509], "flat": [172, 173, 329, 333], "flat_param": 277, "flatten": [0, 29, 30, 114, 115, 116, 117, 205, 242, 244, 247, 267, 271, 291, 304, 305, 310, 329, 502], "flexibl": 8, "flexibli": 400, "flip": [0, 107, 111], "float": [0, 1, 2, 12, 15, 17, 19, 79, 150, 151, 152, 153, 154, 167, 170, 171, 180, 185, 191, 205, 236, 249, 251, 255, 257, 325, 328, 336, 350, 358, 359, 360, 366, 370, 372, 381, 393, 405, 409, 414, 420, 422, 423, 424, 425, 426, 427, 428, 430, 431, 445, 446, 447, 449, 453, 456, 457, 468, 469, 478, 479, 480, 481, 482, 483, 484, 489, 490, 491, 492, 494, 495], "float16": [1, 2, 12, 151, 180, 213, 336, 381, 508, 509], "float16_t": [1, 2], "float32": [0, 1, 2, 12, 19, 149, 151, 154, 180, 181, 191, 200, 201, 205, 207, 212, 240, 253, 255, 256, 257, 262, 263, 289, 313, 323, 336, 423, 424, 425, 426, 427, 428, 429, 430, 431, 444, 446, 453, 491, 492, 493, 494, 495, 502, 503, 504, 505, 506, 508, 509, 510, 511], "float64": [12, 191, 336, 509], "floor": [0, 1, 170], "floor_divid": 0, "flow": [0, 298, 508], "flush": 2, "fn": [183, 327, 330, 331, 332, 510], "follow": [1, 2, 4, 6, 7, 8, 9, 19, 111, 122, 154, 172, 205, 243, 248, 331, 345, 435, 436, 450, 478, 479, 480, 483, 484, 490, 499, 502, 503, 504, 505, 507, 512], "food": 6, "forc": [6, 7, 103, 345, 503, 510], "forg": [9, 503], "formal": [122, 248], "format": [6, 147, 213, 274, 275, 276, 277, 278, 503, 509], "formul": [351, 361], "formula": 456, "forth": [423, 503], "forward": [1, 2, 317, 502, 507, 508], "found": [4, 385], "four": 350, "fourier": [155, 156, 157, 158, 159, 160, 164, 165, 166], "fourth": 504, "frac": [140, 248, 284, 350, 358, 359, 360, 366, 370, 372, 374, 405, 413, 425, 426, 427, 428, 445, 447, 449, 452, 463, 465, 466, 478, 480, 481, 482, 483, 489], "fraction": 19, "framework": 8, "free": 279, "freez": [345, 398, 475], "freq": 153, "frequenc": [153, 409, 414], "frequent": [502, 508], "friend": 6, "fro": 205, "frobeniu": 205, "from": [0, 1, 2, 4, 6, 7, 8, 84, 121, 122, 124, 127, 128, 131, 132, 133, 151, 162, 163, 165, 166, 171, 172, 176, 180, 183, 205, 213, 223, 227, 241, 248, 250, 251, 252, 253, 254, 255, 259, 262, 277, 279, 288, 295, 298, 300, 304, 305, 310, 311, 322, 324, 329, 330, 331, 332, 333, 345, 374, 386, 388, 401, 425, 426, 427, 428, 430, 431, 447, 456, 472, 477, 501, 502, 503, 504, 505, 508, 509, 510, 511, 512], "from_embed": 403, "from_linear": 404, "front": [2, 504], "frozen": [345, 386, 396, 398, 404, 475], "fuction": 135, "full": [0, 1, 2, 7, 66, 80, 111, 151, 203, 290, 399, 400, 447, 502, 503, 504, 508], "full_turn": 414, "fulli": [2, 8, 507, 509, 512], "fun": [99, 146, 148, 177, 192, 317, 320, 321, 502, 504, 506, 508, 512], "fun1": 508, "func": 406, "function": [0, 1, 2, 3, 5, 6, 7, 8, 17, 19, 84, 99, 118, 135, 140, 141, 146, 148, 151, 177, 183, 185, 192, 197, 198, 200, 201, 202, 205, 206, 207, 210, 211, 223, 283, 284, 317, 320, 321, 327, 328, 330, 331, 332, 345, 351, 361, 363, 364, 367, 368, 369, 375, 376, 380, 382, 386, 393, 398, 402, 406, 407, 408, 410, 411, 412, 413, 415, 416, 417, 418, 419, 420, 421, 422, 434, 435, 436, 437, 438, 439, 440, 442, 443, 444, 458, 463, 465, 466, 467, 468, 469, 470, 472, 477, 486, 499, 501, 503, 506, 508, 509, 511], "functionexport": 148, "functool": 502, "further": [2, 9, 505], "fuse": [1, 502], "fusibl": 502, "futur": [6, 85, 146, 148, 183, 404, 506, 508], "fx": 118, "g": [3, 9, 118, 151, 205, 248, 371, 471, 489, 490, 504, 508, 513], "g_t": [371, 478, 480, 481, 482, 483, 484, 489, 490], "gain": [425, 426, 427, 428], "gamma": [350, 366, 370, 372, 405, 425, 426, 427, 428], "gap": 1, "gate": [364, 365, 437], "gather": [0, 127, 172, 173], "gather_mm": [0, 173], "gather_qmm": 0, "gaurante": 319, "gaussian": [5, 363, 434, 435, 436, 447], "gaussian_nll_loss": 345, "gc_func": 422, "gelu": [345, 435, 436, 502], "gelu_approx": [345, 363, 434], "gelu_fast_approx": [345, 363, 434], "geluapprox": 363, "gelufast": 363, "gener": [0, 1, 3, 5, 12, 19, 107, 149, 151, 162, 163, 206, 212, 227, 251, 256, 257, 258, 259, 262, 263, 422, 499, 502, 506, 508, 513], "general_": 2, "generate_stub": 9, "geq": [420, 469], "get": [2, 5, 7, 9, 105, 106, 107, 109, 110, 119, 120, 167, 174, 175, 176, 228, 254, 345, 502, 504, 505, 508, 512], "get_cache_memori": 97, "get_command_encod": 2, "get_kernel": 2, "gguf": [9, 213, 275, 511], "gh": 1, "gii": 1, "git": 9, "github": [5, 7, 9, 502], "give": [2, 6, 7, 29, 502], "given": [0, 2, 9, 16, 18, 29, 39, 84, 85, 95, 98, 100, 114, 115, 116, 117, 122, 124, 137, 142, 144, 154, 155, 156, 157, 158, 159, 160, 164, 165, 166, 171, 172, 203, 205, 222, 224, 226, 232, 236, 238, 246, 256, 258, 259, 271, 272, 279, 281, 290, 292, 297, 301, 303, 309, 310, 311, 313, 314, 315, 318, 334, 358, 385, 401, 445, 447, 453], "gix": 1, "gix_mult": 1, "giy_mult": 1, "global": [125, 127, 128, 129, 131, 132, 133, 138, 260, 325, 328, 499, 502], "glorot": [425, 426], "glorot_norm": 345, "glorot_uniform": 345, "glu": [6, 345], "gm": 1, "gn": 1, "go": [2, 6, 503, 505], "golub": 205, "good": [2, 9, 477, 502, 503, 507, 512], "goroshin": 359, "gower": 6, "gpu": [1, 3, 8, 9, 228, 336, 506, 512], "gputrac": [3, 230], "grad": [2, 5, 7, 118, 317, 328, 477, 485, 502, 503, 504, 505, 506, 508, 510], "grad_fn": [5, 502, 505], "gradient": [0, 5, 7, 118, 177, 298, 317, 325, 327, 328, 345, 386, 399, 404, 422, 452, 475, 477, 478, 479, 481, 482, 483, 484, 485, 488, 490, 502, 503, 505, 506, 508, 509, 510], "grain": 499, "graph": [2, 6, 7, 8, 147, 282, 504, 505], "great": 3, "greater": [0, 6, 29, 145, 179, 244, 328, 420, 469], "greater_equ": 0, "grep": 9, "grid": [2, 151, 227], "grid_dim": 2, "grid_grad": 1, "grid_idx": 1, "grid_sampl": 1, "grid_sample_grad": 1, "grid_sample_ref": 1, "grid_sample_vjp": 1, "grid_shap": 1, "grid_siz": 1, "ground": [5, 6, 446, 456], "group": [0, 1, 104, 105, 106, 107, 108, 109, 110, 122, 127, 128, 129, 131, 132, 133, 154, 173, 248, 249, 319, 325, 326, 352, 353, 366, 403, 404, 503], "group_dim": 2, "group_siz": [0, 122, 173, 248, 249, 326, 403, 404], "groupnorm": 345, "grow": 508, "gru": 345, "guid": [2, 4, 8, 503, 504], "guidelin": 282, "gw": 1, "h": [1, 2, 4, 104, 105, 106, 108, 109, 110, 205, 350, 353, 354, 356, 357, 359, 360, 365, 371, 406, 505, 508], "h_": [365, 371, 406], "h_in": 1, "h_stride": 1, "h_t": [365, 371, 406], "ha": [2, 3, 6, 7, 8, 9, 79, 99, 124, 133, 161, 162, 164, 165, 166, 175, 177, 197, 198, 200, 201, 202, 206, 210, 211, 227, 252, 350, 365, 371, 374, 406, 475, 477, 502, 503, 504, 506, 507, 508, 510, 512], "had": 6, "hadamard": [0, 180], "hadamard_transform": 0, "half": [19, 259, 263, 409, 508], "halv": [364, 437], "hand": [6, 505, 508], "handi": 505, "handl": [2, 345, 502], "happen": [2, 6, 150, 152, 422, 477, 502, 508], "happi": 6, "hard": 6, "hard_shrink": [345, 367], "hard_tanh": [345, 368], "hardcod": 502, "hardshrink": [345, 438], "hardswish": 345, "hardtanh": [345, 439], "hat": [122, 248], "have": [0, 1, 2, 6, 9, 17, 83, 86, 87, 88, 93, 118, 127, 154, 162, 163, 165, 166, 173, 185, 223, 230, 252, 319, 325, 329, 371, 401, 411, 484, 486, 501, 502, 503, 504, 506, 507, 508, 512], "haven": 6, "hazan": 480, "he": [6, 427, 428], "he_norm": 345, "he_uniform": 345, "head": [154, 401, 422], "header": [2, 151], "heart": 6, "heavi": 6, "height": [348, 349, 350, 353, 354, 356, 357, 359, 360, 378, 379], "hello": [329, 333], "help": [2, 6, 502, 512], "helper": [6, 151, 325, 502, 503, 507], "henc": [0, 2, 248, 502], "hendryck": 436, "here": [2, 6, 477, 502, 504, 505, 508, 511, 512], "hermitian": [200, 201], "hf": 371, "hg": 371, "hh": 406, "hi": [6, 371], "hidden": [365, 371, 406, 422], "hidden_dim": [7, 475, 477], "hidden_s": [365, 371, 406], "hierarchi": 336, "high": [259, 263, 345, 362, 431, 472], "high_pad_s": 0, "higher": [2, 184, 283, 453, 503, 505], "highli": 9, "him": 6, "hing": 448, "hinge_loss": 345, "hinton": 489, "hit": 2, "hn": 365, "ho": 371, "hold": [2, 6, 11, 12, 205, 502], "homebrew": 503, "hopkin": 205, "host": 2, "host1": 503, "host2": 503, "host3": 503, "host4": 503, "host_nam": 1, "hostfil": [503, 507], "hostnam": [503, 507], "hostname1": [503, 507], "hostname2": [503, 507], "hostname3": 503, "hostname4": 503, "hot": 446, "hour": 6, "how": [2, 4, 6, 7, 345, 347, 348, 349, 352, 353, 354, 355, 356, 357, 362, 377, 378, 379, 403, 423, 485, 502, 506, 512], "howev": [2, 118, 345, 363, 366, 486, 499, 502, 503, 508, 509], "hr": 365, "http": [366, 370, 372, 380, 405, 436, 458], "huber": 449, "huber_loss": 345, "human": [427, 428], "hundr": 9, "hurri": 6, "hutter": 482, "hyperbol": [0, 21, 23, 26, 113, 287, 307, 421, 470], "hz": 365, "i": [0, 1, 2, 3, 4, 6, 7, 8, 9, 17, 19, 29, 38, 79, 84, 85, 98, 105, 106, 107, 109, 110, 111, 114, 115, 116, 117, 118, 123, 124, 127, 128, 130, 131, 132, 133, 135, 142, 146, 148, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 170, 171, 172, 173, 180, 183, 185, 186, 191, 192, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 213, 218, 222, 223, 227, 229, 243, 244, 247, 248, 249, 256, 257, 258, 269, 271, 274, 275, 276, 282, 283, 284, 290, 292, 297, 298, 303, 304, 305, 308, 311, 312, 316, 317, 318, 319, 320, 321, 322, 325, 326, 328, 329, 330, 331, 332, 336, 338, 345, 347, 348, 349, 350, 352, 353, 354, 355, 356, 357, 358, 359, 360, 363, 365, 366, 370, 371, 372, 374, 377, 378, 379, 385, 386, 392, 394, 395, 397, 398, 400, 401, 402, 404, 405, 406, 409, 414, 420, 422, 423, 427, 428, 434, 436, 444, 445, 447, 452, 453, 456, 457, 459, 464, 469, 475, 477, 479, 481, 482, 484, 485, 486, 491, 493, 494, 499, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513], "i386": 9, "i_n": 1, "i_nw": 1, "i_s": 1, "i_sw": 1, "i_t": 371, "iclr": [481, 482, 483], "id": [7, 9], "idea": [505, 508], "idempot": [386, 398], "ident": [0, 118, 133, 149, 298, 345, 395, 503], "identifi": [2, 329, 501], "idim": 7, "idiom": [7, 502], "idx": [39, 506], "ie": 398, "ieee": 336, "ifac": 503, "ignor": [6, 39, 85, 98, 99, 142, 479, 507], "ih": 406, "ii": 1, "ij": 227, "imag": [0, 353, 354, 356, 357, 359, 360, 423], "imagenet": [427, 428], "imaginari": 182, "immedi": [6, 381], "implement": [0, 1, 5, 7, 153, 154, 362, 385, 401, 409, 411, 414, 420, 422, 423, 469, 478, 479, 480, 483, 484, 485, 497, 502, 505], "impli": 319, "implicit": [499, 502, 505], "implicitli": 508, "import": [2, 3, 5, 6, 7, 9, 118, 129, 180, 183, 205, 277, 317, 329, 330, 331, 332, 333, 345, 347, 348, 349, 350, 370, 377, 378, 379, 388, 423, 444, 446, 453, 472, 475, 477, 502, 503, 505, 506, 508, 509, 510], "import_funct": 504, "imported_ab": 504, "imported_fun": 504, "imported_funct": 504, "improv": [1, 2, 3, 6, 444, 478, 479, 480, 481, 482, 483, 489, 502, 503], "in_ax": [321, 505], "in_channel": [352, 353, 354, 355, 356, 357], "in_dim": [345, 475], "in_proj": 475, "includ": [1, 2, 4, 114, 115, 116, 117, 147, 151, 174, 175, 282, 372, 382, 394, 404, 447, 477, 502, 504, 505, 506, 510, 511, 513], "include_dir": 2, "inclus": [0, 42, 43, 44, 45, 114, 115, 116, 117, 168], "incom": 2, "inconveni": 502, "incorpor": 509, "incorrect": 509, "increas": [283, 507], "increment": 19, "incur": [6, 9], "independ": [126, 359, 360], "index": [0, 1, 2, 8, 10, 29, 39, 144, 149, 177, 227, 244, 288, 289, 304, 305, 317], "indic": [0, 2, 17, 27, 28, 29, 30, 39, 172, 173, 177, 185, 186, 187, 188, 189, 190, 191, 203, 210, 247, 288, 289, 292, 304, 305, 317, 395, 397, 446, 453, 493, 506], "indices_or_sect": [72, 292], "indirectli": 509, "individu": [345, 359, 360], "ineffici": [506, 508], "inexact": [12, 191], "inf": [205, 236, 401], "infer": [8, 171, 213, 311, 316, 503, 504], "infin": [0, 187, 189, 190, 236, 377, 378, 379, 483], "infinit": [17, 185, 186], "info": [6, 9], "inform": [3, 4, 6, 7, 9, 137, 167, 228, 275, 276, 336, 345, 350, 363, 401, 503, 504, 505, 512], "inherit": [7, 501], "inifn": 187, "init": [345, 402, 472, 477, 491, 492, 494, 495, 503], "init_fn": [424, 425, 426, 427, 428, 429, 430, 431, 472], "init_valu": 1, "initi": [1, 3, 5, 6, 129, 332, 345, 350, 366, 370, 372, 374, 402, 405, 424, 425, 426, 427, 428, 429, 430, 431, 475, 486, 491, 492, 494, 495, 502, 503, 504, 508], "initializer_list": 0, "inject": 0, "inlin": 0, "inner": [0, 502], "inorm": 370, "inp": [1, 151], "inp_ndim": 1, "inp_shap": 1, "inp_strid": 1, "inplac": [2, 9], "input": [0, 1, 2, 5, 6, 13, 14, 15, 16, 17, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 121, 123, 124, 127, 128, 133, 134, 135, 136, 137, 139, 140, 141, 143, 144, 145, 146, 147, 148, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 172, 173, 177, 178, 179, 180, 182, 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, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 232, 233, 234, 235, 236, 237, 239, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 258, 261, 264, 265, 266, 267, 269, 270, 271, 272, 273, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 300, 301, 302, 304, 305, 306, 307, 308, 309, 310, 311, 312, 314, 315, 316, 317, 318, 319, 321, 322, 324, 347, 348, 349, 350, 352, 353, 354, 355, 356, 357, 359, 360, 362, 364, 365, 366, 370, 371, 372, 374, 377, 378, 379, 401, 404, 405, 406, 409, 420, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 437, 444, 445, 447, 448, 449, 450, 452, 453, 455, 457, 469, 472, 502, 504, 505, 506, 507, 510, 511], "input_dil": [0, 107], "input_dim": [7, 345, 374, 404], "input_nam": [1, 151], "input_s": [365, 371, 406], "inputs1": 453, "inputs2": 453, "insert": [124, 144, 512], "insid": [502, 504], "inspect": [3, 502, 510], "inspir": 8, "instabl": 457, "instal": [2, 4, 507], "instanc": [6, 39, 118, 248, 333, 345, 370, 381, 382, 383, 386, 388, 389, 390, 395, 398, 399, 400, 411, 475, 503, 507, 509], "instancenorm": 345, "instanti": [1, 2, 7, 508], "instantiate_kernel": 2, "instead": [2, 9, 118, 345, 400, 414, 505, 508], "instruct": [4, 504], "int": [0, 1, 2, 4, 6, 7, 10, 16, 18, 19, 27, 28, 29, 30, 34, 35, 36, 37, 42, 43, 44, 45, 46, 47, 50, 57, 58, 59, 60, 61, 64, 67, 69, 72, 75, 76, 77, 78, 79, 81, 84, 93, 95, 100, 104, 105, 106, 107, 108, 109, 110, 114, 115, 116, 117, 122, 123, 124, 131, 132, 133, 137, 144, 149, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 171, 173, 174, 175, 176, 177, 181, 191, 199, 205, 212, 222, 224, 226, 228, 232, 234, 240, 243, 244, 246, 247, 248, 249, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 267, 269, 271, 272, 279, 282, 283, 288, 289, 290, 291, 292, 295, 296, 297, 301, 302, 304, 305, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 321, 323, 325, 326, 345, 347, 348, 349, 350, 352, 353, 354, 355, 356, 357, 362, 364, 365, 366, 370, 371, 372, 374, 377, 378, 379, 401, 403, 404, 405, 406, 409, 414, 422, 437, 445, 446, 450, 455, 457, 475, 491, 493, 494, 495], "int16": 336, "int32": [0, 1, 12, 19, 39, 168, 191, 193, 205, 259, 288, 316, 336, 423, 506, 510], "int64": [12, 336], "int64_t": 2, "int8": [12, 336], "int_0": 140, "integ": [0, 12, 170, 172, 173, 191, 205, 228, 243, 248, 249, 251, 258, 259, 292, 304, 308, 321, 336, 362, 393, 493, 506], "integr": [19, 304, 508], "intend": [0, 502], "interact": 422, "interest": 512, "interfac": [2, 503, 507], "intermedi": 509, "intern": 350, "interpol": 423, "interpret": 4, "interv": [19, 212, 259, 263], "introduc": [0, 271], "intuit": 345, "invalid": [0, 84], "invers": [0, 20, 21, 22, 23, 24, 25, 26, 90, 141, 158, 159, 160, 161, 162, 163, 198, 202, 206, 211], "invert": 0, "involv": [477, 502], "iogpu": 283, "iostream": 4, "ip": [503, 507], "ip1": [503, 507], "ip2": [503, 507], "ip3": 503, "ip4": 503, "is_avail": 129, "is_equival": 2, "is_leaf": [329, 330, 331, 332], "is_leaf_fn": 385, "isclos": 0, "isfinit": 0, "ish": 6, "ishmael": 6, "isinf": 0, "isnan": 0, "isneginf": 0, "isposinf": 0, "issu": [503, 505, 509], "issubdtyp": [2, 12, 336], "item": [0, 2, 5, 6, 7, 330, 477, 504, 508, 509, 510], "iter": [5, 7, 210, 330, 331, 499, 502, 508], "iterm": 9, "itertool": [6, 330], "its": [0, 1, 2, 9, 154, 198, 206, 223, 244, 261, 313, 327, 333, 345, 404, 477, 481, 482, 483, 503, 508, 509, 512], "itself": [2, 326, 486], "ix": 1, "ix_n": 1, "ix_nw": 1, "ix_s": 1, "ix_sw": 1, "iy_n": 1, "iy_nw": 1, "iy_s": 1, "iy_sw": 1, "j": [6, 9, 205, 359, 480, 481, 483], "j8": 2, "jacobian": [2, 192, 320, 510], "jain": 359, "jax": [8, 499], "jit": 151, "jmlr": 480, "jnp": 509, "john": 205, "join": 493, "join_schedul": 477, "jointli": 256, "json": [503, 507], "just": [2, 4, 7, 372, 502, 504, 506], "jvp": [2, 118, 510], "k": [0, 6, 46, 93, 123, 149, 154, 172, 180, 310, 313, 314, 315, 374, 386], "kaim": 428, "keep": [2, 16, 18, 27, 28, 222, 224, 226, 232, 246, 297, 301, 318, 345, 385, 505, 508], "keepdim": [0, 16, 18, 27, 28, 34, 35, 36, 37, 57, 58, 59, 60, 64, 76, 77, 81, 205, 222, 224, 226, 232, 246, 290, 297, 301, 318], "kei": [1, 3, 6, 146, 154, 183, 228, 251, 252, 253, 255, 256, 257, 258, 259, 261, 262, 263, 329, 330, 385, 386, 398, 401, 486, 499, 501, 504, 505], "kept": 283, "kernel": [2, 8, 9, 104, 105, 106, 107, 108, 109, 110, 151, 347, 377, 502, 506], "kernel_dil": [0, 107], "kernel_s": [347, 348, 349, 352, 353, 354, 355, 356, 357, 377, 378, 379], "key_cach": 6, "key_input_dim": 401, "key_proj": 6, "keyword": [146, 177, 277, 278, 317, 330, 345, 499, 504, 511, 513], "kind": 6, "kingma": [481, 483], "kl_div_loss": 345, "kname": 2, "know": [2, 6], "known": [412, 464], "kron": 0, "kroneck": [0, 193], "kth": [0, 29, 244], "kullback": 450, "kw_onli": 2, "kwarg": [11, 126, 146, 147, 183, 277, 278, 334, 504, 513], "l": [6, 7, 197, 198, 200, 201, 203, 345, 350, 352, 355, 365, 371, 406, 456], "l1": [317, 449, 451, 452, 456], "l1_loss": 345, "l2": [449, 452, 490], "l2_loss": 345, "l_": 449, "la": 205, "label": [3, 5, 446, 453], "label_smooth": 446, "lack": 506, "lambd": [367, 418, 438, 468], "lambda": [330, 331, 332, 345, 367, 381, 386, 393, 418, 438, 462, 468, 478, 479, 480, 481, 482, 483, 484, 489, 490, 502, 503, 504, 505], "languag": [1, 2, 4], "larg": [6, 345, 401, 452, 502, 504, 508], "larger": [1, 153, 283, 409, 484], "largest": [205, 236, 310], "lasso": 317, "last": [0, 1, 6, 30, 79, 150, 152, 157, 160, 162, 163, 165, 166, 168, 172, 173, 176, 184, 197, 198, 200, 201, 202, 206, 207, 210, 211, 223, 252, 291, 308, 319, 352, 353, 354, 355, 356, 357, 359, 360, 366, 423, 509], "later": [3, 9, 477], "launch": [1, 2, 129, 503, 506], "layer": [8, 150, 326, 345, 347, 348, 349, 359, 360, 365, 366, 371, 372, 374, 377, 378, 379, 395, 400, 403, 404, 406, 411, 422, 471, 475, 504, 507], "layer_s": 7, "layernorm": 345, "layout": 1, "lazi": [8, 475, 510], "lazili": [6, 345], "lceil": 93, "ld": [365, 371, 406], "lead": [0, 19, 84, 502], "leaf": [99, 326, 329, 330, 331, 332, 385], "leaf_modul": 345, "leaki": [373, 441], "leaky_relu": 345, "leakyrelu": 345, "learn": [5, 7, 8, 350, 366, 370, 372, 402, 405, 477, 478, 479, 480, 481, 482, 483, 484, 489, 490], "learnabl": [352, 353, 354, 355, 356, 357, 411], "learning_r": [7, 477, 478, 479, 480, 481, 482, 483, 484, 486, 489, 490, 491, 492, 493, 494, 495, 502], "least": [6, 86, 87, 88, 98, 197, 198, 200, 201, 202, 206, 207, 210, 211, 248], "leav": [2, 85, 142, 330, 331, 332], "lectur": 489, "lecun": 359, "left": [0, 6, 153, 194, 205, 248, 271, 363, 409, 423, 435, 436, 447, 449, 457], "left_shift": 0, "leibler": 450, "len": [6, 157, 160, 163, 166, 180, 493], "length": [6, 295, 350, 352, 355, 365, 371, 406, 493], "leq": [449, 462], "less": [0, 1, 6, 29, 196, 244, 283, 325, 409, 456, 503], "less_equ": 0, "let": [1, 2, 5, 6, 198, 502, 504, 505, 508, 509], "level": [0, 172, 173, 427, 428], "lh": [365, 371, 406], "lhs_indic": [0, 172, 173], "lhs_mask": 93, "lib": 503, "libmlx": 9, "libmlx_ext": 2, "libmpi": 503, "librari": [2, 4, 9, 338, 345, 503, 504], "like": [2, 6, 8, 132, 146, 148, 183, 191, 241, 324, 360, 452, 486, 488, 502, 503, 504, 505, 507, 508, 509, 510, 512], "likelihood": [447, 455], "limit": [0, 2, 98, 279, 282, 283, 506], "linalg": 180, "line": [6, 503, 504, 507, 508, 509], "linear": [0, 2, 6, 7, 8, 208, 209, 326, 330, 345, 351, 361, 363, 364, 373, 388, 404, 406, 407, 408, 410, 412, 423, 432, 433, 434, 435, 436, 437, 441, 460, 461, 462, 464, 472, 475, 486, 493, 494, 502, 504], "linear1": 6, "linear2": 6, "linear3": 6, "linear_schedul": [477, 493], "linearli": 401, "link": [2, 4, 9], "linspac": 0, "lion": 477, "list": [1, 6, 11, 16, 18, 31, 72, 79, 84, 85, 86, 87, 88, 95, 99, 100, 107, 137, 142, 151, 156, 157, 159, 160, 162, 163, 165, 166, 171, 177, 192, 205, 222, 224, 226, 227, 232, 240, 243, 246, 251, 252, 253, 255, 256, 257, 259, 262, 263, 275, 290, 292, 296, 297, 301, 308, 309, 312, 317, 318, 320, 323, 329, 332, 333, 345, 386, 388, 389, 390, 391, 396, 398, 399, 400, 475, 477, 481, 482, 483, 484, 493, 501, 502, 503, 504, 505, 507, 508], "listen": 503, "liter": [2, 243, 423, 427, 428, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457], "littl": 6, "liu": 6, "live": [8, 151, 512], "ll": [1, 5, 7, 449, 502, 505], "llama": 6, "llamaattent": 6, "llamaencoderlay": 6, "llm": 8, "load": [7, 8, 338, 388, 503], "load_weight": [345, 508], "loader": 7, "loader_path": 2, "loan": 205, "loc": [1, 255, 257], "local": [345, 359, 503], "localhost": [503, 507], "locat": [0, 2, 4, 84, 288, 289, 399, 400, 503, 512], "log": [0, 216, 218, 222, 375, 376, 442, 443, 444, 447, 450, 452, 455, 467], "log10": 0, "log1p": 0, "log2": 0, "log_cosh_loss": 345, "log_sigmoid": [345, 375], "log_softmax": [345, 376], "logaddexp": 0, "logarithm": [0, 214, 215, 216, 217], "logcosh": 452, "logic": [0, 2, 219, 220, 221, 503], "logical_and": 0, "logical_not": 0, "logical_or": 0, "logist": [0, 5, 284, 436, 464], "logit": [6, 252, 444, 446, 502], "logsigmoid": 345, "logsoftmax": 345, "logsumexp": 0, "long": 6, "longer": [6, 111, 505], "look": [2, 6, 503], "lookup": 362, "loop": [6, 7, 502, 503, 505, 508], "loshchilov": 482, "loss": [5, 7, 317, 345, 477, 502, 503, 505, 508], "loss_and_grad": 345, "loss_and_grad_fn": [7, 477, 502, 505], "loss_fn": [5, 7, 477, 502, 505], "loss_grad_fn": 503, "lot": 505, "low": [259, 263, 431, 472], "low_pad_s": 0, "lower": [197, 198, 200, 201, 209, 211, 248, 259, 262, 263, 313, 431], "lr": [5, 484], "lr_schedul": [491, 492, 493, 494, 495], "lstm": 345, "lto": 2, "lu": [6, 204], "luckili": 508, "lvalu": 317, "m": [0, 2, 4, 6, 9, 93, 149, 172, 180, 205, 313, 478, 502], "m1": [1, 6, 502, 505, 512], "m10": 336, "m7": 336, "m_": [481, 482, 483, 484], "m_t": [481, 482, 483, 484], "mac": 503, "machin": [6, 8, 9, 489, 503], "maco": [9, 283], "macosx": 9, "made": [6, 338], "mai": [2, 4, 85, 146, 148, 183, 205, 326, 359, 503, 505, 506], "main": [4, 8, 124, 149, 151, 311, 330, 331, 345, 503], "maintain": [359, 360, 484], "major": [0, 2, 103], "make": [1, 2, 3, 4, 6, 7, 9, 103, 147, 148, 223, 238, 281, 345, 491, 492, 494, 495, 502, 508, 510, 512], "make_shar": 2, "malloc": 2, "man": 6, "manag": [8, 299, 499, 503, 504, 512], "mani": [2, 84, 292, 352, 353, 354, 355, 356, 357, 362, 403, 502, 503, 504, 508], "manual": [345, 503], "map": [2, 7, 39, 213, 330, 362, 381, 504], "map_fn": [381, 385], "map_torch_to_mlx": 6, "margin": [453, 457], "margin_ranking_loss": 345, "mask": [0, 6, 93, 154, 395, 401, 506], "mask_lh": [0, 93], "mask_n": 1, "mask_nw": 1, "mask_out": [0, 93], "mask_rh": [0, 93], "mask_s": 1, "mask_sw": 1, "matadata": 213, "match": [9, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 174, 388, 423, 446, 506, 509], "materi": [6, 8], "math": [6, 457, 502], "mathbf": 198, "mathcal": 374, "mathemat": 205, "mathrm": [140, 284, 370], "matmul": [0, 172, 512], "matric": [205, 207, 210], "matrix": [0, 5, 15, 46, 93, 122, 123, 149, 172, 173, 180, 181, 197, 198, 200, 201, 202, 203, 204, 205, 206, 207, 210, 211, 223, 227, 248, 249, 256, 403, 404, 429, 472], "matter": [6, 345, 504], "matur": 503, "max": [0, 1, 2, 205, 225, 351, 377, 378, 379, 402, 432, 439, 440, 445, 447, 448, 453, 457, 459, 461, 479, 483, 502, 505, 512], "max_buffer_s": 228, "max_freq": 414, "max_i": 248, "max_norm": 328, "max_recommended_working_set_s": [228, 283], "max_val": 439, "maximum": [0, 7, 27, 39, 98, 114, 176, 282, 328, 345, 373, 377, 378, 379, 407, 414, 435, 436, 441, 460, 475, 508], "maxpool1d": 345, "maxpool2d": 345, "maxpool3d": 345, "maxtotalthreadsperthreadgroup": 2, "mca": [503, 507], "md": 205, "me": 6, "mean": [0, 1, 5, 6, 7, 152, 255, 256, 257, 317, 345, 350, 366, 386, 405, 430, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 502, 503, 505, 509], "meant": 118, "measur": 512, "mechan": 422, "medic": 360, "meet": 9, "member": [345, 391, 396], "memori": [0, 1, 2, 8, 84, 97, 174, 175, 176, 268, 279, 282, 283, 422, 475, 479, 502, 508, 509], "memory_order_relax": 1, "memory_s": [228, 283], "memoryview": [508, 509], "merg": 502, "meshgrid": 0, "metadata": [5, 213, 275, 276], "metal": [2, 4, 8, 151, 282], "metal_captur": 3, "metal_kernel": 1, "metal_path": 9, "metallib": [2, 9], "method": [2, 6, 10, 11, 31, 118, 126, 167, 326, 334, 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, 392, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 475, 478, 479, 480, 481, 482, 483, 484, 486, 489, 490, 497], "millisecond": [9, 502, 512], "min": [0, 2, 205, 233, 351, 402, 432, 439, 440, 459, 461], "min_freq": 414, "min_i": 248, "min_val": 439, "mind": 6, "mine": 6, "minibatch": 7, "minim": [503, 507], "minimum": [0, 28, 39, 98, 115, 414, 444, 445], "minsizerel": 9, "minu": 145, "minut": 6, "mish": 345, "mismatch": 504, "miss": [388, 504, 511], "mix": 506, "mkdir": [3, 9], "ml": 9, "mlp": [7, 345, 422, 477], "mlp_dim": [6, 422], "mlx": [1, 3, 5, 6, 7, 9, 338, 345, 472, 475, 477, 499, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512], "mlx_build_acceler": 4, "mlx_build_benchmark": 9, "mlx_build_cpu": 9, "mlx_build_exampl": 9, "mlx_build_gguf": 9, "mlx_build_met": [2, 4, 9], "mlx_build_metallib": 2, "mlx_build_python_bind": 9, "mlx_build_safetensor": 9, "mlx_build_test": 9, "mlx_cxx_flag": 4, "mlx_disable_compil": [125, 138, 502], "mlx_ext": 2, "mlx_ext_metallib": 2, "mlx_found": 4, "mlx_include_dir": [2, 4], "mlx_librari": 4, "mlx_metal_debug": [3, 9], "mlx_metal_jit": 9, "mlx_root": 4, "mlx_sample_extens": 2, "mlx_trace": 3, "mlxfn": [146, 148, 183, 504], "mnist": 7, "mode": [0, 1, 2, 111, 243, 384, 395, 397, 423, 427, 428], "model": [5, 7, 8, 277, 326, 327, 330, 331, 345, 381, 384, 386, 388, 392, 395, 397, 398, 399, 401, 422, 472, 475, 477, 485, 486, 488, 502, 503, 504, 508], "modest": 2, "modif": 509, "modifi": 509, "modul": [2, 4, 6, 7, 326, 327, 411, 422, 472, 488, 501, 502, 508], "moment": [6, 479, 483, 503], "momentum": [350, 484, 486, 490, 502], "monei": 6, "monitor": 507, "monoton": 458, "moor": 206, "more": [1, 2, 3, 4, 7, 11, 79, 124, 146, 172, 197, 198, 200, 201, 202, 203, 206, 210, 211, 223, 275, 276, 279, 282, 336, 345, 350, 359, 409, 414, 422, 423, 425, 426, 427, 428, 444, 499, 502, 503, 505, 506, 510, 512], "moreov": 507, "most": [2, 154, 252, 316, 345, 488, 502, 503, 505, 506, 508], "move": [0, 2, 234, 512], "moveaxi": 0, "mpi": [129, 338], "mpirun": [503, 507], "mse": 317, "mse_loss": 345, "mtl": 2, "mtl_capture_en": 3, "mtlcommandbuff": 2, "mu": 490, "much": [1, 2, 6, 347, 348, 349, 377, 378, 379, 502, 508], "multi": [8, 154, 352, 353, 354, 355, 356, 357, 504, 506, 509], "multidimension": 227, "multiheadattent": [6, 345], "multipl": [0, 1, 9, 15, 93, 148, 150, 152, 172, 173, 223, 235, 248, 249, 401, 414, 492, 493, 495, 502, 508, 511], "multipli": [0, 2, 39, 173, 248, 249, 358, 414, 423], "murtadha": 6, "must": [0, 1, 2, 3, 9, 93, 98, 146, 153, 154, 171, 173, 200, 201, 205, 251, 252, 256, 259, 262, 263, 322, 423, 509], "mx": [1, 2, 3, 4, 5, 6, 7, 39, 85, 101, 102, 118, 129, 132, 146, 147, 148, 151, 168, 183, 191, 193, 200, 201, 203, 205, 207, 213, 258, 277, 288, 289, 316, 317, 328, 345, 347, 348, 349, 350, 361, 370, 373, 377, 378, 379, 381, 388, 392, 407, 423, 424, 425, 426, 427, 428, 429, 430, 431, 433, 441, 444, 445, 446, 450, 453, 460, 470, 472, 475, 477, 499, 502, 503, 504, 505, 506, 508, 509, 510, 511, 512, 513], "my": [6, 9], "my_devic": 513, "my_path": 277, "my_script": [503, 507], "myexp": [1, 151], "myexp_strid": 1, "mymlp": 475, "n": [0, 1, 2, 6, 31, 93, 104, 105, 106, 107, 108, 109, 110, 149, 154, 155, 157, 158, 160, 161, 164, 166, 180, 181, 256, 297, 313, 318, 350, 352, 353, 354, 355, 356, 357, 359, 360, 365, 371, 406, 423, 452, 457, 503, 507], "n_kv": 154, "n_q": 154, "n_t": 365, "naiv": [2, 505], "naive_add": 505, "name": [1, 2, 118, 147, 151, 173, 213, 248, 249, 275, 276, 277, 278, 345, 366, 385, 388, 390, 503, 506, 511], "named_modul": 345, "namespac": 4, "nan": [0, 17, 83, 185, 186, 188, 236], "nan_to_num": 0, "nanobind": 2, "nanobind_add_modul": 2, "nativ": [9, 503], "natur": [0, 214, 216, 508], "nb": 2, "nb_domain": 2, "nb_modul": 2, "nb_static": 2, "nbyte": 2, "nc": 350, "ndarrai": [31, 506, 508, 510], "ndhwc": [354, 357, 360], "ndim": [0, 1, 2, 168, 205, 210, 423], "ne": 1, "nearest": [1, 423], "necessari": [103, 345], "necessarili": 310, "need": [1, 2, 4, 6, 7, 8, 9, 83, 248, 345, 399, 400, 414, 422, 499, 503, 505, 507, 508, 509, 510, 512], "neg": [0, 124, 168, 189, 236, 271, 311, 373, 377, 378, 379, 401, 447, 455, 457, 506], "negat": [0, 237], "negative_slop": [373, 441], "neginf": [0, 236], "neighbor": [423, 507], "neither": [177, 317], "nelem": 2, "nervou": 6, "nest": [79, 99, 332, 345, 475, 501, 505], "nesterov": 490, "network": [6, 8, 325, 350, 359, 362, 425, 426, 472, 475, 489, 503], "neural": [6, 8, 362, 425, 426, 458, 472, 475, 489], "never": [6, 508], "new": [0, 2, 7, 95, 124, 234, 238, 269, 296, 312, 319, 330, 331, 393, 401, 475, 477, 488, 493, 502, 504, 506, 508, 509], "new_tre": 331, "next": [2, 4, 6, 7, 279, 504], "nh": [365, 371, 406], "nhwc": [350, 353, 356], "nice": [505, 508], "nlc": [350, 352, 355], "nld": [365, 371, 406], "nlh": [365, 371, 406], "nll": [447, 455], "nll_loss": 345, "nn": [2, 6, 7, 277, 330, 345, 472, 475, 477, 486, 488, 502, 504, 508], "nobodi": 6, "node": [85, 99, 142, 321, 331, 332, 503, 507], "nois": 5, "noisi": 5, "nomins": 2, "non": [0, 1, 2, 4, 9, 227, 396, 406, 458, 475], "none": [1, 2, 6, 10, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 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, 121, 122, 123, 124, 125, 127, 128, 131, 132, 133, 134, 135, 136, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 149, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 177, 178, 179, 180, 181, 182, 184, 185, 186, 187, 188, 189, 190, 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, 230, 231, 232, 233, 234, 235, 236, 237, 239, 240, 241, 242, 243, 244, 245, 246, 247, 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, 276, 277, 280, 281, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 321, 322, 323, 324, 325, 326, 329, 330, 331, 332, 347, 348, 349, 363, 377, 378, 379, 381, 385, 386, 393, 398, 401, 406, 414, 422, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 479, 497, 504, 506], "nonlinear": [406, 502], "nonzero": 506, "noop": [398, 503], "nor": [2, 177, 317], "norm": [6, 152, 328, 366, 457, 483, 484], "norm1": 6, "norm2": 6, "norm_first": 422, "normal": [1, 2, 5, 6, 150, 151, 152, 200, 256, 262, 345, 347, 348, 349, 350, 366, 370, 372, 377, 378, 379, 405, 422, 425, 427, 509, 512], "not_equ": 0, "notabl": [6, 8], "notat": [122, 329, 390], "note": [0, 1, 2, 4, 6, 9, 17, 19, 84, 93, 99, 106, 109, 110, 118, 154, 162, 163, 173, 174, 185, 203, 205, 248, 252, 319, 326, 345, 405, 423, 477, 509, 511], "noth": [6, 118, 345, 508], "notic": [6, 504, 505, 511], "now": [1, 2, 6, 9, 404, 502, 509], "np": [1, 6, 7, 503, 509, 510], "npy": [213, 274, 511], "npz": [6, 213, 277, 278, 388, 392, 511], "nuc": 205, "nuclear": 205, "nuisanc": 503, "nullopt": 0, "num": [0, 6, 212, 261], "num_class": [7, 477], "num_decoder_lay": 422, "num_embed": [362, 403], "num_encoder_lay": 422, "num_epoch": [7, 477], "num_exampl": 5, "num_featur": [5, 350], "num_group": 366, "num_head": [6, 401, 422], "num_it": 5, "num_lay": [6, 7, 477], "num_param": 345, "num_paramet": 402, "num_sampl": 252, "num_split": 0, "number": [0, 2, 12, 19, 62, 71, 99, 105, 106, 107, 109, 110, 122, 147, 149, 154, 173, 177, 181, 192, 212, 236, 243, 248, 249, 252, 255, 257, 261, 263, 267, 271, 272, 308, 309, 313, 317, 320, 321, 325, 326, 345, 350, 352, 353, 354, 355, 356, 357, 359, 360, 366, 370, 401, 402, 422, 423, 425, 426, 427, 428, 491, 493, 494, 499, 502, 505, 507, 513], "number_of_el": 0, "numer": [6, 150, 152, 205, 218, 222, 290, 350, 366, 370, 372, 405, 444, 445, 447, 457, 478, 479, 480, 481, 482, 483, 489, 502, 508], "numpi": [2, 6, 7, 8, 14, 17, 19, 89, 91, 92, 94, 95, 134, 135, 139, 178, 179, 185, 194, 195, 196, 218, 223, 225, 233, 235, 239, 245, 266, 270, 300, 508, 510, 511], "nw": 1, "nwhc": 359, "o": [2, 9, 154, 371], "o_t": 371, "obj": 275, "object": [3, 11, 31, 51, 79, 99, 147, 148, 151, 191, 277, 321, 329, 330, 331, 332, 336, 359, 422, 501, 507], "observ": 6, "occupi": [122, 173, 248, 249], "occur": 509, "odim": 7, "odot": [365, 371], "off": [6, 9, 508], "offer": 452, "offset": [0, 1, 2, 6, 47, 84, 124, 150, 153, 311], "often": 360, "ok": [388, 502, 504, 505], "okai": [502, 508], "old": 6, "older": [146, 148, 183], "omit": [483, 503], "onc": [2, 9, 502, 504], "one": [0, 2, 4, 6, 9, 39, 79, 86, 94, 98, 105, 106, 107, 109, 110, 129, 144, 146, 149, 150, 152, 153, 205, 216, 223, 249, 252, 295, 300, 316, 325, 336, 398, 423, 446, 503, 504, 507, 512], "ones": [0, 2, 6, 241, 277, 289, 313, 399, 400, 477, 503, 506], "ones_lik": 0, "onli": [1, 2, 6, 8, 9, 83, 93, 105, 106, 107, 109, 110, 118, 200, 201, 205, 210, 248, 256, 283, 319, 336, 345, 385, 386, 388, 393, 395, 398, 399, 400, 475, 502, 503, 504, 505, 507, 511, 512], "onlin": 480, "op": [1, 2, 242, 319, 386, 508], "open": [3, 9, 19, 259, 263, 503], "openmpi": 503, "oper": [3, 6, 8, 10, 38, 86, 87, 88, 107, 154, 172, 173, 245, 247, 290, 298, 305, 334, 336, 345, 422, 484, 502, 503, 505, 506, 508, 509, 510, 512, 513], "operand": [136, 137, 172], "opportun": 502, "opt": [485, 503], "optim": [1, 3, 5, 7, 8, 399, 502, 503, 505, 508], "option": [0, 3, 6, 15, 16, 18, 19, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 84, 86, 87, 88, 93, 99, 100, 104, 105, 106, 107, 108, 109, 110, 111, 114, 115, 116, 117, 118, 122, 123, 124, 127, 128, 129, 131, 132, 133, 146, 148, 149, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 171, 172, 173, 177, 181, 189, 190, 193, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 222, 224, 226, 227, 232, 236, 240, 243, 244, 246, 248, 249, 251, 252, 253, 255, 256, 257, 258, 259, 261, 262, 263, 267, 269, 271, 290, 291, 292, 295, 296, 297, 301, 303, 304, 308, 310, 311, 312, 313, 314, 315, 316, 317, 318, 321, 323, 325, 326, 329, 330, 331, 332, 347, 348, 349, 350, 352, 353, 354, 355, 356, 357, 365, 371, 374, 377, 378, 379, 381, 385, 386, 388, 393, 398, 401, 403, 404, 406, 409, 414, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 478, 479, 480, 481, 482, 483, 484, 486, 489, 490, 491, 499, 502, 504, 511, 513], "ord": 205, "order": [0, 1, 29, 84, 107, 137, 200, 201, 205, 244, 248, 310, 345, 366, 399, 411, 486, 502, 505, 507], "ordinari": 184, "org": [366, 370, 372, 380, 405, 436, 458], "origin": [6, 124, 328, 350, 394, 425, 426, 427, 428, 478, 479, 480, 483, 484, 504, 509], "orthonorm": 180, "ostream": 2, "ostringstream": 2, "other": [0, 2, 6, 8, 191, 205, 345, 387, 475, 484, 502, 503, 504, 506, 507, 508, 510], "other_input": 345, "otherwis": [19, 107, 129, 258, 326, 329, 330, 331, 332, 386, 388, 398, 420, 422, 423, 438, 444, 449, 456, 468, 469, 508, 509], "our": [1, 2, 6, 7, 411, 478, 479, 480, 483, 484, 503], "out": [0, 1, 2, 9, 93, 151, 183, 359, 360, 395, 502, 503, 504, 505, 506], "out_ax": [321, 505], "out_channel": [352, 353, 354, 355, 356, 357], "out_dim": [345, 475], "out_dtyp": 2, "out_idx": 2, "out_mask": 93, "out_proj": [6, 475], "out_ptr": 2, "out_shap": [1, 2], "outer": [0, 502, 508], "outlier": 452, "output": [0, 1, 2, 6, 9, 16, 17, 18, 19, 29, 84, 93, 94, 95, 99, 101, 102, 103, 114, 115, 116, 117, 118, 136, 147, 149, 150, 151, 152, 153, 154, 161, 164, 165, 166, 171, 172, 177, 180, 181, 185, 205, 212, 222, 224, 226, 227, 232, 236, 240, 241, 244, 246, 247, 251, 252, 253, 255, 256, 257, 259, 262, 263, 277, 278, 288, 289, 290, 295, 297, 301, 305, 311, 313, 317, 318, 319, 320, 321, 322, 323, 324, 350, 352, 353, 354, 355, 356, 357, 370, 374, 401, 404, 420, 422, 423, 425, 426, 427, 428, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 469, 472, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512], "output_dim": [7, 345, 374, 404], "output_directori": 2, "output_dtyp": [1, 151], "output_fil": 6, "output_nam": [1, 151], "output_shap": [1, 151], "output_strip_trailing_whitespac": 4, "output_vari": 4, "outsid": [151, 168], "over": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 104, 105, 106, 107, 108, 109, 110, 114, 115, 116, 117, 157, 160, 163, 166, 184, 205, 210, 212, 222, 224, 226, 232, 244, 246, 273, 290, 291, 297, 301, 308, 310, 318, 350, 352, 353, 354, 355, 356, 357, 366, 372, 405, 446, 491, 494, 503, 505, 507], "overal": 2, "overhead": [502, 508, 512], "overlap": 1, "overload": 19, "overrid": [2, 138], "overview": 3, "overwrit": 6, "own": [9, 503, 509], "owndata": 509, "p": [9, 203, 251, 345, 358, 359, 360, 457, 481, 483], "pack": [173, 248, 249], "packag": [2, 5, 7, 9, 338, 472, 503, 507], "package_data": 2, "pad": [0, 1, 104, 105, 106, 107, 108, 109, 110, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 347, 348, 349, 352, 353, 354, 355, 356, 357, 377, 378, 379], "pad_valu": 0, "pad_width": [0, 243], "padding_hi": 0, "padding_lo": 0, "page": [503, 510], "pain": 6, "pair": [0, 2, 243, 388, 409], "pairwis": 457, "pan": 6, "paper": [350, 414, 478, 479, 480, 483, 484], "parallel": [503, 512], "param": [317, 326, 345, 472, 504, 505], "paramet": [0, 1, 2, 5, 6, 7, 13, 14, 15, 16, 17, 18, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 121, 122, 123, 124, 127, 128, 129, 131, 132, 133, 134, 135, 136, 137, 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, 168, 169, 170, 171, 172, 173, 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, 230, 232, 233, 234, 235, 236, 237, 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, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 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, 328, 329, 330, 331, 332, 333, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 370, 371, 372, 373, 374, 377, 378, 379, 381, 382, 385, 386, 388, 393, 394, 395, 398, 399, 400, 401, 402, 403, 404, 405, 406, 409, 411, 414, 418, 420, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 437, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 469, 471, 472, 475, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 488, 489, 490, 491, 492, 493, 494, 495, 497, 502, 503, 504, 505, 508], "parameter_scal": 479, "parametr": [402, 459], "pars": [6, 147], "parse_arg": 6, "parser": 6, "part": [1, 2, 146, 148, 182, 183, 264, 505, 506], "parti": 503, "partial": [399, 400, 502, 508], "particip": [127, 128, 131, 132, 133], "particular": [248, 366], "particularli": 502, "partit": [0, 29], "pass": [1, 2, 6, 7, 66, 80, 242, 243, 317, 325, 327, 329, 330, 331, 345, 386, 398, 399, 400, 411, 502, 503, 504, 507, 508], "password": [503, 507], "path": [3, 4, 9, 137, 146, 147, 148, 183, 230, 277, 278, 326, 331, 388, 503, 507], "pattern": [345, 508], "peak": [176, 268], "penalti": 490, "penros": 206, "pep": 509, "per": [6, 7, 122, 154, 173, 248, 249, 325, 326, 350, 366, 370, 372, 405, 497, 502, 503, 507, 508], "perceptron": [8, 504], "perf_count": 502, "perfectli": 508, "perform": [0, 1, 2, 3, 6, 8, 15, 93, 107, 114, 115, 116, 117, 133, 136, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 172, 173, 180, 200, 201, 223, 249, 272, 290, 304, 325, 345, 366, 422, 427, 428, 477, 502, 503, 506, 508, 512], "perhap": [2, 6], "perm": 7, "permtuat": 258, "permuat": 203, "permut": [0, 7], "persist": 9, "pg": 205, "phi": [363, 434], "physic": 503, "pi": [140, 363, 414, 435, 505], "pick": 2, "pip": [2, 4, 9], "pipelin": 2, "pivot": [203, 204], "pixel": 359, "place": [6, 39, 271, 272, 326, 503, 508, 509], "placehold": 502, "plai": [2, 6], "plain": 411, "plan": [2, 502], "platform": 9, "plot": 503, "plu": [0, 216], "png": 503, "point": [0, 2, 5, 6, 9, 84, 167, 170, 249, 336], "pool": [347, 348, 349, 377, 378, 379, 512], "popul": 2, "port": 507, "portion": 358, "posinf": [0, 236], "posit": [0, 6, 29, 124, 153, 168, 177, 190, 197, 198, 234, 236, 244, 256, 271, 311, 317, 330, 345, 352, 353, 354, 355, 356, 357, 401, 409, 414, 447, 457, 504], "possibl": [129, 292, 362, 403, 502, 503, 506, 512], "possibli": [6, 15, 93, 172, 223, 328], "postur": 6, "power": [0, 505, 509], "practic": [2, 502], "pre": [9, 154, 444], "preced": 366, "precis": [0, 6, 145, 154, 345, 363, 405, 444, 485, 502], "preclud": 345, "pred": [448, 452], "predic": [326, 393], "predict": [444, 447, 448, 449, 450, 451, 452, 454, 455, 456], "prefix": [321, 329, 331], "prelu": 345, "prepar": [2, 6, 503], "prepend": [3, 223], "preprint": [6, 478, 484], "preprocessor": 9, "present": 1, "preserv": [269, 505], "press": [6, 205], "pretti": [502, 508], "prevent": [298, 457, 509], "previou": [279, 282, 283], "primal": [1, 2, 118, 192, 320], "primit": 505, "print": [1, 2, 5, 6, 7, 9, 85, 118, 193, 328, 329, 330, 331, 333, 345, 499, 502, 503, 504, 505, 506, 507, 508, 509, 510], "prior": [247, 304, 305], "priorit": 505, "privat": [2, 4], "prng": [251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 499], "prob": 444, "probabl": [9, 259, 358, 359, 360, 404, 444, 446, 450, 512], "problem": [5, 7, 345], "process": [6, 107, 111, 126, 127, 128, 129, 131, 132, 133, 325, 330, 331, 360, 362, 422, 501, 503, 507], "processor": 9, "prod": [0, 1], "produc": [0, 2, 9, 99, 401, 472, 504], "product": [0, 2, 15, 84, 116, 184, 192, 193, 199, 223, 242, 246, 308, 320, 401, 510], "profil": 3, "program": [4, 176], "programmat": 400, "project": [3, 4, 6, 401, 504], "project_source_dir": 2, "promot": [2, 154], "promote_typ": 2, "promoted_dtyp": 2, "prompt": 6, "propag": [505, 506], "properti": [32, 39, 48, 52, 62, 63, 69, 71, 394, 397, 487, 503, 505], "proportion": 328, "protocol": 509, "provid": [0, 2, 6, 84, 122, 146, 147, 177, 258, 271, 308, 317, 325, 330, 332, 338, 345, 381, 386, 388, 398, 399, 400, 403, 404, 422, 423, 471, 475, 503, 504, 511, 513], "pseudo": [206, 499], "pth": 6, "public": [2, 345], "pun": 0, "pure": [1, 118, 345, 477], "purpos": [1, 205, 503], "purs": 6, "push": 2, "push_back": 2, "put": [0, 1, 7, 247, 502, 503], "put_along_axi": [0, 203], "py": [2, 6, 9, 503, 507], "pypi": 9, "python": [1, 3, 4, 6, 51, 69, 79, 85, 142, 325, 329, 330, 331, 332, 333, 475, 485, 486, 488, 501, 503, 504, 505, 507, 509], "python_execut": 4, "python_requir": 2, "pytorch": [6, 8, 363, 366, 505], "pytorch_compat": 366, "q": [154, 207], "quantiz": [0, 122, 173, 213, 249, 403, 404], "quantized_matmul": 0, "quantizedembed": 345, "quantizedlinear": 345, "quarter": 6, "queri": [6, 154, 283, 401], "query_input_dim": 401, "query_proj": 6, "question": [6, 508], "queue": 3, "quick": [2, 8], "quit": [505, 509], "quotient": [0, 134, 135, 170], "r": [2, 6, 207, 317, 359, 365], "r_t": 365, "race": 512, "radian": [0, 121], "rag": 6, "rain": 6, "rais": [0, 6, 118, 205, 245, 292, 388, 504], "ram": [6, 282], "random": [1, 2, 3, 5, 6, 7, 8, 151, 347, 348, 349, 350, 370, 377, 378, 379, 388, 395, 502, 504, 505, 512, 513], "randomli": [5, 6, 258, 358, 359, 360], "rang": [0, 2, 3, 5, 6, 7, 9, 19, 168, 172, 212, 426, 428, 435, 436, 477, 491, 492, 493, 494, 495, 499, 502, 505, 508, 512], "rank": [0, 131, 132, 133, 453, 503, 507], "rate": [5, 477, 478, 479, 480, 481, 482, 483, 484, 489, 490], "rather": [2, 505, 512], "ratio": [0, 25], "rceil": 93, "re": [7, 9, 472], "reachabl": 503, "readabl": 3, "real": [0, 161, 162, 163, 164, 165, 166, 197, 198, 200, 201], "realli": 372, "reason": [1, 6, 506], "reboot": 9, "receiv": [131, 132, 326, 493, 503, 509], "reciproc": [0, 273], "reclaim": 279, "recommend": [9, 282, 484], "recompil": [99, 502], "reconstruct": 203, "record": [3, 176, 508], "recreat": [333, 477], "rectifi": [373, 407, 408, 427, 428, 441, 460, 461], "recurr": [365, 371, 406], "recurs": [147, 345, 385, 386, 391, 396, 398, 475], "recv": [132, 503], "reduc": [0, 1, 9, 16, 18, 27, 28, 128, 222, 224, 226, 232, 246, 297, 301, 318, 325, 332, 350, 422, 452], "reduct": [16, 18, 128, 222, 224, 232, 246, 332, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457], "redund": 505, "refer": [205, 370, 380, 394, 425, 426, 427, 428, 436, 458, 506], "reflect": [394, 502, 506, 509], "regard": 363, "regardless": [84, 154, 503], "regist": [2, 7], "register_librari": 2, "regress": [8, 452], "regular": [39, 359, 458, 482, 502, 504, 506], "regularli": 2, "reimplement": 2, "rel": [17, 185, 479, 502, 503], "relative_step": 479, "releas": 4, "relev": 2, "reli": [1, 2], "relu": [345, 402, 422, 459, 472], "relu6": 345, "remain": [0, 6, 283, 317, 331, 358, 359, 360, 503], "remaind": [0, 135], "remov": [0, 124, 223, 252, 295, 446], "rep": [0, 309], "repeat": [0, 309], "repeatedli": 5, "repetit": 267, "replac": [0, 6, 236, 399, 400, 422, 456], "replai": 3, "repli": 6, "repo": [5, 7, 9, 502], "report": [174, 282], "repres": [2, 6, 126, 129, 173, 453, 457, 509], "represent": [6, 204, 248, 319, 329, 333], "requir": [1, 2, 4, 6, 345, 503, 507, 508, 509], "requires_grad": 505, "rerun": [502, 508], "rescal": 328, "research": 8, "reset": 268, "reset_peak_memori": 176, "reshap": [0, 6, 205, 423, 502, 506], "resid": 283, "resolv": 2, "resourc": 2, "resource_limit": 228, "respect": [2, 5, 7, 118, 150, 152, 172, 173, 177, 248, 317, 330, 345, 350, 363, 366, 370, 372, 475, 503, 505, 507, 510], "respons": 2, "rest": [6, 153, 330, 331, 409, 507], "restart": 9, "restor": 271, "result": [0, 6, 15, 19, 39, 79, 84, 99, 147, 150, 152, 173, 193, 205, 223, 249, 256, 267, 282, 296, 330, 331, 332, 336, 414, 444, 502, 503, 505, 509], "resum": 6, "return": [0, 1, 2, 4, 5, 6, 7, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 51, 69, 79, 83, 84, 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, 121, 122, 123, 124, 127, 128, 129, 131, 132, 133, 134, 135, 136, 137, 139, 140, 141, 143, 144, 145, 146, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 171, 172, 173, 175, 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, 232, 233, 234, 235, 236, 237, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 261, 262, 263, 264, 265, 266, 267, 269, 270, 272, 273, 279, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 326, 327, 328, 329, 330, 331, 332, 333, 345, 365, 371, 381, 382, 383, 385, 386, 387, 388, 389, 390, 391, 395, 396, 398, 399, 400, 406, 424, 425, 426, 427, 428, 429, 430, 431, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 472, 475, 485, 501, 502, 503, 504, 505, 506, 508, 509, 511, 512], "return_metadata": 213, "revers": [0, 2, 42, 43, 44, 45, 84, 114, 115, 116, 117, 312, 414], "rf": 9, "rfft": 161, "rfft2": 162, "rfftn": 163, "rho": 478, "rhs_indic": [0, 172, 173], "rhs_mask": 93, "right": [0, 1, 2, 9, 248, 270, 271, 363, 423, 435, 436, 447, 449, 457], "right_shift": 0, "ring": 129, "rm": [6, 9, 152, 479], "rmsnorm": [6, 345], "rmsprop": 477, "rnn": [345, 365], "robust": 452, "roform": [6, 409], "roll": 0, "root": [0, 6, 152, 273, 293, 405], "rope": [6, 345], "rosetta": 9, "rotari": [6, 153, 409], "rotat": [153, 409], "round": [0, 248], "row": [0, 1, 2, 84, 103, 149, 151, 181, 248, 313], "rpath": 2, "rsqrt": 0, "rtol": [0, 17, 185], "rule": [2, 477], "run": [1, 2, 3, 4, 6, 7, 8, 9, 10, 151, 242, 334, 350, 381, 478, 479, 481, 482, 483, 502, 504, 507, 508, 512, 513], "runtim": [6, 129, 338, 502, 503], "runtime_error": 2, "safetensor": [9, 213, 276, 388, 392, 477, 508, 511], "sai": [2, 6, 472, 508], "said": 6, "sake": 505, "same": [0, 2, 6, 9, 17, 39, 83, 94, 95, 99, 105, 106, 107, 109, 110, 111, 127, 150, 152, 161, 164, 165, 166, 173, 177, 185, 192, 243, 252, 271, 272, 289, 319, 320, 322, 325, 331, 345, 348, 349, 350, 358, 366, 370, 378, 379, 403, 424, 425, 426, 427, 428, 429, 430, 431, 446, 457, 475, 485, 499, 502, 503, 504, 506, 507, 512], "sampl": [2, 5, 6, 212, 251, 252, 253, 255, 256, 259, 262, 263, 425, 426, 427, 428, 430, 431, 447, 453, 457, 499, 502, 504], "sat": 6, "save": [3, 6, 8, 213, 230, 248, 275, 276, 277, 278, 392, 504, 508], "save_gguf": 511, "save_safetensor": [392, 477, 511], "save_weight": 345, "savez": [6, 392, 511], "savez_compress": 511, "saw": [6, 505], "scalar": [0, 2, 14, 15, 17, 31, 51, 79, 83, 89, 90, 91, 92, 93, 95, 98, 134, 135, 139, 170, 171, 177, 178, 179, 180, 185, 194, 195, 196, 212, 218, 219, 220, 221, 223, 225, 233, 235, 236, 239, 243, 245, 251, 259, 262, 263, 266, 270, 275, 300, 317, 319, 322, 327, 457, 504, 505, 508, 510], "scale": [0, 2, 6, 15, 122, 150, 152, 153, 154, 173, 180, 248, 249, 255, 257, 328, 359, 360, 372, 401, 409, 410, 414, 423, 462, 479], "scale_arr": 2, "scale_factor": 423, "scale_paramet": 479, "scatter": 0, "scatter_add": 0, "scatter_add_axi": 0, "scatter_max": 0, "scatter_min": 0, "scatter_prod": 0, "schedul": [2, 477, 491, 492, 493, 494, 495, 497, 512], "schema": [3, 507], "scipi": [180, 203], "scope": 345, "score": [6, 154, 453], "script": [503, 507], "sdk": 9, "se": 1, "second": [6, 9, 124, 191, 193, 194, 219, 221, 223, 270, 302, 311, 317, 348, 349, 378, 379, 445, 453, 479, 483, 502, 504, 505, 512], "second_layer_a": 508, "second_layer_b": 508, "secret": 6, "section": [1, 6, 9, 292, 457, 502, 503, 505], "see": [1, 2, 4, 6, 7, 9, 11, 12, 33, 34, 35, 36, 37, 40, 41, 42, 43, 44, 45, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 205, 275, 276, 279, 326, 336, 345, 350, 351, 359, 361, 363, 367, 368, 369, 375, 376, 384, 402, 403, 404, 407, 408, 409, 410, 412, 414, 415, 416, 417, 418, 419, 421, 423, 425, 426, 427, 428, 434, 435, 436, 462, 502, 503, 504, 505, 506, 507, 510, 512], "seed": 254, "seen": [503, 509], "select": [0, 3, 9, 200, 201, 310, 322, 381, 385, 393, 507], "self": [6, 7, 10, 31, 32, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 51, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 118, 167, 345, 458, 475], "selu": 345, "semant": [14, 89, 91, 92, 94, 95, 134, 135, 139, 178, 179, 194, 195, 196, 218, 223, 225, 233, 235, 239, 245, 266, 270, 300, 512], "semi": [197, 198, 256], "send": 503, "sender": 503, "sennrich": 6, "sensit": 452, "sentencepiec": 6, "separ": [6, 66, 80, 366, 453], "sequenc": [6, 16, 18, 34, 35, 57, 58, 59, 60, 64, 72, 75, 76, 77, 81, 84, 95, 107, 131, 144, 151, 156, 157, 159, 160, 162, 163, 165, 166, 171, 177, 222, 224, 226, 232, 240, 246, 251, 252, 253, 255, 256, 257, 259, 262, 263, 269, 288, 289, 290, 292, 295, 297, 301, 308, 309, 312, 316, 317, 318, 323, 350, 352, 355, 365, 371, 406, 422, 499, 512], "sequenti": [345, 472], "seri": 9, "serial": 477, "set": [2, 4, 6, 7, 9, 99, 118, 125, 127, 128, 129, 131, 132, 133, 138, 150, 152, 153, 228, 279, 280, 281, 282, 283, 299, 325, 363, 372, 374, 384, 386, 393, 394, 395, 398, 399, 404, 409, 420, 445, 457, 469, 475, 477, 479, 481, 482, 486, 499, 504, 505, 508], "set_byt": 2, "set_compute_pipeline_st": 2, "set_data": 2, "set_dtyp": 345, "set_input_arrai": 2, "set_memory_limit": 279, "set_output_arrai": 2, "set_vector_byt": 2, "setup": [2, 4, 5, 7, 9, 502, 503, 504], "sever": [6, 9, 104, 105, 106, 107, 108, 109, 110, 277, 278, 325, 502, 503, 507, 511], "sgd": [5, 7, 477, 484, 486, 491, 492, 495, 502], "shade": [1, 2], "shall": 6, "shape": [0, 2, 3, 6, 7, 66, 83, 84, 93, 94, 95, 99, 104, 105, 106, 107, 108, 109, 110, 124, 127, 131, 132, 146, 148, 151, 154, 155, 158, 161, 164, 165, 166, 171, 172, 180, 192, 202, 211, 223, 240, 241, 251, 252, 253, 255, 256, 257, 259, 262, 263, 269, 271, 289, 316, 319, 320, 322, 323, 324, 345, 347, 348, 349, 350, 352, 353, 354, 355, 356, 357, 359, 360, 365, 370, 371, 374, 377, 378, 379, 388, 406, 424, 425, 426, 427, 428, 429, 430, 431, 446, 457, 477, 502, 504, 505, 506, 510, 512], "shapeless": [0, 99, 146, 148], "share": [8, 122, 173, 248, 249, 319, 503], "shazeer": 6, "shift": [0, 194, 270, 271, 350], "shop": 6, "should": [1, 2, 4, 5, 6, 7, 9, 84, 97, 124, 127, 150, 151, 152, 154, 192, 230, 247, 248, 283, 305, 311, 317, 320, 325, 326, 329, 345, 352, 353, 354, 355, 356, 357, 359, 360, 395, 401, 411, 446, 448, 453, 475, 501, 502, 503, 504, 505, 508, 509, 513], "show": [9, 336, 502], "shown": 2, "shuffl": 7, "side": [0, 243, 347, 348, 349, 377, 378, 379, 502], "sigma": [363, 364, 365, 371, 413, 425, 426, 427, 428, 436, 437, 442, 463, 464], "sigmoid": [0, 6, 345, 375, 412, 436, 442, 444, 464], "sign": [0, 17, 185, 336, 484], "signal": [111, 423], "signatur": [1, 151], "signedinteg": [12, 191], "signific": 248, "significantli": 503, "silent": [164, 165, 166], "silicon": [2, 6, 8, 9, 512], "silu": 345, "simd": 1, "simd_sum": 1, "simdgroup": 1, "simdgroup_s": 1, "similar": [6, 173, 191, 330, 399, 400, 401, 445, 503, 509, 511], "similarli": [2, 9, 223, 505, 508], "simpl": [2, 6, 7, 345, 362, 471, 477, 502, 503, 504, 505, 507, 508], "simple_axpbi": 2, "simple_tim": 2, "simplest": [2, 345, 503], "simpli": [2, 6, 9, 361, 373, 407, 433, 441, 460, 470, 475, 502, 503, 505, 507], "simplic": 0, "simplifi": 503, "simultan": 1, "sin": [0, 118, 414, 504, 505, 510], "sinc": [1, 2, 6, 7, 173, 176, 475, 484, 493, 502, 504, 509, 512], "sine": [0, 22, 23, 286, 287, 504, 505], "sing": 205, "singer": 480, "singl": [7, 85, 142, 192, 213, 227, 243, 320, 348, 349, 378, 379, 502, 504, 506, 511], "singleton": [0, 16, 18, 27, 28, 129, 222, 223, 224, 226, 232, 246, 297, 301, 318, 503], "singular": [205, 206, 210], "sinh": 0, "sinusoid": 414, "sinusoidalpositionalencod": 345, "size": [0, 1, 2, 6, 7, 52, 69, 93, 105, 106, 109, 110, 122, 144, 150, 151, 152, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 171, 173, 175, 180, 181, 191, 199, 205, 248, 249, 252, 269, 282, 283, 288, 292, 295, 316, 319, 325, 326, 345, 347, 348, 349, 352, 353, 354, 355, 356, 357, 362, 370, 377, 378, 379, 403, 404, 423, 479, 503, 508, 509], "size_in_megabyt": 283, "size_t": [0, 2], "skip": [3, 84], "slice": [0, 289, 506], "slice_s": [0, 288], "slice_upd": 0, "slide": [347, 348, 349, 377, 378, 379], "slight": [6, 508], "slightli": [409, 512], "slope": 373, "slow": 502, "slowli": 6, "small": [6, 145, 150, 152, 325, 350, 366, 372, 405, 447, 452, 457, 502, 512], "smaller": [0, 9, 244, 325, 484, 502], "smallest": 205, "smile": 6, "smooth": [446, 456, 489], "smooth_l1_loss": 345, "sned": 133, "snippet": 503, "so": [1, 2, 6, 9, 177, 180, 317, 358, 423, 477, 502, 503, 508, 512], "socket": 503, "softmax": [0, 6, 154, 345, 376, 443, 446], "softmin": 345, "softplu": [345, 380, 458], "softshrink": 345, "softsign": 345, "solut": [208, 209], "solv": 345, "some": [0, 2, 5, 6, 7, 147, 386, 398, 477, 486, 502, 503, 504, 505, 507, 508], "someon": 6, "someth": [5, 6, 506], "sometim": 502, "sonoma": 9, "soon": 6, "sort": [0, 29, 30, 244, 310], "sourc": [0, 1, 2, 3, 4, 61, 131, 132, 151, 234, 312, 503], "space": [0, 2, 212, 444, 455], "spars": [0, 227], "spatial": [105, 106, 107, 109, 110, 347, 348, 349, 366, 377, 378, 379, 423], "speak": [6, 205], "specif": [1, 2, 9, 503, 505], "specifi": [0, 2, 19, 38, 105, 106, 107, 109, 110, 124, 162, 163, 171, 177, 199, 205, 212, 234, 240, 247, 252, 267, 302, 304, 305, 308, 311, 312, 317, 321, 323, 350, 420, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 469, 502, 503, 504, 505, 512], "speed": 1, "spent": 6, "split": [0, 364, 366, 437], "splittabl": 499, "sqrt": [0, 6, 140, 154, 180, 350, 363, 366, 370, 372, 374, 405, 414, 425, 426, 427, 428, 435, 478, 480, 481, 482, 489, 502], "squar": [0, 5, 6, 152, 181, 202, 211, 273, 293, 317, 330, 345, 405, 454, 456, 478, 479, 481, 482, 483, 505, 509], "squeez": [0, 423, 502], "src": [0, 131, 132], "ssh": [503, 507], "stabil": [150, 152, 350, 366, 370, 372, 405, 444, 445, 447, 478, 479, 480, 481, 482, 483, 489], "stabl": [218, 222, 290, 452], "stable_abi": 2, "stack": [0, 502], "standard": [0, 1, 4, 7, 51, 79, 223, 253, 257, 297, 422, 425, 427, 430, 503, 510], "starmap": [6, 330], "start": [0, 1, 2, 5, 6, 8, 9, 19, 153, 212, 230, 288, 289, 292, 332, 502, 504, 506, 507, 512], "start_axi": [0, 50, 168], "start_captur": 3, "start_indic": [288, 289], "state": [6, 7, 345, 365, 371, 406, 477, 486, 499, 502], "static": [9, 502], "static_cast": 2, "std": [0, 2, 4, 430, 504], "stderr": 507, "stdout": 507, "step": [0, 3, 4, 6, 7, 19, 325, 345, 365, 371, 406, 479, 486, 491, 493, 494, 495, 502, 503], "step_decai": 477, "step_siz": 495, "still": [6, 9, 205, 502, 508], "stochast": [480, 481, 483, 490, 508], "stood": 6, "stop": [0, 2, 6, 19, 212, 231, 298, 505, 506], "stop_captur": 3, "stop_gradi": [0, 505], "storag": 84, "store": 6, "str": [2, 111, 129, 136, 137, 146, 147, 148, 151, 154, 177, 183, 200, 201, 205, 213, 227, 228, 230, 274, 275, 276, 277, 278, 317, 326, 329, 333, 381, 382, 385, 386, 388, 390, 392, 398, 423, 427, 428, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457], "straight": 6, "strang": 6, "stream": [2, 8, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 98, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 120, 121, 122, 123, 124, 127, 128, 131, 132, 133, 134, 135, 136, 139, 140, 141, 143, 144, 145, 149, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 171, 172, 173, 178, 179, 180, 181, 182, 184, 185, 186, 187, 188, 189, 190, 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, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 255, 256, 257, 258, 259, 261, 262, 263, 264, 265, 266, 267, 269, 270, 271, 272, 273, 281, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 318, 319, 322, 323, 324, 503, 512], "streamcontext": 299, "streamordevic": [0, 2], "street": 6, "strength": [484, 490], "strict": [129, 178, 195, 386, 388, 398], "strictli": [205, 283], "stride": [0, 2, 84, 104, 105, 106, 107, 108, 109, 110, 347, 348, 349, 352, 353, 354, 355, 356, 357, 377, 378, 379, 409, 506], "string": [0, 2, 137, 146, 151, 183, 228, 243, 504, 509, 511], "stronger": 507, "structur": [2, 325, 485, 505], "stub": 9, "style": [2, 14, 17, 89, 91, 92, 134, 135, 139, 178, 179, 185, 194, 195, 196, 218, 223, 225, 233, 235, 239, 245, 266, 270, 300], "su": 6, "sub": [0, 7, 124, 261, 288, 289, 311, 326], "subarrai": [124, 292], "subclass": 475, "subdivid": 1, "subdtyp": 191, "subgradi": 480, "sublinear": 479, "submodul": [6, 7, 345, 382, 386, 387, 398, 400], "subnetwork": 503, "suboptim": 504, "subscript": [136, 137], "subsect": 6, "subsequ": [129, 477, 503, 507], "subset": [345, 385], "substanti": 9, "subtl": 502, "subtract": [0, 39], "subtyp": [191, 336], "succe": 129, "successfulli": 503, "sudo": [9, 283, 503], "suggest": 503, "sum": [0, 2, 5, 14, 117, 128, 148, 184, 205, 222, 290, 308, 311, 345, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 503, 506, 509], "sum_": [205, 452], "sum_i": 443, "sum_j": [465, 466], "summat": [136, 137], "super": [6, 7, 345, 475], "superset": [330, 485], "support": [1, 2, 6, 8, 9, 17, 93, 106, 109, 110, 154, 168, 180, 185, 197, 198, 200, 201, 202, 206, 207, 210, 211, 213, 223, 248, 256, 503, 505, 506, 509, 511], "suppos": [505, 512], "sure": [2, 3, 6, 9, 345, 502], "surpass": [427, 428], "surpris": 6, "sw": 1, "swap": [0, 111, 282, 302, 400], "swapax": [0, 118], "swiglu": 6, "swish": [412, 464], "switch": 9, "symbol": 484, "symmetr": [105, 106, 109, 110, 197, 198, 200, 201], "symmetri": [200, 201], "synchron": [2, 502], "syntax": [39, 506], "synthet": 5, "sysctl": 283, "system": [4, 6, 9, 174, 175, 208, 209, 228, 283], "t": [0, 1, 2, 4, 6, 9, 103, 140, 151, 154, 173, 197, 198, 249, 317, 345, 365, 371, 406, 478, 479, 480, 481, 482, 483, 484, 489, 490, 502, 504, 505, 512], "t_kv": 154, "t_q": 154, "tabl": [1, 205, 336, 362], "take": [0, 2, 6, 7, 89, 90, 91, 92, 99, 146, 172, 177, 192, 225, 233, 241, 249, 305, 317, 320, 321, 324, 331, 332, 347, 348, 349, 377, 378, 379, 401, 444, 499, 503, 504, 505, 506, 507, 511, 512, 513], "take_along_axi": [0, 203, 506], "taken": [124, 304, 311], "talk": 503, "tan": 0, "tangent": [0, 2, 24, 25, 26, 118, 192, 306, 307, 421, 470], "tangent_i": 2, "tangent_x": 2, "tanh": [0, 345, 363, 365, 371, 380, 406, 435, 458], "target": [2, 317, 444, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 502], "target_include_directori": 2, "target_link_librari": [2, 4], "target_link_opt": 2, "target_sourc": 2, "task": 452, "tau": 490, "tcp": 503, "tediou": 503, "tell": [4, 6, 502, 509], "temp": 6, "templat": [0, 1, 2, 151], "ten": 508, "tend": 484, "tensor": [213, 308, 457, 509], "tensordot": 0, "term": [2, 447, 478, 479, 480, 481, 482, 483, 489], "termin": [9, 507], "test": [7, 9, 503, 507], "test_imag": 7, "test_label": 7, "text": [6, 363, 365, 371, 380, 406, 413, 420, 425, 426, 427, 428, 435, 438, 439, 440, 447, 448, 449, 452, 453, 456, 458, 459, 462, 463, 468, 469, 479, 484], "textrm": [248, 363, 364, 434, 437], "tf": 509, "tgp_size": 2, "th": [114, 115, 116, 117, 123, 149, 200, 493], "than": [1, 2, 6, 79, 111, 124, 135, 153, 172, 178, 179, 195, 196, 197, 198, 200, 201, 202, 203, 206, 210, 211, 223, 279, 283, 328, 330, 409, 420, 423, 453, 456, 469, 479, 484, 502, 504, 505, 512], "thank": 508, "thei": [1, 2, 5, 6, 9, 17, 111, 173, 185, 411, 448, 475, 484, 501, 502, 503, 504, 508, 510, 511, 512], "them": [0, 2, 6, 127, 345, 386, 398, 503, 504, 507, 512], "themselv": [2, 502], "thi": [0, 1, 2, 4, 6, 7, 9, 16, 17, 18, 19, 27, 28, 29, 30, 84, 85, 97, 118, 138, 146, 148, 151, 172, 173, 174, 180, 183, 185, 192, 197, 198, 200, 201, 202, 205, 206, 207, 210, 211, 218, 222, 223, 224, 226, 232, 244, 246, 252, 281, 283, 290, 291, 292, 297, 301, 304, 310, 318, 325, 328, 331, 332, 345, 358, 359, 360, 364, 365, 371, 382, 383, 385, 386, 389, 390, 391, 396, 398, 399, 400, 401, 404, 406, 420, 425, 426, 427, 428, 435, 436, 437, 444, 452, 469, 475, 486, 501, 502, 503, 504, 505, 507, 508, 509, 511], "thin": 507, "thing": [2, 6], "third": [199, 349, 379, 503, 504], "thompson": 359, "those": [2, 6, 345], "though": [2, 6, 502, 504, 508, 509], "thousand": 508, "thread": [1, 2], "thread_index_in_simdgroup": 1, "thread_position_in_grid": [1, 2, 151], "threadgroup": [1, 2, 151], "threads_per_simdgroup": 1, "three": [6, 88, 349, 379, 423], "threefri": 499, "threshold": [420, 449, 456, 469], "through": [1, 2, 298, 422, 484, 502, 503, 504, 505, 509], "throw": [2, 99, 129], "thu": [6, 345], "thumb": 477, "tic": 502, "tieleman": 489, "tile": [0, 154], "time": [2, 6, 9, 282, 309, 345, 365, 371, 406, 502, 503, 505, 508, 512], "timeit": [502, 505], "titl": 2, "tmp": [1, 151], "to_quant": 326, "to_stream": 2, "toc": 502, "togeth": [0, 1, 2, 7, 248, 330, 331, 503], "tok_embed": 6, "token": [6, 362, 403], "told": 6, "toler": [0, 17, 185], "too": [191, 502, 508], "took": 6, "tool": 9, "top": [2, 310, 374, 423], "topk": 0, "torch": [6, 509], "torch_weight": 6, "total": [283, 505], "total_norm": 328, "tpi": 502, "tpng": 503, "trace": [0, 3, 148, 502], "trace_fil": 3, "tracer": 399, "track": [2, 345, 350], "track_running_stat": 350, "trade": 508, "tradit": [6, 153, 359, 360, 409], "train": [6, 7, 345, 350, 358, 359, 360, 384, 386, 398, 425, 426, 504], "train_imag": [7, 477], "train_label": [7, 477], "trainabl": [7, 327, 345, 475], "trainable_paramet": [345, 385, 486], "transfer": 507, "transform": [1, 6, 8, 118, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 180, 327, 345, 350, 366, 372, 374, 385, 386, 398, 404, 409, 506], "transformerencod": 277, "transit": 493, "translat": [150, 372], "transpos": [0, 6, 32, 108, 109, 110, 173, 249, 355, 356, 357], "treat": [0, 2, 118, 162, 163, 165, 166, 304, 423, 502], "tree": [8, 85, 99, 142, 177, 317, 321, 325, 329, 330, 331, 332, 333, 485, 486, 488, 497, 505], "tree_flatten": [277, 330, 333, 345, 477, 504], "tree_map": [331, 345, 503], "tree_unflatten": [6, 477, 504], "trembl": 6, "tri": [0, 129], "triangl": [200, 201, 313], "triangular": [197, 198, 209, 211], "trigger": 502, "tril": 0, "trilinear": 423, "triplet": 457, "triplet_loss": 345, "triu": 0, "true": [0, 1, 2, 4, 5, 6, 17, 42, 43, 44, 45, 83, 99, 114, 115, 116, 117, 151, 153, 173, 185, 191, 197, 198, 205, 210, 213, 227, 249, 290, 322, 326, 329, 330, 331, 332, 336, 345, 350, 352, 353, 354, 355, 356, 357, 365, 366, 370, 371, 372, 374, 385, 386, 388, 395, 398, 404, 406, 409, 414, 422, 423, 444, 452, 479, 481, 482, 502, 504], "truncat": [155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 262], "truth": [5, 446, 456], "try": [2, 9, 503], "tupl": [0, 31, 66, 69, 80, 85, 94, 100, 105, 106, 107, 109, 110, 131, 135, 137, 142, 144, 146, 183, 192, 200, 203, 204, 205, 207, 210, 243, 248, 269, 271, 288, 289, 295, 316, 317, 320, 329, 330, 331, 332, 333, 347, 348, 349, 353, 354, 356, 357, 377, 378, 379, 388, 390, 411, 423, 479, 481, 482, 483, 484, 501, 504, 505], "tutori": 2, "twice": 512, "two": [0, 2, 14, 15, 17, 25, 83, 87, 89, 91, 92, 93, 124, 134, 139, 156, 159, 165, 172, 173, 178, 179, 185, 193, 195, 196, 197, 198, 199, 200, 201, 202, 206, 207, 210, 211, 218, 223, 225, 233, 235, 239, 242, 302, 332, 348, 364, 371, 378, 437, 445, 502, 503, 504, 505, 506, 512], "txt": [2, 4], "type": [0, 1, 2, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 38, 69, 79, 83, 84, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 121, 122, 123, 124, 127, 128, 129, 131, 132, 133, 134, 135, 136, 137, 139, 140, 141, 143, 144, 145, 149, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 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, 232, 233, 234, 235, 236, 237, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 261, 262, 263, 264, 265, 266, 267, 269, 270, 272, 273, 279, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 300, 301, 302, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 328, 329, 332, 345, 393, 422, 424, 425, 426, 427, 428, 429, 430, 431, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 502, 504, 506, 509], "type_to_nam": 2, "typenam": [0, 1, 2], "typic": [0, 154, 325, 362, 477, 502, 508], "u": [1, 2, 4, 197, 200, 201, 203, 210, 374, 400, 497, 503, 507, 508], "u_": 478, "u_t": 478, "uint": [1, 2, 151], "uint16": [12, 336], "uint3": 1, "uint32": [12, 27, 28, 29, 30, 252, 336], "uint64": [12, 336], "uint8": [12, 336], "ultra": 6, "unabl": 9, "unam": 9, "unari": 502, "unchang": [153, 298, 409], "uncheck": 9, "uncompress": 277, "undefin": [0, 29, 118, 197, 198, 244, 256, 506], "under": 205, "underli": [2, 319], "understand": [6, 425, 426], "unevalu": 147, "unexpect": [2, 19], "unexpectedli": 507, "unflatten": 0, "unfreez": [345, 386], "unfrozen": 398, "unifi": 8, "uniform": [3, 345, 374, 388, 426, 428, 472, 499, 502, 505, 512], "uniformli": 263, "unintend": 0, "union": [19, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 53, 54, 55, 56, 57, 58, 59, 60, 61, 64, 65, 66, 67, 68, 70, 72, 73, 74, 75, 76, 77, 78, 80, 81, 82, 86, 87, 88, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 189, 190, 191, 193, 200, 201, 210, 228, 275, 299], "uniqu": [2, 208, 209, 499, 503], "unique_ptr": 2, "unit": [351, 361, 363, 364, 365, 373, 407, 408, 410, 412, 425, 426, 427, 428, 432, 433, 434, 435, 436, 437, 441, 460, 461, 462, 464], "unittest": 9, "univers": 205, "unless": [6, 17, 185, 205, 475], "unlik": [6, 17, 185, 203, 359, 360, 394], "unnecessari": [2, 6], "unnorm": [252, 444, 446], "unscal": 479, "unsign": [173, 248, 249, 336], "unsignedinteg": 12, "unspecifi": [16, 18, 19, 27, 28, 29, 30, 100, 114, 115, 116, 117, 171, 222, 224, 226, 232, 240, 244, 246, 267, 290, 291, 297, 301, 304, 310, 311, 318, 323, 513], "unsqueez": 6, "unsupport": 213, "until": [2, 325, 508, 510], "unus": 2, "up": [1, 2, 6, 118, 502], "upcast": 2, "updat": [0, 1, 2, 5, 6, 7, 39, 99, 289, 326, 330, 332, 350, 381, 382, 388, 393, 394, 395, 400, 477, 479, 482, 484, 485, 486, 490, 491, 492, 493, 494, 495, 502, 503, 504, 508], "update_modul": 345, "uplo": [200, 201], "upon": [6, 330, 331], "upper": [197, 198, 200, 201, 209, 211, 248, 259, 262, 263, 431], "upsampl": 345, "us": [0, 3, 5, 6, 7, 8, 9, 19, 39, 84, 118, 122, 125, 127, 128, 131, 132, 133, 135, 151, 153, 168, 173, 174, 175, 176, 194, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 223, 248, 249, 267, 269, 270, 271, 279, 282, 283, 303, 316, 325, 329, 332, 336, 338, 345, 348, 349, 359, 362, 363, 365, 371, 374, 378, 379, 381, 385, 392, 399, 401, 403, 404, 406, 409, 414, 422, 423, 427, 428, 435, 436, 445, 472, 475, 477, 478, 479, 481, 482, 483, 484, 485, 486, 499, 501, 502, 503, 504, 505, 506, 507, 510, 512], "usag": [422, 502, 503], "user": [2, 6, 345], "usual": [362, 403, 501, 508], "util": [1, 2, 6, 8, 9, 277, 345, 477, 507], "v": [6, 111, 154, 200, 345, 386, 509], "v_": [478, 480, 481, 482, 483, 489, 490], "v_t": [478, 480, 481, 482, 483, 489, 490], "val": [0, 31, 171], "valid": [7, 111, 168, 321, 329, 386, 398, 501, 503], "valid_parameter_filt": 381, "valu": [0, 1, 5, 6, 12, 13, 17, 19, 27, 28, 51, 79, 83, 98, 129, 146, 149, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 171, 183, 185, 199, 205, 206, 210, 212, 228, 236, 243, 247, 251, 252, 253, 255, 256, 257, 259, 262, 263, 271, 275, 283, 304, 305, 317, 321, 327, 329, 330, 331, 332, 336, 348, 349, 351, 358, 359, 360, 361, 367, 370, 374, 378, 379, 385, 401, 402, 418, 420, 422, 424, 444, 445, 446, 447, 448, 449, 451, 452, 453, 454, 455, 456, 469, 475, 479, 482, 491, 492, 494, 495, 505], "value_and_grad": [7, 118, 345, 399, 475, 477, 488, 502, 505, 509, 510], "value_and_grad_fn": 508, "value_cach": 6, "value_dim": 401, "value_input_dim": 401, "value_output_dim": 401, "value_proj": 6, "valueerror": [118, 205, 388, 505], "values_hat": 6, "van": 205, "var": [0, 350, 366, 370, 372, 447], "variabl": [9, 99, 118, 125, 138, 146, 147, 148, 177, 192, 317, 320, 321, 502, 503, 504], "varianc": [0, 297, 318, 350, 366, 447], "variant": [6, 456, 483], "variou": 205, "vector": [0, 2, 5, 8, 184, 192, 205, 304, 320, 321, 362, 446, 504, 510], "verbos": [1, 151, 503], "veri": [6, 401, 507, 508, 512], "verifi": [5, 9], "versa": 271, "version": [2, 4, 9, 85, 122, 146, 148, 183, 218, 222, 248, 290, 321, 499, 505, 506], "versu": 502, "via": [9, 118, 485, 488, 503, 507, 508, 509], "vice": 271, "video": 360, "view": [0, 3, 84, 509], "virtual": 2, "visual": 147, "vjp": [2, 118, 510], "vmap": [2, 118, 504, 505, 508, 510], "vmap_add": 505, "vocab_s": 6, "vocabulari": [362, 403], "void": [1, 2], "vt": 210, "w": [0, 1, 5, 105, 106, 109, 110, 122, 173, 200, 248, 249, 317, 331, 350, 353, 354, 356, 357, 359, 360, 374, 477, 490, 505], "w1": [6, 328], "w2": [6, 328], "w3": 6, "w_": [365, 371, 406, 478, 479, 480, 481, 482, 483, 484, 489, 490], "w_1": 248, "w_g": 248, "w_i": [122, 248], "w_in": 1, "w_q": 248, "w_star": 5, "w_stride": 1, "w_t": [478, 480, 481, 482, 483, 484, 489, 490], "wa": [4, 6, 84, 131, 132, 503, 504, 508], "wai": [2, 6, 9, 345, 423, 502, 503, 504, 505, 506, 507], "wait": 6, "walk": [6, 504], "walkthrough": 2, "walsh": 180, "want": [1, 2, 6, 503, 504, 505, 507, 512], "warm": [2, 502], "warmup_init": 479, "watch": [6, 502], "wd": 484, "we": [0, 1, 2, 5, 6, 7, 118, 122, 131, 132, 173, 248, 249, 345, 362, 403, 411, 482, 484, 499, 501, 502, 503, 504, 505, 507, 508, 512], "weight": [0, 5, 104, 105, 106, 107, 108, 109, 110, 150, 152, 330, 345, 388, 392, 403, 404, 444, 446, 475, 479, 482, 484, 486, 490, 505, 508], "weight_decai": [479, 482, 484, 490], "weight_fil": 6, "weights_fp16": 508, "well": [6, 345, 386, 398, 401, 503, 508], "wen": 6, "went": 6, "were": [6, 512], "wet": 6, "what": [2, 6, 330, 507], "whatsoev": 6, "whc": 359, "when": [0, 1, 2, 6, 8, 9, 99, 107, 118, 133, 197, 198, 200, 201, 202, 205, 206, 210, 211, 213, 282, 352, 353, 354, 355, 356, 357, 423, 427, 428, 444, 450, 456, 475, 477, 493, 499, 502, 503, 504, 512], "where": [0, 4, 7, 149, 185, 198, 248, 317, 321, 350, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 363, 365, 366, 370, 371, 372, 374, 385, 402, 405, 406, 420, 427, 428, 433, 434, 436, 447, 453, 459, 462, 464, 469, 486, 503, 505, 506], "wherea": 505, "whether": [146, 148, 151, 173, 200, 201, 209, 211, 249, 365, 371, 385, 401, 406, 444, 447, 453], "which": [0, 1, 2, 6, 7, 8, 9, 19, 38, 84, 85, 99, 107, 124, 127, 128, 129, 131, 132, 133, 142, 146, 148, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 168, 177, 183, 186, 187, 188, 189, 190, 192, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 213, 227, 230, 248, 252, 253, 267, 269, 271, 274, 275, 276, 277, 278, 295, 296, 304, 311, 316, 317, 320, 321, 326, 348, 349, 359, 360, 363, 378, 379, 381, 385, 409, 444, 446, 449, 453, 456, 472, 485, 486, 499, 502, 503, 504, 505, 506, 507, 508, 512, 513], "while": [2, 3, 6, 9, 269, 409, 508, 509], "whistl": 2, "who": 6, "whose": [149, 326, 327], "why": 6, "wi": 503, "wide": 508, "width": [348, 349, 350, 353, 354, 356, 357, 359, 360, 378, 379, 403, 404], "window": [9, 347, 348, 349, 377, 378, 379], "wipe": 9, "wire": 283, "wired_limit_mb": 283, "wise": [0, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 89, 90, 91, 92, 96, 112, 113, 134, 135, 139, 140, 141, 143, 145, 169, 170, 178, 179, 185, 194, 195, 196, 214, 215, 216, 217, 218, 219, 220, 221, 225, 233, 235, 237, 239, 245, 265, 266, 270, 273, 284, 285, 286, 287, 293, 294, 300, 306, 307, 351, 359, 360, 369, 380, 402, 413, 432, 439, 440, 442, 443, 458, 459, 461, 464, 465, 466, 467, 502], "wish": 9, "with_logit": 444, "within": [0, 3, 29, 185], "without": [1, 6, 8, 298, 401, 471, 501, 502, 503, 504, 507, 508, 509, 512], "wk": 6, "wl": 2, "wo": 6, "word": 0, "work": [2, 3, 6, 282, 336, 502, 503, 504, 505, 506, 507, 508], "workhors": 345, "world": [333, 503], "world2": 503, "world_ani": 503, "world_mpi": 503, "world_r": 503, "worri": [1, 508], "would": [2, 6, 423, 503, 504, 506, 508, 509, 512], "wq": 6, "wrap": [118, 345], "wrapper": [504, 507], "write": [0, 1, 6, 345, 509], "written": [2, 504], "wrong": 504, "wrt": 327, "wv": 6, "x": [0, 1, 2, 4, 5, 6, 7, 39, 85, 93, 118, 127, 128, 132, 133, 140, 145, 146, 147, 150, 151, 152, 173, 180, 181, 183, 205, 249, 253, 258, 272, 277, 284, 314, 315, 322, 330, 332, 345, 347, 348, 349, 350, 351, 361, 363, 364, 366, 370, 372, 373, 374, 377, 378, 379, 380, 381, 402, 405, 407, 413, 414, 420, 423, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 475, 477, 484, 502, 503, 504, 505, 506, 508, 509, 510, 512], "x1": 445, "x2": 445, "x86_64": 9, "x_1": [445, 453], "x_2": [445, 453], "x_cast": 2, "x_grad": 1, "x_i": [443, 465, 466], "x_j": [465, 466], "x_offset": 2, "x_ptr": 2, "x_shape": 1, "x_stride": 2, "x_t": [365, 371, 406], "x_view": 509, "xcode": 9, "xcodeproj": 3, "xcrun": 9, "xf": 371, "xg": 371, "xi": 371, "xn": 365, "xo": 371, "xor": 92, "xr": 365, "xy": [0, 227], "xz": 365, "x\u00b2": 509, "y": [0, 2, 4, 5, 6, 7, 39, 85, 118, 146, 147, 180, 183, 322, 345, 350, 359, 366, 370, 372, 374, 405, 448, 453, 456, 477, 480, 502, 503, 504, 505, 508, 509], "y_": [448, 452], "y_cast": 2, "y_hat": 345, "y_offset": 2, "y_ptr": 2, "y_stride": 2, "ye": 6, "year": 6, "yet": [6, 345, 475, 486, 505, 506, 508, 510], "yield": [6, 7, 499], "you": [2, 3, 4, 6, 7, 8, 9, 283, 345, 414, 422, 472, 499, 502, 503, 504, 505, 506, 507, 509, 511, 512], "your": [2, 6, 9, 475, 503, 505, 508], "z": [2, 85, 365, 502, 504, 508], "z_t": 365, "zeiler": 478, "zero": [0, 146, 149, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 199, 227, 268, 289, 313, 314, 315, 324, 345, 347, 348, 349, 358, 359, 360, 388, 424, 425, 426, 427, 428, 429, 430, 431, 472, 477, 479, 504, 506], "zero_grad": 505, "zeros_lik": [0, 203], "zhang": 6, "zip": [6, 7], "zip_saf": 2}, "titles": ["Operations", "Custom Metal Kernels", "Custom Extensions in MLX", "Metal Debugger", "Using MLX in C++", "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.std", "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.async_eval", "mlx.core.atleast_1d", "mlx.core.atleast_2d", "mlx.core.atleast_3d", "mlx.core.bitwise_and", "mlx.core.bitwise_invert", "mlx.core.bitwise_or", "mlx.core.bitwise_xor", "mlx.core.block_masked_mm", "mlx.core.broadcast_arrays", "mlx.core.broadcast_to", "mlx.core.ceil", "mlx.core.clear_cache", "mlx.core.clip", "mlx.core.compile", "mlx.core.concatenate", "mlx.core.conj", "mlx.core.conjugate", "mlx.core.contiguous", "mlx.core.conv1d", "mlx.core.conv2d", "mlx.core.conv3d", "mlx.core.conv_general", "mlx.core.conv_transpose1d", "mlx.core.conv_transpose2d", "mlx.core.conv_transpose3d", "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.distributed.recv", "mlx.core.distributed.recv_like", "mlx.core.distributed.send", "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.export_function", "mlx.core.export_to_dot", "mlx.core.exporter", "mlx.core.eye", "mlx.core.fast.layer_norm", "mlx.core.fast.metal_kernel", "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.finfo", "mlx.core.flatten", "mlx.core.floor", "mlx.core.floor_divide", "mlx.core.full", "mlx.core.gather_mm", "mlx.core.gather_qmm", "mlx.core.get_active_memory", "mlx.core.get_cache_memory", "mlx.core.get_peak_memory", "mlx.core.grad", "mlx.core.greater", "mlx.core.greater_equal", "mlx.core.hadamard_transform", "mlx.core.identity", "mlx.core.imag", "mlx.core.import_function", "mlx.core.inner", "mlx.core.isclose", "mlx.core.isfinite", "mlx.core.isinf", "mlx.core.isnan", "mlx.core.isneginf", "mlx.core.isposinf", "mlx.core.issubdtype", "mlx.core.jvp", "mlx.core.kron", "mlx.core.left_shift", "mlx.core.less", "mlx.core.less_equal", "mlx.core.linalg.cholesky", "mlx.core.linalg.cholesky_inv", "mlx.core.linalg.cross", "mlx.core.linalg.eigh", "mlx.core.linalg.eigvalsh", "mlx.core.linalg.inv", "mlx.core.linalg.lu", "mlx.core.linalg.lu_factor", "mlx.core.linalg.norm", "mlx.core.linalg.pinv", "mlx.core.linalg.qr", "mlx.core.linalg.solve", "mlx.core.linalg.solve_triangular", "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.device_info", "mlx.core.metal.is_available", "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.put_along_axis", "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.permutation", "mlx.core.random.randint", "mlx.core.random.seed", "mlx.core.random.split", "mlx.core.random.truncated_normal", "mlx.core.random.uniform", "mlx.core.real", "mlx.core.reciprocal", "mlx.core.remainder", "mlx.core.repeat", "mlx.core.reset_peak_memory", "mlx.core.reshape", "mlx.core.right_shift", "mlx.core.roll", "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_cache_limit", "mlx.core.set_default_device", "mlx.core.set_default_stream", "mlx.core.set_memory_limit", "mlx.core.set_wired_limit", "mlx.core.sigmoid", "mlx.core.sign", "mlx.core.sin", "mlx.core.sinh", "mlx.core.slice", "mlx.core.slice_update", "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.unflatten", "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.average_gradients", "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", "Export Functions", "Fast", "FFT", "Linear Algebra", "Memory Management", "Metal", "Neural Networks", "mlx.nn.ALiBi", "mlx.nn.AvgPool1d", "mlx.nn.AvgPool2d", "mlx.nn.AvgPool3d", "mlx.nn.BatchNorm", "mlx.nn.CELU", "mlx.nn.Conv1d", "mlx.nn.Conv2d", "mlx.nn.Conv3d", "mlx.nn.ConvTranspose1d", "mlx.nn.ConvTranspose2d", "mlx.nn.ConvTranspose3d", "mlx.nn.Dropout", "mlx.nn.Dropout2d", "mlx.nn.Dropout3d", "mlx.nn.ELU", "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.LogSigmoid", "mlx.nn.LogSoftmax", "mlx.nn.MaxPool1d", "mlx.nn.MaxPool2d", "mlx.nn.MaxPool3d", "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.Sigmoid", "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.celu", "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", "Exporting Functions", "Function Transforms", "Indexing Arrays", "Launching Distributed Programs", "Lazy Evaluation", "Conversion to NumPy and Other Frameworks", "Quick Start Guide", "Saving and Loading Arrays", "Unified Memory", "Using Streams"], "titleterms": {"A": 512, "In": 506, "The": 345, "ab": [13, 33], "adadelta": 478, "adafactor": 479, "adagrad": 480, "adam": 481, "adamax": 483, "adamw": 482, "add": 14, "addmm": 15, "algebra": 342, "alibi": 346, "all": [6, 16, 34, 503], "all_gath": 127, "all_sum": 128, "allclos": 17, "ani": [18, 35], "api": [8, 9], "appli": 381, "apply_gradi": 485, "apply_to_modul": 382, "arang": 19, "arcco": 20, "arccosh": 21, "arcsin": 22, "arcsinh": 23, "arctan": 24, "arctan2": 25, "arctanh": 26, "argmax": [27, 36], "argmin": [28, 37], "argpartit": 29, "argsort": 30, "arrai": [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, 335, 506, 511], "array_equ": 83, "as_strid": 84, "astyp": 38, "async_ev": 85, "atleast_1d": 86, "atleast_2d": 87, "atleast_3d": 88, "attent": 6, "automat": 505, "average_gradi": [325, 503], "avgpool1d": 347, "avgpool2d": 348, "avgpool3d": 349, "back": 2, "backend": 503, "basic": [502, 504, 510], "batchnorm": 350, "benchmark": 6, "bernoulli": 251, "binari": 9, "binary_cross_entropi": 444, "bind": 2, "bitwise_and": 89, "bitwise_invert": 90, "bitwise_or": 91, "bitwise_xor": 92, "block_masked_mm": 93, "broadcast_arrai": 94, "broadcast_to": 95, "build": [2, 9], "c": [4, 8, 9, 504], "categor": 252, "ceil": 96, "celu": [351, 432], "children": 383, "choleski": 197, "cholesky_inv": 198, "class": 345, "clear_cach": 97, "clip": 98, "clip_grad_norm": 328, "cmake": 2, "co": [41, 112], "code": [2, 6], "common": 496, "commun": [338, 503], "compil": [99, 502], "complex": 1, "comput": 508, "concaten": 100, "conj": [40, 101], "conjug": 102, "constant": 424, "contigu": 103, "conv1d": [104, 352], "conv2d": [105, 353], "conv3d": [106, 354], "conv_gener": 107, "conv_transpose1d": 108, "conv_transpose2d": 109, "conv_transpose3d": 110, "convers": 509, "convert": 6, "convolv": 111, "convtranspose1d": 355, "convtranspose2d": 356, "convtranspose3d": 357, "core": [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, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 334], "cosh": 113, "cosine_decai": 491, "cosine_similarity_loss": 445, "cpu": 2, "cross": 199, "cross_entropi": 446, "cummax": [42, 114], "cummin": [43, 115], "cumprod": [44, 116], "cumsum": [45, 117], "custom": [1, 2], "custom_funct": 118, "data": 336, "debug": 502, "debugg": 3, "default_devic": 119, "default_stream": 120, "defin": 503, "degre": 121, "dequant": 122, "devic": [10, 337], "device_info": 228, "diag": [46, 123], "diagon": [47, 124], "differ": 506, "differenti": 505, "disable_compil": 125, "distribut": [126, 127, 128, 129, 130, 131, 132, 133, 338, 503, 507], "divid": 134, "divmod": 135, "download": [2, 6], "dropout": 358, "dropout2d": 359, "dropout3d": 360, "dtype": [11, 48], "dtypecategori": 12, "eigh": 200, "eigvalsh": 201, "einsum": 136, "einsum_path": 137, "elu": [361, 433], "embed": 362, "enable_compil": 138, "encod": 6, "end": 2, "equal": 139, "erf": 140, "erfinv": 141, "eval": [142, 384], "evalu": 508, "exampl": [1, 2, 8, 502, 503, 504, 512], "exp": [49, 143], "expand_dim": 144, "expm1": 145, "exponential_decai": 492, "export": [148, 339, 504], "export_funct": 146, "export_to_dot": 147, "extens": 2, "ey": 149, "fast": [150, 151, 152, 153, 154, 340], "fft": [155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 341], "fft2": 156, "fftn": 157, "filter_and_map": 385, "finfo": 167, "flatten": [50, 168], "floor": 169, "floor_divid": 170, "format": 511, "found": 9, "framework": 509, "freez": 386, "from": [9, 506], "full": [6, 171], "function": [339, 471, 474, 502, 504, 505, 510], "further": 8, "gather_mm": 172, "gather_qmm": 173, "gaussian_nll_loss": 447, "gelu": [363, 434], "gelu_approx": 435, "gelu_fast_approx": 436, "gener": 6, "get": 503, "get_active_memori": 174, "get_cache_memori": 175, "get_peak_memori": 176, "glorot_norm": 425, "glorot_uniform": 426, "glu": [364, 437], "gpu": 2, "grad": [177, 345], "graph": [502, 508, 510], "greater": 178, "greater_equ": 179, "grid": 1, "group": 126, "groupnorm": 366, "gru": 365, "guid": 510, "gumbel": 253, "hadamard_transform": 180, "hard_shrink": 438, "hard_tanh": 439, "hardshrink": 367, "hardswish": [369, 440], "hardtanh": 368, "he_norm": 427, "he_uniform": 428, "hinge_loss": 448, "host": [503, 507], "huber_loss": 449, "ident": [181, 429], "ifft": 158, "ifft2": 159, "ifftn": 160, "imag": 182, "implement": [2, 6], "import": 504, "import_funct": 183, "index": 506, "infer": 6, "init": [129, 424, 425, 426, 427, 428, 429, 430, 431, 486], "initi": 472, "inner": 184, "inspect": 345, "instal": [8, 9, 503], "instancenorm": 370, "introduc": 2, "inv": 202, "irfft": 161, "irfft2": 162, "irfftn": 163, "is_avail": [130, 229], "isclos": 185, "isfinit": 186, "isinf": 187, "isnan": 188, "isneginf": 189, "isposinf": 190, "issubdtyp": 191, "item": 51, "items": 52, "jax": 509, "join_schedul": 493, "jvp": 192, "kei": 254, "kernel": 1, "kl_div_loss": 450, "kron": 193, "l1_loss": 451, "laplac": 255, "launch": 507, "layer": [6, 7, 473], "layer_norm": 150, "layernorm": 372, "lazi": 508, "leaf_modul": 387, "leaky_relu": 441, "leakyrelu": 373, "left_shift": 194, "less": 195, "less_equ": 196, "linalg": [197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211], "linear": [5, 342, 374], "linear_schedul": 494, "linspac": 212, "lion": 484, "llm": 6, "load": [6, 213, 477, 511], "load_weight": 388, "log": [53, 214], "log10": [54, 215], "log1p": [55, 216], "log2": [56, 217], "log_cosh_loss": 452, "log_sigmoid": 442, "log_softmax": 443, "logaddexp": 218, "logical_and": 219, "logical_not": 220, "logical_or": 221, "logsigmoid": 375, "logsoftmax": 376, "logsumexp": [57, 222], "loss": [444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 474], "lstm": 371, "lu": 203, "lu_factor": 204, "manag": 343, "margin_ranking_loss": 453, "matmul": 223, "max": [58, 224], "maximum": 225, "maxpool1d": 377, "maxpool2d": 378, "maxpool3d": 379, "mean": [59, 226], "memori": [343, 512], "meshgrid": 227, "metal": [1, 3, 9, 228, 229, 230, 231, 344], "metal_kernel": 151, "min": [60, 232], "minim": 9, "minimum": 233, "mish": [380, 458], "mlx": [2, 4, 8, 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, 295, 296, 297, 298, 299, 300, 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, 328, 329, 330, 331, 332, 333, 334, 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, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495], "model": 6, "modul": [345, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 475, 504], "more": 504, "moveaxi": [61, 234], "mpi": [503, 507], "mse_loss": 454, "multi": 7, "multiheadattent": 401, "multipl": 504, "multipli": 235, "multivariate_norm": 256, "named_modul": 390, "nan_to_num": 236, "nbyte": 62, "ndim": 63, "neg": 237, "network": 345, "neural": 345, "new_stream": 238, "nll_loss": 455, "nn": [325, 326, 327, 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, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 503], "norm": 205, "normal": [257, 430], "not_equ": 239, "numpi": [506, 509], "ones": 240, "ones_lik": 241, "onli": 508, "oper": [0, 2, 476], "optim": [328, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497], "option": 9, "other": 509, "outer": 242, "packag": 4, "pad": 243, "paramet": [345, 391], "partit": 244, "perceptron": 7, "permut": 258, "pinv": 206, "place": 506, "power": 245, "prelu": [402, 459], "primit": 2, "prod": [64, 246], "program": [503, 507], "provid": 507, "pure": 502, "put": 6, "put_along_axi": 247, "python": [2, 8, 9], "pytorch": 509, "qr": 207, "quantiz": [248, 326], "quantized_matmul": 249, "quantizedembed": 403, "quantizedlinear": 404, "quick": [345, 510], "radian": 250, "randint": 259, "random": [251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 499], "read": 8, "real": 264, "reciproc": [65, 265], "recv": 131, "recv_lik": 132, "reduc": 503, "refer": 8, "regress": 5, "relu": [407, 460], "relu6": [408, 461], "remaind": 266, "remot": [503, 507], "repeat": 267, "requir": 9, "reset_peak_memori": 268, "reshap": [66, 269], "result": 2, "rfft": 164, "rfft2": 165, "rfftn": 166, "right_shift": 270, "ring": [503, 507], "rms_norm": 152, "rmsnorm": 405, "rmsprop": 489, "rnn": 406, "roll": 271, "rope": [153, 409], "round": [67, 272], "rsqrt": [68, 273], "run": 503, "sampl": 1, "save": [274, 477, 511], "save_gguf": 275, "save_safetensor": 276, "save_weight": 392, "savez": 277, "savez_compress": 278, "scaled_dot_product_attent": 154, "schedul": 498, "script": [2, 6], "seed": 260, "select": 503, "selu": [410, 462], "send": 133, "sequenti": 411, "serial": 511, "set": [503, 507], "set_cache_limit": 279, "set_default_devic": 280, "set_default_stream": 281, "set_dtyp": 393, "set_memory_limit": 282, "set_wired_limit": 283, "setuptool": 2, "sgd": 490, "shape": [1, 69], "shapeless": [502, 504], "shell": 9, "sigmoid": [284, 413, 463], "sign": 285, "silu": [412, 464], "simpl": [1, 512], "sin": [70, 286], "sinh": 287, "sinusoidalpositionalencod": 414, "size": [9, 71], "slice": 288, "slice_upd": 289, "smooth_l1_loss": 456, "softmax": [290, 415, 465], "softmin": [416, 466], "softplu": [417, 467], "softshrink": [418, 468], "softsign": 419, "solv": 208, "solve_triangular": 209, "sort": 291, "sourc": 9, "specif": 507, "specifi": 513, "speedup": 502, "split": [72, 261, 292], "sqrt": [73, 293], "squar": [74, 294], "squeez": [75, 295], "stack": 296, "start": [345, 503, 510], "start_captur": 230, "state": [394, 487], "std": [76, 297], "step": [420, 469], "step_decai": 495, "stop_captur": 231, "stop_gradi": 298, "stream": [299, 334, 337, 513], "stride": 1, "subtract": 300, "sum": [77, 301], "support": 336, "svd": 210, "swapax": [78, 302], "synchron": 303, "t": 32, "take": 304, "take_along_axi": 305, "tan": 306, "tanh": [307, 421, 470], "tensordot": 308, "tensorflow": 509, "thunderbolt": 503, "tile": 309, "togeth": 6, "tolist": 79, "topk": 310, "trace": [311, 504], "train": [395, 397, 502, 503], "trainable_paramet": 396, "transform": [2, 422, 500, 502, 504, 505, 508, 510], "transpos": [80, 312], "tree": 501, "tree_flatten": 329, "tree_map": 330, "tree_map_with_path": 331, "tree_reduc": 332, "tree_unflatten": 333, "tri": 313, "tri_inv": 211, "tril": 314, "triplet_loss": 457, "triu": 315, "troubleshoot": 9, "truncated_norm": 262, "tune": 503, "type": 336, "unflatten": 316, "unfreez": 398, "unifi": 512, "uniform": [263, 431], "up": [503, 507], "updat": [345, 399, 488, 506], "update_modul": 400, "upsampl": 423, "us": [1, 2, 4, 508, 513], "usag": [2, 8, 507], "util": [329, 330, 331, 332, 333, 501, 503], "valu": 345, "value_and_grad": [317, 327], "var": [81, 318], "variabl": 4, "vector": 505, "view": [82, 319], "vjp": [1, 320], "vmap": 321, "weight": 6, "what": 508, "when": 508, "where": 322, "why": 508, "workflow": 3, "x86": 9, "xcode": 3, "you": 508, "zero": 323, "zeros_lik": 324}}) \ No newline at end of file diff --git a/docs/build/html/usage/compile.html b/docs/build/html/usage/compile.html index b17091896..9c35252c0 100644 --- a/docs/build/html/usage/compile.html +++ b/docs/build/html/usage/compile.html @@ -8,7 +8,7 @@ - Compilation — MLX 0.24.1 documentation + Compilation — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/distributed.html b/docs/build/html/usage/distributed.html index 927cb62e3..e1eefb505 100644 --- a/docs/build/html/usage/distributed.html +++ b/docs/build/html/usage/distributed.html @@ -8,7 +8,7 @@ - Distributed Communication — MLX 0.24.1 documentation + Distributed Communication — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/export.html b/docs/build/html/usage/export.html index af131d17e..ce8b365db 100644 --- a/docs/build/html/usage/export.html +++ b/docs/build/html/usage/export.html @@ -8,7 +8,7 @@ - Exporting Functions — MLX 0.24.1 documentation + Exporting Functions — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/function_transforms.html b/docs/build/html/usage/function_transforms.html index bb71c125a..16b0ead4b 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.24.1 documentation + Function Transforms — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/indexing.html b/docs/build/html/usage/indexing.html index 8008b02cb..7df919fac 100644 --- a/docs/build/html/usage/indexing.html +++ b/docs/build/html/usage/indexing.html @@ -8,7 +8,7 @@ - Indexing Arrays — MLX 0.24.1 documentation + Indexing Arrays — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/launching_distributed.html b/docs/build/html/usage/launching_distributed.html index 24ebdc779..653c7e4fd 100644 --- a/docs/build/html/usage/launching_distributed.html +++ b/docs/build/html/usage/launching_distributed.html @@ -8,7 +8,7 @@ - Launching Distributed Programs — MLX 0.24.1 documentation + Launching Distributed Programs — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/lazy_evaluation.html b/docs/build/html/usage/lazy_evaluation.html index b9eb05630..97eae9602 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.24.1 documentation + Lazy Evaluation — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/numpy.html b/docs/build/html/usage/numpy.html index dab13c570..d65cc5ccb 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.24.1 documentation + Conversion to NumPy and Other Frameworks — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/quick_start.html b/docs/build/html/usage/quick_start.html index dcdbd7398..553c4f0d5 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.24.1 documentation + Quick Start Guide — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/saving_and_loading.html b/docs/build/html/usage/saving_and_loading.html index a37e90b5b..46d87ca51 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.24.1 documentation + Saving and Loading Arrays — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/unified_memory.html b/docs/build/html/usage/unified_memory.html index 779e6f6c0..40db8f794 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.24.1 documentation + Unified Memory — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms
      diff --git a/docs/build/html/usage/using_streams.html b/docs/build/html/usage/using_streams.html index 13ce3c154..18aa80e94 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.24.1 documentation + Using Streams — MLX 0.24.2 documentation @@ -36,7 +36,7 @@ - + @@ -137,8 +137,8 @@ - MLX 0.24.1 documentation - Home - + MLX 0.24.2 documentation - Home + @@ -291,10 +291,12 @@
    • mlx.core.bitwise_or
    • mlx.core.bitwise_xor
    • mlx.core.block_masked_mm
      • +
    • mlx.core.broadcast_arrays
    • mlx.core.broadcast_to
    • mlx.core.ceil
    • mlx.core.clip
    • mlx.core.concatenate
      • +
    • mlx.core.contiguous
    • mlx.core.conj
    • mlx.core.conjugate
    • mlx.core.convolve
      • @@ -453,6 +455,7 @@
    • Transforms